Normalize all separator comments to width 80.
[u/mrichter/AliRoot.git] / EVE / alice-macros / primary_vertex.C
1 // $Id$
2 // Main authors: Matevz Tadel & Alja Mrak-Tadel: 2006, 2007
3
4 /**************************************************************************
5  * Copyright(c) 1998-2008, ALICE Experiment at CERN, all rights reserved. *
6  * See http://aliceinfo.cern.ch/Offline/AliRoot/License.html for          *
7  * full copyright notice.                                                 *
8  **************************************************************************/
9
10 TPolyMarker3D* make_vertex_marker(AliESDVertex* v, const Text_t* name )
11 {
12   Double_t x[3], e[3];
13   v->GetXYZ(x);
14   v->GetSigmaXYZ(e);
15
16   printf("%16s: %f %f %f   -   %f %f %f\n", name,
17          x[0], x[1], x[2], e[0], e[1], e[2]);
18
19   TPolyMarker3D* m = new TPolyMarker3D(1);
20   m->SetName(name);
21   m->SetPoint(0, x[0], x[1], x[2]);
22
23   return m;
24 }
25
26 TEveBoxSet* make_vertex_boxes(AliESDVertex* v)
27 {
28   Double_t x[3], e[3];
29   v->GetXYZ(x);
30   v->GetSigmaXYZ(e);
31
32   TEveBoxSet* bs;
33
34   bs = new TEveBoxSet("+- 10 x 10 x 20mm");
35   bs->SetRenderMode(TEveBoxSet::RM_TEveLine);
36   bs->AddBox(Box(-1, x[0], x[1], x[2], 1, 1, 2));
37   bs->SetMainColor((Color_t) 2);
38   gEve->AddElement(bs);
39
40   bs = new TEveBoxSet("+- 30 sigma_r x 10 sigma_z");
41   bs->SetRenderMode(TEveBoxSet::RM_TEveLine);
42   bs->AddBox(Box(-1, x[0], x[1], x[2], 30*e[0], 30*e[1], 10*e[2]));
43   bs->SetMainColor((Color_t) 3);
44   gEve->AddElement(bs);
45
46   gEve->Redraw3D();
47 }
48
49 void register_vertex_marker(TPolyMarker3D* m)
50 {
51   Color_t* colp = TEveUtil::FindColorVar(m, "fMarkerColor");
52   TEveElementObjectPtr* rnrEl = new TEveElementObjectPtr(m, *colp);
53   gEve->AddElement(rnrEl);
54   gEve->Redraw3D();
55 }
56
57 void primary_vertex_primitive(Bool_t showSPD=kTRUE, Bool_t showBoxes=kFALSE)
58 {
59   AliESDEvent* esd = AliEveEventManager::AssertESD();
60
61   AliESDVertex*  pv  = esd->GetPrimaryVertex();
62   TPolyMarker3D* pvm = make_vertex_marker(pv, "Primary Vertex");
63   pvm->SetMarkerStyle(5);
64   pvm->SetMarkerColor(5);
65   pvm->SetMarkerSize(1.4);
66   register_vertex_marker(pvm);
67
68   if (showSPD) {
69     AliESDVertex*  spdv  = esd->GetVertex();
70     TPolyMarker3D* spdvm = make_vertex_marker(spdv, "SPD Vertex");
71     spdvm->SetMarkerStyle(2);
72     spdvm->SetMarkerColor(6);
73     register_vertex_marker(spdvm);
74   }
75
76   if (showBoxes)
77     make_vertex_boxes(pv);
78 }
79
80 /******************************************************************************/
81
82 TEveStraightLineSet* ESDvertex_lineset(AliESDVertex* v, const Text_t* name)
83 {
84   Double_t x[3], e[3];
85   v->GetXYZ(x); v->GetSigmaXYZ(e);
86   printf("%16s: %f %f %f   -   %f %f %f\n", name,
87          x[0], x[1], x[2], e[0], e[1], e[2]);
88
89   // dimensions
90   TEveStraightLineSet* ls = new TEveStraightLineSet();
91   ls->SetName(name);
92   ls->AddLine(e[0], 0, 0, -e[0], 0, 0);
93   ls->AddLine(0, e[1], 0, 0, -e[1], 0);
94   ls->AddLine(0, 0, e[2], 0, 0, -e[2]);
95   for(Int_t i =0; i < 3; i++)
96   {
97     ls->AddMarker(i, 0);
98     ls->AddMarker(i, 1);
99   }
100
101   // centre marker
102   ls->AddMarker(0, 0.5);
103   ls->RefHMTrans().SetPos(x);
104   return ls;
105 }
106
107 void make_vertex_ellipses(TEveStraightLineSet* ls, AliESDVertex* v, Bool_t ellipseUseSigma)
108 {
109   Double_t x[3], e[3];
110   v->GetXYZ(x); v->GetSigmaXYZ(e);
111
112   if (ellipseUseSigma)
113   {
114     e[0] *= 30; e[1] *= 30; e[2] *= 10;
115     ls->SetMarkerStyle(5);
116     ls->SetMarkerColor(5);
117     ls->SetMarkerSize(1.4);
118     ls->SetLineColor(7);
119     ls->SetTitle("+- 30 sigma_r x 10 sigma_z");
120   }
121   else
122   {
123     e[0] = 1; e[1] = 1; e[2] = 2;
124     ls->SetMarkerStyle(2);
125     ls->SetMarkerColor(6);
126     ls->SetLineColor(6);
127     ls->SetTitle("+- 10 x 10 x 20mm");
128   }
129   Int_t N = 32;
130   Float_t S = 2*TMath::Pi()/N;
131   Float_t b, a, phi;
132
133   a = e[0]; b = e[1];
134   for (Int_t i = 0; i<N; i++)
135     ls->AddLine(a*TMath::Cos(i*S)  , b*TMath::Sin(i*S)  , 0,
136                 a*TMath::Cos(i*S+S), b*TMath::Sin(i*S+S), 0);
137
138   a = e[0]; b = e[2];
139   for (Int_t i = 0; i<N; i++)
140     ls->AddLine(a*TMath::Cos(i*S)  , 0, b*TMath::Sin(i*S),
141                 a*TMath::Cos(i*S+S), 0, b*TMath::Sin(i*S+S));
142
143   a = e[1]; b = e[2];
144   for (Int_t i = 0; i<N; i++)
145     ls->AddLine(0, a*TMath::Cos(i*S)  ,  b*TMath::Sin(i*S),
146                 0, a*TMath::Cos(i*S+S),  b*TMath::Sin(i*S+S));
147 }
148
149 void primary_vertex(Bool_t showSPD=kTRUE, Bool_t rnrEllipse=kTRUE)
150 {
151   AliESDEvent* esd = AliEveEventManager::AssertESD();
152   TEveStraightLineSet* ls;
153
154   AliESDVertex* PV  =  esd->GetPrimaryVertex();
155   ls = ESDvertex_lineset(PV, "Primary Vertex");
156   if (rnrEllipse) make_vertex_ellipses(ls, PV, kTRUE);
157   gEve->AddElement(ls);
158
159   if (showSPD)
160   {
161     AliESDVertex*  SPDV  = esd->GetVertex();
162     ls = ESDvertex_lineset(SPDV, "SPD Vertex");
163     if (rnrEllipse) make_vertex_ellipses(ls, SPDV, kFALSE);
164     gEve->AddElement(ls);
165   }
166 }