TPC/AliTPCCalibKrTask.cxx

   1 /**************************************************************************
   2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3  *                                                                        *
   4  * Author: The ALICE Off-line Project.                                    *
   5  * Contributors are mentioned in the code where appropriate.              *
   6  *                                                                        *
   7  * Permission to use, copy, modify and distribute this software and its   *
   8  * documentation strictly for non-commercial purposes is hereby granted   *
   9  * without fee, provided that the above copyright notice appears in all   *
  10  * copies and that both the copyright notice and this permission notice   *
  11  * appear in the supporting documentation. The authors make no claims     *
  12  * about the suitability of this software for any purpose. It is          *
  13  * provided "as is" without express or implied warranty.                  *
  14  **************************************************************************/
  15
  16 //----------------------------------------------------------------------------
  17 // The AliTPCCalibKrTask class description (TPC Kr calibration).
  18 // The AliTPCCalibKrTask loops over tree of TPC Kr clusters and fills AliTPCCalibKr
  19 // calibration component.
  20 //
  21 // As the input it requires the tree with reconstructed Kr clusters (AliTPCclusterKr objects).
  22 // The AliTPCCalibKr output calibration component contains an array of TH3F histograms which can be stored
  23 // in the ouptut file.
  24 //
  25 //  Author: Jacek Otwinowski (J.Otwinowski@gsi.de)
  26
  27 /*
  28
  29 // Usage example:
  30 //
  31
  32 // -- Load toolkit
  33 gSystem->AddIncludePath("-I$ALICE_ROOT/TPC/macros");
  34 gROOT->LoadMacro("$ALICE_ROOT/TPC/macros/AliXRDPROOFtoolkit.cxx+");
  35 AliXRDPROOFtoolkit tool;
  36
  37 // -- Make chain of files
  38 TChain * chain = tool.MakeChain("KrClusters.txt","Kr","",1000,0);
  39
  40 // -- Create the analysis manager
  41 AliAnalysisManager *mgr = new AliAnalysisManager("testAnalysis");
  42
  43 // -- Calibration component
  44 AliTPCCalibKr *calibObj = new AliTPCCalibKr;
  45 calibObj->SetIrocHistogram(200,100,6000);
  46 calibObj->SetOrocHistogram(200,100,5500);
  47 calibObj->Init();
  48
  49 // -- Add task
  50 AliTPCCalibKrTask *task = new AliTPCCalibKrTask;
  51 task->SetInputChain(chain);
  52 task->SetTPCCalibKr(calibObj);
  53 mgr->AddTask(task);
  54
  55 // -- Attach input
  56 cInput  = mgr->CreateContainer("cInput", TChain::Class(), AliAnalysisManager::kInputContainer);
  57 mgr->ConnectInput(task, 0, cInput);
  58
  59 // -- Attach output
  60 cOutput = mgr->CreateContainer("cOutput", TList::Class(), AliAnalysisManager::kOutputContainer,"outHistFile.root");
  61 mgr->ConnectOutput(task, 0, cOutput);
  62
  63 // -- Run analysis
  64 mgr->InitAnalysis();
  65 mgr->PrintStatus();
  66 mgr->StartAnalysis("local", chain);
  67
  68 */
  69
  70 // system includes
  71 #include <cstdio>
  72 using namespace std;
  73
  74 //Root includes
  75 #include <TH1F.h>
  76 #include <TH2F.h>
  77 #include <TH3F.h>
  78 #include <TString.h>
  79 #include <TMath.h>
  80 #include <TF1.h>
  81 #include <TRandom.h>
  82 #include <TDirectory.h>
  83 #include <TFile.h>
  84 //AliRoot includes
  85 #include "AliRawReader.h"
  86 #include "AliRawReaderRoot.h"
  87 #include "AliRawReaderDate.h"
  88 #include "AliTPCRawStream.h"
  89 #include "AliTPCCalROC.h"
  90 #include "AliTPCCalPad.h"
  91 #include "AliTPCROC.h"
  92 #include "AliMathBase.h"
  93 #include "TTreeStream.h"
  94
  95 //date
  96 #include "event.h"
  97
  98 //header file
  99 #include "AliTPCCalibKr.h"
 100 #include "AliTPCCalibKrTask.h"
 101
 102 Int_t AliTPCCalibKrTask::fEvtNumber = 0;
 103
 104 ClassImp(AliTPCCalibKrTask)
 105
 106 AliTPCCalibKrTask::AliTPCCalibKrTask(const char *name) :
 107   AliAnalysisTask(name,""),
 108   fClustKr(0),
 109   fTPCCalibKr(0),
 110   fTree(0),
 111   fOutput(0)
 112 {
 113   //
 114   // default constructor
 115   //
 116
 117   DefineInput(0, TChain::Class());
 118   DefineOutput(0, TList::Class());
 119 }
 120
 121 //_____________________________________________________________________
 122 AliTPCCalibKrTask::~AliTPCCalibKrTask()
 123 {
 124   //
 125   // destructor
 126   //
 127   if(fOutput) fOutput->Delete();
 128   delete fOutput; fOutput = 0;
 129 }
 130
 131 //_____________________________________________________________________
 132 void AliTPCCalibKrTask::ConnectInputData(Option_t *)
 133 {
 134   // Connect input data
 135   // Called once
 136
 137   fTree = dynamic_cast<TTree*> (GetInputData(0));
 138
 139   if(!fTree) {
 140    Printf("ERROR: Could not read chain from input");
 141    return;
 142   }
 143   else {
 144    fTree->SetBranchStatus("*",1);
 145    fTree->SetBranchStatus("Cl.fCluster",0);
 146   }
 147
 148   // set branch address
 149   if(!fTree->GetBranch("Cl.")) {
 150     Printf("ERROR: Could not get Cl. branch from input");
 151   } else {
 152    fTree->GetBranch("Cl.")->SetAddress(&fClustKr);
 153   }
 154 }
 155
 156 //_____________________________________________________________________
 157 void AliTPCCalibKrTask::CreateOutputObjects()
 158 {
 159   // create object to the output
 160   fOutput = new TList;
 161   fOutput->SetOwner(); // is owner of the fTPCCalibKr objects
 162
 163   if(fTPCCalibKr) fOutput->Add(fTPCCalibKr);
 164   //fTPCCalibKr = new AliTPCCalibKr;
 165   //if(fTPCCalibKr) fOutput->Add(fTPCCalibKr);
 166   else
 167      Printf("WARNING: AliTPCCalibKr is not added to the output");
 168 }
 169
 170 //_____________________________________________________________________
 171 Bool_t AliTPCCalibKrTask::ReadEntry(Int_t evt)
 172 {
 173   // read entry
 174   Long64_t centry = fTree->LoadTree(evt);
 175   if(centry < 0) return kFALSE;
 176
 177   if(!fTree->GetBranch("Cl."))
 178   {
 179     Printf("ERROR: Could not get Cl. branch from input");
 180         return kFALSE;
 181   } else {
 182    fTree->GetBranch("Cl.")->SetAddress(&fClustKr);
 183   }
 184
 185   fTree->GetEntry(evt);
 186
 187 return kTRUE;
 188 }
 189
 190 //_____________________________________________________________________
 191 void AliTPCCalibKrTask::Exec(Option_t *)
 192 {
 193   // Main loop
 194   // Called for each event
 195
 196   // read entry
 197   if(fClustKr) delete fClustKr; fClustKr=0;
 198   Bool_t status = ReadEntry(fEvtNumber);
 199   if(status==kTRUE)
 200   {
 201           // Process output objects
 202       if(fClustKr) fTPCCalibKr->Process(fClustKr);
 203   }
 204
 205   if( !( fEvtNumber % 100000) ) {
 206     cout << fEvtNumber << endl; }
 207
 208   fEvtNumber++;
 209
 210   // Post output data.
 211   PostData(0, fOutput);
 212 }
 213
 214 //_____________________________________________________________________
 215 void AliTPCCalibKrTask::Terminate(Option_t *)
 216 {
 217   // Called once at the end of the event loop
 218   cout << "Terminate " << endl;
 219
 220   fOutput = dynamic_cast<TList*> (GetOutputData(0));
 221   if (!fOutput) {
 222     Printf("ERROR: fOutput not available");
 223     return;
 224   }
 225
 226   fTPCCalibKr = dynamic_cast<AliTPCCalibKr*> (fOutput->FindObject("AliTPCCalibKr"));
 227   if (!fTPCCalibKr) {
 228     Printf("WARNING: AliTPCCalibKr not available");
 229     return;
 230   }
 231
 232   for(int i=0; i<72; ++i) {
 233          if( fTPCCalibKr->IsCSide(i) == kTRUE )
 234            printf("C side chamber: %d, 3D histo entries: %10.f \n",i,fTPCCalibKr->GetHistoKr(i)->GetEntries());
 235
 236          if( fTPCCalibKr->IsCSide(i) == kFALSE )
 237            printf("A side chamber: %d, 3D histo entries: %10.f \n",i,fTPCCalibKr->GetHistoKr(i)->GetEntries());
 238   }
 239 }