Simulation: separate code from node tree parsing and solver

1 files changed, 53 insertions, 0 deletions

diff --git a/source/blender/simulation/intern/simulation_solver.hh b/source/blender/simulation/intern/simulation_solver.hh
new file mode 100644
index 00000000000..c96cfee52da
--- /dev/null
+++ b/source/blender/simulation/intern/simulation_solver.hh
@@ -0,0 +1,53 @@
+/*
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License
+ * as published by the Free Software Foundation; either version 2
+ * of the License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+ */
+
+#ifndef __SIM_SIMULATION_SOLVER_HH__
+#define __SIM_SIMULATION_SOLVER_HH__
+
+#include "BLI_float3.hh"
+#include "BLI_span.hh"
+
+#include "DNA_simulation_types.h"
+
+#include "FN_attributes_ref.hh"
+
+struct Depsgraph;
+
+namespace blender::sim {
+
+class ParticleForce {
+ public:
+  virtual ~ParticleForce();
+  virtual void add_force(fn::AttributesRef attributes,
+                         MutableSpan<float3> r_combined_force) const = 0;
+};
+
+struct SimulationInfluences {
+  Map<const ParticleSimulationState *, Vector<const ParticleForce *>> particle_forces;
+};
+
+void initialize_simulation_states(Simulation &simulation,
+                                  Depsgraph &depsgraph,
+                                  const SimulationInfluences &influences);
+
+void solve_simulation_time_step(Simulation &simulation,
+                                Depsgraph &depsgraph,
+                                const SimulationInfluences &influences,
+                                float time_step);
+
+}  // namespace blender::sim
+
+#endif /* __SIM_SIMULATION_SOLVER_HH__ */