plugin for hiding all the elements except a layer around some given elements....

plugin for hiding all the elements except a layer around some given elements. To be integrated to the visibility window instead?

Plugin/CMakeLists.txt

@@ -36,6 +36,7 @@ set(SRC
   FaultZone.cpp
   MeshSubEntities.cpp
   CVTRemesh.cpp
+  ShowNeighborElements.cpp
 )
 
 file(GLOB HDR RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} *.h)

Plugin/PluginManager.cpp

@@ -65,6 +65,7 @@
 #include "FaultZone.h"
 #include "MeshSubEntities.h"
 #include "CVTRemesh.h"
+#include "ShowNeighborElements.h"
 
 // for testing purposes only :-)
 #undef HAVE_DLOPEN
@@ -261,6 +262,8 @@ void PluginManager::registerDefaultPlugins()
                       ("DuplicateBoundaries", GMSH_RegisterDuplicateBoundariesPlugin()));
     allPlugins.insert(std::pair<std::string, GMSH_Plugin*>
                       ("ThinLayerFixMesh", GMSH_RegisterThinLayerFixMeshPlugin()));
+    allPlugins.insert(std::pair<std::string, GMSH_Plugin*>
+                      ("ShowNeighborElements", GMSH_RegisterShowNeighborElementsPlugin()));
     allPlugins.insert(std::pair<std::string, GMSH_Plugin*>
                       ("MeshSubEntities", GMSH_RegisterMeshSubEntitiesPlugin()));
 #if defined(HAVE_MESH)

Plugin/ShowNeighborElements.cpp

