From 9f84fbe9bbd476209ad93eedea76d49aa8ce75c7 Mon Sep 17 00:00:00 2001
From: bgbsww <120601209+bgbsww@users.noreply.github.com>
Date: Wed, 24 Jan 2024 22:24:08 -0500
Subject: [PATCH] Tweak resetElementMap to default to empty defined element map
 (#12098)

* Tweak resetElementMap tp default to empty defined element map

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
---
 src/App/ComplexGeoData.h         | 5 ++++-
 tests/src/App/ComplexGeoData.cpp | 2 +-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/App/ComplexGeoData.h b/src/App/ComplexGeoData.h
index d5630c3560..0bacdd6229 100644
--- a/src/App/ComplexGeoData.h
+++ b/src/App/ComplexGeoData.h
@@ -265,7 +265,10 @@ public:
      *
      * @return Returns the existing element map.
      */
-    virtual ElementMapPtr resetElementMap(ElementMapPtr elementMap=ElementMapPtr()) {
+    virtual ElementMapPtr resetElementMap(ElementMapPtr elementMap=ElementMapPtr(), bool forceEmpty=false) {
+        if ( ! elementMap && ! forceEmpty ) {
+            elementMap = std::make_shared<Data::ElementMap>();
+        }
         _elementMap.swap(elementMap);
         return elementMap;
     }
diff --git a/tests/src/App/ComplexGeoData.cpp b/tests/src/App/ComplexGeoData.cpp
index 4c344495ea..e7dc527e86 100644
--- a/tests/src/App/ComplexGeoData.cpp
+++ b/tests/src/App/ComplexGeoData.cpp
@@ -444,7 +444,7 @@ TEST_F(ComplexGeoDataTest, saveDocFileWithNoElementMap)
 {
     // Arrange
     Base::StringWriter writer;
-    cgd().resetElementMap();
+    cgd().resetElementMap(nullptr, true);  // Force undefined map
 
     // Act
     cgd().SaveDocFile(writer);