KVFlowTensor.h

#ifndef __KVFLOWTENSOR_H
#define __KVFLOWTENSOR_H
 
#include "KVVarGlob.h"
#include "TMatrixDSym.h"
#include "TVectorT.h"
 
class KVFlowTensor: public KVVarGlob {
 
    TMatrixDSym fTensor;
   enum {
      kONE,
      kNRKE,
      kRKE
   } weight;
    mutable TVectorD fEVal;
    mutable TVector3 fEVec[3];
   Double_t f(int i) const
   {
      return fEVal[i - 1];
   }
   const TVector3& e(int i) const
   {
      return fEVec[i - 1];
   }
    TRotation fAziReacPlane;
    TRotation fFlowReacPlane;
    Double_t fSqueezeAngle;
    Double_t fSqOutRatio;
    Double_t fSphericity;
    Double_t fCoplanarity;
    Int_t fNParts;
   Bool_t fCalculated;
   Double_t sum_val_prop;
 
protected:
   Double_t getvalue_int(Int_t) const override;
 
public:
   enum {
      kFlowAngle,
      kSphericity,
      kCoplanarity,
      kNumberParts,
      kKinFlowRatio13,
      kKinFlowRatio23,
      kPhiReacPlane,
      kSqueezeAngle,
      kSqueezeRatio
   };
 
   KVFlowTensor();
   KVFlowTensor(const Char_t* nom);
   KVFlowTensor(const KVFlowTensor& a);
 
   virtual ~KVFlowTensor(void);
 
   void Copy(TObject& obj) const override;
 
   KVFlowTensor& operator=(const KVFlowTensor& a);
 
   void Init() override;
   void Reset() override;
   void Calculate() override;
 
private:
   void init_KVFlowTensor();
public:
   void fill(const KVNucleus* n) override;
   const TRotation& GetAziReacPlaneRotation() const;
   const TRotation& GetFlowReacPlaneRotation() const;
 
   KVFlowTensor& operator+=(const KVFlowTensor& other)
   {
 
      fTensor += other.fTensor;
      fNParts += other.fNParts;
      return (*this);
   }
   Double_t GetNormalisedEigenValue(Int_t i)
   {
 
      return f(i) / sum_val_prop;
   }
 
   void Print(Option_t* = "") const override;
 
   Bool_t HasNaNElements() const
   {
      for (int i = 0; i < 3; ++i) {
         for (int j = 0; j < 3; ++j) {
            if (TMath::IsNaN(fTensor(i, j))) return kTRUE;
         }
      }
      return kFALSE;
   }
   Char_t GetValueType(Int_t i) const override
   {
      if (i == 3) return 'I';
      return 'D';
   }
 
   ClassDefOverride(KVFlowTensor, 1) //Kinetic energy flow tensor of Gyulassy et al
};
 
 
#endif