MeasureGui: Set document when creating measurement

hlorus · Sep 29, 2024 · 5ec4d13 · 5ec4d13
1 parent 2a3978d
commit 5ec4d13
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/Mod/Measure/Gui/TaskMeasure.cpp b/src/Mod/Measure/Gui/TaskMeasure.cpp
@@ -204,6 +204,10 @@ App::DocumentObject* TaskMeasure::createObject(const App::MeasureType* measureTy
         Py_XDECREF(result);
     }
 
+    // Store the document in the document object
+    _mDocument = App::GetApplication().getActiveDocument();
+    _mMeasureObject->setTemporaryDocument(_mDocument);
+
     return static_cast<App::DocumentObject*>(_mMeasureObject);
 }
 
@@ -249,7 +253,6 @@ void TaskMeasure::saveObject()
         _mViewObject = nullptr;
     }
 
-    _mDocument = App::GetApplication().getActiveDocument();
     _mDocument->addObject(_mMeasureObject,
                           modeSwitch->currentIndex() != 0 ? modeSwitch->currentText().toLatin1()
                                                           : QString().toLatin1());