#include <rabit.h>
#include <vector>
#include <cstdlib>
#include <cstdio>
#include <cmath>

namespace rabit {
/*! \brief sparse matrix, CSR format */
struct SparseMat {
  // sparse matrix entry
  struct Entry {
    // feature index 
    unsigned findex;
    // feature value
    float fvalue;
  };
  // sparse vector
  struct Vector {
    const Entry *data;
    unsigned length;
    inline const Entry &operator[](size_t i) const {
      return data[i];
    }
  };
  inline Vector operator[](size_t i) const {
    Vector v;
    v.data = &data[0] + row_ptr[i];
    v.length = static_cast<unsigned>(row_ptr[i + 1]-row_ptr[i]);
    return v;
  }
  // load data from file
  inline void Load(const char *fname) {
    FILE *fi;
    if (!strcmp(fname, "stdin")) {
      fi = stdin;
    } else {
      fi = utils::FopenCheck(fname, "r");
    }
    row_ptr.clear();
    row_ptr.push_back(0);
    data.clear();    
    feat_dim = 0;
    unsigned num_feat;
    while (fscanf(fi, "%u", &num_feat) == 1) {
      Entry e;
      for (unsigned i = 0; i < num_feat; ++i) {
        utils::Check(fscanf(fi, "%u:%f", &e.findex, &e.fvalue) == 2,
                     "invalid format");
        data.push_back(e);
        feat_dim = std::max(e.findex, feat_dim);
      }
      row_ptr.push_back(data.size());
    }
    feat_dim += 1;
    // close the filed
    if (fi != stdin) fclose(fi);
  }
  inline size_t NumRow(void) const {
    return row_ptr.size() - 1;
  }
  // maximum feature dimension
  unsigned feat_dim;
  std::vector<size_t> row_ptr;
  std::vector<Entry> data;
};
// dense matrix
struct Matrix {
  inline void Init(size_t nrow, size_t ncol, float v = 0.0f) {
    this->nrow = nrow;
    this->ncol = ncol;
    data.resize(nrow * ncol);
    std::fill(data.begin(), data.end(), v);
  }
  inline float *operator[](size_t i) {
    return &data[0] + i * ncol;
  }
  inline const float *operator[](size_t i) const {
    return &data[0] + i * ncol;
  }
  inline void Print(const char *fname) {
    FILE *fo;
    if (!strcmp(fname, "stdout")) {
      fo = stdout;
    } else {
      fo = utils::FopenCheck(fname, "w");
    }
    fprintf(fo, "%lu %lu\n", nrow, ncol);
    for (size_t i = 0; i < data.size(); ++i) {
      fprintf(fo, "%g", data[i]);
      if ((i+1) % ncol == 0) {
        fprintf(fo, "\n");
      } else {
        fprintf(fo, " ");
      }
    }
    // close the filed
    if (fo != stdout) fclose(fo);
  }
  // number of data
  size_t nrow, ncol;
  std::vector<float> data;
};

/*!\brief computes a random number modulo the value */
inline int Random(int value) {
  return rand() % value;
}
} // namespace rabit