csr_reader.h

Go to the documentation of this file.

#ifndef CSR_READER_H
#define CSR_READER_H
 
#include <fstream>
 
#include "file_reader.h"
#include "sparse_matrix.h"
 
namespace ModuleIO
{
 
template <typename T>
class csrFileReader : public FileReader
{
  public:
    // Constructor
    csrFileReader(const std::string& filename);
 
    // read all matrices of all R coordinates
    void parseFile();
 
    // get number of R
    int getNumberOfR() const;
 
    // get sparse matrix of a specific R coordinate
    SparseMatrix<T> getMatrix(int Rx, int Ry, int Rz);
 
    // get matrix by using index
    SparseMatrix<T> getMatrix(int index) const;
 
    // get R coordinate using index
    std::vector<int> getRCoordinate(int index) const;
 
    // get step
    int getStep() const;
 
    // get matrix dimension
    int getMatrixDimension() const;
 
  private:
    std::vector<std::vector<int>> RCoordinates;
    std::vector<SparseMatrix<T>> sparse_matrices;
    int step;
    int matrixDimension;
    int numberOfR;
};
 
} // namespace ModuleIO
 
#endif // READ_CSR_H