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

#include "AliHMPIDTracker.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)
{
// 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();
  Float_t xRa=0,yRa=0,theta=0,phi=0;                                                            //track intersection at PC and angles at RAD, LORS  
  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())==kFALSE) continue;                    //not in active area  
    pTrk->SetHMPIDtrk      (xRa,yRa,theta,phi);                                                 //store track intersection info
    pTrk->SetHMPIDcluIdx   (i,0);
    return i;
  }                                                                                             //chambers loop
  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 21 TF1
  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;
  for(Int_t iTrk=0;iTrk<pEsd->GetNumberOfTracks();iTrk++){                                       //ESD tracks loop
    AliESDtrack *pTrk = pEsd->GetTrack(iTrk);                                                    //get reconstructed track    
    Int_t cham=IntTrkCha(pTrk,xPc,yPc);                                                          //get chamber intersected by this track 
    if(cham<0) continue;                                                                         //no intersection at all, go after next track
    Double_t nmean=((TF1*)pNmean->At(3*cham))->Eval(pEsd->GetTimeStamp());                       //C6F14 Nmean for this chamber
    Int_t hvsec = AliHMPIDParam::InHVSector(xPc,yPc);
    Double_t qthre=((TF1*)pQthre->At(6*cham+hvsec))->Eval(pEsd->GetTimeStamp());
    recon.SetImpPC(xPc,yPc);                                                                     //store track impact to PC
    recon.CkovAngle(pTrk,(TClonesArray *)pClus->At(cham),nmean,qthre);                           //search for Cerenkov angle of this track
//    Printf("AliHMPIDTracker::Recon: nmean %f, qthre %f",nmean,qthre);
  }                                                                                              //ESD tracks loop
  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=((TF1*)pQthre->At(iCh+iHVsec))->Eval(0);                                             //C6F14 Nmean for this chamber
  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()
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Commit	Line	Data
3c6274c1	1	#include "AliHMPIDTracker.h" //class header
	2	#include "AliHMPIDCluster.h" //GetTrackPoint(),PropagateBack()
	3	#include "AliHMPIDParam.h" //GetTrackPoint(),PropagateBack()
59d9d4b3	4	#include "AliHMPIDRecon.h" //Recon()
57fcdc2b	5	#include "AliHMPIDReconHTA.h" //ReconHTA()
	6	#include <AliESDEvent.h> //PropagateBack(),Recon()
	7	#include <AliESDtrack.h> //Intersect()
d3da6dc4	8	#include <AliRun.h> //GetTrackPoint(),PropagateBack()
	9	#include <AliTrackPointArray.h> //GetTrackPoint()
	10	#include <AliAlignObj.h> //GetTrackPoint()
59d9d4b3	11	#include <AliCDBManager.h> //PropageteBack()
59d9d4b3	12	#include <AliCDBEntry.h> //PropageteBack()
423554a3	13	//.
	14	// HMPID base class fo tracking
	15	//.
	16	//.
	17	//.
d3da6dc4	18	ClassImp(AliHMPIDTracker)
94b1fbfa	19	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
c61a7285	20	AliHMPIDTracker::AliHMPIDTracker():
	21	AliTracker(),
	22	fClu(new TObjArray(AliHMPIDParam::kMaxCh+1))
