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

#include "AliRICHTracker.h"
#include <AliESD.h>
#include <TVector3.h>
#include <TTree.h>
#include "AliRICH.h"
#include "AliRICHHelix.h"
#include <AliMagF.h>
#include "AliRICHRecon.h"
#include <AliStack.h>
#include <TParticle.h>
#include <TMath.h>
#include <AliRun.h>
ClassImp(AliRICHTracker)
//__________________________________________________________________________________________________
Int_t AliRICHTracker::PropagateBack(AliESD *pESD)
{
// Interface callback methode invoked by AliRecontruction during tracking after TOF
// It steers to different way to provide the final reconstructed information sutable for analisys:
// 1. AliESD  - reconstructed tracks are used     
// 2. RICH private ntuple for debug- stack particles used instead of reconstructed tracks     
  AliDebug(1,"Start pattern recognition");
  if(pESD->GetNumberOfTracks()) {
    Int_t iNtracks=pESD->GetNumberOfTracks();
    AliDebug(1,Form("Start with %i tracks",iNtracks));
    AliRICH *pRich=((AliRICH*)gAlice->GetDetector("RICH"));  
    for(Int_t iTrackN=0;iTrackN<iNtracks;iTrackN++){//ESD tracks loop
      RecWithESD(pESD,pRich,iTrackN);
    }
  }
  else RecWithStack(0);
  AliDebug(1,"Stop pattern recognition");
  return 0; // error code: 0=no error;
}//PropagateBack()
//__________________________________________________________________________________________________
void AliRICHTracker::RecWithESD(AliESD *pESD,AliRICH *pRich,Int_t iTrackN)
{
//recontruction from ESD- primary way to reconstruct particle ID signal from tracks provided by core detectors
    fnPhotBKG = 0;

    Double_t fField=GetFieldMap()->SolenoidField()/10;// magnetic field in Tesla
    AliESDtrack *pTrack = pESD->GetTrack(iTrackN);// get next reconstructed track
//  if((pTrack->GetStatus()&AliESDtrack::kTOFout)==0) continue; //ignore tracks not recontructed by TOF
//    pTrack->GetXYZ(xb); 
//    pTrack->GetPxPyPz(pb); 
    Int_t status=pTrack->GetStatus()&AliESDtrack::kTOFout;//get running track parameters
    Int_t charge = (Int_t)(-TMath::Sign(1.,pTrack->GetSign()*fField));
    AliDebug(1,Form("Track %i pmod=%f charge=%i stat=%i",iTrackN,pTrack->GetP(),charge,status));
    AliRICHHelix helix(pTrack->X3(),pTrack->P3(),charge,fField);   
    Int_t iChamber=helix.RichIntersect(pRich->P());        
    AliDebug(1,Form("intersection with %i chamber found",iChamber));
    if(!iChamber) {
      pTrack->SetRICHsignal(-999); //to be improved by flags...
      return;//intersection with no chamber found
    }
//find MIP cluster candidate (cluster which is closest to track intersection point)
    Double_t distMip=9999,distX=0,distY=0; //min distance between clusters and track position on PC 
    Int_t iMipId=0; //index of that min distance cluster
    Double_t chargeMip=0; //charge of the MIP
    Bool_t kFound = kFALSE;
    for(Int_t iClusN=0;iClusN<pRich->Clusters(iChamber)->GetEntries();iClusN++){//clusters loop for intersected chamber
      AliRICHCluster *pClus=(AliRICHCluster*)pRich->Clusters(iChamber)->UncheckedAt(iClusN);//get pointer to current cluster
      if(pClus->Q()<AliRICHParam::QthMIP()) continue;
      Double_t distCurrent=pClus->DistTo(helix.PosPc());//distance between current cluster and helix intersection with PC
      if(distCurrent<distMip){
        kFound = kTRUE;
        distMip=distCurrent;
        iMipId=iClusN;
        distX=pClus->DistX(helix.PosPc());
        distY=pClus->DistY(helix.PosPc());
        chargeMip=pClus->Q();
      }//find cluster nearest to the track       
      AliDebug(1,Form("Ploc (%f,%f,%f) dist= %f",helix.Ploc().Mag(),helix.Ploc().Theta()*TMath::RadToDeg(),
                                       helix.Ploc().Phi()*TMath::RadToDeg(),pClus->DistTo(helix.PosPc())));
    }//clusters loop for intersected chamber

    if(!kFound) {
      pTrack->SetRICHsignal(-999);
      return;
    }
    AliDebug(1,Form("Min distance cluster: %i dist is %f",iMipId,distMip));
    pTrack->SetRICHcluster(((Int_t)chargeMip)+1000000*iChamber);
    pTrack->SetRICHdxdy(distX,distY);
    pTrack->SetRICHthetaPhi(helix.Ploc().Theta(),helix.Ploc().Phi());
//
// HERE CUTS ON GOLD RINGS....
//
    if(distMip>AliRICHParam::DmatchMIP()) {
      //track not accepted for pattern recognition
      pTrack->SetRICHsignal(-990); //to be improved by flags...
      return;
    }
//
    AliRICHRecon recon(&helix,pRich->Clusters(iChamber),iMipId); //actual job is done there

    Double_t thetaCerenkov=recon.ThetaCerenkov(); //search for mean Cerenkov angle for this track
    
    pTrack->SetRICHsignal(thetaCerenkov);
    pTrack->SetRICHnclusters(recon.GetHoughPhotons());

    fnPhotBKG = recon.GetPhotBKG();
        
    AliDebug(1,Form("FINAL Theta Cerenkov=%f",pTrack->GetRICHsignal()));
//
    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 iPart=0;iPart<AliPID::kSPECIES;iPart++) {
        sigmaPID[iPart] = 0;
        fErrPar[iPart] = 0;
        for(Int_t iphot=0;iphot<pRich->Clusters(iChamber)->GetEntries();iphot++) {
          recon.SetPhotonIndex(iphot);
          if(recon.GetPhotonFlag() == 2) {
            Double_t theta_g=recon.GetTrackTheta();
            Double_t phi_g=(recon.GetPhiPoint()-recon.GetTrackPhi());
            Double_t sigma2 = AliRICHParam::SigmaSinglePhoton(iPart,pTrack->GetP(),theta_g,phi_g).Mag2(); 
            if(sigma2>0) sigmaPID[iPart] += 1/sigma2;
          }
        }
	if (sigmaPID[iPart]>0)
          sigmaPID[iPart] *= (Double_t)(recon.GetHoughPhotons()-fnPhotBKG)/(Double_t)(recon.GetHoughPhotons()); // n total phots, m are background...the sigma are scaled..
	  if(sigmaPID[iPart]>0) sigmaPID[iPart] = 1/TMath::Sqrt(sigmaPID[iPart])*0.001; // sigma from parametrization are in mrad...
          else                  sigmaPID[iPart] = 0;
          fErrPar[iPart]=sigmaPID[iPart];
        AliDebug(1,Form("sigma for %s is %f rad",AliPID::ParticleName(iPart),sigmaPID[iPart]));
      }
      CalcProb(thetaCerenkov,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]));
    }    
  AliDebug(1,"Stop.");  
} //RecWithESD
//__________________________________________________________________________________________________
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   
  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);

  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;}