+// Gmsh - Copyright (C) 1997-2016 C. Geuzaine, J.-F. Remacle
+//
+// See the LICENSE.txt file for license information. Please report all
+// bugs and problems to the public mailing list <gmsh@onelab.info>.
+
+#include "ShowNeighborElements.h"
+#include "GModel.h"
+#include "MElement.h"
+#include "MVertex.h"
+#include "Options.h"
+#include "Context.h"
+#include "GmshDefines.h"
+#if defined(HAVE_OPENGL)
+#include "drawContext.h"
+#endif
+
+StringXNumber ShowNeighborElementsOptions_Number[] = {
+  {GMSH_FULLRC, "NumLayers", NULL, 1},
+  {GMSH_FULLRC, "Element1", NULL, 0},
+  {GMSH_FULLRC, "Element2", NULL, 0},
+  {GMSH_FULLRC, "Element3", NULL, 0},
+  {GMSH_FULLRC, "Element4", NULL, 0},
+  {GMSH_FULLRC, "Element5", NULL, 0}
+};
+extern "C"
+{
+  GMSH_Plugin *GMSH_RegisterShowNeighborElementsPlugin()
+  {
+    return new GMSH_ShowNeighborElementsPlugin();
+  }
+}
+int GMSH_ShowNeighborElementsPlugin::getNbOptions() const
+{
+  return sizeof(ShowNeighborElementsOptions_Number) / sizeof(StringXNumber);
+}
+StringXNumber *GMSH_ShowNeighborElementsPlugin::getOption(int iopt)
+{
+  return &ShowNeighborElementsOptions_Number[iopt];
+}
+std::string GMSH_ShowNeighborElementsPlugin::getHelp() const
+{
+  return "Plugin(ShowNeighborElements) allows to set visible some given elements "
+      " and a layer of elements around them, the other being set invisible.";
+}
+
+// Execution
+PView* GMSH_ShowNeighborElementsPlugin::execute(PView *v)
+{
+  GModel *m = GModel::current();
+
+  _nLayers = static_cast<int>(ShowNeighborElementsOptions_Number[0].def);
+  _nel1 = static_cast<int>(ShowNeighborElementsOptions_Number[1].def);
+  _nel2 = static_cast<int>(ShowNeighborElementsOptions_Number[2].def);
+  _nel3 = static_cast<int>(ShowNeighborElementsOptions_Number[3].def);
+  _nel4 = static_cast<int>(ShowNeighborElementsOptions_Number[4].def);
+  _nel5 = static_cast<int>(ShowNeighborElementsOptions_Number[5].def);
+
+  for (GModel::fiter it = m->firstFace(); it != m->lastFace(); it++) {
+    GFace *f = *it;
+    _init(f);
+    _showLayers(f, _nLayers);
+  }
+
+  for (GModel::riter it = m->firstRegion(); it != m->lastRegion(); it++) {
+    GRegion *r = *it;
+    _init(r);
+    _showLayers(r, _nLayers);
+  }
+
+#if defined(HAVE_OPENGL)
+  CTX::instance()->mesh.changed = (ENT_ALL);
+  drawContext::global()->draw();
+#endif
+
+  return NULL;
+}
+
+void GMSH_ShowNeighborElementsPlugin::_init(GEntity *ent)
+{
+  _vert2elem.clear();
+  for (unsigned i = 0; i < ent->getNumMeshElements(); ++i) {
+    MElement *el = ent->getMeshElement(i);
+    if (el->getNum() == _nel1 ||
+        el->getNum() == _nel2 ||
+        el->getNum() == _nel3 ||
+        el->getNum() == _nel4 ||
+        el->getNum() == _nel5) {
+      el->setVisibility(true);
+      for (int k = 0; k < el->getNumVertices(); ++k) { // TODO only corner vertices?
+        _vertices.insert(el->getVertex(k));
+      }
+    }
+    else {
+      el->setVisibility(false);
+      for (int k = 0; k < el->getNumVertices(); ++k) { // TODO only corner vertices?
+        _vert2elem.insert(std::make_pair(el->getVertex(k), el));
+      }
+    }
+  }
+}
+
+void GMSH_ShowNeighborElementsPlugin::_showLayers(GEntity *ent, int nLayer)
+{
+//  std::pair<std::multimap<MVertex*, MElement*>::iterator,
+//            std::multimap<MVertex*, MElement*>::iterator> lowUpBound;
+
+  if (_vertices.empty() || nLayer < 1) return;
+
+  std::set<MVertex*> &vert = _vertices;
+//  _vertices.clear();
+  std::map<MElement*, int> el2cnt;
+
+  std::set<MVertex*>::iterator it;
+  for (it = vert.begin(); it != vert.end(); ++it) {
+    MVertex *v = *it;
+    std::multimap<MVertex*, MElement*>::iterator ite, itstop;
+    ite = _vert2elem.lower_bound(v);
+    itstop = _vert2elem.upper_bound(v);
+    for (; ite != itstop; ++ite) {
+      MElement *el = ite->second;
+      if (el2cnt.find(el) == el2cnt.end()) el2cnt[el] = 0;
+      ++el2cnt[el];
+//      el->setVisibility(true);
+//      for (int k = 0; k < el->getNumVertices(); ++k) { // TODO only corner vertices?
+//        _vertices.insert(el->getVertex(k));
+//      }
+    }
+  }
+
+  std::map<MElement*, int>::iterator it2;
+  for (it2 = el2cnt.begin(); it2 != el2cnt.end(); ++it2) {
+    if (it2->second && it2->second > 3-nLayer) {
+      it2->first->setVisibility(true);
+    }
+  }
+
+  //_showLayers(ent, nLayer-1);
+}

Plugin/ShowNeighborElements.h

+// Gmsh - Copyright (C) 1997-2016 C. Geuzaine, J.-F. Remacle
+//
+// See the LICENSE.txt file for license information. Please report all
+// bugs and problems to the public mailing list <gmsh@onelab.info>.
+
+#ifndef _SHOWNEIGHBORELEMENTS_H_
+#define _SHOWNEIGHBORELEMENTS_H_
+
+#include "Plugin.h"
+#include <set>
+
+class MVertex;
+
+extern "C"
+{
+  GMSH_Plugin *GMSH_RegisterShowNeighborElementsPlugin();
+}
+
+class GMSH_ShowNeighborElementsPlugin : public GMSH_PostPlugin
+{
+private :
+  int _nLayers;
+  int _nel1, _nel2, _nel3, _nel4, _nel5;
+  std::multimap<MVertex*, MElement*> _vert2elem;
+  std::set<MVertex*> _vertices;
+
+public :
+  GMSH_ShowNeighborElementsPlugin() {}
+  std::string getName() const { return "ShowNeighborElements"; }
+  std::string getShortHelp() const {
+    return "Choose which type of element to hide.";
+  }
+  std::string getHelp() const;
+  std::string getAuthor() const { return "Amaury Johnen"; }
+  int getNbOptions() const;
+  StringXNumber* getOption(int);
+  PView* execute(PView *);
+
+private :
+  void _init(GEntity *);
+  void _showLayers(GEntity *, int nLayer);
+};
+
+#endif