[u/mrichter/AliRoot.git] / HLT / MUON / macros / DisplaydHLTData.C

/**************************************************************************
 * This file is property of and copyright by the ALICE HLT Project        *
 * All rights reserved.                                                   *
 *                                                                        *
 * Primary Authors:                                                       *
 *   Artur Szostak <artursz@iafrica.com>                                  *
 *                                                                        *
 * 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.                  *
 **************************************************************************/

// $Id: DisplaydHLTData.C 37070 2009-11-20 13:53:08Z aszostak $

/**
 * \ingroup macros
 * \file DisplaydHLTData.C
 * \brief Macro for displaying rootified dHLT data generated with RunChain.C
 *
 * This macro is used to display the dHLT data generated by the RunChain.C macro
 * with the "root" option for making the output data. Optionally the dHLTRawData.root
 * file which is generated during a AliRoot offline reconstruction can also be
 * displayed using this macro.
 * After having generated the dHLT raw data file (called <i>output.root</i> for example)
 * the following command can be executed in an alieve session to display both
 * MUON and dHLT data:
 * \code
 *   > gSystem->Load("libAliHLTMUON.so");
 *   > .L $ALICE_ROOT/EVE/macros/alieve_init.C
 *   > .L $ALICE_ROOT/EVE/alice-macros/event_goto.C
 *   > .L $ALICE_ROOT/EVE/alice-macros/MUON_displayData.C
 *   > .L $ALICE_ROOT/HLT/MUON/macros/DisplaydHLTData.C
 *   > alieve_init("OCDBpath","DataPath",EventNumber);
 *   > MUON_displayData(1,1,1);
 *   > DisplaydHLTData("output.root");
 * \endcode
 * Where <i>OCDBpath</i> should be replaced by the path to the OCDB database,
 * <i>DataPath</i> should be the directory of the raw data or raw data root file and
 * <i>EventNumber</i> is the event number to load.
 *
 * \author Artur Szostak <artursz@iafrica.com>
 */

#if !defined(__CINT__) || defined(__MAKECINT__)
#include "AliEveEventManager.h"
#include "AliHLTMUONTriggerRecord.h"
#include "AliHLTMUONRecHit.h"
#include "AliHLTMUONMansoTrack.h"
#include "AliHLTMUONEvent.h"
#include "TEveManager.h"
#include "TEveElement.h"
#include "TEvePointSet.h"
#include "TEveStraightLineSet.h"
#include "TArrayD.h"
#include "TFile.h"
#include "TString.h"
#include "TClassTable.h"
#include "TSystem.h"
#include <iostream>
using std::cerr;
using std::endl;
#endif


TFile* gDHLTFile = NULL;

void FillData(Int_t eventId);


void DisplaydHLTData(const char* dHLTfilename = "output.root")
{
	// Display dHLT data.
	
	if (gClassTable->GetID("AliHLTMUONEvent") < 0)
	{
		gSystem->Load("libAliHLTMUON.so");
	}
	
	if (AliEveEventManager::GetMaster() == NULL)
	{
		cerr << "ERROR: alieve event not initialised, use alieve_init(...)" << endl;
		return;
	}
	
	gEve->DisableRedraw();
	
	TString filename = dHLTfilename;
	TString tmpfilename = filename;
	if (gSystem->FindFile("", tmpfilename) == NULL)
	{
		tmpfilename = filename;
		if (gSystem->FindFile(".", tmpfilename) == NULL)
		{
			TString filename = TString(AliEveEventManager::GetMaster()->GetTitle());
			filename += "/";
			filename += dHLTfilename;
		}
	}
	if (gDHLTFile == NULL)
	{
		gDHLTFile = new TFile(filename.Data(), "READ");
	}
	else if (filename != gDHLTFile->GetName())
	{
		delete gDHLTFile;
		gDHLTFile = new TFile(filename.Data(), "READ");
	}
	
	FillData(AliEveEventManager::GetMaster()->GetEventId());
	
	gEve->Redraw3D(kTRUE);
	gEve->EnableRedraw();
}


