[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(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  

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