Core/src/FWGeometryTableViewManager.cc

0001 // -*- C++ -*-
0002 //
0003 // Package:     Core
0004 // Class  :     FWGeometryTableViewManager
0005 //
0006 // Implementation:
0007 //     [Notes on implementation]
0008 //
0009 // Original Author:  Alja Mrak-Tadel
0010 //         Created:  Fri Jul  8 00:40:37 CEST 2011
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   // Function used in geometry table views.
0081
0082   assert(s_geoManager);
0083   return s_geoManager;
0084 }
0085
0086 //______________________________________________________________________________
0087 void FWGeometryTableViewManager::setGeoManagerRuntime(TGeoManager* x) {
0088   // Function called from FWFFLooper to set geometry created in runtime.
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       // Try it as a GDML file
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 }