Hamiltonian.hpp

Go to the documentation of this file.

#ifndef HAMILTONIAN_HPP
#define HAMILTONIAN_HPP

#include <armadillo>
#include "EnumGamma.hpp"

template <class T>
class Hamiltonian
{

   friend class Solver;

private:
   int deg;

public:
   int N;
   T elements;
   Hamiltonian(int _N, int _deg) : deg{_deg}, N{_N}
   {
      elements.set_size(deg * N, deg * N);

      // [!] only for arma::mat
      if constexpr (std::is_same<T, arma::mat>::value)
         elements.fill(0);
   }

   void InsertBlock(Gamma gammai, int i, Gamma gammaj, int j, double value)
   {
      int ishift = static_cast<int>(gammai) % deg;
      int jshift = static_cast<int>(gammaj) % deg;
      this->elements(deg * i + ishift, deg * j + jshift) += value;
   }

   void Print()
   {
      this->elements.print();
   }
};

#endif