Render strategy (wip)

Author: mikolasan

graphics/CMakeLists.txt

@@ -85,6 +85,9 @@ add_executable(neurons_gui
     thirdparty/imgui/backends/imgui_impl_sdl2.cpp
     neurons/renderer.cpp
     neurons/network.hpp
+    neurons/render/neuron_render.hpp
+    neurons/render/neuron_render.cpp
+    neurons/render/network_render.hpp
     neurons/neurons_gui.cpp
 )
 
@@ -97,6 +100,7 @@ target_include_directories(neurons_gui PRIVATE
     thirdparty/imgui/backends 
     bgfx_ext
     imgui_ext 
+    neurons
 )
 
 if (MSVC)

graphics/neurons/network.hpp

@@ -1,14 +1,21 @@
 #pragma once
+
+#include <memory>
 #include <vector>
+
 #include "neuron.hpp"
 #include "synapse.hpp"
+#include "render_strategy.h"
+
 
 struct Network {
     std::vector<Neuron> neurons;
     std::vector<std::vector<Synapse>> synapses;
     float dt = 1.0f;
     float time = 0.0f;
 
+    std::shared_ptr<RenderStrategy> render;
+
     explicit Network() = default;
 
     Network(int N) {
@@ -52,4 +59,22 @@ struct Network {
 
         time += dt;
     }
-};
+
+    void init() {
+        if (render) render->init();
+    }
+
+    void draw() const {
+        if (render) render->draw();
+    }
+
+    // void draw() const {
+    //     for (auto& n : neurons) {
+    //         n.draw();
+    //     }
+    // }
+
+    void update(float input, float dt, float t) {
+        if (render) render->update(dt);
+    }
+};

graphics/neurons/neuron.hpp

@@ -4,6 +4,7 @@
 
 #include "render_strategy.h"
 
+
 struct Neuron {
     size_t idx = 0;
     float v = 0.0f;
@@ -13,16 +14,20 @@ struct Neuron {
     float spike_time = -1.0f;
     bool spiked = false;
 
-    std::shared_ptr<NeuronStrategy> strategy;
+    std::shared_ptr<RenderStrategy> render;
+
+    void init() {
+        if (render) render->init();
+    }
 
     void simulate(float input, float dt) {
         v += dt * (-v + input) / 20.0f;
         spiked = (v > 1.0f);
-        if (strategy) strategy->update(*this, dt);
+        if (render) render->update(dt);
     }
 
     void draw() const {
-        if (strategy) strategy->draw(*this);
+        if (render) render->draw();
     }
 
     void update(float input, float dt, float t) {
@@ -42,21 +47,3 @@ struct Neuron {
         }
     }
 };
-
-struct RenderNeuronStrategy : NeuronStrategy {
-    std::shared_ptr<VisualContext> ctx;
-
-    RenderNeuronStrategy(std::shared_ptr<VisualContext> ctx) : ctx(ctx) {}
-
-    void update(Neuron&, float) override {
-        // Optional: nothing for now
-    }
-
-    void draw(const Neuron& n) const override {
-        size_t idx = n.idx;
-        const bx::Vec3& pos = ctx->positions[idx];
-
-        bx::Vec3 c = n.spiked ? bx::Vec3{1, 1, 0} : bx::Vec3{n.v, 0, 1 - n.v};
-        // draw_sphere(pos, 0.2f, c);
-    }
-};