ITS/AliITSpidESD2.cxx

   1 /**************************************************************************
   2  * Copyright(c) 2005-2007, 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 // ITS PID class --- method # 2                                          //
  20 //                                                                       //
  21 //                                                                       //
  22 //The PID is based on the likelihood of all the four ITS' layers,        //
  23 //without using the truncated mean for the dE/dx. The response           //
  24 //functions for each layer are convoluted Landau-Gaussian functions.     //
  25 // Origin: Elena Bruna bruna@to.infn.it, Massimo Masera masera@to.infn.it//
  26 //-----------------------------------------------------------------------//
  27
  28 #include "AliITSpidESD2.h"
  29 #include "AliESDEvent.h"
  30 #include "AliESDtrack.h"
  31 #include "AliITStrackV2.h"
  32 #include "AliITSRecPoint.h"
  33 #include "AliITStrackerMI.h"
  34 #include "AliITSLoader.h"
  35 #include "AliITSPident.h"
  36 #include "AliITSSteerPid.h"
  37 #include "AliLog.h"
  38
  39 ClassImp(AliITSpidESD2)
  40 //_________________________________________________________________________
  41   AliITSpidESD2::AliITSpidESD2():AliITSpidESD(),
  42 fTracker(0),
  43 fSp(0)
  44 { //
  45   //  The main constructor
  46 }
  47 //_________________________________________________________________________
  48 AliITSpidESD2::AliITSpidESD2(AliITStrackerMI* tracker):AliITSpidESD(),
  49 fTracker(tracker),
  50 fSp(0)
  51 { //
  52   //  The main constructor
  53   fSp=new AliITSSteerPid();
  54   fSp->InitLayer();
  55 }
  56 //_________________________________________________________________________
  57 AliITSpidESD2::~AliITSpidESD2(){
  58   //destructor
  59
  60   if(fSp)delete fSp;
  61
  62 }
  63 //______________________________________________________________________
  64 AliITSpidESD2::AliITSpidESD2(const AliITSpidESD2 &ob) :AliITSpidESD(ob),
  65 fTracker(ob.fTracker),
  66 fSp(ob.fSp)
  67 {
  68   // Copy constructor
  69
  70 }
  71
  72 //______________________________________________________________________
  73 AliITSpidESD2& AliITSpidESD2::operator=(const AliITSpidESD2& ob ){
  74   // Assignment operator
  75   this->~AliITSpidESD2();
  76   new(this) AliITSpidESD2(ob);
  77   return *this;
  78 }
  79
  80 //_________________________________________________________________________
  81 Int_t AliITSpidESD2::MakePID(TTree *clustersTree, AliESDEvent *event)
  82 {
  83
  84   //
  85   //  This function calculates the "detector response" PID probabilities
  86   //
  87   Double_t xr,par[5];
  88   AliITStrackV2* track=0;
  89   fTracker->LoadClusters(clustersTree);
  90   printf("==== Landau Fit PID ITS ====== \n");
  91   Int_t ntrk=event->GetNumberOfTracks();
  92   Double_t momits;
  93   // for (Int_t i=0; i<ntrk; i++) {
  94   for (Int_t i=0; i<ntrk; i++) {
  95     AliESDtrack *esdtr=event->GetTrack(i);
  96     if ((esdtr->GetStatus()&AliESDtrack::kITSin )==0)
  97       if ((esdtr->GetStatus()&AliESDtrack::kITSout)==0) continue;
  98
  99     track = new AliITStrackV2(*esdtr);
 100     Double_t dEdxsignal=track->GetdEdx();
 101     track->GetExternalParameters(xr,par);
 102     if (par[4]!=0) {
 103       Float_t lamb=TMath::ATan(par[3]);
 104       momits=1/(TMath::Abs(par[4])*TMath::Cos(lamb));
 105     }
 106     else {
 107       AliWarning("Null particle momentum in ITS");
 108       momits = 0.;
 109     }
 110     Double_t snp=track->GetSnp();
 111     Double_t tgl=track->GetTgl();
 112     const Int_t kns=AliPID::kSPECIES;
 113     Double_t condprobfun[kns];
 114     for(Int_t ii=0;ii<kns;ii++)condprobfun[ii]=0;
 115     Int_t cluind[12];
 116     for(Int_t ii=0;ii<12;ii++){
 117       cluind[ii]=track->GetClusterIndex(ii);
 118     }
 119     AliITSRecPoint* cluarr[12];
 120     Float_t qclu[8],qclucorr[8],nlay[8];
 121     for(Int_t ii=0;ii<8;ii++){
 122       qclu[ii]=0;
 123       qclucorr[ii]=0;
 124       nlay[ii]=0;
 125     }
 126     Int_t jj=0;
 127     for(Int_t ij=0;ij<12;ij++){
 128       cluind[ij]=track->GetClusterIndex(ij);
 129       if(cluind[ij]>0){
 130         cluarr[ij]=(AliITSRecPoint*)fTracker->GetCluster(cluind[ij]);
 131         Int_t lay=cluarr[ij]->GetLayer();
 132         if(lay>1){//sdd+ssd only
 133           qclu[jj]=cluarr[ij]->GetQ();
 134           qclucorr[jj]=qclu[jj]*TMath::Sqrt((1-snp*snp)/(1+tgl*tgl));
 135           nlay[jj]=lay;
 136           jj++;
 137         }
 138         else qclucorr[jj]=-1;
 139       }
 140     }
 141
 142     Float_t prip=0.33;
 143     Float_t prik=0.33;
 144     Float_t pripi=0.33;
 145     Float_t prie=0.;
 146     AliITSPident mypid(momits,dEdxsignal,fSp,qclucorr,nlay,prip,prik,pripi,prie);
 147     condprobfun[0]=mypid.GetProdCondFunPi();//el --PID in the ITS does not distinguish among Pi,el,mu
 148     condprobfun[1]=mypid.GetProdCondFunPi();//mu
 149     condprobfun[2]=mypid.GetProdCondFunPi();//pi
 150     condprobfun[3]=mypid.GetProdCondFunK();//kaon
 151     condprobfun[4]=mypid.GetProdCondFunPro();//pro
 152
 153     esdtr->SetITSpid(condprobfun);
 154
 155     delete track;
 156   }
 157   fTracker->UnloadClusters();
 158   return 0;
 159 }