1 files changed, 10 insertions, 0 deletions
diff --git a/src/slic3r/GUI/ProjectDirtyStateManager.hpp b/src/slic3r/GUI/ProjectDirtyStateManager.hpp
index 81ac28915..b488c00bb 100644
--- a/src/slic3r/GUI/ProjectDirtyStateManager.hpp
+++ b/src/slic3r/GUI/ProjectDirtyStateManager.hpp
@@ -4,6 +4,9 @@
 #if ENABLE_PROJECT_DIRTY_STATE
 
 namespace Slic3r {
+namespace UndoRedo {
+class Stack;
+} // namespace UndoRedo
 namespace GUI {
 
 class ProjectDirtyStateManager
@@ -14,12 +17,19 @@ class ProjectDirtyStateManager
         bool presets{ false };
 
         bool is_dirty() const { return plater || presets; }
+        void reset() {
+            plater = false;
+            presets = false;
+        }
     };
 
     DirtyState m_state;
 
 public:
     bool is_dirty() const { return m_state.is_dirty(); }
+    void update_from_undo_redo_stack(const Slic3r::UndoRedo::Stack& main_stack, const Slic3r::UndoRedo::Stack& active_stack);
+    void update_from_presets();
+    void reset_after_save();
 
 #if ENABLE_PROJECT_DIRTY_STATE_DEBUG_WINDOW
     void render_debug_window() const;