ATO-97 Function to connect CalPads trees into the common Tree as friend trees. naming...

diff --git a/TPC/Base/AliTPCcalibDButil.cxx b/TPC/Base/AliTPCcalibDButil.cxx
index 0d59e7737e0034bb7c35e81b940c57a51a18ff98..140692e8e357b5e99a8b8f10cabf3a00bf728844 100644 (file)
--- a/TPC/Base/AliTPCcalibDButil.cxx
+++ b/TPC/Base/AliTPCcalibDButil.cxx
@@ -57,6 +57,7 @@
 #include "AliRelAlignerKalman.h"
 #include "TTree.h"
 #include "TROOT.h"
+#include "TKey.h"
 
 const Float_t kAlmost0=1.e-30;
 
@@ -3313,11 +3314,85 @@ TTree* AliTPCcalibDButil::ConnectPulserTrees(TString baseDir, TTree *tMain)
                     TString::Format("(%s.CETmean.fElements-%s.CETmean_LTM)",
                                     friendName.Data(),friendName.Data()).Data());    
     tMain->SetAlias((friendName+".CETmean_MedianDelta").Data(),
-                    TString::Format("(%s.CETmean.fElements-s.CETmean_Median)",
+                    TString::Format("(%s.CETmean.fElements-%s.CETmean_Median)",
                                     friendName.Data(),friendName.Data()).Data());
     tMain->SetAlias((friendName+".CETmean_MeanDelta").Data(),
                     TString::Format("(%s.CETmean.fElements-%s.CETmean_Mean)",
                                     friendName.Data(),friendName.Data()).Data());        
   }
+  return tMain;
 }  
   
+
+TTree* AliTPCcalibDButil::ConnectDistortionTrees(TString baseDir, TTree *tMain){
+  //
+  // baseDir:   Base directory with Distortion information
+  // TTrees are added to the base tree as a friend tree
+  // If base tree not provide - first tree from list is used as base
+  //  
+  // === add the calibDistortion trees ======================================
+  //TString inputTreesDistortionCalib       = gSystem->GetFromPipe(Form("ls %s/calibDistortion/20*/*.tree.root",baseDir.Data()));
+  // TString baseDir="$NOTES/reconstruction/distortionFit/"; TTree *tMain=0;
+
+
+  TString inputTreesDistortionCalib       = gSystem->GetFromPipe(Form("find  %s -iname \"calibTimeResHisto.root\"",baseDir.Data()));
+  TObjArray *arrInputTreesDistortionCalib = inputTreesDistortionCalib.Tokenize("\n");  
+  //
+  for (Int_t itree=0; itree<arrInputTreesDistortionCalib->GetEntriesFast(); ++itree) {
+    TFile *finput= TFile::Open(arrInputTreesDistortionCalib->At(itree)->GetName());
+    TString strFile=arrInputTreesDistortionCalib->At(itree)->GetName();
+    TObjArray *path=strFile.Tokenize("/");
+    Int_t plength=path->GetEntries();
+    if (!finput) continue;
+    TList* list = finput->GetListOfKeys();
+    Int_t nkeys=list->GetEntries();
+    for (Int_t ikey=0; ikey<nkeys; ikey++){
+      TKey * key = (TKey*)list->At(ikey);
+      if (strstr(key->GetClassName(),"TTree")==0) continue;
+      TTree * tree  = dynamic_cast<TTree*>(finput->Get(list->At(ikey)->GetName()));
+      if (!tree) continue;
+      TString friendName=TString::Format("%s.%s.%s",path->At(plength-3)->GetName(),path->At(plength-2)->GetName(), tree->GetName()); 
+      ::Info("AliTPCcalibDButil::ConnectDistortionTrees","%s",friendName.Data());  
+      if (tMain==0) tMain=tree;
+      tMain->AddFriend(tree,friendName.Data());  
+    }
+  }
+  //  tMain->SetAlias("");
+  return tMain;
+}  
+  
+
+//_____________________________________________________________________________________
+TTree* AliTPCcalibDButil::ConnectCalPadTrees(TString baseDir, TString pattern, TTree *tMain)
+{
+  //
+  // baseDir:   Base directory with per Pad information
+  // TTrees are added to the base tree as a friend tree
+  /* Example
+     TString baseDir="/hera/alice/fsozzi/summarymaps/calib/";  // prefix directory with calibration
+     TString pattern="QA root"; 
+  */
+  //  
+  // === add the calibPulser trees ======================================
+  TString inputTreesCalPad       = gSystem->GetFromPipe(Form("ls %s/%s",baseDir.Data(), pattern.Data()));
+  TObjArray *arrInputTreesCalPad = inputTreesCalPad.Tokenize("\n");
+  //
+  for (Int_t itree=0; itree<arrInputTreesCalPad->GetEntriesFast(); ++itree) {
+    TFile *fin2 = TFile::Open(arrInputTreesCalPad->At(itree)->GetName());
+    TTree *tin = (TTree*)fin2->Get("calPads");
+    gROOT->cd();
+    TString friendName=arrInputTreesCalPad->At(itree)->GetName();
+    friendName.ReplaceAll(baseDir.Data(),"");
+    friendName.ReplaceAll("/",".");
+    friendName.ReplaceAll(".root","");
+    friendName.ReplaceAll("^/","");
+    printf("%s\n", friendName.Data());
+    ::Info("AliTPCcalibDButil::ConnectCalPadTrees","%s",friendName.Data());  
+    if (tMain==0) tMain=tin;
+    tMain->AddFriend(tin,friendName.Data());  
+  }
+  return tMain;
+} 
+  
+
+

diff --git a/TPC/Base/AliTPCcalibDButil.h b/TPC/Base/AliTPCcalibDButil.h
index 12b7a755dfa71f3f7b1c021333f082b80d75f10a..8f408d4692bef8965b614033a8553153d29927ef 100644 (file)
--- a/TPC/Base/AliTPCcalibDButil.h
+++ b/TPC/Base/AliTPCcalibDButil.h
@@ -181,7 +181,9 @@ public:
   static TTree* ConnectGainTrees(TString baseDir);
   static TTree* ConnectPulserTrees(TString baseDir, TTree * tree=0);
   //static TTree* ConnectNoiseTrees(TString baseDir,  TTree * tree=0); 
-  
+  static TTree* ConnectDistortionTrees(TString baseDir, TTree *tMain);
+  static  TTree* ConnectCalPadTrees(TString baseDir, TString pattern,  TTree *tMain);
+
 private:
   AliTPCcalibDB *fCalibDB;            //pointer to calibDB object
   AliTPCCalPad  *fPadNoise;           //noise information