Fireworks/Core/src/FWGeometryTableView.cc - master - githubmirror/cmssw

Fireworks/Core/src/FWGeometryTableView.cc
408 строк15 KB
Shahzad Malik Muzaffar
[Fireworks] Move public headers in to interface directory
16 авг 2021, 19:47
16 авг 2021, 19:4751ca82e
// -*- C++ -*-
//
// Package:     Core
// Class  :     FWGeometryTableView
//
// Implementation:
//     [Notes on implementation]
//
// Original Author:
//         Created:  Wed Jan  4 00:05:34 CET 2012
//

// system include files
#include <functional>

// user include files
#include "Fireworks/Core/src/FWGeometryTableView.h"
#include "Fireworks/Core/src/FWGeoTopNodeScene.h"
#include "Fireworks/Core/interface/FWGeometryTableViewManager.h"
#include "Fireworks/Core/interface/FWViewType.h"
#include "Fireworks/Core/interface/FWGeometryTableManagerBase.h"
#include "Fireworks/Core/interface/CmsShowViewPopup.h"
#include "Fireworks/Core/src/FWGeometryTableManager.h"
#include "Fireworks/Core/interface/fwLog.h"
#include "Fireworks/Core/interface/FWEnumParameter.h"
#include "Fireworks/Core/interface/FWEnumParameterSetter.h"

#include "Fireworks/Core/src/FWGUIValidatingTextEntry.h"
#include "Fireworks/Core/interface/FWGUIManager.h"
#include "Fireworks/Core/interface/FWValidatorBase.h"
#include "Fireworks/Core/src/FWEveDetectorGeo.h"

#include "KeySymbols.h"
#include "TGButton.h"
#include "TGComboBox.h"
#include "TGLabel.h"
#include "TGListBox.h"
#include "TGLViewer.h"
#include "TGeoMatrix.h"
#include "TGeoBBox.h"
#include "TVirtualX.h"
#include "TEveViewer.h"
#include "TEveScene.h"
#include "TEveSceneInfo.h"
#include "TEveManager.h"
#include "TGeoManager.h"
#include "TGLCamera.h"

//==============================================================================
//==============================================================================
//==============================================================================
//==============================================================================

class FWGeoMaterialValidator : public FWValidatorBase {
public:
  FWGeometryTableView* m_browser;
  mutable std::vector<const char*> m_list;
  FWGeoMaterialValidator(FWGeometryTableView* v) { m_browser = v; }
  ~FWGeoMaterialValidator() override {}

  virtual void addDaughtersRec(TGeoVolume* v) const {
    switch (m_browser->getFilterType()) {
      case FWGeometryTableView::kFilterMaterialName:
        m_list.push_back(v->GetMaterial()->GetName());
        break;
      case FWGeometryTableView::kFilterMaterialTitle:
        m_list.push_back(v->GetMaterial()->GetTitle());
        break;
      case FWGeometryTableView::kFilterShapeName:
        m_list.push_back(v->GetShape()->GetName());
        break;
      case FWGeometryTableView::kFilterShapeClassName:
        m_list.push_back(v->GetShape()->ClassName());
        break;
      default:
        std::cerr << "FWGeoMaterialValidator unhandeled case. \n";
    }

    for (int i = 0; i < v->GetNdaughters(); ++i)
      addDaughtersRec(v->GetNode(i)->GetVolume());
  }

  void fillOptions(const char* iBegin,
                   const char* iEnd,
                   std::vector<std::pair<std::shared_ptr<std::string>, std::string> >& oOptions) const override {
    oOptions.clear();
    m_list.clear();

    FWGeometryTableManagerBase::Entries_i tnit = m_browser->getTableManager()->refEntries().begin();
    std::advance(tnit, m_browser->getTopNodeIdx());
    FWGeometryTableViewManager::getGeoMangeur();
    addDaughtersRec(tnit->m_node->GetVolume());

    std::sort(m_list.begin(), m_list.end());
    std::vector<const char*>::iterator ui = std::unique(m_list.begin(), m_list.end());
    m_list.resize(ui - m_list.begin());

    std::string part(iBegin, iEnd);
    unsigned int part_size = part.size();
    std::string h = "";
    // int cnt = 0;
    oOptions.push_back(std::make_pair(std::make_shared<std::string>(*m_list.begin()), h));
    std::vector<const char*>::iterator startIt = m_list.begin();
    startIt++;
    for (std::vector<const char*>::iterator i = startIt; i != m_list.end(); ++i) {
      //      std::cout << *i << " " << cnt++ << std::endl;
      if ((strlen(*i) >= part_size) && strncmp(*i, part.c_str(), part_size) == 0) {
        oOptions.push_back(std::make_pair(std::make_shared<std::string>((*i)), &((*i)[part_size])));
      }
    }
  }

