From 7d7649bd0611ed4b1da98b6513366efb500d0b14 Mon Sep 17 00:00:00 2001 From: sp Date: Fri, 19 Jan 2024 10:40:23 +0100 Subject: [PATCH] include all action with indices --- util/PrismModulesPrinter.h | 1 + util/PrismPrinter.h | 4 ++++ 2 files changed, 5 insertions(+) diff --git a/util/PrismModulesPrinter.h b/util/PrismModulesPrinter.h index c0d192a..bfdd58f 100644 --- a/util/PrismModulesPrinter.h +++ b/util/PrismModulesPrinter.h @@ -109,6 +109,7 @@ namespace prism { std::vector configuration; std::vector viewDirections = {0, 1, 2, 3}; std::map viewDirectionToString = {{0, "East"}, {1, "South"}, {2, "West"}, {3, "North"}}; + std::vector> nonMovementActions = { {PICKUP, "pickup"}, {DROP, "drop"}, {TOGGLE, "toggle"}, {DONE, "done"} }; std::map>> agentNameActionMap; }; diff --git a/util/PrismPrinter.h b/util/PrismPrinter.h index a75e9f1..1f82cf4 100644 --- a/util/PrismPrinter.h +++ b/util/PrismPrinter.h @@ -9,6 +9,10 @@ #define LEFT 0 #define RIGHT 1 #define FORWARD 2 +#define PICKUP 3 +#define DROP 4 +#define TOGGLE 5 +#define DONE 6 typedef std::string AgentName;