From 949b6a7afae5791b28709ca1731b1bb3d5065400 Mon Sep 17 00:00:00 2001
From: neo <cossu@post.kek.jp>
Date: Thu, 4 Jun 2015 18:29:55 +0900
Subject: [PATCH] Added support for Ta to Lattice types

---
 lib/lattice/Lattice_reality.h | 11 +++++++++++
 lib/tensors/Tensor_Ta.h       | 17 +++++++++++------
 tests/Test_main.cc            | 13 +++++++------
 3 files changed, 29 insertions(+), 12 deletions(-)

diff --git a/lib/lattice/Lattice_reality.h b/lib/lattice/Lattice_reality.h
index 2f52ed22..4ad6fb7b 100644
--- a/lib/lattice/Lattice_reality.h
+++ b/lib/lattice/Lattice_reality.h
@@ -48,5 +48,16 @@ PARALLEL_FOR_LOOP
     }
 
 
+    template<class vobj> inline auto Ta(const Lattice<vobj> &z) -> Lattice<decltype(Ta(z._odata[0]))>
+    {
+      Lattice<decltype(Ta(z._odata[0]))> ret(z._grid);
+PARALLEL_FOR_LOOP
+        for(int ss=0;ss<z._grid->oSites();ss++){
+            ret._odata[ss] = Ta(z._odata[ss]);
+        }
+      return ret;
+    }
+
+
 }
 #endif
diff --git a/lib/tensors/Tensor_Ta.h b/lib/tensors/Tensor_Ta.h
index ef4b86e1..3a903364 100644
--- a/lib/tensors/Tensor_Ta.h
+++ b/lib/tensors/Tensor_Ta.h
@@ -5,10 +5,10 @@ namespace Grid {
   /////////////////////////////////////////////// 
   // Ta function for scalar, vector, matrix
   /////////////////////////////////////////////// 
-  /* inline ComplexF Ta( const ComplexF &arg){    return arg;} */
-  /* inline ComplexD Ta( const ComplexD &arg){    return arg;} */
-  /* inline RealF Ta( const RealF &arg){    return arg;} */
-  /* inline RealD Ta( const RealD &arg){    return arg;} */
+  inline ComplexF Ta( const ComplexF &arg){    return arg;}
+  inline ComplexD Ta( const ComplexD &arg){    return arg;}
+  inline RealF Ta( const RealF &arg){    return arg;}
+  inline RealD Ta( const RealD &arg){    return arg;}
 
 
   template<class vtype> inline iScalar<vtype> Ta(const iScalar<vtype>&r)
@@ -28,8 +28,13 @@ namespace Grid {
   template<class vtype,int N> inline iMatrix<vtype,N> Ta(const iMatrix<vtype,N> &arg)
     {
       iMatrix<vtype,N> ret(arg);
-      vtype factor = (1/(double)N);
-      ret = (ret - adj(arg))*0.5;
+      double factor = (1/(double)N);
+      for(int c1=0;c1<N;c1++){
+	for(int c2=0;c2<N;c2++){
+	  ret._internal[c1][c2]= (ret._internal[c1][c2] - adj(arg._internal[c2][c1]));
+	  ret._internal[c1][c2] *= 0.5;
+	}}
+      //ret = (ret - adj(arg))*0.5;
       ret -= trace(ret)*factor;
       return ret;
     }
diff --git a/tests/Test_main.cc b/tests/Test_main.cc
index 724e1563..cef2d848 100644
--- a/tests/Test_main.cc
+++ b/tests/Test_main.cc
@@ -123,7 +123,9 @@ int main (int argc, char ** argv)
     cMat = outerProduct(cVec,cVec);
     scalar = localInnerProduct(cVec,cVec);
 
- 
+    cMat = Ta(cMat);  //traceless antihermitian
+
+
     scalar += scalar;
     scalar -= scalar;
     scalar *= scalar;
@@ -214,13 +216,12 @@ int main (int argc, char ** argv)
     scm=transposeIndex<1>(scm);
     
    
-    random(SerialRNG, cm);
-    std::cout << cm << std::endl;
+    //random(SerialRNG, cm);
+    //std::cout << cm << std::endl;
 
     cm = Ta(cm);
-    //cm = adj(cm);
-    TComplex tracecm= trace(cm);      
-    std::cout << cm << "  "<< tracecm << std::endl;
+    //TComplex tracecm= trace(cm);      
+    //std::cout << cm << "  "<< tracecm << std::endl;
 
 
 //    Foo = Foo+scalar; // LatticeColourMatrix+Scalar