Simulation: use better api for adding and removing simulation states

1 files changed, 9 insertions, 1 deletions

diff --git a/source/blender/blenkernel/BKE_simulation.h b/source/blender/blenkernel/BKE_simulation.h
index ff6aaa5e30e..36f4122f4dc 100644
--- a/source/blender/blenkernel/BKE_simulation.h
+++ b/source/blender/blenkernel/BKE_simulation.h
@@ -17,13 +17,15 @@
 #ifndef __BKE_SIMULATION_H__
 #define __BKE_SIMULATION_H__
 
+#include "DNA_simulation_types.h"
+
 #ifdef __cplusplus
 extern "C" {
 #endif
 
 struct Depsgraph;
 struct Main;
-struct Simulation;
+struct Scene;
 
 void *BKE_simulation_add(struct Main *bmain, const char *name);
 
@@ -31,6 +33,12 @@ void BKE_simulation_data_update(struct Depsgraph *depsgraph,
                                 struct Scene *scene,
                                 struct Simulation *simulation);
 
+SimulationState *BKE_simulation_state_add(Simulation *simulation,
+                                          eSimulationStateType type,
+                                          const char *name);
+void BKE_simulation_state_remove(Simulation *simulation, SimulationState *state);
+void BKE_simulation_state_remove_all(Simulation *simulation);
+
 #ifdef __cplusplus
 }
 #endif