94b1fbfa	23	{
59d9d4b3	24	// ctor. Create TObjArray of TClonesArray of AliHMPIDCluster
	25	//
	26	//
c61a7285	27	fClu->SetOwner(kTRUE);
ae5a42aa	28	for(int i=AliHMPIDParam::kMinCh;i<=AliHMPIDParam::kMaxCh;i++) fClu->AddAt(new TClonesArray("AliHMPIDCluster"),i);
59d9d4b3	29	}//ctor
d3da6dc4	30	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
	31	Bool_t AliHMPIDTracker::GetTrackPoint(Int_t idx, AliTrackPoint& point) const
	32	{
	33	// Interface callback methode invoked from AliReconstruction::WriteAlignmentData() to get position of MIP cluster in MARS associated to a current track.
	34	// MIP cluster is reffered by index which is stored in AliESDtrack ???????
	35	// Arguments: idx- cluster index which is stored by HMPID in AliESDtrack
	36	// point- reference to the object where to store the point
	37	// Returns: status of operation if FALSE then AliReconstruction::WriteAlignmentData() do not store this point to array of points for current track.
	38	if(idx<0) return kFALSE; //no MIP cluster assigned to this track in PropagateBack()
59d9d4b3	39	Int_t iCham=idx/1000000; Int_t iClu=idx%1000000;
ae079791	40	point.SetVolumeID(AliGeomManager::LayerToVolUID(AliGeomManager::kHMPID,iCham-1));//layer and chamber number
94b1fbfa	41	TClonesArray pArr=(TClonesArray)(*fClu)[iCham];
94b1fbfa	42	AliHMPIDCluster pClu=(AliHMPIDCluster)pArr->UncheckedAt(iClu);//get pointer to cluster
d3da6dc4	43	Double_t mars[3];
	44	AliHMPIDParam::Instance()->Lors2Mars(iCham,pClu->X(),pClu->Y(),mars);
	45	point.SetXYZ(mars[0],mars[1],mars[2]);
	46	return kTRUE;
59d9d4b3	47	}//GetTrackPoint()
	48	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
	49	Int_t AliHMPIDTracker::IntTrkCha(AliESDtrack *pTrk,Float_t &xPc,Float_t &yPc)
	50	{
	51	// Static method to find intersection in between given track and HMPID chambers
	52	// Arguments: pTrk- ESD track; xPc,yPc- track intersection with PC in LORS [cm]
	53	// Returns: intersected chamber ID or -1
	54	AliHMPIDParam *pParam=AliHMPIDParam::Instance();
	55	Float_t xRa=0,yRa=0,theta=0,phi=0; //track intersection at PC and angles at RAD, LORS
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
ae5a42aa	64	if(AliHMPIDParam::IsInside(xPc,yPc,pParam->DistCut())==kFALSE) continue; //not in active area
59d9d4b3	65	pTrk->SetHMPIDtrk (xRa,yRa,theta,phi); //store track intersection info
	66	pTrk->SetHMPIDcluIdx (i,0);
	67	return i;
	68	} //chambers loop
	69	return -1; //no intersection with HMPID chambers
	70	}//IntTrkCha()