void FillData(TEveElementList* list, const AliHLTMUONTriggerRecord* trigrec)
{
	// Count the number of points in the trigger record.
	Int_t pointCount = 0;
	for (int i = 11; i <= 14; ++i)
	{
		if (trigrec->Hit(i) != TVector3(0,0,0)) ++pointCount;
	}
	
	// Create and fill the point set for the trigger record.
	TEvePointSet* pointset = new TEvePointSet(pointCount);
	pointset->SetName(Form("TriggerRecord%d", trigrec->Id()));
	pointset->SetTitle(Form("dHLT trigger record ID = %d", trigrec->Id()));
	pointset->SetMarkerStyle(28);
	pointset->SetMarkerColor(kYellow);
	pointset->SetMarkerSize(3);
	Int_t ipnt = 0;
	for (int i = 11; i <= 14; ++i)
	{
		if (trigrec->Hit(i) == TVector3(0,0,0)) continue;
		pointset->SetPoint(ipnt, trigrec->Hit(i).X(), trigrec->Hit(i).Y(), trigrec->Hit(i).Z());
		++ipnt;
	}
	
	list->AddElement(pointset);
}


void FillData(TEveElementList* list, const AliHLTMUONMansoTrack* track)
{
	// Count the points in the track.
	Int_t pointCount = 0;
	for (int i = 7; i <= 10; ++i)
	{
		const AliHLTMUONRecHit* hit = track->Hit(i);
		if (hit != NULL) ++pointCount;
	}
	const AliHLTMUONTriggerRecord* trigrec = track->TriggerRecord();
	if (trigrec != NULL)
	{
		for (int i = 11; i <= 14; ++i)
		{
			if (trigrec->Hit(i) != TVector3(0,0,0)) ++pointCount;
		}
	}
	
	// Create and fill the point set for the track.
	TEvePointSet* pointset = new TEvePointSet(pointCount);
	pointset->SetName(Form("Track%d", track->Id()));
	pointset->SetTitle(Form("dHLT track ID = %d", track->Id()));
	if (track->Id() != -1)
	{
		pointset->SetMarkerStyle(20);
		pointset->SetMarkerColor(kGreen);
		pointset->SetMarkerSize(1.5);
	}
	else
	{
		pointset->SetMarkerStyle(28);
		pointset->SetMarkerColor(kYellow);
		pointset->SetMarkerSize(3);
	}
	Int_t ipnt = 0;
	if (trigrec != NULL)
	{
		for (int i = 11; i <= 14; ++i)
		{
			if (trigrec->Hit(i) == TVector3(0,0,0)) continue;
			pointset->SetPoint(ipnt, trigrec->Hit(i).X(), trigrec->Hit(i).Y(), trigrec->Hit(i).Z());
			++ipnt;
		}
	}
	for (int j = 7; j <= 10; ++j)
	{
		const AliHLTMUONRecHit* hit = track->Hit(j);
		if (hit == NULL) continue;
		pointset->SetPoint(ipnt, hit->X(), hit->Y(), hit->Z());
		++ipnt;
	}
	
	for (int j = 7; j <= 10; ++j)
	{
		if (track->RoIRadius(j) == -1) continue;
		TEveStraightLineSet* roi = new TEveStraightLineSet(
				Form("RoI%d", j),
				Form("Region of interest for chamber %d", j)
			);
		Float_t p0x = track->RoICentre(j).X() + track->RoIRadius(j);
		Float_t p0y = track->RoICentre(j).Y();
		Float_t p0z = track->RoICentre(j).Z();
		Float_t p1x = p0x;
		Float_t p1y = p0y;
		Float_t p1z = p0z;
		for (int i = 0; i < 32; ++i)
		{
			Double_t t = Double_t(i) / 32. * 2. * TMath::Pi();
			Float_t p2x = track->RoICentre(j).X() + track->RoIRadius(j)*cos(t);
			Float_t p2y = track->RoICentre(j).Y() + track->RoIRadius(j)*sin(t);
			Float_t p2z = track->RoICentre(j).Z();
			roi->AddLine(p1x, p1y, p1z, p2x, p2y, p2z);
			p1x = p2x;
			p1y = p2y;
			p1z = p2z;
		}
		roi->AddLine(p1x, p1y, p1z, p0x, p0y, p0z);
		pointset->AddElement(roi);
	}
	
	list->AddElement(pointset);
}


