[u/mrichter/AliRoot.git] / RICH / AliRICHTracker.cxx

#include "AliRICHTracker.h" //class header
#include "AliRICH.h"
#include "AliRICHRecon.h"
#include <AliESD.h>
#include <TVector3.h>
#include <TTree.h>          //EsdPrint() 
#include <TFile.h>          //EsdPrint()    
#include "AliRICHHelix.h"
#include <AliMagF.h>
#include <AliStack.h>
#include <TParticle.h>
#include <TMath.h>
#include <AliRun.h>
#include <TNtupleD.h>            //RecWithStack();
#include <AliTrackPointArray.h>  //GetTrackPoint()
#include <AliAlignObj.h>         //GetTrackPoint()
ClassImp(AliRICHTracker)
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
AliRICHTracker::AliRICHTracker():AliTracker()
{
// AliRICHTracker is created from AliReconstraction::Run() which invokes AliReconstraction::CreateTrackers() 
// which in turn invokes AliRICHReconstructor::CreateTracker(). 
// Note that this is done just once per session before AliReconstruction::Run() goes to events loop.
  AliRICHParam::Instance()->CdbRead(0,0); 
  for(Int_t i=0;i<5;i++)fErrPar[i]=0;
}
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++    
Bool_t AliRICHTracker::GetTrackPoint(Int_t idx, AliTrackPoint& point) const
{
// Interface callback methode invoked from AliReconstruction::WriteAlignmentData() to get position of MIP cluster in MARS associated to a current track.
// MIP cluster is reffered by index which is stored in AliESDtrack  ???????
// Arguments: idx- cluster index which is stored by RICH in AliESDtrack
//            point- reference to the object where to store the point     
//   Returns: status of operation  if FALSE then AliReconstruction::WriteAlignmentData() do not store this point to array of points for current track. 
  if(idx<0) return kFALSE; //no MIP cluster assigned to this track in PropagateBack()
  Int_t iCham=idx/1000000;
  Int_t iClu=idx%1000000;
  point.SetVolumeID(AliAlignObj::LayerToVolUID(AliAlignObj::kRICH,iCham-1));//layer and chamber number
  AliRICH *pRich=((AliRICH*)gAlice->GetDetector("RICH"));  
  AliRICHCluster *pClu=(AliRICHCluster*)pRich->Clus(iCham)->UncheckedAt(iClu);//get pointer to cluster
  TVector3 mars=AliRICHParam::Instance()->Lors2Mars(iCham,pClu->X(),pClu->Y());
  point.SetXYZ(mars.X(),mars.Y(),mars.Z());
  return kTRUE;
}
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Int_t AliRICHTracker::LoadClusters(TTree *pCluTree)
{
// Interface callback methode invoked from AliReconstruction::RunTracking() to load RICH clusters for RICH
// Arguments: pCluTree- pointer to clusters tree got by AliRICHLoader::LoadRecPoints("read") then AliRICHLoader::TreeR()
//   Returns: error code (currently ignored in AliReconstruction::RunTraking())    
  AliDebug(1,"Start.");  pCluTree->GetEntry(0);  AliDebug(1,"Stop."); return 0;
}
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Int_t AliRICHTracker::PropagateBack(AliESD *pESD)
{
// Interface callback methode invoked by AliRecontruction::RunTracking() during tracking after TOF. It's done just once per event
// Arguments: pESD - pointer to Event Summary Data class instance which contains a list of tracks
//   Returns: error code, 0 if no errors   
  Int_t iNtracks=pESD->GetNumberOfTracks();
  AliDebug(1,Form("Start with %i tracks",iNtracks));
  AliRICH *pRich=((AliRICH*)gAlice->GetDetector("RICH"));  
  AliRICHRecon recon;
  AliPID pid; // needed to retrive all the PID info
   
  for(Int_t iTrk=0;iTrk<iNtracks;iTrk++){//ESD tracks loop
    AliESDtrack *pTrack = pESD->GetTrack(iTrk);// get next reconstructed track    
    Double_t mom[3], pos[3];
    pTrack->GetPxPyPz(mom); TVector3 mom3(mom[0],mom[1],mom[2]);
    pTrack->GetXYZ(pos); TVector3 pos3(pos[0],pos[1],pos[2]);
    AliRICHHelix helix(pos3,mom3,(Int_t)pTrack->GetSign(),-0.1*GetBz()); //construct helix out of track running parameters  
     //Printf(" magnetic field %f charged %f\n",GetBz(),pTrack->GetSign()); helix.Print("Track");
    Int_t iChamber=helix.RichIntersect(AliRICHParam::Instance());    
    if(!iChamber) continue;                                         //no intersection with chambers, ignore this track go after the next one
  
    //find MIP cluster candidate (closest to track intersection point cluster with large enough QDC)
    Double_t    dR=9999,   dX=9999,   dY=9999;                     //distance between track-PC intersection point and current cluster
    Double_t mipDr=9999,mipDx=9999,mipDy=9999,mipX=9999,mipY=9999; //nearest cluster parameters
    Int_t   iMipId=-1;                         //index of this nearest cluster
    for(Int_t iClu=0;iClu<pRich->Clus(iChamber)->GetEntries();iClu++){//clusters loop for intersected chamber
      AliRICHCluster *pClu=(AliRICHCluster*)pRich->Clus(iChamber)->UncheckedAt(iClu);//get pointer to current cluster
      if(pClu->Q()<AliRICHParam::QthMIP()) continue; //to low QDC, go after another one
      pClu->DistXY(helix.PosPc(),dX,dY); dR=TMath::Sqrt(dX*dX+dY*dY); //get distance for current cluster
      if(dR<mipDr){iMipId=iClu; mipDr=dR; mipDx=dX; mipDy=dY; mipX=pClu->X(); mipY=pClu->Y();} //current cluster is closer, overwrite data for min cluster
    }//clusters loop for intersected chamber

    pTrack->SetRICHthetaPhi(helix.Ploc().Theta(),helix.Ploc().Phi()); //store track impact angles with respect to RICH planes
    pTrack->SetRICHdxdy(mipDx,mipDy);                                 //distance between track-PC intersection and closest cluster with Qdc>100
    pTrack->SetRICHmipXY(mipX,mipY);                                  //position of that closest cluster with Qdc>100
    
    if(iMipId==-1)                        {pTrack->SetRICHsignal(kMipQdcCut);  continue;} //no cluster with enough QDC found
    if(mipDr>AliRICHParam::DmatchMIP())   {pTrack->SetRICHsignal(kMipDistCut); continue;} //closest cluster with enough carge is still too far 
  
    pTrack->SetRICHcluster(iMipId+1000000*iChamber);                                //set mip cluster index
    pTrack->SetRICHsignal(recon.ThetaCerenkov(&helix,pRich->Clus(iChamber),iMipId));//search for mean Cerenkov angle for this track
    pTrack->SetRICHnclusters(iMipId);                                               //on return iMipId is number of photon clusters accepted in reconstruction

    AliDebug(1,Form("Ch=%i PC Intersection=(%5.2f,%5.2f) cm MIP cluster dist=(%5.2f,%5.2f)=%5.2f cm ThetaCkov=%f",
                     iChamber,helix.PosPc().X(),helix.PosPc().Y(),            mipDx,mipDy,mipDr,     pTrack->GetRICHsignal()));
    
//here comes PID calculations    
    if(pTrack->GetRICHsignal()>0) {
      AliDebug(1,Form("Start to assign the probabilities"));
      Double_t sigmaPID[AliPID::kSPECIES];  Double_t richPID[AliPID::kSPECIES];
      for (Int_t iPid=0;iPid<AliPID::kSPECIES;iPid++){//PID loop
        sigmaPID[iPid] = 0; fErrPar[iPid] = 0;
        for(Int_t iCkov=0;iCkov<pRich->Clus(iChamber)->GetEntries();iCkov++){//Ckov candidates loop ????????????? somehting fomr AliRICHRecon must be
          recon.SetPhotonIndex(iCkov);
          if(recon.GetPhotonFlag() == 2) {
            Double_t thetaCkov=0.6; //??????????????????
            Double_t   phiCkov=0.6; //??????????????????
            Double_t  thetaTrk=recon.GetTrackTheta();
            Double_t    phiTrk=(recon.GetPhiPoint()-recon.GetTrackPhi());
            Double_t     beta =pTrack->GetP()/TMath::Sqrt((pTrack->GetP()*pTrack->GetP()+pid.ParticleMass(iPid)*pid.ParticleMass(iPid)));
            Double_t   sigma2 = AliRICHParam::SigmaSinglePhotonFormula(thetaCkov,phiCkov,thetaTrk,phiTrk,beta);
            if(sigma2>0) sigmaPID[iPid] += 1/sigma2;
          }
        }//Ckov candidates loop
        
        if (sigmaPID[iPid]>0)
          sigmaPID[iPid] *= (Double_t)(iMipId-recon.GetPhotBKG())/(Double_t)(iMipId); // n total phots, m are background...the sigma are scaled..
          if(sigmaPID[iPid]>0) sigmaPID[iPid] = 1/TMath::Sqrt(sigmaPID[iPid])*0.001; // sigma from parametrization are in mrad...
          else                  sigmaPID[iPid] = 0;
          fErrPar[iPid]=sigmaPID[iPid];
        AliDebug(1,Form("sigma for %s is %f rad",AliPID::ParticleName(iPid),sigmaPID[iPid]));
      }
      CalcProb(pTrack->GetRICHsignal(),pTrack->GetP(),sigmaPID,richPID);
      pTrack->SetRICHpid(richPID);         
      AliDebug(1,Form("PROBABILITIES ---> %f - %f - %f - %f - %f",richPID[0],richPID[1],richPID[2],richPID[3],richPID[4]));
    }//if(pTrack->GetRICHsignal())    
  }//ESD tracks loop
  AliDebug(1,"Stop pattern recognition");
  return 0; // error code: 0=no error;
}//PropagateBack()
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
void AliRICHTracker::RecWithStack(TNtupleD *hn)
{
// Reconstruction for particles from STACK. This methode is to be used for RICH standalone when no other detectors are switched on, so normal tracking is not available.
// Arguments: hn- output ntuple where to store all variables
//   Returns: none       
  AliDebug(1,"Start.");  
  AliRICH *pRich=((AliRICH*)gAlice->GetDetector("RICH"));
  
//  pRich->GetLoader()->GetRunLoader()->LoadHeader();
  if(!pRich->GetLoader()->GetRunLoader()->TreeK()) pRich->GetLoader()->GetRunLoader()->LoadKinematics();
  AliStack *pStack =   pRich->GetLoader()->GetRunLoader()->Stack();
  if(!pStack) {AliDebug(1,Form("No STACK found in AliRoot"));return;}
  Int_t iNtracks=pStack->GetNtrack();
  AliDebug(1,Form(" Start reconstruction with %i track(s) from Stack",iNtracks));
  
  Double_t hnvec[20];
  
  Double_t b=GetFieldMap()->SolenoidField()/10;// magnetic field in Tesla
  AliDebug(1,Form("Start with simulated %i tracks in %f Tesla field",iNtracks,b));
  TVector3 x0(0,0,0); TVector3 p0(0,0,0);//tmp storage for AliRICHHelix
  

  if(pRich->GetLoader()->LoadRecPoints()) {AliDebug(1,Form("No clusters found in RICH"));return;}
  pRich->GetLoader()->TreeR()->GetEntry(0);

  AliRICHRecon recon;
  for(Int_t iTrackN=0;iTrackN<iNtracks;iTrackN++){//stack particles loop
    TParticle *pParticle = pStack->Particle(iTrackN);
    if(!pParticle) {AliDebug(1,Form("Not a valid TParticle pointer. Track skipped"));continue;}
    AliDebug(1,Form(" PDG code : %i",pParticle->GetPdgCode()));
//
// problem of PDG code of some extra particles to be solved!!!!!!!!!
//
// found problem! Look in TRD directory : codes from Fluka are :
//
//    if ((pdg_code == 10010020) ||
//        (pdg_code == 10010030) ||
//        (pdg_code == 50000050) ||
//        (pdg_code == 50000051) ||
//        (pdg_code == 10020040)) {
//
    if(pParticle->GetPdgCode()>=50000050||pParticle->GetPdgCode()==0||pParticle->GetPdgCode()>10000) {AliDebug(1,Form("A photon as track... Track skipped"));continue;}
    if(!pParticle->GetPDG()) continue;
//
// to be updated for us!!
//
    AliDebug(1,Form("Track %i is a %s with charge %i and momentum %f",
            iTrackN,pParticle->GetPDG()->GetName(),Int_t(pParticle->GetPDG()->Charge()),pParticle->P()));
//    if(pParticle->GetMother(0)!=-1) continue; //consider only primaries
    if(pParticle->GetPDG()->Charge()==0||TMath::Abs(Int_t(pParticle->GetPDG()->Charge()))!=3) continue; //to avoid photons from stack...
    hnvec[0]=pParticle->P();
    hnvec[1]=pParticle->GetPDG()->Charge();
    
    p0.SetMagThetaPhi(pParticle->P(),pParticle->Theta(),pParticle->Phi());   x0.SetXYZ(pParticle->Vx(),pParticle->Vy(),pParticle->Vz());
    AliRICHHelix helix(x0,p0,TMath::Sign(1,(Int_t)pParticle->GetPDG()->Charge()),b);   
    Int_t iChamber=helix.RichIntersect(AliRICHParam::Instance());        
    if(!iChamber) continue;// no intersection with RICH found
    
    hnvec[2]=helix.Ploc().Theta();
    hnvec[3]=helix.Ploc().Phi();
    hnvec[4]=helix.PosPc().X();
    hnvec[5]=helix.PosPc().Y();
    
    Double_t dX,dY,dR,dRmip=9999;      //min distance between clusters and track position on PC 
    Int_t iMipId=-1;       //index of that min distance cluster 
    for(Int_t iClu=0;iClu<pRich->Clus(iChamber)->GetEntries();iClu++){//clusters loop for intersected chamber
      AliRICHCluster *pClu=(AliRICHCluster*)pRich->Clus(iChamber)->UncheckedAt(iClu);//get pointer to current cluster
      pClu->DistXY(helix.PosPc(),dX,dY); dR=TMath::Sqrt(dX*dX+dY*dY);//ditance between current cluster and helix intersection with PC
      if(dR<dRmip){dRmip=dR; hnvec[6]=pClu->X();hnvec[7]=pClu->Y();hnvec[8]=pClu->Q();
                                                  iMipId=1000000*iChamber+iClu;}//find cluster nearest to the track 
      
    }//clusters loop for intersected chamber
    
    
    hnvec[9]=recon.ThetaCerenkov(&helix,pRich->Clus(iChamber),iMipId); //search for mean Cerenkov angle for this track
    hnvec[10]=iMipId;//on return from ThetaCerenkov() contains number of photon candidates accepted
    hnvec[11]=(Double_t)iMipId;
    hnvec[12]=(Double_t)iChamber;
    hnvec[13]=(Double_t)pParticle->GetPdgCode();
    if(hn) hn->Fill(hnvec);
    AliDebug(1,Form("FINAL Theta Cerenkov=%f",hnvec[9]));
  }//stack particles loop
  
  pRich->GetLoader()->UnloadRecPoints();
  AliDebug(1,"Stop.");  
}//RecWithStack
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
void AliRICHTracker::CalcProb(Double_t thetaCer,Double_t pmod, Double_t *sigmaPID, Double_t *richPID)
{
// Calculates probability to be a electron-muon-pion-kaon-proton
// from the given Cerenkov angle and momentum assuming no initial particle composition
// (i.e. apriory probability to be the particle of the given sort is the same for all sorts)
  Double_t height[AliPID::kSPECIES];Double_t totalHeight=0;
  Double_t thetaTh[AliPID::kSPECIES];
  for(Int_t iPart=0;iPart<AliPID::kSPECIES;iPart++){
    height[iPart]=0;
    Double_t mass = AliRICHParam::fgMass[iPart];
    Double_t refIndex=AliRICHParam::Instance()->IdxC6F14(AliRICHParam::EckovMean());
    Double_t cosThetaTh = TMath::Sqrt(mass*mass+pmod*pmod)/(refIndex*pmod);
    thetaTh[iPart]=0;
    if(cosThetaTh>=1) continue;
    thetaTh[iPart] = TMath::ACos(cosThetaTh);
//    Double_t sinThetaThNorm = TMath::Sin(thetaTh)/TMath::Sqrt(1-1/(refIndex*refIndex));
//    Double_t sigmaThetaTh = (0.014*(1/sinThetaThNorm-1) + 0.0043)*1.25;
//    height[iPart] = TMath::Gaus(thetaCer,thetaTh,sigmaThetaTh);
    if(sigmaPID[iPart]>0) height[iPart] = TMath::Gaus(thetaCer,thetaTh[iPart],sigmaPID[iPart],kTRUE);
    else                  height[iPart] = 0;
    totalHeight +=height[iPart];
    AliDebugClass(1,Form(" Particle %s with mass %f with height %f and thetaTH %f",AliPID::ParticleName(iPart),mass,height[iPart],thetaTh[iPart]));
    AliDebugClass(1,Form(" partial height %15.14f total height %15.14f",height[iPart],totalHeight));
  }
  if(totalHeight<1e-5) {for(Int_t iPart=0;iPart<AliPID::kSPECIES;iPart++)richPID[iPart]=1.0/AliPID::kSPECIES;return;}
  for(Int_t iPart=0;iPart<AliPID::kSPECIES;iPart++) richPID[iPart] = height[iPart]/totalHeight;
  Int_t iPartNear = TMath::LocMax(AliPID::kSPECIES,richPID);
  if(TMath::Abs(thetaCer-thetaTh[iPartNear])>5.*sigmaPID[iPartNear]) for(Int_t iPart=0;iPart<AliPID::kSPECIES;iPart++)richPID[iPart]=1.0/AliPID::kSPECIES;
  //last line is to check if the nearest thetacerenkov to the teorethical one is within 5 sigma, otherwise no response (equal prob to every particle

}//CalcProb
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
void AliRICHTracker::EsdPrint()
{
// Reads a set of ESD files and print out some information
// Arguments: probCut - cut on probability 
//   Returns: none
    
  TFile *pFile=TFile::Open("AliESDs.root","read");              if(!pFile) {Printf("ERROR: AliESDs.root does not exist!");return;} 
  TTree *pTr=(TTree*)pFile->Get("esdTree");                     if(!pTr)   {Printf("ERROR: AliESDs.root, no ESD tree inside!");return;} 
  AliESD *pEsd=new AliESD;  pTr->SetBranchAddress("ESD", &pEsd); 
  
  Int_t iNevt=pTr->GetEntries();  Printf("This ESD contains %i events",iNevt);
  for(Int_t iEvt=0;iEvt<iNevt;iEvt++){//ESD events loop
    pTr->GetEvent(iEvt);
    Int_t iNtracks=pEsd->GetNumberOfTracks(); Printf("ESD contains %i tracks created in Bz=%.2f Tesla",iNtracks,pEsd->GetMagneticField()/10.);
    for(Int_t iTrk=0;iTrk<iNtracks;iTrk++){//ESD tracks loop
      AliESDtrack *pTrack = pEsd->GetTrack(iTrk);// get next reconstructed track
      Float_t dx,dy;         pTrack->GetRICHdxdy(dx,dy);
      Float_t theta,phi;     pTrack->GetRICHthetaPhi(theta,phi);
      TString comment;
      if(pTrack->GetRICHsignal()>0)
        comment="OK";
      else if(pTrack->GetRICHsignal()==kMipQdcCut)
        comment="no enough QDC";
      else if(pTrack->GetRICHsignal()==kMipDistCut)
        comment="nearest cluster is too far";
      else if(pTrack->GetRICHsignal()==-1)
        comment="no intersection";            
      Printf("Track %2i Q=%4.1f P=%.3f GeV RICH: ChamClus %7i Track-Mip=(%7.2f,%7.2f)=%5.2f cm ThetaCer %7.1f rad %s",
                   iTrk,pTrack->GetSign(),pTrack->GetP(),
                   pTrack->GetRICHcluster(),dx,dy,TMath::Sqrt(dx*dx+dy*dy),pTrack->GetRICHsignal(), comment.Data());
    }//ESD tracks loop
  }//ESD events loop
  delete pEsd;  pFile->Close();//close AliESDs.root
}
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
void AliRICHTracker::MatrixPrint(Double_t probCut)
{
// Reads a set of 3  ESD files from current directory and prints out the matrix of probabilities to pion kaon or proton completely blindly withou nay assumption on the contents of files.
// Normally it implies that those 3 ESDs contain only particles of the same sort namly pions, kaons and protons in that order.  
// Arguments: probCut - cut on probability 
//   Returns: none
  for(Int_t iFile=0;iFile<3;iFile++){
    TFile *pFile=TFile::Open(Form("Esd%1i.root",iFile+1),"read"); if(!pFile) {Printf("ERROR: Esd%1i.root does not exist!",iFile+1);return;} 
    TTree *pTr=(TTree*)pFile->Get("esdTree");                     if(!pTr)   {Printf("ERROR: Esd%1i.root, no ESD tree inside!",iFile+1);return;} 
    AliESD *pEsd=new AliESD;  pTr->SetBranchAddress("ESD", &pEsd); 
    Int_t iProtCnt=0,iKaonCnt=0,iPionCnt=0,iUnreconCnt=0,iTrkCnt=0; //counters
    
    for(Int_t iEvt=0;iEvt<pTr->GetEntries();iEvt++){//ESD events loop
      pTr->GetEvent(iEvt);
      iTrkCnt+=pEsd->GetNumberOfTracks();
      for(Int_t iTrk=0;iTrk<pEsd->GetNumberOfTracks();iTrk++){//ESD tracks loop
        AliESDtrack *pTrack = pEsd->GetTrack(iTrk);// get next reconstructed track
        Float_t dx,dy;         pTrack->GetRICHdxdy(dx,dy);
        Float_t theta,phi;     pTrack->GetRICHthetaPhi(theta,phi);
        Double_t prob[5];       pTrack->GetRICHpid(prob);
        if(pTrack->GetRICHsignal()>0){
          if(prob[4]>probCut)                         iProtCnt++; 
          if(prob[3]>probCut)                         iKaonCnt++;
          if((prob[0]+prob[1]+prob[2])>probCut)       iPionCnt++;
        } else
          iUnreconCnt++;       
      }//ESD tracks loop
      
    }//ESD events loop
    Printf("Bz=%5.2f Events=%i Total tracks=%i No recognized tracks=%i Pion=%i Kaon=%i Proton=%i ProbCut=%.2f",
        0.1*pEsd->GetMagneticField(),pTr->GetEntries(),iTrkCnt,iUnreconCnt,iPionCnt,iKaonCnt,iProtCnt,probCut);
    delete pEsd;  pFile->Close();//close AliESDs.root
  }//files loop
}
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Commit	Line	Data
db910db9	1	#include "AliRICHTracker.h" //class header
	2	#include "AliRICH.h"
	3	#include "AliRICHRecon.h"
