Quick renaming of stuff to be more generic as 'AI'. Now maybe I can get some sweet sweet investor money.

ai.cpp

@@ -0,0 +1,151 @@
+#include "dbc.hpp"
+#include "ai.hpp"
+
+namespace ai {
+  using namespace nlohmann;
+
+  bool is_subset(State& source, State& target) {
+    State result = source & target;
+    return result == target;
+  }
+
+  void Action::needs(int name, bool val) {
+    if(val) {
+      $positive_preconds[name] = true;
+      $negative_preconds[name] = false;
+    } else {
+      $negative_preconds[name] = true;
+      $positive_preconds[name] = false;
+    }
+  }
+
+  void Action::effect(int name, bool val) {
+    if(val) {
+      $positive_effects[name] = true;
+      $negative_effects[name] = false;
+    } else {
+      $negative_effects[name] = true;
+      $positive_effects[name] = false;
+    }
+  }
+
+  void Action::load(nlohmann::json& profile, nlohmann::json& config) {
+    dbc::check(config.contains("needs"),
+        fmt::format("Action.load({}): no 'needs' field", $name));
+    dbc::check(config.contains("effects"),
+        fmt::format("Action.load({}): no 'effects' field", $name));
+
+    for(auto& [name_key, value] : profile.items()) {
+      dbc::check(value < STATE_MAX, fmt::format("Action.load({}): profile field {} has value {} greater than STATE_MAX {}", $name, (std::string)name_key, (int)value, STATE_MAX));
+    }
+
+    for(auto& [name_key, value] : config["needs"].items()) {
+      dbc::check(profile.contains(name_key), fmt::format("Action.load({}): profile does not have name {}", $name, name_key));
+      int name = profile[name_key].template get<int>();
+
+      needs(name, bool(value));
+    }
+
+    for(auto& [name_key, value] : config["effects"].items()) {
+      dbc::check(profile.contains(name_key), fmt::format("Action.load({}): profile does not have name {}", $name, name_key));
+
+      int name = profile[name_key].template get<int>();
+
+      effect(name, bool(value));
+    }
+  }
+
+  bool Action::can_effect(State& state) {
+    return ((state & $positive_preconds) == $positive_preconds) &&
+      ((state & $negative_preconds) == ALL_ZERO);
+  }
+
+  State Action::apply_effect(State& state) {
+    return (state | $positive_effects) & ~$negative_effects;
+  }
+
+  int distance_to_goal(State& from, State& to) {
+    auto result = from ^ to;
+    return result.count();
+  }
+
+  Script reconstruct_path(std::unordered_map<Action, Action>& came_from, Action& current) {
+    Script total_path{current};
+    int count = 0;
+
+    while(came_from.contains(current) && count++ < 10) {
+      current = came_from.at(current);
+      if(current != FINAL_ACTION) {
+        total_path.push_front(current);
+      }
+    }
+
+    return total_path;
+  }
+
+  inline int h(State& start, State& goal) {
+    return distance_to_goal(start, goal);
+  }
+
+  inline int d(State& start, State& goal) {
+    return distance_to_goal(start, goal);
+  }
+
+  ActionState find_lowest(std::unordered_map<ActionState, int>& open_set) {
+    dbc::check(!open_set.empty(), "open set can't be empty in find_lowest");
+    const ActionState *result = nullptr;
+    int lowest_score = SCORE_MAX;
+
+    for(auto& kv : open_set) {
+      if(kv.second < lowest_score) {
+        lowest_score = kv.second;
+        result = &kv.first;
+      }
+    }
+
+    return *result;
+  }
+
+  std::optional<Script> plan_actions(std::vector<Action>& actions, State& start, State& goal) {
+    std::unordered_map<ActionState, int> open_set;
+    std::unordered_map<Action, Action> came_from;
+    std::unordered_map<State, int> g_score;
+
+    ActionState start_state{FINAL_ACTION, start};
+
+    g_score[start] = 0;
+    open_set[start_state] = g_score[start] + h(start, goal);
+
+    while(!open_set.empty()) {
+      auto current = find_lowest(open_set);
+
+      if(is_subset(current.state, goal)) {
+        return std::make_optional<Script>(reconstruct_path(came_from, current.action));
+      }
+
+      open_set.erase(current);
+
+      for(auto& neighbor_action : actions) {
+        // calculate the State being current/neighbor
+        if(!neighbor_action.can_effect(current.state)) {
+          continue;
+        }
+
+        auto neighbor = neighbor_action.apply_effect(current.state);
+        int d_score = d(current.state, neighbor);
+        int tentative_g_score = g_score[current.state] + d_score;
+        int neighbor_g_score = g_score.contains(neighbor) ? g_score[neighbor] : SCORE_MAX;
+        if(tentative_g_score < neighbor_g_score) {
+          came_from.insert_or_assign(neighbor_action, current.action);
+
+          g_score[neighbor] = tentative_g_score;
+          // open_set gets the fScore
+          ActionState neighbor_as{neighbor_action, neighbor};
+          open_set[neighbor_as] = tentative_g_score + h(neighbor, goal);
+        }
+      }
+    }
+
+    return std::nullopt;
+  }
+}