FMD/analysis/AliFMDAnalysisTaskCollector.cxx

   1
   2 #include <TROOT.h>
   3 #include <TSystem.h>
   4 #include <TInterpreter.h>
   5 #include <TChain.h>
   6 #include <TFile.h>
   7 #include <TList.h>
   8 #include <iostream>
   9
  10 #include "AliFMDAnalysisTaskCollector.h"
  11 #include "AliAnalysisManager.h"
  12 #include "AliESDFMD.h"
  13 #include "AliESDEvent.h"
  14 #include "AliAODEvent.h"
  15 #include "AliAODHandler.h"
  16 #include "AliMCEventHandler.h"
  17 #include "AliStack.h"
  18 #include "AliESDVertex.h"
  19 #include "AliFMDAnaParameters.h"
  20 //#include "AliFMDGeometry.h"
  21 ClassImp(AliFMDAnalysisTaskCollector)
  22
  23
  24 AliFMDAnalysisTaskCollector::AliFMDAnalysisTaskCollector()
  25 : fDebug(0),
  26   fChain(0x0),
  27   fESD(0x0),
  28   fOutputList(0),
  29   fArray(0),
  30   fEdistHist(0),
  31   fZvtxDist(0)
  32 {
  33   // Default constructor
  34   DefineInput (0, TChain::Class());
  35   DefineOutput(0, TList::Class());
  36 }
  37 //____________________________________________________________________
  38 AliFMDAnalysisTaskCollector::AliFMDAnalysisTaskCollector(const char* name):
  39     AliAnalysisTask(name, "AnalysisTaskFMD"),
  40     fDebug(0),
  41     fChain(0x0),
  42     fESD(0x0),
  43     fOutputList(0),
  44     fArray(0),
  45     fEdistHist(0),
  46     fZvtxDist(0)
  47 {
  48   // Default constructor
  49   DefineInput (0, TChain::Class());
  50   DefineOutput(0, TList::Class());
  51 }
  52 //____________________________________________________________________
  53 void AliFMDAnalysisTaskCollector::CreateOutputObjects()
  54 {
  55   // Create the output container
  56   printf("AnalysisTaskFMD::CreateOutPutData() \n");
  57
  58   fOutputList = new TList();//(TList*)GetOutputData(0);
  59   AliFMDAnaParameters* pars = AliFMDAnaParameters::Instance();
  60
  61   fArray     = new TObjArray();
  62   fArray->SetName("FMD");
  63   fArray->SetOwner();
  64   TH1F* hEdist = 0;
  65
  66   for(Int_t nEta = 0; nEta <= pars->GetNetaBins()+1; nEta++) {
  67     TObjArray* etaArray = new TObjArray();
  68     fArray->AddAtAndExpand(etaArray,nEta);
  69     for(Int_t det =1; det<=3;det++)
  70       {
  71         TObjArray* detArray = new TObjArray();
  72         detArray->SetName(Form("FMD%d",det));
  73         etaArray->AddAtAndExpand(detArray,det);
  74         Int_t nRings = (det==1 ? 1 : 2);
  75         for(Int_t ring = 0;ring<nRings;ring++)
  76           {
  77             Char_t ringChar = (ring == 0 ? 'I' : 'O');
  78             hEdist = new TH1F(Form("FMD%d%c_etabin%d",det,ringChar,nEta),Form("FMD%d%c_etabin%d",det,ringChar,nEta),200,0,6);
  79             hEdist->SetXTitle("#Delta E / E_{MIP}");
  80             fOutputList->Add(hEdist);
  81             detArray->AddAtAndExpand(hEdist,ring);
  82           }
  83       }
  84
  85   }
  86
  87   fZvtxDist  = new TH1F("ZvtxDist","Vertex distribution",100,-30,30);
  88   fZvtxDist->SetXTitle("z vertex");
  89   //fOutputList->Add(fArray);
  90   fOutputList->Add(fZvtxDist);
  91 }
  92 //____________________________________________________________________
  93 void AliFMDAnalysisTaskCollector::Init()
  94 {
  95   // Initialization
  96   printf("AnalysisTaskFMD::Init() \n");
  97
  98
  99 }
 100 //____________________________________________________________________
 101 void AliFMDAnalysisTaskCollector::ConnectInputData(Option_t */*option*/)
 102 {
 103   fChain = (TChain*)GetInputData(0);
 104   fESD = new AliESDEvent();
 105   fESD->ReadFromTree(fChain);
 106
 107 }
 108 //____________________________________________________________________
 109 void AliFMDAnalysisTaskCollector::Exec(Option_t */*option*/)
 110 {
 111   AliESD* old = fESD->GetAliESDOld();
 112   if (old) {
 113     fESD->CopyFromOldESD();
 114   }
 115   AliFMDAnaParameters* pars = AliFMDAnaParameters::Instance();
 116
 117   Double_t vertex[3];
 118
 119   GetVertex(vertex);
 120   //if(vertex[0] == 0 && vertex[1] == 0 && vertex[2] == 0)
 121   //  return;
 122   fZvtxDist->Fill(vertex[2]);
 123
 124   if(TMath::Abs(vertex[2]) > pars->GetVtxCutZ())
 125     return;
 126
 127   AliESDFMD* fmd = fESD->GetFMDData();
 128   if (!fmd) return;
 129
 130
 131
 132   for(UShort_t det=1;det<=3;det++) {
 133
 134
 135     Int_t nRings = (det==1 ? 1 : 2);
 136     for (UShort_t ir = 0; ir < nRings; ir++) {
 137
 138       Char_t   ring = (ir == 0 ? 'I' : 'O');
 139       UShort_t nsec = (ir == 0 ? 20  : 40);
 140       UShort_t nstr = (ir == 0 ? 512 : 256);
 141       TH2F* hBg = pars->GetBackgroundCorrection(det,ring,0);
 142
 143       for(UShort_t sec =0; sec < nsec;  sec++)  {
 144         for(UShort_t strip = 0; strip < nstr; strip++) {
 145
 146
 147           Float_t mult = fmd->Multiplicity(det,ring,sec,strip);
 148           if(mult == AliESDFMD::kInvalidMult || mult == 0) continue;
 149           //AliFMDGeometry* geo = AliFMDGeometry::Instance();
 150
 151           //  Double_t x,y,z;
 152           //geo->Detector2XYZ(det,ring,sec,strip,x,y,z);
 153
 154           // Double_t r = TMath::Sqrt(x*x+y*y);
 155
 156           // Double_t z_real      = z-vertex[2];
 157           // Double_t theta       = TMath::ATan2(r,z_real);
 158           // std::cout<<"From EtaFromStrip "<<theta<<std::endl;
 159           //Double_t eta         =  -1*TMath::Log(TMath::Tan(0.5*theta));
 160           Float_t eta = pars->GetEtaFromStrip(det,ring,sec,strip,vertex[2]);
 161           //Float_t eta = fmd->Eta(det,ring,sec,strip);
 162           Int_t nEta = hBg->GetXaxis()->FindBin(eta);
 163
 164           TObjArray* etaArray = (TObjArray*)fArray->At(nEta);
 165           TObjArray* detArray = (TObjArray*)etaArray->At(det);
 166           TH1F* Edist = (TH1F*)detArray->At(ir);
 167
 168           Edist->Fill(mult);
 169
 170         }
 171       }
 172     }
 173   }
 174
 175   PostData(0, fOutputList);
 176
 177 }
 178 //____________________________________________________________________
 179 void AliFMDAnalysisTaskCollector::Terminate(Option_t */*option*/)
 180 {
 181   /*
 182   for(UShort_t det=1;det<=3;det++) {
 183     TObjArray* detArray = (TObjArray*)fArray->At(det);
 184     Int_t nRings = (det==1 ? 1 : 2);
 185     for (UShort_t ir = 0; ir < nRings; ir++) {
 186       TH1F* hEdist = (TH1F*)detArray->At(ir);
 187       hEdist->SetAxisRange(0.4,hEdist->GetXaxis()->GetXmax());
 188       Float_t max = hEdist->GetBinCenter(hEdist->GetMaximumBin());
 189       hEdist->Fit("landau","","",max-0.1,2*max);
 190     }
 191   }
 192   */
 193 }
 194 //_____________________________________________________________________
 195 void AliFMDAnalysisTaskCollector::GetVertex(Double_t* vertexXYZ)
 196 {
 197   const AliESDVertex* vertex = 0;
 198   vertex = fESD->GetPrimaryVertex();
 199   if (!vertex || (vertexXYZ[0] == 0 && vertexXYZ[1] == 0 && vertexXYZ[2] == 0))
 200     vertex = fESD->GetPrimaryVertexSPD();
 201   if (!vertex || (vertexXYZ[0] == 0 && vertexXYZ[1] == 0 && vertexXYZ[2] == 0))
 202     vertex = fESD->GetPrimaryVertexTPC();
 203
 204   if (!vertex || (vertexXYZ[0] == 0 && vertexXYZ[1] == 0 && vertexXYZ[2] == 0))
 205     vertex = fESD->GetVertex();
 206
 207   if (vertex && (vertexXYZ[0] != 0 || vertexXYZ[1] != 0 || vertexXYZ[2] != 0)) {
 208     vertex->GetXYZ(vertexXYZ);
 209     //std::cout<<vertex->GetName()<<"   "<< vertex->GetTitle() <<"   "<< vertex->GetZv()<<std::endl;
 210     return;
 211   }
 212   else if (fESD->GetESDTZERO()) {
 213     vertexXYZ[0] = 0;
 214     vertexXYZ[1] = 0;
 215     vertexXYZ[2] = fESD->GetT0zVertex();
 216
 217     return;
 218   }
 219
 220   return;
 221 }
 222 //____________________________________________________________________
 223 //
 224 // EOF
 225 //