[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 <TCanvas.h>       //TestXXX() 
#include <TLatex.h>        //TestTrans()  
#include <TView.h>         //TestTrans()
#include <TPolyMarker3D.h> //TestTrans()
#include <TRotation.h>
#include <AliRunLoader.h>  //Stack()
#include <AliStack.h>      //Stack()
#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():TNamed("HmpidParam","default version") 
{
// Here all the intitializition is taken place when AliHMPIDParam::Instance() is invoked for the first time.
// In particulare, matrices to be used for LORS<->MARS trasnformations are initialized from TGeo structure.    
// Note that TGeoManager should be already initialized from geometry.root file  

if(!gGeoManager) 
{
  TGeoManager::Import("geometry.root");
  if(!gGeoManager) AliFatal("!!!!!!No geometry loaded!!!!!!!");
}


    Float_t dead=2.6;// cm of the dead zones between PCs-> See 2CRC2099P1
    TGeoVolume *pCellVol = gGeoManager->GetVolume("Hcel");
    if(!pCellVol) {
      fgCellX=0.8;fgCellY=0.84;
      } else { 
      TGeoBBox *bcell = (TGeoBBox *)pCellVol->GetShape();
      fgCellX=2.*bcell->GetDX(); fgCellY = 2.*bcell->GetDY();
    }
    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()) {
//      fM[i]=(TGeoHMatrix*)gGeoManager->GetVolume("ALIC")->GetNode(Form("HMPID_%i",i))->GetMatrix(); // previous style
      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
	17	#include <TCanvas.h> //TestXXX()
	18	#include <TLatex.h> //TestTrans()
	19	#include <TView.h> //TestTrans()
	20	#include <TPolyMarker3D.h> //TestTrans()
	21	#include <TRotation.h>
	22	#include <AliRunLoader.h> //Stack()
	23	#include <AliStack.h> //Stack()
	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	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
2df6b16e	48	AliHMPIDParam::AliHMPIDParam():TNamed("HmpidParam","default version")
d3da6dc4	49	{
	50	// Here all the intitializition is taken place when AliHMPIDParam::Instance() is invoked for the first time.
	51	// In particulare, matrices to be used for LORS<->MARS trasnformations are initialized from TGeo structure.
	52	// Note that TGeoManager should be already initialized from geometry.root file
ae5a42aa	53
	54	if(!gGeoManager)
	55	{
e4290ede	56	TGeoManager::Import("geometry.root");
3a08ff86	57	if(!gGeoManager) AliFatal("!!!!!!No geometry loaded!!!!!!!");
ae5a42aa	58	}
	59
	60
3a08ff86	61	Float_t dead=2.6;// cm of the dead zones between PCs-> See 2CRC2099P1
b6eacec7	62	TGeoVolume *pCellVol = gGeoManager->GetVolume("Hcel");
	63	if(!pCellVol) {
	64	fgCellX=0.8;fgCellY=0.84;
	65	} else {
	66	TGeoBBox bcell = (TGeoBBox )pCellVol->GetShape();
	67	fgCellX=2.bcell->GetDX(); fgCellY = 2.bcell->GetDY();
	68	}
ae5a42aa	69	fgPcX=80.fgCellX; fgPcY = 48.fgCellY;
3a08ff86	70	fgAllX=2.*fgPcX+dead;
3a08ff86	71	fgAllY=3.fgPcY+2.dead;
ae5a42aa	72
3a08ff86	73	fgkMinPcX[1]=fgPcX+dead; fgkMinPcX[3]=fgkMinPcX[1]; fgkMinPcX[5]=fgkMinPcX[3];
ae5a42aa	74	fgkMaxPcX[0]=fgPcX; fgkMaxPcX[2]=fgkMaxPcX[0]; fgkMaxPcX[4]=fgkMaxPcX[2];
	75	fgkMaxPcX[1]=fgAllX; fgkMaxPcX[3]=fgkMaxPcX[1]; fgkMaxPcX[5]=fgkMaxPcX[3];
	76
3a08ff86	77	fgkMinPcY[2]=fgPcY+dead; fgkMinPcY[3]=fgkMinPcY[2];
3a08ff86	78	fgkMinPcY[4]=2.fgPcY+2.dead; fgkMinPcY[5]=fgkMinPcY[4];
ae5a42aa	79	fgkMaxPcY[0]=fgPcY; fgkMaxPcY[1]=fgkMaxPcY[0];
3a08ff86	80	fgkMaxPcY[2]=2.*fgPcY+dead; fgkMaxPcY[3]=fgkMaxPcY[2];
ae5a42aa	81	fgkMaxPcY[4]=fgAllY; fgkMaxPcY[5]=fgkMaxPcY[4];
	82
	83	fX=0.5*SizeAllX();
	84	fY=0.5*SizeAllY();
	85	for(Int_t i=kMinCh;i<=kMaxCh;i++)
97eadc2b	86	if(gGeoManager && gGeoManager->IsClosed()) {
97eadc2b	87	// fM[i]=(TGeoHMatrix*)gGeoManager->GetVolume("ALIC")->GetNode(Form("HMPID_%i",i))->GetMatrix(); // previous style
2df6b16e	88	TGeoPNEntry* pne = gGeoManager->GetAlignableEntry(Form("/HMPID/Chamber%i",i));
97eadc2b	89	if (!pne) {
	90	AliErrorClass(Form("The symbolic volume %s does not correspond to any physical entry!",Form("HMPID_%i",i)));
	91	fM[i]=new TGeoHMatrix;
	92	IdealPosition(i,fM[i]);
	93	} else {
	94	TGeoPhysicalNode *pnode = pne->GetPhysicalNode();
f80e1da4	95	if(pnode) fM[i]=pnode->GetMatrix();
	96	else {
	97	fM[i]=new TGeoHMatrix;
	98	IdealPosition(i,fM[i]);
	99	}
97eadc2b	100	}
97eadc2b	101	} else{
1d4857c5	102	fM[i]=new TGeoHMatrix;
	103	IdealPosition(i,fM[i]);
	104	}
d3da6dc4	105	fgInstance=this;
	106	}//ctor
	107	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
	108	void AliHMPIDParam::Print(Option_t* opt) const
	109	{
	110	// print some usefull (hopefully) info on some internal guts of HMPID parametrisation
	111
	112	for(Int_t i=0;i<7;i++) fM[i]->Print(opt);
	113	}//Print()
	114	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
1d4857c5	115	void AliHMPIDParam::IdealPosition(Int_t iCh, TGeoHMatrix *pMatrix)
	116	{
	117	// Construct ideal position matrix for a given chamber
	118	// Arguments: iCh- chamber ID; pMatrix- pointer to precreated unity matrix where to store the results
	119	// Returns: none
423554a3	120	const Double_t kAngHor=19.5; // horizontal angle between chambers 19.5 grad
	121	const Double_t kAngVer=20; // vertical angle between chambers 20 grad
	122	const Double_t kAngCom=30; // common HMPID rotation with respect to x axis 30 grad
	123	const Double_t kTrans[3]={490,0,0}; // center of the chamber is on window-gap surface
	124	pMatrix->RotateY(90); // rotate around y since initial position is in XY plane -> now in YZ plane
	125	pMatrix->SetTranslation(kTrans); // now plane in YZ is shifted along x
1d4857c5	126	switch(iCh){
	127	case 0: pMatrix->RotateY(kAngHor); pMatrix->RotateZ(-kAngVer); break; //right and down
	128	case 1: pMatrix->RotateZ(-kAngVer); break; //down
	129	case 2: pMatrix->RotateY(kAngHor); break; //right
	130	case 3: break; //no rotation
	131	case 4: pMatrix->RotateY(-kAngHor); break; //left
	132	case 5: pMatrix->RotateZ(kAngVer); break; //up
	133	case 6: pMatrix->RotateY(-kAngHor); pMatrix->RotateZ(kAngVer); break; //left and up
	134	}
	135	pMatrix->RotateZ(kAngCom); //apply common rotation in XY plane
	136
	137	}
	138	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
d3da6dc4	139	Int_t AliHMPIDParam::Stack(Int_t evt,Int_t tid)
d3da6dc4	140	{
d1bf51e1	141	// Prints some useful info from stack
d3da6dc4	142	// Arguments: evt - event number. if not -1 print info only for that event
	143	// tid - track id. if not -1 then print it and all it's mothers if any
	144	// Returns: mother tid of the given tid if any
	145	AliRunLoader *pAL=AliRunLoader::Open();
	146	if(pAL->LoadHeader()) return -1;
	147	if(pAL->LoadKinematics()) return -1;
	148
	149	Int_t mtid=-1;
cf7e313e	150	Int_t iNevt=pAL->GetNumberOfEvents();
d3da6dc4	151
	152	for(Int_t iEvt=0;iEvt<iNevt;iEvt++){//events loop
	153	if(evt!=-1 && evt!=iEvt) continue; //in case one needs to print the requested event, ignore all others
	154	pAL->GetEvent(iEvt);
	155	AliStack *pStack=pAL->Stack();
	156	if(tid==-1){ //print all tids for this event
	157	for(Int_t i=0;i<pStack->GetNtrack();i++) pStack->Particle(i)->Print();
	158	Printf("totally %i tracks including %i primaries for event %i out of %i event(s)",pStack->GetNtrack(),pStack->GetNprimary(),iEvt,iNevt);
	159	}else{ //print only this tid and it;s mothers
	160	if(tid<0 \|\| tid>pStack->GetNtrack()) {Printf("Wrong tid, valid tid range for event %i is 0-%i",iEvt,pStack->GetNtrack());break;}
	161	TParticle *pTrack=pStack->Particle(tid); mtid=pTrack->GetFirstMother();
	162	TString str=pTrack->GetName();
	163	while((tid=pTrack->GetFirstMother()) >= 0){
	164	pTrack=pStack->Particle(tid);
	165	str+=" from ";str+=pTrack->GetName();
	166	}
d3da6dc4	167	}//if(tid==-1)
	168	}//events loop
	169	pAL->UnloadHeader(); pAL->UnloadKinematics();
	170	return mtid;
	171	}
	172	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
	173	Int_t AliHMPIDParam::StackCount(Int_t pid,Int_t evt)
	174	{
	175	// Counts total number of particles of given sort (including secondary) for a given event
	176	AliRunLoader *pAL=AliRunLoader::Open();
	177	pAL->GetEvent(evt);
	178	if(pAL->LoadHeader()) return 0;
	179	if(pAL->LoadKinematics()) return 0;
	180	AliStack *pStack=pAL->Stack();
	181
	182	Int_t iCnt=0;
	183	for(Int_t i=0;i<pStack->GetNtrack();i++) if(pStack->Particle(i)->GetPdgCode()==pid) iCnt++;
	184
	185	pAL->UnloadHeader(); pAL->UnloadKinematics();
	186	return iCnt;
	187	}
	188	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++