comment slow function

[u/mrichter/AliRoot.git] / PWGLF / SPECTRA / ChargedHadrons / dNdPt / AlidNdPtTask.cxx

/**************************************************************************\r
* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *\r
*                                                                        *\r
* Author: The ALICE Off-line Project.                                    *\r
* Contributors are mentioned in the code where appropriate.              *\r
*                                                                        *\r
* Permission to use, copy, modify and distribute this software and its   *\r
* documentation strictly for non-commercial purposes is hereby granted   *\r
* without fee, provided that the above copyright notice appears in all   *\r
* copies and that both the copyright notice and this permission notice   *\r
* appear in the supporting documentation. The authors make no claims     *\r
* about the suitability of this software for any purpose. It is          *\r
* provided "as is" without express or implied warranty.                  *\r
**************************************************************************/\r
\r
#include "iostream"\r
\r
#include "TChain.h"\r
#include "TTree.h"\r
#include "TH1F.h"\r
#include "TCanvas.h"\r
#include "TList.h"\r
#include "TFile.h"\r
\r
#include "AliAnalysisTask.h"\r
#include "AliAnalysisManager.h"\r
#include "AliESDEvent.h"\r
#include "AliESDInputHandler.h"\r
#include "AliESDVertex.h"\r
#include "AliTracker.h"\r
#include "AliGeomManager.h"\r
\r
#include "AliCentrality.h"\r
#include "AliESDVZERO.h"\r
#include "AliMultiplicity.h"\r
\r
#include "AliESDtrackCuts.h"\r
#include "AliMCEventHandler.h"\r
#include "AlidNdPt.h"\r
#include "AlidNdPtEventCuts.h"\r
#include "AlidNdPtAcceptanceCuts.h"\r
\r
#include "AlidNdPtTask.h"\r
\r
using namespace std;\r
\r
ClassImp(AlidNdPtTask)\r
\r
//_____________________________________________________________________________\r
AlidNdPtTask::AlidNdPtTask(const char *name) \r
  : AliAnalysisTaskSE(name)\r
  , fESD(0)\r
  , fMC(0)\r
  , fOutput(0)\r
  , fPitList(0)\r
  , fCompList(0)\r
  , fUseMCInfo(kFALSE)\r
{\r
  // Constructor\r
\r
  // Define input and output slots here\r
  DefineOutput(1, TList::Class());\r
\r
  // create the list for comparison objects\r
  fCompList = new TList;\r
}\r
\r
//_____________________________________________________________________________\r
AlidNdPtTask::~AlidNdPtTask()\r
{\r
  if(fOutput) delete fOutput;  fOutput =0; \r
  if(fCompList) delete fCompList;  fCompList =0; \r
}\r
\r
//____________________________________________________________________________\r
Bool_t AlidNdPtTask::Notify()\r
{\r
  static Int_t count = 0;\r
  count++;\r
  //Printf("Processing %d. file: %s", count, ((TTree*) GetInputData(0))->GetCurrentFile()->GetName());\r
  TTree *chain = (TChain*)GetInputData(0);\r
  if(chain)\r
    Printf("Processing %d. file: %s", count, chain->GetCurrentFile()->GetName());\r
\r
  /*\r
  TChain *chain = (TChain*)GetInputData(0);\r
  AliESDInputHandler *esdH = dynamic_cast<AliESDInputHandler*> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());\r
  if (!esdH) {\r
    Printf("ERROR: Could not get ESDInputHandler");\r
    return kFALSE;\r
  } else {\r
    if(chain)\r
    Printf("chain->GetCurrentFile()->GetName() %s", chain->GetCurrentFile()->GetName());\r
  }\r
  */\r
\r
return kTRUE;\r
}\r
\r
//_____________________________________________________________________________\r
Bool_t AlidNdPtTask::AddAnalysisObject(AlidNdPt *pObj) \r
{\r
  // add analysis object to the list\r
  if(pObj == 0) {\r
    Printf("ERROR: Could not add comparison object");\r
    return kFALSE;\r
  }\r
\r
  // add object to the list\r
  fCompList->AddLast(pObj);\r
       \r
return kTRUE;\r
}\r
\r
//_____________________________________________________________________________\r
void AlidNdPtTask::UserCreateOutputObjects()\r
{\r
  // Create histograms\r
  // Called once\r
\r
  OpenFile(1, "RECREATE");\r
\r
  //\r
  // create output list\r
  //\r
  fOutput = new TList;\r
  fOutput->SetOwner();\r
  fPitList = fOutput->MakeIterator();\r
\r
  // add dNdPt analysis objects to the output\r
  AlidNdPt *pObj=0;\r
  Int_t count=0;\r
  TIterator *pitCompList = fCompList->MakeIterator();\r
  pitCompList->Reset();\r
  while(( pObj = (AlidNdPt *)pitCompList->Next()) != NULL) {\r
    fOutput->Add(pObj);\r
    count++;\r
  }\r
  Printf("UserCreateOutputObjects(): Number of output objects: %d \n", count);\r
\r
  PostData(1, fOutput);\r
}\r
\r
//_____________________________________________________________________________\r
void AlidNdPtTask::UserExec(Option_t *) \r
{\r
  //\r
  // Called for each event\r
  //\r
\r
  // ESD event\r
  fESD = (AliESDEvent*) (InputEvent());\r
  if (!fESD) {\r
    Printf("ERROR: ESD event not available");\r
    return;\r
  }\r
\r
  // MC event\r
  if(fUseMCInfo) {\r
    fMC = MCEvent();\r
    if (!fMC) {\r
      Printf("ERROR: MC event not available");\r
      return;\r
    }\r
  }\r
\r
\r
    AlidNdPt *pObj = 0;\r
    fPitList->Reset();\r
    while((pObj = (AlidNdPt *)fPitList->Next()) != NULL) {\r
      pObj->Process(fESD,fMC);\r
    }\r
\r
\r
  // Post output data.\r
  PostData(1, fOutput);\r
}\r
\r
//_____________________________________________________________________________\r
void AlidNdPtTask::FinishTaskOutput() \r
{\r
  //\r
  // Called one at the end \r
  // locally on working node\r
  //\r
   // check output data\r
  fOutput = dynamic_cast<TList*> (GetOutputData(1));\r
  if (!fOutput) {\r
    Printf("ERROR: AlidNdPtTask::FinishTaskOutput(): Output data not avaiable GetOutputData(1)==0x0 ..." );\r
    return;\r
  }\r
\r
  AlidNdPt* pObj=0;\r
  TIterator* itOut = fOutput->MakeIterator();\r
  itOut->Reset();\r
  while(( pObj = dynamic_cast<AlidNdPt*>(itOut->Next())) != NULL) {\r
    if(pObj->GetAnalyseOutput()) { \r
      pObj->Analyse();\r
    }\r
  }\r
\r
  // Post output data.\r
  PostData(1, fOutput);\r
}\r
\r
//_____________________________________________________________________________\r
void AlidNdPtTask::Terminate(Option_t *) \r
{\r
  // Called one at the end \r
  \r
  // check output data\r
  fOutput = dynamic_cast<TList*> (GetOutputData(1));\r
  if (!fOutput) {\r
    Printf("ERROR: AlidNdPtTask::Terminate(): Output data not avaiable GetOutputData(0)==0x0 ..." );\r
    return;\r
  }\r
}\r