Merge branch 'master' of https://git.cern.ch/reps/AliRoot

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

diff --git a/PWGLF/FORWARD/analysis2/AliForwardMCFlowTaskQC.h b/PWGLF/FORWARD/analysis2/AliForwardMCFlowTaskQC.h
index f6d04d81e797a69dd4fc1e1148449b1acdafcf82..7e12151bd4f249d977cbd2d5f160ea848cba6f59 100644 (file)
--- a/PWGLF/FORWARD/analysis2/AliForwardMCFlowTaskQC.h
+++ b/PWGLF/FORWARD/analysis2/AliForwardMCFlowTaskQC.h
@@ -6,22 +6,15 @@
 /**
  * @file   AliForwardMCFlowTaskQC.h
  * @author Alexander Hansen alexander.hansen@cern.ch
- * @date   Tue Feb 14 2012
  * 
  * @brief  
  * 
- * 
  * @ingroup pwglf_forward_flow
  */
 #include "AliForwardFlowTaskQC.h"
 #include "AliForwardFlowWeights.h"
-#include <TH2D.h>
 class TGraph;
 
- /**
- * @defgroup pwg2_forward_tasks_flow Flow tasks 
- * @ingroup pwg2_forward_tasks
- */
 /**
  * Calculate the flow in the forward regions using the Q cumulants method
  *
@@ -29,11 +22,10 @@ class TGraph;
  *   - AliAODEvent
  *
  * Outputs:
- *   - AnalysisResults.root
- *
- * @ingroup pwg2_forward_tasks_flow
- * @ingroup pwg2_forward_flow
+ *   - forward_flow.root
  *
+ * @ingroup pwglf_forward_tasks_flow
+ * @ingroup pwglf_forward_flow
  *
  */
 class AliForwardMCFlowTaskQC : public AliForwardFlowTaskQC
@@ -43,7 +35,7 @@ public:
    * Constructor
    */
   AliForwardMCFlowTaskQC();
-  /*
+  /**
    * Constructor
    *
    * @param name Name of task
@@ -54,34 +46,30 @@ public:
    */
   virtual ~AliForwardMCFlowTaskQC() {}
   /**
-   * Check for centrality in AliAODForwardMult object, 
-   * if present return true - also sets fCent value
-   * can be used to get centrality from impact parameter
-   *
-   * @param aodfm AliAODForwardMultObject
-   * 
-   * @return Bool_t 
-   */
-  virtual Bool_t GetCentrality(const AliAODForwardMult* aodfm);
-  /*
    * Set use parametrization from impact parameter for centrality
    *
    * @param use Use impact par
    */
-  void SetUseImpactParameter(Bool_t use) { fUseImpactPar = use; }
-  /*
-   * Set string to add flow to MC truth particles
+  void SetUseImpactParameter(Bool_t use = kTRUE) { fUseImpactPar = use; }
+  /**
+   * Set to get vertex from MC header
+   *
+   * @param use Get from MC header
+   */
+  void SetUseMCHeaderVertex(Bool_t use = kTRUE) { fUseMCVertex = use; }
+  /**
+   * Set string to add flow to MC particles particles
    *
    * @param type String
    */
   void AddFlow(TString type = "") { fAddFlow = type; }
-  /*
+  /**
    * Set which function fAddFlow should use
    *
-   * @param type of AddFlow 
+   * @param number Type of AddFlow 
    */
   void AddFlowType(Int_t number = 0) { fAddType = number; }
-  /*
+  /**
    * Set which order of flow to add
    *
    * @param order Flow order 
@@ -89,7 +77,7 @@ public:
   void AddFlowOrder(Int_t order = 2) { fAddOrder = order; }
  
 protected:
-  /*
+  /**
    * Copy constructor
    *
    * @param o Object to copy from
@@ -98,49 +86,86 @@ protected:
   /** 
    * Assignment operator 
    * 
+   * @param o Object to assing from 
+   *
    * @return Reference to this object 
    */
   AliForwardMCFlowTaskQC& operator=(const AliForwardMCFlowTaskQC& o);
