ESDCheck/AliAnalysisTaskPt.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 /* $Id$ */
  17
  18 //_________________________________________________________________________
  19 // A test analysis task to check the pt of tracks distribution in simulated data
  20 //
  21 //*-- Panos
  22 //////////////////////////////////////////////////////////////////////////////
  23
  24 #include <TCanvas.h>
  25 #include <TChain.h>
  26 #include <TFile.h>
  27 #include <TH1.h>
  28 #include <TROOT.h>
  29 #include <TSystem.h>
  30 #include <TString.h>
  31
  32 #include "AliAnalysisTaskPt.h"
  33 #include "AliESD.h"
  34 #include "AliLog.h"
  35
  36 //________________________________________________________________________
  37 AliAnalysisTaskPt::AliAnalysisTaskPt(const char *name) :
  38   AliAnalysisTask(name,""),
  39   fChain(0),
  40   fESD(0),
  41   fhPt(0),
  42   fOutputContainer(0)
  43 {
  44   // Constructor.
  45   // Input slot #0 works with an Ntuple
  46   DefineInput(0, TChain::Class());
  47   // Output slot #0 writes into a TH1 container
  48   DefineOutput(0, TObjArray::Class());
  49 }
  50
  51 //________________________________________________________________________
  52 void AliAnalysisTaskPt::ConnectInputData(Option_t *)
  53 {
  54   // Initialisation of branch container and histograms
  55
  56   AliInfo(Form("*** Initialization of %s", GetName())) ;
  57
  58   // Get input data
  59   fChain = dynamic_cast<TChain *>(GetInputData(0)) ;
  60   if (!fChain) {
  61     AliError(Form("Input 0 for %s not found\n", GetName()));
  62     return ;
  63   }
  64
  65   // One should first check if the branch address was taken by some other task
  66   char ** address = (char **)GetBranchAddress(0, "ESD");
  67   if (address) {
  68     fESD = (AliESD*)(*address);
  69   } else {
  70     fESD = new AliESD();
  71     SetBranchAddress(0, "ESD", &fESD);
  72   }
  73 }
  74
  75 //________________________________________________________________________
  76 void AliAnalysisTaskPt::CreateOutputObjects()
  77 {
  78   // create histograms
  79
  80   OpenFile(0) ;
  81
  82   fhPt = new TH1F("fhPt","This is the Pt distribution",15,0.1,3.1);
  83   fhPt->SetStats(kTRUE);
  84   fhPt->GetXaxis()->SetTitle("P_{T} [GeV]");
  85   fhPt->GetYaxis()->SetTitle("#frac{dN}{dP_{T}}");
  86   fhPt->GetXaxis()->SetTitleColor(1);
  87   fhPt->SetMarkerStyle(kFullCircle);
  88
  89   // create output container
  90
  91   fOutputContainer = new TObjArray(1) ;
  92   fOutputContainer->SetName(GetName()) ;
  93
  94   fOutputContainer->AddAt(fhPt, 0) ;
  95 }
  96
  97 //________________________________________________________________________
  98 void AliAnalysisTaskPt::Exec(Option_t *)
  99 {
 100   // Processing of one event
 101
 102   Long64_t entry = fChain->GetReadEntry() ;
 103
 104   if (!fESD) {
 105     AliError("fESD is not connected to the input!") ;
 106     return ;
 107   }
 108
 109   if ( !((entry-1)%100) )
 110     AliInfo(Form("%s ----> Processing event # %lld",  (dynamic_cast<TChain *>(fChain))->GetFile()->GetName(), entry)) ;
 111
 112   //************************  Pt tracks  *************************************
 113
 114   for(Int_t iTracks = 0; iTracks < fESD->GetNumberOfTracks(); iTracks++) {
 115     AliESDtrack * ESDTrack = fESD->GetTrack(iTracks);
 116     Double_t momentum[3];
 117     ESDTrack->GetPxPyPz(momentum);
 118     Double_t Pt = sqrt(pow(momentum[0],2) + pow(momentum[1],2));
 119     fhPt->Fill(Pt);
 120   }//track loop
 121
 122   PostData(0, fOutputContainer);
 123 }
 124
 125 //________________________________________________________________________
 126 void AliAnalysisTaskPt::Terminate(Option_t *)
 127 {
 128   // Processing when the event loop is ended
 129
 130   Bool_t problem=kFALSE;
 131
 132   AliInfo(Form(" *** %s Report:", GetName())) ;
 133
 134   TCanvas *c1 = new TCanvas("c1","Pt",10,10,310,310);
 135   c1->SetFillColor(10);
 136   c1->SetHighLightColor(10);
 137
 138   c1->cd(1)->SetLeftMargin(0.15);
 139   c1->cd(1)->SetBottomMargin(0.15);
 140   if (fhPt->GetMaximum() > 0 )
 141     c1->cd(1)->SetLogy();
 142   fOutputContainer = (TObjArray*)GetOutputData(0);
 143   fhPt = (TH1F*)fOutputContainer->At(0);
 144   if (fhPt) fhPt->DrawCopy("E");
 145
 146   c1->Print("TracksPt.eps");
 147
 148   char line[1024] ;
 149   sprintf(line, ".!tar -zcf %s.tar.gz *.eps", GetName()) ;
 150   gROOT->ProcessLine(line);
 151   sprintf(line, ".!rm -fR *.eps");
 152   gROOT->ProcessLine(line);
 153
 154   AliInfo(Form("!!! All the eps files are in %s.tar.gz !!!", GetName())) ;
 155
 156   TString report ;
 157   if(problem)
 158     report="Problems found, please check!!!";
 159   else
 160     report="OK";
 161
 162   AliInfo(Form("*** %s Summary Report: %s \n",GetName(), report.Data())) ;
 163 }