Separate import of standard macros into a special function so that

[u/mrichter/AliRoot.git] / EVE / alice-macros / jetplane.C

Alieve::JetPlane* jetplane(Int_t iev)

{	
	TFile* f       = new TFile("aod.root");
	TTree* treeAOD = (TTree*) f->Get("AOD");
	AliAODEvent* aod = new AliAODEvent();
  aod->ReadFromTree(treeAOD);
  treeAOD->GetEntry(iev);

	using namespace Alieve;
	JetPlane* jp = new JetPlane(iev);

	// Read Jets in current event

	TClonesArray* jets = aod->GetJets();
	Int_t njets = jets->GetEntries();
	printf("Event: %5d Number of jets: %5d \n", iev, njets);

	for (Int_t ij = 0; ij < njets; ij++)
	{
		AliAODJet jet = (AliAODJet) jets->At(ij);
		jp->AddJet(jet);
	}

// Read tracks in current event

	TClonesArray* tracks = aod->GetTracks();
	Int_t ntracks = tracks->GetEntries();
	printf("Event: %5d Number of tracks: %5d \n", iev, ntracks);

	for (Int_t ij = 0; ij < ntracks; ij++)
	{
		AliAODTrack track = (AliAODTrack) tracks->At(ij);
		jp->AddTrack(track);
	}

// Render Jet Plane
	gStyle->SetPalette(1, 0);
	gReve->AddRenderElement(jp);
	gReve->Redraw3D();

	return jp;
}