RICH/menu.C

   1 void Show()
   2 {
   3 //How to get number of events:
   4   gpRL->LoadHeader();//without this the following will be zero:
   5   Info("RICH/menu.C::Show","3 ways to get number of events 1=%i 2=%i 3=%i",
   6                           gAlice->GetEventsPerRun(),
   7                           gpRL->TreeE()->GetEntries(),
   8                           gAlice->TreeE()->GetEntries());
   9 //
  10   return;
  11   for(Int_t iEventN=0;iEventN<iNevents;iEventN++){// loop on events
  12       Int_t iNparticles=gAlice->GetEvent(iEventN);
  13       Int_t iNtracks=gAlice->TreeH()->GetEntries();
  14       cout<<"Event "<<iEventN<<" contains "<<iNparticles<<" particles and "<<iNtracks<<" tracks\n";
  15
  16       for(Int_t iTrackN=0;iTrackN<iNtracks;iTrackN++){ // loop on tracks
  17          gAlice->TreeH()->GetEntry(iTrackN);
  18         // gAlice->Particle(iTrackN)->Print();
  19          for(AliRICHhit *pRICHhit=(AliRICHhit*)pRICH->FirstHit(-1);pRICHhit;pRICHhit->(AliRICHhit*)pRICH->NextHit()){
  20              TVector3 mrsV3(pRICHhit->X(),pRICHhit->Y(),pRICHhit->Z());
  21              cout<<"Before\n";mrsV3.Dump();
  22              TVector3 armV3=pRICH->ToArm(mrsV3);
  23              cout<<"After\n";armV3.Dump();
  24          }//loop on hits of given track
  25       }// loop on tracks
  26    }//loop on events
  27
  28 }//void Show()
  29
  30 void menu(Int_t iNevents=5)// How many events to generate.
  31 {
  32   Info("RICH/menu.C","%i event(s) are requested",iNevents);
  33
  34   TString runString="gAlice->Run(";  runString=runString+iNevents+");";
  35
  36   TControlBar *pMenu = new TControlBar("vertical","RICH main");
  37
  38   if(GetAlice()){//it's from file, reconstruct
  39     pMenu->AddButton("Show","Show()","Shows the structure of events in files");
  40     pMenu->AddButton("RingViewer","RingViewer()","Show rings with reconstructed info");
  41   }else{//it's aliroot, simulate
  42     pMenu->AddButton("Init",         "Init()",                "Normal init");
  43     pMenu->AddButton("Run",           runString.Data(),       "Process!");
  44   }
  45   pMenu->AddButton("Debug ON",     "gAlice->SetDebug(1)",   "Switch debug on");
  46   pMenu->AddButton("Debug OFF",    "gAlice->SetDebug()",    "Switch debug off");
  47   pMenu->AddButton("Geo",          "Geo()",                 "Geomentry submenu");
  48   pMenu->AddButton("Browser",      "new TBrowser;",         "Start ROOT TBrowser");
  49   pMenu->AddButton("Quit",         ".q",                    "Close session");
  50   pMenu->Show();
  51 }//void menu(Int_t iNevents)
  52
  53 void Init()
  54 {
  55   gAlice->Init("~/my/AliceConfig.C");
  56 }
  57
  58 AliRunLoader *rl,*gpRL;
  59 AliLoader *rrl,*gpRichLoader;
  60 AliRICH *r,*gpRich;
  61 AliRun *a;
  62 Bool_t GetAlice()
  63 {
  64   if(gAlice){//it's aliroot
  65     Info("RICH/menu.C::GetAlice","gAlice!=NULL, it's aliroot, execute Init()");
  66     Init();
  67     return kFALSE;
  68   }else{//it's root with ALICE libs loaded
  69     Info("RICH/menu.C::GetAlice","gAlice=0, getting it from simulated file.");
  70
  71     gpRL=AliRunLoader::Open("galice.root","AlicE","update");
  72     if(!gpRL) Fatal("GetAlice","Can't get AliRunLoader");
  73
  74     gpRL->LoadgAlice();
  75     if(!gAlice) Fatal("RICH/menu.C::GetAlice","No gAlice in file");
  76 //    gAlice=gpRL->GetAliRun();
  77
  78     gpRich=(AliRICH*)gAlice->GetDetector("RICH");
  79     if(!gpRich) Fatal("RICH/menu.C::GetAlice","No RICH in file");
  80
  81     gpRichLoader=gpRL->GetLoader("RICHLoader");
  82     if(!gpRichLoader) Fatal("RICH/menu.C::GetAlice","No RICH loader in file");
  83
  84     Info("RICH/menu.C::GetAlice","contains %i event(s)",gAlice->GetEventsPerRun());
  85     rl=gpRL;rrl=gpRichLoader;r=gpRich;a=gAlice;//for manual manipulation convinience
  86     return kTRUE;
  87   }
  88 }//void GetAlice()
  89
  90
  91
  92 void Geo()
  93 {
  94   TControlBar *pMenu = new TControlBar("vertical","RICH draw");
  95   pMenu->AddButton("Draw Front", "gMC->Gdraw(\"ALIC\", 0,0,0, 10,10, 0.01,0.01)","Draws ALIC volume in XY view");
  96   pMenu->AddButton("Draw Side",  "gMC->Gdraw(\"ALIC\",90,180, 0, 10,10, 0.01,0.01)","Draws ALIC volume in YZ view");
  97   pMenu->AddButton("Draw Top",   "gMC->Gdraw(\"ALIC\",90, 90, 0, 10,10, 0.03,0.03)","Draws ALIC volume in XZ view");
  98   pMenu->AddButton("ALICE Tree", "((TGeant3*)gMC)->Gdtree(\"ALIC\")","Draws ALICE tree");
  99   pMenu->AddButton("RICH Tree",  "((TGeant3*)gMC)->Gdtree(\"RICH\")","Draws RICH tree");
 100   pMenu->AddButton("Geo test",  "GeoTest()",   "Draw RICH geo as a macro");
 101   pMenu->AddButton("Print ref", "PrintGeo()",  "Print RICH chambers default position");
 102   pMenu->AddButton("Print act", "r->Print();", "Print RICH chambers default position");
 103   pMenu->Show();
 104 }//void Draw()
 105
 106 void GeoTest()
 107 {
 108
 109    TBRIK *pAliceBRIK=new TBRIK("aliceBRIK","ALICE mother volume","void",500,500,500);
 110    AliRICH *pRICH=(AliRICH*)gAlice->GetDetector("RICH");
 111    TBRIK *pArmBRIK=new TBRIK("armBRIK","RICH arm1","void",pRICH->GetSizeX(),pRICH->GetSizeY(),pRICH->GetSizeZ());
 112
 113    TNode *pAliceNode=new TNode("aliceNode","Mother volume","aliceBRIK");
 114    pAliceNode->cd();
 115
 116 // ARM 1 LEFT
 117    TRotation rot1;
 118    TVector3  v1(0,pRICH->GetOffset(),0);
 119
 120
 121    rot1.RotateX(pRICH->GetYZAngle()*kDegrad);        v1.RotateX(pRICH->GetYZAngle()*kDegrad);
 122    rot1.RotateZ(pRICH->GetXYAngle()*kDegrad);        v1.RotateZ(pRICH->GetXYAngle()*kDegrad);
 123    rot1.RotateZ(pRICH->GetRotAngle()*kDegrad);       v1.RotateZ(pRICH->GetRotAngle()*kDegrad);
 124
 125    TRotMatrix *pArm1RotMatrix=new TRotMatrix("rotArm1","rotArm1",rot1.ThetaX()*kRaddeg, rot1.PhiX()*kRaddeg,
 126                                                                  rot1.ThetaY()*kRaddeg, rot1.PhiY()*kRaddeg,
 127                                                                  rot1.ThetaZ()*kRaddeg, rot1.PhiZ()*kRaddeg);
 128
 129    TNode *pArm1Node=new TNode("arm1Node","Left arm","armBRIK",v1.X(),v1.Y(),v1.Z(),"rotArm1");
 130    arm1Node->SetLineColor(kRed);
 131
 132 // ARM 2 LEFT
 133    TRotation rot2;
 134    TVector3  v2(0,pRICH->GetOffset(),0);
 135
 136
 137    rot2.RotateX(pRICH->GetYZAngle()*kDegrad);        v2.RotateX(pRICH->GetYZAngle()*kDegrad);
 138    rot2.RotateZ(-pRICH->GetXYAngle()*kDegrad);        v2.RotateZ(-pRICH->GetXYAngle()*kDegrad);
 139    rot2.RotateZ(pRICH->GetRotAngle()*kDegrad);        v2.RotateZ(pRICH->GetRotAngle()*kDegrad);
 140
 141    TRotMatrix *pArm2RotMatrix=new TRotMatrix("rotArm2","rotArm2",rot2.ThetaX()*kRaddeg, rot2.PhiX()*kRaddeg,
 142                                                                  rot2.ThetaY()*kRaddeg, rot2.PhiY()*kRaddeg,
 143                                                                  rot2.ThetaZ()*kRaddeg, rot2.PhiZ()*kRaddeg);
 144
 145    TNode *pArm2Node=new TNode("arm2Node","Left arm","armBRIK",v2.X(),v2.Y(),v2.Z(),"rotArm2");
 146    arm2Node->SetLineColor(kBlue);
 147
 148    aliceNode->Draw();
 149 }//void GeoTest()
 150
 151 void RingViewer()
 152 {
 153   gStyle->SetPalette(1);
 154   TCanvas *view = new TCanvas("Display","ALICE RICH Display",0,0,1200,750);
 155
 156   TH2F *p2F = new TH2F("p2F","RICH DISPLAY",160,0,160,144,0,144);
 157   p2F->SetStats(0);
 158   p2F->SetMaximum(100);
 159
 160   Int_t Nevents = gAlice->GetEventsPerRun();
 161 }
 162
 163 void PrintGeo(Float_t rotDeg=0)
 164 {
 165   AliRICHParam *p=new AliRICHParam;
 166   Double_t r=p->Offset();
 167   Double_t kP=p->AngleXY();
 168   Double_t kT=p->AngleYZ();
 169   Double_t kRot;
 170
 171   if(rotDeg==0)
 172     kRot=p->AngleRot();
 173   else
 174     kRot=rotDeg*deg;
 175
 176   cout<<endl;
 177   Double_t phi=90*deg+kRot,theta=90*deg-kT;
 178   Info("   menu for          0","r=%8.3f theta=%5.1f phi=%5.1f x=%8.3f y=%8.3f z=%8.3f",
 179                                  r,      theta*r2g,  phi*r2g,
 180                                                                r*sin(theta)*cos(phi),
 181                                                                        r*sin(theta)*sin(phi),
 182                                                                                r*cos(theta));
 183
 184   phi=90*deg+kRot+kP,theta=90*deg;
 185   Info("   menu for          1","r=%8.3f theta=%5.1f phi=%5.1f x=%8.3f y=%8.3f z=%8.3f",
 186                                  r,      theta*r2g,  phi*r2g,
 187                                                                r*sin(theta)*cos(phi),
 188                                                                        r*sin(theta)*sin(phi),
 189                                                                                r*cos(theta));
 190
 191   phi=90*deg+kRot,theta=90*deg;
 192   Info("   menu for          2","r=%8.3f theta=%5.1f phi=%5.1f x=%8.3f y=%8.3f z=%8.3f",
 193                                  r,      theta*r2g,  phi*r2g,
 194                                                                r*sin(theta)*cos(phi),
 195                                                                        r*sin(theta)*sin(phi),
 196                                                                                r*cos(theta));
 197
 198
 199   phi=90*deg+kRot-kP,theta=90*deg;
 200   Info("   menu for          3","r=%8.3f theta=%5.1f phi=%5.1f x=%8.3f y=%8.3f z=%8.3f",
 201                                  r,      theta*r2g,  phi*r2g,
 202                                                                r*sin(theta)*cos(phi),
 203                                                                        r*sin(theta)*sin(phi),
 204                                                                                r*cos(theta));
 205
 206
 207   phi=90*deg+kRot+kP,theta=90*deg+kT;
 208   Info("   menu for          4","r=%8.3f theta=%5.1f phi=%5.1f x=%8.3f y=%8.3f z=%8.3f",
 209                                  r,      theta*r2g,  phi*r2g,
 210                                                                r*sin(theta)*cos(phi),
 211                                                                        r*sin(theta)*sin(phi),
 212                                                                                r*cos(theta));
 213
 214
 215   phi=90*deg+kRot,theta=90*deg+kT;
 216   Info("   menu for          5","r=%8.3f theta=%5.1f phi=%5.1f x=%8.3f y=%8.3f z=%8.3f",
 217                                  r,      theta*r2g,  phi*r2g,
 218                                                                 r*sin(theta)*cos(phi),
 219                                                                        r*sin(theta)*sin(phi),
 220                                                                                r*cos(theta));
 221
 222   phi=90*deg+kRot-kP,theta=90*deg+kT;
 223   Info("   menu for          6","r=%8.3f theta=%5.1f phi=%5.1f x=%8.3f y=%8.3f z=%8.3f",
 224                                  r,      theta*r2g,  phi*r2g,
 225                                                                r*sin(theta)*cos(phi),
 226                                                                        r*sin(theta)*sin(phi),
 227                                                                                r*cos(theta));
 228
 229   delete p;
 230 }//void PrintGeo()