PWG0/dNdPt/AlidNdPtTask.cxx

   1 /**************************************************************************\r
   2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *\r
   3 *                                                                        *\r
   4 * Author: The ALICE Off-line Project.                                    *\r
   5 * Contributors are mentioned in the code where appropriate.              *\r
   6 *                                                                        *\r
   7 * Permission to use, copy, modify and distribute this software and its   *\r
   8 * documentation strictly for non-commercial purposes is hereby granted   *\r
   9 * without fee, provided that the above copyright notice appears in all   *\r
  10 * copies and that both the copyright notice and this permission notice   *\r
  11 * appear in the supporting documentation. The authors make no claims     *\r
  12 * about the suitability of this software for any purpose. It is          *\r
  13 * provided "as is" without express or implied warranty.                  *\r
  14 **************************************************************************/\r
  15 \r
  16 #include "iostream"\r
  17 \r
  18 #include "TChain.h"\r
  19 #include "TTree.h"\r
  20 #include "TH1F.h"\r
  21 #include "TCanvas.h"\r
  22 #include "TList.h"\r
  23 #include "TFile.h"\r
  24 \r
  25 #include "AliAnalysisTask.h"\r
  26 #include "AliAnalysisManager.h"\r
  27 #include "AliESDEvent.h"\r
  28 #include "AliESDInputHandler.h"\r
  29 #include "AliESDVertex.h"\r
  30 #include "AliTracker.h"\r
  31 #include "AliGeomManager.h"\r
  32 \r
  33 #include "AliESDtrackCuts.h"\r
  34 #include "AliMCEventHandler.h"\r
  35 #include "dNdPt/AlidNdPt.h"\r
  36 #include "dNdPt/AlidNdPtEventCuts.h"\r
  37 #include "dNdPt/AlidNdPtAcceptanceCuts.h"\r
  38 \r
  39 #include "dNdPt/AlidNdPtTask.h"\r
  40 \r
  41 using namespace std;\r
  42 \r
  43 ClassImp(AlidNdPtTask)\r
  44 \r
  45 //_____________________________________________________________________________\r
  46 AlidNdPtTask::AlidNdPtTask(const char *name) \r
  47   : AliAnalysisTask(name, "dNdPt analysis")\r
  48   , fESD(0)\r
  49   , fMC(0)\r
  50   , fOutput(0)\r
  51   , fPitList(0)\r
  52   , fCompList(0)\r
  53   , fUseMCInfo(kFALSE)\r
  54 {\r
  55   // Constructor\r
  56 \r
  57   // Define input and output slots here\r
  58   DefineInput(0, TChain::Class());\r
  59   DefineOutput(0, TList::Class());\r
  60 \r
  61   // create the list for comparison objects\r
  62   fCompList = new TList;\r
  63 }\r
  64 \r
  65 //_____________________________________________________________________________\r
  66 AlidNdPtTask::~AlidNdPtTask()\r
  67 {\r
  68   if(fOutput) delete fOutput;  fOutput =0; \r
  69 }\r
  70 \r
  71 //____________________________________________________________________________\r
  72 Bool_t AlidNdPtTask::Notify()\r
  73 {\r
  74   static Int_t count = 0;\r
  75   count++;\r
  76   //Printf("Processing %d. file: %s", count, ((TTree*) GetInputData(0))->GetCurrentFile()->GetName());\r
  77   TTree *chain = (TChain*)GetInputData(0);\r
  78   if(chain)\r
  79     Printf("Processing %d. file: %s", count, chain->GetCurrentFile()->GetName());\r
  80 \r
  81   /*\r
  82   TChain *chain = (TChain*)GetInputData(0);\r
  83   AliESDInputHandler *esdH = dynamic_cast<AliESDInputHandler*> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());\r
  84   if (!esdH) {\r
  85     Printf("ERROR: Could not get ESDInputHandler");\r
  86     return kFALSE;\r
  87   } else {\r
  88     if(chain)\r
  89     Printf("chain->GetCurrentFile()->GetName() %s", chain->GetCurrentFile()->GetName());\r
  90   }\r
  91   */\r
  92 \r
  93 return kTRUE;\r
  94 }\r
  95 \r
  96 //_____________________________________________________________________________\r
  97 void AlidNdPtTask::ConnectInputData(Option_t *) \r
  98 {\r
  99 // connect input data \r
 100 // called once\r
 101 \r
 102   TTree *tree = dynamic_cast<TTree*> (GetInputData(0));\r
 103   if (!tree) {\r
 104     Printf("ERROR: Could not read chain from input slot 0");\r
 105     return;\r
 106   }\r
 107 \r
 108   AliESDInputHandler *esdH = dynamic_cast<AliESDInputHandler*> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());\r
 109 \r
 110   if (!esdH) {\r
 111     Printf("ERROR: Could not get ESDInputHandler");\r
 112     return;\r
 113   } else {\r
 114     fESD = esdH->GetEvent();\r
 115 \r
 116     // Enable only the needed branches\r
 117     //esdH->SetActiveBranches("AliESDHeader Vertex Tracks");\r
 118   }\r
 119 \r
 120   if (fUseMCInfo) {\r
 121     AliMCEventHandler* eventHandler = dynamic_cast<AliMCEventHandler*> (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());\r
 122     if (!eventHandler) {\r
 123       Printf("ERROR: Could not retrieve MC event handler");\r
 124       return;\r
 125     } else { \r
 126       fMC = eventHandler->MCEvent();\r
 127     }\r
 128   }\r
 129 }\r
 130 \r
 131 //_____________________________________________________________________________\r
 132 Bool_t AlidNdPtTask::AddAnalysisObject(AlidNdPt *pObj) \r
 133 {\r
 134   // add analysis object to the list\r
 135   if(pObj == 0) {\r
 136     Printf("ERROR: Could not add comparison object");\r
 137     return kFALSE;\r
 138   }\r
 139 \r
 140   // add object to the list\r
 141   fCompList->AddLast(pObj);\r
 142        \r
 143 return kTRUE;\r
 144 }\r
 145 \r
 146 //_____________________________________________________________________________\r
 147 void AlidNdPtTask::CreateOutputObjects()\r
 148 {\r
 149   // Create histograms\r
 150   // Called once\r
 151 \r
 152   OpenFile(0, "RECREATE");\r
 153 \r
 154   //\r
 155   // create output list\r
 156   //\r
 157   fOutput = new TList;\r
 158   fOutput->SetOwner();\r
 159   fPitList = fOutput->MakeIterator();\r
 160 \r
 161   // add dNdPt analysis objects to the output\r
 162   AlidNdPt *pObj=0;\r
 163   Int_t count=0;\r
 164   TIterator *pitCompList = fCompList->MakeIterator();\r
 165   pitCompList->Reset();\r
 166   while(( pObj = (AlidNdPt *)pitCompList->Next()) != NULL) {\r
 167     fOutput->Add(pObj);\r
 168     count++;\r
 169   }\r
 170   Printf("CreateOutputObjects(): Number of output comparison objects: %d \n", count);\r
 171 }\r
 172 \r
 173 //_____________________________________________________________________________\r
 174 void AlidNdPtTask::Exec(Option_t *) \r
 175 {\r
 176   // Called for each event\r
 177   if (!fESD) {\r
 178     Printf("ERROR: ESD event not available");\r
 179     return;\r
 180   }\r
 181 \r
 182   // Process analysis\r
 183   AlidNdPt *pObj = 0;\r
 184   fPitList->Reset();\r
 185   while((pObj = (AlidNdPt *)fPitList->Next()) != NULL) {\r
 186     pObj->Process(fESD,fMC);\r
 187   }\r
 188 \r
 189   // Post output data.\r
 190   PostData(0, fOutput);\r
 191 }\r
 192 \r
 193 //_____________________________________________________________________________\r
 194 void AlidNdPtTask::Terminate(Option_t *) \r
 195 {\r
 196   // Called one at the end \r
 197   \r
 198   // check output data\r
 199   fOutput = dynamic_cast<TList*> (GetOutputData(0));\r
 200   if (!fOutput) {\r
 201     Printf("ERROR: AlidNdPtTask::Terminate(): Output data not avaiable GetOutputData(0)==0x0 ..." );\r
 202     return;\r
 203   }\r
 204 }\r