From 4f9b4e3fb3c7eecfc9290b53b078bf1b99357648 Mon Sep 17 00:00:00 2001
From: mpv <mpv@opencascade.com>
Date: Fri, 7 Feb 2020 15:42:25 +0300
Subject: [PATCH] Make the dead shape correctly referenced by SMESH mesh after
 loading the python dump script

---
 src/SMESH_I/SMESH_Mesh_i.cxx | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/SMESH_I/SMESH_Mesh_i.cxx b/src/SMESH_I/SMESH_Mesh_i.cxx
index 3f65f9f9b..8c2a1a91b 100644
--- a/src/SMESH_I/SMESH_Mesh_i.cxx
+++ b/src/SMESH_I/SMESH_Mesh_i.cxx
@@ -273,6 +273,18 @@ void SMESH_Mesh_i::ReplaceShape(GEOM::GEOM_Object_ptr theNewGeom)
   // re-assign global hypotheses to the new shape
   _mainShapeTick = -1;
   CheckGeomModif( true );
+
+  // update the reference to theNewGeom (needed for correct execution of a dumped python script)
+  SALOMEDS::SObject_var aSO = _gen_i->ObjectToSObject(_this());
+  if (!aSO->_is_nil()) {
+    SALOMEDS::SObject_var aShapeRefSO;
+    if (aSO->FindSubObject(1, aShapeRefSO)) {
+      _gen_i->getStudyServant()->NewBuilder()->Addreference(
+        aShapeRefSO, _gen_i->getStudyServant()->FindObjectID(theNewGeom->GetStudyEntry()));
+    }
+  }
+
+  TPythonDump() <<  SMESH::SMESH_Mesh_var(_this()) << ".ReplaceShape( " << theNewGeom->GetStudyEntry() << " )";
 }
 
 //================================================================================
-- 
2.39.2