d3da6dc4	71	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
	72	Int_t AliHMPIDTracker::LoadClusters(TTree *pCluTree)
	73	{
59d9d4b3	74	// Interface callback methode invoked from AliReconstruction::RunTracking() to load HMPID clusters before PropagateBack() gets control. Done once per event.
d3da6dc4	75	// Arguments: pCluTree- pointer to clusters tree got by AliHMPIDLoader::LoadRecPoints("read") then AliHMPIDLoader::TreeR()
d3da6dc4	76	// Returns: error code (currently ignored in AliReconstruction::RunTraking())
ae5a42aa	77	for(int i=AliHMPIDParam::kMinCh;i<=AliHMPIDParam::kMaxCh;i++) pCluTree->SetBranchAddress(Form("HMPID%d",i),&((*fClu)[i]));
94b1fbfa	78	pCluTree->GetEntry(0);
94b1fbfa	79	return 0;
59d9d4b3	80	}//LoadClusters()
d3da6dc4	81	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
af885e0f	82	Int_t AliHMPIDTracker::PropagateBack(AliESDEvent *pEsd)
abb5f786	83	{
59d9d4b3	84	// Interface pure virtual in AliTracker. Invoked from AliReconstruction::RunTracking() after invocation of AliTracker::LoadClusters() once per event
abb5f786	85	// Agruments: pEsd - pointer to ESD
abb5f786	86	// Returns: error code
f6998a13	87	AliCDBEntry *pNmeanEnt =AliCDBManager::Instance()->Get("HMPID/Calib/Nmean"); //contains TObjArray of 21 TF1
49881df7	88	AliCDBEntry pQthreEnt =AliCDBManager::Instance()->Get("HMPID/Calib/Qthre"); //contains TObjArray of 42 (7ch 6sec) TF1
abb5f786	89	if(!pNmeanEnt) AliFatal("No Nmean C6F14 ");
abb5f786	90	if(!pQthreEnt) AliFatal("No Qthre");
94b1fbfa	91
afe12692	92	return Recon(pEsd,fClu,(TObjArray)pNmeanEnt->GetObject(),(TObjArray)pQthreEnt->GetObject());
611e810d	93	}//PropagateBack()
abb5f786	94	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
afe12692	95	Int_t AliHMPIDTracker::Recon(AliESDEvent pEsd,TObjArray pClus,TObjArray pNmean, TObjArray pQthre)
d3da6dc4	96	{
59d9d4b3	97	// Static method to reconstruct Theta Ckov for all valid tracks of a given event.
59d9d4b3	98	// Arguments: pEsd- pointer ESD; pClu- pointer to clusters for all chambers; pNmean - pointer to all function Nmean=f(time)
d3da6dc4	99	// Returns: error code, 0 if no errors
abb5f786	100	AliHMPIDRecon recon; //instance of reconstruction class, nothing important in ctor
59d9d4b3	101	Float_t xPc,yPc;
	102	for(Int_t iTrk=0;iTrk<pEsd->GetNumberOfTracks();iTrk++){ //ESD tracks loop
	103	AliESDtrack *pTrk = pEsd->GetTrack(iTrk); //get reconstructed track
a591e55f	104	Int_t cham=IntTrkCha(pTrk,xPc,yPc); //get chamber intersected by this track
59280a5a	105	if(cham<0) continue; //no intersection at all, go after next track
a591e55f	106	Double_t nmean=((TF1)pNmean->At(3cham))->Eval(pEsd->GetTimeStamp()); //C6F14 Nmean for this chamber
49881df7	107	Int_t hvsec = AliHMPIDParam::InHVSector(xPc,yPc);
49881df7	108	Double_t qthre=((TF1)pQthre->At(6cham+hvsec))->Eval(pEsd->GetTimeStamp());
a591e55f	109	recon.SetImpPC(xPc,yPc); //store track impact to PC
afe12692	110	recon.CkovAngle(pTrk,(TClonesArray *)pClus->At(cham),nmean,qthre); //search for Cerenkov angle of this track
8d6043fc	111	// Printf("AliHMPIDTracker::Recon: nmean %f, qthre %f",nmean,qthre);
59280a5a	112	} //ESD tracks loop
d3da6dc4	113	return 0; // error code: 0=no error;
611e810d	114	}//Recon()
611e810d	115	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
49881df7	116	Int_t AliHMPIDTracker::ReconHiddenTrk(Int_t iCh,Int_t iHVsec,AliESDtrack pTrk,TClonesArray pCluLst,TObjArray pNmean,TObjArray pQthre)
611e810d	117	{
611e810d	118	// Static method to reconstruct Theta Ckov for all valid tracks of a given event.
afe12692	119	// 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	120	// Returns: error code, 0 if no errors
57fcdc2b	121	AliHMPIDReconHTA reconHTA; //instance of reconstruction class, nothing important in ctor
611e810d	122	Double_t nmean=((TF1)pNmean->At(3iCh))->Eval(0); //C6F14 Nmean for this chamber
49881df7	123	Double_t qthre=((TF1*)pQthre->At(iCh+iHVsec))->Eval(0); //C6F14 Nmean for this chamber
611e810d	124	if(pCluLst->GetEntriesFast()<4) return 1; //min 4 clusters (3 + 1 mip) to find a ring!
57fcdc2b	125	if(reconHTA.CkovHiddenTrk(pTrk,pCluLst,nmean,qthre)) return 0; //search for track parameters and Cerenkov angle of this track
5b2b2013	126	else return 1; // error code: 0=no error,1=fit not performed;
611e810d	127	}//Recon()
d3da6dc4	128	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++