[u/mrichter/AliRoot.git] / RICH / menu.C

void Show()
{  
//How to get number of events:  
  gpRL->LoadHeader();//without this the following will be zero:
  Info("RICH/menu.C::Show","3 ways to get number of events 1=%i 2=%i 3=%i",
                          gAlice->GetEventsPerRun(),
                          gpRL->TreeE()->GetEntries(),
                          gAlice->TreeE()->GetEntries());
//
  return;
  for(Int_t iEventN=0;iEventN<iNevents;iEventN++){// loop on events
      Int_t iNparticles=gAlice->GetEvent(iEventN);
      Int_t iNtracks=gAlice->TreeH()->GetEntries();
      cout<<"Event "<<iEventN<<" contains "<<iNparticles<<" particles and "<<iNtracks<<" tracks\n";

      for(Int_t iTrackN=0;iTrackN<iNtracks;iTrackN++){ // loop on tracks
         gAlice->TreeH()->GetEntry(iTrackN);
        // gAlice->Particle(iTrackN)->Print();
         for(AliRICHhit *pRICHhit=(AliRICHhit*)pRICH->FirstHit(-1);pRICHhit;pRICHhit->(AliRICHhit*)pRICH->NextHit()){
             TVector3 mrsV3(pRICHhit->X(),pRICHhit->Y(),pRICHhit->Z());
             cout<<"Before\n";mrsV3.Dump();
             TVector3 armV3=pRICH->ToArm(mrsV3);
             cout<<"After\n";armV3.Dump();
         }//loop on hits of given track
      }// loop on tracks
   }//loop on events

}//void Show()

void menu(Int_t iNevents=5)// How many events to generate.
{ 
  Info("RICH/menu.C","%i event(s) are requested",iNevents);
   
  TString runString="gAlice->Run(";  runString=runString+iNevents+");";

  TControlBar *pMenu = new TControlBar("vertical","RICH main");
       
  if(GetAlice()){//it's from file, reconstruct
    pMenu->AddButton("Show","Show()","Shows the structure of events in files");
    pMenu->AddButton("RingViewer","RingViewer()","Show rings with reconstructed info");
  }else{//it's aliroot, simulate
    pMenu->AddButton("Init",         "Init()",                "Normal init");   
    pMenu->AddButton("Run",           runString.Data(),       "Process!");
  }
  pMenu->AddButton("Debug ON",     "gAlice->SetDebug(1)",   "Switch debug on");   
  pMenu->AddButton("Debug OFF",    "gAlice->SetDebug()",    "Switch debug off");     
  pMenu->AddButton("Geo",          "Geo()",                 "Geomentry submenu");
  pMenu->AddButton("Browser",      "new TBrowser;",         "Start ROOT TBrowser");
  pMenu->AddButton("Quit",         ".q",                    "Close session");
  pMenu->Show();
}//void menu(Int_t iNevents)

void Init()
{
  gAlice->Init("~/my/AliceConfig.C");
}

AliRunLoader *rl,*gpRL;
AliLoader *rrl,*gpRichLoader;
AliRICH *r,*gpRich;
AliRun *a;
Bool_t GetAlice()
{
  if(gAlice){//it's aliroot
    Info("RICH/menu.C::GetAlice","gAlice!=NULL, it's aliroot, execute Init()");
    Init();
    return kFALSE;
  }else{//it's root with ALICE libs loaded
    Info("RICH/menu.C::GetAlice","gAlice=0, getting it from simulated file.");
    
    gpRL=AliRunLoader::Open("galice.root","AlicE","update");
    if(!gpRL) Fatal("GetAlice","Can't get AliRunLoader");
    
    gpRL->LoadgAlice();
    if(!gAlice) Fatal("RICH/menu.C::GetAlice","No gAlice in file");
//    gAlice=gpRL->GetAliRun();    
    
    gpRich=(AliRICH*)gAlice->GetDetector("RICH"); 
    if(!gpRich) Fatal("RICH/menu.C::GetAlice","No RICH in file");
    
    gpRichLoader=gpRL->GetLoader("RICHLoader");     
    if(!gpRichLoader) Fatal("RICH/menu.C::GetAlice","No RICH loader in file");    
    
    Info("RICH/menu.C::GetAlice","contains %i event(s)",gAlice->GetEventsPerRun());
    rl=gpRL;rrl=gpRichLoader;r=gpRich;a=gAlice;//for manual manipulation convinience
    return kTRUE;
  }       
}//void GetAlice()         


