Properly reset flow state after each process iteration

CURA-11044
Ultimaker · Sep 20, 2023 · 4ff21e1 · 4ff21e1
1 parent d86bbfa
commit 4ff21e1
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/include/gradual_flow/gcode_path.h b/include/gradual_flow/gcode_path.h
@@ -378,6 +378,7 @@ struct GCodeState
         {
             current_flow = target_flow;
             discretized_duration_remaining = 0;
+            flow_state = FlowState::STABLE;
             return { path };
         }
 
@@ -400,6 +401,7 @@ struct GCodeState
             }
             else
             {
+                flow_state = FlowState::TRANSITION;
                 return { partitioned_gcode_path };
             }
         }
@@ -418,6 +420,7 @@ struct GCodeState
             {
                 remaining_path.speed = segment_speed;
                 discretized_duration_remaining = std::max(discretized_duration_remaining - remaining_path.totalDuration(), .0);
+                flow_state = discretized_duration_remaining > 0. ? FlowState::TRANSITION : FlowState::STABLE;
                 discretized_paths.emplace_back(remaining_path);
                 return discretized_paths;
             }
@@ -439,12 +442,15 @@ struct GCodeState
             }
             else
             {
+                flow_state = FlowState::TRANSITION;
                 discretized_duration_remaining = remaining_partition_duration;
                 return discretized_paths;
             }
         }
         discretized_paths.emplace_back(remaining_path);
 
+        flow_state = discretized_duration_remaining > 0. ? FlowState::TRANSITION : FlowState::STABLE;
+
         return discretized_paths;
     }
 };