[u/mrichter/AliRoot.git] / ITS / analITSgeom.C

void analITSgeom (const char *filename="galice.root",Int_t evNumber=0){
/////////////////////////////////////////////////////////////////////////
//   This macro is a small example of a ROOT macro
//   illustrating how to read the output of GALICE
//   and fill some histograms.
//   
//     Root > .L analITSgeom.C   //this loads the macro in memory
//     Root > analITSgeom();     //by default process first event   
//     Root > analITgeomS(2);    //process third event
//Begin_Html
/*
<img src="figures/analITSgeom_ref.gif">
</pre>
<br clear=left>
<font size=+2 color=red>
<p>A reference plot produced by analITSgeom.C.
</font>
<pre>
 */
//End_Html
/////////////////////////////////////////////////////////////////////////


// Dynamically link some shared libs
   if (gClassTable->GetID("AliRun") < 0) {
      gROOT->LoadMacro("loadlibs.C");
      loadlibs();
   } // end if gClassTable...
      
// Connect the Root Galice file containing Geometry, Kine and Hits
   TFile *file = (TFile*)gROOT->GetListOfFiles()->FindObject(filename);
   if (!file) file = new TFile(filename);

// Get AliRun object from file or create it if not on file
   if (!gAlice) {
      gAlice = (AliRun*)file->Get("gAlice");
      if (gAlice) printf("AliRun object found on file\n");
      if (!gAlice) gAlice = new AliRun("gAlice","Alice test program");
   }
      
// Import the Kine and Hits Trees for the event evNumber in the file
   Int_t nparticles = gAlice->GetEvent(evNumber);
   if (nparticles <= 0) return;
   Float_t x,y,z,mass,e,r,phi,flad;
   Int_t nbytes = 0;
   Int_t j,hit,ipart,lay;
   Int_t nhits;
   Int_t sector,plane;
   TParticle *particle;
   AliITShit  *itsHit;

// Get pointers to Alice detectors and Hits containers
   AliDetector *ITS  = gAlice->GetDetector("ITS");
   if(!ITS) return;
   TClonesArray *Particles = gAlice->Particles();
   TClonesArray *ITShits   = ITS->Hits();
   AliITSgeom   *gm        = ((AliITS *)ITS)->GetITSgeom();

   TTree *TH        = gAlice->TreeH();
   Int_t ntracks    = TH->GetEntries();

   // Create histograms
   Float_t pi2 = 2.0*TMath::Pi();
   Int_t   Nlad = gm->GetNladders(1);
   TH2F *hITS1 = new TH2F("hITS1","Ladder# vs. angle for Layer 1",
			  Nlad,0.5,(Float_t)Nlad+0.5,100,0.0,pi2);
   Int_t   Nlad = gm->GetNladders(2);
   TH2F *hITS2 = new TH2F("hITS2","Ladder# vs. angle for Layer 2",
			  Nlad,0.5,(Float_t)Nlad+0.5,100,0.0,pi2);
   Int_t   Nlad = gm->GetNladders(3);
   TH2F *hITS3 = new TH2F("hITS3","Ladder# vs. angle for Layer 3",
			  Nlad,0.5,(Float_t)Nlad+0.5,100,0.0,pi2);
   Int_t   Nlad = gm->GetNladders(4);
   TH2F *hITS4 = new TH2F("hITS4","Ladder# vs. angle for Layer 4",
			  Nlad,0.5,(Float_t)Nlad+0.5,100,0.0,pi2);
   Int_t   Nlad = gm->GetNladders(5);
   TH2F *hITS5 = new TH2F("hITS5","Ladder# vs. angle for Layer 5",
			  Nlad,0.5,(Float_t)Nlad+0.5,100,0.0,pi2);
   Int_t   Nlad = gm->GetNladders(6);
   TH2F *hITS6 = new TH2F("hITS6","Ladder# vs. angle for Layer 6",
			  Nlad,0.5,(Float_t)Nlad+0.5,100,0.0,pi2);

// Start loop on tracks in the hits containers
   for (Int_t track=0; track<ntracks;track++) {
     gAlice->ResetHits();
     nbytes += TH->GetEvent(track);
     Int_t i=0;
       nhits = ITShits->GetEntriesFast();
       for (hit=0;hit<nhits;hit++) {
	 itsHit   = (AliITShit*)ITShits->UncheckedAt(hit);
	 // With this new version, to be able to do proper detector
	 // simulations, the requirment that a "hit" leave some
	 // energy deposited has been removed.
	 if(itsHit->GetIonization()<=0.0) continue;
	 phi = TMath::ATan2(itsHit->GetYG(),itsHit->GetXG());
	 if(phi<0.0) phi += pi2;
	 lay = itsHit->GetLayer();
	 flad = ((Float_t)itsHit->GetLadder()) + 0.5;
         if(lay==1) hITS1->Fill(flad,phi,1.0);
         if(lay==2) hITS2->Fill(flad,phi,1.0);
         if(lay==3) hITS3->Fill(flad,phi,1.0);
         if(lay==4) hITS4->Fill(flad,phi,1.0);
         if(lay==5) hITS5->Fill(flad,phi,1.0);
         if(lay==6) hITS6->Fill(flad,phi,1.0);
       } // end for hit
   }  // end for track

//Create a canvas, set the view range, show histograms
   TCanvas *c1 = new TCanvas("c1"," ITS geometry test",400,10,600,700);
   c1->Divide(2,3);
   c1->cd(1);
   hITS1->Draw();
   c1->cd(2);
   hITS2->Draw();
   c1->cd(3);
   hITS3->Draw();
   c1->cd(4);
   hITS4->Draw();
   c1->cd(5);
   hITS5->Draw();
   c1->cd(6);
   hITS6->Draw();
   c1->Print("analITSgeom.gif");
}
Commit	Line	Data
58005f18	1	void analITSgeom (const char *filename="galice.root",Int_t evNumber=0){
	2	/////////////////////////////////////////////////////////////////////////
	3	// This macro is a small example of a ROOT macro
	4	// illustrating how to read the output of GALICE
	5	// and fill some histograms.
	6	//
	7	// Root > .L analITSgeom.C //this loads the macro in memory
	8	// Root > analITSgeom(); //by default process first event
	9	// Root > analITgeomS(2); //process third event
	10	//Begin_Html
	11	/*
	12	<img src="figures/analITSgeom_ref.gif">
	13	</pre>
	14	<br clear=left>
	15	<font size=+2 color=red>
	16	<p>A reference plot produced by analITSgeom.C.
	17	</font>
	18	<pre>
	19	*/
	20	//End_Html
	21	/////////////////////////////////////////////////////////////////////////
	22
	23
	24	// Dynamically link some shared libs
	25	if (gClassTable->GetID("AliRun") < 0) {
	26	gROOT->LoadMacro("loadlibs.C");
	27	loadlibs();
	28	} // end if gClassTable...
	29
	30	// Connect the Root Galice file containing Geometry, Kine and Hits
	31	TFile file = (TFile)gROOT->GetListOfFiles()->FindObject(filename);
	32	if (!file) file = new TFile(filename);
	33
	34	// Get AliRun object from file or create it if not on file
	35	if (!gAlice) {
	36	gAlice = (AliRun*)file->Get("gAlice");
	37	if (gAlice) printf("AliRun object found on file\n");
	38	if (!gAlice) gAlice = new AliRun("gAlice","Alice test program");
	39	}
	40
	41	// Import the Kine and Hits Trees for the event evNumber in the file
	42	Int_t nparticles = gAlice->GetEvent(evNumber);
	43	if (nparticles <= 0) return;
	44	Float_t x,y,z,mass,e,r,phi,flad;
	45	Int_t nbytes = 0;
	46	Int_t j,hit,ipart,lay;
	47	Int_t nhits;
	48	Int_t sector,plane;
	49	TParticle *particle;
	50	AliITShit *itsHit;
	51
	52	// Get pointers to Alice detectors and Hits containers
	53	AliDetector *ITS = gAlice->GetDetector("ITS");
	54	if(!ITS) return;
	55	TClonesArray *Particles = gAlice->Particles();
	56	TClonesArray *ITShits = ITS->Hits();
	57	AliITSgeom gm = ((AliITS )ITS)->GetITSgeom();
	58
	59	TTree *TH = gAlice->TreeH();
	60	Int_t ntracks = TH->GetEntries();
	61
	62	// Create histograms
	63	Float_t pi2 = 2.0*TMath::Pi();
	64	Int_t Nlad = gm->GetNladders(1);
65	TH2F *hITS1 = new TH2F("hITS1","Ladder# vs. angle for Layer 1",
66	Nlad,0.5,(Float_t)Nlad+0.5,100,0.0,pi2);
67	Int_t Nlad = gm->GetNladders(2);
68	TH2F *hITS2 = new TH2F("hITS2","Ladder# vs. angle for Layer 2",
69	Nlad,0.5,(Float_t)Nlad+0.5,100,0.0,pi2);
70	Int_t Nlad = gm->GetNladders(3);
71	TH2F *hITS3 = new TH2F("hITS3","Ladder# vs. angle for Layer 3",
72	Nlad,0.5,(Float_t)Nlad+0.5,100,0.0,pi2);
73	Int_t Nlad = gm->GetNladders(4);
74	TH2F *hITS4 = new TH2F("hITS4","Ladder# vs. angle for Layer 4",
75	Nlad,0.5,(Float_t)Nlad+0.5,100,0.0,pi2);
76	Int_t Nlad = gm->GetNladders(5);
77	TH2F *hITS5 = new TH2F("hITS5","Ladder# vs. angle for Layer 5",
78	Nlad,0.5,(Float_t)Nlad+0.5,100,0.0,pi2);
79	Int_t Nlad = gm->GetNladders(6);
80	TH2F *hITS6 = new TH2F("hITS6","Ladder# vs. angle for Layer 6",
81	Nlad,0.5,(Float_t)Nlad+0.5,100,0.0,pi2);
82
83	// Start loop on tracks in the hits containers
84	for (Int_t track=0; track<ntracks;track++) {
85	gAlice->ResetHits();
86	nbytes += TH->GetEvent(track);
87	Int_t i=0;
88	nhits = ITShits->GetEntriesFast();
89	for (hit=0;hit<nhits;hit++) {
90	itsHit = (AliITShit*)ITShits->UncheckedAt(hit);
91	// With this new version, to be able to do proper detector
92	// simulations, the requirment that a "hit" leave some
93	// energy deposited has been removed.
94	if(itsHit->GetIonization()<=0.0) continue;
95	phi = TMath::ATan2(itsHit->GetYG(),itsHit->GetXG());
96	if(phi<0.0) phi += pi2;
97	lay = itsHit->GetLayer();
98	flad = ((Float_t)itsHit->GetLadder()) + 0.5;
99	if(lay==1) hITS1->Fill(flad,phi,1.0);
100	if(lay==2) hITS2->Fill(flad,phi,1.0);
101	if(lay==3) hITS3->Fill(flad,phi,1.0);
102	if(lay==4) hITS4->Fill(flad,phi,1.0);
103	if(lay==5) hITS5->Fill(flad,phi,1.0);
104	if(lay==6) hITS6->Fill(flad,phi,1.0);
105	} // end for hit
106	} // end for track
107
108	//Create a canvas, set the view range, show histograms
109	TCanvas *c1 = new TCanvas("c1"," ITS geometry test",400,10,600,700);
110	c1->Divide(2,3);
111	c1->cd(1);
112	hITS1->Draw();
113	c1->cd(2);
114	hITS2->Draw();
115	c1->cd(3);
116	hITS3->Draw();
117	c1->cd(4);
118	hITS4->Draw();
119	c1->cd(5);
120	hITS5->Draw();
121	c1->cd(6);
122	hITS6->Draw();
123	c1->Print("analITSgeom.gif");
124	}