  bool isStringValid(std::string& exp) {
    std::cerr << "validation not used \n";
    return false;
  }
};

//==============================================================================
//==============================================================================
//==============================================================================
//==============================================================================
FWGeometryTableView::FWGeometryTableView(TEveWindowSlot* iParent, FWColorManager* colMng)
    : FWGeometryTableViewBase(iParent, FWViewType::kGeometryTable, colMng),
      m_tableManager(nullptr),
      m_filterEntry(nullptr),
      m_filterValidator(nullptr),
      m_mode(this, "Mode", 0l, 0l, 1l),
      m_disableTopNode(this, "HideTopNode", true),
      m_visLevel(this, "VisLevel", 3l, 1l, 100l),
      m_filter(this, "Materials", std::string()),
      m_filterType(this, "FilterType:", 0l, 0l, 3l),
      m_visLevelFilter(this, "IgnoreVisLevelOnFilter", true),
      m_selectRegion(this, "SelectNearCameraCenter", false),
      m_regionRadius(this, "SphereRadius", 10.0, 1.0, 300.0),
      m_proximityAlgo(this, "Proximity algorithm", 1l, 0l, 1l) {
  FWGeoTopNodeGLScene* gls = new FWGeoTopNodeGLScene(nullptr);
#if ROOT_VERSION_CODE < ROOT_VERSION(5, 32, 0)
  m_eveScene = new FWGeoTopNodeEveScene(gls, "TopGeoNodeScene", "");
#else
  m_eveScene = new TEveScene(gls, "TopGeoNodeScene", "");
#endif
  gEve->GetScenes()->AddElement(m_eveScene);

  m_eveTopNode = new FWEveDetectorGeo(this);
  m_eveTopNode->IncDenyDestroy();
  m_eveTopNode->SetPickable(true);
  m_eveScene->AddElement(m_eveTopNode);

  gls->m_eveTopNode = m_eveTopNode;
  m_eveTopNode->m_scene = gls;

  // top row
  TGHorizontalFrame* hp = new TGHorizontalFrame(m_frame);
  {
    TGTextButton* rb = new TGTextButton(hp, "CdTop");
    hp->AddFrame(rb, new TGLayoutHints(kLHintsNormal, 2, 2, 0, 0));
    rb->Connect("Clicked()", "FWGeometryTableViewBase", this, "cdTop()");
  }
  {
    TGTextButton* rb = new TGTextButton(hp, "CdUp");
    hp->AddFrame(rb, new TGLayoutHints(kLHintsNormal, 2, 2, 0, 0));
    rb->Connect("Clicked()", "FWGeometryTableViewBase", this, "cdUp()");
  }
  {
    m_viewBox = new FWViewCombo(hp, this);
    hp->AddFrame(m_viewBox, new TGLayoutHints(kLHintsExpandY, 2, 2, 0, 0));
  }
  {
    m_filterType.addEntry(kFilterMaterialName, "MaterialName");
    m_filterType.addEntry(kFilterMaterialTitle, "MaterialTitle");
    m_filterType.addEntry(kFilterShapeName, "ShapeName");
    m_filterType.addEntry(kFilterShapeClassName, "ShapeClassName");

    std::shared_ptr<FWParameterSetterBase> ptr(FWParameterSetterBase::makeSetterFor((FWParameterBase*)&m_filterType));
    ptr->attach((FWParameterBase*)&m_filterType, this);

    TGFrame* pframe = ptr->build(hp, false);
    hp->AddFrame(pframe, new TGLayoutHints(kLHintsLeft));
    m_filterTypeSetter.swap(ptr);
  }
  {
    hp->AddFrame(new TGLabel(hp, "FilterExp:"), new TGLayoutHints(kLHintsBottom, 0, 0, 0, 2));
    m_filterEntry = new FWGUIValidatingTextEntry(hp);
    m_filterEntry->SetHeight(20);
    m_filterValidator = new FWGeoMaterialValidator(this);
    m_filterEntry->setValidator(m_filterValidator);
    hp->AddFrame(m_filterEntry, new TGLayoutHints(kLHintsExpandX, 0, 2, 1, 0));
    m_filterEntry->setMaxListBoxHeight(150);
    m_filterEntry->getListBox()->Connect("Selected(int)", "FWGeometryTableView", this, "filterListCallback()");
    m_filterEntry->Connect("ReturnPressed()", "FWGeometryTableView", this, "filterTextEntryCallback()");

    gVirtualX->GrabKey(m_filterEntry->GetId(), gVirtualX->KeysymToKeycode((int)kKey_A), kKeyControlMask, true);
  }
  m_frame->AddFrame(hp, new TGLayoutHints(kLHintsLeft | kLHintsExpandX, 2, 2, 2, 0));

  m_tableManager = new FWGeometryTableManager(this);
  {
    TEveGeoManagerHolder gmgr(FWGeometryTableViewManager::getGeoMangeur());
    m_tableManager->loadGeometry(gGeoManager->GetTopNode(), gGeoManager->GetListOfVolumes());
  }
  cdTop();

  m_mode.addEntry(kNode, "Node");
  m_mode.addEntry(kVolume, "Volume");

  m_mode.changed_.connect(std::bind(&FWGeometryTableView::refreshTable3D, this));
  m_autoExpand.changed_.connect(std::bind(&FWGeometryTableView::autoExpandCallback, this));
  m_visLevel.changed_.connect(std::bind(&FWGeometryTableView::refreshTable3D, this));

  m_visLevelFilter.changed_.connect(std::bind(&FWGeometryTableView::refreshTable3D, this));

  m_disableTopNode.changed_.connect(std::bind(&FWGeometryTableView::updateVisibilityTopNode, this));
  postConst();

  m_proximityAlgo.addEntry(kBBoxCenter, "BBox center");
  m_proximityAlgo.addEntry(kBBoxSurface, "BBox surface");

  m_selectRegion.changed_.connect(std::bind(&FWGeometryTableView::checkRegionOfInterest, this));
  m_regionRadius.changed_.connect(std::bind(&FWGeometryTableView::checkRegionOfInterest, this));
  m_proximityAlgo.changed_.connect(std::bind(&FWGeometryTableView::checkRegionOfInterest, this));
}

