add possibility to use the pre-defined set of TPC priors (M.Figueredo, R.Romita)
authorzconesa <zaida.conesa.del.valle@cern.ch>
Wed, 29 Jan 2014 13:06:54 +0000 (14:06 +0100)
committerzconesa <zaida.conesa.del.valle@cern.ch>
Wed, 29 Jan 2014 13:07:23 +0000 (14:07 +0100)
PWGHF/vertexingHF/AliAODPidHF.cxx [changed mode: 0644->0755]
PWGHF/vertexingHF/AliAODPidHF.h [changed mode: 0644->0755]

old mode 100644 (file)
new mode 100755 (executable)
index 1eb8b67..a0aa952
@@ -73,7 +73,8 @@ AliAODPidHF::AliAODPidHF():
   fTPCResponse(new AliTPCPIDResponse()),
   fPriorsH(),
   fCombDetectors(kTPCTOF),
-  fUseCombined(kFALSE)
+  fUseCombined(kFALSE),
+  fDefaultPriors(kTRUE)
 {
   //
   // Default constructor
@@ -154,7 +155,8 @@ AliAODPidHF::AliAODPidHF(const AliAODPidHF& pid) :
   fPidCombined(0x0),
   fTPCResponse(0x0),
   fCombDetectors(pid.fCombDetectors),
-  fUseCombined(pid.fUseCombined)
+  fUseCombined(pid.fUseCombined),
+  fDefaultPriors(pid.fDefaultPriors)  
 {
   
   fnSigmaCompat=new Double_t[fnNSigmaCompat];
@@ -963,8 +965,12 @@ void AliAODPidHF::SetUpCombinedPID(){
   // Configuration of combined Bayesian PID
 
   fPidCombined->SetSelectedSpecies(AliPID::kSPECIES);
-  for (Int_t ispecies=0;ispecies<AliPID::kSPECIES;++ispecies) {
-    fPidCombined->SetPriorDistribution(static_cast<AliPID::EParticleType>(ispecies),fPriorsH[ispecies]);
+  if(!fDefaultPriors){
+       for (Int_t ispecies=0;ispecies<AliPID::kSPECIES;++ispecies) {
+       fPidCombined->SetPriorDistribution(static_cast<AliPID::EParticleType>(ispecies),fPriorsH[ispecies]);
+       }
+  }else{
+       fPidCombined->SetDefaultTPCPriors(); 
   }
   switch (fCombDetectors){
   case kTPCTOF:
old mode 100644 (file)
new mode 100755 (executable)
index c095271..a0f8c9d
@@ -158,7 +158,8 @@ class AliAODPidHF : public TObject{
     return fCombDetectors;
   }
   Bool_t GetUseCombined() {return fUseCombined;}
-
+  Bool_t GetDefaultPriors() {return fDefaultPriors;}
   Int_t RawSignalPID (AliAODTrack *track, TString detector) const;
   Bool_t IsKaonRaw (AliAODTrack *track, TString detector) const;
   Bool_t IsPionRaw (AliAODTrack *track, TString detector) const;
@@ -190,7 +191,7 @@ class AliAODPidHF : public TObject{
   void SetPriorsHistos(TString priorFileName);
   void SetUpCombinedPID();
   void SetUseCombined(Bool_t useCombined=kTRUE) {fUseCombined=useCombined;}
-
+  void SetUseDefaultPriors(Bool_t defaultP)        {fDefaultPriors=defaultP;}
   Int_t ApplyPidTPCRaw(AliAODTrack *track,Int_t specie) const;
   Int_t ApplyPidTOFRaw(AliAODTrack *track,Int_t specie) const;
   Int_t ApplyPidITSRaw(AliAODTrack *track,Int_t specie) const;
@@ -240,8 +241,9 @@ class AliAODPidHF : public TObject{
   TH1F* fPriorsH[AliPID::kSPECIES]; // priors histos
   ECombDetectors fCombDetectors; // detectors to be involved for combined PID
   Bool_t fUseCombined; // detectors to be involved for combined PID
+  Bool_t fDefaultPriors; // use default priors for combined PID
 
-  ClassDef(AliAODPidHF,20) // AliAODPid for heavy flavor PID
+  ClassDef(AliAODPidHF,21) // AliAODPid for heavy flavor PID
 
     };