3
0
mirror of https://github.com/triqs/dft_tools synced 2024-12-27 06:43:40 +01:00
dft_tools/test/triqs/arrays/views3.cpp
2013-07-23 08:13:52 +02:00

91 lines
2.6 KiB
C++

/*******************************************************************************
*
* TRIQS: a Toolbox for Research in Interacting Quantum Systems
*
* Copyright (C) 2011 by O. Parcollet
*
* TRIQS is free software: you can redistribute it and/or modify it under the
* terms of the GNU General Public License as published by the Free Software
* Foundation, either version 3 of the License, or (at your option) any later
* version.
*
* TRIQS is distributed in the hope that it will be useful, but WITHOUT ANY
* WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
* FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
* details.
*
* You should have received a copy of the GNU General Public License along with
* TRIQS. If not, see <http://www.gnu.org/licenses/>.
*
******************************************************************************/
#include "./common.hpp"
#define ARRAY_DEBUG_SLICE
#include "./src/array.hpp"
#include <iostream>
using std::cout; using std::endl;
using namespace triqs::arrays;
///
template<typename IM, typename A>
void print_in_indexmap_order (std::ostream & out, IM const & im, A const & a) {
out<<"[";
//for (typename IM::iterator it(im); it; ++it) out<<a[it.indices()]<<" ";
for (typename IM::iterator it(im); it; ++it) {
auto i = it.indices();
out<< a(i[0],i[1],i[2])<<" ";
}
out<<"]";
}
template<class AA>
void f(AA & A) {
for (int i =0; i<2; ++i)
for (int j=0; j<3; ++j)
for (int k=0; k<4; ++k)
A(i,j,k) = 100*(i+1)+ 10*(j+1) + (k+1);
print_in_indexmap_order(std::cout,A.indexmap(),A);std::cout<<std::endl;
std::cout<< A(0,range(), range())<<std::endl;
std::cout<< A(1,range(), range())<<std::endl;
std::cout<< A(range(), 0, range())<<std::endl;
std::cout<< A(range(), range(),1)<<std::endl;
std::cout<< A(range(), 0,1)<<std::endl;
std::cout<<"-------------------------------------------------- "<<std::endl;
}
int main(int argc, char **argv) {
try {
//array<long, 3, Option::options<Option::memory_order <2,1,0> > > A0 (2,3,4);
//array<long, 3, Option::options<Option::memory_order <0,1,2> > > A1 (2,3,4);
//array<long, 3, Option::options<Option::memory_order <1,0,2> > > A2 (2,3,4);
// permutation in triqs::arrays
array<long, 3, 0, permutation(2,1,0)> A0 (2,3,4);
array<long, 3, 0, permutation(0,1,2)> A1 (2,3,4);
array<long, 3, 0, permutation(1,0,2)> A2 (2,3,4);
array<long, 3> A3 (2,3,4);
array<long, 3, TRAVERSAL_ORDER_FORTRAN> A4 (2,3,4,FORTRAN_LAYOUT);
f(A0);
f(A1);
f(A2);
f(A3);
f(A4);
}
catch (const char * ERR) {
std::cout<<"Error is "<<ERR<<std::endl;
}
return 0;
}