mirror of
https://github.com/triqs/dft_tools
synced 2024-11-01 03:33:50 +01:00
65 lines
1.4 KiB
C++
65 lines
1.4 KiB
C++
#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;
|
|
}
|