Construct shapes in memory and support more shapes

src/axom/klee/Geometry.cpp

@@ -6,29 +6,134 @@
 #include "axom/klee/Geometry.hpp"
 
 #include "axom/klee/GeometryOperators.hpp"
+#include "conduit_blueprint_mesh.hpp"
 
 #include <utility>
 
 namespace axom
 {
 namespace klee
 {
-bool operator==(const TransformableGeometryProperties &lhs,
-                const TransformableGeometryProperties &rhs)
+bool operator==(const TransformableGeometryProperties& lhs,
+                const TransformableGeometryProperties& rhs)
 {
   return lhs.dimensions == rhs.dimensions && lhs.units == rhs.units;
 }
 
-Geometry::Geometry(const TransformableGeometryProperties &startProperties,
+Geometry::Geometry(const TransformableGeometryProperties& startProperties,
                    std::string format,
                    std::string path,
                    std::shared_ptr<GeometryOperator const> operator_)
   : m_startProperties(startProperties)
   , m_format(std::move(format))
   , m_path(std::move(path))
+  , m_levelOfRefinement(0)
   , m_operator(std::move(operator_))
 { }
 
+Geometry::Geometry(const TransformableGeometryProperties& startProperties,
+                   const axom::sidre::Group* meshGroup,
+                   const std::string& topology,
+                   std::shared_ptr<GeometryOperator const> operator_)
+  : m_startProperties(startProperties)
+  , m_format("blueprint-tets")
+  , m_path()
+  , m_meshGroup(meshGroup)
+  , m_topology(topology)
+  , m_levelOfRefinement(0)
+  , m_operator(std::move(operator_))
+{
+#ifdef AXOM_DEBUG
+  SLIC_ASSERT_MSG(isBlueprintTetMesh(m_meshGroup),
+                  "Mesh provided to Geometry is not a valid blueprint "
+                  "unstructured tetrahedral mesh.");
+#endif
+}
+
+Geometry::Geometry(const TransformableGeometryProperties& startProperties,
+                   const axom::primal::Tetrahedron<double, 3>& tet,
+                   std::shared_ptr<GeometryOperator const> operator_)
+  : m_startProperties(startProperties)
+  , m_format("tet3D")
+  , m_path()
+  , m_meshGroup(nullptr)
+  , m_topology()
+  , m_tet(tet)
+  , m_levelOfRefinement(0)
+  , m_operator(std::move(operator_))
+{ }
+
+Geometry::Geometry(const TransformableGeometryProperties& startProperties,
+                   const axom::primal::Hexahedron<double, 3>& hex,
+                   std::shared_ptr<GeometryOperator const> operator_)
+  : m_startProperties(startProperties)
+  , m_format("hex3D")
+  , m_path()
+  , m_meshGroup(nullptr)
+  , m_topology()
+  , m_hex(hex)
+  , m_levelOfRefinement(0)
+  , m_operator(std::move(operator_))
+{ }
+
+Geometry::Geometry(const TransformableGeometryProperties& startProperties,
+                   const Sphere3D& sphere,
+                   axom::IndexType levelOfRefinement,
+                   std::shared_ptr<GeometryOperator const> operator_)
+  : m_startProperties(startProperties)
+  , m_format("sphere3D")
+  , m_path()
+  , m_meshGroup(nullptr)
+  , m_topology()
+  , m_sphere(sphere)
+  , m_levelOfRefinement(levelOfRefinement)
+  , m_operator(std::move(operator_))
+{ }
+
+Geometry::Geometry(const TransformableGeometryProperties& startProperties,
+                   const axom::Array<double, 2>& discreteFunction,
+                   const Point3D& vorBase,
+                   const Vector3D& vorDirection,
+                   axom::IndexType levelOfRefinement,
+                   std::shared_ptr<GeometryOperator const> operator_)
+  : m_startProperties(startProperties)
+  , m_format("vor3D")
+  , m_path()
+  , m_meshGroup(nullptr)
+  , m_topology()
+  , m_sphere()
+  , m_discreteFunction(discreteFunction)
+  , m_vorBase(vorBase)
+  , m_vorDirection(vorDirection)
+  , m_levelOfRefinement(levelOfRefinement)
+  , m_operator(std::move(operator_))
+{ }
+
+Geometry::Geometry(const TransformableGeometryProperties& startProperties,
+                   const axom::primal::Plane<double, 3>& plane,
+                   std::shared_ptr<GeometryOperator const> operator_)
+  : m_startProperties(startProperties)
+  , m_format("plane3D")
+  , m_path()
+  , m_meshGroup(nullptr)
+  , m_topology()
+  , m_plane(plane)
+  , m_levelOfRefinement(0)
+  , m_operator(std::move(operator_))
+{ }
+
+bool Geometry::hasGeometry() const
+{
+  bool isInMemory = m_format == "blueprint-tets" || m_format == "sphere3D" ||
+    m_format == "tet3D" || m_format == "hex3D" || m_format == "plane3D" ||
+    m_format == "cone3D" || m_format == "cylinder3D";
+  if(isInMemory)
+  {
+    return true;
+  }
+  return !m_path.empty();
+}
+
 TransformableGeometryProperties Geometry::getEndProperties() const
 {
   if(m_operator)
@@ -38,5 +143,69 @@ TransformableGeometryProperties Geometry::getEndProperties() const
   return m_startProperties;
 }
 
+const axom::sidre::Group* Geometry::getBlueprintMesh() const
+{
+  SLIC_ASSERT_MSG(
+    m_meshGroup,
+    axom::fmt::format(
+      "The Geometry format '{}' is not specified "
+      "as a blueprint mesh and/or has not been converted into one.",
+      m_format));
+  return m_meshGroup;
+}
+
+const std::string& Geometry::getBlueprintTopology() const
+{
+  SLIC_ASSERT_MSG(
+    m_meshGroup,
+    axom::fmt::format(
+      "The Geometry format '{}' is not specified "
+      "as a blueprint mesh and/or has not been converted into one.",
+      m_format));
+  return m_topology;
+}
+
+bool Geometry::isBlueprintTetMesh(const axom::sidre::Group* meshGroup) const
+{
+  conduit::Node bpMesh;
+  meshGroup->createNativeLayout(bpMesh);
+
+  conduit::Node info;
+  bool isValid = conduit::blueprint::mesh::verify(bpMesh, info);
+  // The above call to verify causes the crash, even though this
+  // function is never entered.
+  if(!isValid)
+  {
+    return false;
+  }
+
+  const auto& topology =
+    bpMesh.fetch_existing(axom::fmt::format("topologies/{}", m_topology));
+
+  std::string coordsetName = topology.fetch_existing("coordset").as_string();
+  const auto& coordSet =
+    bpMesh.fetch_existing(axom::fmt::format("coordsets/{}", coordsetName));
+
+  auto dim = conduit::blueprint::mesh::coordset::dims(coordSet);
+  if(dim != 3)
+  {
+    return false;
+  }
+
+  auto topoType = topology.fetch_existing("type").as_string();
+  if(topoType != "unstructured")
+  {
+    return false;
+  }
+
+  auto shapeType = topology.fetch_existing("elements/shape").as_string();
+  if(shapeType != "tet")
+  {
+    return false;
+  }
+
+  return true;
+}
+
 }  // namespace klee
 }  // namespace axom