#ifndef ALIHFEVARMANAGER_H
#define ALIHFEVARMANAGER_H
#ifndef ROOT_TNamed
#include <TNamed.h>
#endif
#ifndef ROOT_TArrayD
#include <TArrayD.h>
#endif
template <class X>
class THnSparseT;
class TArrayF;
typedef class THnSparseT<TArrayF> THnSparseF;
class TH3F;
class TF3;
class AliCFContainer;
class AliHFEcontainer;
class AliHFEsignalCuts;
class AliVParticle;
class AliHFEvarManager : public TNamed{
public:
enum EVarCode_t{
kPt = 1,
kEta,
kPhi,
kCharge,
kSource,
kCentrality,
kSpecies,
};
AliHFEvarManager();
AliHFEvarManager(const Char_t *name);
AliHFEvarManager(const AliHFEvarManager &ref);
AliHFEvarManager &operator=(const AliHFEvarManager &ref);
void Copy(TObject &o) const;
~AliHFEvarManager();
TObjArray *GetVariables() const { return fVariables; }
void SetOwner(Bool_t owner = kTRUE) { SetBit(kOwner, owner); }
Bool_t IsOwner() const { return TestBit(kOwner); }
void AddVariable(TString name);
void AddVariable(TString name, Int_t nBins, Double_t min, Double_t max, Bool_t isLogarithmic = kFALSE);
void AddVariable(TString name, Int_t nBins, const Double_t *binning);
Bool_t IsVariableDefined(TString name);
void DefineVariables(AliHFEcontainer *cont);
void NewTrack(AliVParticle *track, AliVParticle *mcTrack = NULL, Float_t centrality = 99.0, Int_t aprioriPID = -1, Bool_t signal = kTRUE);
Bool_t IsSignalTrack() const { return fSignalTrack; }
void FillContainer(AliCFContainer *const cont, Int_t step, Bool_t useMC = kFALSE) const;
void FillContainer(const AliHFEcontainer *const cont, const Char_t *contname, UInt_t step, Bool_t useMC = kFALSE, Double_t externalWeight = 1.) const;
void FillContainerStepname(const AliHFEcontainer *const cont, const Char_t *contname, const Char_t *step, Bool_t useMC = kFALSE, Double_t externalWeight = 1.) const;
void FillCorrelationMatrix(THnSparseF *matrix) const;
void SetSignalCuts(AliHFEsignalCuts *signal) { fSignal = signal; }
void SetWeightFactors(TH3F *weightFactors);
void SetWeightFactorsFunction(TF3*weightFactorsFunction);
struct AliHFEvariable : public TNamed{
public:
AliHFEvariable();
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);
AliHFEvariable(const Char_t *name, const Char_t *title, UInt_t fCode, UInt_t nBins, const Double_t *binning);
AliHFEvariable(const AliHFEvariable &ref);
AliHFEvariable &operator=(const AliHFEvariable &ref);
~AliHFEvariable();
UInt_t GetVarCode() const { return fCode; }
UInt_t GetNumberOfBins() const { return fNBins; }
Double_t* GetBinning();
Bool_t HasUserDefinedBinning() const { return fUserDefinedBinning; }
Double_t GetMinimum() const { return fMin; }
Double_t GetMaximum() const { return fMax; }
Int_t IsLogarithmic() const { return fIsLogarithmic; }
private:
UInt_t fCode;
UInt_t fNBins;
Double_t fMin;
Double_t fMax;
TArrayD fBinning;
Bool_t fIsLogarithmic;
Bool_t fUserDefinedBinning;
ClassDef(AliHFEvarManager::AliHFEvariable, 1)
};
protected:
Double_t GetValue(AliVParticle *track, UInt_t code, Float_t centrality = 99.0, Int_t aprioriPID = -1) const;
void FillArray(AliVParticle *track, Double_t *container, Float_t centrality = 99.0, Int_t aprioriPID = -1) const;
Double_t FindWeight(Double_t pt, Double_t eta, Double_t phi) const;
private:
enum{
kOwner = BIT(14)
};
TObjArray *fVariables;
Double_t *fContent;
Double_t *fContentMC;
Double_t fWeightFactor;
Bool_t fSignalTrack;
Bool_t fWeighting;
AliHFEsignalCuts *fSignal;
TH3F *fWeightFactors;
TF3 *fWeightFactorsFunction;
ClassDef(AliHFEvarManager, 1)
};
#endif /* ALIHFEVARMANAGER_H */