FWGeometryTableView::~FWGeometryTableView() {}

//______________________________________________________________________________
void FWGeometryTableView::setPath(int parentIdx, std::string&) {
  m_eveTopNode->clearSelection();

  m_topNodeIdx.set(parentIdx);
  getTableManager()
      ->refEntries()
      .at(getTopNodeIdx())
      .setBitVal(FWGeometryTableManagerBase::kVisNodeSelf, !m_disableTopNode.value());
  getTableManager()->setLevelOffset(getTableManager()->refEntries().at(getTopNodeIdx()).m_level);

  checkExpandLevel();
  refreshTable3D();
}

//______________________________________________________________________________
FWGeometryTableManagerBase* FWGeometryTableView::getTableManager() { return m_tableManager; }

//______________________________________________________________________________
void FWGeometryTableView::autoExpandCallback() {
  if (!m_enableRedraw)
    return;
  checkExpandLevel();
  getTableManager()->redrawTable(true);
}

//______________________________________________________________________________
void FWGeometryTableView::filterTextEntryCallback() {
  // std::cout << "text entry click ed \n" ;
  std::string exp = m_filterEntry->GetText();
  updateFilter(exp);
}

//______________________________________________________________________________
void FWGeometryTableView::filterListCallback() {
  // std::cout << "list click ed [" << m_filterEntry->GetText() << "] \n" ;

  std::string exp = m_filterEntry->GetText();
  updateFilter(exp);
}

//______________________________________________________________________________
void FWGeometryTableView::updateFilter(std::string& exp) {
  // std::cout << "=FWGeometryTableViewBase::updateFilter()" << m_filterEntry->GetText() <<std::endl;

  if (exp.empty()) {
    // std::cout << "FITLER OFF \n";
    for (FWGeometryTableManagerBase::Entries_i i = m_tableManager->refEntries().begin();
         i != m_tableManager->refEntries().end();
         ++i) {
      m_tableManager->setVisibility(*i, true);
      m_tableManager->setVisibilityChld(*i, true);
    }

    // NOTE: entry should be cleared automatically
    m_filterEntry->Clear();
  }

  m_filter.set(exp);
  m_tableManager->updateFilter(m_filterType.value());
  refreshTable3D();
}

//==============================================================================