void FillData(Int_t eventId)
{
	AliHLTMUONEvent* dhltEvent = dynamic_cast<AliHLTMUONEvent*>( gDHLTFile->Get(Form("AliHLTMUONEvent;%d", eventId+1)) );
	if (dhltEvent == NULL)
	{
		cerr << "ERROR: Could not find dHLT data for event " << eventId << endl;
		return;
	}
	
	TEveElementList* list = new TEveElementList("dHLTData", "dHLT data");
	gEve->AddElement(list);
	
	TArrayD px, py, pz;
	for (int i = 0; i < dhltEvent->DataObjects().GetEntriesFast(); i++)
	{
		if (dhltEvent->DataObjects()[i]->IsA() == AliHLTMUONRecHit::Class())
		{
			const AliHLTMUONRecHit* hit = dynamic_cast<const AliHLTMUONRecHit*>( dhltEvent->DataObjects()[i] );
			px.Set(px.GetSize()+1);
			py.Set(py.GetSize()+1);
			pz.Set(pz.GetSize()+1);
			px[px.GetSize()-1] = hit->X();
			py[py.GetSize()-1] = hit->Y();
			pz[pz.GetSize()-1] = hit->Z();
			continue;
		}
		else if (dhltEvent->DataObjects()[i]->IsA() == AliHLTMUONTriggerRecord::Class())
		{
			const AliHLTMUONTriggerRecord* trigrec = dynamic_cast<const AliHLTMUONTriggerRecord*>( dhltEvent->DataObjects()[i] );
			FillData(list, trigrec);
		}
		else if (dhltEvent->DataObjects()[i]->IsA() == AliHLTMUONMansoTrack::Class())
		{
			const AliHLTMUONMansoTrack* track = dynamic_cast<const AliHLTMUONMansoTrack*>( dhltEvent->DataObjects()[i] );
			FillData(list, track);
		}
	}
	
	TEvePointSet* mps = new TEvePointSet(px.GetSize());
	mps->SetName("RecHits");
	mps->SetTitle("dHLT reconstructed hits");
	mps->SetMarkerStyle(28);
	mps->SetMarkerColor(kYellow);
	mps->SetMarkerSize(3);
	for (int i = 0; i < px.GetSize(); ++i)
	{
		mps->SetPoint(i, px[i], py[i], pz[i]);
	}
	list->AddElement(mps);
}
Commit	Line	Data
bc5cb6d6	1	/**************************************************************************
	2	* This file is property of and copyright by the ALICE HLT Project *
	3	* All rights reserved. *
	4	* *
	5	* Primary Authors: *
	6	* Artur Szostak <artursz@iafrica.com> *
	7	* *
	8	* Permission to use, copy, modify and distribute this software and its *
	9	* documentation strictly for non-commercial purposes is hereby granted *
	10	* without fee, provided that the above copyright notice appears in all *
	11	* copies and that both the copyright notice and this permission notice *
	12	* appear in the supporting documentation. The authors make no claims *
	13	* about the suitability of this software for any purpose. It is *
	14	* provided "as is" without express or implied warranty. *
	15	**************************************************************************/
	16
	17	// $Id: DisplaydHLTData.C 37070 2009-11-20 13:53:08Z aszostak $
	18
	19	/**
	20	* \ingroup macros
	21	* \file DisplaydHLTData.C
	22	* \brief Macro for displaying rootified dHLT data generated with RunChain.C
	23	*
	24	* This macro is used to display the dHLT data generated by the RunChain.C macro
	25	* with the "root" option for making the output data. Optionally the dHLTRawData.root
	26	* file which is generated during a AliRoot offline reconstruction can also be
	27	* displayed using this macro.
	28	* After having generated the dHLT raw data file (called <i>output.root</i> for example)
	29	* the following command can be executed in an alieve session to display both
	30	* MUON and dHLT data:
	31	* \code
	32	* > gSystem->Load("libAliHLTMUON.so");
	33	* > .L $ALICE_ROOT/EVE/macros/alieve_init.C
	34	* > .L $ALICE_ROOT/EVE/alice-macros/event_goto.C
	35	* > .L $ALICE_ROOT/EVE/alice-macros/MUON_displayData.C
	36	* > .L $ALICE_ROOT/HLT/MUON/macros/DisplaydHLTData.C
	37	* > alieve_init("OCDBpath","DataPath",EventNumber);
	38	* > MUON_displayData(1,1,1);
	39	* > DisplaydHLTData("output.root");
	40	* \endcode
	41	* Where <i>OCDBpath</i> should be replaced by the path to the OCDB database,
	42	* <i>DataPath</i> should be the directory of the raw data or raw data root file and
	43	* <i>EventNumber</i> is the event number to load.
	44	*
	45	* \author Artur Szostak <artursz@iafrica.com>
	46	*/
	47
	48	#if !defined(__CINT__) \|\| defined(__MAKECINT__)
	49	#include "AliEveEventManager.h"
	50	#include "AliHLTMUONTriggerRecord.h"
	51	#include "AliHLTMUONRecHit.h"
	52	#include "AliHLTMUONMansoTrack.h"
	53	#include "AliHLTMUONEvent.h"
	54	#include "TEveManager.h"
	55	#include "TEveElement.h"
	56	#include "TEvePointSet.h"
	57	#include "TEveStraightLineSet.h"
	58	#include "TArrayD.h"
	59	#include "TFile.h"
	60	#include "TString.h"
	61	#include "TClassTable.h"
	62	#include "TSystem.h"
	63	#include <iostream>
	64	using std::cerr;
