Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2021-08-17 23:10:45

0001 // -*- C++ -*-
0002 //
0003 // Package:     Core
0004 // Class  :     FWGeometryTableView
0005 //
0006 // Implementation:
0007 //     [Notes on implementation]
0008 //
0009 // Original Author:
0010 //         Created:  Wed Jan  4 00:05:34 CET 2012
0011 //
0012 
0013 // system include files
0014 #include <functional>
0015 
0016 // user include files
0017 #include "Fireworks/Core/src/FWGeometryTableView.h"
0018 #include "Fireworks/Core/src/FWGeoTopNodeScene.h"
0019 #include "Fireworks/Core/interface/FWGeometryTableViewManager.h"
0020 #include "Fireworks/Core/interface/FWViewType.h"
0021 #include "Fireworks/Core/interface/FWGeometryTableManagerBase.h"
0022 #include "Fireworks/Core/interface/CmsShowViewPopup.h"
0023 #include "Fireworks/Core/src/FWGeometryTableManager.h"
0024 #include "Fireworks/Core/interface/fwLog.h"
0025 #include "Fireworks/Core/interface/FWEnumParameter.h"
0026 #include "Fireworks/Core/interface/FWEnumParameterSetter.h"
0027 
0028 #include "Fireworks/Core/src/FWGUIValidatingTextEntry.h"
0029 #include "Fireworks/Core/interface/FWGUIManager.h"
0030 #include "Fireworks/Core/interface/FWValidatorBase.h"
0031 #include "Fireworks/Core/src/FWEveDetectorGeo.h"
0032 
0033 #include "KeySymbols.h"
0034 #include "TGButton.h"
0035 #include "TGComboBox.h"
0036 #include "TGLabel.h"
0037 #include "TGListBox.h"
0038 #include "TGLViewer.h"
0039 #include "TGeoMatrix.h"
0040 #include "TGeoBBox.h"
0041 #include "TVirtualX.h"
0042 #include "TEveViewer.h"
0043 #include "TEveScene.h"
0044 #include "TEveSceneInfo.h"
0045 #include "TEveManager.h"
0046 #include "TGeoManager.h"
0047 #include "TGLCamera.h"
0048 
0049 //==============================================================================
0050 //==============================================================================
0051 //==============================================================================
0052 //==============================================================================
0053 
0054 class FWGeoMaterialValidator : public FWValidatorBase {
0055 public:
0056   FWGeometryTableView* m_browser;
0057   mutable std::vector<const char*> m_list;
0058   FWGeoMaterialValidator(FWGeometryTableView* v) { m_browser = v; }
0059   ~FWGeoMaterialValidator() override {}
0060 
0061   virtual void addDaughtersRec(TGeoVolume* v) const {
0062     switch (m_browser->getFilterType()) {
0063       case FWGeometryTableView::kFilterMaterialName:
0064         m_list.push_back(v->GetMaterial()->GetName());
0065         break;
0066       case FWGeometryTableView::kFilterMaterialTitle:
0067         m_list.push_back(v->GetMaterial()->GetTitle());
0068         break;
0069       case FWGeometryTableView::kFilterShapeName:
0070         m_list.push_back(v->GetShape()->GetName());
0071         break;
0072       case FWGeometryTableView::kFilterShapeClassName:
0073         m_list.push_back(v->GetShape()->ClassName());
0074         break;
0075       default:
0076         std::cerr << "FWGeoMaterialValidator unhandeled case. \n";
0077     }
0078 
0079     for (int i = 0; i < v->GetNdaughters(); ++i)
0080       addDaughtersRec(v->GetNode(i)->GetVolume());
0081   }
0082 
0083   void fillOptions(const char* iBegin,
0084                    const char* iEnd,
0085                    std::vector<std::pair<std::shared_ptr<std::string>, std::string> >& oOptions) const override {
0086     oOptions.clear();
0087     m_list.clear();
0088 
0089     FWGeometryTableManagerBase::Entries_i tnit = m_browser->getTableManager()->refEntries().begin();
0090     std::advance(tnit, m_browser->getTopNodeIdx());
0091     FWGeometryTableViewManager::getGeoMangeur();
0092     addDaughtersRec(tnit->m_node->GetVolume());
0093 
0094     std::sort(m_list.begin(), m_list.end());
0095     std::vector<const char*>::iterator ui = std::unique(m_list.begin(), m_list.end());
0096     m_list.resize(ui - m_list.begin());
0097 
0098     std::string part(iBegin, iEnd);
0099     unsigned int part_size = part.size();
0100     std::string h = "";
0101     // int cnt = 0;
0102     oOptions.push_back(std::make_pair(std::make_shared<std::string>(*m_list.begin()), h));
0103     std::vector<const char*>::iterator startIt = m_list.begin();
0104     startIt++;
0105     for (std::vector<const char*>::iterator i = startIt; i != m_list.end(); ++i) {
0106       //      std::cout << *i << " " << cnt++ << std::endl;
0107       if ((strlen(*i) >= part_size) && strncmp(*i, part.c_str(), part_size) == 0) {
0108         oOptions.push_back(std::make_pair(std::make_shared<std::string>((*i)), &((*i)[part_size])));
0109       }
0110     }
0111   }
0112 
0113   bool isStringValid(std::string& exp) {
0114     std::cerr << "validation not used \n";
0115     return false;
0116   }
0117 };
0118 
0119 //==============================================================================
0120 //==============================================================================
0121 //==============================================================================
0122 //==============================================================================
0123 FWGeometryTableView::FWGeometryTableView(TEveWindowSlot* iParent, FWColorManager* colMng)
0124     : FWGeometryTableViewBase(iParent, FWViewType::kGeometryTable, colMng),
0125       m_tableManager(nullptr),
0126       m_filterEntry(nullptr),
0127       m_filterValidator(nullptr),
0128       m_mode(this, "Mode", 0l, 0l, 1l),
0129       m_disableTopNode(this, "HideTopNode", true),
0130       m_visLevel(this, "VisLevel", 3l, 1l, 100l),
0131       m_filter(this, "Materials", std::string()),
0132       m_filterType(this, "FilterType:", 0l, 0l, 3l),
0133       m_visLevelFilter(this, "IgnoreVisLevelOnFilter", true),
0134       m_selectRegion(this, "SelectNearCameraCenter", false),
0135       m_regionRadius(this, "SphereRadius", 10.0, 1.0, 300.0),
0136       m_proximityAlgo(this, "Proximity algorithm", 1l, 0l, 1l) {
0137   FWGeoTopNodeGLScene* gls = new FWGeoTopNodeGLScene(nullptr);
0138 #if ROOT_VERSION_CODE < ROOT_VERSION(5, 32, 0)
0139   m_eveScene = new FWGeoTopNodeEveScene(gls, "TopGeoNodeScene", "");
0140 #else
0141   m_eveScene = new TEveScene(gls, "TopGeoNodeScene", "");
0142 #endif
0143   gEve->GetScenes()->AddElement(m_eveScene);
0144 
0145   m_eveTopNode = new FWEveDetectorGeo(this);
0146   m_eveTopNode->IncDenyDestroy();
0147   m_eveTopNode->SetPickable(true);
0148   m_eveScene->AddElement(m_eveTopNode);
0149 
0150   gls->m_eveTopNode = m_eveTopNode;
0151   m_eveTopNode->m_scene = gls;
0152 
0153   // top row
0154   TGHorizontalFrame* hp = new TGHorizontalFrame(m_frame);
0155   {
0156     TGTextButton* rb = new TGTextButton(hp, "CdTop");
0157     hp->AddFrame(rb, new TGLayoutHints(kLHintsNormal, 2, 2, 0, 0));
0158     rb->Connect("Clicked()", "FWGeometryTableViewBase", this, "cdTop()");
0159   }
0160   {
0161     TGTextButton* rb = new TGTextButton(hp, "CdUp");
0162     hp->AddFrame(rb, new TGLayoutHints(kLHintsNormal, 2, 2, 0, 0));
0163     rb->Connect("Clicked()", "FWGeometryTableViewBase", this, "cdUp()");
0164   }
0165   {
0166     m_viewBox = new FWViewCombo(hp, this);
0167     hp->AddFrame(m_viewBox, new TGLayoutHints(kLHintsExpandY, 2, 2, 0, 0));
0168   }
0169   {
0170     m_filterType.addEntry(kFilterMaterialName, "MaterialName");
0171     m_filterType.addEntry(kFilterMaterialTitle, "MaterialTitle");
0172     m_filterType.addEntry(kFilterShapeName, "ShapeName");
0173     m_filterType.addEntry(kFilterShapeClassName, "ShapeClassName");
0174 
0175     std::shared_ptr<FWParameterSetterBase> ptr(FWParameterSetterBase::makeSetterFor((FWParameterBase*)&m_filterType));
0176     ptr->attach((FWParameterBase*)&m_filterType, this);
0177 
0178     TGFrame* pframe = ptr->build(hp, false);
0179     hp->AddFrame(pframe, new TGLayoutHints(kLHintsLeft));
0180     m_filterTypeSetter.swap(ptr);
0181   }
0182   {
0183     hp->AddFrame(new TGLabel(hp, "FilterExp:"), new TGLayoutHints(kLHintsBottom, 0, 0, 0, 2));
0184     m_filterEntry = new FWGUIValidatingTextEntry(hp);
0185     m_filterEntry->SetHeight(20);
0186     m_filterValidator = new FWGeoMaterialValidator(this);
0187     m_filterEntry->setValidator(m_filterValidator);
0188     hp->AddFrame(m_filterEntry, new TGLayoutHints(kLHintsExpandX, 0, 2, 1, 0));
0189     m_filterEntry->setMaxListBoxHeight(150);
0190     m_filterEntry->getListBox()->Connect("Selected(int)", "FWGeometryTableView", this, "filterListCallback()");
0191     m_filterEntry->Connect("ReturnPressed()", "FWGeometryTableView", this, "filterTextEntryCallback()");
0192 
0193     gVirtualX->GrabKey(m_filterEntry->GetId(), gVirtualX->KeysymToKeycode((int)kKey_A), kKeyControlMask, true);
0194   }
0195   m_frame->AddFrame(hp, new TGLayoutHints(kLHintsLeft | kLHintsExpandX, 2, 2, 2, 0));
0196 
0197   m_tableManager = new FWGeometryTableManager(this);
0198   {
0199     TEveGeoManagerHolder gmgr(FWGeometryTableViewManager::getGeoMangeur());
0200     m_tableManager->loadGeometry(gGeoManager->GetTopNode(), gGeoManager->GetListOfVolumes());
0201   }
0202   cdTop();
0203 
0204   m_mode.addEntry(kNode, "Node");
0205   m_mode.addEntry(kVolume, "Volume");
0206 
0207   m_mode.changed_.connect(std::bind(&FWGeometryTableView::refreshTable3D, this));
0208   m_autoExpand.changed_.connect(std::bind(&FWGeometryTableView::autoExpandCallback, this));
0209   m_visLevel.changed_.connect(std::bind(&FWGeometryTableView::refreshTable3D, this));
0210 
0211   m_visLevelFilter.changed_.connect(std::bind(&FWGeometryTableView::refreshTable3D, this));
0212 
0213   m_disableTopNode.changed_.connect(std::bind(&FWGeometryTableView::updateVisibilityTopNode, this));
0214   postConst();
0215 
0216   m_proximityAlgo.addEntry(kBBoxCenter, "BBox center");
0217   m_proximityAlgo.addEntry(kBBoxSurface, "BBox surface");
0218 
0219   m_selectRegion.changed_.connect(std::bind(&FWGeometryTableView::checkRegionOfInterest, this));
0220   m_regionRadius.changed_.connect(std::bind(&FWGeometryTableView::checkRegionOfInterest, this));
0221   m_proximityAlgo.changed_.connect(std::bind(&FWGeometryTableView::checkRegionOfInterest, this));
0222 }
0223 
0224 FWGeometryTableView::~FWGeometryTableView() {}
0225 
0226 //______________________________________________________________________________
0227 void FWGeometryTableView::setPath(int parentIdx, std::string&) {
0228   m_eveTopNode->clearSelection();
0229 
0230   m_topNodeIdx.set(parentIdx);
0231   getTableManager()
0232       ->refEntries()
0233       .at(getTopNodeIdx())
0234       .setBitVal(FWGeometryTableManagerBase::kVisNodeSelf, !m_disableTopNode.value());
0235   getTableManager()->setLevelOffset(getTableManager()->refEntries().at(getTopNodeIdx()).m_level);
0236 
0237   checkExpandLevel();
0238   refreshTable3D();
0239 }
0240 
0241 //______________________________________________________________________________
0242 FWGeometryTableManagerBase* FWGeometryTableView::getTableManager() { return m_tableManager; }
0243 
0244 //______________________________________________________________________________
0245 void FWGeometryTableView::autoExpandCallback() {
0246   if (!m_enableRedraw)
0247     return;
0248   checkExpandLevel();
0249   getTableManager()->redrawTable(true);
0250 }
0251 
0252 //______________________________________________________________________________
0253 void FWGeometryTableView::filterTextEntryCallback() {
0254   // std::cout << "text entry click ed \n" ;
0255   std::string exp = m_filterEntry->GetText();
0256   updateFilter(exp);
0257 }
0258 
0259 //______________________________________________________________________________
0260 void FWGeometryTableView::filterListCallback() {
0261   // std::cout << "list click ed [" << m_filterEntry->GetText() << "] \n" ;
0262 
0263   std::string exp = m_filterEntry->GetText();
0264   updateFilter(exp);
0265 }
0266 
0267 //______________________________________________________________________________
0268 void FWGeometryTableView::updateFilter(std::string& exp) {
0269   // std::cout << "=FWGeometryTableViewBase::updateFilter()" << m_filterEntry->GetText() <<std::endl;
0270 
0271   if (exp.empty()) {
0272     // std::cout << "FITLER OFF \n";
0273     for (FWGeometryTableManagerBase::Entries_i i = m_tableManager->refEntries().begin();
0274          i != m_tableManager->refEntries().end();
0275          ++i) {
0276       m_tableManager->setVisibility(*i, true);
0277       m_tableManager->setVisibilityChld(*i, true);
0278     }
0279 
0280     // NOTE: entry should be cleared automatically
0281     m_filterEntry->Clear();
0282   }
0283 
0284   m_filter.set(exp);
0285   m_tableManager->updateFilter(m_filterType.value());
0286   refreshTable3D();
0287 }
0288 
0289 //==============================================================================
0290 
0291 void FWGeometryTableView::populateController(ViewerParameterGUI& gui) const {
0292   gui.requestTab("Style")
0293       .addParam(&m_mode)
0294       .addParam(&m_autoExpand)
0295       .separator()
0296       .addParam(&m_disableTopNode)
0297       .addParam(&m_minParentTransparency)
0298       .addParam(&m_minLeafTransparency)
0299       .addParam(&m_visLevel)
0300       .addParam(&m_visLevelFilter)
0301       .separator()
0302       .addParam(&m_selectRegion)
0303       .addParam(&m_regionRadius)
0304       .addParam(&m_proximityAlgo)
0305       .separator();
0306 
0307   TGTextButton* butt = new TGTextButton(gui.getTabContainer(), "ReloadColors");
0308   gui.getTabContainer()->AddFrame(butt);
0309   butt->Connect("Clicked()", "FWGeometryTableViewBase", (FWGeometryTableViewBase*)this, "reloadColors()");
0310 }
0311 
0312 //------------------------------------------------------------------------------
0313 
0314 /*
0315 void FWGeometryTableView::setPath(int parentIdx, std::string& path)
0316 {
0317    //   printf("Set Path to [%s], current node \n", path.c_str());
0318    m_topNodeIdx.set(parentIdx);
0319    getTableManager()->refEntries().at(getTopNodeIdx()).setBitVal(FWGeometryTableManagerBase::kVisNodeSelf,!m_disableTopNode.value() );
0320    getTableManager()->setLevelOffset(getTableManager()->refEntries().at(getTopNodeIdx()).m_level);
0321 
0322    m_eveTopNode->clearSelection(); 
0323 
0324    checkExpandLevel();
0325    refreshTable3D(); 
0326    FWGUIManager::getGUIManager()->updateStatus(path.c_str());
0327    }*/
0328 
0329 //--------------------------------------------------------------
0330 bool viewIsChecked(TEveViewer* v, TEveElement* el) {
0331   if (strstr(v->GetElementName(), "3D")) {
0332     for (TEveElement::List_i eit = v->BeginChildren(); eit != v->EndChildren(); ++eit) {
0333       TEveScene* s = ((TEveSceneInfo*)*eit)->GetScene();
0334       if (el && s->HasChildren() && s->FirstChild() == el)
0335         return true;
0336     }
0337   }
0338   return false;
0339 }
0340 
0341 void FWGeometryTableView::checkRegionOfInterest() {
0342   if (m_selectRegion.value()) {
0343     double* center = nullptr;
0344     for (TEveElement::List_i it = gEve->GetViewers()->BeginChildren(); it != gEve->GetViewers()->EndChildren(); ++it) {
0345       TEveViewer* v = ((TEveViewer*)(*it));
0346       if (viewIsChecked(v, m_eveTopNode)) {
0347         if (center) {
0348           fwLog(fwlog::kWarning) << "Center picked from first view \n";
0349         } else {
0350           center = v->GetGLViewer()->CurrentCamera().GetCenterVec();
0351           fwLog(fwlog::kInfo) << Form(
0352               "Center picked (%.1f, %.1f, %.1f) from first selected 3D view \n", center[0], center[1], center[2]);
0353         }
0354       }
0355     }
0356 
0357     if (!center) {
0358       fwLog(fwlog::kError) << "No 3D view selected \n";
0359       return;
0360     }
0361 
0362     m_tableManager->checkRegionOfInterest(center, m_regionRadius.value(), m_proximityAlgo.value());
0363   } else {
0364     m_tableManager->resetRegionOfInterest();
0365   }
0366 
0367   refreshTable3D();
0368 }
0369 
0370 //------------------------------------------------------------------------------
0371 
0372 void FWGeometryTableView::setFrom(const FWConfiguration& iFrom) {
0373   m_enableRedraw = false;
0374   for (const_iterator it = begin(), itEnd = end(); it != itEnd; ++it) {
0375     // printf("set from %s \n",(*it)->name().c_str() );
0376     if ((*it)->name() == m_topNodeIdx.name())
0377       setTopNodePathFromConfig(iFrom);
0378     else
0379       (*it)->setFrom(iFrom);
0380   }
0381 
0382   cdNode(m_topNodeIdx.value());
0383 
0384   TGComboBox* cbox = ((FWEnumParameterSetter*)m_filterTypeSetter.get())->getWidget();
0385   cbox->Select(m_filterType.value(), false);
0386   m_viewersConfig = iFrom.valueForKey("Viewers");
0387 
0388   m_enableRedraw = true;
0389   m_filterEntry->SetText(m_filter.value().c_str(), false);
0390   m_tableManager->updateFilter(m_filterType.value());
0391   checkExpandLevel();
0392   refreshTable3D();
0393   /*
0394      getTableManager()->redrawTable();
0395      m_eveTopNode->ElementChanged();
0396      gEve->FullRedraw3D(false, true);
0397    */
0398 }
0399 
0400 //------------------------------------------------------------------------------
0401 
0402 void FWGeometryTableView::updateVisibilityTopNode() {
0403   getTableManager()
0404       ->refEntries()
0405       .at(getTopNodeIdx())
0406       .setBitVal(FWGeometryTableManagerBase::kVisNodeSelf, !m_disableTopNode.value());
0407   refreshTable3D();
0408 }