|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
// -*- C++ -*-
//
// Package: Core
// Class : FWGeometryTableViewManager
//
// Implementation:
// [Notes on implementation]
//
// Original Author: Alja Mrak-Tadel
// Created: Fri Jul 8 00:40:37 CEST 2011
//
#include <functional>
#include <memory>
#include "TFile.h"
#include "TSystem.h"
#include "TGeoManager.h"
#include "TGeoMatrix.h"
#include "TEveManager.h"
#include "Fireworks/Core/interface/FWGeometryTableViewManager.h"
#include "Fireworks/Core/src/FWGeometryTableView.h"
#include "Fireworks/Core/src/FWOverlapTableView.h"
#include "Fireworks/Core/interface/FWGeometry.h"
#include "Fireworks/Core/interface/FWGUIManager.h"
#include "Fireworks/Core/interface/FWColorManager.h"
#include "Fireworks/Core/interface/fwLog.h"
TGeoManager* FWGeometryTableViewManager::s_geoManager = nullptr;
TGeoManager* FWGeometryTableViewManager_GetGeoManager() { return FWGeometryTableViewManager::getGeoMangeur(); }
FWGeometryTableViewManager::FWGeometryTableViewManager(FWGUIManager* iGUIMgr, std::string fileName, std::string geoName)
: FWViewManagerBase(), m_fileName(fileName), m_TGeoName(geoName) {
FWGUIManager::ViewBuildFunctor f;
f = std::bind(&FWGeometryTableViewManager::buildView, this, std::placeholders::_1, std::placeholders::_2);
iGUIMgr->registerViewBuilder(FWViewType::idToName(FWViewType::kGeometryTable), f);
iGUIMgr->registerViewBuilder(FWViewType::idToName(FWViewType::kOverlapTable), f);
}
FWGeometryTableViewManager::~FWGeometryTableViewManager() {}
FWViewBase* FWGeometryTableViewManager::buildView(TEveWindowSlot* iParent, const std::string& type) {
if (!s_geoManager)
setGeoManagerFromFile();
std::shared_ptr<FWGeometryTableViewBase> view;
FWViewType::EType typeId =
(type == FWViewType::sName[FWViewType::kGeometryTable]) ? FWViewType::kGeometryTable : FWViewType::kOverlapTable;
if (typeId == FWViewType::kGeometryTable)
view = std::make_shared<FWGeometryTableView>(iParent, &colorManager());
else
view = std::make_shared<FWOverlapTableView>(iParent, &colorManager());
view->setBackgroundColor();
m_views.push_back(std::shared_ptr<FWGeometryTableViewBase>(view));
view->beingDestroyed_.connect(std::bind(&FWGeometryTableViewManager::beingDestroyed, this, std::placeholders::_1));
return view.get();
}
void FWGeometryTableViewManager::beingDestroyed(const FWViewBase* iView) {
for (std::vector<std::shared_ptr<FWGeometryTableViewBase> >::iterator it = m_views.begin(); it != m_views.end();
++it) {
if (it->get() == iView) {
m_views.erase(it);
return;
}
}
}
void FWGeometryTableViewManager::colorsChanged() {
for (std::vector<std::shared_ptr<FWGeometryTableViewBase> >::iterator it = m_views.begin(); it != m_views.end(); ++it)
(*it)->setBackgroundColor();
}
//______________________________________________________________________________
TGeoManager* FWGeometryTableViewManager::getGeoMangeur() {
// Function used in geometry table views.
assert(s_geoManager);
return s_geoManager;
}
//______________________________________________________________________________
void FWGeometryTableViewManager::setGeoManagerRuntime(TGeoManager* x) {
// Function called from FWFFLooper to set geometry created in runtime.
s_geoManager = x;
}
//______________________________________________________________________________
void FWGeometryTableViewManager::setGeoManagerFromFile() {
TFile* file = FWGeometry::findFile(m_fileName.c_str());
fwLog(fwlog::kInfo) << "Geometry table file: " << m_fileName.c_str() << std::endl;
try {
if (!file) {
// Try it as a GDML file
s_geoManager = TGeoManager::Import(m_fileName.c_str(), m_TGeoName.c_str());
} else {
file->ls();
s_geoManager = (TGeoManager*)file->Get(m_TGeoName.c_str());
}
if (!s_geoManager)
throw std::runtime_error("Can't find TGeoManager object in selected file.");
} catch (std::runtime_error& e) {
fwLog(fwlog::kError) << e.what();
exit(0);
}
}
|