Updates EvtGen Code

[u/mrichter/AliRoot.git] / TEvtGen / EvtGenBase / EvtTensor3C.hh

//--------------------------------------------------------------------------
//
// Environment:
//      This software is part of the EvtGen package developed jointly
//      for the BaBar and CLEO collaborations.  If you use all or part
//      of it, please give an appropriate acknowledgement.
//
// Copyright Information: See EvtGen/COPYRIGHT
//      Copyright (C) 1998      Caltech, UCSB
//
// Module: EvtGen/EvtTensor3C.hh
//
// Description: Class to handle complex 3 tensors
//
// Modification history:
//
//    RYD     September 14, 1997         Module created
//
//------------------------------------------------------------------------

#ifndef EVTTENSOR3C_HH
#define EVTTENSOR3C_HH

#include <iostream>
#include "EvtGenBase/EvtComplex.hh"

class EvtVector3C;
class EvtVector3R;

class EvtTensor3C;

namespace EvtGenFunctions {
  EvtTensor3C eps(const EvtVector3R& v);
  EvtTensor3C rotateEuler(const EvtTensor3C& v,
				 double phi,double theta,double ksi);
  EvtTensor3C directProd(const EvtVector3C& c1,const EvtVector3C& c2); 
  EvtTensor3C directProd(const EvtVector3C& c1,const EvtVector3R& c2); 
  EvtTensor3C directProd(const EvtVector3R& c1,const EvtVector3R& c2);
};

class EvtTensor3C {
  friend EvtTensor3C operator*(
	          const EvtComplex& c,const EvtTensor3C& t2);
  friend EvtTensor3C operator*(const double d,const EvtTensor3C& t2);
  friend EvtTensor3C operator*(
                  const EvtTensor3C& t2,const EvtComplex& c);
  friend EvtTensor3C operator*(const EvtTensor3C& t2,const double d);
  friend EvtTensor3C operator+(
                  const EvtTensor3C& t1,const EvtTensor3C& t2);
  friend EvtTensor3C operator-(
                  const EvtTensor3C& t1,const EvtTensor3C& t2);
  friend EvtTensor3C EvtGenFunctions::directProd(const EvtVector3C& c1,const EvtVector3C& c2); 
  friend EvtTensor3C EvtGenFunctions::directProd(const EvtVector3C& c1,const EvtVector3R& c2); 
  friend EvtTensor3C EvtGenFunctions::directProd(const EvtVector3R& c1,const EvtVector3R& c2); 
  friend EvtTensor3C conj(const EvtTensor3C& t2);
  //Contract the second index of two tensors result(i,j) = t1(i,k)t2(j,k)
  friend EvtTensor3C cont22(const EvtTensor3C& t1,const EvtTensor3C& t2); 
  //Contract the first index of two tensors result(i,j) = t1(k,i)t2(k,j)
  friend EvtTensor3C cont11(const EvtTensor3C& t1,const EvtTensor3C& t2);
  //Contract the last index of eps_{ijk} with w
  friend EvtTensor3C EvtGenFunctions::eps(const EvtVector3R& v);
  friend std::ostream& operator<<(std::ostream& c,const EvtTensor3C& v); 

public:
  EvtTensor3C();
  EvtTensor3C(const EvtTensor3C& t1 );
  EvtTensor3C(double d11, double d22, double d33);
  virtual ~EvtTensor3C();
  EvtTensor3C& operator=(const EvtTensor3C& t1);
  inline void set(int i,int j,const EvtComplex& c);
  inline const EvtComplex& get(int i, int j) const;
  inline EvtComplex trace() const;
  static const EvtTensor3C& id();
  void zero();
  void applyRotateEuler(double phi,double theta,double ksi);

  EvtTensor3C operator+=(const EvtTensor3C& t2);
  EvtTensor3C operator-=(const EvtTensor3C& t2);
  EvtTensor3C operator*=(const double d);
  EvtTensor3C operator*=(const EvtComplex& c);
  EvtTensor3C conj() const;
  EvtVector3C cont1(const EvtVector3C& v) const; 
  EvtVector3C cont2(const EvtVector3C& v) const; 
  EvtVector3C cont1(const EvtVector3R& v) const; 
  EvtVector3C cont2(const EvtVector3R& v) const; 
  
private:

  EvtComplex t[3][3];

};

inline EvtTensor3C operator*(const EvtComplex& c,const EvtTensor3C& t2){
  return EvtTensor3C(t2)*=c;
}

inline EvtTensor3C operator*(const double d,const EvtTensor3C& t2){
  return EvtTensor3C(t2)*=d;
}

inline EvtTensor3C operator*(const EvtTensor3C& t2,const EvtComplex& c){
  return EvtTensor3C(t2)*=c;
}

inline EvtTensor3C operator*(const EvtTensor3C& t2,const double d){
  return EvtTensor3C(t2)*=d;
}

inline EvtTensor3C operator+(const EvtTensor3C& t1,const EvtTensor3C& t2){
  return EvtTensor3C(t1)+=t2;
}

inline EvtTensor3C operator-(const EvtTensor3C& t1,const EvtTensor3C& t2){
  return EvtTensor3C(t1)-=t2;
}

inline void EvtTensor3C::set(int i,int j,const EvtComplex& c){
   t[i][j]=c;
}

inline const EvtComplex& EvtTensor3C::get(int i,int j) const{
   return t[i][j];
}

inline EvtComplex EvtTensor3C::trace() const{
   return t[0][0]+t[1][1]+t[2][2];
}

#endif