Removing compilation warnings and failures
[u/mrichter/AliRoot.git] / TPC / macros / AliEveTree.C
CommitLineData
5cdd5f7f 1/*
2
3 .L $ALICE_ROOT/TPC/macros/AliEveTree.C+
4 MakeEveTree();
5
6 TFile fesd("AliESDs.root");
7 TTree * treeESD = (TTree*)fesd.Get("esdTree");
8
9
10
11 TFile ftpc("TPCdebug.root");
12 TFile fits("ITSdebug.root");
13 TTree * treeTPC = (TTree*)ftpc.Get("Transform");
14 TTree * treeITS = (TTree*)fits.Get("Clusters");
15
16 treeTPC->Draw("gx2/sqrt(gx0^2+gx1^2):sqrt(gx0^2+gx1^2)>>his(100,0,200,100,-1,1)","event==4","")
17 treeITS->Draw("gz/sqrt(gx^2+gy^2):sqrt(gx^2+gy^2)>>his(100,0,200,100,-1,1)","event==4","same")
18
19 treeTPC->Draw("atan2(gx1,gx0):sqrt(gx0^2+gx1^2)>>his(100,0,200,100,-1,1)","event==4","*")
20 treeITS->Draw("atan2(gy,gx):sqrt(gx^2+gy^2)>>his(100,0,200,100,-1,1)","event==4","same*");
21
22*/
23#include "TFile.h"
24#include "TTree.h"
25#include "TTreeStream.h"
26#include "TPolyMarker3D.h"
27#include "TVectorD.h"
28
29#include "AliESDEvent.h"
30#include "AliESDtrack.h"
31#include "AliESDfriendTrack.h"
32#include "AliTrackPointArray.h"
33
34
35AliTrackPointArray* MakeArray(TPolyMarker3D *pol);
36void MakeESDTree(AliESDEvent*esd, TTreeSRedirector &cstream);
37
38void MakeEveTree(){
39 TFile fesd("AliESDs.root");
40 TTree * esdTree = (TTree*)fesd.Get("esdTree");
41 AliESDEvent cesd;
42 esdTree->SetBranchStatus("*",kTRUE);
43 cesd.ReadFromTree(esdTree);
44 TTreeSRedirector cstream("eveTree.root");
45 for (Int_t ievent=0; ievent<esdTree->GetEntries();ievent++) {
46 //cout<<endl<<endl<<"********* Processing event number: "<<event<<"*******\n";
47 esdTree->GetEvent(ievent);
48 AliESDfriend *fESDfriend = (AliESDfriend*)cesd.FindListObject("AliESDfriend");
49 cesd.SetESDfriend(fESDfriend);
50 MakeESDTree(&cesd,cstream);
51 }
52}
53
54
55void MakeESDTree(AliESDEvent*esd, TTreeSRedirector &cstream){
56 //
57 //
58 //
59 Float_t bz = esd->GetMagneticField();
60 //AliTPCseed dummyTPC;
61 //AliTPCseed dummyTRD;
62 for (Int_t i=0;i<esd->GetNumberOfTracks(); i++){
63 AliESDtrack * track = esd->GetTrack(i);
64 if (!track) continue;
65 TPolyMarker3D polA;
66 TPolyMarker3D polV;
67 TPolyMarker3D polI;
68 TPolyMarker3D polO;
69 //
70 AliTrackPointArray * arrayA=0;
71 AliTrackPointArray * arrayV=0;
72 AliTrackPointArray * arrayI=0;
73 AliTrackPointArray * arrayO=0;
74 //
75 if (track->GetInnerParam()) {
76 ((AliExternalTrackParam*)track->GetInnerParam())->FillPolymarker(&polA,bz,0,350,1);}
77 else{
78 track->FillPolymarker(&polA,bz,0,300,5);
79 }
80 arrayA=MakeArray(&polA);
81 track->FillPolymarker(&polV,bz,0,60,1);
82 arrayV=MakeArray(&polV);
83 if (track->GetInnerParam()) {
84 ((AliExternalTrackParam*)track->GetInnerParam())->FillPolymarker(&polI,bz,60,170,1);
85 arrayI = MakeArray(&polI);
86 }
87 if (track->GetOuterParam()) {
88 ((AliExternalTrackParam*)track->GetOuterParam())->FillPolymarker(&polO,bz,170,350,1);
89 arrayO=MakeArray(&polO);
90 }
91 static AliTrackPointArray cldummy(5);
92 AliTrackPointArray *clarray= &cldummy;
93 const AliESDfriendTrack *ftrack = track->GetFriendTrack();
94 if (ftrack && ftrack->GetTrackPointArray()) {
95 clarray=(AliTrackPointArray *)ftrack->GetTrackPointArray();
96 }
97 Int_t event = esd->GetEventNumberInFile();
98 Int_t id = track->GetID();
99 cstream<<"Tracks"<<
100 "eventNr="<<event<<
101 "trackNr="<<id<<
102 "Tr.="<<track<<
103 "Cl.="<<clarray<<
104 //
105 "pA.="<<&polA<<
106 "pV.="<<&polV<<
107 "pI.="<<&polI<<
108 "pO.="<<&polO<<
109 //
110 "aA.="<<arrayA<<
111 "aV.="<<arrayV<<
112 "aI.="<<arrayI<<
113 "aO.="<<arrayO<<
114 "\n";
115 }
116}
117
118
119
120AliTrackPointArray *MakeArray(TPolyMarker3D *pol){
121 //
122 // Make a aray of points with errors
123 //
124 Int_t entries = pol->GetN();
125 AliTrackPointArray * array = new AliTrackPointArray(entries);
126 for (Int_t i=0;i<entries;i++){
127 Double_t xyz[3]={0,0,0};
128 pol->GetPoint(i,xyz[0],xyz[1],xyz[2]);
129 ((Float_t*)array->GetX())[i]=xyz[0];
130 ((Float_t*)array->GetY())[i]=xyz[1];
131 ((Float_t*)array->GetZ())[i]=xyz[2];
132 }
133 return array;
134}
135