From a8faade43f822a5dfc7e51e1d237858384a2178e Mon Sep 17 00:00:00 2001
From: "zhou.xu" <zhou.xu@bambulab.com>
Date: Wed, 13 Mar 2024 15:11:30 +0800
Subject: [PATCH] FIX:reload paint after background process apply

Jira: STUDIO-6493
Change-Id: I9a1986152f05163f236f58bb24210b690ca3d562
---
 src/slic3r/GUI/GUI_ObjectList.cpp |  1 +
 src/slic3r/GUI/GUI_Preview.hpp    |  3 +++
 src/slic3r/GUI/Plater.cpp         | 10 ++++++++++
 src/slic3r/GUI/Plater.hpp         |  1 +
 4 files changed, 15 insertions(+)

diff --git a/src/slic3r/GUI/GUI_ObjectList.cpp b/src/slic3r/GUI/GUI_ObjectList.cpp
index a0ca20de5..c1baf13e8 100644
--- a/src/slic3r/GUI/GUI_ObjectList.cpp
+++ b/src/slic3r/GUI/GUI_ObjectList.cpp
@@ -5930,6 +5930,7 @@ void ObjectList::toggle_printable_state()
 
     // update scene
     wxGetApp().plater()->update();
+    wxGetApp().plater()->reload_paint_after_background_process_apply();
 }
 
 ModelObject* ObjectList::object(const int obj_idx) const
diff --git a/src/slic3r/GUI/GUI_Preview.hpp b/src/slic3r/GUI/GUI_Preview.hpp
index 2ce7e67ec..9cb20f657 100644
--- a/src/slic3r/GUI/GUI_Preview.hpp
+++ b/src/slic3r/GUI/GUI_Preview.hpp
@@ -107,6 +107,7 @@ class Preview : public wxPanel
     const Slic3r::PrintBase* m_loaded_print { nullptr };
     //BBS: add only gcode mode
     bool m_only_gcode { false };
+    bool m_reload_paint_after_background_process_apply{false};
 
 public:
     enum class OptionType : unsigned int
@@ -160,6 +161,8 @@ public:
     void show_sliders(bool show = true);
     void show_moves_sliders(bool show = true);
     void show_layers_sliders(bool show = true);
+    void set_reload_paint_after_background_process_apply(bool flag) { m_reload_paint_after_background_process_apply = flag; }
+    bool get_reload_paint_after_background_process_apply() { return m_reload_paint_after_background_process_apply; }
 
 private:
     bool init(wxWindow* parent, Bed3D& bed, Model* model);
diff --git a/src/slic3r/GUI/Plater.cpp b/src/slic3r/GUI/Plater.cpp
index b9776eca0..7d02e958a 100644
--- a/src/slic3r/GUI/Plater.cpp
+++ b/src/slic3r/GUI/Plater.cpp
@@ -4963,6 +4963,12 @@ unsigned int Plater::priv::update_background_process(bool force_validation, bool
 
         notification_manager->set_slicing_progress_hidden();
     }
+    else {
+        if (preview && preview->get_reload_paint_after_background_process_apply()) {
+            preview->set_reload_paint_after_background_process_apply(false);
+            preview->reload_print();
+        }
+    }
 
     if ((invalidated != Print::APPLY_STATUS_UNCHANGED || force_validation) && ! background_process.empty()) {
         // The delayed error message is no more valid.
@@ -10419,6 +10425,10 @@ void Plater::select_view(const std::string& direction) { p->select_view(directio
 //BBS: add no_slice logic
 void Plater::select_view_3D(const std::string& name, bool no_slice) { p->select_view_3D(name, no_slice); }
 
+void Plater::reload_paint_after_background_process_apply() {
+    p->preview->set_reload_paint_after_background_process_apply(true);
+}
+
 bool Plater::is_preview_shown() const { return p->is_preview_shown(); }
 bool Plater::is_preview_loaded() const { return p->is_preview_loaded(); }
 bool Plater::is_view3D_shown() const { return p->is_view3D_shown(); }
diff --git a/src/slic3r/GUI/Plater.hpp b/src/slic3r/GUI/Plater.hpp
index 3d1b12964..58d96a080 100644
--- a/src/slic3r/GUI/Plater.hpp
+++ b/src/slic3r/GUI/Plater.hpp
@@ -296,6 +296,7 @@ public:
     //BBS: add no_slice logic
     void select_view_3D(const std::string& name, bool no_slice = true);
 
+    void reload_paint_after_background_process_apply();
     bool is_preview_shown() const;
     bool is_preview_loaded() const;
     bool is_view3D_shown() const;