Transition PWG2/FORWARD -> PWGLF

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

diff --git a/PWGLF/FORWARD/analysis2/AliForwardFlowTaskQC.h b/PWGLF/FORWARD/analysis2/AliForwardFlowTaskQC.h
new file mode 100644 (file)
index 0000000..a8a7f30
--- /dev/null
+++ b/PWGLF/FORWARD/analysis2/AliForwardFlowTaskQC.h
@@ -0,0 +1,188 @@
+//
+// Calculate the flow in the forward regions using the Q cumulants method
+//
+#ifndef ALIFORWARDFLOWTASKQC_H
+#define ALIFORWARDFLOWTASKQC_H
+/**
+ * @file   AliForwardFlowTaskQC.h
+ * @author Alexander Hansen alexander.hansen@cern.ch
+ * @date   Fri Mar 25 13:53:00 2011
+ * 
+ * @brief  
+ * 
+ * 
+ * @ingroup pwg2_forward_flow
+ */
+#include "AliAnalysisTaskSE.h"
+#include "AliForwardFlowUtil.h"
+class AliAODEvent;
+
+ /**
+ * @defgroup pwg2_forward_tasks_flow Flow tasks 
+ * @ingroup pwg2_forward_tasks
+ */
+/**
+ * Calculate the flow in the forward regions using the Q cumulants method
+ *
+ * @par Inputs:
+ *   - AliAODEvent
+ *
+ * Outputs:
+ *   - AnalysisResults.root
+ *
+ * @ingroup pwg2_forward_tasks_flow
+ * @ingroup pwg2_forward_flow
+ *
+ * @todo Add centrality stuff
+ *
+ */
+class AliForwardFlowTaskQC : public AliAnalysisTaskSE
+{
+public:
+  /** 
+   * Constructor 
+   */
+  AliForwardFlowTaskQC();
+  /** 
+   * Constructor
+   * 
+   * @param name Name of task 
+   */
+  AliForwardFlowTaskQC(const char* name);
+  /**
+   * Destructor
+   */
+  virtual ~AliForwardFlowTaskQC() {}
+  /** 
+   * @{ 
+   * @name Task interface methods 
+   */
+  /** 
+   * Create output objects 
+   * 
+   */
+  virtual void UserCreateOutputObjects();
+  /**
+   * Initialize the task
+   *
+   */
+  virtual void Init() {}
+  /** 
+   * Process each event 
+   *
+   * @param option Not used
+   */  
+  virtual void UserExec(Option_t *option);
+  /** 
+   * End of job
+   * 
+   * @param option Not used 
+   */
+  virtual void Terminate(Option_t *option);
+  /* @} */
+  /*
+   * Returns the outputlist
+   *
+   */
+  TList* GetOutputList() { return fOutputList; }
+  /* 
+   * Set Number of @f$ \eta@f$ bins to be used in flow analysis
+   *
+   */
+  void SetUseNEtaBins(Int_t nbins) { fEtaBins = nbins; }
+  /*
+   * Set which harmonics to calculate. @f$ v_{1}@f$ to @f$ v_{4}@f$ is
+   * available and calculated as default
+   *
+   * @param v1  Do @f$ v_{1}@f$ 
+   * @param v2  Do @f$ v_{2}@f$ 
+   * @param v3  Do @f$ v_{3}@f$ 
+   * @param v4  Do @f$ v_{4}@f$ 
+   */
+  void SetDoHarmonics(Bool_t v1 = kTRUE, Bool_t v2 = kTRUE, 
+                     Bool_t v3 = kTRUE, Bool_t v4 = kTRUE,
+                     Bool_t v5 = kTRUE, Bool_t v6 = kTRUE) { 
+    fv[1] = v1; fv[2] = v2; fv[3] = v3; fv[4] = v4; fv[5] = v5; fv[6] = v6;}
+  /*
+   * Set string to add flow to MC truth particles
+   *
+   * @param type String
+   */
+  void AddFlow(TString type = "") { fAddFlow = type; }
+  /*
+   * Set which function fAddFlow should use
+   *
+   * @param type of AddFlow 
+   */
+  void AddFlowType(Int_t number = 0) { fAddType = number; }
+  /*
+   * Set which order of flow to add
+   *
+   * @param order Flow order 
+   */
+  void AddFlowOrder(Int_t order = 2) { fAddOrder = order; }
+  /*
+   *
+   * Set MC input flag
+   *
+   * @param mc MC input
+   */
+  void SetMCinput(Bool_t mc = kTRUE) { fMC = mc; }
+  /*
+   * Set number of eta bins to be used in reference flow
+   *
+   * @param bins Ref Eta Bins
+   */
+  void SetRefEtaBins(Int_t bins) { fEtaRef = bins; } 
+protected:
+  /** 
+   * Copy constructor 
+   * 
+   * @param o Object to copy from 
+   */
+  AliForwardFlowTaskQC(const AliForwardFlowTaskQC& o);
+  /** 
+   * Assignment operator 
+   * 
+   * @return Reference to this object 
+   */
+  AliForwardFlowTaskQC& operator=(const AliForwardFlowTaskQC&) { return *this; }
+  /*
+   * if MC information is available do analysis on Monte Carlo truth
+   * and track references
+   *
+   */
+  void ProcessPrimary();
+  /**
+   * Calculate Q cumulant
+   * 
+   * @param type     Determines which histograms should be used
+   *                 - "FMD" = FMD data histograms
+   *                 - "SPD" = SPD data histograms
+   *                 - "TR" = track reference histograms
+   *                 - "MC" = MC truth histograms
+   * @param harmonic Which harmonic to calculate
+   */
+  void CumulantsMethod(TString type, Int_t harmonic);
+
+  TList*         fOutputList;   //  Output list
+  AliForwardFlowUtil* fFlowUtil;//  AliForwardFlowUtil
+  AliAODEvent*   fAOD;          //  AOD event
+  Bool_t         fMC;           //  Is MC flags
+  Int_t          fEtaBins;      //  Number of eta bins in histograms
+  Int_t          fEtaRef;       //  Number of eta bins for reference flow
+  Bool_t         fv[7];         //  Calculate v_{n} flag
+  TString        fAddFlow;     //  Add flow string
+  Int_t          fAddType;     //  Add flow type #
+  Int_t          fAddOrder;    //  Add flow order
+  Float_t       fZvertex;      //  Z vertex bin
+  Double_t       fCent;         //  Centrality
+  
+
+  ClassDef(AliForwardFlowTaskQC, 2); // Analysis task for FMD analysis
+};
+ 
+#endif
+// Local Variables:
+//   mode: C++ 
+// End: