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

//  **************************************************************************
//  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
//  *                                                                        *
//  * Author: The ALICE Off-line Project.                                    *
//  * Contributors are mentioned in the code where appropriate.              *
//  *                                                                        *
//  * Permission to use, copy, modify and distribute this software and its   *
//  * documentation strictly for non-commercial purposes is hereby granted   *
//  * without fee, provided that the above copyright notice appears in all   *
//  * copies and that both the copyright notice and this permission notice   *
//  * appear in the supporting documentation. The authors make no claims     *
//  * about the suitability of this software for any purpose. It is          *
//  * provided "as is" without express or implied warranty.                  *
//  **************************************************************************
#include "AliHMPIDParam.h"  //class header
#include "AliHMPIDDigit.h"  //ctor
#include "AliLog.h"         //general
#include <AliRunLoader.h>   //Stack()
#include <AliStack.h>       //Stack()
#include <TLatex.h>         //TestTrans()  
#include <TView.h>          //TestTrans()
#include <TPolyMarker3D.h>  //TestTrans()
#include <TRotation.h>
#include <TParticle.h>      //Stack()    
#include <TGeoPhysicalNode.h> //ctor
#include <TGeoBBox.h>
ClassImp(AliHMPIDParam)


Float_t AliHMPIDParam::fgkMinPcX[]={0.,0.,0.,0.,0.,0.};
Float_t AliHMPIDParam::fgkMaxPcX[]={0.,0.,0.,0.,0.,0.};
Float_t AliHMPIDParam::fgkMinPcY[]={0.,0.,0.,0.,0.,0.};
Float_t AliHMPIDParam::fgkMaxPcY[]={0.,0.,0.,0.,0.,0.};

Float_t AliHMPIDParam::fgCellX=0.;
Float_t AliHMPIDParam::fgCellY=0.;

Float_t AliHMPIDParam::fgPcX=0;
Float_t AliHMPIDParam::fgPcY=0;

Float_t AliHMPIDParam::fgAllX=0;
Float_t AliHMPIDParam::fgAllY=0;

Int_t AliHMPIDParam::fgSigmas=4;

AliHMPIDParam* AliHMPIDParam::fgInstance=0x0;        //singleton pointer               
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
AliHMPIDParam::AliHMPIDParam(Bool_t noGeo=kFALSE):TNamed("HmpidParam","default version") 
{
// Here all the intitializition is taken place when AliHMPIDParam::Instance() is invoked for the first time.
// In particular, matrices to be used for LORS<->MARS trasnformations are initialized from TGeo structure.    
// Note that TGeoManager should be already initialized from geometry.root file  

  fRadNmean = MeanIdxRad(); //initialization of the running ref. index of freon
  
  Float_t dead=2.6;// cm of the dead zones between PCs-> See 2CRC2099P1
  
  if(noGeo==kFALSE && !gGeoManager)  
  {
    TGeoManager::Import("geometry.root");
    if(!gGeoManager) AliFatal("!!!!!!No geometry loaded!!!!!!!");
  }
  
  fgCellX=0.8;fgCellY=0.84;
  
  if(!noGeo==kTRUE){
    TGeoVolume *pCellVol = gGeoManager->GetVolume("Hcel");
    if(pCellVol) {
      TGeoBBox *bcell = (TGeoBBox *)pCellVol->GetShape();
      fgCellX=2.*bcell->GetDX(); fgCellY = 2.*bcell->GetDY();  // overwrite the values with the read ones
    }
  }    
  fgPcX=80.*fgCellX; fgPcY = 48.*fgCellY;
  fgAllX=2.*fgPcX+dead;
  fgAllY=3.*fgPcY+2.*dead;

  fgkMinPcX[1]=fgPcX+dead; fgkMinPcX[3]=fgkMinPcX[1];  fgkMinPcX[5]=fgkMinPcX[3];
  fgkMaxPcX[0]=fgPcX; fgkMaxPcX[2]=fgkMaxPcX[0];  fgkMaxPcX[4]=fgkMaxPcX[2];
  fgkMaxPcX[1]=fgAllX; fgkMaxPcX[3]=fgkMaxPcX[1];  fgkMaxPcX[5]=fgkMaxPcX[3];

  fgkMinPcY[2]=fgPcY+dead; fgkMinPcY[3]=fgkMinPcY[2];  
  fgkMinPcY[4]=2.*fgPcY+2.*dead; fgkMinPcY[5]=fgkMinPcY[4];
  fgkMaxPcY[0]=fgPcY; fgkMaxPcY[1]=fgkMaxPcY[0];  
  fgkMaxPcY[2]=2.*fgPcY+dead; fgkMaxPcY[3]=fgkMaxPcY[2]; 
  fgkMaxPcY[4]=fgAllY; fgkMaxPcY[5]=fgkMaxPcY[4];   
    
  fX=0.5*SizeAllX();
  fY=0.5*SizeAllY();
  
  for(Int_t i=kMinCh;i<=kMaxCh;i++) 
    if(gGeoManager && gGeoManager->IsClosed()) {
      TGeoPNEntry* pne = gGeoManager->GetAlignableEntry(Form("/HMPID/Chamber%i",i));
      if (!pne) {
        AliErrorClass(Form("The symbolic volume %s does not correspond to any physical entry!",Form("HMPID_%i",i)));
        fM[i]=new TGeoHMatrix;
        IdealPosition(i,fM[i]);
      } else {
        TGeoPhysicalNode *pnode = pne->GetPhysicalNode();
        if(pnode) fM[i]=pnode->GetMatrix();
        else {
          fM[i]=new TGeoHMatrix;
          IdealPosition(i,fM[i]);
        }
      }
    } else{
      fM[i]=new TGeoHMatrix;
      IdealPosition(i,fM[i]);
    } 
  fgInstance=this; 
}//ctor
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
void AliHMPIDParam::Print(Option_t* opt) const
{
// print some usefull (hopefully) info on some internal guts of HMPID parametrisation 
  
  for(Int_t i=0;i<7;i++) fM[i]->Print(opt);
}//Print()
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
void AliHMPIDParam::IdealPosition(Int_t iCh, TGeoHMatrix *pMatrix)
{
// Construct ideal position matrix for a given chamber
// Arguments: iCh- chamber ID; pMatrix- pointer to precreated unity matrix where to store the results
//   Returns: none
  const Double_t kAngHor=19.5;        //  horizontal angle between chambers  19.5 grad
  const Double_t kAngVer=20;          //  vertical angle between chambers    20   grad     
  const Double_t kAngCom=30;          //  common HMPID rotation with respect to x axis  30   grad     
  const Double_t kTrans[3]={490,0,0}; //  center of the chamber is on window-gap surface
  pMatrix->RotateY(90);               //  rotate around y since initial position is in XY plane -> now in YZ plane
  pMatrix->SetTranslation(kTrans);    //  now plane in YZ is shifted along x 
  switch(iCh){
    case 0:                pMatrix->RotateY(kAngHor);  pMatrix->RotateZ(-kAngVer);  break; //right and down 
    case 1:                                            pMatrix->RotateZ(-kAngVer);  break; //down              
    case 2:                pMatrix->RotateY(kAngHor);                               break; //right 
    case 3:                                                                         break; //no rotation
    case 4:                pMatrix->RotateY(-kAngHor);                              break; //left   
    case 5:                                            pMatrix->RotateZ(kAngVer);   break; //up
    case 6:                pMatrix->RotateY(-kAngHor); pMatrix->RotateZ(kAngVer);   break; //left and up 
  }
  pMatrix->RotateZ(kAngCom);     //apply common rotation  in XY plane    
   
}
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Int_t AliHMPIDParam::Stack(Int_t evt,Int_t tid)
{
// Prints some useful info from stack
// Arguments: evt - event number. if not -1 print info only for that event
//            tid - track id. if not -1 then print it and all it's mothers if any   
//   Returns: mother tid of the given tid if any
  AliRunLoader *pAL=AliRunLoader::Open(); 
  if(pAL->LoadHeader()) return -1;
  if(pAL->LoadKinematics()) return -1;
  
  Int_t mtid=-1;
  Int_t iNevt=pAL->GetNumberOfEvents();
  
  for(Int_t iEvt=0;iEvt<iNevt;iEvt++){//events loop
    if(evt!=-1 && evt!=iEvt) continue; //in case one needs to print the requested event, ignore all others
    pAL->GetEvent(iEvt);    
    AliStack *pStack=pAL->Stack();  
    if(tid==-1){                        //print all tids for this event
      for(Int_t i=0;i<pStack->GetNtrack();i++) pStack->Particle(i)->Print();
          Printf("totally %i tracks including %i primaries for event %i out of %i event(s)",
          pStack->GetNtrack(),pStack->GetNprimary(),iEvt,iNevt);
    }else{                              //print only this tid and it;s mothers
      if(tid<0 || tid>pStack->GetNtrack()) {Printf("Wrong tid, valid tid range for event %i is 0-%i",iEvt,pStack->GetNtrack());break;}
      TParticle *pTrack=pStack->Particle(tid); mtid=pTrack->GetFirstMother();
      TString str=pTrack->GetName();
      while((tid=pTrack->GetFirstMother()) >= 0){
        pTrack=pStack->Particle(tid);
        str+=" from ";str+=pTrack->GetName();
      } 
    }//if(tid==-1)      
  }//events loop
  pAL->UnloadHeader();  pAL->UnloadKinematics();
  return mtid;
}
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Int_t AliHMPIDParam::StackCount(Int_t pid,Int_t evt)
{
// Counts total number of particles of given sort (including secondary) for a given event
  AliRunLoader *pAL=AliRunLoader::Open(); 
  pAL->GetEvent(evt);    
  if(pAL->LoadHeader()) return 0;
  if(pAL->LoadKinematics()) return 0;
  AliStack *pStack=pAL->Stack();
  
  Int_t iCnt=0;
  for(Int_t i=0;i<pStack->GetNtrack();i++) if(pStack->Particle(i)->GetPdgCode()==pid) iCnt++;
  
  pAL->UnloadHeader();  pAL->UnloadKinematics();
  return iCnt;
}
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Commit	Line	Data
d3da6dc4	1	// **************************************************************************
	2	// * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	// * *
	4	// * Author: The ALICE Off-line Project. *
	5	// * Contributors are mentioned in the code where appropriate. *
	6	// * *
	7	// * Permission to use, copy, modify and distribute this software and its *
	8	// * documentation strictly for non-commercial purposes is hereby granted *
	9	// * without fee, provided that the above copyright notice appears in all *
	10	// * copies and that both the copyright notice and this permission notice *
	11	// * appear in the supporting documentation. The authors make no claims *
	12	// * about the suitability of this software for any purpose. It is *
	13	// * provided "as is" without express or implied warranty. *
	14	// **************************************************************************
	15	#include "AliHMPIDParam.h" //class header
	16	#include "AliHMPIDDigit.h" //ctor
a8ff381e	17	#include "AliLog.h" //general
	18	#include <AliRunLoader.h> //Stack()
	19	#include <AliStack.h> //Stack()
	20	#include <TLatex.h> //TestTrans()
	21	#include <TView.h> //TestTrans()
	22	#include <TPolyMarker3D.h> //TestTrans()
d3da6dc4	23	#include <TRotation.h>
a8ff381e	24	#include <TParticle.h> //Stack()
97eadc2b	25	#include <TGeoPhysicalNode.h> //ctor
ae5a42aa	26	#include <TGeoBBox.h>
d3da6dc4	27	ClassImp(AliHMPIDParam)
d3da6dc4	28
ae5a42aa	29
	30	Float_t AliHMPIDParam::fgkMinPcX[]={0.,0.,0.,0.,0.,0.};
	31	Float_t AliHMPIDParam::fgkMaxPcX[]={0.,0.,0.,0.,0.,0.};
	32	Float_t AliHMPIDParam::fgkMinPcY[]={0.,0.,0.,0.,0.,0.};
	33	Float_t AliHMPIDParam::fgkMaxPcY[]={0.,0.,0.,0.,0.,0.};
	34
	35	Float_t AliHMPIDParam::fgCellX=0.;
	36	Float_t AliHMPIDParam::fgCellY=0.;
	37
	38	Float_t AliHMPIDParam::fgPcX=0;
	39	Float_t AliHMPIDParam::fgPcY=0;
	40
	41	Float_t AliHMPIDParam::fgAllX=0;
	42	Float_t AliHMPIDParam::fgAllY=0;
	43
	44	Int_t AliHMPIDParam::fgSigmas=4;
	45
d3da6dc4	46	AliHMPIDParam* AliHMPIDParam::fgInstance=0x0; //singleton pointer
d3da6dc4	47	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
58fc9564	48	AliHMPIDParam::AliHMPIDParam(Bool_t noGeo=kFALSE):TNamed("HmpidParam","default version")
d3da6dc4	49	{
d3da6dc4	50	// Here all the intitializition is taken place when AliHMPIDParam::Instance() is invoked for the first time.
58fc9564	51	// In particular, matrices to be used for LORS<->MARS trasnformations are initialized from TGeo structure.
d3da6dc4	52	// Note that TGeoManager should be already initialized from geometry.root file
ae5a42aa	53
a8ff381e	54	fRadNmean = MeanIdxRad(); //initialization of the running ref. index of freon
a8ff381e	55
8e2a911a	56	Float_t dead=2.6;// cm of the dead zones between PCs-> See 2CRC2099P1
58fc9564	57
	58	if(noGeo==kFALSE && !gGeoManager)
	59	{
	60	TGeoManager::Import("geometry.root");
	61	if(!gGeoManager) AliFatal("!!!!!!No geometry loaded!!!!!!!");
	62	}
	63
8e2a911a	64	fgCellX=0.8;fgCellY=0.84;
58fc9564	65
8e2a911a	66	if(!noGeo==kTRUE){
	67	TGeoVolume *pCellVol = gGeoManager->GetVolume("Hcel");
	68	if(pCellVol) {
	69	TGeoBBox bcell = (TGeoBBox )pCellVol->GetShape();
	70	fgCellX=2.bcell->GetDX(); fgCellY = 2.bcell->GetDY(); // overwrite the values with the read ones
	71	}
	72	}
58fc9564	73	fgPcX=80.fgCellX; fgPcY = 48.fgCellY;
	74	fgAllX=2.*fgPcX+dead;
	75	fgAllY=3.fgPcY+2.dead;
ae5a42aa	76
58fc9564	77	fgkMinPcX[1]=fgPcX+dead; fgkMinPcX[3]=fgkMinPcX[1]; fgkMinPcX[5]=fgkMinPcX[3];
	78	fgkMaxPcX[0]=fgPcX; fgkMaxPcX[2]=fgkMaxPcX[0]; fgkMaxPcX[4]=fgkMaxPcX[2];
	79	fgkMaxPcX[1]=fgAllX; fgkMaxPcX[3]=fgkMaxPcX[1]; fgkMaxPcX[5]=fgkMaxPcX[3];
ae5a42aa	80
58fc9564	81	fgkMinPcY[2]=fgPcY+dead; fgkMinPcY[3]=fgkMinPcY[2];
	82	fgkMinPcY[4]=2.fgPcY+2.dead; fgkMinPcY[5]=fgkMinPcY[4];
	83	fgkMaxPcY[0]=fgPcY; fgkMaxPcY[1]=fgkMaxPcY[0];
	84	fgkMaxPcY[2]=2.*fgPcY+dead; fgkMaxPcY[3]=fgkMaxPcY[2];
	85	fgkMaxPcY[4]=fgAllY; fgkMaxPcY[5]=fgkMaxPcY[4];
ae5a42aa	86
	87	fX=0.5*SizeAllX();
	88	fY=0.5*SizeAllY();
58fc9564	89
ae5a42aa	90	for(Int_t i=kMinCh;i<=kMaxCh;i++)
97eadc2b	91	if(gGeoManager && gGeoManager->IsClosed()) {
2df6b16e	92	TGeoPNEntry* pne = gGeoManager->GetAlignableEntry(Form("/HMPID/Chamber%i",i));
97eadc2b	93	if (!pne) {
	94	AliErrorClass(Form("The symbolic volume %s does not correspond to any physical entry!",Form("HMPID_%i",i)));
	95	fM[i]=new TGeoHMatrix;
	96	IdealPosition(i,fM[i]);
	97	} else {
	98	TGeoPhysicalNode *pnode = pne->GetPhysicalNode();
f80e1da4	99	if(pnode) fM[i]=pnode->GetMatrix();
	100	else {
	101	fM[i]=new TGeoHMatrix;
	102	IdealPosition(i,fM[i]);
	103	}
97eadc2b	104	}
97eadc2b	105	} else{
1d4857c5	106	fM[i]=new TGeoHMatrix;
	107	IdealPosition(i,fM[i]);
	108	}
d3da6dc4	109	fgInstance=this;
	110	}//ctor
	111	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
	112	void AliHMPIDParam::Print(Option_t* opt) const
	113	{
	114	// print some usefull (hopefully) info on some internal guts of HMPID parametrisation
	115
	116	for(Int_t i=0;i<7;i++) fM[i]->Print(opt);
	117	}//Print()
	118	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
1d4857c5	119	void AliHMPIDParam::IdealPosition(Int_t iCh, TGeoHMatrix *pMatrix)
	120	{
	121	// Construct ideal position matrix for a given chamber
	122	// Arguments: iCh- chamber ID; pMatrix- pointer to precreated unity matrix where to store the results
	123	// Returns: none
423554a3	124	const Double_t kAngHor=19.5; // horizontal angle between chambers 19.5 grad
	125	const Double_t kAngVer=20; // vertical angle between chambers 20 grad
	126	const Double_t kAngCom=30; // common HMPID rotation with respect to x axis 30 grad
	127	const Double_t kTrans[3]={490,0,0}; // center of the chamber is on window-gap surface
	128	pMatrix->RotateY(90); // rotate around y since initial position is in XY plane -> now in YZ plane
	129	pMatrix->SetTranslation(kTrans); // now plane in YZ is shifted along x
1d4857c5	130	switch(iCh){
	131	case 0: pMatrix->RotateY(kAngHor); pMatrix->RotateZ(-kAngVer); break; //right and down
	132	case 1: pMatrix->RotateZ(-kAngVer); break; //down
	133	case 2: pMatrix->RotateY(kAngHor); break; //right
	134	case 3: break; //no rotation
	135	case 4: pMatrix->RotateY(-kAngHor); break; //left
	136	case 5: pMatrix->RotateZ(kAngVer); break; //up
	137	case 6: pMatrix->RotateY(-kAngHor); pMatrix->RotateZ(kAngVer); break; //left and up
	138	}
	139	pMatrix->RotateZ(kAngCom); //apply common rotation in XY plane
	140
	141	}
	142	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
d3da6dc4	143	Int_t AliHMPIDParam::Stack(Int_t evt,Int_t tid)
d3da6dc4	144	{
d1bf51e1	145	// Prints some useful info from stack
d3da6dc4	146	// Arguments: evt - event number. if not -1 print info only for that event
	147	// tid - track id. if not -1 then print it and all it's mothers if any
	148	// Returns: mother tid of the given tid if any
	149	AliRunLoader *pAL=AliRunLoader::Open();
	150	if(pAL->LoadHeader()) return -1;
	151	if(pAL->LoadKinematics()) return -1;
	152
	153	Int_t mtid=-1;
cf7e313e	154	Int_t iNevt=pAL->GetNumberOfEvents();
d3da6dc4	155
	156	for(Int_t iEvt=0;iEvt<iNevt;iEvt++){//events loop
	157	if(evt!=-1 && evt!=iEvt) continue; //in case one needs to print the requested event, ignore all others
	158	pAL->GetEvent(iEvt);
	159	AliStack *pStack=pAL->Stack();
	160	if(tid==-1){ //print all tids for this event
	161	for(Int_t i=0;i<pStack->GetNtrack();i++) pStack->Particle(i)->Print();
a8ff381e	162	Printf("totally %i tracks including %i primaries for event %i out of %i event(s)",
a8ff381e	163	pStack->GetNtrack(),pStack->GetNprimary(),iEvt,iNevt);
d3da6dc4	164	}else{ //print only this tid and it;s mothers
	165	if(tid<0 \|\| tid>pStack->GetNtrack()) {Printf("Wrong tid, valid tid range for event %i is 0-%i",iEvt,pStack->GetNtrack());break;}
	166	TParticle *pTrack=pStack->Particle(tid); mtid=pTrack->GetFirstMother();
	167	TString str=pTrack->GetName();
	168	while((tid=pTrack->GetFirstMother()) >= 0){
	169	pTrack=pStack->Particle(tid);
	170	str+=" from ";str+=pTrack->GetName();
	171	}
d3da6dc4	172	}//if(tid==-1)
	173	}//events loop
	174	pAL->UnloadHeader(); pAL->UnloadKinematics();
	175	return mtid;
	176	}
	177	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
	178	Int_t AliHMPIDParam::StackCount(Int_t pid,Int_t evt)
	179	{
	180	// Counts total number of particles of given sort (including secondary) for a given event
	181	AliRunLoader *pAL=AliRunLoader::Open();
	182	pAL->GetEvent(evt);
	183	if(pAL->LoadHeader()) return 0;
	184	if(pAL->LoadKinematics()) return 0;
	185	AliStack *pStack=pAL->Stack();
	186
	187	Int_t iCnt=0;
	188	for(Int_t i=0;i<pStack->GetNtrack();i++) if(pStack->Particle(i)->GetPdgCode()==pid) iCnt++;
	189
	190	pAL->UnloadHeader(); pAL->UnloadKinematics();
	191	return iCnt;
	192	}
	193	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++