SparseMatrix_CCS.h

Go to the documentation of this file.

/* Copyright (C) 2014  Ward Poelmans

   This file is part of Hubbard-GPU.

   Hubbard-GPU is free software: you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation, either version 3 of the License, or
   (at your option) any later version.

   Hubbard-GPU is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with Hubbard-GPU.  If not, see <http://www.gnu.org/licenses/>.
   */
#ifndef SPARSEMATRIX_CCS_H
#define SPARSEMATRIX_CCS_H

#include <iostream>
#include <vector>

class matrix;

using std::ostream;

class SparseMatrix_CCS
{
   friend ostream &operator<<(ostream &output,SparseMatrix_CCS &matrix_p);

   public:

      //constructor
      SparseMatrix_CCS(unsigned int n, unsigned int m);

      //copy constructor
      SparseMatrix_CCS(const SparseMatrix_CCS &);

      //destructor
      virtual ~SparseMatrix_CCS();

      //overload equality operator
      SparseMatrix_CCS &operator=(const SparseMatrix_CCS &);

      //easy to change the numbers
   //   double &operator()(unsigned int i,unsigned int j);

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

      unsigned int gn() const;

      unsigned int gm() const;

      void ConvertFromMatrix(const matrix &dense);

      void ConvertToMatrix(matrix &dense) const;

      void PrintRaw() const;

      void PushToCol(unsigned int j, double value);

      void NewCol();

      void mvprod(const matrix &, matrix &) const;

      unsigned int NumOfElInCol(unsigned int idx) const;

      double GetElementInCol(unsigned int col_index, unsigned int element_index) const;

      unsigned int GetElementRowIndexInCol(unsigned int col_index, unsigned int element_index) const;

      void prod(const matrix &, const SparseMatrix_CCS &);

   private:

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

      unsigned int n;
      unsigned int m;
};

#endif /* SPARSEMATRIX_CCS_H */

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