65	using std::endl;
66	#endif
67
68
69	TFile* gDHLTFile = NULL;
70
71	void FillData(Int_t eventId);
72
73
74	void DisplaydHLTData(const char* dHLTfilename = "output.root")
75	{
76	// Display dHLT data.
77
78	if (gClassTable->GetID("AliHLTMUONEvent") < 0)
79	{
80	gSystem->Load("libAliHLTMUON.so");
81	}
82
83	if (AliEveEventManager::GetMaster() == NULL)
84	{
85	cerr << "ERROR: alieve event not initialised, use alieve_init(...)" << endl;
86	return;
87	}
88
89	gEve->DisableRedraw();
90
91	TString filename = dHLTfilename;
92	TString tmpfilename = filename;
93	if (gSystem->FindFile("", tmpfilename) == NULL)
94	{
95	tmpfilename = filename;
96	if (gSystem->FindFile(".", tmpfilename) == NULL)
97	{
98	TString filename = TString(AliEveEventManager::GetMaster()->GetTitle());
99	filename += "/";
100	filename += dHLTfilename;
101	}
102	}
103	if (gDHLTFile == NULL)
104	{
105	gDHLTFile = new TFile(filename.Data(), "READ");
106	}
107	else if (filename != gDHLTFile->GetName())
108	{
109	delete gDHLTFile;
110	gDHLTFile = new TFile(filename.Data(), "READ");
111	}
112
113	FillData(AliEveEventManager::GetMaster()->GetEventId());
114
115	gEve->Redraw3D(kTRUE);
116	gEve->EnableRedraw();
117	}
118
119
120	void FillData(TEveElementList* list, const AliHLTMUONTriggerRecord* trigrec)
121	{
122	// Count the number of points in the trigger record.
123	Int_t pointCount = 0;
124	for (int i = 11; i <= 14; ++i)
125	{
126	if (trigrec->Hit(i) != TVector3(0,0,0)) ++pointCount;
127	}
128
129	// Create and fill the point set for the trigger record.
130	TEvePointSet* pointset = new TEvePointSet(pointCount);
131	pointset->SetName(Form("TriggerRecord%d", trigrec->Id()));
132	pointset->SetTitle(Form("dHLT trigger record ID = %d", trigrec->Id()));
133	pointset->SetMarkerStyle(28);
134	pointset->SetMarkerColor(kYellow);
135	pointset->SetMarkerSize(3);
136	Int_t ipnt = 0;
137	for (int i = 11; i <= 14; ++i)
138	{
139	if (trigrec->Hit(i) == TVector3(0,0,0)) continue;
140	pointset->SetPoint(ipnt, trigrec->Hit(i).X(), trigrec->Hit(i).Y(), trigrec->Hit(i).Z());
141	++ipnt;
142	}
143
144	list->AddElement(pointset);
145	}
146
147
148	void FillData(TEveElementList* list, const AliHLTMUONMansoTrack* track)
149	{
150	// Count the points in the track.
151	Int_t pointCount = 0;
152	for (int i = 7; i <= 10; ++i)
153	{
154	const AliHLTMUONRecHit* hit = track->Hit(i);
155	if (hit != NULL) ++pointCount;
156	}
157	const AliHLTMUONTriggerRecord* trigrec = track->TriggerRecord();
158	if (trigrec != NULL)
159	{
160	for (int i = 11; i <= 14; ++i)
161	{
162	if (trigrec->Hit(i) != TVector3(0,0,0)) ++pointCount;
163	}
164	}
165
166	// Create and fill the point set for the track.
167	TEvePointSet* pointset = new TEvePointSet(pointCount);
168	pointset->SetName(Form("Track%d", track->Id()));
169	pointset->SetTitle(Form("dHLT track ID = %d", track->Id()));
170	if (track->Id() != -1)
171	{
172	pointset->SetMarkerStyle(20);
173	pointset->SetMarkerColor(kGreen);
174	pointset->SetMarkerSize(1.5);
175	}
176	else
177	{
178	pointset->SetMarkerStyle(28);
179	pointset->SetMarkerColor(kYellow);
180	pointset->SetMarkerSize(3);
181	}
182	Int_t ipnt = 0;
183	if (trigrec != NULL)
184	{
185	for (int i = 11; i <= 14; ++i)
186	{
187	if (trigrec->Hit(i) == TVector3(0,0,0)) continue;
188	pointset->SetPoint(ipnt, trigrec->Hit(i).X(), trigrec->Hit(i).Y(), trigrec->Hit(i).Z());
189	++ipnt;
190	}
191	}
192	for (int j = 7; j <= 10; ++j)
193	{
194	const AliHLTMUONRecHit* hit = track->Hit(j);
195	if (hit == NULL) continue;
196	pointset->SetPoint(ipnt, hit->X(), hit->Y(), hit->Z());
197	++ipnt;
198	}
199
200	for (int j = 7; j <= 10; ++j)
201	{
202	if (track->RoIRadius(j) == -1) continue;
203	TEveStraightLineSet* roi = new TEveStraightLineSet(
204	Form("RoI%d", j),
205	Form("Region of interest for chamber %d", j)
206	);
207	Float_t p0x = track->RoICentre(j).X() + track->RoIRadius(j);
208	Float_t p0y = track->RoICentre(j).Y();
209	Float_t p0z = track->RoICentre(j).Z();
210	Float_t p1x = p0x;
211	Float_t p1y = p0y;
212	Float_t p1z = p0z;
213	for (int i = 0; i < 32; ++i)
214	{
215	Double_t t = Double_t(i) / 32. * 2. * TMath::Pi();
216	Float_t p2x = track->RoICentre(j).X() + track->RoIRadius(j)*cos(t);
217	Float_t p2y = track->RoICentre(j).Y() + track->RoIRadius(j)*sin(t);
218	Float_t p2z = track->RoICentre(j).Z();
219	roi->AddLine(p1x, p1y, p1z, p2x, p2y, p2z);
220	p1x = p2x;
221	p1y = p2y;
222	p1z = p2z;
223	}
224	roi->AddLine(p1x, p1y, p1z, p0x, p0y, p0z);
225	pointset->AddElement(roi);
226	}
227
228	list->AddElement(pointset);
229	}
230
231
232	void FillData(Int_t eventId)
233	{
234	AliHLTMUONEvent* dhltEvent = dynamic_cast<AliHLTMUONEvent*>( gDHLTFile->Get(Form("AliHLTMUONEvent;%d", eventId+1)) );
235	if (dhltEvent == NULL)
236	{
237	cerr << "ERROR: Could not find dHLT data for event " << eventId << endl;
238	return;
239	}
240
241	TEveElementList* list = new TEveElementList("dHLTData", "dHLT data");
242	gEve->AddElement(list);
243
244	TArrayD px, py, pz;
245	for (int i = 0; i < dhltEvent->DataObjects().GetEntriesFast(); i++)
246	{
247	if (dhltEvent->DataObjects()[i]->IsA() == AliHLTMUONRecHit::Class())
248	{
249	const AliHLTMUONRecHit* hit = dynamic_cast<const AliHLTMUONRecHit*>( dhltEvent->DataObjects()[i] );
250	px.Set(px.GetSize()+1);
251	py.Set(py.GetSize()+1);
252	pz.Set(pz.GetSize()+1);
253	px[px.GetSize()-1] = hit->X();
254	py[py.GetSize()-1] = hit->Y();
255	pz[pz.GetSize()-1] = hit->Z();
256	continue;
257	}
258	else if (dhltEvent->DataObjects()[i]->IsA() == AliHLTMUONTriggerRecord::Class())
259	{
260	const AliHLTMUONTriggerRecord* trigrec = dynamic_cast<const AliHLTMUONTriggerRecord*>( dhltEvent->DataObjects()[i] );
261	FillData(list, trigrec);
262	}
263	else if (dhltEvent->DataObjects()[i]->IsA() == AliHLTMUONMansoTrack::Class())
264	{
265	const AliHLTMUONMansoTrack* track = dynamic_cast<const AliHLTMUONMansoTrack*>( dhltEvent->DataObjects()[i] );
266	FillData(list, track);
267	}
268	}
269
270	TEvePointSet* mps = new TEvePointSet(px.GetSize());
271	mps->SetName("RecHits");
272	mps->SetTitle("dHLT reconstructed hits");
273	mps->SetMarkerStyle(28);
274	mps->SetMarkerColor(kYellow);
275	mps->SetMarkerSize(3);
276	for (int i = 0; i < px.GetSize(); ++i)
277	{
278	mps->SetPoint(i, px[i], py[i], pz[i]);
279	}
280	list->AddElement(mps);
281	}