copy TRD performance train to PWG1

[u/mrichter/AliRoot.git] / PWG1 / AliComparisonTask.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
//------------------------------------------------------------------------------\r
// Implementation of the AliComparisonTask class. It compares properties of the \r
// reconstructed and MC particle tracks under several conditions. \r
// As the input it requires the tree with AliRecInfo and AliMCInfo branches. Such\r
// tree can be prepared in advance by runing AliGenInfoMaker and then AliRecInfoMaker\r
// (details in description of these classes).\r
//\r
// The comparison output objects deriving from AliComparisonObject \r
// (e.g. AliComparisonRes, AliComparisonEff, AliComparisonDEdxA, AliComparisonDCA ...) \r
// are stored in the output file (details in description of these classes).\r
// \r
// Author: J.Otwinowski 04/02/2008 \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 "AliMagF.h"\r
#include "AliTracker.h"\r
#include "AliGeomManager.h"\r
\r
#include "AliMCInfo.h"\r
#include "AliESDRecInfo.h"\r
#include "AliMCInfoCuts.h"\r
#include "AliRecInfoCuts.h"\r
#include "AliComparisonRes.h"\r
#include "AliComparisonEff.h"\r
#include "AliComparisonDEdx.h"\r
#include "AliComparisonDCA.h"\r
#include "AliComparisonObject.h"\r
#include "AliComparisonTask.h"\r
\r
using namespace std;\r
\r
ClassImp(AliComparisonTask)\r
\r
Int_t AliComparisonTask::fEvtNumber = 0;\r
\r
//_____________________________________________________________________________\r
AliComparisonTask::AliComparisonTask(const char *name) \r
  : AliAnalysisTask(name, "")\r
  , fTree(0)\r
  , fInfoMC(0)\r
  , fInfoRC(0)\r
  , fOutput(0)\r
  , fPitList(0)\r
  , fCompList(0)\r
{\r
  // Constructor\r
\r
  // Define input and output slots here\r
  DefineInput(0, TChain::Class());\r
  DefineOutput(0, TList::Class());\r
\r
  // create the list for comparison objects\r
  fCompList = new TList;\r
}\r
\r
//_____________________________________________________________________________\r
AliComparisonTask::~AliComparisonTask()\r
{\r
  if(fOutput)   delete fOutput;  fOutput =0; \r
  if(fCompList)   delete fCompList;  fCompList =0; \r
}\r
\r
//_____________________________________________________________________________\r
void AliComparisonTask::ConnectInputData(Option_t *) \r
{\r
  // Connect input data \r
  // Called once\r
\r
  fTree = dynamic_cast<TTree*> (GetInputData(0));\r
  if (!fTree) {\r
    Printf("ERROR: Could not read chain from input slot 0");\r
  } else {\r
    fTree->SetBranchStatus("*",1);\r
  }\r
\r
  if(fTree->GetBranch("MC") &&  fTree->GetBranch("RC")) {\r
    fTree->GetBranch("MC")->SetAddress(&fInfoMC);\r
    fTree->GetBranch("RC")->SetAddress(&fInfoRC);\r
  } else {\r
      Printf("ERROR: Could not get MC and RC branches");\r
  }\r
}\r
\r
//_____________________________________________________________________________\r
Bool_t AliComparisonTask::AddComparisonObject(AliComparisonObject *pObj) \r
{\r
  // add comparison 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 AliComparisonTask::CreateOutputObjects()\r
{\r
  // Create histograms\r
  // Called once\r
\r
  // create output list\r
  fOutput = new TList;\r
  fOutput->SetOwner();\r
  fPitList = fOutput->MakeIterator();\r
\r
  AliComparisonObject *pObj=0;\r
  Int_t count=0;\r
\r
  // add comparison objects to the output\r
  TIterator *pitCompList = fCompList->MakeIterator();\r
  pitCompList->Reset();\r
  while(( pObj = (AliComparisonObject *)pitCompList->Next()) != NULL) {\r
    fOutput->Add(pObj);\r
	count++;\r
  }\r
  Printf("CreateOutputObjects(): Number of output comparison objects: %d \n", count);\r
}\r
\r
//_____________________________________________________________________________\r
Bool_t AliComparisonTask::ReadEntry(Int_t evt) \r
{\r
// Read entry from the tree\r
  Long64_t centry = fTree->LoadTree(evt);\r
  if(centry < 0) return kFALSE;\r
\r
  if(fTree->GetBranch("MC") &&  fTree->GetBranch("RC")) {\r
    fTree->GetBranch("MC")->SetAddress(&fInfoMC);\r
    fTree->GetBranch("RC")->SetAddress(&fInfoRC);\r
  } else {\r
      Printf("ERROR: Could not get MC and RC branches");\r
	  return kFALSE;\r
  }\r
  fTree->GetEntry(evt);\r
\r
return kTRUE;\r
}\r
//_____________________________________________________________________________\r
void AliComparisonTask::Exec(Option_t *) \r
{\r
  // Main loop\r
  // Called for each event\r
\r
  AliComparisonObject *pObj=0;\r
\r
  if (!fInfoMC && !fInfoRC) {\r
    Printf("ERROR: fInfoMC && fInfoRC not available");\r
    return;\r
  }\r
\r
  // Process comparison\r
  Bool_t status = ReadEntry(fEvtNumber);\r
  if(status == kTRUE) \r
  {\r
    fPitList->Reset();\r
    while(( pObj = (AliComparisonObject *)fPitList->Next()) != NULL) {\r
       pObj->Exec(fInfoMC,fInfoRC);\r
    }\r
  }\r
\r
  if( !( fEvtNumber % 10000) ) { \r
    cout << fEvtNumber << endl;\r
  }\r
  fEvtNumber++;\r
\r
  // Post output data.\r
  PostData(0, fOutput);\r
}\r
\r
//_____________________________________________________________________________\r
void AliComparisonTask::Terminate(Option_t *) \r
{\r
  // Called one at the end \r
  \r
  // check output data\r
  fOutput = dynamic_cast<TList*> (GetOutputData(0));\r
  if (!fOutput) {\r
    Printf("ERROR: AliComparisonTask::Terminate(): Output data not avaiable GetOutputData(0)==0x0 ..." );\r
    return;\r
  }\r
}\r