[u/mrichter/AliRoot.git] / HMPID / AliHMPIDTracker.cxx

#include "AliHMPIDTracker.h"     //class header
#include "AliHMPIDtrack.h"     //class header
#include "AliHMPIDCluster.h"     //GetTrackPoint(),PropagateBack() 
#include "AliHMPIDParam.h"       //GetTrackPoint(),PropagateBack()
#include "AliHMPIDRecon.h"       //Recon()
#include "AliHMPIDReconHTA.h"    //ReconHTA()
#include <AliESDEvent.h>         //PropagateBack(),Recon()  
#include <AliESDtrack.h>         //Intersect()  
#include <AliRun.h>              //GetTrackPoint(),PropagateBack()  
#include <AliTrackPointArray.h>  //GetTrackPoint()
#include <AliAlignObj.h>         //GetTrackPoint()
#include <AliCDBManager.h>       //PropageteBack()
#include <AliCDBEntry.h>         //PropageteBack()
//.
// HMPID base class fo tracking
//.
//.
//.
ClassImp(AliHMPIDTracker)
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
AliHMPIDTracker::AliHMPIDTracker():
  AliTracker(),
  fClu(new TObjArray(AliHMPIDParam::kMaxCh+1))  
{
// ctor. Create TObjArray of TClonesArray of AliHMPIDCluster  
// 
//  
  fClu->SetOwner(kTRUE);
  for(int i=AliHMPIDParam::kMinCh;i<=AliHMPIDParam::kMaxCh;i++) fClu->AddAt(new TClonesArray("AliHMPIDCluster"),i);
}//ctor
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++    
Bool_t AliHMPIDTracker::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 HMPID 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(AliGeomManager::LayerToVolUID(AliGeomManager::kHMPID,iCham-1));//layer and chamber number
  TClonesArray *pArr=(TClonesArray*)(*fClu)[iCham];
  AliHMPIDCluster *pClu=(AliHMPIDCluster*)pArr->UncheckedAt(iClu);//get pointer to cluster
  Double_t mars[3];
  AliHMPIDParam::Instance()->Lors2Mars(iCham,pClu->X(),pClu->Y(),mars);
  point.SetXYZ(mars[0],mars[1],mars[2]);
  return kTRUE;
}//GetTrackPoint()
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Int_t AliHMPIDTracker::IntTrkCha(AliESDtrack *pTrk,Float_t &xPc,Float_t &yPc,Float_t &xRa,Float_t &yRa,Float_t &theta,Float_t &phi)
{
// Static method to find intersection in between given track and HMPID chambers
// Arguments: pTrk- ESD track; xPc,yPc- track intersection with PC in LORS [cm]
//   Returns: intersected chamber ID or -1
  AliHMPIDParam *pParam=AliHMPIDParam::Instance();
  for(Int_t i=AliHMPIDParam::kMinCh;i<=AliHMPIDParam::kMaxCh;i++){                              //chambers loop
    Double_t p1[3],n1[3]; pParam->Norm(i,n1); pParam->Point(i,p1,AliHMPIDParam::kRad);          //point & norm  for middle of radiator plane
    Double_t p2[3],n2[3]; pParam->Norm(i,n2); pParam->Point(i,p2,AliHMPIDParam::kPc);           //point & norm  for entrance to PC plane
    if(pTrk->Intersect(p1,n1,-GetBz())==kFALSE) continue;                                       //try to intersect track with the middle of radiator
    if(pTrk->Intersect(p2,n2,-GetBz())==kFALSE) continue;                                       //try to intersect track with PC
    pParam->Mars2LorsVec(i,n1,theta,phi);                                                       //track angles at RAD
    pParam->Mars2Lors   (i,p1,xRa,yRa);                                                         //TRKxRAD position
    pParam->Mars2Lors   (i,p2,xPc,yPc);                                                         //TRKxPC position
    if(AliHMPIDParam::IsInside(xPc,yPc,pParam->DistCut())==kTRUE) return i;                     //return intersected chamber  
  }                                                                                             //chambers loop
  return -1;                                                                                    //no intersection with HMPID chambers
}//IntTrkCha()
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Int_t AliHMPIDTracker::IntTrkCha(Int_t ch,AliHMPIDtrack *pTrk,Float_t &xPc,Float_t &yPc,Float_t &xRa,Float_t &yRa,Float_t &theta,Float_t &phi)
{
// Static method to find intersection in between given track and HMPID chambers
// Arguments: pTrk- HMPID track; xPc,yPc- track intersection with PC in LORS [cm]
//   Returns: intersected chamber ID or -1
    AliHMPIDParam *pParam=AliHMPIDParam::Instance();
    Double_t p1[3],n1[3]; 
    pParam->Norm(ch,n1); 
    pParam->Point(ch,p1,AliHMPIDParam::kRad);                                                    //point & norm  for middle of radiator plane
    Double_t p2[3],n2[3]; 
    pParam->Norm(ch,n2); 
    pParam->Point(ch,p2,AliHMPIDParam::kPc);                                                     //point & norm  for entrance to PC plane
    if(pTrk->Intersect(pTrk,p1,n1)==kFALSE) return -1;                                           //try to intersect track with the middle of radiator
    if(pTrk->Intersect(pTrk,p2,n2)==kFALSE) return -1;   
    pParam->Mars2LorsVec(ch,n1,theta,phi);                                                       //track angles at RAD
    pParam->Mars2Lors   (ch,p1,xRa,yRa);                                                         //TRKxRAD position
    pParam->Mars2Lors   (ch,p2,xPc,yPc);                                                         //TRKxPC position
    if(AliHMPIDParam::IsInside(xPc,yPc,pParam->DistCut())==kTRUE) return ch;                     //return intersected chamber  
  return -1;                                                                                     //no intersection with HMPID chambers
}//IntTrkCha()
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Int_t AliHMPIDTracker::LoadClusters(TTree *pCluTree)
{
// Interface callback methode invoked from AliReconstruction::RunTracking() to load HMPID clusters before PropagateBack() gets control. Done once per event.
// Arguments: pCluTree- pointer to clusters tree got by AliHMPIDLoader::LoadRecPoints("read") then AliHMPIDLoader::TreeR()
//   Returns: error code (currently ignored in AliReconstruction::RunTraking())    
  for(int i=AliHMPIDParam::kMinCh;i<=AliHMPIDParam::kMaxCh;i++) pCluTree->SetBranchAddress(Form("HMPID%d",i),&((*fClu)[i]));
  pCluTree->GetEntry(0);
  return 0;  
}//LoadClusters()
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Int_t AliHMPIDTracker::PropagateBack(AliESDEvent *pEsd)
{
// Interface pure virtual in AliTracker. Invoked from AliReconstruction::RunTracking() after invocation of AliTracker::LoadClusters() once per event
// Agruments: pEsd - pointer to ESD
//   Returns: error code    
  AliCDBEntry *pNmeanEnt =AliCDBManager::Instance()->Get("HMPID/Calib/Nmean"); //contains TObjArray of 42 TF1 + 1 EPhotMean
  AliCDBEntry *pQthreEnt =AliCDBManager::Instance()->Get("HMPID/Calib/Qthre"); //contains TObjArray of 42 (7ch * 6sec) TF1
  if(!pNmeanEnt) AliFatal("No Nmean C6F14 ");
  if(!pQthreEnt) AliFatal("No Qthre");
    
  return Recon(pEsd,fClu,(TObjArray*)pNmeanEnt->GetObject(),(TObjArray*)pQthreEnt->GetObject());  
}//PropagateBack()
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Int_t AliHMPIDTracker::Recon(AliESDEvent *pEsd,TObjArray *pClus,TObjArray *pNmean, TObjArray *pQthre)
{
// Static method to reconstruct Theta Ckov for all valid tracks of a given event.
// Arguments: pEsd- pointer ESD; pClu- pointer to clusters for all chambers; pNmean - pointer to all function Nmean=f(time)
//   Returns: error code, 0 if no errors   
  
  AliHMPIDRecon recon;                                                                           //instance of reconstruction class, nothing important in ctor
  Float_t xPc,yPc,xRa,yRa,theta,phi;
  Double_t cluLORS[2]={0},cluMARS[3]={0},trkMARS[3]={0};
//  Double_t bestcluMARS[3]={0,0,0};
  Double_t radClu,radInitTrk;   
  Int_t nMipClusTot=0;
  Double_t d3d=0,dmin=999999,bz=0;
  Bool_t isMatched=kFALSE;
  Int_t bestCluCh=-1;
  Int_t cluSiz=0;
  Double_t qthre = 0;   Double_t nmean=0; Int_t cham=0; Int_t hvsec=0;
  Int_t index=0;                                                                                //index of the "best" matching cluster
  Double_t bestChi2=-1;                                                                         //Chi2 of the "best" matching cluster
  Double_t chi2=0;   
  Int_t nClusCh[AliHMPIDParam::kMaxCh+1];
  Bool_t isOkQcut=kFALSE;
  Bool_t isOkDcut=kFALSE;
  
  AliHMPIDParam *pParam = AliHMPIDParam::Instance();                                             //Instance of AliHMPIDParam
  
  for(Int_t iTrk=0;iTrk<pEsd->GetNumberOfTracks();iTrk++){                                        //loop on the ESD tracks in the event
    isMatched=kFALSE;bestCluCh=-1;dmin=999999;bestChi2=99999;chi2=99999;cluSiz=0;                 //init. track matching params
    isOkQcut = kFALSE;
    AliHMPIDCluster *bestHmpCluster=0x0;                                                          //the best matching cluster
    AliESDtrack *pTrk = pEsd->GetTrack(iTrk);                                                     //get reconstructed track    
    AliHMPIDtrack *hmpTrk = new AliHMPIDtrack(*pTrk);                                             //create a hmpid track to be used for propagation and matching 
    bz=AliTracker::GetBz();  
    
    Int_t ipCh=IntTrkCha(pTrk,xPc,yPc,xRa,yRa,theta,phi);
    if(ipCh<0) {                                                                                 //no intersection at all, go after next track
      pTrk->SetHMPIDtrk(0,0,0,0);                                                                //no intersection found
      pTrk->SetHMPIDcluIdx   (99,99999);                                                         //chamber not found, mip not yet considered
      pTrk->SetHMPIDsignal(AliHMPIDRecon::kNotPerformed);                                        //ring reconstruction not yet performed
      continue;                                                                         
    }
    
// track intersects the chamber ipCh: find the MIP          
    
    TClonesArray *pMipCluLst=(TClonesArray *)pClus->At(ipCh);                                   //get the list of clusters
    nMipClusTot = pMipCluLst->GetEntries();                                                     //total number of clusters in the given chamber
    nClusCh[ipCh] = nMipClusTot;
    
    for (Int_t iClu=0; iClu<nMipClusTot;iClu++) {                                               //clusters loop
      
      AliHMPIDCluster *pClu=(AliHMPIDCluster*)pMipCluLst->UncheckedAt(iClu);                    //get the cluster
// evaluate qThre
      if(pQthre->GetEntriesFast()==pParam->kMaxCh+1) {                                             // just for backward compatibility
        qthre=((TF1*)pQthre->At(pClu->Ch()))->Eval(pEsd->GetTimeStamp());                          //
      } else {                                                                                     // in the past just 1 qthre
        hvsec = pParam->InHVSector(pClu->Y());                                              //  per chamber
        if(hvsec>=0)
	  qthre=((TF1*)pQthre->At(6*cham+hvsec))->Eval(pEsd->GetTimeStamp());                      //
      }                                                                                            //
//
      if(pClu->Q()<qthre) continue;                                                                      //charge compartible with MIP clusters      
      isOkQcut = kTRUE;

      cluLORS[0]=pClu->X(); cluLORS[1]=pClu->Y();                                            //get the LORS coordinates of the cluster
      pParam->Lors2Mars(ipCh,cluLORS[0],cluLORS[1],cluMARS);              //convert cluster coors. from LORS to MARS
      radClu=TMath::Sqrt(cluMARS[0]*cluMARS[0]+cluMARS[1]*cluMARS[1]);                       //radial distance of candidate cluster in MARS                                          
      Double_t trkx0[3]; 
      hmpTrk->GetXYZ(trkx0);                                                                 //get track position in MARS
      radInitTrk=TMath::Sqrt(trkx0[0]*trkx0[0]+trkx0[1]*trkx0[1]);
      hmpTrk->PropagateToR(radClu,10);
      hmpTrk->GetXYZ(trkx0);                                                                   //get track position in MARS
      hmpTrk->GetXYZAt(radClu,bz,trkMARS);                                                     //get the track coordinates at the rad distance after prop. 
      d3d=TMath::Sqrt((cluMARS[0]-trkMARS[0])*(cluMARS[0]-trkMARS[0])+(cluMARS[1]-trkMARS[1])*(cluMARS[1]-trkMARS[1])+(cluMARS[2]-trkMARS[2])*(cluMARS[2]-trkMARS[2]));
      chi2=hmpTrk->GetPredictedChi2(pClu);
      if(dmin > d3d ) {                                                                         //to be saved for the moment...
        cluSiz = pClu->Size();
        dmin=d3d;
        bestCluCh=ipCh;
        bestHmpCluster=pClu;
        index=iClu;
        bestChi2=chi2;
        cluLORS[0]=pClu->X(); cluLORS[1]=pClu->Y();
//        pParam->Lors2Mars(ipCh,cluLORS[0],cluLORS[1],bestcluMARS); 
      }//global dmin cut 
    }//clus loop

    if(!isOkQcut) {
      pTrk->SetHMPIDcluIdx(ipCh,9999);                                                          
      pTrk->SetHMPIDsignal(pParam->kMipQdcCut);
      continue;                                                                     
    }
    
    if(dmin < pParam->DistCut()) {
      isOkDcut = kTRUE;
    }

    if(!isOkDcut) {
      pTrk->SetHMPIDcluIdx(ipCh,index+1000*cluSiz);                                             //set chamber, index of cluster + cluster size
      pTrk->SetHMPIDsignal(pParam->kMipDistCut);                                                //closest cluster with enough charge is still too far from intersection
    }
    
    if(isOkQcut*isOkDcut) isMatched = kTRUE;                                                    // MIP-Track matched !!    
    
    if(!isMatched) continue;                                                                    // If matched continue...
    
    Int_t indexAll = 0;
    for(Int_t iC=0;iC<bestCluCh;iC++) indexAll+=nClusCh[iC]; indexAll+=index;                    //to be verified...

    Bool_t isOk = hmpTrk->Update(bestHmpCluster,bestChi2,indexAll);
    if(!isOk) continue;
    pTrk->SetOuterParam((AliExternalTrackParam*)&hmpTrk,AliESDtrack::kHMPIDout);                 

//    cham=IntTrkCha(bestCluCh,hmpTrk,xPc,yPc,xRa,yRa,theta,phi);
    cham=IntTrkCha(pTrk,xPc,yPc,xRa,yRa,theta,phi);
    if(cham<0) {                                                                                  //no intersection at all, go after next track
      pTrk->SetHMPIDtrk(0,0,0,0);                                                                //no intersection found
      pTrk->SetHMPIDcluIdx   (99,99999);                                                         //chamber not found, mip not yet considered
      pTrk->SetHMPIDsignal(AliHMPIDRecon::kNotPerformed);                                        //ring reconstruction not yet performed
      continue;                                                                         
    }

    pTrk->SetHMPIDtrk(xRa,yRa,theta,phi);                                                        //store initial infos
    //evaluate nMean
    if(pNmean->GetEntries()==21) {                                                              //for backward compatibility
      nmean=((TF1*)pNmean->At(3*cham))->Eval(pEsd->GetTimeStamp());                             //C6F14 Nmean for this chamber
    } else {
      Int_t iRad     = pParam->Radiator(yRa);                                                   //evaluate the radiator involved
      Double_t tLow  = ((TF1*)pNmean->At(6*cham+2*iRad  ))->Eval(pEsd->GetTimeStamp());         //C6F14 low  temp for this chamber
      Double_t tHigh = ((TF1*)pNmean->At(6*cham+2*iRad+1))->Eval(pEsd->GetTimeStamp());         //C6F14 high temp for this chamber
      Double_t tExp  = pParam->FindTemp(tLow,tHigh,yRa);                                        //estimated temp for that chamber at that y
      nmean = pParam->NIdxRad(AliHMPIDParam::Instance()->GetEPhotMean(),tExp);                  //mean ref idx @ a given temp
      if(nmean < 0){                                                                            //track didn' t pass through the radiator
         pTrk->SetHMPIDsignal(AliHMPIDRecon::kNoRad);                                           //set the appropriate flag
         pTrk->SetHMPIDcluIdx(ipCh,index+1000*cluSiz);                                          //set index of cluster
         continue;
      }
    }
    //
    recon.SetImpPC(xPc,yPc);                                                                     //store track impact to PC
    recon.CkovAngle(pTrk,(TClonesArray *)pClus->At(cham),index,nmean);                           //search for Cerenkov angle of this track
  }//iTrk

  return 0; // error code: 0=no error;
}//Recon()
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Int_t AliHMPIDTracker::ReconHiddenTrk(Int_t iCh,Int_t iHVsec,AliESDtrack *pTrk,TClonesArray *pCluLst,TObjArray *pNmean,TObjArray *pQthre)
{
// Static method to reconstruct Theta Ckov for all valid tracks of a given event.
// Arguments: pEsd- pointer ESD; pClu- pointer to clusters for all chambers; pNmean - pointer to all function Nmean=f(time), pQthre - pointer to all function Qthre=f(time)
//   Returns: error code, 0 if no errors
  AliHMPIDReconHTA reconHTA;                                                                          //instance of reconstruction class, nothing important in ctor
  Double_t nmean=((TF1*)pNmean->At(3*iCh))->Eval(0);                                            //C6F14 Nmean for this chamber
  Double_t qthre = 0;
  if(pQthre->GetEntriesFast()==AliHMPIDParam::kMaxCh+1)                                         //
    qthre=((TF1*)pQthre->At(iCh))->Eval(0);                                                     //just for backward compatibi
  else  qthre=((TF1*)pQthre->At(6*iCh+iHVsec))->Eval(0);                                        //
  if(pCluLst->GetEntriesFast()<4) return 1;                                                     //min 4 clusters (3 + 1 mip) to find a ring! 
  if(reconHTA.CkovHiddenTrk(pTrk,pCluLst,nmean,qthre)) return 0;                                   //search for track parameters and Cerenkov angle of this track
  else return 1;                                                                                // error code: 0=no error,1=fit not performed;
}//Recon()
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
void AliHMPIDTracker::FillClusterArray(TObjArray* array) const {
  
 // Publishes all pointers to clusters known to the tracker into the
  // passed object array.
  // The ownership is not transfered - the caller is not expected to delete
  // the clusters
 
  for(Int_t iCh=AliHMPIDParam::kMinCh;iCh<=AliHMPIDParam::kMaxCh;iCh++){    
    TClonesArray *pCluArr=(TClonesArray*)(*fClu)[iCh];
    for (Int_t iClu=0; iClu<pCluArr->GetEntriesFast();iClu++){
      AliHMPIDCluster *pClu=(AliHMPIDCluster*)pCluArr->UncheckedAt(iClu);    
      array->AddLast(pClu);
    }//cluster loop in iCh
    pCluArr->Delete();
  }//Ch loop
    
  return;
}
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Commit	Line	Data
3c6274c1	1	#include "AliHMPIDTracker.h" //class header
496c71b0	2	#include "AliHMPIDtrack.h" //class header
3c6274c1	3	#include "AliHMPIDCluster.h" //GetTrackPoint(),PropagateBack()
3c6274c1	4	#include "AliHMPIDParam.h" //GetTrackPoint(),PropagateBack()
59d9d4b3	5	#include "AliHMPIDRecon.h" //Recon()
57fcdc2b	6	#include "AliHMPIDReconHTA.h" //ReconHTA()
	7	#include <AliESDEvent.h> //PropagateBack(),Recon()
	8	#include <AliESDtrack.h> //Intersect()