-  /*
+  /**
    * Initiate vertex bin objects
    */
   void InitVertexBins();
-   /*
+  /**
    * Initiate diagnostics histograms
    */
   void InitHists();
-  /*
+  /**
    * Analyze event
+   *
+   * @return true on success 
    */
   Bool_t Analyze();
-  /*
+  /**
    * Finalize analysis
    */
   void Finalize();
+  /**
+   * Check trigger from AODForwardMult object
+   * returns true if B trigger is present
+   *
+   * @param aodfm AliAODForwardMultObject
+   * 
+   * @return Bool_t 
+   */
+  virtual Bool_t CheckTrigger(const AliAODForwardMult* aodfm) const;
+  /**
+   * Check for centrality in AliAODForwardMult object, 
+   * if present return true - also sets fCent value
+   * can be used to get centrality from impact parameter
+   *
+   * @param aodfm AliAODForwardMultObject
+   * 
+   * @return Bool_t 
+   */
+  virtual Bool_t GetCentrality(const AliAODForwardMult* aodfm);
+  /**
+   * Check for vertex in MCHeader
+   * returns true if in range of fVtxAXis, also sets fVtx value
+   *
+   * @param aodfm Not used
+   * 
+   * @return Bool_t 
+   */
+  virtual Bool_t GetVertex(const AliAODForwardMult* aodfm);
   /**
    * Loop over AliAODMCParticle branch object and fill d^2N/detadphi histograms
    * add flow if arguments are set
    * 
    * @return true on success
    */
-  Bool_t LoopAODMC();
+  Bool_t FillMCHist();
   /**
    * Get centrality form MC impact parameter
+   *
+   * @return Centrality
    */
   Double_t GetCentFromB() const;
   
-  TList         fBinsFMDTR;         //  List with FMDTR VertexBin objects
-  TList         fBinsSPDTR;         //  List with SPDTR VertexBin objects
-  TList         fBinsMC;            //  List with MC VertexBin objects
-  TH2D          fdNdedpMC;          //  d^2N/detadphi MC truth histogram
-  AliForwardFlowWeights fWeights;   //  Flow after burner 
-  TGraph*       fImpactParToCent;   //  Parametrization of b to centrality
-  Bool_t        fUseImpactPar;      //  Flag to use impact parameter for cent
-  TString       fAddFlow;           //  Add flow string
-  Int_t         fAddType;           //  Add flow type #
-  Int_t         fAddOrder;          //  Add flow order
+  TList                 fBinsForwardTR;   //  List with FMDTR VertexBin objects
+  TList                 fBinsCentralTR;   //  List with SPDTR VertexBin objects
+  TList                 fBinsMC;          //  List with MC VertexBin objects
+  TH2D                  fHistdNdedpMC;    //  d^2N/detadphi MC particles histogram
+  TH2D*                 fHistFMDMCCorr;   //  Diagnostics for mult. corr. between FMD and MC
+  TH2D*                 fHistSPDMCCorr;   //  Diagnostics for mult. corr. between SPD and MC
+  AliForwardFlowWeights fWeights;         //  Flow after burner 
+  TGraph*               fImpactParToCent; //  Parametrization of b to centrality
+  Bool_t                fUseImpactPar;    //  Flag to use impact parameter for cent
+  Bool_t                fUseMCVertex;     //  Get vertex from MC header
+  TString               fAddFlow;         //  Add flow string
+  Int_t                 fAddType;         //  Add flow type #
+  Int_t                 fAddOrder;        //  Add flow order
 
-  ClassDef(AliForwardMCFlowTaskQC, 2); // FMD MC analysis task 
+  ClassDef(AliForwardMCFlowTaskQC, 4); // FMD MC analysis task 
 };
  
 #endif