Merge pull request #11533 from KratosMultiphysics/core/entities-utili…

…ties-has-entity [FastPR][Core] Adding HasPrototypeEntity to EntityIdentifier
KratosMultiphysics · Sep 3, 2023 · 5af3577 · 5af3577
2 parents 864cdfa + 8a87677
commit 5af3577
Show file tree

Hide file tree

Showing 3 changed files with 52 additions and 0 deletions.
diff --git a/kratos/tests/cpp_tests/utilities/test_entities_utilities.cpp b/kratos/tests/cpp_tests/utilities/test_entities_utilities.cpp
@@ -0,0 +1,35 @@
+//    |  /           |
+//    ' /   __| _` | __|  _ \   __|
+//    . \  |   (   | |   (   |\__ `
+//   _|\_\_|  \__,_|\__|\___/ ____/
+//                   Multi-Physics
+//
+//  License:         BSD License
+//                   Kratos default license: kratos/license.txt
+//
+//  Main authors:    Ruben Zorrilla
+//
+//
+
+// Project includes
+#include "testing/testing.h"
+#include "containers/model.h"
+#include "includes/expect.h"
+#include "utilities/entities_utilities.h"
+
+namespace Kratos::Testing
+{
+
+KRATOS_TEST_CASE_IN_SUITE(EntityIdentifierHasPrototype, KratosCoreFastSuite)
+{
+    Model model;
+    auto& r_model_part = model.CreateModelPart("MainModelPart");
+    r_model_part.CreateNewNode(1, 0.0, 0.0, 0.0);
+    r_model_part.CreateNewNode(2, 1.0, 0.0, 0.0);
+    r_model_part.CreateNewNode(3, 0.0, 1.0, 0.0);
+    auto p_tri_geom = r_model_part.CreateNewGeometry("Triangle2D3", 1, {{1, 2, 3}});
+    auto entity_identifier = EntitiesUtilities::EntitityIdentifier<Element>("Element2D3N");
+    KRATOS_EXPECT_TRUE(entity_identifier.HasPrototypeEntity(*p_tri_geom));
+}
+
+}  // namespace Kratos::Testing.
diff --git a/kratos/utilities/entities_utilities.cpp b/kratos/utilities/entities_utilities.cpp
@@ -62,6 +62,15 @@ bool EntitityIdentifier<TEntity>::IsInitialized() const
 /***********************************************************************************/
 /***********************************************************************************/
 
+template<class TEntity>
+bool EntitityIdentifier<TEntity>::HasPrototypeEntity(const GeometryType& rGeometry) const
+{
+    return mTypes[static_cast<std::size_t>(rGeometry.GetGeometryType())] != nullptr;
+}
+
+/***********************************************************************************/
+/***********************************************************************************/
+
 template<class TEntity>
 const TEntity& EntitityIdentifier<TEntity>::GetPrototypeEntity(typename GeometryType::Pointer pGeometry) const
 {

diff --git a/kratos/utilities/entities_utilities.h b/kratos/utilities/entities_utilities.h
@@ -103,6 +103,14 @@ namespace EntitiesUtilities
          */
         bool IsInitialized() const;
 
+        /**
+         * @brief Get the prototype entity.
+         * @param rGeometry The reference to the geometry.
+         * @return true there is a prototype for the provided entity.
+         * @return false there is no prototype for the provided entity.
+         */
+        bool HasPrototypeEntity(const GeometryType& rGeometry) const;
+
         /**
          * @brief Get the prototype entity.
          * @param pGeometry The pointer to the geometry.