void Geo()
{
  TControlBar *pMenu = new TControlBar("vertical","RICH draw");
  pMenu->AddButton("Draw Front", "gMC->Gdraw(\"ALIC\", 0,0,0, 10,10, 0.01,0.01)","Draws ALIC volume in XY view");
  pMenu->AddButton("Draw Side",  "gMC->Gdraw(\"ALIC\",90,180, 0, 10,10, 0.01,0.01)","Draws ALIC volume in YZ view");
  pMenu->AddButton("Draw Top",   "gMC->Gdraw(\"ALIC\",90, 90, 0, 10,10, 0.03,0.03)","Draws ALIC volume in XZ view");
  pMenu->AddButton("ALICE Tree", "((TGeant3*)gMC)->Gdtree(\"ALIC\")","Draws ALICE tree");      
  pMenu->AddButton("RICH Tree",  "((TGeant3*)gMC)->Gdtree(\"RICH\")","Draws RICH tree");      
  pMenu->AddButton("Geo test",  "GeoTest()",   "Draw RICH geo as a macro");
  pMenu->AddButton("Print ref", "PrintGeo()",  "Print RICH chambers default position");
  pMenu->AddButton("Print act", "r->Print();", "Print RICH chambers default position");
  pMenu->Show();  
}//void Draw()

void GeoTest()
{

   TBRIK *pAliceBRIK=new TBRIK("aliceBRIK","ALICE mother volume","void",500,500,500);
   AliRICH *pRICH=(AliRICH*)gAlice->GetDetector("RICH");
   TBRIK *pArmBRIK=new TBRIK("armBRIK","RICH arm1","void",pRICH->GetSizeX(),pRICH->GetSizeY(),pRICH->GetSizeZ());
   
   TNode *pAliceNode=new TNode("aliceNode","Mother volume","aliceBRIK");
   pAliceNode->cd();

// ARM 1 LEFT      
   TRotation rot1;
   TVector3  v1(0,pRICH->GetOffset(),0);
   
         
   rot1.RotateX(pRICH->GetYZAngle()*kDegrad);        v1.RotateX(pRICH->GetYZAngle()*kDegrad);
   rot1.RotateZ(pRICH->GetXYAngle()*kDegrad);        v1.RotateZ(pRICH->GetXYAngle()*kDegrad);
   rot1.RotateZ(pRICH->GetRotAngle()*kDegrad);       v1.RotateZ(pRICH->GetRotAngle()*kDegrad);
         
   TRotMatrix *pArm1RotMatrix=new TRotMatrix("rotArm1","rotArm1",rot1.ThetaX()*kRaddeg, rot1.PhiX()*kRaddeg,
                                                                 rot1.ThetaY()*kRaddeg, rot1.PhiY()*kRaddeg,
                                                                 rot1.ThetaZ()*kRaddeg, rot1.PhiZ()*kRaddeg);

   TNode *pArm1Node=new TNode("arm1Node","Left arm","armBRIK",v1.X(),v1.Y(),v1.Z(),"rotArm1");
   arm1Node->SetLineColor(kRed);
   
// ARM 2 LEFT      
   TRotation rot2;
   TVector3  v2(0,pRICH->GetOffset(),0);
   
         
   rot2.RotateX(pRICH->GetYZAngle()*kDegrad);        v2.RotateX(pRICH->GetYZAngle()*kDegrad);
   rot2.RotateZ(-pRICH->GetXYAngle()*kDegrad);        v2.RotateZ(-pRICH->GetXYAngle()*kDegrad);
   rot2.RotateZ(pRICH->GetRotAngle()*kDegrad);        v2.RotateZ(pRICH->GetRotAngle()*kDegrad);
         
   TRotMatrix *pArm2RotMatrix=new TRotMatrix("rotArm2","rotArm2",rot2.ThetaX()*kRaddeg, rot2.PhiX()*kRaddeg,
                                                                 rot2.ThetaY()*kRaddeg, rot2.PhiY()*kRaddeg,
                                                                 rot2.ThetaZ()*kRaddeg, rot2.PhiZ()*kRaddeg);

   TNode *pArm2Node=new TNode("arm2Node","Left arm","armBRIK",v2.X(),v2.Y(),v2.Z(),"rotArm2");
   arm2Node->SetLineColor(kBlue);
   
   aliceNode->Draw();
}//void GeoTest()

void RingViewer()
{
  gStyle->SetPalette(1);
  TCanvas *view = new TCanvas("Display","ALICE RICH Display",0,0,1200,750);
  
  TH2F *p2F = new TH2F("p2F","RICH DISPLAY",160,0,160,144,0,144);
  p2F->SetStats(0);
  p2F->SetMaximum(100);

  Int_t Nevents = gAlice->GetEventsPerRun();
}

