File indexing completed on 2025-01-07 03:06:34
0001
0002
0003
0004
0005
0006
0007
0008
0009
0010
0011
0012
0013 #include <functional>
0014 #include <memory>
0015
0016 #include "TFile.h"
0017 #include "TSystem.h"
0018 #include "TGeoManager.h"
0019 #include "TGeoMatrix.h"
0020 #include "TEveManager.h"
0021
0022 #include "Fireworks/Core/interface/FWGeometryTableViewManager.h"
0023 #include "Fireworks/Core/src/FWGeometryTableView.h"
0024 #include "Fireworks/Core/src/FWOverlapTableView.h"
0025 #include "Fireworks/Core/interface/FWGeometry.h"
0026 #include "Fireworks/Core/interface/FWGUIManager.h"
0027 #include "Fireworks/Core/interface/FWColorManager.h"
0028 #include "Fireworks/Core/interface/fwLog.h"
0029
0030 TGeoManager* FWGeometryTableViewManager::s_geoManager = nullptr;
0031
0032 TGeoManager* FWGeometryTableViewManager_GetGeoManager() { return FWGeometryTableViewManager::getGeoMangeur(); }
0033
0034 FWGeometryTableViewManager::FWGeometryTableViewManager(FWGUIManager* iGUIMgr, std::string fileName, std::string geoName)
0035 : FWViewManagerBase(), m_fileName(fileName), m_TGeoName(geoName) {
0036 FWGUIManager::ViewBuildFunctor f;
0037 f = std::bind(&FWGeometryTableViewManager::buildView, this, std::placeholders::_1, std::placeholders::_2);
0038 iGUIMgr->registerViewBuilder(FWViewType::idToName(FWViewType::kGeometryTable), f);
0039 iGUIMgr->registerViewBuilder(FWViewType::idToName(FWViewType::kOverlapTable), f);
0040 }
0041
0042 FWGeometryTableViewManager::~FWGeometryTableViewManager() {}
0043
0044 FWViewBase* FWGeometryTableViewManager::buildView(TEveWindowSlot* iParent, const std::string& type) {
0045 if (!s_geoManager)
0046 setGeoManagerFromFile();
0047 std::shared_ptr<FWGeometryTableViewBase> view;
0048
0049 FWViewType::EType typeId =
0050 (type == FWViewType::sName[FWViewType::kGeometryTable]) ? FWViewType::kGeometryTable : FWViewType::kOverlapTable;
0051 if (typeId == FWViewType::kGeometryTable)
0052 view = std::make_shared<FWGeometryTableView>(iParent, &colorManager());
0053 else
0054 view = std::make_shared<FWOverlapTableView>(iParent, &colorManager());
0055
0056 view->setBackgroundColor();
0057 m_views.push_back(std::shared_ptr<FWGeometryTableViewBase>(view));
0058 view->beingDestroyed_.connect(std::bind(&FWGeometryTableViewManager::beingDestroyed, this, std::placeholders::_1));
0059
0060 return view.get();
0061 }
0062
0063 void FWGeometryTableViewManager::beingDestroyed(const FWViewBase* iView) {
0064 for (std::vector<std::shared_ptr<FWGeometryTableViewBase> >::iterator it = m_views.begin(); it != m_views.end();
0065 ++it) {
0066 if (it->get() == iView) {
0067 m_views.erase(it);
0068 return;
0069 }
0070 }
0071 }
0072
0073 void FWGeometryTableViewManager::colorsChanged() {
0074 for (std::vector<std::shared_ptr<FWGeometryTableViewBase> >::iterator it = m_views.begin(); it != m_views.end(); ++it)
0075 (*it)->setBackgroundColor();
0076 }
0077
0078
0079 TGeoManager* FWGeometryTableViewManager::getGeoMangeur() {
0080
0081
0082 assert(s_geoManager);
0083 return s_geoManager;
0084 }
0085
0086
0087 void FWGeometryTableViewManager::setGeoManagerRuntime(TGeoManager* x) {
0088
0089
0090 s_geoManager = x;
0091 }
0092
0093
0094 void FWGeometryTableViewManager::setGeoManagerFromFile() {
0095 TFile* file = FWGeometry::findFile(m_fileName.c_str());
0096 fwLog(fwlog::kInfo) << "Geometry table file: " << m_fileName.c_str() << std::endl;
0097 try {
0098 if (!file) {
0099
0100 s_geoManager = TGeoManager::Import(m_fileName.c_str(), m_TGeoName.c_str());
0101 } else {
0102 file->ls();
0103 s_geoManager = (TGeoManager*)file->Get(m_TGeoName.c_str());
0104 }
0105 if (!s_geoManager)
0106 throw std::runtime_error("Can't find TGeoManager object in selected file.");
0107
0108 } catch (std::runtime_error& e) {
0109 fwLog(fwlog::kError) << e.what();
0110 exit(0);
0111 }
0112 }