d3da6dc4	9	#include <AliRun.h> //GetTrackPoint(),PropagateBack()
	10	#include <AliTrackPointArray.h> //GetTrackPoint()
	11	#include <AliAlignObj.h> //GetTrackPoint()
59d9d4b3	12	#include <AliCDBManager.h> //PropageteBack()
59d9d4b3	13	#include <AliCDBEntry.h> //PropageteBack()
423554a3	14	//.
	15	// HMPID base class fo tracking
	16	//.
	17	//.
	18	//.
d3da6dc4	19	ClassImp(AliHMPIDTracker)
94b1fbfa	20	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
c61a7285	21	AliHMPIDTracker::AliHMPIDTracker():
	22	AliTracker(),
	23	fClu(new TObjArray(AliHMPIDParam::kMaxCh+1))
94b1fbfa	24	{
59d9d4b3	25	// ctor. Create TObjArray of TClonesArray of AliHMPIDCluster
	26	//
	27	//
c61a7285	28	fClu->SetOwner(kTRUE);
ae5a42aa	29	for(int i=AliHMPIDParam::kMinCh;i<=AliHMPIDParam::kMaxCh;i++) fClu->AddAt(new TClonesArray("AliHMPIDCluster"),i);
59d9d4b3	30	}//ctor
d3da6dc4	31	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
	32	Bool_t AliHMPIDTracker::GetTrackPoint(Int_t idx, AliTrackPoint& point) const
	33	{
	34	// Interface callback methode invoked from AliReconstruction::WriteAlignmentData() to get position of MIP cluster in MARS associated to a current track.
	35	// MIP cluster is reffered by index which is stored in AliESDtrack ???????
	36	// Arguments: idx- cluster index which is stored by HMPID in AliESDtrack
	37	// point- reference to the object where to store the point
	38	// Returns: status of operation if FALSE then AliReconstruction::WriteAlignmentData() do not store this point to array of points for current track.
	39	if(idx<0) return kFALSE; //no MIP cluster assigned to this track in PropagateBack()
59d9d4b3	40	Int_t iCham=idx/1000000; Int_t iClu=idx%1000000;
ae079791	41	point.SetVolumeID(AliGeomManager::LayerToVolUID(AliGeomManager::kHMPID,iCham-1));//layer and chamber number
94b1fbfa	42	TClonesArray pArr=(TClonesArray)(*fClu)[iCham];
94b1fbfa	43	AliHMPIDCluster pClu=(AliHMPIDCluster)pArr->UncheckedAt(iClu);//get pointer to cluster
d3da6dc4	44	Double_t mars[3];
	45	AliHMPIDParam::Instance()->Lors2Mars(iCham,pClu->X(),pClu->Y(),mars);
	46	point.SetXYZ(mars[0],mars[1],mars[2]);
	47	return kTRUE;
59d9d4b3	48	}//GetTrackPoint()
59d9d4b3	49	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
39cd22e6	50	Int_t AliHMPIDTracker::IntTrkCha(AliESDtrack *pTrk,Float_t &xPc,Float_t &yPc,Float_t &xRa,Float_t &yRa,Float_t &theta,Float_t &phi)
59d9d4b3	51	{
	52	// Static method to find intersection in between given track and HMPID chambers
	53	// Arguments: pTrk- ESD track; xPc,yPc- track intersection with PC in LORS [cm]
	54	// Returns: intersected chamber ID or -1
	55	AliHMPIDParam *pParam=AliHMPIDParam::Instance();
ae5a42aa	56	for(Int_t i=AliHMPIDParam::kMinCh;i<=AliHMPIDParam::kMaxCh;i++){ //chambers loop
59d9d4b3	57	Double_t p1[3],n1[3]; pParam->Norm(i,n1); pParam->Point(i,p1,AliHMPIDParam::kRad); //point & norm for middle of radiator plane
	58	Double_t p2[3],n2[3]; pParam->Norm(i,n2); pParam->Point(i,p2,AliHMPIDParam::kPc); //point & norm for entrance to PC plane
	59	if(pTrk->Intersect(p1,n1,-GetBz())==kFALSE) continue; //try to intersect track with the middle of radiator
	60	if(pTrk->Intersect(p2,n2,-GetBz())==kFALSE) continue; //try to intersect track with PC
	61	pParam->Mars2LorsVec(i,n1,theta,phi); //track angles at RAD
	62	pParam->Mars2Lors (i,p1,xRa,yRa); //TRKxRAD position
	63	pParam->Mars2Lors (i,p2,xPc,yPc); //TRKxPC position
39cd22e6	64	if(AliHMPIDParam::IsInside(xPc,yPc,pParam->DistCut())==kTRUE) return i; //return intersected chamber
59d9d4b3	65	} //chambers loop
39cd22e6	66	return -1; //no intersection with HMPID chambers
59d9d4b3	67	}//IntTrkCha()
d3da6dc4	68	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
496c71b0	69	Int_t AliHMPIDTracker::IntTrkCha(Int_t ch,AliHMPIDtrack *pTrk,Float_t &xPc,Float_t &yPc,Float_t &xRa,Float_t &yRa,Float_t &theta,Float_t &phi)
	70	{
	71	// Static method to find intersection in between given track and HMPID chambers
	72	// Arguments: pTrk- HMPID track; xPc,yPc- track intersection with PC in LORS [cm]
	73	// Returns: intersected chamber ID or -1
	74	AliHMPIDParam *pParam=AliHMPIDParam::Instance();
	75	Double_t p1[3],n1[3];
	76	pParam->Norm(ch,n1);
	77	pParam->Point(ch,p1,AliHMPIDParam::kRad); //point & norm for middle of radiator plane
	78	Double_t p2[3],n2[3];
	79	pParam->Norm(ch,n2);
	80	pParam->Point(ch,p2,AliHMPIDParam::kPc); //point & norm for entrance to PC plane
	81	if(pTrk->Intersect(pTrk,p1,n1)==kFALSE) return -1; //try to intersect track with the middle of radiator
	82	if(pTrk->Intersect(pTrk,p2,n2)==kFALSE) return -1;
	83	pParam->Mars2LorsVec(ch,n1,theta,phi); //track angles at RAD
	84	pParam->Mars2Lors (ch,p1,xRa,yRa); //TRKxRAD position
	85	pParam->Mars2Lors (ch,p2,xPc,yPc); //TRKxPC position
	86	if(AliHMPIDParam::IsInside(xPc,yPc,pParam->DistCut())==kTRUE) return ch; //return intersected chamber
	87	return -1; //no intersection with HMPID chambers
	88	}//IntTrkCha()
	89	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