void PrintGeo(Float_t rotDeg=0)
{
  AliRICHParam *p=new AliRICHParam;  
  Double_t r=p->Offset();
  Double_t kP=p->AngleXY();
  Double_t kT=p->AngleYZ();
  Double_t kRot;
  
  if(rotDeg==0)
    kRot=p->AngleRot();
  else
    kRot=rotDeg*deg;
        
  cout<<endl;
  Double_t phi=90*deg+kRot,theta=90*deg-kT;    
  Info("   menu for          0","r=%8.3f theta=%5.1f phi=%5.1f x=%8.3f y=%8.3f z=%8.3f",
                                 r,      theta*r2g,  phi*r2g,  
                                                               r*sin(theta)*cos(phi),
                                                                       r*sin(theta)*sin(phi),
                                                                               r*cos(theta));
  
  phi=90*deg+kRot+kP,theta=90*deg;
  Info("   menu for          1","r=%8.3f theta=%5.1f phi=%5.1f x=%8.3f y=%8.3f z=%8.3f",
                                 r,      theta*r2g,  phi*r2g,  
                                                               r*sin(theta)*cos(phi),
                                                                       r*sin(theta)*sin(phi),
                                                                               r*cos(theta));
  
  phi=90*deg+kRot,theta=90*deg;
  Info("   menu for          2","r=%8.3f theta=%5.1f phi=%5.1f x=%8.3f y=%8.3f z=%8.3f",
                                 r,      theta*r2g,  phi*r2g,  
                                                               r*sin(theta)*cos(phi),
                                                                       r*sin(theta)*sin(phi),
                                                                               r*cos(theta));

  
  phi=90*deg+kRot-kP,theta=90*deg;
  Info("   menu for          3","r=%8.3f theta=%5.1f phi=%5.1f x=%8.3f y=%8.3f z=%8.3f",
                                 r,      theta*r2g,  phi*r2g,  
                                                               r*sin(theta)*cos(phi),
                                                                       r*sin(theta)*sin(phi),
                                                                               r*cos(theta));


  phi=90*deg+kRot+kP,theta=90*deg+kT;
  Info("   menu for          4","r=%8.3f theta=%5.1f phi=%5.1f x=%8.3f y=%8.3f z=%8.3f",
                                 r,      theta*r2g,  phi*r2g,  
                                                               r*sin(theta)*cos(phi),
                                                                       r*sin(theta)*sin(phi),
                                                                               r*cos(theta));
  
  
  phi=90*deg+kRot,theta=90*deg+kT;
  Info("   menu for          5","r=%8.3f theta=%5.1f phi=%5.1f x=%8.3f y=%8.3f z=%8.3f",
                                 r,      theta*r2g,  phi*r2g,  
                                                                r*sin(theta)*cos(phi),
                                                                       r*sin(theta)*sin(phi),
                                                                               r*cos(theta));
  
  phi=90*deg+kRot-kP,theta=90*deg+kT;
  Info("   menu for          6","r=%8.3f theta=%5.1f phi=%5.1f x=%8.3f y=%8.3f z=%8.3f",
                                 r,      theta*r2g,  phi*r2g,  
                                                               r*sin(theta)*cos(phi),
                                                                       r*sin(theta)*sin(phi),
                                                                               r*cos(theta));

  delete p;
}//void PrintGeo()
Commit	Line	Data
d48cca74	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=90deg+kRot,theta=90deg-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, thetar2g, phir2g,
180	rsin(theta)cos(phi),
181	rsin(theta)sin(phi),
182	r*cos(theta));
183
184	phi=90deg+kRot+kP,theta=90deg;
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, thetar2g, phir2g,
187	rsin(theta)cos(phi),
188	rsin(theta)sin(phi),
189	r*cos(theta));
190
191	phi=90deg+kRot,theta=90deg;
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, thetar2g, phir2g,
194	rsin(theta)cos(phi),
195	rsin(theta)sin(phi),
196	r*cos(theta));
197
198
199	phi=90deg+kRot-kP,theta=90deg;
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, thetar2g, phir2g,
202	rsin(theta)cos(phi),
203	rsin(theta)sin(phi),
204	r*cos(theta));
205
206
207	phi=90deg+kRot+kP,theta=90deg+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, thetar2g, phir2g,
210	rsin(theta)cos(phi),
211	rsin(theta)sin(phi),
212	r*cos(theta));
213
214
215	phi=90deg+kRot,theta=90deg+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, thetar2g, phir2g,
218	rsin(theta)cos(phi),
219	rsin(theta)sin(phi),
220	r*cos(theta));
221
222	phi=90deg+kRot-kP,theta=90deg+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, thetar2g, phir2g,
225	rsin(theta)cos(phi),
226	rsin(theta)sin(phi),
227	r*cos(theta));
228
229	delete p;
230	}//void PrintGeo()