New branches with run number and OCDB file ITS/Align/Data in spTree

[u/mrichter/AliRoot.git] / PWG1 / ITS / AliAlignmentDataFilterITS.cxx

diff --git a/PWG1/ITS/AliAlignmentDataFilterITS.cxx b/PWG1/ITS/AliAlignmentDataFilterITS.cxx
index 11a5996774ad0619d38d7674bdaafbe1d15a7467..ee8703c7166788415c6676d7926cea2f91b45d1e 100644 (file)
--- a/PWG1/ITS/AliAlignmentDataFilterITS.cxx
+++ b/PWG1/ITS/AliAlignmentDataFilterITS.cxx
@@ -285,10 +285,14 @@ void AliAlignmentDataFilterITS::CreateOutputObjects()
 
   fspTree = new TTree("spTree","Tree with ITS track points");
   const AliTrackPointArray *array = 0;
-  Float_t curv,curverr;
+  Float_t curv,curverr,runNumber;
+  const TObjString *itsaligndata = 0;
   fspTree->Branch("SP","AliTrackPointArray",&array);
   fspTree->Branch("curv",&curv);
   fspTree->Branch("curverr",&curverr);
+  fspTree->Branch("run",&runNumber);
+  fspTree->Branch("ITSAlignData",&itsaligndata);
+  
 
   return;
 }
@@ -365,10 +369,29 @@ void AliAlignmentDataFilterITS::FilterCosmic(const AliESDEvent *esd)
 
   // Set branch addresses for space points tree
   AliTrackPointArray *arrayForTree=0;
-  Float_t curv,curverr;
+  Float_t curv,curverr,runNumber;
+  TObjString *itsaligndata=0;
   fspTree->SetBranchAddress("SP",&arrayForTree);
   fspTree->SetBranchAddress("curv",&curv);
   fspTree->SetBranchAddress("curverr",&curverr);
+  fspTree->SetBranchAddress("run",&runNumber);
+  fspTree->SetBranchAddress("ITSAlignData",&itsaligndata);
+
+
+  runNumber = (Float_t)esd->GetRunNumber();
+
+  TTree* esdTree = dynamic_cast<TTree*> (GetInputData(0));
+  // Get the list of OCDB objects used for reco 
+  TList *cdbList = (TList*)(esdTree->GetTree()->GetUserInfo())->FindObject("cdbList");
+  TIter iter2(cdbList);             
+  TObjString* cdbEntry=0;
+  TString cdbEntryString;
+  while((cdbEntry =(TObjString*)(iter2.Next()))) {
+  cdbEntryString = cdbEntry->GetString();
+  if(cdbEntryString.Contains("ITS/Align/Data")) 
+    itsaligndata = new TObjString(*cdbEntry);
+  }     
+
 
   TString triggeredClass = fESD->GetFiredTriggerClasses(); 
   if(fOnlySPDFO && !triggeredClass.Contains("C0SCO-ABCE-NOPF-CENT")) return;
@@ -687,10 +710,28 @@ void AliAlignmentDataFilterITS::FilterCollision(const AliESDEvent *esd)
 
   // Set branch addresses for space points tree
   AliTrackPointArray *arrayForTree=0;
-  Float_t curv,curverr;
+  Float_t curv,curverr,runNumber;
+  TObjString *itsaligndata=0;
   fspTree->SetBranchAddress("SP",&arrayForTree);
   fspTree->SetBranchAddress("curv",&curv);
   fspTree->SetBranchAddress("curverr",&curverr);
+  fspTree->SetBranchAddress("run",&runNumber);
+  fspTree->SetBranchAddress("ITSAlignData",&itsaligndata);
+
+
+  runNumber = (Float_t)esd->GetRunNumber();
+
+  TTree* esdTree = dynamic_cast<TTree*> (GetInputData(0));
+  // Get the list of OCDB objects used for reco 
+  TList *cdbList = (TList*)(esdTree->GetTree()->GetUserInfo())->FindObject("cdbList");
+  TIter iter2(cdbList);             
+  TObjString* cdbEntry=0;
+  TString cdbEntryString;
+  while((cdbEntry =(TObjString*)(iter2.Next()))) {
+  cdbEntryString = cdbEntry->GetString();
+  if(cdbEntryString.Contains("ITS/Align/Data")) 
+    itsaligndata = new TObjString(*cdbEntry);
+  }     
 
   Int_t ntracks = esd->GetNumberOfTracks();