d3da6dc4	90	Int_t AliHMPIDTracker::LoadClusters(TTree *pCluTree)
d3da6dc4	91	{
59d9d4b3	92	// Interface callback methode invoked from AliReconstruction::RunTracking() to load HMPID clusters before PropagateBack() gets control. Done once per event.
d3da6dc4	93	// Arguments: pCluTree- pointer to clusters tree got by AliHMPIDLoader::LoadRecPoints("read") then AliHMPIDLoader::TreeR()
d3da6dc4	94	// Returns: error code (currently ignored in AliReconstruction::RunTraking())
ae5a42aa	95	for(int i=AliHMPIDParam::kMinCh;i<=AliHMPIDParam::kMaxCh;i++) pCluTree->SetBranchAddress(Form("HMPID%d",i),&((*fClu)[i]));
94b1fbfa	96	pCluTree->GetEntry(0);
94b1fbfa	97	return 0;
59d9d4b3	98	}//LoadClusters()
d3da6dc4	99	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
af885e0f	100	Int_t AliHMPIDTracker::PropagateBack(AliESDEvent *pEsd)
abb5f786	101	{
59d9d4b3	102	// Interface pure virtual in AliTracker. Invoked from AliReconstruction::RunTracking() after invocation of AliTracker::LoadClusters() once per event
abb5f786	103	// Agruments: pEsd - pointer to ESD
abb5f786	104	// Returns: error code
f455af6e	105	AliCDBEntry *pNmeanEnt =AliCDBManager::Instance()->Get("HMPID/Calib/Nmean"); //contains TObjArray of 42 TF1 + 1 EPhotMean
49881df7	106	AliCDBEntry pQthreEnt =AliCDBManager::Instance()->Get("HMPID/Calib/Qthre"); //contains TObjArray of 42 (7ch 6sec) TF1
496c71b0	107	if(!pNmeanEnt) AliFatal("No Nmean C6F14 ");
496c71b0	108	if(!pQthreEnt) AliFatal("No Qthre");
94b1fbfa	109
afe12692	110	return Recon(pEsd,fClu,(TObjArray)pNmeanEnt->GetObject(),(TObjArray)pQthreEnt->GetObject());
611e810d	111	}//PropagateBack()
abb5f786	112	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
afe12692	113	Int_t AliHMPIDTracker::Recon(AliESDEvent pEsd,TObjArray pClus,TObjArray pNmean, TObjArray pQthre)
d3da6dc4	114	{
59d9d4b3	115	// Static method to reconstruct Theta Ckov for all valid tracks of a given event.
59d9d4b3	116	// Arguments: pEsd- pointer ESD; pClu- pointer to clusters for all chambers; pNmean - pointer to all function Nmean=f(time)
d3da6dc4	117	// Returns: error code, 0 if no errors
496c71b0	118
496c71b0	119	AliHMPIDRecon recon; //instance of reconstruction class, nothing important in ctor
39cd22e6	120	Float_t xPc,yPc,xRa,yRa,theta,phi;
496c71b0	121	Double_t cluLORS[2]={0},cluMARS[3]={0},trkMARS[3]={0};
	122	// Double_t bestcluMARS[3]={0,0,0};
	123	Double_t radClu,radInitTrk;
	124	Int_t nMipClusTot=0;
	125	Double_t d3d=0,dmin=999999,bz=0;
	126	Bool_t isMatched=kFALSE;
	127	Int_t bestCluCh=-1;
	128	Int_t cluSiz=0;
	129	Double_t qthre = 0; Double_t nmean=0; Int_t cham=0; Int_t hvsec=0;
	130	Int_t index=0; //index of the "best" matching cluster
	131	Double_t bestChi2=-1; //Chi2 of the "best" matching cluster
	132	Double_t chi2=0;
	133	Int_t nClusCh[AliHMPIDParam::kMaxCh+1];
	134	Bool_t isOkQcut=kFALSE;
	135	Bool_t isOkDcut=kFALSE;
	136
	137	AliHMPIDParam *pParam = AliHMPIDParam::Instance(); //Instance of AliHMPIDParam
	138
	139	for(Int_t iTrk=0;iTrk<pEsd->GetNumberOfTracks();iTrk++){ //loop on the ESD tracks in the event
	140	isMatched=kFALSE;bestCluCh=-1;dmin=999999;bestChi2=99999;chi2=99999;cluSiz=0; //init. track matching params
	141	isOkQcut = kFALSE;
	142	AliHMPIDCluster *bestHmpCluster=0x0; //the best matching cluster
	143	AliESDtrack *pTrk = pEsd->GetTrack(iTrk); //get reconstructed track
	144	AliHMPIDtrack hmpTrk = new AliHMPIDtrack(pTrk); //create a hmpid track to be used for propagation and matching
	145	bz=AliTracker::GetBz();
	146
	147	Int_t ipCh=IntTrkCha(pTrk,xPc,yPc,xRa,yRa,theta,phi);
	148	if(ipCh<0) { //no intersection at all, go after next track
	149	pTrk->SetHMPIDtrk(0,0,0,0); //no intersection found
	150	pTrk->SetHMPIDcluIdx (99,99999); //chamber not found, mip not yet considered
	151	pTrk->SetHMPIDsignal(AliHMPIDRecon::kNotPerformed); //ring reconstruction not yet performed
	152	continue;
	153	}
	154
	155	// track intersects the chamber ipCh: find the MIP
	156
	157	TClonesArray pMipCluLst=(TClonesArray )pClus->At(ipCh); //get the list of clusters
	158	nMipClusTot = pMipCluLst->GetEntries(); //total number of clusters in the given chamber
	159	nClusCh[ipCh] = nMipClusTot;
	160
	161	for (Int_t iClu=0; iClu<nMipClusTot;iClu++) { //clusters loop
	162
	163	AliHMPIDCluster pClu=(AliHMPIDCluster)pMipCluLst->UncheckedAt(iClu); //get the cluster
	164	// evaluate qThre
	165	if(pQthre->GetEntriesFast()==pParam->kMaxCh+1) { // just for backward compatibility
	166	qthre=((TF1*)pQthre->At(pClu->Ch()))->Eval(pEsd->GetTimeStamp()); //
	167	} else { // in the past just 1 qthre
	168	hvsec = pParam->InHVSector(pClu->Y()); // per chamber
	169	if(hvsec>=0)
	170	qthre=((TF1)pQthre->At(6cham+hvsec))->Eval(pEsd->GetTimeStamp()); //
	171	} //
	172	//
	173	if(pClu->Q()<qthre) continue; //charge compartible with MIP clusters
	174	isOkQcut = kTRUE;
	175
	176	cluLORS[0]=pClu->X(); cluLORS[1]=pClu->Y(); //get the LORS coordinates of the cluster
	177	pParam->Lors2Mars(ipCh,cluLORS[0],cluLORS[1],cluMARS); //convert cluster coors. from LORS to MARS
	178	radClu=TMath::Sqrt(cluMARS[0]cluMARS[0]+cluMARS[1]cluMARS[1]); //radial distance of candidate cluster in MARS
	179	Double_t trkx0[3];
	180	hmpTrk->GetXYZ(trkx0); //get track position in MARS
	181	radInitTrk=TMath::Sqrt(trkx0[0]trkx0[0]+trkx0[1]trkx0[1]);
	182	hmpTrk->PropagateToR(radClu,10);
	183	hmpTrk->GetXYZ(trkx0); //get track position in MARS
	184	hmpTrk->GetXYZAt(radClu,bz,trkMARS); //get the track coordinates at the rad distance after prop.
185	d3d=TMath::Sqrt((cluMARS[0]-trkMARS[0])(cluMARS[0]-trkMARS[0])+(cluMARS[1]-trkMARS[1])(cluMARS[1]-trkMARS[1])+(cluMARS[2]-trkMARS[2])*(cluMARS[2]-trkMARS[2]));
186	chi2=hmpTrk->GetPredictedChi2(pClu);
187	if(dmin > d3d ) { //to be saved for the moment...
188	cluSiz = pClu->Size();
189	dmin=d3d;
190	bestCluCh=ipCh;
191	bestHmpCluster=pClu;
192	index=iClu;
193	bestChi2=chi2;
194	cluLORS[0]=pClu->X(); cluLORS[1]=pClu->Y();
195	// pParam->Lors2Mars(ipCh,cluLORS[0],cluLORS[1],bestcluMARS);
196	}//global dmin cut
197	}//clus loop
198
199	if(!isOkQcut) {
200	pTrk->SetHMPIDcluIdx(ipCh,9999);
201	pTrk->SetHMPIDsignal(pParam->kMipQdcCut);
202	continue;
203	}
204
205	if(dmin < pParam->DistCut()) {
206	isOkDcut = kTRUE;
207	}
208
209	if(!isOkDcut) {
210	pTrk->SetHMPIDcluIdx(ipCh,index+1000*cluSiz); //set chamber, index of cluster + cluster size
211	pTrk->SetHMPIDsignal(pParam->kMipDistCut); //closest cluster with enough charge is still too far from intersection
212	}
213
214	if(isOkQcut*isOkDcut) isMatched = kTRUE; // MIP-Track matched !!
215
216	if(!isMatched) continue; // If matched continue...
217
218	Int_t indexAll = 0;
219	for(Int_t iC=0;iC<bestCluCh;iC++) indexAll+=nClusCh[iC]; indexAll+=index; //to be verified...
220
221	Bool_t isOk = hmpTrk->Update(bestHmpCluster,bestChi2,indexAll);
222	if(!isOk) continue;
223	pTrk->SetOuterParam((AliExternalTrackParam*)&hmpTrk,AliESDtrack::kHMPIDout);
224
225	// cham=IntTrkCha(bestCluCh,hmpTrk,xPc,yPc,xRa,yRa,theta,phi);
226	cham=IntTrkCha(pTrk,xPc,yPc,xRa,yRa,theta,phi);
227	if(cham<0) { //no intersection at all, go after next track
39cd22e6	228	pTrk->SetHMPIDtrk(0,0,0,0); //no intersection found
	229	pTrk->SetHMPIDcluIdx (99,99999); //chamber not found, mip not yet considered
	230	pTrk->SetHMPIDsignal(AliHMPIDRecon::kNotPerformed); //ring reconstruction not yet performed
	231	continue;
	232	}
496c71b0	233
39cd22e6	234	pTrk->SetHMPIDtrk(xRa,yRa,theta,phi); //store initial infos
496c71b0	235	//evaluate nMean
	236	if(pNmean->GetEntries()==21) { //for backward compatibility
	237	nmean=((TF1)pNmean->At(3cham))->Eval(pEsd->GetTimeStamp()); //C6F14 Nmean for this chamber
	238	} else {
	239	Int_t iRad = pParam->Radiator(yRa); //evaluate the radiator involved
	240	Double_t tLow = ((TF1)pNmean->At(6cham+2*iRad ))->Eval(pEsd->GetTimeStamp()); //C6F14 low temp for this chamber
	241	Double_t tHigh = ((TF1)pNmean->At(6cham+2*iRad+1))->Eval(pEsd->GetTimeStamp()); //C6F14 high temp for this chamber
	242	Double_t tExp = pParam->FindTemp(tLow,tHigh,yRa); //estimated temp for that chamber at that y
	243	nmean = pParam->NIdxRad(AliHMPIDParam::Instance()->GetEPhotMean(),tExp); //mean ref idx @ a given temp
	244	if(nmean < 0){ //track didn' t pass through the radiator
	245	pTrk->SetHMPIDsignal(AliHMPIDRecon::kNoRad); //set the appropriate flag
	246	pTrk->SetHMPIDcluIdx(ipCh,index+1000*cluSiz); //set index of cluster
	247	continue;
	248	}
	249	}
	250	//
a591e55f	251	recon.SetImpPC(xPc,yPc); //store track impact to PC
496c71b0	252	recon.CkovAngle(pTrk,(TClonesArray *)pClus->At(cham),index,nmean); //search for Cerenkov angle of this track
	253	}//iTrk
	254
d3da6dc4	255	return 0; // error code: 0=no error;
611e810d	256	}//Recon()
611e810d	257	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
49881df7	258	Int_t AliHMPIDTracker::ReconHiddenTrk(Int_t iCh,Int_t iHVsec,AliESDtrack pTrk,TClonesArray pCluLst,TObjArray pNmean,TObjArray pQthre)
611e810d	259	{
611e810d	260	// Static method to reconstruct Theta Ckov for all valid tracks of a given event.
afe12692	261	// Arguments: pEsd- pointer ESD; pClu- pointer to clusters for all chambers; pNmean - pointer to all function Nmean=f(time), pQthre - pointer to all function Qthre=f(time)
611e810d	262	// Returns: error code, 0 if no errors
57fcdc2b	263	AliHMPIDReconHTA reconHTA; //instance of reconstruction class, nothing important in ctor
611e810d	264	Double_t nmean=((TF1)pNmean->At(3iCh))->Eval(0); //C6F14 Nmean for this chamber
6b7f1827	265	Double_t qthre = 0;
fd47aebf	266	if(pQthre->GetEntriesFast()==AliHMPIDParam::kMaxCh+1) //
	267	qthre=((TF1*)pQthre->At(iCh))->Eval(0); //just for backward compatibi
	268	else qthre=((TF1)pQthre->At(6iCh+iHVsec))->Eval(0); //
611e810d	269	if(pCluLst->GetEntriesFast()<4) return 1; //min 4 clusters (3 + 1 mip) to find a ring!
57fcdc2b	270	if(reconHTA.CkovHiddenTrk(pTrk,pCluLst,nmean,qthre)) return 0; //search for track parameters and Cerenkov angle of this track
5b2b2013	271	else return 1; // error code: 0=no error,1=fit not performed;
611e810d	272	}//Recon()
d3da6dc4	273	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
c1af14f7	274	void AliHMPIDTracker::FillClusterArray(TObjArray* array) const {
	275
	276	// Publishes all pointers to clusters known to the tracker into the
	277	// passed object array.
	278	// The ownership is not transfered - the caller is not expected to delete
	279	// the clusters
	280
	281	for(Int_t iCh=AliHMPIDParam::kMinCh;iCh<=AliHMPIDParam::kMaxCh;iCh++){
	282	TClonesArray pCluArr=(TClonesArray)(*fClu)[iCh];
	283	for (Int_t iClu=0; iClu<pCluArr->GetEntriesFast();iClu++){
	284	AliHMPIDCluster pClu=(AliHMPIDCluster)pCluArr->UncheckedAt(iClu);
	285	array->AddLast(pClu);
	286	}//cluster loop in iCh
	287	pCluArr->Delete();
	288	}//Ch loop
	289
	290	return;
	291	}
	292	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++