fei_CSRMat.hpp

#ifndef _fei_CSRMat_hpp_
#define _fei_CSRMat_hpp_

/*--------------------------------------------------------------------*/
/*    Copyright 2005 Sandia Corporation.                              */
/*    Under the terms of Contract DE-AC04-94AL85000, there is a       */
/*    non-exclusive license for use of this work by or on behalf      */
/*    of the U.S. Government.  Export of this program may require     */
/*    a license from the United States Government.                    */
/*--------------------------------------------------------------------*/

#include "fei_macros.hpp"
#include "fei_FillableMat.hpp"
#include "fei_SparseRowGraph.hpp"
#include "fei_CSVec.hpp"

namespace fei {

class CSRMat {
 public:
  CSRMat();
  CSRMat(const FillableMat& fmat);
  virtual ~CSRMat();

  SparseRowGraph& getGraph() {return srg_;}
  const SparseRowGraph& getGraph() const {return srg_;}
 
  std::vector<double>& getPackedCoefs() {return packedcoefs_;}
  const std::vector<double>& getPackedCoefs() const {return packedcoefs_;}

  unsigned getNumRows() const {return srg_.rowNumbers.size();}

  CSRMat& operator=(const FillableMat& src);

  CSRMat& operator+=(const CSRMat& src);

  bool operator==(const CSRMat& rhs) const;

  bool operator!=(const CSRMat& rhs) const;

 private:
  SparseRowGraph srg_;
  std::vector<double> packedcoefs_;
};//class CSRMat

void multiply_CSRMat_CSVec(const CSRMat& A, const CSVec& x, CSVec& y);

void multiply_trans_CSRMat_CSVec(const CSRMat& A, const CSVec& x, CSVec& y);

void multiply_CSRMat_CSRMat(const CSRMat& A, const CSRMat& B, CSRMat& C,
                            bool storeResultZeros=false);

void multiply_trans_CSRMat_CSRMat(const CSRMat& A, const CSRMat& B, CSRMat& C,
                                  bool storeResultZeros=false);

void add_CSRMat_to_FillableMat(const CSRMat& csrm, FillableMat& fm);

}//namespace fei

#endif