From Cvetan: as we switched to 10MHz sampling rate change the default max-time-bin...
[u/mrichter/AliRoot.git] / EVE / EveDet / AliEveTPCSectorViz.cxx
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 #include "AliEveTPCSectorViz.h"
11
12 #include <EveDet/AliEveTPCData.h>
13 #include <EveDet/AliEveTPCSectorData.h>
14
15 #include <TEveTrans.h>
16 #include <TStyle.h>
17 #include <TMath.h>
18
19 //==============================================================================
20 //==============================================================================
21 // AliEveTPCSectorViz
22 //==============================================================================
23
24 //______________________________________________________________________________
25 //
26 // Base class for TPC raw-data visualization.
27 // See AliEveTPCSector2D and AliEveTPCSector3D for concrete implementations.
28
29 ClassImp(AliEveTPCSectorViz)
30
31 /******************************************************************************/
32
33 AliEveTPCSectorViz::AliEveTPCSectorViz(const Text_t* n, const Text_t* t) :
34   TEveElement(fFrameColor),
35   TNamed(n, t),
36
37   fTPCData  (0),
38   fSectorID (0),
39
40   fMinTime   (0),
41   fMaxTime   (1000),
42   fThreshold (5),
43   fMaxVal    (128),
44
45   fRnrInn   (kTRUE),
46   fRnrOut1  (kTRUE),
47   fRnrOut2  (kTRUE),
48
49   fFrameColor (4),
50   fRnrFrame   (kTRUE),
51   fAutoTrans  (kFALSE),
52   fRTS        (1),
53
54   fColorArray (0)
55 {
56   // Constructor.
57
58    InitMainTrans();
59 }
60
61 AliEveTPCSectorViz::~AliEveTPCSectorViz()
62 {
63   // Destructor.
64
65   if (fTPCData) fTPCData->DecRefCount();
66   delete [] fColorArray;
67 }
68
69 void AliEveTPCSectorViz::CopyVizParams(const TEveElement* el)
70 {
71   // Copy basic viualization parameters from another TPCSectorViz.
72
73   const AliEveTPCSectorViz* v = dynamic_cast<const AliEveTPCSectorViz*>(el);
74   if (v) {
75     fMinTime   = v->fMinTime;
76     fMaxTime   = v->fMaxTime;
77     fThreshold = v->fThreshold;
78     fMaxVal    = v->fMaxVal;
79
80     fRnrInn    = v->fRnrInn;
81     fRnrOut1   = v->fRnrOut1;
82     fRnrOut2   = v->fRnrOut2;
83   }
84 }
85
86 /******************************************************************************/
87
88 void AliEveTPCSectorViz::SetDataSource(AliEveTPCData* data)
89 {
90   // Set the data source.
91
92   if (data == fTPCData) return;
93   if (fTPCData) fTPCData->DecRefCount();
94   fTPCData = data;
95   if (fTPCData) fTPCData->IncRefCount();
96   IncRTS();
97 }
98
99 void AliEveTPCSectorViz::SetSectorID(Int_t id)
100 {
101   // Set sector id.
102
103   if (id < 0)  id = 0;
104   if (id > 35) id = 35;
105   fSectorID = id;
106   if (fAutoTrans)
107     SetAutoTrans(kTRUE); // Force repositioning.
108   IncRTS();
109 }
110
111 AliEveTPCSectorData* AliEveTPCSectorViz::GetSectorData() const
112 {
113   // Get sector-data.
114
115   return fTPCData ? fTPCData->GetSectorData(fSectorID) : 0;
116 }
117
118 /******************************************************************************/
119
120 void AliEveTPCSectorViz::SetThreshold(Short_t t)
121 {
122   // Set visualization threshold.
123
124   fThreshold = TMath::Min(t, (Short_t)(fMaxVal - 1));
125   ClearColorArray();
126   IncRTS();
127 }
128
129 void AliEveTPCSectorViz::SetMaxVal(Int_t mv)
130 {
131   // Set visualization max signal value.
132   // Signals above this will have the same color.
133
134   fMaxVal = TMath::Max(mv, (Int_t)(fThreshold + 1));
135   ClearColorArray();
136   IncRTS();
137 }
138
139 /******************************************************************************/
140
141 void AliEveTPCSectorViz::SetAutoTrans(Bool_t trans)
142 {
143   // Set automatic update of transformation matrix.
144   // The position is calculated immediately.
145
146   fAutoTrans = trans;
147   if (fAutoTrans)
148   {
149     using namespace TMath;
150     Float_t c = Cos((fSectorID + 0.5)*20*Pi()/180 - PiOver2());
151     Float_t s = Sin((fSectorID + 0.5)*20*Pi()/180 - PiOver2());
152     Float_t z = AliEveTPCSectorData::GetZLength();
153
154     InitMainTrans();
155     TEveTrans& t = RefMainTrans();
156
157     if (fSectorID < 18) {
158       // column major
159       t[0]  = -c;  t[1]  = -s;
160       t[4]  = -s;  t[5]  =  c;
161       t[10] = -1;
162       t[14] =  z;
163     } else {
164       t[0]  =  c;  t[1]  =  s;
165       t[4]  = -s;  t[5]  =  c;
166       t[10] =  1;
167       t[14] = -z;
168     }
169   }
170 }
171 void AliEveTPCSectorViz::SetUseTrans(Bool_t t)
172 {
173   // Set flag spcifying if transformation matrix should be applied.
174
175   RefMainTrans().SetUseTrans(t);
176 }
177
178 /******************************************************************************/
179
180 void AliEveTPCSectorViz::SetupColor(Int_t val, UChar_t* pixel) const
181 {
182   // Set pixel color to represent signal val from the current palette.
183
184   Float_t div  = TMath::Max(1, fMaxVal - fThreshold);
185   Int_t   nCol = gStyle->GetNumberOfColors();
186   Int_t   cBin = TMath::Nint(nCol*(val - fThreshold)/div);
187
188   TEveUtil::ColorFromIdx(gStyle->GetColorPalette(TMath::Min(nCol - 1, cBin)), pixel);
189 }
190
191 void AliEveTPCSectorViz::ClearColorArray()
192 {
193   // Clear cached color array.
194
195   if (fColorArray) {
196     delete [] fColorArray;
197     fColorArray = 0;
198   }
199 }
200
201 void AliEveTPCSectorViz::SetupColorArray() const
202 {
203   // Initialize cached color array.
204
205   if (fColorArray)
206     return;
207
208   fColorArray = new UChar_t [4 * (fMaxVal - fThreshold + 1)];
209   UChar_t* p = fColorArray;
210   for (Int_t v=fThreshold; v<=fMaxVal; ++v, p+=4)
211     SetupColor(v, p);
212 }