void FWGeometryTableView::populateController(ViewerParameterGUI& gui) const {
  gui.requestTab("Style")
      .addParam(&m_mode)
      .addParam(&m_autoExpand)
      .separator()
      .addParam(&m_disableTopNode)
      .addParam(&m_minParentTransparency)
      .addParam(&m_minLeafTransparency)
      .addParam(&m_visLevel)
      .addParam(&m_visLevelFilter)
      .separator()
      .addParam(&m_selectRegion)
      .addParam(&m_regionRadius)
      .addParam(&m_proximityAlgo)
      .separator();

  TGTextButton* butt = new TGTextButton(gui.getTabContainer(), "ReloadColors");
  gui.getTabContainer()->AddFrame(butt);
  butt->Connect("Clicked()", "FWGeometryTableViewBase", (FWGeometryTableViewBase*)this, "reloadColors()");
}

//------------------------------------------------------------------------------

/*
void FWGeometryTableView::setPath(int parentIdx, std::string& path)
{
   //   printf("Set Path to [%s], current node \n", path.c_str());
   m_topNodeIdx.set(parentIdx);
   getTableManager()->refEntries().at(getTopNodeIdx()).setBitVal(FWGeometryTableManagerBase::kVisNodeSelf,!m_disableTopNode.value() );
   getTableManager()->setLevelOffset(getTableManager()->refEntries().at(getTopNodeIdx()).m_level);

   m_eveTopNode->clearSelection(); 

   checkExpandLevel();
   refreshTable3D(); 
   FWGUIManager::getGUIManager()->updateStatus(path.c_str());
   }*/

//--------------------------------------------------------------
bool viewIsChecked(TEveViewer* v, TEveElement* el) {
  if (strstr(v->GetElementName(), "3D")) {
    for (TEveElement::List_i eit = v->BeginChildren(); eit != v->EndChildren(); ++eit) {
      TEveScene* s = ((TEveSceneInfo*)*eit)->GetScene();
      if (el && s->HasChildren() && s->FirstChild() == el)
        return true;
    }
  }
  return false;
}

void FWGeometryTableView::checkRegionOfInterest() {
  if (m_selectRegion.value()) {
    double* center = nullptr;
    for (TEveElement::List_i it = gEve->GetViewers()->BeginChildren(); it != gEve->GetViewers()->EndChildren(); ++it) {
      TEveViewer* v = ((TEveViewer*)(*it));
      if (viewIsChecked(v, m_eveTopNode)) {
        if (center) {
          fwLog(fwlog::kWarning) << "Center picked from first view \n";
        } else {
          center = v->GetGLViewer()->CurrentCamera().GetCenterVec();
          fwLog(fwlog::kInfo) << Form(
              "Center picked (%.1f, %.1f, %.1f) from first selected 3D view \n", center[0], center[1], center[2]);
        }
      }
    }

    if (!center) {
      fwLog(fwlog::kError) << "No 3D view selected \n";
      return;
    }

    m_tableManager->checkRegionOfInterest(center, m_regionRadius.value(), m_proximityAlgo.value());
  } else {
    m_tableManager->resetRegionOfInterest();
  }

  refreshTable3D();
}

//------------------------------------------------------------------------------

void FWGeometryTableView::setFrom(const FWConfiguration& iFrom) {
  m_enableRedraw = false;
  for (const_iterator it = begin(), itEnd = end(); it != itEnd; ++it) {
    // printf("set from %s \n",(*it)->name().c_str() );
    if ((*it)->name() == m_topNodeIdx.name())
      setTopNodePathFromConfig(iFrom);
    else
      (*it)->setFrom(iFrom);
  }

  cdNode(m_topNodeIdx.value());

  TGComboBox* cbox = ((FWEnumParameterSetter*)m_filterTypeSetter.get())->getWidget();
  cbox->Select(m_filterType.value(), false);
  m_viewersConfig = iFrom.valueForKey("Viewers");

  m_enableRedraw = true;
  m_filterEntry->SetText(m_filter.value().c_str(), false);
  m_tableManager->updateFilter(m_filterType.value());
  checkExpandLevel();
  refreshTable3D();
  /*
     getTableManager()->redrawTable();
     m_eveTopNode->ElementChanged();
     gEve->FullRedraw3D(false, true);
   */
}

//------------------------------------------------------------------------------

void FWGeometryTableView::updateVisibilityTopNode() {
  getTableManager()
      ->refEntries()
      .at(getTopNodeIdx())
      .setBitVal(FWGeometryTableManagerBase::kVisNodeSelf, !m_disableTopNode.value());
  refreshTable3D();
}