Mega-commit by Alexander - added Event plane to analysis task, and other flow updates...

[u/mrichter/AliRoot.git] / PWGLF / FORWARD / analysis2 / AliSPDMCTrackDensity.h

diff --git a/PWGLF/FORWARD/analysis2/AliSPDMCTrackDensity.h b/PWGLF/FORWARD/analysis2/AliSPDMCTrackDensity.h
index a0e5f64da4ac6acd02b7fa18c6f1054d67288bc2..36af38438a5735548e8de6b876d1710056776948 100644 (file)
--- a/PWGLF/FORWARD/analysis2/AliSPDMCTrackDensity.h
+++ b/PWGLF/FORWARD/analysis2/AliSPDMCTrackDensity.h
@@ -117,7 +117,8 @@ protected:
                     const AliMCParticle* mother,
                     Int_t                refNo,
                     Double_t             vz, 
-                    TH2D&                output) const;
+                    TH2D&                output,
+                     Double_t             w) const;
   /** 
    * Get ultimate mother of a track 
    * 
@@ -137,6 +138,21 @@ protected:
    */
   Double_t GetTrackRefTheta(const AliTrackReference* ref,
                            Double_t vz) const;
+  /** 
+   * Calculate flow weight 
+   *
+   * @param eta  Pseudo rapidity 
+   * @param pt   Transverse momemtum 
+   * @param b    Impact parameter
+   * @param phi  Azimuthal angle 
+   * @param rp   Reaction plance angle 
+   * @param id   Particle PDG code
+   *
+   * @return Flow weight for the particle
+   */
+  Double_t CalculateWeight(Double_t eta, Double_t pt, Double_t b, 
+                          Double_t phi, Double_t rp, Int_t id) const;
+
   Bool_t   fUseOnlyPrimary;       // Only use primaries 
   Double_t fMinR;             // Min radius 
   Double_t fMaxR;             // Max radius