dft_tools/test/triqs/gfs/block.cpp

#define TRIQS_ARRAYS_ENFORCE_BOUNDCHECK
#include <triqs/gfs.hpp> 
using namespace triqs::gfs;
using namespace triqs;
#define TEST(X) std::cout << BOOST_PP_STRINGIZE((X)) << " ---> "<< (X) <<std::endl<<std::endl;

int main() {
 try {
  double beta =1;
  auto G1 = gf<imfreq> ({beta, Fermion}, {2,2});
  auto G2 = G1;
  auto G3 = G2;

  // construct some block functions
  auto B0 = block_gf<imfreq> (3); 
  auto B1 = make_block_gf<imfreq> (3, G1); 
  auto B2 = make_block_gf<imfreq> ({G1,G1,G1});
  auto B3 = make_block_gf<imfreq> ({"a","b","c"}, {G1,G1,G1}); 
  auto B4 = block_gf<imfreq> (1); 

  // test hdf5 
  {
    H5::H5File file("ess_gf.h5", H5F_ACC_TRUNC );
    h5_write(file, "B3", B3);
  }

  {
    H5::H5File file("ess_gf.h5", H5F_ACC_RDONLY);
    std::cout  << "B4 mesh" << B4.mesh().size()<<std::endl;
    h5_read(file, "B3", B4);
    std::cout  << "B4 mesh" << B4.mesh().size()<<std::endl;
  }

  B1[0][0] = 98;
  //not implemented yet
  //B3["a"][0] = 98;

  auto View =  make_block_gf_view(G1,G2,G3);

  std::cout  << "Number of blocks " << View.mesh().size()<<std::endl ;
  auto g0 = View[0];
  auto g0v = View[0]();

  auto Gv = g0();

  Gv[0] = 20;
  TEST( G1( 0) ) ;
  Gv[0] = 0;

  g0v[0] = 3.2;
  TEST( G1( 0) ) ;

  // Operation
  g0[0] = 3.2;
  TEST( View[0](0) ) ;
  View = View/2; 
  TEST( View[0](0) ) ;

  // try the loop over the block.
  for (auto g : View) { g[0] = 20;}

 }
TRIQS_CATCH_AND_ABORT;
}