998b831f	4	#include <AliESD.h>
998b831f	5	#include <TVector3.h>
db910db9	6	#include <TTree.h> //EsdPrint()
db910db9	7	#include <TFile.h> //EsdPrint()
998b831f	8	#include "AliRICHHelix.h"
998b831f	9	#include <AliMagF.h>
998b831f	10	#include <AliStack.h>
998b831f	11	#include <TParticle.h>
30c60010	12	#include <TMath.h>
0fe8fa07	13	#include <AliRun.h>
db910db9	14	#include <TNtupleD.h> //RecWithStack();
	15	#include <AliTrackPointArray.h> //GetTrackPoint()
	16	#include <AliAlignObj.h> //GetTrackPoint()
998b831f	17	ClassImp(AliRICHTracker)
db910db9	18	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
db910db9	19	AliRICHTracker::AliRICHTracker():AliTracker()
998b831f	20	{
db910db9	21	// AliRICHTracker is created from AliReconstraction::Run() which invokes AliReconstraction::CreateTrackers()
	22	// which in turn invokes AliRICHReconstructor::CreateTracker().
	23	// Note that this is done just once per session before AliReconstruction::Run() goes to events loop.
	24	AliRICHParam::Instance()->CdbRead(0,0);
	25	for(Int_t i=0;i<5;i++)fErrPar[i]=0;
	26	}
	27	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
	28	Bool_t AliRICHTracker::GetTrackPoint(Int_t idx, AliTrackPoint& point) const
30c60010	29	{
db910db9	30	// Interface callback methode invoked from AliReconstruction::WriteAlignmentData() to get position of MIP cluster in MARS associated to a current track.
	31	// MIP cluster is reffered by index which is stored in AliESDtrack ???????
	32	// Arguments: idx- cluster index which is stored by RICH in AliESDtrack
	33	// point- reference to the object where to store the point
	34	// Returns: status of operation if FALSE then AliReconstruction::WriteAlignmentData() do not store this point to array of points for current track.
	35	if(idx<0) return kFALSE; //no MIP cluster assigned to this track in PropagateBack()
	36	Int_t iCham=idx/1000000;
	37	Int_t iClu=idx%1000000;
	38	point.SetVolumeID(AliAlignObj::LayerToVolUID(AliAlignObj::kRICH,iCham-1));//layer and chamber number
	39	AliRICH pRich=((AliRICH)gAlice->GetDetector("RICH"));
	40	AliRICHCluster pClu=(AliRICHCluster)pRich->Clus(iCham)->UncheckedAt(iClu);//get pointer to cluster
	41	TVector3 mars=AliRICHParam::Instance()->Lors2Mars(iCham,pClu->X(),pClu->Y());
	42	point.SetXYZ(mars.X(),mars.Y(),mars.Z());
	43	return kTRUE;
	44	}
	45	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
	46	Int_t AliRICHTracker::LoadClusters(TTree *pCluTree)
	47	{
	48	// Interface callback methode invoked from AliReconstruction::RunTracking() to load RICH clusters for RICH
	49	// Arguments: pCluTree- pointer to clusters tree got by AliRICHLoader::LoadRecPoints("read") then AliRICHLoader::TreeR()
	50	// Returns: error code (currently ignored in AliReconstruction::RunTraking())
	51	AliDebug(1,"Start."); pCluTree->GetEntry(0); AliDebug(1,"Stop."); return 0;
	52	}
	53	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
	54	Int_t AliRICHTracker::PropagateBack(AliESD *pESD)
	55	{
	56	// Interface callback methode invoked by AliRecontruction::RunTracking() during tracking after TOF. It's done just once per event
	57	// Arguments: pESD - pointer to Event Summary Data class instance which contains a list of tracks
	58	// Returns: error code, 0 if no errors
	59	Int_t iNtracks=pESD->GetNumberOfTracks();
	60	AliDebug(1,Form("Start with %i tracks",iNtracks));
	61	AliRICH pRich=((AliRICH)gAlice->GetDetector("RICH"));
	62	AliRICHRecon recon;
cb801d41	63	AliPID pid; // needed to retrive all the PID info
db910db9	64
	65	for(Int_t iTrk=0;iTrk<iNtracks;iTrk++){//ESD tracks loop
	66	AliESDtrack *pTrack = pESD->GetTrack(iTrk);// get next reconstructed track
e1e6896f	67	Double_t mom[3], pos[3];
	68	pTrack->GetPxPyPz(mom); TVector3 mom3(mom[0],mom[1],mom[2]);
	69	pTrack->GetXYZ(pos); TVector3 pos3(pos[0],pos[1],pos[2]);
	70	AliRICHHelix helix(pos3,mom3,(Int_t)pTrack->GetSign(),-0.1*GetBz()); //construct helix out of track running parameters
2a92c3fe	71	//Printf(" magnetic field %f charged %f\n",GetBz(),pTrack->GetSign()); helix.Print("Track");
db910db9	72	Int_t iChamber=helix.RichIntersect(AliRICHParam::Instance());
	73	if(!iChamber) continue; //no intersection with chambers, ignore this track go after the next one
	74
	75	//find MIP cluster candidate (closest to track intersection point cluster with large enough QDC)
2714766e	76	Double_t dR=9999, dX=9999, dY=9999; //distance between track-PC intersection point and current cluster
2714766e	77	Double_t mipDr=9999,mipDx=9999,mipDy=9999,mipX=9999,mipY=9999; //nearest cluster parameters
db910db9	78	Int_t iMipId=-1; //index of this nearest cluster
	79	for(Int_t iClu=0;iClu<pRich->Clus(iChamber)->GetEntries();iClu++){//clusters loop for intersected chamber
	80	AliRICHCluster pClu=(AliRICHCluster)pRich->Clus(iChamber)->UncheckedAt(iClu);//get pointer to current cluster
	81	if(pClu->Q()<AliRICHParam::QthMIP()) continue; //to low QDC, go after another one
	82	pClu->DistXY(helix.PosPc(),dX,dY); dR=TMath::Sqrt(dXdX+dYdY); //get distance for current cluster
2714766e	83	if(dR<mipDr){iMipId=iClu; mipDr=dR; mipDx=dX; mipDy=dY; mipX=pClu->X(); mipY=pClu->Y();} //current cluster is closer, overwrite data for min cluster
a25b3368	84	}//clusters loop for intersected chamber
01c81d9d	85
db910db9	86	pTrack->SetRICHthetaPhi(helix.Ploc().Theta(),helix.Ploc().Phi()); //store track impact angles with respect to RICH planes
2714766e	87	pTrack->SetRICHdxdy(mipDx,mipDy); //distance between track-PC intersection and closest cluster with Qdc>100
2714766e	88	pTrack->SetRICHmipXY(mipX,mipY); //position of that closest cluster with Qdc>100
d3eb6079	89
db910db9	90	if(iMipId==-1) {pTrack->SetRICHsignal(kMipQdcCut); continue;} //no cluster with enough QDC found
2714766e	91	if(mipDr>AliRICHParam::DmatchMIP()) {pTrack->SetRICHsignal(kMipDistCut); continue;} //closest cluster with enough carge is still too far
db910db9	92
	93	pTrack->SetRICHcluster(iMipId+1000000*iChamber); //set mip cluster index
	94	pTrack->SetRICHsignal(recon.ThetaCerenkov(&helix,pRich->Clus(iChamber),iMipId));//search for mean Cerenkov angle for this track
	95	pTrack->SetRICHnclusters(iMipId); //on return iMipId is number of photon clusters accepted in reconstruction
fab9e039	96
db910db9	97	AliDebug(1,Form("Ch=%i PC Intersection=(%5.2f,%5.2f) cm MIP cluster dist=(%5.2f,%5.2f)=%5.2f cm ThetaCkov=%f",
2714766e	98	iChamber,helix.PosPc().X(),helix.PosPc().Y(), mipDx,mipDy,mipDr, pTrack->GetRICHsignal()));
db910db9	99
db910db9	100	//here comes PID calculations
101624cd	101	if(pTrack->GetRICHsignal()>0) {
101624cd	102	AliDebug(1,Form("Start to assign the probabilities"));
cb801d41	103	Double_t sigmaPID[AliPID::kSPECIES]; Double_t richPID[AliPID::kSPECIES];
	104	for (Int_t iPid=0;iPid<AliPID::kSPECIES;iPid++){//PID loop
	105	sigmaPID[iPid] = 0; fErrPar[iPid] = 0;
	106	for(Int_t iCkov=0;iCkov<pRich->Clus(iChamber)->GetEntries();iCkov++){//Ckov candidates loop ????????????? somehting fomr AliRICHRecon must be
	107	recon.SetPhotonIndex(iCkov);
101624cd	108	if(recon.GetPhotonFlag() == 2) {
cb801d41	109	Double_t thetaCkov=0.6; //??????????????????
	110	Double_t phiCkov=0.6; //??????????????????
	111	Double_t thetaTrk=recon.GetTrackTheta();
	112	Double_t phiTrk=(recon.GetPhiPoint()-recon.GetTrackPhi());
	113	Double_t beta =pTrack->GetP()/TMath::Sqrt((pTrack->GetP()pTrack->GetP()+pid.ParticleMass(iPid)pid.ParticleMass(iPid)));
	114	Double_t sigma2 = AliRICHParam::SigmaSinglePhotonFormula(thetaCkov,phiCkov,thetaTrk,phiTrk,beta);
	115	if(sigma2>0) sigmaPID[iPid] += 1/sigma2;
101624cd	116	}
cb801d41	117	}//Ckov candidates loop
	118
	119	if (sigmaPID[iPid]>0)
	120	sigmaPID[iPid] *= (Double_t)(iMipId-recon.GetPhotBKG())/(Double_t)(iMipId); // n total phots, m are background...the sigma are scaled..
	121	if(sigmaPID[iPid]>0) sigmaPID[iPid] = 1/TMath::Sqrt(sigmaPID[iPid])*0.001; // sigma from parametrization are in mrad...
	122	else sigmaPID[iPid] = 0;
	123	fErrPar[iPid]=sigmaPID[iPid];
	124	AliDebug(1,Form("sigma for %s is %f rad",AliPID::ParticleName(iPid),sigmaPID[iPid]));
101624cd	125	}
db910db9	126	CalcProb(pTrack->GetRICHsignal(),pTrack->GetP(),sigmaPID,richPID);
101624cd	127	pTrack->SetRICHpid(richPID);
101624cd	128	AliDebug(1,Form("PROBABILITIES ---> %f - %f - %f - %f - %f",richPID[0],richPID[1],richPID[2],richPID[3],richPID[4]));
cb801d41	129	}//if(pTrack->GetRICHsignal())
db910db9	130	}//ESD tracks loop
	131	AliDebug(1,"Stop pattern recognition");
	132	return 0; // error code: 0=no error;
	133	}//PropagateBack()
	134	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
1cb5a97c	135	void AliRICHTracker::RecWithStack(TNtupleD *hn)
30c60010	136	{
db910db9	137	// Reconstruction for particles from STACK. This methode is to be used for RICH standalone when no other detectors are switched on, so normal tracking is not available.
	138	// Arguments: hn- output ntuple where to store all variables
	139	// Returns: none
d3eb6079	140	AliDebug(1,"Start.");
30c60010	141	AliRICH pRich=((AliRICH)gAlice->GetDetector("RICH"));
30c60010	142
1cb5a97c	143	// pRich->GetLoader()->GetRunLoader()->LoadHeader();
d3eb6079	144	if(!pRich->GetLoader()->GetRunLoader()->TreeK()) pRich->GetLoader()->GetRunLoader()->LoadKinematics();
30c60010	145	AliStack *pStack = pRich->GetLoader()->GetRunLoader()->Stack();
39853df5	146	if(!pStack) {AliDebug(1,Form("No STACK found in AliRoot"));return;}
30c60010	147	Int_t iNtracks=pStack->GetNtrack();
1cb5a97c	148	AliDebug(1,Form(" Start reconstruction with %i track(s) from Stack",iNtracks));
1cb5a97c	149
ab92fbe6	150	Double_t hnvec[20];
1cb5a97c	151
30c60010	152	Double_t b=GetFieldMap()->SolenoidField()/10;// magnetic field in Tesla
	153	AliDebug(1,Form("Start with simulated %i tracks in %f Tesla field",iNtracks,b));
	154	TVector3 x0(0,0,0); TVector3 p0(0,0,0);//tmp storage for AliRICHHelix
	155
	156
1cb5a97c	157	if(pRich->GetLoader()->LoadRecPoints()) {AliDebug(1,Form("No clusters found in RICH"));return;}
	158	pRich->GetLoader()->TreeR()->GetEntry(0);
	159
db910db9	160	AliRICHRecon recon;
a25b3368	161	for(Int_t iTrackN=0;iTrackN<iNtracks;iTrackN++){//stack particles loop
30c60010	162	TParticle *pParticle = pStack->Particle(iTrackN);
39853df5	163	if(!pParticle) {AliDebug(1,Form("Not a valid TParticle pointer. Track skipped"));continue;}
39853df5	164	AliDebug(1,Form(" PDG code : %i",pParticle->GetPdgCode()));
ab92fbe6	165	//
	166	// problem of PDG code of some extra particles to be solved!!!!!!!!!
	167	//
	168	// found problem! Look in TRD directory : codes from Fluka are :
	169	//
	170	// if ((pdg_code == 10010020) \|\|
	171	// (pdg_code == 10010030) \|\|
	172	// (pdg_code == 50000050) \|\|
	173	// (pdg_code == 50000051) \|\|
	174	// (pdg_code == 10020040)) {
	175	//
	176	if(pParticle->GetPdgCode()>=50000050\|\|pParticle->GetPdgCode()==0\|\|pParticle->GetPdgCode()>10000) {AliDebug(1,Form("A photon as track... Track skipped"));continue;}
1e066c96	177	if(!pParticle->GetPDG()) continue;
ab92fbe6	178	//
	179	// to be updated for us!!
	180	//
9f873529	181	AliDebug(1,Form("Track %i is a %s with charge %i and momentum %f",
	182	iTrackN,pParticle->GetPDG()->GetName(),Int_t(pParticle->GetPDG()->Charge()),pParticle->P()));
	183	// if(pParticle->GetMother(0)!=-1) continue; //consider only primaries
	184	if(pParticle->GetPDG()->Charge()==0\|\|TMath::Abs(Int_t(pParticle->GetPDG()->Charge()))!=3) continue; //to avoid photons from stack...
1cb5a97c	185	hnvec[0]=pParticle->P();
1cb5a97c	186	hnvec[1]=pParticle->GetPDG()->Charge();
db910db9	187
db910db9	188	p0.SetMagThetaPhi(pParticle->P(),pParticle->Theta(),pParticle->Phi()); x0.SetXYZ(pParticle->Vx(),pParticle->Vy(),pParticle->Vz());
9f873529	189	AliRICHHelix helix(x0,p0,TMath::Sign(1,(Int_t)pParticle->GetPDG()->Charge()),b);
db910db9	190	Int_t iChamber=helix.RichIntersect(AliRICHParam::Instance());
	191	if(!iChamber) continue;// no intersection with RICH found
	192
ef210ba6	193	hnvec[2]=helix.Ploc().Theta();
ef210ba6	194	hnvec[3]=helix.Ploc().Phi();
1cb5a97c	195	hnvec[4]=helix.PosPc().X();
1cb5a97c	196	hnvec[5]=helix.PosPc().Y();
db910db9	197
db910db9	198	Double_t dX,dY,dR,dRmip=9999; //min distance between clusters and track position on PC
0fe8fa07	199	Int_t iMipId=-1; //index of that min distance cluster
db910db9	200	for(Int_t iClu=0;iClu<pRich->Clus(iChamber)->GetEntries();iClu++){//clusters loop for intersected chamber
	201	AliRICHCluster pClu=(AliRICHCluster)pRich->Clus(iChamber)->UncheckedAt(iClu);//get pointer to current cluster
	202	pClu->DistXY(helix.PosPc(),dX,dY); dR=TMath::Sqrt(dXdX+dYdY);//ditance between current cluster and helix intersection with PC
	203	if(dR<dRmip){dRmip=dR; hnvec[6]=pClu->X();hnvec[7]=pClu->Y();hnvec[8]=pClu->Q();
	204	iMipId=1000000*iChamber+iClu;}//find cluster nearest to the track
30c60010	205
a25b3368	206	}//clusters loop for intersected chamber
30c60010	207
db910db9	208
	209	hnvec[9]=recon.ThetaCerenkov(&helix,pRich->Clus(iChamber),iMipId); //search for mean Cerenkov angle for this track
	210	hnvec[10]=iMipId;//on return from ThetaCerenkov() contains number of photon candidates accepted
1cb5a97c	211	hnvec[11]=(Double_t)iMipId;
ab92fbe6	212	hnvec[12]=(Double_t)iChamber;
ab92fbe6	213	hnvec[13]=(Double_t)pParticle->GetPdgCode();
1cb5a97c	214	if(hn) hn->Fill(hnvec);
db910db9	215	AliDebug(1,Form("FINAL Theta Cerenkov=%f",hnvec[9]));
a25b3368	216	}//stack particles loop
1cb5a97c	217
39853df5	218	pRich->GetLoader()->UnloadRecPoints();
30c60010	219	AliDebug(1,"Stop.");
a25b3368	220	}//RecWithStack
db910db9	221	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
101624cd	222	void AliRICHTracker::CalcProb(Double_t thetaCer,Double_t pmod, Double_t sigmaPID, Double_t richPID)
30c60010	223	{
12b0e5f4	224	// Calculates probability to be a electron-muon-pion-kaon-proton
a25b3368	225	// from the given Cerenkov angle and momentum assuming no initial particle composition
12b0e5f4	226	// (i.e. apriory probability to be the particle of the given sort is the same for all sorts)
304864ab	227	Double_t height[AliPID::kSPECIES];Double_t totalHeight=0;
101624cd	228	Double_t thetaTh[AliPID::kSPECIES];
304864ab	229	for(Int_t iPart=0;iPart<AliPID::kSPECIES;iPart++){
101624cd	230	height[iPart]=0;
068217e3	231	Double_t mass = AliRICHParam::fgMass[iPart];
db910db9	232	Double_t refIndex=AliRICHParam::Instance()->IdxC6F14(AliRICHParam::EckovMean());
30c60010	233	Double_t cosThetaTh = TMath::Sqrt(massmass+pmodpmod)/(refIndex*pmod);
101624cd	234	thetaTh[iPart]=0;
	235	if(cosThetaTh>=1) continue;
	236	thetaTh[iPart] = TMath::ACos(cosThetaTh);
	237	// Double_t sinThetaThNorm = TMath::Sin(thetaTh)/TMath::Sqrt(1-1/(refIndex*refIndex));
	238	// Double_t sigmaThetaTh = (0.014(1/sinThetaThNorm-1) + 0.0043)1.25;
	239	// height[iPart] = TMath::Gaus(thetaCer,thetaTh,sigmaThetaTh);
01c81d9d	240	if(sigmaPID[iPart]>0) height[iPart] = TMath::Gaus(thetaCer,thetaTh[iPart],sigmaPID[iPart],kTRUE);
01c81d9d	241	else height[iPart] = 0;
30c60010	242	totalHeight +=height[iPart];
db910db9	243	AliDebugClass(1,Form(" Particle %s with mass %f with height %f and thetaTH %f",AliPID::ParticleName(iPart),mass,height[iPart],thetaTh[iPart]));
db910db9	244	AliDebugClass(1,Form(" partial height %15.14f total height %15.14f",height[iPart],totalHeight));
30c60010	245	}
12b0e5f4	246	if(totalHeight<1e-5) {for(Int_t iPart=0;iPart<AliPID::kSPECIES;iPart++)richPID[iPart]=1.0/AliPID::kSPECIES;return;}
101624cd	247	for(Int_t iPart=0;iPart<AliPID::kSPECIES;iPart++) richPID[iPart] = height[iPart]/totalHeight;
101624cd	248	Int_t iPartNear = TMath::LocMax(AliPID::kSPECIES,richPID);
01c81d9d	249	if(TMath::Abs(thetaCer-thetaTh[iPartNear])>5.*sigmaPID[iPartNear]) for(Int_t iPart=0;iPart<AliPID::kSPECIES;iPart++)richPID[iPart]=1.0/AliPID::kSPECIES;
07625543	250	//last line is to check if the nearest thetacerenkov to the teorethical one is within 5 sigma, otherwise no response (equal prob to every particle
101624cd	251
30c60010	252	}//CalcProb
db910db9	253	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
	254	void AliRICHTracker::EsdPrint()
	255	{
2a92c3fe	256	// Reads a set of ESD files and print out some information
	257	// Arguments: probCut - cut on probability
	258	// Returns: none
	259
	260	TFile *pFile=TFile::Open("AliESDs.root","read"); if(!pFile) {Printf("ERROR: AliESDs.root does not exist!");return;}
	261	TTree pTr=(TTree)pFile->Get("esdTree"); if(!pTr) {Printf("ERROR: AliESDs.root, no ESD tree inside!");return;}
	262	AliESD *pEsd=new AliESD; pTr->SetBranchAddress("ESD", &pEsd);
db910db9	263
2a92c3fe	264	Int_t iNevt=pTr->GetEntries(); Printf("This ESD contains %i events",iNevt);
db910db9	265	for(Int_t iEvt=0;iEvt<iNevt;iEvt++){//ESD events loop
2a92c3fe	266	pTr->GetEvent(iEvt);
2a92c3fe	267	Int_t iNtracks=pEsd->GetNumberOfTracks(); Printf("ESD contains %i tracks created in Bz=%.2f Tesla",iNtracks,pEsd->GetMagneticField()/10.);
db910db9	268	for(Int_t iTrk=0;iTrk<iNtracks;iTrk++){//ESD tracks loop
2a92c3fe	269	AliESDtrack *pTrack = pEsd->GetTrack(iTrk);// get next reconstructed track
2714766e	270	Float_t dx,dy; pTrack->GetRICHdxdy(dx,dy);
	271	Float_t theta,phi; pTrack->GetRICHthetaPhi(theta,phi);
	272	TString comment;
	273	if(pTrack->GetRICHsignal()>0)
	274	comment="OK";
	275	else if(pTrack->GetRICHsignal()==kMipQdcCut)
	276	comment="no enough QDC";
	277	else if(pTrack->GetRICHsignal()==kMipDistCut)
	278	comment="nearest cluster is too far";
	279	else if(pTrack->GetRICHsignal()==-1)
	280	comment="no intersection";
	281	Printf("Track %2i Q=%4.1f P=%.3f GeV RICH: ChamClus %7i Track-Mip=(%7.2f,%7.2f)=%5.2f cm ThetaCer %7.1f rad %s",
	282	iTrk,pTrack->GetSign(),pTrack->GetP(),
	283	pTrack->GetRICHcluster(),dx,dy,TMath::Sqrt(dxdx+dydy),pTrack->GetRICHsignal(), comment.Data());
db910db9	284	}//ESD tracks loop
db910db9	285	}//ESD events loop
2a92c3fe	286	delete pEsd; pFile->Close();//close AliESDs.root
	287	}
	288	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
	289	void AliRICHTracker::MatrixPrint(Double_t probCut)
	290	{
	291	// Reads a set of 3 ESD files from current directory and prints out the matrix of probabilities to pion kaon or proton completely blindly withou nay assumption on the contents of files.
	292	// Normally it implies that those 3 ESDs contain only particles of the same sort namly pions, kaons and protons in that order.
	293	// Arguments: probCut - cut on probability
	294	// Returns: none
	295	for(Int_t iFile=0;iFile<3;iFile++){
	296	TFile *pFile=TFile::Open(Form("Esd%1i.root",iFile+1),"read"); if(!pFile) {Printf("ERROR: Esd%1i.root does not exist!",iFile+1);return;}
	297	TTree pTr=(TTree)pFile->Get("esdTree"); if(!pTr) {Printf("ERROR: Esd%1i.root, no ESD tree inside!",iFile+1);return;}
	298	AliESD *pEsd=new AliESD; pTr->SetBranchAddress("ESD", &pEsd);
	299	Int_t iProtCnt=0,iKaonCnt=0,iPionCnt=0,iUnreconCnt=0,iTrkCnt=0; //counters
	300
	301	for(Int_t iEvt=0;iEvt<pTr->GetEntries();iEvt++){//ESD events loop
	302	pTr->GetEvent(iEvt);
	303	iTrkCnt+=pEsd->GetNumberOfTracks();
	304	for(Int_t iTrk=0;iTrk<pEsd->GetNumberOfTracks();iTrk++){//ESD tracks loop
	305	AliESDtrack *pTrack = pEsd->GetTrack(iTrk);// get next reconstructed track
2714766e	306	Float_t dx,dy; pTrack->GetRICHdxdy(dx,dy);
2714766e	307	Float_t theta,phi; pTrack->GetRICHthetaPhi(theta,phi);
2a92c3fe	308	Double_t prob[5]; pTrack->GetRICHpid(prob);
	309	if(pTrack->GetRICHsignal()>0){
	310	if(prob[4]>probCut) iProtCnt++;
	311	if(prob[3]>probCut) iKaonCnt++;
	312	if((prob[0]+prob[1]+prob[2])>probCut) iPionCnt++;
	313	} else
	314	iUnreconCnt++;
	315	}//ESD tracks loop
	316
	317	}//ESD events loop
	318	Printf("Bz=%5.2f Events=%i Total tracks=%i No recognized tracks=%i Pion=%i Kaon=%i Proton=%i ProbCut=%.2f",
	319	0.1*pEsd->GetMagneticField(),pTr->GetEntries(),iTrkCnt,iUnreconCnt,iPionCnt,iKaonCnt,iProtCnt,probCut);
	320	delete pEsd; pFile->Close();//close AliESDs.root
	321	}//files loop
db910db9	322	}
db910db9	323	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++