//
// 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(pRich->P());        
    hnvec[2]=helix.Ploc().Theta();
    hnvec[3]=helix.Ploc().Phi();
    AliDebug(1,Form("intersection with %i chamber found",iChamber));
    if(!iChamber) continue;// no intersection with RICH found
    hnvec[4]=helix.PosPc().X();
    hnvec[5]=helix.PosPc().Y();
    Double_t distMip=9999;   //min distance between clusters and track position on PC 
    Double_t mipX=-1;      //min distance between clusters and track position on PC 
    Double_t mipY=-1;      //min distance between clusters and track position on PC 
    Double_t chargeMip=-1; // charge MIP to find
    Int_t iMipId=-1;       //index of that min distance cluster 
    for(Int_t iClusN=0;iClusN<pRich->Clusters(iChamber)->GetEntries();iClusN++){//clusters loop for intersected chamber
      AliRICHCluster *pClus=(AliRICHCluster*)pRich->Clusters(iChamber)->UncheckedAt(iClusN);//get pointer to current cluster
      Double_t distCurrent=pClus->DistTo(helix.PosPc());//ditance between current cluster and helix intersection with PC
      if(distCurrent<distMip){distMip=distCurrent;mipX=pClus->X();
                                                  mipY=pClus->Y();
                                                  chargeMip=pClus->Q();iMipId=1000000*iChamber+iClusN;}//find cluster nearest to the track 
      
      AliDebug(1,Form("Ploc (%f,%f,%f) dist= %f",helix.Ploc().Mag(),helix.Ploc().Theta()*TMath::RadToDeg(),
                                                                    helix.Ploc().Phi()*TMath::RadToDeg(),pClus->DistTo(helix.PosPc())));
    }//clusters loop for intersected chamber
    
    AliDebug(1,Form("Min distance cluster: %i dist is %f",iMipId,distMip));
    hnvec[6]=mipX;hnvec[7]=mipY;
    hnvec[8]=chargeMip;
    AliRICHRecon recon(&helix,pRich->Clusters(iChamber),iMipId);
    Double_t thetaCerenkov=recon.ThetaCerenkov(); //search for mean Cerenkov angle for this track
    hnvec[9]=thetaCerenkov;
    hnvec[10]=recon.GetHoughPhotons();
    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",thetaCerenkov));
  }//stack particles loop
  
  pRich->GetLoader()->UnloadRecPoints();
  AliDebug(1,"Stop.");  
}//RecWithStack
//__________________________________________________________________________________________________
Int_t AliRICHTracker::LoadClusters(TTree *pTree)
{
// Load clusters for RICH
  AliDebug(1,"Start.");  pTree->GetEntry(0);  AliDebug(1,"Stop."); return 0;
}
//__________________________________________________________________________________________________
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::RefIdxC6F14(AliRICHParam::MeanCkovEnergy());
    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];
    AliDebug(1,Form(" Particle %s with mass %f with height %f and thetaTH %f",AliPID::ParticleName(iPart),mass,height[iPart],thetaTh[iPart]));
    AliDebug(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
//__________________________________________________________________________________________________
Commit	Line	Data
998b831f	1	#include "AliRICHTracker.h"
	2	#include <AliESD.h>
	3	#include <TVector3.h>
	4	#include <TTree.h>
	5	#include "AliRICH.h"
	6	#include "AliRICHHelix.h"
	7	#include <AliMagF.h>
	8	#include "AliRICHRecon.h"
	9	#include <AliStack.h>
	10	#include <TParticle.h>
30c60010	11	#include <TMath.h>
0fe8fa07	12	#include <AliRun.h>
998b831f	13	ClassImp(AliRICHTracker)
30c60010	14	//__________________________________________________________________________________________________
998b831f	15	Int_t AliRICHTracker::PropagateBack(AliESD *pESD)
998b831f	16	{
a25b3368	17	// Interface callback methode invoked by AliRecontruction during tracking after TOF
	18	// It steers to different way to provide the final reconstructed information sutable for analisys:
	19	// 1. AliESD - reconstructed tracks are used
	20	// 2. RICH private ntuple for debug- stack particles used instead of reconstructed tracks
30c60010	21	AliDebug(1,"Start pattern recognition");
910735ae	22	if(pESD->GetNumberOfTracks()) {
	23	Int_t iNtracks=pESD->GetNumberOfTracks();
	24	AliDebug(1,Form("Start with %i tracks",iNtracks));
	25	AliRICH pRich=((AliRICH)gAlice->GetDetector("RICH"));
	26	for(Int_t iTrackN=0;iTrackN<iNtracks;iTrackN++){//ESD tracks loop
	27	RecWithESD(pESD,pRich,iTrackN);
	28	}
	29	}
	30	else RecWithStack(0);
30c60010	31	AliDebug(1,"Stop pattern recognition");
30c60010	32	return 0; // error code: 0=no error;
a25b3368	33	}//PropagateBack()
30c60010	34	//__________________________________________________________________________________________________
910735ae	35	void AliRICHTracker::RecWithESD(AliESD pESD,AliRICH pRich,Int_t iTrackN)
30c60010	36	{
d3eb6079	37	//recontruction from ESD- primary way to reconstruct particle ID signal from tracks provided by core detectors
fab9e039	38	fnPhotBKG = 0;
d3eb6079	39
910735ae	40	Double_t fField=GetFieldMap()->SolenoidField()/10;// magnetic field in Tesla
998b831f	41	AliESDtrack *pTrack = pESD->GetTrack(iTrackN);// get next reconstructed track
998b831f	42	// if((pTrack->GetStatus()&AliESDtrack::kTOFout)==0) continue; //ignore tracks not recontructed by TOF
d3eb6079	43	// pTrack->GetXYZ(xb);
d3eb6079	44	// pTrack->GetPxPyPz(pb);
1cb5a97c	45	Int_t status=pTrack->GetStatus()&AliESDtrack::kTOFout;//get running track parameters
910735ae	46	Int_t charge = (Int_t)(-TMath::Sign(1.,pTrack->GetSign()*fField));
39853df5	47	AliDebug(1,Form("Track %i pmod=%f charge=%i stat=%i",iTrackN,pTrack->GetP(),charge,status));
910735ae	48	AliRICHHelix helix(pTrack->X3(),pTrack->P3(),charge,fField);
998b831f	49	Int_t iChamber=helix.RichIntersect(pRich->P());
998b831f	50	AliDebug(1,Form("intersection with %i chamber found",iChamber));
01c81d9d	51	if(!iChamber) {
	52	pTrack->SetRICHsignal(-999); //to be improved by flags...
	53	return;//intersection with no chamber found
	54	}
	55	//find MIP cluster candidate (cluster which is closest to track intersection point)
d3eb6079	56	Double_t distMip=9999,distX=0,distY=0; //min distance between clusters and track position on PC
f770edb5	57	Int_t iMipId=0; //index of that min distance cluster
f770edb5	58	Double_t chargeMip=0; //charge of the MIP
01c81d9d	59	Bool_t kFound = kFALSE;
998b831f	60	for(Int_t iClusN=0;iClusN<pRich->Clusters(iChamber)->GetEntries();iClusN++){//clusters loop for intersected chamber
0fe8fa07	61	AliRICHCluster pClus=(AliRICHCluster)pRich->Clusters(iChamber)->UncheckedAt(iClusN);//get pointer to current cluster
01c81d9d	62	if(pClus->Q()<AliRICHParam::QthMIP()) continue;
d3eb6079	63	Double_t distCurrent=pClus->DistTo(helix.PosPc());//distance between current cluster and helix intersection with PC
a25b3368	64	if(distCurrent<distMip){
01c81d9d	65	kFound = kTRUE;
a25b3368	66	distMip=distCurrent;
a25b3368	67	iMipId=iClusN;
d3eb6079	68	distX=pClus->DistX(helix.PosPc());
d3eb6079	69	distY=pClus->DistY(helix.PosPc());
f770edb5	70	chargeMip=pClus->Q();
a25b3368	71	}//find cluster nearest to the track
998b831f	72	AliDebug(1,Form("Ploc (%f,%f,%f) dist= %f",helix.Ploc().Mag(),helix.Ploc().Theta()*TMath::RadToDeg(),
a25b3368	73	helix.Ploc().Phi()*TMath::RadToDeg(),pClus->DistTo(helix.PosPc())));
a25b3368	74	}//clusters loop for intersected chamber
01c81d9d	75
	76	if(!kFound) {
	77	pTrack->SetRICHsignal(-999);
	78	return;
	79	}
998b831f	80	AliDebug(1,Form("Min distance cluster: %i dist is %f",iMipId,distMip));
01c81d9d	81	pTrack->SetRICHcluster(((Int_t)chargeMip)+1000000*iChamber);
	82	pTrack->SetRICHdxdy(distX,distY);
	83	pTrack->SetRICHthetaPhi(helix.Ploc().Theta(),helix.Ploc().Phi());
f770edb5	84	//
	85	// HERE CUTS ON GOLD RINGS....
	86	//
01c81d9d	87	if(distMip>AliRICHParam::DmatchMIP()) {
f770edb5	88	//track not accepted for pattern recognition
01c81d9d	89	pTrack->SetRICHsignal(-990); //to be improved by flags...
910735ae	90	return;
f770edb5	91	}
f770edb5	92	//
d3eb6079	93	AliRICHRecon recon(&helix,pRich->Clusters(iChamber),iMipId); //actual job is done there
101624cd	94
101624cd	95	Double_t thetaCerenkov=recon.ThetaCerenkov(); //search for mean Cerenkov angle for this track
d3eb6079	96
101624cd	97	pTrack->SetRICHsignal(thetaCerenkov);
d3eb6079	98	pTrack->SetRICHnclusters(recon.GetHoughPhotons());
fab9e039	99
	100	fnPhotBKG = recon.GetPhotBKG();
	101
d3eb6079	102	AliDebug(1,Form("FINAL Theta Cerenkov=%f",pTrack->GetRICHsignal()));
101624cd	103	//
	104	if(pTrack->GetRICHsignal()>0) {
	105	AliDebug(1,Form("Start to assign the probabilities"));
	106	Double_t sigmaPID[AliPID::kSPECIES];
	107	Double_t richPID[AliPID::kSPECIES];
	108	for (Int_t iPart=0;iPart<AliPID::kSPECIES;iPart++) {
	109	sigmaPID[iPart] = 0;
068217e3	110	fErrPar[iPart] = 0;
101624cd	111	for(Int_t iphot=0;iphot<pRich->Clusters(iChamber)->GetEntries();iphot++) {
	112	recon.SetPhotonIndex(iphot);
	113	if(recon.GetPhotonFlag() == 2) {
fab9e039	114	Double_t theta_g=recon.GetTrackTheta();
fab9e039	115	Double_t phi_g=(recon.GetPhiPoint()-recon.GetTrackPhi());
01c81d9d	116	Double_t sigma2 = AliRICHParam::SigmaSinglePhoton(iPart,pTrack->GetP(),theta_g,phi_g).Mag2();
01c81d9d	117	if(sigma2>0) sigmaPID[iPart] += 1/sigma2;
101624cd	118	}
101624cd	119	}
d1be0d8a	120	if (sigmaPID[iPart]>0)
01c81d9d	121	sigmaPID[iPart] *= (Double_t)(recon.GetHoughPhotons()-fnPhotBKG)/(Double_t)(recon.GetHoughPhotons()); // n total phots, m are background...the sigma are scaled..
	122	if(sigmaPID[iPart]>0) sigmaPID[iPart] = 1/TMath::Sqrt(sigmaPID[iPart])*0.001; // sigma from parametrization are in mrad...
	123	else sigmaPID[iPart] = 0;
	124	fErrPar[iPart]=sigmaPID[iPart];
101624cd	125	AliDebug(1,Form("sigma for %s is %f rad",AliPID::ParticleName(iPart),sigmaPID[iPart]));
	126	}
	127	CalcProb(thetaCerenkov,pTrack->GetP(),sigmaPID,richPID);
	128	pTrack->SetRICHpid(richPID);
	129	AliDebug(1,Form("PROBABILITIES ---> %f - %f - %f - %f - %f",richPID[0],richPID[1],richPID[2],richPID[3],richPID[4]));
910735ae	130	}
998b831f	131	AliDebug(1,"Stop.");
30c60010	132	} //RecWithESD
30c60010	133	//__________________________________________________________________________________________________
1cb5a97c	134	void AliRICHTracker::RecWithStack(TNtupleD *hn)
30c60010	135	{
d3eb6079	136	//Reconstruction for particles from STACK. This methode is to be used for RICH standalone when no other detectors are switched on,
	137	//so normal tracking is not available
	138	AliDebug(1,"Start.");
30c60010	139	AliRICH pRich=((AliRICH)gAlice->GetDetector("RICH"));
30c60010	140
1cb5a97c	141	// pRich->GetLoader()->GetRunLoader()->LoadHeader();
d3eb6079	142	if(!pRich->GetLoader()->GetRunLoader()->TreeK()) pRich->GetLoader()->GetRunLoader()->LoadKinematics();
30c60010	143	AliStack *pStack = pRich->GetLoader()->GetRunLoader()->Stack();
39853df5	144	if(!pStack) {AliDebug(1,Form("No STACK found in AliRoot"));return;}
30c60010	145	Int_t iNtracks=pStack->GetNtrack();
1cb5a97c	146	AliDebug(1,Form(" Start reconstruction with %i track(s) from Stack",iNtracks));
1cb5a97c	147
ab92fbe6	148	Double_t hnvec[20];
1cb5a97c	149
30c60010	150	Double_t b=GetFieldMap()->SolenoidField()/10;// magnetic field in Tesla
	151	AliDebug(1,Form("Start with simulated %i tracks in %f Tesla field",iNtracks,b));
	152	TVector3 x0(0,0,0); TVector3 p0(0,0,0);//tmp storage for AliRICHHelix
	153
	154
1cb5a97c	155	if(pRich->GetLoader()->LoadRecPoints()) {AliDebug(1,Form("No clusters found in RICH"));return;}
	156	pRich->GetLoader()->TreeR()->GetEntry(0);
	157
a25b3368	158	for(Int_t iTrackN=0;iTrackN<iNtracks;iTrackN++){//stack particles loop
30c60010	159	TParticle *pParticle = pStack->Particle(iTrackN);
39853df5	160	if(!pParticle) {AliDebug(1,Form("Not a valid TParticle pointer. Track skipped"));continue;}
39853df5	161	AliDebug(1,Form(" PDG code : %i",pParticle->GetPdgCode()));
ab92fbe6	162	//
	163	// problem of PDG code of some extra particles to be solved!!!!!!!!!
	164	//
	165	// found problem! Look in TRD directory : codes from Fluka are :
	166	//
	167	// if ((pdg_code == 10010020) \|\|
	168	// (pdg_code == 10010030) \|\|
	169	// (pdg_code == 50000050) \|\|
	170	// (pdg_code == 50000051) \|\|
	171	// (pdg_code == 10020040)) {
	172	//
	173	if(pParticle->GetPdgCode()>=50000050\|\|pParticle->GetPdgCode()==0\|\|pParticle->GetPdgCode()>10000) {AliDebug(1,Form("A photon as track... Track skipped"));continue;}
	174	//
	175	// to be updated for us!!
	176	//
9f873529	177	AliDebug(1,Form("Track %i is a %s with charge %i and momentum %f",
	178	iTrackN,pParticle->GetPDG()->GetName(),Int_t(pParticle->GetPDG()->Charge()),pParticle->P()));
	179	// if(pParticle->GetMother(0)!=-1) continue; //consider only primaries
	180	if(pParticle->GetPDG()->Charge()==0\|\|TMath::Abs(Int_t(pParticle->GetPDG()->Charge()))!=3) continue; //to avoid photons from stack...
1cb5a97c	181	hnvec[0]=pParticle->P();
1cb5a97c	182	hnvec[1]=pParticle->GetPDG()->Charge();
30c60010	183	p0.SetMagThetaPhi(pParticle->P(),pParticle->Theta(),pParticle->Phi());
30c60010	184	x0.SetXYZ(pParticle->Vx(),pParticle->Vy(),pParticle->Vz());
9f873529	185	AliRICHHelix helix(x0,p0,TMath::Sign(1,(Int_t)pParticle->GetPDG()->Charge()),b);
30c60010	186	Int_t iChamber=helix.RichIntersect(pRich->P());
ef210ba6	187	hnvec[2]=helix.Ploc().Theta();
ef210ba6	188	hnvec[3]=helix.Ploc().Phi();
30c60010	189	AliDebug(1,Form("intersection with %i chamber found",iChamber));
1cb5a97c	190	if(!iChamber) continue;// no intersection with RICH found
	191	hnvec[4]=helix.PosPc().X();
	192	hnvec[5]=helix.PosPc().Y();
	193	Double_t distMip=9999; //min distance between clusters and track position on PC
0fe8fa07	194	Double_t mipX=-1; //min distance between clusters and track position on PC
	195	Double_t mipY=-1; //min distance between clusters and track position on PC
	196	Double_t chargeMip=-1; // charge MIP to find
	197	Int_t iMipId=-1; //index of that min distance cluster
30c60010	198	for(Int_t iClusN=0;iClusN<pRich->Clusters(iChamber)->GetEntries();iClusN++){//clusters loop for intersected chamber
0fe8fa07	199	AliRICHCluster pClus=(AliRICHCluster)pRich->Clusters(iChamber)->UncheckedAt(iClusN);//get pointer to current cluster
30c60010	200	Double_t distCurrent=pClus->DistTo(helix.PosPc());//ditance between current cluster and helix intersection with PC
1cb5a97c	201	if(distCurrent<distMip){distMip=distCurrent;mipX=pClus->X();
	202	mipY=pClus->Y();
	203	chargeMip=pClus->Q();iMipId=1000000*iChamber+iClusN;}//find cluster nearest to the track
30c60010	204
	205	AliDebug(1,Form("Ploc (%f,%f,%f) dist= %f",helix.Ploc().Mag(),helix.Ploc().Theta()*TMath::RadToDeg(),
	206	helix.Ploc().Phi()*TMath::RadToDeg(),pClus->DistTo(helix.PosPc())));
a25b3368	207	}//clusters loop for intersected chamber
30c60010	208
30c60010	209	AliDebug(1,Form("Min distance cluster: %i dist is %f",iMipId,distMip));
1cb5a97c	210	hnvec[6]=mipX;hnvec[7]=mipY;
1cb5a97c	211	hnvec[8]=chargeMip;
30c60010	212	AliRICHRecon recon(&helix,pRich->Clusters(iChamber),iMipId);
30c60010	213	Double_t thetaCerenkov=recon.ThetaCerenkov(); //search for mean Cerenkov angle for this track
1cb5a97c	214	hnvec[9]=thetaCerenkov;
	215	hnvec[10]=recon.GetHoughPhotons();
	216	hnvec[11]=(Double_t)iMipId;
ab92fbe6	217	hnvec[12]=(Double_t)iChamber;
ab92fbe6	218	hnvec[13]=(Double_t)pParticle->GetPdgCode();
1cb5a97c	219	if(hn) hn->Fill(hnvec);
30c60010	220	AliDebug(1,Form("FINAL Theta Cerenkov=%f",thetaCerenkov));
a25b3368	221	}//stack particles loop
1cb5a97c	222
39853df5	223	pRich->GetLoader()->UnloadRecPoints();
30c60010	224	AliDebug(1,"Stop.");
a25b3368	225	}//RecWithStack
a25b3368	226	//__________________________________________________________________________________________________
998b831f	227	Int_t AliRICHTracker::LoadClusters(TTree *pTree)
	228	{
	229	// Load clusters for RICH
	230	AliDebug(1,"Start."); pTree->GetEntry(0); AliDebug(1,"Stop."); return 0;
	231	}
30c60010	232	//__________________________________________________________________________________________________
101624cd	233	void AliRICHTracker::CalcProb(Double_t thetaCer,Double_t pmod, Double_t sigmaPID, Double_t richPID)
30c60010	234	{
12b0e5f4	235	// Calculates probability to be a electron-muon-pion-kaon-proton
a25b3368	236	// from the given Cerenkov angle and momentum assuming no initial particle composition
12b0e5f4	237	// (i.e. apriory probability to be the particle of the given sort is the same for all sorts)
304864ab	238	Double_t height[AliPID::kSPECIES];Double_t totalHeight=0;
101624cd	239	Double_t thetaTh[AliPID::kSPECIES];
304864ab	240	for(Int_t iPart=0;iPart<AliPID::kSPECIES;iPart++){
101624cd	241	height[iPart]=0;
068217e3	242	Double_t mass = AliRICHParam::fgMass[iPart];
101624cd	243	Double_t refIndex=AliRICHParam::RefIdxC6F14(AliRICHParam::MeanCkovEnergy());
30c60010	244	Double_t cosThetaTh = TMath::Sqrt(massmass+pmodpmod)/(refIndex*pmod);
101624cd	245	thetaTh[iPart]=0;
	246	if(cosThetaTh>=1) continue;
	247	thetaTh[iPart] = TMath::ACos(cosThetaTh);
	248	// Double_t sinThetaThNorm = TMath::Sin(thetaTh)/TMath::Sqrt(1-1/(refIndex*refIndex));
	249	// Double_t sigmaThetaTh = (0.014(1/sinThetaThNorm-1) + 0.0043)1.25;
	250	// height[iPart] = TMath::Gaus(thetaCer,thetaTh,sigmaThetaTh);
01c81d9d	251	if(sigmaPID[iPart]>0) height[iPart] = TMath::Gaus(thetaCer,thetaTh[iPart],sigmaPID[iPart],kTRUE);
01c81d9d	252	else height[iPart] = 0;
30c60010	253	totalHeight +=height[iPart];
101624cd	254	AliDebug(1,Form(" Particle %s with mass %f with height %f and thetaTH %f",AliPID::ParticleName(iPart),mass,height[iPart],thetaTh[iPart]));
12b0e5f4	255	AliDebug(1,Form(" partial height %15.14f total height %15.14f",height[iPart],totalHeight));
30c60010	256	}
12b0e5f4	257	if(totalHeight<1e-5) {for(Int_t iPart=0;iPart<AliPID::kSPECIES;iPart++)richPID[iPart]=1.0/AliPID::kSPECIES;return;}
101624cd	258	for(Int_t iPart=0;iPart<AliPID::kSPECIES;iPart++) richPID[iPart] = height[iPart]/totalHeight;
101624cd	259	Int_t iPartNear = TMath::LocMax(AliPID::kSPECIES,richPID);
01c81d9d	260	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	261	//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	262
30c60010	263	}//CalcProb
a25b3368	264	//__________________________________________________________________________________________________