Updated tratment of histograms with priors

diff --git a/PWGHF/vertexingHF/AliAODPidHF.cxx b/PWGHF/vertexingHF/AliAODPidHF.cxx
index 12c65594a03bb3e20ff93af42365ebdf3fb53d63..e7187bb93bd6e0aa79b1c64966bc97377e877336 100644 (file)
--- a/PWGHF/vertexingHF/AliAODPidHF.cxx
+++ b/PWGHF/vertexingHF/AliAODPidHF.cxx
@@ -899,18 +899,24 @@ Bool_t AliAODPidHF::IsExcluded(AliAODTrack *track, Int_t labelTrack, Double_t ns
 //-----------------------------
 void AliAODPidHF::SetPriorsHistos(TString priorFileName){
   // Set histograms with priors
+
   for (Int_t ispecies=0;ispecies<AliPID::kSPECIES;++ispecies) {
     if(fPriorsH[ispecies]) delete fPriorsH[ispecies];
     TString nt ="name";
     nt+="_prior_";
     nt+=AliPID::ParticleName(ispecies);
-    fPriorsH[ispecies]=new TH1F(nt,nt,100,0,10);
   }
+  TDirectory *current = gDirectory;
   TFile *priorFile=TFile::Open(priorFileName);
   if (priorFile) {
-    fPriorsH[AliPID::kProton]->Add(static_cast<TH1*>(priorFile->Get("priors3step9")));
-    fPriorsH[AliPID::kKaon  ]->Add(static_cast<TH1*>(priorFile->Get("priors2step9")));
-    fPriorsH[AliPID::kPion  ]->Add(static_cast<TH1*>(priorFile->Get("priors1step9")));
+    TH1F* h3=static_cast<TH1F*>(priorFile->Get("priors3step9"));
+    TH1F* h2=static_cast<TH1F*>(priorFile->Get("priors2step9"));
+    TH1F* h1=static_cast<TH1F*>(priorFile->Get("priors1step9"));
+    current->cd();
+    fPriorsH[AliPID::kProton] = new TH1F(*h3);
+    fPriorsH[AliPID::kKaon  ] = new TH1F(*h2);
+    fPriorsH[AliPID::kPion  ] = new TH1F(*h1);
+    priorFile->Close();
     delete priorFile;
     TF1 *salt=new TF1("salt","1.e-10",0,10);
     fPriorsH[AliPID::kProton]->Add(salt);

diff --git a/PWGHF/vertexingHF/AliAODPidHF.h b/PWGHF/vertexingHF/AliAODPidHF.h
index 2e93f353e5bf7ac472c4aad467a07180c5e9ce2a..b55f698d352666c346f81765db327a4adee13352 100644 (file)
--- a/PWGHF/vertexingHF/AliAODPidHF.h
+++ b/PWGHF/vertexingHF/AliAODPidHF.h
@@ -70,6 +70,19 @@ class AliAODPidHF : public AliAODPid{
  void SetCombDetectors(ECombDetectors pidComb) {
    fCombDetectors=pidComb;
   }
+ void SetPionPriorHisto(TH1F* histo){
+   if(fPriorsH[AliPID::kPion]) delete fPriorsH[AliPID::kPion];
+   fPriorsH[AliPID::kPion] = new TH1F(*histo);
+ }
+ void SetKaonPriorHisto(TH1F* histo){
+   if(fPriorsH[AliPID::kKaon]) delete fPriorsH[AliPID::kKaon];
+   fPriorsH[AliPID::kKaon] = new TH1F(*histo);
+ }
+ void SetProtonPriorHisto(TH1F* histo){
+   if(fPriorsH[AliPID::kProton]) delete fPriorsH[AliPID::kProton];
+   fPriorsH[AliPID::kProton] = new TH1F(*histo);
+ }
+
  
  //Getters