class AliExternalTrackParam;
class AliStack;
class AliESDVertex;
+class AliProtonAnalysisBase;
class AliProtonAnalysis : public TObject {
public:
Int_t nbinsPt, Float_t fLowPt, Float_t fHighPt);
virtual ~AliProtonAnalysis();
- void SetEtaMode() {fAnalysisEtaMode = kTRUE;}
+ void SetBaseAnalysis(AliProtonAnalysisBase *baseAnalysis) {
+ fProtonAnalysisBase = baseAnalysis;}
+ AliProtonAnalysisBase *GetProtonAnalysisBaseObject() {
+ return fProtonAnalysisBase;}
- void UseTPCOnly() {fUseTPCOnly = kTRUE;}
- void UseHybridTPC() {fUseHybridTPC = kTRUE;}
-
void InitAnalysisHistograms(Int_t nbinsY, Float_t fLowY, Float_t fHighY,
Int_t nbinsPt, Float_t fLowPt, Float_t fHighPt);
Bool_t ReadFromFile(const char* filename);
Bool_t PrintMean(TH1 *hist, Double_t edge);
Bool_t PrintYields(TH1 *hist, Double_t edge);
- //Cut functions
- void SetPointOnITSLayer1() {fPointOnITSLayer1Flag = kTRUE;}
- void SetPointOnITSLayer2() {fPointOnITSLayer2Flag = kTRUE;}
- void SetPointOnITSLayer3() {fPointOnITSLayer3Flag = kTRUE;}
- void SetPointOnITSLayer4() {fPointOnITSLayer4Flag = kTRUE;}
- void SetPointOnITSLayer5() {fPointOnITSLayer5Flag = kTRUE;}
- void SetPointOnITSLayer6() {fPointOnITSLayer6Flag = kTRUE;}
- void SetMinITSClusters(Int_t minITSClusters) {
- fMinITSClusters = minITSClusters;
- fMinITSClustersFlag = kTRUE;
- }
- void SetMaxChi2PerITSCluster(Double_t maxChi2PerITSCluster) {
- fMaxChi2PerITSCluster = maxChi2PerITSCluster;
- fMaxChi2PerITSClusterFlag = kTRUE;
- }
- void SetMinTPCClusters(Int_t minTPCClusters) {
- fMinTPCClusters = minTPCClusters;
- fMinTPCClustersFlag = kTRUE;
- }
- void SetMaxChi2PerTPCCluster(Double_t maxChi2PerTPCCluster) {
- fMaxChi2PerTPCCluster = maxChi2PerTPCCluster;
- fMaxChi2PerTPCClusterFlag = kTRUE;
- }
- void SetMaxCov11(Double_t maxCov11) {
- fMaxCov11 = maxCov11; fMaxCov11Flag = kTRUE;}
- void SetMaxCov22(Double_t maxCov22) {
- fMaxCov22 = maxCov22; fMaxCov22Flag = kTRUE;}
- void SetMaxCov33(Double_t maxCov33) {
- fMaxCov33 = maxCov33; fMaxCov33Flag = kTRUE;}
- void SetMaxCov44(Double_t maxCov44) {
- fMaxCov44 = maxCov44; fMaxCov44Flag = kTRUE;}
- void SetMaxCov55(Double_t maxCov55) {
- fMaxCov55 = maxCov55; fMaxCov55Flag = kTRUE;}
- void SetMaxSigmaToVertex(Double_t maxSigmaToVertex) {
- fMaxSigmaToVertex = maxSigmaToVertex;
- fMaxSigmaToVertexFlag = kTRUE;
- }
- void SetMaxSigmaToVertexTPC(Double_t maxSigmaToVertex) {
- fMaxSigmaToVertexTPC = maxSigmaToVertex;
- fMaxSigmaToVertexTPCFlag = kTRUE;
- }
- void SetMaxDCAXY(Double_t maxDCAXY) {
- fMaxDCAXY = maxDCAXY;
- fMaxDCAXYFlag = kTRUE;
- }
- void SetMaxDCAXYTPC(Double_t maxDCAXY) {
- fMaxDCAXYTPC = maxDCAXY;
- fMaxDCAXYTPCFlag = kTRUE;
- }
- void SetMaxDCAZ(Double_t maxDCAZ) {
- fMaxDCAZ = maxDCAZ;
- fMaxDCAZFlag = kTRUE;
- }
- void SetMaxDCAZTPC(Double_t maxDCAZ) {
- fMaxDCAZTPC = maxDCAZ;
- fMaxDCAZTPCFlag = kTRUE;
- }
- void SetMaxConstrainChi2(Double_t maxConstrainChi2) {
- fMaxConstrainChi2 = maxConstrainChi2;
- fMaxConstrainChi2Flag = kTRUE;
- }
- void SetITSRefit() {fITSRefitFlag = kTRUE;}
- void SetTPCRefit() {fTPCRefitFlag = kTRUE;}
- void SetESDpid() {fESDpidFlag = kTRUE;}
- void SetTPCpid() {fTPCpidFlag = kTRUE;}
-
- //Prior probabilities
- void SetPriorProbabilities(Double_t * const partFrac) {
- for(Int_t i = 0; i < AliPID::kSPECIESN; i++) fPartFrac[i] = partFrac[i];}
- void SetPriorProbabilityFunctions(TF1 *const felectron,
- TF1 *const fmuon,
- TF1 *const fpion,
- TF1 *const fkaon,
- TF1 *const fproton) {
- fFunctionProbabilityFlag = kTRUE;
- fElectronFunction = felectron;
- fMuonFunction = fmuon;
- fPionFunction = fpion;
- fKaonFunction = fkaon;
- fProtonFunction = fproton;
- }
- Double_t GetParticleFraction(Int_t i, Double_t p);
-
//interface to the correction framework
void Correct(Int_t step);
Bool_t ReadCorrectionContainer(const char* filename);
private:
AliProtonAnalysis(const AliProtonAnalysis&); // Not implemented
AliProtonAnalysis& operator=(const AliProtonAnalysis&); // Not implemented
-
- Bool_t IsAccepted(AliESDEvent *esd,
- const AliESDVertex *vertex,
- AliESDtrack *track);
- Bool_t IsInPhaseSpace(AliESDtrack *track);
-
- Float_t GetSigmaToVertex(AliESDtrack* esdTrack) const;
- Double_t Rapidity(Double_t Px, Double_t Py, Double_t Pz) const;
- Bool_t fAnalysisEtaMode; //run the analysis in eta or y
+ AliProtonAnalysisBase *fProtonAnalysisBase;//base analysis object
- Int_t fNBinsY; //number of bins in y
- Float_t fMinY, fMaxY; //min & max value of y
+ Int_t fNBinsY; //number of bins in y or eta
+ Double_t fMinY, fMaxY; //min & max value of y or eta
Int_t fNBinsPt; //number of bins in pT
- Float_t fMinPt, fMaxPt; //min & max value of pT
-
- //cuts
- Int_t fMinTPCClusters, fMinITSClusters; //min TPC & ITS clusters
- Double_t fMaxChi2PerTPCCluster, fMaxChi2PerITSCluster; //max chi2 per TPC & ITS cluster
- Double_t fMaxCov11, fMaxCov22, fMaxCov33, fMaxCov44, fMaxCov55; //max values of cov. matrix
- Double_t fMaxSigmaToVertex; //max sigma to vertex cut
- Double_t fMaxSigmaToVertexTPC; //max sigma to vertex cut
- Double_t fMaxDCAXY, fMaxDCAXYTPC; //max DCA xy
- Double_t fMaxDCAZ, fMaxDCAZTPC; //max DCA z
- Double_t fMaxConstrainChi2; //max constrain chi2 - vertex
- Bool_t fMinTPCClustersFlag, fMinITSClustersFlag; //shows if this cut is used or not
- Bool_t fMaxChi2PerTPCClusterFlag, fMaxChi2PerITSClusterFlag; //shows if this cut is used or not
- Bool_t fMaxCov11Flag, fMaxCov22Flag, fMaxCov33Flag, fMaxCov44Flag, fMaxCov55Flag; //shows if this cut is used or not
- Bool_t fMaxSigmaToVertexFlag; //shows if this cut is used or not
- Bool_t fMaxSigmaToVertexTPCFlag; //shows if this cut is used or not
- Bool_t fMaxDCAXYFlag, fMaxDCAXYTPCFlag; //shows if this cut is used or not
- Bool_t fMaxDCAZFlag, fMaxDCAZTPCFlag; //shows if this cut is used or not
- Bool_t fMaxConstrainChi2Flag; //shows if this cut is used or not
- Bool_t fITSRefitFlag, fTPCRefitFlag; //shows if this cut is used or not
- Bool_t fESDpidFlag, fTPCpidFlag; //shows if this cut is used or not
- Bool_t fPointOnITSLayer1Flag, fPointOnITSLayer2Flag; //shows if this cut is used or not
- Bool_t fPointOnITSLayer3Flag, fPointOnITSLayer4Flag; //shows if this cut is used or not
- Bool_t fPointOnITSLayer5Flag, fPointOnITSLayer6Flag; //shows if this cut is used or not
+ Double_t fMinPt, fMaxPt; //min & max value of pT
- //pid
- Bool_t fFunctionProbabilityFlag; //flag: kTRUE if functions used
- Double_t fPartFrac[10]; //prior probabilities
- TF1 *fElectronFunction; //momentum dependence of the prior probs
- TF1 *fMuonFunction; //momentum dependence of the prior probs
- TF1 *fPionFunction; //momentum dependence of the prior probs
- TF1 *fKaonFunction; //momentum dependence of the prior probs
- TF1 *fProtonFunction; //momentum dependence of the prior probs
-
- //Detectors
- Bool_t fUseTPCOnly; //kTRUE if TPC only information is used
- Bool_t fUseHybridTPC; //kTRUE if TPC info is used for momentum - PID and ITS for vertex & points
-
//Analysis containers
AliCFContainer *fProtonContainer; //container for protons
AliCFContainer *fAntiProtonContainer; //container for antiprotons
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff