Changed pt cut value (Andrea)

diff --git a/PWG3/AliAnalysisVertexingHFTest.C b/PWG3/AliAnalysisVertexingHFTest.C
index 8a3afd79639f465746aab6026320eb2c1a5993ca..5ae11f367a75d72a171458e69ef3089ef9aa55db 100644 (file)
--- a/PWG3/AliAnalysisVertexingHFTest.C
+++ b/PWG3/AliAnalysisVertexingHFTest.C
@@ -1,31 +1,33 @@
 //--------------------------------------------------------------------------
 // Test macro for reconstruction of heavy-flavour vertexing candidates
 //
+// mode: 0 -> input ESD, output AOD
+//       1 -> input ESD, output TTree
+//
 //     Andrea Dainese, andrea.dainese@lnl.infn.it
 //--------------------------------------------------------------------------
 
-void AliAnalysisVertexingHFTest(Int_t evFirst=0,
-                               Int_t evLast=0,
-                               TString infile="AliESDs.root",
-                               TString outfile="VertexingHF.root") {
+void AliAnalysisVertexingHFTest(Int_t mode=0)
+{
   
-  gSystem->Load("libANALYSIS");
+  gSystem->Load("libANALYSIS.so");
   gSystem->Load("libAOD.so");
   gSystem->Load("libPWG3base.so");
 
   AliAnalysisVertexingHF *vHF = new AliAnalysisVertexingHF();
+  //vHF->SetUseTRef();
   //--- switch-off candidates finding (default: all on)
   //vHF->SetD0toKpiOff();
-  //vHF->SetJPSItoEleOff();
+  vHF->SetJPSItoEleOff();
   vHF->Set3ProngOff();
   vHF->Set4ProngOff();
   //--- secondary vertex with KF?
-  //vHF->SetSecVtxWithKF();
+  vHF->SetSecVtxWithKF();
   //--- set cuts for single-track selection
   vHF->SetITSrefitRequired();
   vHF->SetBothSPDNotRequired();
   vHF->SetMinITSCls(5);
-  vHF->SetMinPtCut(0.1);
+  vHF->SetMinPtCut(0.);
   vHF->SetMind0Cut(0.);
   //--- set cuts for candidates selection
   //vHF->SetD0toKpiCuts(); 
@@ -42,41 +44,47 @@ void AliAnalysisVertexingHFTest(Int_t evFirst=0,
   //--- check the settings
   vHF->PrintStatus();
   //--- verbose
-  vHF->SetDebug(1);
-
-  TTree *trees = new TTree[2];
-  AliAODRecoDecayHF2Prong *rd2=0;
-  //AliAODRecoDecayHF3Prong *rd3=0;
-  //AliAODRecoDecayHF4Prong *rd4=0;
-  trees[0].Branch("D0toKpi","AliAODRecoDecayHF2Prong",&rd2);
-  trees[1].Branch("JPSItoEle","AliAODRecoDecayHF2Prong",&rd2);
-  //trees[2].Branch("Charmto3Prong","AliAODRecoDecayHF3Prong",&rd3);
-  //trees[3].Branch("D0to4Prong","AliAODRecoDecayHF4Prong",&rd4);
-
-  TFile *inesd = new TFile(infile.Data());
-  AliESDEvent *esd = new AliESDEvent();
-  TTree *esdTree = (TTree*)inesd->Get("esdTree");
-  esd->ReadFromTree(esdTree);
+  //vHF->SetDebug(1);
 
-  if(esdTree->GetEntries()<evLast) evLast=esdTree->GetEntries()-1;
-  for(Int_t i=evFirst; i<=evLast; i++) {
-    esdTree->GetEvent(i);
-    vHF->FindCandidates(esd,trees);
-    //if(i==evLast) i=evFirst;
+  switch(mode) {
+  case 0:
+    vHF->FindCandidatesESDtoAOD();
+    break;
+  case 1:
+    TTree *trees = new TTree[1];
+    AliAODRecoDecayHF2Prong *rd2=0;
+    // AliAODRecoDecayHF3Prong *rd3=0;
+    //AliAODRecoDecayHF4Prong *rd4=0;
+    trees[0].Branch("D0toKpi","AliAODRecoDecayHF2Prong",&rd2);
+    //trees[1].Branch("JPSItoEle","AliAODRecoDecayHF2Prong",&rd2);
+    //trees[2].Branch("Charmto3Prong","AliAODRecoDecayHF3Prong",&rd3);
+    //trees[3].Branch("D0to4Prong","AliAODRecoDecayHF4Prong",&rd4);
+    
+    TFile *inesd = new TFile("AliESDs.root");
+    AliESDEvent *esd = new AliESDEvent();
+    TTree *esdTree = (TTree*)inesd->Get("esdTree");
+    esd->ReadFromTree(esdTree);
+    
+    for(Int_t i=0; i<esdTree->GetEntries(); i++) {
+      esdTree->GetEvent(i);
+      vHF->FindCandidates(esd,trees);
+      //if(i==evLast) i=evFirst;
+    }
+    
+    delete esdTree;
+    inesd->Close();
+    delete inesd;
+    
+    // Write trees with candidates
+    TFile *fout = new TFile("AliAnalysisVertexingHF.root","recreate");
+    trees[0].Write("TreeD0toKpi");
+    //trees[1].Write("TreeJPSItoEle");
+    //trees[2].Write("TreeCharm3Prong");
+    //trees[3].Write("TreeD0to4Prong");
+    fout->Close();
+    delete fout;
+    break;
   }
 
-  delete esdTree;
-  inesd->Close();
-  delete inesd;
-
-  // Write trees with candidates
-  TFile *fout = new TFile("AliAnalysisVertexingHF.root","recreate");
-  trees[0].Write("TreeD0toKpi");
-  trees[1].Write("TreeJPSItoEle");
-  //trees[2].Write("TreeCharm3Prong");
-  //trees[3].Write("TreeD0to4Prong");
-  fout->Close();
-  delete fout;
-
   return;
 }