SparseMatrix_CRS.h

Go to the documentation of this file.

#ifndef SPARSEMATRIX_CRS_H
#define SPARSEMATRIX_CRS_H

#include <iostream>
#include <vector>

#include "helpers.h"

// dark magic to get the friend operator<< to work...
namespace helpers { class SparseMatrix_CRS; };
std::ostream &operator<<(std::ostream &output,helpers::SparseMatrix_CRS &matrix_p);

namespace helpers {

class SparseMatrix_CRS
{
   friend std::ostream &::operator<<(std::ostream &output,helpers::SparseMatrix_CRS &matrix_p);

   public:

      SparseMatrix_CRS(unsigned int n);

      virtual ~SparseMatrix_CRS() = default;

      //easy to access the numbers
      double operator()(unsigned int i,unsigned int j) const;

      unsigned int gn() const;

      void ConvertFromMatrix(const helpers::matrix &dense);

      void ConvertToMatrix(helpers::matrix &dense) const;

      void PrintRaw() const;

      void PushToRow(unsigned int j, double value);

      void PushToRowNext(unsigned int j, double value);

      void NewRow();

      void mvprod(const double *, double *) const;

      void mvprod(const double *, double *, double) const;

      void SetGuess(unsigned int);

      int WriteToFile(const char*,const char*,bool=false) const;

      int ReadFromFile(const char*,const char*);

      unsigned int NumOfElInRow(unsigned int idx) const;

      double GetElementInRow(unsigned int row_index, unsigned int element_index) const;

      unsigned int GetElementColIndexInRow(unsigned int row_index, unsigned int element_index) const;

      void AddList(std::vector< std::unique_ptr<SparseMatrix_CRS> > &);

   private:

      std::vector<double> data;
      std::vector<unsigned int> col;
      std::vector<unsigned int> row;

      unsigned int n;
};

}

#endif /* SPARSEMATRIX_CRS_H */

/* vim: set ts=3 sw=3 expandtab :*/