3
0
mirror of https://github.com/triqs/dft_tools synced 2024-12-27 06:43:40 +01:00
dft_tools/test/triqs/gf/block.cpp
2013-07-29 11:53:02 +02:00

82 lines
2.1 KiB
C++

//#define TRIQS_ARRAYS_ENFORCE_BOUNDCHECK
#include <triqs/gfs/imfreq.hpp>
#include <triqs/gfs/imtime.hpp>
#include <triqs/gfs/block.hpp>
namespace tql= triqs::clef;
namespace tqa= triqs::arrays;
using tqa::range;
using triqs::arrays::make_shape;
using triqs::gfs::gf;
using triqs::gfs::gf_view;
using triqs::gfs::block_index;
using triqs::gfs::Fermion;
using triqs::gfs::imfreq;
using triqs::gfs::imtime;
using triqs::gfs::make_gf;
using triqs::gfs::make_block_gf;
using triqs::gfs::make_gf_view;
#define TEST(X) std::cout << BOOST_PP_STRINGIZE((X)) << " ---> "<< (X) <<std::endl<<std::endl;
int main() {
double beta =1;
auto G1 = make_gf<imfreq> (beta, Fermion, make_shape(2,2));
auto G2 = make_gf<imfreq> (beta, Fermion, make_shape(2,2));
auto G3 = make_gf<imfreq> (beta, Fermion, make_shape(2,2));
//auto BBB = make_block_gf<imfreq> ({G1,G2,G2});
//auto BBB2 = make_gf<block_index, gf<imfreq>> ({G1,G2,G2});
std::vector<gf<imfreq> > V ;
V.push_back(G1); V.push_back(G2); V.push_back(G3);
std::vector<gf_view<imfreq> > Vv; // = { G1,G2,G3};
Vv.push_back(G1); Vv.push_back(G2); Vv.push_back(G3);
std::cout <<" Building gf_view of view"<< std::endl ;
auto GF_v = make_gf_view<block_index,gf<imfreq>> (Vv);
std::cout <<" Building gf_view of gf"<< std::endl ;
auto GF = make_gf_view<block_index,gf<imfreq>> (V); //{G1,G2,G3});
//auto GF = make_gf_view<block_index,gf<imfreq>> ( std::vector<gf_view<imfreq> > {G1,G2,G3});
std::cout << "Number of blocks " << GF.mesh().size()<<std::endl ;
auto g0 = GF[0];
auto g0v = GF_v[0]();
auto Gv = g0();
Gv.on_mesh(0) = 20;
TEST( Gv( 0) ) ;
TEST( G1( 0) ) ;
Gv.on_mesh(0) = 0;
g0v.on_mesh(0) = 3.2;
//g0v[0]= 3.2;
// Vv[0](0) = -2.1;
TEST( Gv( 0) ) ;
TEST( G1( 0) ) ;
//TEST( GF_v(0)( 0) ) ;
//TEST( GF_v[0]( 0) ) ;
// bug fixed for this
gf<block_index,gf<imfreq>> G9;
G9 = make_gf<block_index,gf<imfreq>> (2, make_gf<imfreq>(beta, Fermion, make_shape(2,2)));
// Operation
g0.on_mesh(0) = 3.2;
TEST( GF[0](0) ) ;
GF = GF/2;
TEST( GF[0](0) ) ;
//TEST( g0("3.2") ) ;
//TEST( GF(0)(0) ) ;
// try the loop over the block.
for (auto g : GF) { g.on_mesh(0) = 20;}
}