ITS/UPGRADE/testITSU/runanLoc.C

   1 TChain* loadChain(const char* inpData, const char* chName="esdTree");
   2 const TObjArray* GetTrackingConditions(const char* recopar="ITS/Calib/RecoParam/Run0_999999999_v0_s0.root");
   3
   4 void runanLoc(const char* inpData="AliESDs.root", const char* outFile="tstPerfAn.root")
   5 {
   6   // runs analysis task on local data
   7   //
   8   gROOT->ProcessLine(".x LoadLibs.C(1)");
   9   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  10   if (!mgr) mgr = new AliAnalysisManager("Test reco analysis");
  11   AliAnalysisDataContainer *cinp = (AliAnalysisDataContainer*)mgr->GetCommonInputContainer();
  12   //
  13   AliESDInputHandler *esdInputHandler = dynamic_cast<AliESDInputHandler*>(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());
  14   if (!esdInputHandler) esdInputHandler = new AliESDInputHandlerRP();
  15   mgr->SetInputEventHandler(esdInputHandler);
  16   //
  17   AliMCEventHandler* mcInputHandler = dynamic_cast<AliMCEventHandler*>(AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());
  18   if (!mcInputHandler) mcInputHandler = new AliMCEventHandler();
  19   mgr->SetMCtruthEventHandler(mcInputHandler);
  20   mcInputHandler->SetPreReadMode(AliMCEventHandler::kLmPreRead);
  21   //
  22   gROOT->ProcessLine(".L AliTaskITSUPerf.cxx++");
  23   AliTaskITSUPerf* task = new AliTaskITSUPerf("tstAnalysis");
  24   TString outFString = outFile;
  25   if (outFString.IsNull()) outFString = "recAnOut.root";
  26   AliAnalysisDataContainer *coutput1 = mgr->CreateContainer("clist", TList::Class(),AliAnalysisManager::kOutputContainer,outFString.Data());
  27   mgr->AddTask(task);
  28   mgr->ConnectInput(task, 0,  mgr->GetCommonInputContainer());
  29   mgr->ConnectOutput(task,1,coutput1);
  30   task->SetUseMC(1);  // at the moment not irrelevant
  31   //
  32   // configure task
  33   task->SetEtaMin(-3);
  34   task->SetEtaMax(3);
  35   task->SetNPtBins(10);
  36   task->SetPtMax(2);
  37   //
  38   // optionally set reconstructability condition
  39   task->SetTrackingConditions(GetTrackingConditions());
  40   //
  41   TChain *chain = loadChain(inpData);
  42   if (!chain) return;
  43   mgr->InitAnalysis();
  44   mgr->StartAnalysis("local",chain);
  45   //
  46 }
  47
  48 const TObjArray* GetTrackingConditions(const char* recopar)
  49 {
  50   // load tracking conditions used in reco
  51   TString rpfstr = recopar;
  52   Bool_t fail = kFALSE;
  53   const TObjArray* trCondArr = 0;
  54   while(1) {
  55     TFile* rpf = 0;
  56     if (rpfstr.IsNull() || !(rpf=TFile::Open(rpfstr.Data()))) {fail = kTRUE; break;}
  57     AliCDBEntry* cdbe = (AliCDBEntry*)rpf->Get("AliCDBEntry");
  58     if (!cdbe) {fail = kTRUE; break;}
  59     TObjArray* recoParArr = (TObjArray*)cdbe->GetObject();
  60     if (!recoParArr) {fail = kTRUE; break;}
  61     AliITSURecoParam* recoPar = (AliITSURecoParam*) recoParArr->At(1);
  62     if (!recoPar) {fail = kTRUE; break;}
  63     trCondArr = recoPar->GetTrackingConditions();
  64     if (!trCondArr) {fail = kTRUE; break;}
  65     break;
  66   }
  67   if (fail) {printf("Tracking conditions must be extracted from valid recoparams, path is %s\n",recopar); exit(1);}
  68   //
  69   return trCondArr;
  70 }
  71
  72 //________________________________________________________________________________
  73 TChain* loadChain(const char* inpData, const char* chName)
  74 {
  75   TChain* chain = new TChain(chName);
  76   //
  77   TString inpDtStr = inpData;
  78   if (inpDtStr.EndsWith(".root")) {
  79     if (!inpDtStr.Contains("/")) inpDtStr = Form("%s/%s",gSystem->pwd(),inpDtStr.Data());
  80     printf("Setting %s as an input\n",inpDtStr.Data());
  81     chain->AddFile(inpDtStr.Data());
  82   }
  83   else {
  84     //
  85     ifstream inpf(inpData);
  86     if (!inpf.good()) {
  87       printf("Failed on input filename %s\n",inpData);
  88       return kFALSE;
  89     }
  90     //
  91     TString flName;
  92     flName.ReadLine(inpf);
  93     while ( !flName.IsNull() ) {
  94       flName = flName.Strip(TString::kBoth,' ');
  95       if (flName.BeginsWith("//") || flName.BeginsWith("#")) {flName.ReadLine(inpf); continue;}
  96       flName = flName.Strip(TString::kBoth,',');
  97       flName = flName.Strip(TString::kBoth,'"');
  98       if (!flName.Contains("/")) flName = Form("%s/%s",gSystem->pwd(),flName);
  99       printf("Adding %s\n",flName.Data());
 100       chain->AddFile(flName.Data());
 101       flName.ReadLine(inpf);
 102     }
 103   }
 104   //
 105   int n = chain->GetEntries();
 106   if (n<1) {
 107     printf("Obtained chain is empty\n");
 108     return 0;
 109   }
 110   else printf("Opened %s chain with %d entries\n",chName,n);
 111   return chain;
 112 }