Working on graph evaluations. WIP.

src/AnimGraph/AnimGraph.h

@@ -9,30 +9,57 @@
 #include "AnimGraphNodes.h"
 
 struct AnimDataWorkBuffer {
-  std::vector<AnimData> m_eval_anim_data;
-  std::vector<AnimData*> m_available_data;
+  struct AnimDataList {
+    AnimData* m_anim_data = nullptr;
+    AnimDataList* next = nullptr;
+  };
 
-  AnimDataWorkBuffer(size_t stack_size) {
-    m_eval_anim_data.resize(stack_size);
-    m_available_data.resize(stack_size);
+  AnimDataList* m_anim_data_list = nullptr;
+  size_t m_num_allocations = 0;
 
-    for (size_t i = 0; i < stack_size; i++) {
-      m_available_data[i] = &m_eval_anim_data[i];
+  ~AnimDataWorkBuffer() {
+    AnimDataList* list_node = m_anim_data_list;
+    while (list_node != nullptr) {
+      AnimDataList* current_node = list_node;
+      list_node = list_node->next;
+      //delete current_node->m_anim_data;
+      current_node->m_anim_data = nullptr;
+      delete current_node;
     }
   }
 
-  void push(AnimData* anim_data) {
-    assert (m_available_data.size() < m_eval_anim_data.size());
-    m_available_data.push_back(anim_data);
+  AnimData* allocate(AnimGraphContext& graph_context) {
+    if (m_anim_data_list == nullptr) {
+      AnimData* result = new AnimData();
+      result->m_local_matrices.resize(graph_context.m_skeleton->num_soa_joints());
+      m_num_allocations++;
+      return result;
+    }
+
+    AnimData* result = m_anim_data_list->m_anim_data;
+    m_anim_data_list = m_anim_data_list->next;
+    delete m_anim_data_list;
+
+    return result;
   }
 
-  void pop() {
-    assert (m_available_data.size() > 0);
-    m_available_data.pop_back();
+  void free(AnimData* anim_data) {
+    AnimDataList* list_node = new AnimDataList;
+    list_node->next = m_anim_data_list;
+    list_node->m_anim_data = anim_data;
+    m_anim_data_list = list_node;
   }
 
-  AnimData* peek() {
-    return m_available_data.back();
+  size_t size() {
+    size_t result = 0;
+
+    AnimDataList* node = m_anim_data_list;
+    while (node != nullptr) {
+      result ++;
+      node = node->next;
+    }
+
+    return result;
   }
 };
 
@@ -53,7 +80,7 @@ struct AnimGraph {
   std::vector<Socket>& getGraphOutputs() { return m_socket_accessor->m_inputs; }
   std::vector<Socket>& getGraphInputs() { return m_socket_accessor->m_outputs; }
 
-  AnimDataWorkBuffer m_anim_data_work_buffer = AnimDataWorkBuffer(5);
+  AnimDataWorkBuffer m_anim_data_work_buffer;
 
   ~AnimGraph() {
     delete[] m_input_buffer;
@@ -76,7 +103,7 @@ struct AnimGraph {
   }
 
   void evalInputNode();
-  void prepareNodeEval(size_t node_index);
+  void prepareNodeEval(AnimGraphContext& graph_context, size_t node_index);
   void finishNodeEval(size_t node_index);
   void evalOutputNode();
 
@@ -106,6 +133,16 @@ struct AnimGraph {
     return nullptr;
   }
 
+  void* getOutputPtr(const std::string& name) const {
+    const Socket* input_socket = getOutputSocket(name);
+    if (input_socket != nullptr) {
+      return input_socket->m_reference.ptr;
+    }
+
+    return nullptr;
+  }
+
+
   int getNodeEvalOrderIndex(const AnimNode* node) {
     for (size_t i = 0, n = m_eval_ordered_nodes.size(); i < n; i++) {
       if (m_eval_ordered_nodes[i] == node) {