PWG2/FORWARD/analysis2/AliMCTruthdNdetaTask.h

   1 //
   2 // Task to analyse the AOD for for dN/deta in the forward regions
   3 //
   4 #ifndef ALIMCTRUTHDNDETATASK_H
   5 #define ALIMCTRUTHDNDETATASK_H
   6 /**
   7  * @file   AliMCTruthdNdetaTask.h
   8  * @author Christian Holm Christensen <cholm@dalsgaard.hehi.nbi.dk>
   9  * @date   Wed Mar 23 14:04:54 2011
  10  *
  11  * @brief
  12  *
  13  *
  14  * @ingroup pwg2_forward_dndeta
  15  */
  16 #include "AliBasedNdetaTask.h"
  17 class TList;
  18 class TH2D;
  19 class TH1D;
  20
  21 /**
  22  * Tasks to determine @f$ dN/d\eta@f$ in the forward regions
  23  *
  24  * @ingroup pwg2_forward_tasks_dndeta
  25  * @ingroup pwg2_forward_dndeta
  26  */
  27 class AliMCTruthdNdetaTask : public AliBasedNdetaTask
  28 {
  29 public:
  30   /**
  31    * Constructor
  32    *
  33    */
  34   AliMCTruthdNdetaTask();
  35   /**
  36    * Constructor
  37    *
  38    * @param name    Name of task
  39    */
  40   AliMCTruthdNdetaTask(const char* name);
  41   /**
  42    * Destructor
  43    *
  44    */
  45   virtual ~AliMCTruthdNdetaTask() {}
  46   /**
  47    * Called at end of event processing.
  48    *
  49    * This is called once in the master
  50    *
  51    * @param option Not used
  52    */
  53   virtual void Terminate(Option_t* option);
  54 protected:
  55   /**
  56    * Copy constructor
  57    */
  58   AliMCTruthdNdetaTask(const AliMCTruthdNdetaTask& o);
  59   /**
  60    * Assigmement operator
  61    *
  62    * @return Reference to this
  63    */
  64   AliMCTruthdNdetaTask& operator=(const AliMCTruthdNdetaTask&) { return *this; }
  65
  66   /**
  67    * Retrieve the histogram
  68    *
  69    * @param aod AOD event
  70    * @param mc  Whether to get the MC histogram or not
  71    *
  72    * @return Retrieved histogram or null
  73    */
  74   TH2D* GetHistogram(const AliAODEvent* aod, Bool_t mc);
  75   /**
  76    * Get the marker style
  77    *
  78    * @return Marker style
  79    */
  80   virtual Int_t GetMarker() const { return GetMarkerStyle(kStar); }
  81   /**
  82    * Get the colour to use for markers (only pp - in PbPb we use a rainbow)
  83    *
  84    * @return Marker colour
  85    */
  86   virtual Int_t GetColor() const { return kGray+2; }
  87   /**
  88    * Make a new centrality bin
  89    *
  90    * @param name   Histogram names
  91    * @param l      Lower cut
  92    * @param h      Upper cut
  93    *
  94    * @return Newly allocated object (of our type)
  95    */
  96   AliBasedNdetaTask::CentralityBin*
  97   MakeCentralityBin(const char* name, Short_t l, Short_t h) const;
  98
  99   /**
 100    * Class that holds data for a single centrality bin
 101    *
 102    */
 103   class CentralityBin : public AliBasedNdetaTask::CentralityBin
 104   {
 105   public:
 106     /**
 107      * Constructor
 108      */
 109     CentralityBin() : AliBasedNdetaTask::CentralityBin(), fSumTruth(0) {}
 110     /**
 111      * Constructor
 112      *
 113      * @param name Name used for histograms (e.g., Forward)
 114      * @param low  Lower centrality cut in percent
 115      * @param high Upper centrality cut in percent
 116      */
 117     CentralityBin(const char* name, Short_t low, Short_t high)
 118       : AliBasedNdetaTask::CentralityBin(name, low, high),
 119         fSumTruth(0)
 120     {}
 121     /**
 122      * Copy constructor
 123      *
 124      * @param other Object to copy from
 125      */
 126     CentralityBin(const CentralityBin& other)
 127       : AliBasedNdetaTask::CentralityBin(other),
 128         fSumTruth(other.fSumTruth)
 129     {}
 130     /**
 131      * Destructor
 132      */
 133     virtual ~CentralityBin() {}
 134     /**
 135      * Assignement operator
 136      *
 137      *
 138      * @return
 139      */
 140     CentralityBin& operator=(const CentralityBin&) { return *this; }
 141     /**
 142      * Process an event
 143      *
 144      * @param forward     Forward data (for trigger, vertex, & centrality)
 145      * @param triggerMask Trigger mask
 146      * @param isZero      True if this is a zero bin event
 147      * @param vzMin       Minimum IP z coordinate
 148      * @param vzMax       Maximum IP z coordinate
 149      * @param data        Data histogram
 150      * @param mc          MC histogram
 151      */
 152     virtual void ProcessEvent(const AliAODForwardMult* forward,
 153                               Int_t                    triggerMask,
 154                               Bool_t                   isZero,
 155                               Double_t                 vzMin,
 156                               Double_t                 vzMax,
 157                               const TH2D*              data,
 158                               const TH2D*              mc);
 159     /**
 160      * End of processing
 161      *
 162      * @param sums        List of sums
 163      * @param results     Output list of results
 164      * @param scheme      Normalisation scheme options
 165      * @param shapeCorr   Shape correction or nil
 166      * @param trigEff     Trigger efficiency
 167      * @param symmetrice  Whether to symmetrice the results
 168      * @param rebin       Whether to rebin the results
 169      * @param rootProj    If true, use TH2::ProjectionX
 170      * @param corrEmpty   Whether to correct for empty bins
 171      * @param cutEdges    Whether to cut edges when rebinning
 172      * @param triggerMask Trigger mask
 173      * @param color       Marker colour
 174      * @param marker      Marker style
 175      * @param mclist      List of MC results
 176      * @param truthlist   List of MC truth results
 177      */
 178     virtual void End(TList*      sums,
 179                      TList*      results,
 180                      UShort_t    scheme,
 181                      const TH1*  shapeCorr,
 182                      Double_t    trigEff,
 183                      Bool_t      symmetrice,
 184                      Int_t       rebin,
 185                      Bool_t      rootProj,
 186                      Bool_t      corrEmpty,
 187                      Bool_t      cutEdges,
 188                      Int_t       triggerMask,
 189                      Int_t       marker,
 190                      Int_t       color,
 191                      TList*      mclist,
 192                      TList*      truthlist);
 193   protected:
 194     TH2D*           fSumTruth;    //  Sum of primary histograms
 195     ClassDef(CentralityBin,1); // A centrality bin
 196   };
 197
 198   ClassDef(AliMCTruthdNdetaTask,1); // Determine multiplicity in forward region
 199 };
 200
 201 #endif
 202 //
 203 // Local Variables:
 204 //  mode: C++
 205 // End:
 206 //