]> git.uio.no Git - u/mrichter/AliRoot.git/blob - EVE/Alieve/JetPlane.cxx
From Cesar: First version of jet visualization code.
[u/mrichter/AliRoot.git] / EVE / Alieve / JetPlane.cxx
1 // $Header$
2
3 #include "JetPlane.h"
4 #include <TString.h>
5 #include <TBuffer3D.h>
6 #include <TBuffer3DTypes.h>
7 #include <TVirtualPad.h>
8 #include <TVirtualViewer3D.h>
9
10 using namespace Alieve;
11
12 //______________________________________________________________________
13 // JetPlane
14 //
15
16
17 Bool_t JetPlane::fgOneMomentumXYZ      = kFALSE;
18 Bool_t JetPlane::fgOneMomentumPhiTheta = kFALSE;
19 Bool_t JetPlane::fgOneEta     = kFALSE;
20 Bool_t JetPlane::fgOneE       = kFALSE;
21 Bool_t JetPlane::fgOneChgMass = kFALSE;
22
23
24 ClassImp(JetPlane)
25
26 JetPlane::JetPlane(Int_t iev) :
27   RenderElementList(Form("JetPlane %i",iev), Form("%i",iev)),
28
29   fMinEta (-1.5 ),
30   fMaxEta ( 1.5 ),
31   fMinPhi (-TMath::Pi() ),
32   fMaxPhi ( TMath::Pi() ),
33
34   fNEtaDiv(30),
35   fNPhiDiv(30),
36
37   fEtaScale(350/1.5),
38   fPhiScale(350/TMath::Pi()),
39   fEnergyScale(100.0),
40
41   fEnergyColorScale (0.),
42
43   fGridColor(5),
44
45   fRnrJets (kTRUE),
46   fRnrTracks (kTRUE),
47
48   fOneSelection (kTRUE),
49   fTwoSelection (kFALSE),
50
51   fSelectionFlag (1)
52 {
53   SetMainColorPtr(&fGridColor);
54 }
55
56 /**************************************************************************/
57
58 void JetPlane::AddJet(AliAODJet jet)
59 {
60   fJets.push_back(jet);
61 }
62
63 /**************************************************************************/
64
65 void JetPlane::AddTrack(AliAODTrack track)
66 {
67   fTracks.push_back(track);
68 }
69
70
71 /**************************************************************************/
72
73 void JetPlane::ComputeBBox()
74 {
75   BBoxInit();
76   BBoxCheckPoint(-350, -350, -20);
77   BBoxCheckPoint( 350,  350,  20);
78 }
79
80 void JetPlane::Paint(Option_t* /*option*/)
81 {
82   TBuffer3D buff(TBuffer3DTypes::kGeneric);
83
84   // Section kCore
85   buff.fID           = this;
86   buff.fColor        = fGridColor;
87   buff.fTransparency = 0;
88   fHMTrans.SetBuffer3D(buff);
89   buff.SetSectionsValid(TBuffer3D::kCore);
90
91   Int_t reqSections = gPad->GetViewer3D()->AddObject(buff);
92   if (reqSections == TBuffer3D::kNone) {
93     // printf("JetPlane::Paint viewer was happy with Core buff3d.\n");
94     return;
95   }
96 }