FMD/AliFMDReconstructor.h

   1 #ifndef ALIFMDRECONSTRUCTOR_H
   2 #define ALIFMDRECONSTRUCTOR_H
   3 //
   4 //  Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights
   5 //  reserved.
   6 //
   7 //  See cxx source for full Copyright notice
   8 //
   9 //  AliFMDReconstructor.h
  10 //  Task Class for making TreeR for FMD
  11 //
  12 //-- Authors: Evgeny Karpechev (INR) and Alla Maevskaia (INR)
  13 //   Latest changes by Christian Holm Christensen <cholm@nbi.dk>
  14 /* $Id$ */
  15 /** @file    AliFMDReconstructor.h
  16     @author  Christian Holm Christensen <cholm@nbi.dk>
  17     @date    Mon Mar 27 12:47:09 2006
  18     @brief   FMD reconstruction
  19 */
  20
  21 //____________________________________________________________________
  22 // Header guards in the header files speeds up the compilation
  23 // considerably.  Please leave them in.
  24 #ifndef ALIRECONSTRUCTOR_H
  25 # include <AliReconstructor.h>
  26 #endif
  27 #include "AliLog.h"
  28
  29 //____________________________________________________________________
  30 class TTree;
  31 class TClonesArray;
  32 class AliFMDDigit;
  33 class AliRawReader;
  34 class AliESDEvent;
  35 class AliESDFMD;
  36 class AliFMDRecoParam;
  37 class TH1;
  38
  39
  40 /** @defgroup FMD_rec Reconstruction */
  41 //____________________________________________________________________
  42 /**
  43  * @brief This is a class that reconstructs AliFMDRecPoint objects
  44  *        from of Digits.
  45  *
  46  * This class reads either digits from a TClonesArray or raw data
  47  * from a DDL file (or similar), and applies calibrations to get
  48  * psuedo-inclusive multiplicities per strip.
  49  *
  50  * @ingroup FMD_rec
  51  */
  52 class AliFMDReconstructor: public AliReconstructor
  53 {
  54 public:
  55   /**
  56    * CTOR
  57    */
  58   AliFMDReconstructor();
  59   /**
  60    * DTOR
  61    */
  62   virtual ~AliFMDReconstructor();
  63
  64   /**
  65    * Initialize the reconstructor.  Here, we initialize the geometry
  66    * manager, and finds the local to global transformations from the
  67    * geometry.   The calibration parameter manager is also
  68    * initialized (meaning that the calibration parameters is read
  69    * from CDB).
  70    */
  71   virtual void   Init();
  72   /**
  73    * Flag that we can convert raw data into digits.
  74    *
  75    * @return always @c true
  76    */
  77   virtual Bool_t HasDigitConversion() const { return kTRUE; }
  78   /**
  79    * Convert raw data read from the AliRawReader @a reader into
  80    * digits.  This is done using AliFMDRawReader and
  81    * AliFMDAltroReader.  The digits are put in the passed TTree @a
  82    * digitsTree.
  83    *
  84    * @param reader     Raw reader.
  85    * @param digitsTree Tree to store read digits in.
  86    */
  87   virtual void   ConvertDigits(AliRawReader* reader, TTree* digitsTree) const;
  88   /**
  89    * Reconstruct one event from the digits passed in @a digitsTree.
  90    * The member function creates AliFMDRecPoint objects and stores
  91    * them on the output tree @a clusterTree.  An FMD ESD object is
  92    * created in parallel.
  93    *
  94    * @param digitsTree  Tree holding the digits of this event
  95    * @param clusterTree Tree to store AliFMDRecPoint objects in.
  96    */
  97   virtual void   Reconstruct(TTree* digitsTree, TTree* clusterTree) const;
  98   /**
  99    * Not used
 100    * @todo Implement this, such that we'll reconstruct directly from
 101    *       the read ADC values rather than going via an intermedant
 102    *       TClonesArray of AliFMDDigits
 103    */
 104   virtual void   Reconstruct(AliRawReader *, TTree*) const;
 105   /**
 106    * Put in the ESD data, the FMD ESD data.  The object created by
 107    * the Reconstruct member function is copied to the ESD object.
 108    *
 109    * @param digitsTree   Tree of digits for this event - not used
 110    * @param clusterTree  Tree of reconstructed points for this event -
 111    *        not used.
 112    * @param esd ESD object to store data in.
 113    */
 114   virtual void   FillESD(TTree* digitsTree, TTree* clusterTree,
 115                          AliESDEvent* esd) const;
 116   /**
 117    * Forwards to above member function
 118    */
 119   virtual void   FillESD(AliRawReader*, TTree* clusterTree,
 120                          AliESDEvent* esd) const;
 121
 122   /**
 123    * Create SDigits from raw data
 124    *
 125    * @param reader  The raw reader
 126    * @param sdigits Array to fill with AliFMDSDigit objects.
 127    */
 128   virtual void Digitize(AliRawReader* reader,
 129                         TClonesArray* sdigits) const;
 130
 131   /**
 132    * Not used
 133    */
 134   virtual void   SetESD(AliESDEvent* esd) { fESD = esd; }
 135   /**
 136    * Set the noise factor
 137    *
 138    * @param f Factor to use
 139    */
 140   virtual void SetNoiseFactor(Float_t f=3) { fNoiseFactor = f; }
 141   /**
 142    * Set whether we should do angle correction or nor
 143    *
 144    * @param use If true, do angle correction
 145    */
 146   virtual void SetAngleCorrect(Bool_t use=kTRUE) { fAngleCorrect = use; }
 147   /**
 148    * Set whether we want to do diagnostics.  If this is enabled, a
 149    * file named @c FMD.Diag.root will be made.  It contains a set of
 150    * histograms for each event, filed in separate directories in the
 151    * file.  The histograms are
 152    * @verbatim
 153    * diagStep1   Read ADC vs. Noise surpressed ADC
 154    * diagStep2   Noise surpressed ADC vs. calculated Energy dep.
 155    * diagStep3   Energy deposition vs. angle corrected Energy dep.
 156    * diagStep4   Energy deposition vs. calculated multiplicity
 157    * diagAll     Read ADC vs. calculated multiplicity
 158    * @endverbatim
 159    *
 160    * @param use If true, make the diagnostics file
 161    */
 162   void SetDiagnose(Bool_t use=kTRUE) { fDiagnostics = use; }
 163
 164 protected:
 165   /**
 166    * Copy CTOR
 167    *
 168    * @param other Object to copy from.
 169    */
 170   AliFMDReconstructor(const AliFMDReconstructor&); //Not implemented
 171   /**
 172    * Assignment operator
 173    *
 174    * @param other Object to assign from
 175    *
 176    * @return reference to this object
 177    */
 178   AliFMDReconstructor& operator=(const AliFMDReconstructor&); //Not implemented
 179   /**
 180    * Try to get the vertex from either ESD or generator header.  Sets
 181    * @c fCurrentVertex to the found Z posistion of the vertex (if
 182    * found), and sets the flag @c fVertexType accordingly
 183    *
 184    * @param esd ESD structure to get Vz from
 185    */
 186   virtual void GetVertex(AliESDEvent* esd) const;
 187   /**
 188    * Process AliFMDDigit objects in @a digits.  For each digit, find
 189    * the psuedo-rapidity @f$ \eta@f$, azimuthal angle @f$ \varphi@f$,
 190    * energy deposited @f$ E@f$, and psuedo-inclusive multiplicity @f$
 191    * M@f$.
 192    *
 193    * @param digits Array of digits.
 194    */
 195   virtual void     ProcessDigits(TClonesArray* digits) const;
 196   /**
 197    * Process a single digit
 198    *
 199    * @param digit Digiti to process
 200    */
 201   virtual void ProcessDigit(AliFMDDigit* digit) const;
 202   /**
 203    * Process the signal from a single strip.
 204    *
 205    * @param det Detector number
 206    * @param rng Ring identifier
 207    * @param sec Sector number
 208    * @param str Strip number
 209    * @param adc Number of ADC counts for this strip
 210    */
 211   virtual void ProcessSignal(UShort_t det,
 212                              Char_t   rng,
 213                              UShort_t sec,
 214                              UShort_t str,
 215                              Short_t  adc) const;
 216   /**
 217    * Process the signal from a single strip.
 218    *
 219    * @param sdigits Array to fill
 220    * @param det     Detector number
 221    * @param rng     Ring identifier
 222    * @param sec     Sector number
 223    * @param str     Strip number
 224    * @param sam     Sample number
 225    * @param adc     Number of ADC counts for this strip
 226    */
 227   virtual void DigitizeSignal(TClonesArray* sdigits,
 228                               UShort_t      det,
 229                               Char_t         rng,
 230                               UShort_t       sec,
 231                               UShort_t       str,
 232                               UShort_t       sam,
 233                               Short_t        adc) const;
 234   /**
 235    * Subtract the pedestal off the ADC counts.
 236    *
 237    * @param det           Detector number
 238    * @param rng           Ring identifier
 239    * @param sec           Sector number
 240    * @param str           Strip number
 241    * @param adc           ADC counts
 242    * @param noiseFactor   If pedestal substracted pedestal is less then
 243    *        this times the noise, then consider this to be 0.
 244    * @param zsEnabled     Whether zero-suppression is on.
 245    * @param zsNoiseFactor Noise factor used in on-line pedestal
 246    *        subtraction.
 247    *
 248    * @return The pedestal subtracted ADC counts (possibly 0), or @c
 249    *         USHRT_MAX in case of problems.
 250    */
 251   virtual UShort_t SubtractPedestal(UShort_t det,
 252                                     Char_t   rng,
 253                                     UShort_t sec,
 254                                     UShort_t str,
 255                                     UShort_t adc,
 256                                     Float_t  noiseFactor,
 257                                     Bool_t   zsEnabled,
 258                                     UShort_t zsNoiseFactor) const;
 259   /**
 260    * Substract pedestals from raw ADC in @a digit
 261    *
 262    * @param det   Detector number
 263    * @param rng   Ring identifier
 264    * @param sec   Sector number
 265    * @param str   Strip number
 266    * @param adc   Number of ADC counts
 267    *
 268    * @return Pedestal subtracted ADC count.
 269    */
 270   virtual UShort_t SubtractPedestal(UShort_t det,
 271                                     Char_t   rng,
 272                                     UShort_t sec,
 273                                     UShort_t str,
 274                                     Short_t  adc) const;
 275   /**
 276    * Converts number of ADC counts to energy deposited.   This is
 277    * done by
 278    * @f[
 279    * E_i = A_i g_i
 280    * @f]
 281    * where @f$ A_i@f$ is the pedestal subtracted ADC counts, and @f$
 282    * g_i@f$ is the gain for the @f$ i^{\mbox{th}}@f$ strip.
 283    *
 284    * @param det   Detector number
 285    * @param rng   Ring identifier
 286    * @param sec   Sector number
 287    * @param str   Strip number
 288    * @param eta   Psuedo-rapidity of digit.
 289    * @param count Pedestal subtracted ADC counts
 290    *
 291    * @return Energy deposited @f$ E_i@f$
 292    */
 293   virtual Float_t  Adc2Energy(UShort_t det,
 294                               Char_t   rng,
 295                               UShort_t sec,
 296                               UShort_t str,
 297                               UShort_t count) const;
 298   /**
 299    * Converts number of ADC counts to energy deposited.   This is
 300    * done by
 301    * @f[
 302    * E_i = A_i g_i
 303    * @f]
 304    * where @f$ A_i@f$ is the pedestal subtracted ADC counts, and @f$
 305    * g_i@f$ is the gain for the @f$ i^{\mbox{th}}@f$ strip.
 306    *
 307    * @param det   Detector number
 308    * @param rng   Ring identifier
 309    * @param sec   Sector number
 310    * @param str   Strip number
 311    * @param eta   Psuedo-rapidity of digit.
 312    * @param count Pedestal subtracted ADC counts
 313    *
 314    * @return Energy deposited @f$ E_i@f$
 315    */
 316   virtual Float_t  Adc2Energy(UShort_t det,
 317                               Char_t   rng,
 318                               UShort_t sec,
 319                               UShort_t str,
 320                               Float_t  eta,
 321                               UShort_t count) const;
 322   /**
 323    * Converts an energy signal to number of particles. In this
 324    * implementation, it's done by
 325    * @f[
 326    * M_i = E_i / E_{\mbox{MIP}}
 327    * @f]
 328    * where @f$ E_i@f$ is the energy deposited, and
 329    * @f$ E_{\mbox{MIP}}@f$ is the average energy deposited by a
 330    * minimum ionizing particle
 331    *
 332    * @param det   Detector number
 333    * @param rng   Ring identifier
 334    * @param sec   Sector number
 335    * @param str   Strip number
 336    * @param eta   On return, psuedo-rapidity @f$ \eta@f$
 337    * @param phi   On return, azimuthal angle @f$ \varphi@f$
 338    * @param edep Energy deposited @f$ E_i@f$
 339    *
 340    * @return Psuedo-inclusive multiplicity @f$ M@f$
 341    */
 342   virtual Float_t  Energy2Multiplicity(UShort_t det,
 343                                        Char_t   rng,
 344                                        UShort_t sec,
 345                                        UShort_t str,
 346                                        Float_t  edep) const;
 347   /**
 348    * Calculate the physical coordinates psuedo-rapidity @f$ \eta@f$,
 349    * azimuthal angle @f$ \varphi@f$ of the strip corresponding to
 350    * the digit @a digit.   This is done by using the information
 351    * obtained, and previously cached by AliFMDGeometry, from the
 352    * TGeoManager.
 353    *
 354    * @param det   Detector number
 355    * @param rng   Ring identifier
 356    * @param sec   Sector number
 357    * @param str   Strip number
 358    * @param eta   On return, psuedo-rapidity @f$ \eta@f$
 359    * @param phi   On return, azimuthal angle @f$ \varphi@f$
 360    */
 361   virtual void     PhysicalCoordinates(UShort_t det,
 362                                        Char_t   rng,
 363                                        UShort_t sec,
 364                                        UShort_t str,
 365                                        Float_t& eta,
 366                                        Float_t& phi) const;
 367   /**
 368    * Set-up reconstructor to use values from reconstruction
 369    * parameters, if present, for this event.   If the argument @a set
 370    * is @c false, then restore preset values.
 371    *
 372    * @param set
 373    */
 374   virtual void UseRecoParam(Bool_t set=kTRUE) const;
 375   /**
 376    * Utility member function to get the reconstruction parameters for
 377    * this event
 378    *
 379    * @return Pointer to AliFMDRecoParam object or null if not
 380    * available.
 381    */
 382   const AliFMDRecoParam* GetParameters() const;
 383   /**
 384    * Get the numeric identifier of this detector
 385    *
 386    * @return Should be 12
 387    */
 388   Int_t GetIdentifier() const;
 389   enum Vertex_t {
 390     kNoVertex,   // Got no vertex
 391     kGenVertex,  // Got generator vertex
 392     kESDVertex   // Got ESD vertex
 393   };
 394   mutable TClonesArray* fMult;          // Cache of RecPoints
 395   mutable Int_t         fNMult;         // Number of entries in fMult
 396   mutable TTree*        fTreeR;         // Output tree
 397   mutable Float_t       fCurrentVertex; // Z-coordinate of primary vertex
 398   mutable AliESDFMD*    fESDObj;        // ESD output object
 399   mutable Float_t       fNoiseFactor;   // Factor of noise to check
 400   mutable Bool_t        fAngleCorrect;  // Whether to angle correct
 401   mutable Vertex_t      fVertexType;    // What kind of vertex we got
 402   AliESDEvent*          fESD;           // ESD object(?)
 403   Bool_t                fDiagnostics;   // Wheter to do diagnostics
 404   TH1*                  fDiagStep1;     // Diagnostics histogram
 405   TH1*                  fDiagStep2;     // Diagnostics histogram
 406   TH1*                  fDiagStep3;     // Diagnostics histogram
 407   TH1*                  fDiagStep4;     // Diagnostics histogram
 408   TH1*                  fDiagAll;       // Diagnostics histogram
 409   mutable Bool_t        fZS[3];         // Zero-suppredded?
 410   mutable UShort_t      fZSFactor[3];   // Noise factor for Zero-suppression
 411 private:
 412
 413   ClassDef(AliFMDReconstructor, 3)  // class for the FMD reconstruction
 414 };
 415 #endif
 416 //____________________________________________________________________
 417 //
 418 // Local Variables:
 419 //   mode: C++
 420 // End:
 421 //
 422 // EOF
 423 //