mirror of
				https://github.com/paboyle/Grid.git
				synced 2025-10-30 19:44:32 +00:00 
			
		
		
		
	Examples of how to access Grid Tensors
This commit is contained in:
		| @@ -35,7 +35,7 @@ | |||||||
| #include <Hadrons/ModuleFactory.hpp> | #include <Hadrons/ModuleFactory.hpp> | ||||||
|  |  | ||||||
| /****************************************************************************** | /****************************************************************************** | ||||||
|  Needed to make sure envCreate() (see Hadrons) works with specialisations |  Needed to make sure envCreate() (see Hadrons) work with specialisations | ||||||
|  with more than one parameter, eg obj<T1 COMMA T2> |  with more than one parameter, eg obj<T1 COMMA T2> | ||||||
|  I imagine this exists already? |  I imagine this exists already? | ||||||
|  ******************************************************************************/ |  ******************************************************************************/ | ||||||
| @@ -191,8 +191,8 @@ public: | |||||||
|   OperatorFunction<Field>   & _poly; |   OperatorFunction<Field>   & _poly; | ||||||
|   LinearOperatorBase<Field> &_Linop; |   LinearOperatorBase<Field> &_Linop; | ||||||
|    |    | ||||||
|   LinOpPeardonNablaHerm(OperatorFunction<Field> & poly,LinearOperatorBase<Field>& linop) : _poly(poly), _Linop(linop) { |   LinOpPeardonNablaHerm(OperatorFunction<Field> & poly,LinearOperatorBase<Field>& linop) | ||||||
|   } |   : _poly{poly}, _Linop{linop} {} | ||||||
|    |    | ||||||
|   void operator()(const Field& in, Field& out) { |   void operator()(const Field& in, Field& out) { | ||||||
|     _poly(_Linop,in,out); |     _poly(_Linop,in,out); | ||||||
|   | |||||||
| @@ -27,6 +27,7 @@ | |||||||
|  *************************************************************************************/ |  *************************************************************************************/ | ||||||
| /*  END LEGAL */ | /*  END LEGAL */ | ||||||
|  |  | ||||||
|  | #include <typeinfo> | ||||||
| #include <Hadrons/Application.hpp> | #include <Hadrons/Application.hpp> | ||||||
| #include <Hadrons/Modules.hpp> | #include <Hadrons/Modules.hpp> | ||||||
|  |  | ||||||
| @@ -294,6 +295,66 @@ void DebugShowTensor(MyTensor &x, const char * n) | |||||||
|   std::cout << std::endl; |   std::cout << std::endl; | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // Test whether typedef and underlying types are the same | ||||||
|  |  | ||||||
|  | void DebugTestTypeEqualities(void) | ||||||
|  | { | ||||||
|  |   Real    r1; | ||||||
|  |   RealD   r2; | ||||||
|  |   double  r3; | ||||||
|  |   const std::type_info &tr1{typeid(r1)}; | ||||||
|  |   const std::type_info &tr2{typeid(r2)}; | ||||||
|  |   const std::type_info &tr3{typeid(r3)}; | ||||||
|  |   if( tr1 == tr2 && tr2 == tr3 ) | ||||||
|  |     std::cout << "r1, r2 and r3 are the same type" << std::endl; | ||||||
|  |   else | ||||||
|  |     std::cout << "r1, r2 and r3 are different types" << std::endl; | ||||||
|  |   std::cout << "r1 is a " << tr1.name() << std::endl; | ||||||
|  |   std::cout << "r2 is a " << tr2.name() << std::endl; | ||||||
|  |   std::cout << "r3 is a " << tr3.name() << std::endl; | ||||||
|  |    | ||||||
|  |   // These are the same | ||||||
|  |   Complex             c1; | ||||||
|  |   std::complex<Real>  c2; | ||||||
|  |   const std::type_info &tc1{typeid(c1)}; | ||||||
|  |   const std::type_info &tc2{typeid(c2)}; | ||||||
|  |   const std::type_info &tc3{typeid(SpinVector::scalar_type)}; | ||||||
|  |   if( tc1 == tc2 && tc2 == tc3) | ||||||
|  |     std::cout << "c1, c2 and SpinVector::scalar_type are the same type" << std::endl; | ||||||
|  |   else | ||||||
|  |     std::cout << "c1, c2 and SpinVector::scalar_type are different types" << std::endl; | ||||||
|  |   std::cout << "c1                      is a " << tc1.name() << std::endl; | ||||||
|  |   std::cout << "c2                      is a " << tc2.name() << std::endl; | ||||||
|  |   std::cout << "SpinVector::scalar_type is a " << tc3.name() << std::endl; | ||||||
|  |    | ||||||
|  |   // These are the same | ||||||
|  |   SpinVector                              s1; | ||||||
|  |   iSpinVector<Complex >                   s2; | ||||||
|  |   iScalar<iVector<iScalar<Complex>, Ns> > s3; | ||||||
|  |   const std::type_info &ts1{typeid(s1)}; | ||||||
|  |   const std::type_info &ts2{typeid(s2)}; | ||||||
|  |   const std::type_info &ts3{typeid(s3)}; | ||||||
|  |   if( ts1 == ts2 && ts2 == ts3 ) | ||||||
|  |     std::cout << "s1, s2 and s3 are the same type" << std::endl; | ||||||
|  |   else | ||||||
|  |     std::cout << "s1, s2 and s3 are different types" << std::endl; | ||||||
|  |   std::cout << "s1 is a " << ts1.name() << std::endl; | ||||||
|  |   std::cout << "s2 is a " << ts2.name() << std::endl; | ||||||
|  |   std::cout << "s3 is a " << ts3.name() << std::endl; | ||||||
|  |    | ||||||
|  |   // These are the same | ||||||
|  |   SpinColourVector            sc1; | ||||||
|  |   iSpinColourVector<Complex > sc2; | ||||||
|  |   const std::type_info &tsc1{typeid(sc1)}; | ||||||
|  |   const std::type_info &tsc2{typeid(sc2)}; | ||||||
|  |   if( tsc1 == tsc2 ) | ||||||
|  |     std::cout << "sc1 and sc2 are the same type" << std::endl; | ||||||
|  |   else | ||||||
|  |     std::cout << "sc1 and sc2 are different types" << std::endl; | ||||||
|  |   std::cout << "sc1 is a " << tsc1.name() << std::endl; | ||||||
|  |   std::cout << "sc2 is a " << tsc2.name() << std::endl; | ||||||
|  | } | ||||||
|  |  | ||||||
| bool DebugEigenTest() | bool DebugEigenTest() | ||||||
| { | { | ||||||
|   const char pszTestFileName[] = "test_tensor.bin"; |   const char pszTestFileName[] = "test_tensor.bin"; | ||||||
| @@ -311,10 +372,39 @@ bool DebugEigenTest() | |||||||
|   for( auto a : p.IndexNames ) |   for( auto a : p.IndexNames ) | ||||||
|     std::cout << a << std::endl; |     std::cout << a << std::endl; | ||||||
|   // Now see whether we can read a tensor back |   // Now see whether we can read a tensor back | ||||||
|   std::array<std::string,3> a2={"Alpha", "Gamma", "Delta"}; |   std::array<std::string,3> Names2={"Alpha", "Gamma", "Delta"}; | ||||||
|   MyTensor y(a2, 2,4,1); |   MyTensor y(Names2, 2,4,1); | ||||||
|   y.ReadBinary(pszTestFileName); |   y.ReadBinary(pszTestFileName); | ||||||
|   DebugShowTensor(y, "y"); |   DebugShowTensor(y, "y"); | ||||||
|  |  | ||||||
|  |   // Testing whether typedef produces the same type - yes it does | ||||||
|  |  | ||||||
|  |   DebugTestTypeEqualities(); | ||||||
|  |   std::cout << std::endl; | ||||||
|  |  | ||||||
|  |   // How to access members of SpinColourVector | ||||||
|  |   SpinColourVector  sc; | ||||||
|  |   for( int s = 0 ; s < Ns ; s++ ) { | ||||||
|  |     auto cv{sc()(s)}; | ||||||
|  |     iVector<Complex,Nc> c2{sc()(s)}; | ||||||
|  |     std::cout << " cv is a " << typeid(cv).name() << std::endl; | ||||||
|  |     std::cout << " c2 is a " << typeid(c2).name() << std::endl; | ||||||
|  |     for( int c = 0 ; c < Nc ; c++ ) { | ||||||
|  |       Complex & z{cv(c)}; | ||||||
|  |       std::cout << "  sc[spin=" << s << ", colour=" << c << "] = " << z << std::endl; | ||||||
|  |     } | ||||||
|  |   } | ||||||
|  |   // We could have removed the Lorentz index independently, but much easier to do as we do above | ||||||
|  |   iVector<iVector<Complex,Nc>,Ns> sc2{sc()}; | ||||||
|  |   std::cout << "sc() is a " << typeid(sc()).name() << std::endl; | ||||||
|  |   std::cout << "sc2  is a " << typeid(sc2 ).name() << std::endl; | ||||||
|  |  | ||||||
|  |   // Or you can access elements directly | ||||||
|  |   std::complex<Real> z = sc()(0)(0); | ||||||
|  |   std::cout << "z = " << z << std::endl; | ||||||
|  |   sc()(3)(2) = std::complex<Real>{3.141,-3.141}; | ||||||
|  |   std::cout << "sc()(3)(2) = " << sc()(3)(2) << std::endl; | ||||||
|  |  | ||||||
|   return true; |   return true; | ||||||
| } | } | ||||||
| #endif | #endif | ||||||
| @@ -325,7 +415,7 @@ int main(int argc, char *argv[]) | |||||||
|   // Debug only - test of Eigen::Tensor |   // Debug only - test of Eigen::Tensor | ||||||
|   std::cout << "sizeof(std::streamsize) = " << sizeof(std::streamsize) << std::endl; |   std::cout << "sizeof(std::streamsize) = " << sizeof(std::streamsize) << std::endl; | ||||||
|   std::cout << "sizeof(Eigen::Index) = " << sizeof(Eigen::Index) << std::endl; |   std::cout << "sizeof(Eigen::Index) = " << sizeof(Eigen::Index) << std::endl; | ||||||
|   //if( DebugEigenTest() ) return 0; |   if( DebugEigenTest() ) return 0; | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
|   // Decode command-line parameters. 1st one is which test to run |   // Decode command-line parameters. 1st one is which test to run | ||||||
|   | |||||||
		Reference in New Issue
	
	Block a user