fopr_CRS.h

Go to the documentation of this file.

#ifndef FOPR_CRS_INCLUDED
#define FOPR_CRS_INCLUDED

#include <valarray>
#include <string>

#include "fopr.h"

#include "bridgeIO.h"
using Bridge::vout;


class Fopr_CRS : public Fopr
{
 private:
  int                   m_Nin, m_Nvol, m_Nex;
  int                   m_Nsize, m_Nnz;
  std::valarray<int>    m_rowidx_nz;
  std::valarray<int>    m_column_nz;
  std::valarray<double> m_elem_nz;
  std::string           m_mode;
  Fopr                  *m_fopr;

 public:

  Fopr_CRS(Fopr *fopr)
    : Fopr(), m_fopr(fopr)
  {
    set_matrix();
  }

  Fopr_CRS(std::string fname)
    : Fopr(), m_fopr(0)
  {
    set_matrix(fname);
  }

  void set_parameters(const Parameters&);

  void write_matrix(std::string);

  void set_config(Field *U)
  {
    if (m_fopr == 0) {
      vout.crucial(m_vl, "Fopr_CRS: fopr is not set.\n");
      abort();
    } else {
      m_fopr->set_config(U);
    }
  }

  void set_mode(std::string mode)
  {
    m_mode = mode;
  }

  std::string get_mode() const
  {
    return m_mode;
  }

  const Field mult(const Field& f)
  {
    Field v(f.nin(), f.nvol(), f.nex());

    mult(v, f);
    return v;
  }

  const Field mult_dag(const Field& f)
  {
    Field v(f.nin(), f.nvol(), f.nex());

    mult_dag(v, f);
    return v;
  }

  void mult(Field& v, const Field& f)
  {
    if (m_mode == "D") {
      D(v, f);
    } else if (m_mode == "DdagD") {
      DdagD(v, f);
    } else if (m_mode == "Ddag") {
      Ddag(v, f);
    } else {
      vout.crucial(m_vl, "Fopr_CRS: mode undefined.\n");
      abort();
    }
  }

  void mult_dag(Field& v, const Field& f)
  {
    if (m_mode == "D") {
      Ddag(v, f);
    } else if (m_mode == "DdagD") {
      DdagD(v, f);
    } else if (m_mode == "Ddag") {
      D(v, f);
    } else {
      vout.crucial(m_vl, "Fopr_CRS: mode undefined.\n");
      abort();
    }
  }

  void DdagD(Field&, const Field&);
  void D(Field&, const Field&);
  void Ddag(Field&, const Field&);
  void H(Field&, const Field&);

  int field_nvol() { return m_Nvol; }
  int field_nin() { return m_Nin; }
  int field_nex() { return m_Nex; }

 private:
  void set_matrix();
  void set_matrix(std::string);

  void set_matrix_1row(int&, std::valarray<int>&,
                       std::valarray<double>&, Field&);
};
#endif