CMS 3D CMS Logo

 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
FWRPZView.cc
Go to the documentation of this file.
1 // -*- C++ -*-
2 //
3 // Package: Core
4 // Class : FWRPZView
5 //
6 // Implementation:
7 // <Notes on implementation>
8 //
9 // Original Author: Chris Jones
10 // Created: Tue Feb 19 10:33:25 EST 2008
11 //
12 
13 // system include files
14 #include <stdexcept>
15 #include <boost/bind.hpp>
16 #include <boost/shared_ptr.hpp>
17 
18 #include "TGLViewer.h"
19 #include "TGLScenePad.h"
20 #include "TEveManager.h"
21 #include "TEveElement.h"
22 #include "TEveScene.h"
23 #include "TEveProjections.h"
24 #include "TEveProjectionAxes.h"
25 #include "TGLabel.h"
26 #include "TEveProjectionManager.h"
27 
29 #define protected public
30 #include "TEveCalo.h"
31 #undef protected
32 
33 // user include files
43 
44 
45 const float FWRPZView::s_distortF = 0.001;
46 const float FWRPZView::s_distortFInv = 1000;
47 //
48 // constructors and destructor
49 //
50 FWRPZView::FWRPZView(TEveWindowSlot* iParent, FWViewType::EType id) :
51  FWEveView(iParent, id, 7),
52  m_geometryList(0),
53  m_projMgr(0),
54  m_axes(0),
55 
56  m_calo(0),
57 
58  m_showPixelBarrel(this, "Show Pixel Barrel", false ),
59  m_showPixelEndcap(this, "Show Pixel Endcap", false),
60  m_showTrackerBarrel(this, "Show Tracker Barrel", false ),
61  m_showTrackerEndcap(this, "Show Tracker Endcap", false),
62  m_showRpcEndcap(this, "Show RPC Endcap", false ),
63  m_showGEM(this, "Show GEM", false ),
64 
65  m_shiftOrigin(this,"Shift origin to beam-spot", false),
66  m_fishEyeDistortion(this,"Distortion",0., 0., 100.),
67  m_fishEyeR(this,"FixedRadius",(double)fireworks::Context::caloR1(), 0.0, 150.0),
68 
69  m_caloDistortion(this,"Calo compression",1.0,0.01,10.),
70  m_muonDistortion(this,"Muon compression",0.2,0.01,10.),
71  m_showProjectionAxes(this,"Show projection axis", false),
72  m_projectionAxesLabelSize(this,"Projection axis label size", 0.015, 0.001, 0.2),
73  m_compressMuon(this,"Compress detectors",false),
74  m_showHF(0),
75  m_showEndcaps(0)
76 {
77  TEveProjection::EPType_e projType = (id == FWViewType::kRhoZ) ? TEveProjection::kPT_RhoZ : TEveProjection::kPT_RPhi;
78 
79  m_projMgr = new TEveProjectionManager(projType);
80  m_projMgr->IncDenyDestroy();
81  m_projMgr->SetImportEmpty(kTRUE);
82 
83  m_projMgr->GetProjection()->SetDistortion(m_fishEyeDistortion.value()*s_distortF);
84  m_projMgr->GetProjection()->SetFixR(m_fishEyeR.value());
85 
86 #ifdef TEVEPROJECTIONS_DISPLACE_ORIGIN_MODE
87  m_projMgr->GetProjection()->SetDisplaceOrigin( m_shiftOrigin.value());
88 #endif
89 
90  if ( id == FWViewType::kRhoPhi || id == FWViewType::kRhoPhiPF) {
91  m_projMgr->GetProjection()->AddPreScaleEntry(0, fireworks::Context::caloR1(), 1.0);
92  m_projMgr->GetProjection()->AddPreScaleEntry(0, 300, 0.2);
93  } else {
94  m_projMgr->GetProjection()->AddPreScaleEntry(0, fireworks::Context::caloR1(), 1.0);
95  m_projMgr->GetProjection()->AddPreScaleEntry(1, 310, 1.0);
96  m_projMgr->GetProjection()->AddPreScaleEntry(0, 370, 0.2);
97  m_projMgr->GetProjection()->AddPreScaleEntry(1, 580, 0.2);
98  }
99 
100 
101  // camera
102  viewerGL()->SetCurrentCamera(TGLViewer::kCameraOrthoXOY);
103  if ( TGLOrthoCamera* camera = dynamic_cast<TGLOrthoCamera*>( &(viewerGL()->CurrentCamera()) ) ) {
104  camera->SetZoomMax(1e6);
105  }
106  geoScene()->GetGLScene()->SetSelectable(kFALSE);
107 
108  m_axes = new TEveProjectionAxes(m_projMgr);
109  m_axes->SetRnrState(m_showProjectionAxes.value());
110  m_axes->SetLabelSize(m_projectionAxesLabelSize.value());
111  m_showProjectionAxes.changed_.connect(boost::bind(&FWRPZView::showProjectionAxes,this));
113  eventScene()->AddElement(m_axes);
114 
115  if ( id != FWViewType::kRhoZ ) {
116  m_showEndcaps = new FWBoolParameter(this,"Include EndCaps", true);
117  m_showEndcaps->changed_.connect( boost::bind(&FWRPZView::setEtaRng, this) );
118  m_showHF = new FWBoolParameter(this,"Include HF", true);
119  m_showHF->changed_.connect( boost::bind(&FWRPZView::setEtaRng, this) );
120  }
121 
122  m_shiftOrigin.changed_.connect(boost::bind(&FWRPZView::doShiftOriginToBeamSpot,this));
123 
124  m_fishEyeDistortion.changed_.connect(boost::bind(&FWRPZView::doFishEyeDistortion,this));
125 
126  m_fishEyeR.changed_.connect(boost::bind(&FWRPZView::doFishEyeDistortion,this));
127 
128  m_caloDistortion.changed_.connect(boost::bind(&FWRPZView::doPreScaleDistortion,this));
129  m_muonDistortion.changed_.connect(boost::bind(&FWRPZView::doPreScaleDistortion,this));
130  m_compressMuon.changed_.connect(boost::bind(&FWRPZView::doCompression,this,_1));
131 }
132 
134 {
135  m_calo->Destroy();
136  m_projMgr->DecDenyDestroy();
137 }
138 
139 //
140 // member functions
141 //
142 
143 TEveCaloViz*
145 {
146  return static_cast<TEveCaloViz*>(m_calo);
147 }
148 
149 void
151 {
153 
155  m_geometryList->IncDenyDestroy();
157  TEveElement* p = m_projMgr->ImportElements(m_geometryList);
158  geoScene()->AddElement(p);
159 
160  TEveCaloData* data = context().getCaloData();
161 
162  TEveCalo3D* calo3d = new TEveCalo3D(data);
163 
164  m_calo = static_cast<TEveCalo2D*> (m_projMgr->ImportElements(calo3d, eventScene()));
165 
166  if (typeId() == FWViewType::kRhoPhiPF)
167  m_calo->SetBarrelRadius(177);
168  else
169  m_calo->SetBarrelRadius(context().caloR1(false));
170 
171  m_calo->SetEndCapPos(context().caloZ1(false));
172  m_calo->SetAutoRange(false);
173  m_calo->SetScaleAbs(true);
174 
181 
182 }
183 
184 void
186 {
187  if (context().getBeamSpot())
188  {
189  FWBeamSpot& b = *(context().getBeamSpot());
190  fwLog(fwlog::kDebug) << Form("%s::eventBegin Set projection center (%f, %f, %f) \n", typeName().c_str(), b.x0(), b.y0(), b.z0());
191 
192  // projection center
193  TEveVector center(b.x0(), b.y0(), b.z0());
194  m_projMgr->GetProjection()->SetCenter(center);
195 
196  // camera move
197  TGLCamera& cam = viewerGL()->CurrentCamera();
198  cam.SetExternalCenter(true);
199  if (typeId() != FWViewType::kRhoZ)
200  {
201  double r = TMath::Sqrt( b.x0()*b.x0() + b.y0()*b.y0());
202  cam.SetCenterVec(b.z0(), TMath::Sign(r, b.y0()), 0);
203  }
204  else
205  {
206  cam.SetCenterVec(b.x0(), b.y0(), b.z0());
207  }
208  }
209 }
210 
211 void
213 {
214 #ifdef TEVEPROJECTIONS_DISPLACE_ORIGIN_MODE
215 
216  TEveProjection* p = m_projMgr->GetProjection();
217  if (p->GetDisplaceOrigin() != m_shiftOrigin.value())
218  {
219  p->SetDisplaceOrigin( m_shiftOrigin.value());
220  m_projMgr->ProjectChildren();
221  gEve->Redraw3D();
222  }
223 #endif
224 }
225 
226 void
227 FWRPZView::shiftOrigin(TEveVector& center)
228 {
229 #ifdef TEVEPROJECTIONS_DISPLACE_ORIGIN_MODE
230  // re-project with new center
231  m_projMgr->GetProjection()->SetCenter(center);
232  m_projMgr->ProjectChildren();
233 
234  // draw projected center
235  float* pc = m_projMgr->GetProjection()->GetProjectedCenter();
236  viewerGL()->CurrentCamera().SetExternalCenter(true);
237  viewerGL()->CurrentCamera().SetCenterVec(pc[0], pc[1], pc[2]);
238  viewerGL()->SetDrawCameraCenter(true);
239 
240  gEve->Redraw3D();
241 #endif
242 }
243 
244 void
246 {
247  // set center back to beam spot
248 
249 #ifdef TEVEPROJECTIONS_DISPLACE_ORIGIN_MODE
250  FWBeamSpot& b = *(context().getBeamSpot());
251  TEveVector center(b.x0(), b.y0(), b.z0());
252  m_projMgr->GetProjection()->SetCenter(center);
253 
254  m_projMgr->ProjectChildren();
255  gEve->Redraw3D();
256 #endif
257 }
258 
259 
260 void
262 {
263  TEveProjection* p = m_projMgr->GetProjection();
264  if (p->GetDistortion() != m_fishEyeDistortion.value()*s_distortFInv)
265  p->SetDistortion(m_fishEyeDistortion.value()*s_distortF);
266  if (p->GetFixR() != m_fishEyeR.value())
267  p->SetFixR(m_fishEyeR.value());
268 
269  m_projMgr->ProjectChildren();
270  gEve->Redraw3D();
271 }
272 
273 void
275 {
277  m_projMgr->GetProjection()->ChangePreScaleEntry(0,1,m_caloDistortion.value());
278  m_projMgr->GetProjection()->ChangePreScaleEntry(0,2,m_muonDistortion.value());
279  } else {
280  m_projMgr->GetProjection()->ChangePreScaleEntry(0,1,m_caloDistortion.value());
281  m_projMgr->GetProjection()->ChangePreScaleEntry(0,2,m_muonDistortion.value());
282  m_projMgr->GetProjection()->ChangePreScaleEntry(1,1,m_caloDistortion.value());
283  m_projMgr->GetProjection()->ChangePreScaleEntry(1,2,m_muonDistortion.value());
284  }
285  m_projMgr->UpdateName();
286  m_projMgr->ProjectChildren();
287  gEve->Redraw3D();
288 }
289 
290 void
292 {
293  m_projMgr->GetProjection()->SetUsePreScale(flag);
294  m_projMgr->UpdateName();
295  m_projMgr->ProjectChildren();
296  gEve->Redraw3D();
297 }
298 
299 void
300 FWRPZView::importElements(TEveElement* iChildren, float iLayer, TEveElement* iProjectedParent)
301 {
302  float oldLayer = m_projMgr->GetCurrentDepth();
303  m_projMgr->SetCurrentDepth(iLayer);
304  //make sure current depth is reset even if an exception is thrown
305  boost::shared_ptr<TEveProjectionManager> sentry(m_projMgr,
306  boost::bind(&TEveProjectionManager::SetCurrentDepth,
307  _1,oldLayer));
308  m_projMgr->ImportElements(iChildren,iProjectedParent);
309 }
310 
311 
312 void
314 {
315  FWEveView::addTo(iTo);
316  TGLOrthoCamera* camera = dynamic_cast<TGLOrthoCamera*>( &(viewerGL()->CurrentCamera()) );
317  if (camera) addToOrthoCamera(camera, iTo);
318 }
319 
320 void
322 {
323  FWEveView::setFrom(iFrom);
324 
325  TGLOrthoCamera* camera = dynamic_cast<TGLOrthoCamera*>( &(viewerGL()->CurrentCamera()) );
326  if (camera) setFromOrthoCamera(camera, iFrom);
327 
328  if (iFrom.version() < 7)
329  {
330  const FWConfiguration* value = iFrom.valueForKey("Show projection axes");
331  if (value)
332  m_showProjectionAxes.set(value->value() == "1");
333  }
334 }
335 
336 void
338 {
339  if (typeId() != FWViewType::kRhoZ)
340  {
341  // rng controllers only in RhoPhi
342  double eta_range = context().caloMaxEta();
343  if (!m_showHF->value() ) eta_range = 3.0;
344  if (!m_showEndcaps->value() ) eta_range = context().caloTransEta();
345  m_calo->SetEta(-eta_range,eta_range);
346  }
347 
349 }
350 
351 void
353 {
354  if (! m_calo->GetData()->Empty())
355  {
356  m_calo->AssertCellIdCache();
357  Float_t sumEt, sumE;
358  TEveCaloData::CellData_t cellData;
359  typedef std::vector<TEveCaloData::vCellId_t*> vBinCells_t;
360  typedef std::vector<TEveCaloData::vCellId_t*>::iterator vBinCells_i;
361 
362  vBinCells_t cellLists = m_calo->fCellLists;
363  for (vBinCells_i it = cellLists.begin(); it != cellLists.end(); it++)
364  {
365  TEveCaloData::vCellId_t* binCells = *it;
366  if (binCells) {
367  sumEt = 0; sumE = 0;
368  TEveCaloData::vCellId_i a = binCells->end();
369 
370  for (TEveCaloData::vCellId_i k = binCells->begin(); k != a; ++k)
371  {
372  m_calo->GetData()->GetCellData((*k), cellData);
373  sumEt += cellData.Value(true);
374  sumE += cellData.Value(false);
375  }
376  // printf("vote sum %f %f \n", sumEt , sumE);
377  context().voteMaxEtAndEnergy(sumEt, sumE);
378  }
379  }
380  }
381 }
382 
384 {
385  m_axes->SetRnrState(m_showProjectionAxes.value());
386  gEve->Redraw3D();
387  viewerGL()->RequestDraw();
388 }
389 
391 {
392  m_axes->SetLabelSize(m_projectionAxesLabelSize.value());
393  viewerGL()->RequestDraw();
394 }
395 
396 void
398 {
400 
401  ViewerParameterGUI& det = gui.requestTab("Detector");;
403 
404  if (typeId() == FWViewType::kRhoZ)
405  {
409  bool showGEM = m_context->getGeom()->versionInfo().haveExtraDet("GEM");
410  if (showGEM) det.addParam(&m_showGEM);
411  }
412 
413 #ifdef TEVEPROJECTIONS_DISPLACE_ORIGIN_MODE
414  gui.requestTab("Projection").addParam(&m_shiftOrigin);
415 #endif
416 
417  gui.requestTab("Projection").
418  addParam(&m_showProjectionAxes).
419  addParam(&m_projectionAxesLabelSize).
420  separator();
421 
422  TGCompositeFrame* f = gui.getTabContainer();
423 
424  f->AddFrame(new TGLabel(f, "FishEye:"));
426  addParam(&m_fishEyeR).
427  separator();
428 
429  f->AddFrame(new TGLabel(f, "PreScales:"));
430 
431  gui.requestTab("Projection").
432  addParam(&m_compressMuon).
433  addParam(&m_muonDistortion).
434  addParam(&m_caloDistortion);
435 
436 
438  {
439  gui.requestTab("Calo").
440  addParam(m_showHF).
441  addParam(m_showEndcaps);
442  }
443 }
444 
static float caloTransEta()
Definition: Context.cc:230
void doShiftOriginToBeamSpot()
Definition: FWRPZView.cc:212
void addToOrthoCamera(TGLOrthoCamera *, FWConfiguration &) const
Definition: FWEveView.cc:418
void doFishEyeDistortion()
Definition: FWRPZView.cc:261
virtual void populateController(ViewerParameterGUI &) const
Definition: FWRPZView.cc:397
FWBoolParameter m_shiftOrigin
Definition: FWRPZView.h:98
TEveScene * eventScene()
Definition: FWEveView.h:82
FWBoolParameter m_showProjectionAxes
Definition: FWRPZView.h:104
double x0() const
Definition: FWBeamSpot.cc:30
FWRPZView(TEveWindowSlot *iParent, FWViewType::EType)
Definition: FWRPZView.cc:50
T Sign(T A, T B)
Definition: MathUtil.h:54
virtual TEveCaloViz * getEveCalo() const
Definition: FWRPZView.cc:144
TEveCalo2D * m_calo
Definition: FWRPZView.h:86
double z0() const
Definition: FWBeamSpot.cc:40
double y0() const
Definition: FWBeamSpot.cc:35
virtual void setContext(const fireworks::Context &)
Definition: FWRPZView.cc:150
FWDoubleParameter m_projectionAxesLabelSize
Definition: FWRPZView.h:105
virtual void populateController(ViewerParameterGUI &) const
Definition: FWEveView.cc:556
const FWGeometry * getGeom() const
Definition: Context.h:83
void voteMaxEtAndEnergy(float Et, float energy) const
Definition: Context.cc:185
ViewerParameterGUI & requestTab(const char *)
virtual void setFrom(const FWConfiguration &)
Definition: FWEveView.cc:364
const std::string & typeName() const
Definition: FWViewBase.cc:120
unsigned int version() const
virtual void setupEnergyScale()
Definition: FWEveView.cc:307
TEveProjectionAxes * m_axes
Definition: FWRPZView.h:85
TGLViewer * viewerGL() const
Definition: FWEveView.cc:194
sigc::signal< void, T > changed_
FWParameters::Bool FWBoolParameter
Definition: FWParameters.h:38
const fireworks::Context * m_context
Definition: FWEveView.h:111
const VersionInfo & versionInfo() const
Definition: FWGeometry.h:117
FWBoolParameter * m_showHF
Definition: FWRPZView.h:108
TEveProjectionManager * m_projMgr
Definition: FWRPZView.h:84
FWDoubleParameter m_fishEyeR
Definition: FWRPZView.h:100
void resetOrigin()
Definition: FWRPZView.cc:245
FWBoolParameter m_showRpcEndcap
Definition: FWRPZView.h:95
float caloR1()
Definition: FWPFGeom.h:21
void initStdGeoElements(const FWViewType::EType id)
bool haveExtraDet(const char *) const
Definition: FWGeometry.cc:379
void doCompression(bool)
Definition: FWRPZView.cc:291
FWBeamSpot * getBeamSpot() const
Definition: Context.h:78
virtual void setFrom(const FWConfiguration &)
Definition: FWRPZView.cc:321
void setEtaRng()
Definition: FWRPZView.cc:337
virtual void voteCaloMaxVal()
Definition: FWRPZView.cc:352
FWBoolParameter m_showTrackerBarrel
Definition: FWRPZView.h:93
void showProjectionAxes()
Definition: FWRPZView.cc:383
static float caloR1(bool offset=true)
Definition: Context.cc:210
double f[11][100]
void importElements(TEveElement *iProjectableChild, float layer, TEveElement *iProjectedParent=0)
Definition: FWRPZView.cc:300
TEveCaloDataHist * getCaloData() const
Definition: Context.h:80
virtual void addTo(FWConfiguration &) const
Definition: FWRPZView.cc:313
FWBoolParameter m_compressMuon
Definition: FWRPZView.h:106
FWRPZViewGeometry * m_geometryList
Definition: FWRPZView.h:83
void shiftOrigin(TEveVector &center)
Definition: FWRPZView.cc:227
const std::string & value(unsigned int iIndex=0) const
const fireworks::Context & context()
Definition: FWEveView.h:67
virtual ~FWRPZView()
Definition: FWRPZView.cc:133
virtual void eventBegin()
Definition: FWRPZView.cc:185
#define fwLog(_level_)
Definition: fwLog.h:50
void doPreScaleDistortion()
Definition: FWRPZView.cc:274
FWBoolParameter * m_showEndcaps
Definition: FWRPZView.h:109
FWBoolParameter m_showGEM
Definition: FWRPZView.h:96
float caloZ1()
Definition: FWPFGeom.h:22
static const float s_distortFInv
Definition: FWRPZView.h:81
double b
Definition: hdecay.h:120
void showTrackerBarrel(bool)
FWDoubleParameter m_fishEyeDistortion
Definition: FWRPZView.h:99
TEveScene * geoScene()
Definition: FWEveView.h:83
ViewerParameterGUI & addParam(const FWParameterBase *)
FWBoolParameter m_showTrackerEndcap
Definition: FWRPZView.h:94
TGCompositeFrame * getTabContainer()
char data[epos_bytes_allocation]
Definition: EPOS_Wrapper.h:82
static double caloMaxEta()
Definition: Context.cc:240
double a
Definition: hdecay.h:121
FWDoubleParameter m_muonDistortion
Definition: FWRPZView.h:103
FWBoolParameter m_showPixelEndcap
Definition: FWRPZView.h:92
FWDoubleParameter m_caloDistortion
Definition: FWRPZView.h:102
void showTrackerEndcap(bool)
const FWConfiguration * valueForKey(const std::string &iKey) const
volatile std::atomic< bool > shutdown_flag false
FWBoolParameter m_showPixelBarrel
Definition: FWRPZView.h:91
void setFromOrthoCamera(TGLOrthoCamera *, const FWConfiguration &)
Definition: FWEveView.cc:438
virtual void addTo(FWConfiguration &) const
Definition: FWEveView.cc:346
static const float s_distortF
Definition: FWRPZView.h:80
void projectionAxesLabelSize()
Definition: FWRPZView.cc:390
virtual void setContext(const fireworks::Context &x)
Definition: FWEveView.cc:275
FWViewType::EType typeId() const
Definition: FWViewBase.h:41