PWG2/FORWARD/analysis2/AliCentralCorrAcceptance.h

   1 //
   2 // This class contains the acceptance correction due to dead channels
   3 //
   4 //
   5 #ifndef ALICENTRALCORRACCEPTANCE_H
   6 #define ALICENTRALCORRACCEPTANCE_H
   7 #include <TObject.h>
   8 #include <TObjArray.h>
   9 #include <TAxis.h>
  10 class TH1D;
  11
  12 /**
  13  * This class contains the acceptance correction due to dead channels
  14  *
  15  * These are generated from the on-line dead channel calculations
  16  *
  17  * @ingroup pwg2_forward_corr
  18  */
  19 class AliCentralCorrAcceptance : public TObject
  20 {
  21 public:
  22   /**
  23    * Default constructor
  24    */
  25   AliCentralCorrAcceptance();
  26   /**
  27    * Copy constructor
  28    *
  29    * @param o Object to copy from
  30    */
  31   AliCentralCorrAcceptance(const AliCentralCorrAcceptance& o);
  32   /**
  33    * Destructor
  34    *
  35    */
  36   virtual ~AliCentralCorrAcceptance();
  37   /**
  38    * @{
  39    * @name Get corrections and parameters
  40    */
  41   /**
  42    * Assignment operator
  43    *
  44    * @param o Object to assign from
  45    *
  46    * @return Reference to this object
  47    */
  48   AliCentralCorrAcceptance& operator=(const AliCentralCorrAcceptance& o);
  49   /**
  50    * Get the acceptance correction @f$ a_{r,v}@f$
  51    *
  52    * @param d  Detector number (1-3)
  53    * @param r  Ring identifier (I or O)
  54    * @param v  Primary interaction point @f$z@f$ coordinate
  55    *
  56    * @return The correction @f$ a_{r,v}@f$
  57    */
  58   TH1D* GetCorrection(Double_t v) const;
  59   /**
  60    * Get the acceptance correction @f$ a_{r,v}@f$
  61    *
  62    * @param d  Detector number (1-3)
  63    * @param r  Ring identifier (I or O)
  64    * @param b  Bin corresponding to the primary interaction point
  65    *           @f$z@f$ coordinate (1 based)
  66    *
  67    * @return The correction @f$ a_{r,v}@f$
  68    */
  69   TH1D* GetCorrection(UShort_t b) const;
  70   /**
  71    * Get the vertex axis used
  72    *
  73    * @return vertex axis
  74    */
  75   const TAxis& GetVertexAxis() const { return fVertexAxis; }
  76   /* @} */
  77
  78   /**
  79    * @{
  80    * @name Set corrections and parameters
  81    */
  82   /**
  83    * Set the acceptance correction @f$ a_{r,v}(\eta)@f$.
  84    * Note, that the object takes ownership of the passed pointer.
  85    *
  86    * @param d    Detector number (1-3)
  87    * @param r    Ring identifier (I or O)
  88    * @param v    Primary interaction point @f$z@f$ coordinate
  89    * @param h    @f$ a_{r,v}(\eta)@f$
  90    *
  91    * @return true if operation succeeded
  92    */
  93   Bool_t SetCorrection(Double_t v, TH1D* h);
  94   /**
  95    * Set the acceptance correction @f$ a_{r,v}(\eta)@f$
  96    * Note, that the object takes ownership of the passed pointer.
  97    *
  98    * @param d    Detector number (1-3)
  99    * @param r    Ring identifier (I or O)
 100    * @param b    Bin corresponding to the primary interaction point
 101    *             @f$z@f$ coordinate  (1 based)
 102    * @param h    @f$ a_{r,v}(\eta)@f$
 103    *
 104    * @return true if operation succeeded
 105    */
 106   Bool_t SetCorrection(UShort_t b, TH1D* h);
 107   /**
 108    * Set the vertex axis to use
 109    *
 110    * @param axis Vertex axis
 111    */
 112   void SetVertexAxis(const TAxis& axis);
 113   /**
 114    * Set the vertex axis to use
 115    *
 116    * @param nBins   Number of bins
 117    * @param min     Minimum
 118    * @param max     Maximum
 119    */
 120   void SetVertexAxis(Int_t nBins, Double_t min, Double_t max);
 121   /* @} */
 122
 123   /**
 124    * @{
 125    * @name Auxiliary member functions
 126    */
 127   /**
 128    * Declare this as a folder
 129    *
 130    * @return Always true
 131    */
 132   Bool_t IsFolder() const { return true; }
 133   /**
 134    * Browse this object in the browser
 135    *
 136    * @param b
 137    */
 138   void Browse(TBrowser* b);
 139   /**
 140    * Print this object
 141    *
 142    * @param option
 143    */
 144   void Print(Option_t* option="R") const; //*MENU*
 145   /* @} */
 146 protected:
 147   /**
 148    * Find the vertex bin that corresponds to the passed vertex
 149    *
 150    * @param vertex The interaction points @f$z@f$-coordinate
 151    *
 152    * @return Vertex bin in @f$[1,N_{\mbox{vertex}}]@f$ or negative if
 153    * out of range
 154    */
 155   Int_t FindVertexBin(Double_t vertex) const;
 156
 157   TObjArray fArray;          // Array of per-vertex acceptance corr
 158   TAxis     fVertexAxis;     // The vertex axis
 159   ClassDef(AliCentralCorrAcceptance,1); // Acceptance correction due to dead areas
 160 };
 161
 162 //____________________________________________________________________
 163 inline void
 164 AliCentralCorrAcceptance::SetVertexAxis(Int_t nBins, Double_t min,
 165                                            Double_t max)
 166 {
 167   fVertexAxis.Set(nBins, min, max);
 168 }
 169 //____________________________________________________________________
 170 inline void
 171 AliCentralCorrAcceptance::SetVertexAxis(const TAxis& e)
 172 {
 173   fVertexAxis.Set(e.GetNbins(), e.GetXmin(), e.GetXmax());
 174 }
 175 #endif
 176 // Local Variables:
 177 //  mode: C++
 178 // End: