HTA included. A running mean ref. index is added to take intoaccount variations.
[u/mrichter/AliRoot.git] / HMPID / AliHMPIDParam.cxx
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 "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()
23 #include <TRotation.h>
24 #include <TParticle.h>      //Stack()    
25 #include <TGeoPhysicalNode.h> //ctor
26 #include <TGeoBBox.h>
27 ClassImp(AliHMPIDParam)
28
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
46 AliHMPIDParam* AliHMPIDParam::fgInstance=0x0;        //singleton pointer               
47 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
48 AliHMPIDParam::AliHMPIDParam(Bool_t noGeo=kFALSE):TNamed("HmpidParam","default version") 
49 {
50 // Here all the intitializition is taken place when AliHMPIDParam::Instance() is invoked for the first time.
51 // In particular, 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  
53
54   fRadNmean = MeanIdxRad(); //initialization of the running ref. index of freon
55   
56   Float_t dead=2.6;// cm of the dead zones between PCs-> See 2CRC2099P1
57   
58   if(noGeo==kFALSE && !gGeoManager)  
59   {
60     TGeoManager::Import("geometry.root");
61     if(!gGeoManager) AliFatal("!!!!!!No geometry loaded!!!!!!!");
62   }
63   
64   fgCellX=0.8;fgCellY=0.84;
65   
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   }    
73   fgPcX=80.*fgCellX; fgPcY = 48.*fgCellY;
74   fgAllX=2.*fgPcX+dead;
75   fgAllY=3.*fgPcY+2.*dead;
76
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];
80
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];   
86     
87   fX=0.5*SizeAllX();
88   fY=0.5*SizeAllY();
89   
90   for(Int_t i=kMinCh;i<=kMaxCh;i++) 
91     if(gGeoManager && gGeoManager->IsClosed()) {
92       TGeoPNEntry* pne = gGeoManager->GetAlignableEntry(Form("/HMPID/Chamber%i",i));
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();
99         if(pnode) fM[i]=pnode->GetMatrix();
100         else {
101           fM[i]=new TGeoHMatrix;
102           IdealPosition(i,fM[i]);
103         }
104       }
105     } else{
106       fM[i]=new TGeoHMatrix;
107       IdealPosition(i,fM[i]);
108     } 
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 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
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
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 
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 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
143 Int_t AliHMPIDParam::Stack(Int_t evt,Int_t tid)
144 {
145 // Prints some useful info from stack
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;
154   Int_t iNevt=pAL->GetNumberOfEvents();
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();
162           Printf("totally %i tracks including %i primaries for event %i out of %i event(s)",
163           pStack->GetNtrack(),pStack->GetNprimary(),iEvt,iNevt);
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       } 
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 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++