KVReactionPlaneEstimator.h

#ifndef __KVREACTIONPLANEESTIMATOR_H
#define __KVREACTIONPLANEESTIMATOR_H
 
#include "KVVGVectorSum.h"
 
 
class KVReactionPlaneEstimator : public KVVGVectorSum {
   double fProjCMRapidity{0};// rapidity of projectile in CM
 
   void init();
   Double_t getvalue_int(Int_t) const;
   std::function<double(const KVNucleus*)> weight_function = [&](const KVNucleus* n)
   {
      auto rapidity = n->Rapidity() / fProjCMRapidity;
      if (rapidity > 0.3) return 1.0;
      else if (rapidity < -0.3) return -1.0;
      return 0.0;
   };
   void fill(const KVNucleus* n)
   {
      Add(weight_function(n) * n->GetTransverseMomentum());
   }
public:
   KVReactionPlaneEstimator() : KVVGVectorSum()
   {
      init();
   }
   KVReactionPlaneEstimator(const Char_t* name) : KVVGVectorSum(name)
   {
      init();
   }
 
   virtual ~KVReactionPlaneEstimator() {}
 
   void SetNormalization(Double_t r)
   {
      fProjCMRapidity = r;
   }
   TVector3 GetQ() const
   {
      return KVVGVectorSum::GetSumObject();
   }
   TVector3 GetQForParticle(const KVNucleus* n);
   static TF1* GetDeltaPhiRFitFunction();
 
   void Init()
   {
 
      if (fProjCMRapidity == 0)
         KVError::Error(this, "Init", "Projectile CM rapidity not set for this variable: %s. Use method SetNormalization() before calculating.",
               GetName());
   }
 
   ClassDef(KVReactionPlaneEstimator, 1) //Calculates the transverse momentum Q vector of Danielewicz & Odyniec
};
 
#endif