PWG3/hfe/AliHFEvarManager.h

   1 /**************************************************************************
   2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3 *                                                                        *
   4 * Author: The ALICE Off-line Project.                                    *
   5 * Contributors are mentioned in the code where appropriate.              *
   6 *                                                                        *
   7 * Permission to use, copy, modify and distribute this software and its   *
   8 * documentation strictly for non-commercial purposes is hereby granted   *
   9 * without fee, provided that the above copyright notice appears in all   *
  10 * copies and that both the copyright notice and this permission notice   *
  11 * appear in the supporting documentation. The authors make no claims     *
  12 * about the suitability of this software for any purpose. It is          *
  13 * provided "as is" without express or implied warranty.                  *
  14 **************************************************************************/
  15 //
  16 // Class AliHFEvarManager
  17 // Common place for definiton of variables to be filled into the
  18 // correction framework container
  19 // More information can be found inside the implementation file
  20 //
  21 #ifndef ALIHFEVARMANAGER_H
  22 #define ALIHFEVARMANAGER_H
  23
  24 #ifndef ROOT_TNamed
  25 #include <TNamed.h>
  26 #endif
  27
  28 template <class X>
  29 class THnSparseT;
  30 class TArrayF;
  31 typedef class THnSparseT<TArrayF> THnSparseF;
  32 class TH3F;
  33 class TF3;
  34 class AliCFContainer;
  35 class AliHFEcontainer;
  36 class AliHFEsignalCuts;
  37 class AliVParticle;
  38
  39
  40 class AliHFEvarManager : public TNamed{
  41 public:
  42   enum EVarCode_t{
  43     kPt = 1,
  44     kEta,
  45     kPhi,
  46     kCharge,
  47     kSource,
  48     kCentrality,
  49     kSpecies,
  50   };
  51         AliHFEvarManager();
  52         AliHFEvarManager(const Char_t *name);
  53   AliHFEvarManager(const AliHFEvarManager &ref);
  54   AliHFEvarManager &operator=(const AliHFEvarManager &ref);
  55   void Copy(TObject &o) const;
  56         ~AliHFEvarManager();
  57
  58   void SetOwner(Bool_t owner = kTRUE) { SetBit(kOwner, owner); }
  59   Bool_t IsOwner() const { return TestBit(kOwner); }
  60
  61   void AddVariable(TString name);
  62   Bool_t IsVariableDefined(TString name);
  63   void DefineVariables(AliHFEcontainer *cont);
  64   void NewTrack(AliVParticle *track, AliVParticle *mcTrack = NULL, Float_t centrality = 99.0, Int_t aprioriPID = -1, Bool_t signal = kTRUE);
  65   Bool_t IsSignalTrack() const { return fSignalTrack; }
  66   void FillContainer(AliCFContainer *cont, Int_t step, Bool_t useMC = kFALSE) const;
  67   void FillContainer(AliHFEcontainer *cont, const Char_t *contname, UInt_t step, Bool_t useMC = kFALSE, Double_t externalWeight = 1.) const;
  68   void FillContainerStepname(AliHFEcontainer *cont, const Char_t *contname, const Char_t *step, Bool_t useMC = kFALSE, Double_t externalWeight = 1.) const;
  69   void FillCorrelationMatrix(THnSparseF *matrix) const;
  70
  71   void SetSignalCuts(AliHFEsignalCuts *signal) { fSignal = signal; }
  72   void SetWeightFactors(TH3F *weightFactors);
  73   void SetWeightFactorsFunction(TF3*weightFactorsFunction);
  74
  75   struct AliHFEvariable : public TNamed{
  76     public:
  77       AliHFEvariable();
  78       AliHFEvariable(const Char_t *name, const Char_t *title, UInt_t fCode, UInt_t nBins, Double_t min, Double_t max, Bool_t isLogarithmic = kFALSE);
  79       AliHFEvariable(const AliHFEvariable &ref);
  80       AliHFEvariable &operator=(const AliHFEvariable &ref);
  81       ~AliHFEvariable(){}
  82
  83       UInt_t GetVarCode() const { return fCode; }
  84       UInt_t GetNumberOfBins() const { return fNBins; }
  85       Double_t GetMinimum() const { return fMin; }
  86       Double_t GetMaximum() const { return fMax; }
  87       Int_t IsLogarithmic() const { return fIsLogarithmic; }
  88     private:
  89       UInt_t    fCode;              // Unique variable code
  90       UInt_t    fNBins;             // Number of bins
  91       Double_t  fMin;               // Minimum
  92       Double_t  fMax;               // Maximum
  93       Bool_t    fIsLogarithmic;     // Logarithmic binning
  94
  95       ClassDef(AliHFEvarManager::AliHFEvariable, 1) // HFE variable definition
  96   };
  97
  98 protected:
  99   Double_t GetValue(AliVParticle *track, UInt_t code, Float_t centrality = 99.0, Int_t aprioriPID = -1) const;
 100   void FillArray(AliVParticle *track, Double_t *container, Float_t centrality = 99.0, Int_t aprioriPID = -1) const;
 101   Double_t FindWeight(Double_t pt, Double_t eta, Double_t phi) const;
 102
 103 private:
 104   enum{
 105     kOwner = BIT(14)
 106   };
 107   TObjArray *fVariables;                // Variables to process
 108   Double_t *fContent;                   //! Cache values for track in classmember
 109   Double_t *fContentMC;                 //! Cache content of the asssociated MC track in class member
 110   Double_t fWeightFactor;               // Cache weighting factor
 111   Bool_t  fSignalTrack;                 // Signal Track
 112         Bool_t  fWeighting;                   // Weighting or not for the efficiency maps
 113   AliHFEsignalCuts *fSignal;            // MC Signal Definition
 114         TH3F *fWeightFactors;                 // Weight factors
 115         TF3  *fWeightFactorsFunction;         // Weight factors
 116
 117         ClassDef(AliHFEvarManager, 1)         // The variable Manager for the HFE Analysis
 118 };
 119
 120 #endif /* ALIHFEVARMANAGER_H */