You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

944 lines
47 KiB

  1. #include "PrismModulesPrinter.h"
  2. #include <map>
  3. #include <string>
  4. namespace prism {
  5. PrismModulesPrinter::PrismModulesPrinter(const ModelType &modelType, const size_t &numberOfPlayer, std::vector<Configuration> config, const bool enforceOneWays)
  6. : modelType(modelType), numberOfPlayer(numberOfPlayer), enforceOneWays(enforceOneWays), configuration(config), viewDirectionMapping({{0, "East"}, {1, "South"}, {2, "West"}, {3, "North"}}) {
  7. }
  8. std::ostream& PrismModulesPrinter::printModel(std::ostream &os, const ModelType &modelType) {
  9. switch(modelType) {
  10. case(ModelType::MDP):
  11. os << "mdp";
  12. break;
  13. case(ModelType::SMG):
  14. os << "smg";
  15. break;
  16. }
  17. os << "\n\n";
  18. return os;
  19. }
  20. std::ostream& PrismModulesPrinter::printBackgroundLabels(std::ostream &os, const AgentName &agentName, const std::pair<Color, cells> &backgroundTiles) {
  21. if(backgroundTiles.second.size() == 0) return os;
  22. bool first = true;
  23. std::string color = getColor(backgroundTiles.first);
  24. color.at(0) = std::toupper(color.at(0));
  25. os << "formula " << agentName << "On" << color << " = ";
  26. for(auto const& cell : backgroundTiles.second) {
  27. if(first) first = false; else os << " | ";
  28. os << "(x" << agentName << "=" << cell.column << "&y" << agentName << "=" << cell.row << ")";
  29. }
  30. os << ";\n";
  31. os << "label \"" << agentName << "On" << color << "\" = " << agentName << "On" << color << ";\n";
  32. return os;
  33. }
  34. std::ostream& PrismModulesPrinter::printRestrictionFormula(std::ostream& os, const AgentName &agentName, const std::string &direction, const cells &grid_cells, const cells &keys, const cells &doors) {
  35. bool first = true;
  36. os << "formula " << agentName << "CannotMove" << direction << " = " ;
  37. for(auto const& cell : grid_cells) {
  38. if(first) first = false;
  39. else os << " | ";
  40. os << "(x" << agentName << "=" << cell.column << "&y" << agentName << "=" << cell.row << ")";
  41. }
  42. if (!keys.empty()) {
  43. os << " | " << agentName << "CannotMove" << direction << "BecauseOfKey";
  44. }
  45. if (!doors.empty()) {
  46. os << " | " << agentName << "CannotMove" << direction << "BecauseOfDoor";
  47. }
  48. os << ";\n";
  49. return os;
  50. }
  51. std::ostream& PrismModulesPrinter::printKeyRestrictionFormula(std::ostream& os, const AgentName &agentName, const std::string &direction, const cells &keys) {
  52. bool first = true;
  53. os << "formula " << agentName << "CannotMove" << direction << "BecauseOfKey" << " = ";
  54. for (auto const& key : keys) {
  55. if(first) first = false;
  56. else os << " | ";
  57. std::string keyColor = key.getColor();
  58. std::string xKey = "xKey" + keyColor;
  59. std::string yKey = "yKey" + keyColor;
  60. coordinates coords;
  61. os << "(!" << agentName << "_has_" << keyColor << "_key & ";
  62. if (direction == "North") {
  63. os << " x" << agentName << " = " << xKey << "&y" << agentName << "-1 = " << yKey << ")";
  64. } else if (direction == "South") {
  65. os << " x" << agentName << " = " << xKey << "&y" << agentName << "+1 = " << yKey << ")";
  66. } else if (direction == "East") {
  67. os << " x" << agentName << "+1 = " << xKey << "&y" << agentName << " = " << yKey << ")";
  68. } else if (direction == "West") {
  69. os << " x" << agentName << "-1 = " << xKey << "&y" << agentName << " = " << yKey << ")";
  70. } else {
  71. os << "Invalid Direction! in Key Restriction";
  72. }
  73. }
  74. os << ";\n";
  75. return os;
  76. }
  77. std::ostream& PrismModulesPrinter::printDoorRestrictionFormula(std::ostream& os, const AgentName &agentName, const std::string &direction, const cells &doors) {
  78. bool first = true;
  79. os << "formula " << agentName << "CannotMove" << direction << "BecauseOfDoor" << " = ";
  80. for (auto const& door : doors) {
  81. if (first) first = false;
  82. else os << " | ";
  83. std::string doorColor = door.getColor();
  84. size_t y = door.getCoordinates().first;
  85. size_t x = door.getCoordinates().second;
  86. os << "(!Door" << doorColor << "open & ";
  87. if (direction == "North") {
  88. os << " x" << agentName << " = " << x << "&y" << agentName << "-1 = " << y << ")";
  89. } else if (direction == "South") {
  90. os << " x" << agentName << " = " << x << "&y" << agentName << "+1 = " << y << ")";
  91. } else if (direction == "East") {
  92. os << " x" << agentName << "+1 = " << x << "&y" << agentName << " = " << y << ")";
  93. } else if (direction == "West") {
  94. os << " x" << agentName << "-1 = " << y << "&y" << agentName << " = " << y << ")";
  95. } else {
  96. os << "Invalid Direction! in Key Restriction";
  97. }
  98. }
  99. os << ";\n";
  100. return os;
  101. }
  102. std::ostream& PrismModulesPrinter::printIsOnSlipperyFormula(std::ostream& os, const AgentName &agentName, const std::vector<std::reference_wrapper<cells>> &slipperyCollection, const cells &slipperyNorth, const cells &slipperyEast, const cells &slipperySouth, const cells &slipperyWest) {
  103. if(std::find_if(slipperyCollection.cbegin(), slipperyCollection.cend(), [=](const std::reference_wrapper<cells>& c) { return !c.get().empty(); }) == slipperyCollection.cend()) {
  104. os << "formula " << agentName << "IsOnSlippery = false;\n";
  105. return os;
  106. }
  107. bool first = true;
  108. os << "formula " << agentName << "IsOnSlippery = ";
  109. for (const auto& slippery: slipperyCollection) {
  110. for(const auto& cell : slippery.get()) {
  111. if(first) first = false; else os << " | ";
  112. os << "(x" << agentName << "=" << cell.column << "&y" << agentName << "=" << cell.row << ")";
  113. }
  114. }
  115. os << ";\n";
  116. if(enforceOneWays) {
  117. first = true;
  118. os << "formula " << agentName << "IsOnSlipperyNorth = ";
  119. for (const auto& cell: slipperyNorth) {
  120. if(first) first = false; else os << " | ";
  121. os << "(x" << agentName << "=" << cell.column << "&y" << agentName << "=" << cell.row << ")";
  122. }
  123. os << ";\n";
  124. first = true;
  125. os << "formula " << agentName << "IsOnSlipperyEast = ";
  126. for (const auto& cell: slipperyEast) {
  127. if(first) first = false; else os << " | ";
  128. os << "(x" << agentName << "=" << cell.column << "&y" << agentName << "=" << cell.row << ")";
  129. }
  130. os << ";\n";
  131. first = true;
  132. os << "formula " << agentName << "IsOnSlipperySouth = ";
  133. for (const auto& cell: slipperySouth) {
  134. if(first) first = false; else os << " | ";
  135. os << "(x" << agentName << "=" << cell.column << "&y" << agentName << "=" << cell.row << ")";
  136. }
  137. os << ";\n";
  138. first = true;
  139. os << "formula " << agentName << "IsOnSlipperyWest = ";
  140. for (const auto& cell: slipperyWest) {
  141. if(first) first = false; else os << " | ";
  142. os << "(x" << agentName << "=" << cell.column << "&y" << agentName << "=" << cell.row << ")";
  143. }
  144. os << ";\n";
  145. }
  146. return os;
  147. }
  148. std::ostream& PrismModulesPrinter::printIsInLavaFormula(std::ostream& os, const AgentName &agentName, const cells &lava) {
  149. if(lava.size() == 0) {
  150. os << "formula " << agentName << "IsInLava = false;\n";
  151. return os;
  152. }
  153. bool first = true;
  154. os << "formula " << agentName << "IsInLava = ";
  155. for(auto const& cell : lava) {
  156. if(first) first = false; else os << " | ";
  157. os << "(x" << agentName << "=" << cell.column << "&y" << agentName << "=" << cell.row << ")";
  158. }
  159. os << ";\n";
  160. os << "formula " << agentName << "IsInLavaAndNotDone = " << agentName << "IsInLava & !" << agentName << "Done;\n";
  161. os << "label \"" << agentName << "IsInLavaAndNotDone\" = " << agentName << "IsInLava & !" << agentName << "Done;\n";
  162. return os;
  163. }
  164. std::ostream& PrismModulesPrinter::printTurningNotAllowedFormulas(std::ostream& os, const AgentName &agentName, const cells &noTurnFloor) {
  165. if( (!enforceOneWays or noTurnFloor.size() == 0) or (noTurnFloor.size() == 0) ) {
  166. os << "formula " << agentName << "CannotTurn = false;\n";
  167. return os;
  168. }
  169. bool first = true;
  170. os << "formula " << agentName << "CannotTurn = ";
  171. for(auto const& cell : noTurnFloor) {
  172. if(first) first = false; else os << " | ";
  173. os << "(x" << agentName << "=" << cell.column << "&y" << agentName << "=" << cell.row << ")";
  174. }
  175. os << " | " << agentName << "IsOnSlippery;\n";
  176. return os;
  177. }
  178. std::ostream& PrismModulesPrinter::printIsFixedFormulas(std::ostream& os, const AgentName &agentName) {
  179. os << "formula " << agentName << "IsFixed = false;\n";
  180. os << "formula " << agentName << "SlipperyTurnLeftAllowed = true;\n";
  181. os << "formula " << agentName << "SlipperyTurnRightAllowed = true;\n";
  182. os << "formula " << agentName << "SlipperyMoveForwardAllowed = true;\n";
  183. os << "label \"FixedStates" << agentName << "\" = " << agentName << "IsFixed | !" << agentName << "SlipperyTurnRightAllowed | !" << agentName << "SlipperyTurnLeftAllowed | !" << agentName << "SlipperyMoveForwardAllowed | " << agentName << "IsInGoal | " << agentName << "IsInLava";
  184. if(enforceOneWays) {
  185. os << " | " << agentName << "CannotTurn";
  186. }
  187. os << ";\n";
  188. //os << "label \"FixedStates\" = " << agentName << "IsFixed | " << agentName << "IsOnSlippery | " << agentName << "IsInGoal | " << agentName << "IsInLava;\n";
  189. return os;
  190. }
  191. std::ostream& PrismModulesPrinter::printWallFormula(std::ostream& os, const AgentName &agentName, const cells &walls) {
  192. os << "formula " << agentName << "IsOnWall = ";
  193. bool first = true;
  194. for(auto const& cell : walls) {
  195. if(first) first = false; else os << " | ";
  196. os << "(x" << agentName << "=" << cell.column << "&y" << agentName << "=" << cell.row << ")";
  197. }
  198. os << ";\n";
  199. return os;
  200. }
  201. std::ostream& PrismModulesPrinter::printFormulas(std::ostream& os,
  202. const AgentName &agentName,
  203. const cells &restrictionNorth,
  204. const cells &restrictionEast,
  205. const cells &restrictionSouth,
  206. const cells &restrictionWest,
  207. const std::vector<std::reference_wrapper<cells>> &slipperyCollection,
  208. const cells &lava,
  209. const cells &walls,
  210. const cells &noTurnFloor,
  211. const cells &slipperyNorth,
  212. const cells &slipperyEast,
  213. const cells &slipperySouth,
  214. const cells &slipperyWest,
  215. const cells &keys,
  216. const cells &doors) {
  217. printRestrictionFormula(os, agentName, "North", restrictionNorth, keys, doors);
  218. printRestrictionFormula(os, agentName, "East", restrictionEast, keys, doors);
  219. printRestrictionFormula(os, agentName, "South", restrictionSouth, keys, doors);
  220. printRestrictionFormula(os, agentName, "West", restrictionWest, keys, doors);
  221. if (!keys.empty()) {
  222. printKeyRestrictionFormula(os, agentName, "North", keys);
  223. printKeyRestrictionFormula(os, agentName, "East", keys);
  224. printKeyRestrictionFormula(os, agentName, "South", keys);
  225. printKeyRestrictionFormula(os, agentName, "West", keys);
  226. }
  227. if (!doors.empty()) {
  228. printDoorRestrictionFormula(os, agentName, "North", doors);
  229. printDoorRestrictionFormula(os, agentName, "East", doors);
  230. printDoorRestrictionFormula(os, agentName, "South", doors);
  231. printDoorRestrictionFormula(os, agentName, "West", doors);
  232. }
  233. printIsOnSlipperyFormula(os, agentName, slipperyCollection, slipperyNorth, slipperyEast, slipperySouth, slipperyWest);
  234. printIsInLavaFormula(os, agentName, lava);
  235. printWallFormula(os, agentName, walls);
  236. printTurningNotAllowedFormulas(os, agentName, noTurnFloor);
  237. printIsFixedFormulas(os, agentName);
  238. os << "\n";
  239. return os;
  240. }
  241. std::ostream& PrismModulesPrinter::printGoalLabel(std::ostream& os, const AgentName &agentName, const cells &goals) {
  242. if(goals.size() == 0) {
  243. os << "formula " << agentName << "IsInGoal = false;\n";
  244. return os;
  245. }
  246. bool first = true;
  247. os << "formula " << agentName << "IsInGoal = ";
  248. for(auto const& cell : goals) {
  249. if(first) first = false; else os << " | ";
  250. os << "(x" << agentName << "=" << cell.column << "&y" << agentName << "=" << cell.row << ")";
  251. }
  252. os << ";\n";
  253. os << "formula " << agentName << "IsInGoalAndNotDone = " << agentName << "IsInGoal & !" << agentName << "Done;\n";
  254. os << "label \"" << agentName << "IsInGoalAndNotDone\" = " << agentName << "IsInGoal & !" << agentName << "Done;\n";
  255. return os;
  256. }
  257. std::ostream& PrismModulesPrinter::printCrashLabel(std::ostream &os, const std::vector<AgentName> agentNames) {
  258. os << "label crash = ";
  259. bool first = true;
  260. for(auto const& agentName : agentNames) {
  261. if(agentName == "Agent") continue;
  262. if(first) first = false; else os << " | ";
  263. os << "(xAgent=x" << agentName << ")&(yAgent=y" << agentName << ")";
  264. }
  265. os << ";\n\n";
  266. return os;
  267. }
  268. std::ostream& PrismModulesPrinter::printConfiguration(std::ostream& os, const std::vector<Configuration>& configurations) {
  269. for (auto& configuration : configurations) {
  270. if (configuration.overwrite_ || configuration.type_ == ConfigType::Module) {
  271. continue;
  272. }
  273. os << configuration.expression_ << std::endl;
  274. }
  275. return os;
  276. }
  277. std::ostream& PrismModulesPrinter::printAvoidanceLabel(std::ostream &os, const std::vector<AgentName> agentNames, const int &distance) {
  278. os << "label avoidance = ";
  279. bool first = true;
  280. for(auto const& agentName : agentNames) {
  281. if(agentName == "Agent") continue;
  282. if(first) first = false; else os << " | ";
  283. os << "max(xAgent-x" << agentName << ",x" << agentName << "-xAgent)+";
  284. os << "max(yAgent-y" << agentName << ",y" << agentName << "-yAgent) ";
  285. }
  286. os << ";\n\n";
  287. return os;
  288. }
  289. // TODO this does not account for multiple agents yet, i.e. key can be picked up multiple times
  290. std::ostream& PrismModulesPrinter::printKeysLabels(std::ostream& os, const AgentName &agentName, const cells &keys) {
  291. if(keys.size() == 0) return os;
  292. for(auto const& key : keys) {
  293. std::string keyColor = key.getColor();
  294. std::string xKey = "xKey" + keyColor;
  295. std::string yKey = "yKey" + keyColor;
  296. os << "label \"" << agentName << "PickedUp" << keyColor << "Key\" = " << agentName << "_has_" << keyColor << "_key = true;\n";
  297. os << "formula " << agentName << "CanPickUp" << keyColor << "Key = ";
  298. os << "((x" << agentName << "-1 = " << xKey << "&y" << agentName << " = " << yKey << "&view" << agentName << " = 2) |";
  299. os << " (x" << agentName << "+1 = " << xKey << "&y" << agentName << " = " << yKey << "&view" << agentName << " = 0) |";
  300. os << " (x" << agentName << " = " << xKey << "&y" << agentName << "-1 = " << yKey << "&view" << agentName << " = 3) |";
  301. os << " (x" << agentName << " = " << xKey << "&y" << agentName << "+1 = " << yKey << "&view" << agentName << " = 1) ) &";
  302. os << "!" << agentName << "_has_" << keyColor << "_key;";
  303. }
  304. os << "\n";
  305. return os;
  306. }
  307. std::ostream& PrismModulesPrinter::printBooleansForKeys(std::ostream &os, const AgentName &agentName, const cells &keys) {
  308. for(auto const& key : keys) {
  309. os << "\t" << agentName << "_has_"<< key.getColor() << "_key : bool;\n";//init false;\n";
  310. }
  311. os << "\n";
  312. return os;
  313. }
  314. std::ostream& PrismModulesPrinter::printBooleansForBackground(std::ostream &os, const AgentName &agentName, const std::map<Color, cells> &backgroundTiles) {
  315. for(auto const& [color, cells] : backgroundTiles) {
  316. if(cells.size() == 0) continue;
  317. std::string c = getColor(color);
  318. c.at(0) = std::toupper(c.at(0));
  319. os << "\t" << agentName << "_picked_up_" << c << " : bool init false;\n";
  320. }
  321. os << "\n";
  322. return os;
  323. }
  324. std::ostream& PrismModulesPrinter::printActionsForKeys(std::ostream &os, const AgentName &agentName, const cells &keys) {
  325. for(auto const& key : keys) { // TODO ADD Drop action and enforce that pickup only possible if pockets empty (nothing picked up already)
  326. os << "\n";
  327. std::string keyColor = key.getColor();
  328. os << "\t[pickup_" << keyColor << "_key]\t" << pickupGuard(agentName, keyColor) << "-> ";
  329. os << "(" << agentName << "_has_" << keyColor << "_key'=true) & (" << agentName << "_is_carrying_object'=true);\n";
  330. os << "\n";
  331. os << "\t[drop_" << keyColor << "_key_north]\t" << dropGuard(agentName, keyColor, 3) << "-> ";
  332. os << "(" << agentName << "_has_" << keyColor << "_key'=false) & (" << agentName << "_is_carrying_object'=false);\n";
  333. os << "\t[drop_" << keyColor << "_key_west]\t" << dropGuard(agentName, keyColor, 2) << "-> ";
  334. os << "(" << agentName << "_has_" << keyColor << "_key'=false) & (" << agentName << "_is_carrying_object'=false);\n";
  335. os << "\t[drop_" << keyColor << "_key_south]\t" << dropGuard(agentName, keyColor, 1) << "-> ";
  336. os << "(" << agentName << "_has_" << keyColor << "_key'=false) & (" << agentName << "_is_carrying_object'=false);\n";
  337. os << "\t[drop_" << keyColor << "_key_east]\t" << dropGuard(agentName, keyColor, 0) << "-> ";
  338. os << "(" << agentName << "_has_" << keyColor << "_key'=false) & (" << agentName << "_is_carrying_object'=false);\n";
  339. }
  340. return os;
  341. }
  342. std::string PrismModulesPrinter::pickupGuard(const AgentName &agentName, const std::string keyColor ) {
  343. return "!" + agentName + "_is_carrying_object &\t" + agentName + "CanPickUp" + keyColor + "Key ";
  344. }
  345. std::string PrismModulesPrinter::dropGuard(const AgentName &agentName, const std::string keyColor, size_t view) {
  346. return viewVariable(agentName, view, true) + "\t!" + agentName + "CannotMove" + viewDirectionMapping.at(view) + "\t&\t" + agentName + "_has_" + keyColor + "_key\t";
  347. }
  348. std::ostream& PrismModulesPrinter::printActionsForBackground(std::ostream &os, const AgentName &agentName, const std::map<Color, cells> &backgroundTiles) {
  349. for(auto const& [color, cells] : backgroundTiles) {
  350. if(cells.size() == 0) continue;
  351. std::string c = getColor(color);
  352. c.at(0) = std::toupper(c.at(0));
  353. os << "\t[" << agentName << "_pickup_" << c << "] " << agentName << "On" << c << " & !" << agentName << "_picked_up_" << c << " -> ";
  354. os << "(" << agentName << "_picked_up_" << c << "'=true);\n";
  355. }
  356. os << "\n";
  357. return os;
  358. }
  359. std::ostream& PrismModulesPrinter::printInitStruct(std::ostream &os, const AgentNameAndPositionMap &agents, const KeyNameAndPositionMap &keys, const cells &lockedDoors, const cells &unlockedDoors) {
  360. os << "init\n";
  361. os << "\t";
  362. bool first = true;
  363. for (auto const& agent : agents) {
  364. if (first) first = false;
  365. else os << " & ";
  366. os << "(!" << agent.first << "IsInGoal & !" << agent.first << "IsInLava & !" << agent.first << "Done & !" << agent.first << "IsOnWall & ";
  367. os << "x" << agent.first << "=" << agent.second.second << " & y" << agent.first << "=" << agent.second.first << ")";
  368. os << " & !" << agent.first << "_is_carrying_object";
  369. if(enforceOneWays) {
  370. os << " & ( !AgentCannotTurn ) ";
  371. } else {
  372. os << " & ( !AgentIsOnSlippery ) ";
  373. }
  374. for (auto const& key : keys) {
  375. os << " & ( !" << agent.first << "_has_" << key.first << "_key )";
  376. }
  377. }
  378. for (auto const& key : keys) {
  379. os << " & ( xKey" << key.first << "="<< key.second.second<< ")";
  380. os << " & ( yKey" << key.first << "=" << key.second.first << ")";
  381. }
  382. for (auto const& locked : lockedDoors) {
  383. os << " & (Door" << locked.getColor() << "locked & !Door" << locked.getColor() << "open)";
  384. }
  385. for (auto const& unlocked : unlockedDoors) {
  386. os << " & (!Door" << unlocked.getColor() << "locked & !Door" << unlocked.getColor() << "open)";
  387. }
  388. // os << " & ( !" << agentName << "_is_carrying_object" << ")";
  389. os << "\nendinit\n\n";
  390. return os;
  391. }
  392. std::ostream& PrismModulesPrinter::printModule(std::ostream &os, const AgentName &agentName, const size_t &agentIndex, const coordinates &boundaries, const coordinates& initialPosition, const cells &keys, const std::map<Color, cells> &backgroundTiles, const bool agentWithView, const std::vector<float> &probabilities) {
  393. os << "module " << agentName << "\n";
  394. os << "\tx" << agentName << " : [1.." << boundaries.second << "];\n";
  395. os << "\ty" << agentName << " : [1.." << boundaries.first << "];\n";
  396. os << "\t" << agentName << "_is_carrying_object : bool;\n";
  397. printBooleansForKeys(os, agentName, keys);
  398. printBooleansForBackground(os, agentName, backgroundTiles);
  399. os << "\t" << agentName << "Done : bool;\n";
  400. if(agentWithView) {
  401. os << "\tview" << agentName << " : [0..3];\n";
  402. os << "\n";
  403. os << "\t[" << agentName << "_turn_right] " << moveGuard(agentIndex) << " !" << agentName << "CannotTurn & " << " !" << agentName << "IsFixed & " << " !" << agentName << "IsInGoal & !" << agentName << "IsInLava & !" << agentName << "IsOnSlippery -> (view" << agentName << "'=mod(view" << agentName << " + 1, 4)) " << moveUpdate(agentIndex) << ";\n";
  404. os << "\t[" << agentName << "_turn_left] " << moveGuard(agentIndex) << " !" << agentName << "CannotTurn & " << " !" << agentName << "IsFixed & " << " !" << agentName << "IsInGoal & !" << agentName << "IsInLava & !" << agentName << "IsOnSlippery & view" << agentName << ">0 -> (view" << agentName << "'=view" << agentName << " - 1) " << moveUpdate(agentIndex) << ";\n";
  405. os << "\t[" << agentName << "_turn_left] " << moveGuard(agentIndex) << " !" << agentName << "CannotTurn & " << " !" << agentName << "IsFixed & " << " !" << agentName << "IsInGoal & !" << agentName << "IsInLava & !" << agentName << "IsOnSlippery & view" << agentName << "=0 -> (view" << agentName << "'=3) " << moveUpdate(agentIndex) << ";\n";
  406. if(enforceOneWays) {
  407. os << "\t[" << agentName << "_stuck] !" << agentName << "IsFixed & " << agentName << "CannotTurn & view" << agentName << " = 0 & " << agentName << "CannotMoveEast -> true;\n";
  408. os << "\t[" << agentName << "_stuck] !" << agentName << "IsFixed & " << agentName << "CannotTurn & view" << agentName << " = 1 & " << agentName << "CannotMoveSouth -> true;\n";
  409. os << "\t[" << agentName << "_stuck] !" << agentName << "IsFixed & " << agentName << "CannotTurn & view" << agentName << " = 2 & " << agentName << "CannotMoveWest -> true;\n";
  410. os << "\t[" << agentName << "_stuck] !" << agentName << "IsFixed & " << agentName << "CannotTurn & view" << agentName << " = 3 & " << agentName << "CannotMoveNorth -> true;\n";
  411. }
  412. } else {
  413. os << "\t[" << agentName << "_turns] " << " !" << agentName << "CannotTurn & " << " !" << agentName << "IsFixed & " << moveGuard(agentIndex) << " true -> (x" << agentName << "'=x" << agentName << ")" << moveUpdate(agentIndex) << ";\n";
  414. }
  415. printActionsForKeys(os, agentName, keys);
  416. printActionsForBackground(os, agentName, backgroundTiles);
  417. os << "\n";
  418. printMovementActions(os, agentName, agentIndex, agentWithView);
  419. for(auto const& probability : probabilities) {
  420. printMovementActions(os, agentName, agentIndex, agentWithView, probability);
  421. }
  422. printDoneActions(os, agentName, agentIndex);
  423. printConfiguredActions(os, agentName);
  424. os << "\n";
  425. return os;
  426. }
  427. std::ostream& PrismModulesPrinter::printKeyModule(std::ostream &os, const cell &key, const coordinates &boundaries, AgentName agentName) {
  428. std::string keyIdentifier = "Key" + key.getColor();
  429. os << "module " << keyIdentifier << "\n";
  430. os << "\tx" << keyIdentifier << " : [-1.." << boundaries.second << "];\n";
  431. os << "\ty" << keyIdentifier << " : [-1.." << boundaries.first << "];\n";
  432. os << "\n";
  433. printKeyActions(os, key ,keyIdentifier, agentName);
  434. os << "\n";
  435. return os;
  436. }
  437. std::ostream& PrismModulesPrinter::printKeyActions(std::ostream &os, const cell &key ,const std::string &keyIdentifier, AgentName agentName) {
  438. std::string keyColor = key.getColor();
  439. os << "\t[pickup_" << keyColor << "key]\t" << pickupGuard(agentName, keyColor) << "-> ";
  440. os << "(xKey" << keyColor << "'=-1) & (yKey" << keyColor << "'=-1)" << ";\n";
  441. os << "\t[drop_" << keyColor << "_key_north]\t" << dropGuard(agentName, keyColor, 3) << "-> ";
  442. os << "(xKey" << keyColor << "'=x" << agentName << ") & (yKey" << keyColor << "'=y" <<agentName << "-1) ;\n";
  443. os << "\t[drop_" << keyColor << "_key_west]\t" << dropGuard(agentName, keyColor, 2) << "-> ";
  444. os << "(xKey" << keyColor << "'=x" << agentName << "-1) & (yKey" << keyColor << "'=y" <<agentName << ") ;\n";
  445. os << "\t[drop_" << keyColor << "_key_south]\t" << dropGuard(agentName, keyColor, 1) << "-> ";
  446. os << "(xKey" << keyColor << "'=x" << agentName << ") & (yKey" << keyColor << "'=y" <<agentName << "+1) ;\n";
  447. os << "\t[drop_" << keyColor << "_key_east]\t" << dropGuard(agentName, keyColor, 0) << "-> ";
  448. os << "(xKey" << keyColor << "'=x" << agentName << "+1) & (yKey" << keyColor << "'=y" <<agentName << ") ;\n";
  449. return os;
  450. }
  451. std::ostream& PrismModulesPrinter::printDoorModule(std::ostream &os, const cell &door, const coordinates &boundaries, AgentName agent) {
  452. std::string doorIdentifier = "Door" + door.getColor();
  453. os << "module " << doorIdentifier << "\n";
  454. os << "\t" << doorIdentifier << "locked : bool;\n";
  455. os << "\t" << doorIdentifier << "open : bool;\n";
  456. printDoorActions(os, door, doorIdentifier, agent);
  457. return os;
  458. }
  459. std::ostream& PrismModulesPrinter::printDoorActions(std::ostream &os, const cell &door ,const std::string &doorIdentifier, AgentName agentName) {
  460. os << "\t[" << "unlock_" << doorIdentifier << "]\t" << unlockGuard(agentName, door) << " -> ";
  461. os << "(" << doorIdentifier << "locked'=false) & (" << doorIdentifier << "open'=true) ;\n";
  462. os << "\t[" << "toggle_" << doorIdentifier << "]\t" << toggleGuard(agentName, door) << " -> ";
  463. os << "(" << doorIdentifier << "open'=!" << doorIdentifier << "open) ;\n";
  464. return os;
  465. }
  466. std::string PrismModulesPrinter::unlockGuard(const AgentName &agentName, const cell& door) {
  467. std::string doorColor = door.getColor();
  468. std::string ret;
  469. ret += agentName + "_has_" + doorColor + "_key & ";
  470. ret += "((" + viewVariable(agentName, 0, true) + "x" + agentName + "+ 1 = " + std::to_string(door.column) + " & y" + agentName + "= " + std::to_string(door.row) + ")";
  471. ret += " | (" + viewVariable(agentName, 1, true) + "x" + agentName + " = " + std::to_string(door.column) + " & y" + agentName + " + 1 = " + std::to_string(door.row) + ")";
  472. ret += " | (" + viewVariable(agentName, 2, true) + "x" + agentName + "- 1 = " + std::to_string(door.column) + " & y" + agentName + "= " + std::to_string(door.row) + ")";
  473. ret += " | (" + viewVariable(agentName, 3, true) + "x" + agentName + " = " + std::to_string(door.column) + " & y" + agentName + " - 1 = " + std::to_string(door.row) + "))";
  474. return ret;
  475. }
  476. std::string PrismModulesPrinter::toggleGuard(const AgentName &agentName, const cell& door) {
  477. std::string doorColor = door.getColor();
  478. std::string ret;
  479. ret += "!Door" + doorColor + "locked & (";
  480. ret += "(" + viewVariable(agentName, 0, true) + "x" + agentName + "+ 1 = " + std::to_string(door.column) + " & y" + agentName + "= " + std::to_string(door.row) + ")";
  481. ret += " | (" + viewVariable(agentName, 1, true) + "x" + agentName + " = " + std::to_string(door.column) + " & y" + agentName + " + 1 = " + std::to_string(door.row) + ")";
  482. ret += " | (" + viewVariable(agentName, 2, true) + "x" + agentName + "- 1 = " + std::to_string(door.column) + " & y" + agentName + "= " + std::to_string(door.row) + ")";
  483. ret += " | (" + viewVariable(agentName, 3, true) + "x" + agentName + " = " + std::to_string(door.column) + " & y" + agentName + " - 1 = " + std::to_string(door.row) + "))";
  484. return ret;
  485. }
  486. std::ostream& PrismModulesPrinter::printConfiguredActions(std::ostream &os, const AgentName &agentName) {
  487. for (auto& config : configuration) {
  488. if (config.type_ == ConfigType::Module && !config.overwrite_ && agentName == config.module_) {
  489. os << config.expression_ ;
  490. }
  491. }
  492. os << "\n";
  493. return os;
  494. }
  495. std::ostream& PrismModulesPrinter::printMovementActions(std::ostream &os, const AgentName &agentName, const size_t &agentIndex, const bool agentWithView, const float &probability) {
  496. if(probability >= 1) {
  497. os << "\t[" << agentName << "_move_north]" << moveGuard(agentIndex) << viewVariable(agentName, 3, agentWithView) << " !" << agentName << "IsFixed & " << " !" << agentName << "IsOnSlippery & !" << agentName << "IsInLava &!" << agentName << "IsInGoal & !" << agentName << "CannotMoveNorth -> (y" << agentName << "'=y" << agentName << "-1)" << moveUpdate(agentIndex) << ";\n";
  498. os << "\t[" << agentName << "_move_east] " << moveGuard(agentIndex) << viewVariable(agentName, 0, agentWithView) << " !" << agentName << "IsFixed & " << " !" << agentName << "IsOnSlippery & !" << agentName << "IsInLava &!" << agentName << "IsInGoal & !" << agentName << "CannotMoveEast -> (x" << agentName << "'=x" << agentName << "+1)" << moveUpdate(agentIndex) << ";\n";
  499. os << "\t[" << agentName << "_move_south]" << moveGuard(agentIndex) << viewVariable(agentName, 1, agentWithView) << " !" << agentName << "IsFixed & " << " !" << agentName << "IsOnSlippery & !" << agentName << "IsInLava &!" << agentName << "IsInGoal & !" << agentName << "CannotMoveSouth -> (y" << agentName << "'=y" << agentName << "+1)" << moveUpdate(agentIndex) << ";\n";
  500. os << "\t[" << agentName << "_move_west] " << moveGuard(agentIndex) << viewVariable(agentName, 2, agentWithView) << " !" << agentName << "IsFixed & " << " !" << agentName << "IsOnSlippery & !" << agentName << "IsInLava &!" << agentName << "IsInGoal & !" << agentName << "CannotMoveWest -> (x" << agentName << "'=x" << agentName << "-1)" << moveUpdate(agentIndex) << ";\n";
  501. } else {
  502. std::string probabilityString = std::to_string(probability);
  503. std::string percentageString = std::to_string((int)(100 * probability));
  504. std::string complementProbabilityString = std::to_string(1 - probability);
  505. os << "\t[" << agentName << "_move_north_" << percentageString << "] ";
  506. os << moveGuard(agentIndex) << viewVariable(agentName, 3, agentWithView) << " !" << agentName << "IsFixed & " << " !" << agentName << "IsOnSlippery & !" << agentName << "IsInLava & !" << agentName << "CannotMoveNorth -> ";
  507. os << probabilityString << ": (y" << agentName << "'=y" << agentName << "-1)" << moveUpdate(agentIndex) << " + ";
  508. os << complementProbabilityString << ": (y" << agentName << "'=y" << agentName << ") " << moveUpdate(agentIndex) << ";\n";
  509. os << "\t[" << agentName << "_move_east_" << percentageString << "] ";
  510. os << moveGuard(agentIndex) << viewVariable(agentName, 0, agentWithView) << " !" << agentName << "IsFixed & " << " !" << agentName << "IsOnSlippery & !" << agentName << "IsInLava & !" << agentName << "CannotMoveEast -> ";
  511. os << probabilityString << ": (x" << agentName << "'=x" << agentName << "+1)" << moveUpdate(agentIndex) << " + ";
  512. os << complementProbabilityString << ": (x" << agentName << "'=x" << agentName << ") " << moveUpdate(agentIndex) << ";\n";
  513. os << "\t[" << agentName << "_move_south_" << percentageString << "] ";
  514. os << moveGuard(agentIndex) << viewVariable(agentName, 1, agentWithView) << " !" << agentName << "IsFixed & " << " !" << agentName << "IsOnSlippery & !" << agentName << "IsInLava & !" << agentName << "CannotMoveSouth -> ";
  515. os << probabilityString << ": (y" << agentName << "'=y" << agentName << "+1)" << moveUpdate(agentIndex) << " + ";
  516. os << complementProbabilityString << ": (y" << agentName << "'=y" << agentName << ") " << moveUpdate(agentIndex) << ";\n";
  517. os << "\t[" << agentName << "_move_west_" << percentageString << "] ";
  518. os << moveGuard(agentIndex) << viewVariable(agentName, 2, agentWithView) << " !" << agentName << "IsFixed & " << " !" << agentName << "IsOnSlippery & !" << agentName << "IsInLava & !" << agentName << "CannotMoveWest -> ";
  519. os << probabilityString << ": (x" << agentName << "'=x" << agentName << "-1)" << moveUpdate(agentIndex) << " + ";
  520. os << complementProbabilityString << ": (x" << agentName << "'=x" << agentName << ") " << moveUpdate(agentIndex) << ";\n";
  521. }
  522. return os;
  523. }
  524. std::ostream& PrismModulesPrinter::printDoneActions(std::ostream &os, const AgentName &agentName, const size_t &agentIndex) {
  525. os << "\t[" << agentName << "_done]" << moveGuard(agentIndex) << agentName << "IsInGoal | " << agentName << "IsInLava -> (" << agentName << "Done'=true);\n";
  526. return os;
  527. }
  528. std::ostream& PrismModulesPrinter::printSlipperyTurn(std::ostream &os, const AgentName &agentName, const size_t &agentIndex, const coordinates &c, std::set<std::string> &slipperyActions, const std::array<bool, 8>& neighborhood, SlipperyType orientation) {
  529. constexpr std::size_t PROB_PIECES = 9, ALL_POSS_DIRECTIONS = 9;
  530. std::array<std::string, ALL_POSS_DIRECTIONS> positionTransition = {
  531. /* north */ "(y" + agentName + "'=y" + agentName + "-1)",
  532. /* north east */ "(x" + agentName + "'=x" + agentName + "+1) & (y" + agentName + "'=y" + agentName + "-1)",
  533. /* east */ "(x" + agentName + "'=x" + agentName + "+1)",
  534. /* east south */ "(x" + agentName + "'=x" + agentName + "+1) & (y" + agentName + "'=y" + agentName + "+1)",
  535. /* south */ "(y" + agentName + "'=y" + agentName + "+1)",
  536. /* south west */ "(x" + agentName + "'=x" + agentName + "-1) & (y" + agentName + "'=y" + agentName + "+1)",
  537. /* west */ "(x" + agentName + "'=x" + agentName + "-1)",
  538. /* north west */ "(x" + agentName + "'=x" + agentName + "-1) & (y" + agentName + "'=y" + agentName + "-1)",
  539. /* own position */ "(x" + agentName + "'=x" + agentName + ") & (y" + agentName + "'=y" + agentName + ")"
  540. };
  541. // view transition appdx in form (guard, update part)
  542. // IMPORTANT: No mod() usage for turn left due to bug in mod() function for decrement
  543. std::array<std::tuple<std::string, std::string, std::string>, 3> viewTransition = {
  544. std::make_tuple(" & " + agentName + "SlipperyTurnRightAllowed ", " & (view" + agentName + "'=mod(view" + agentName + " + 1, 4))", "_right]"),
  545. std::make_tuple(" & " + agentName + "SlipperyTurnLeftAllowed & view" + agentName + ">0", " & (view" + agentName + "'=view" + agentName + " - 1)", "_left]"),
  546. std::make_tuple(" & " + agentName + "SlipperyTurnLeftAllowed & view" + agentName + "=0", " & (view" + agentName + "'=3)", "_left]")
  547. };
  548. // direction specifics
  549. std::string actionName;
  550. std::size_t remainPosIndex = 8;
  551. std::array<std::size_t, ALL_POSS_DIRECTIONS> prob_piece_dir; // { n, ne, w, se, s, sw, w, nw, CURRENT POS }
  552. switch (orientation)
  553. {
  554. case SlipperyType::North:
  555. actionName = "\t[" + agentName + "turn_at_slip_north";
  556. prob_piece_dir = { 0, 0, 0, 1, 1, 1, 0, 0, 0 /* <- R */ };
  557. break;
  558. case SlipperyType::South:
  559. actionName = "\t[" + agentName + "turn_at_slip_south";
  560. prob_piece_dir = { 1, 1, 0, 0, 0, 0, 0, 1, 0 /* <- R */ };
  561. break;
  562. case SlipperyType::East:
  563. actionName = "\t[" + agentName + "turn_at_slip_east";
  564. prob_piece_dir = { 0, 0, 0, 0, 0, 1, 1, 1, 0 /* <- R */ };
  565. break;
  566. case SlipperyType::West:
  567. actionName = "\t[" + agentName + "turn_at_slip_west";
  568. prob_piece_dir = { 0, 1, 1, 1, 0, 0, 0, 0, 0 /* <- R */ };
  569. break;
  570. }
  571. slipperyActions.insert(actionName);
  572. // override probability to 0 if corresp. direction is blocked
  573. for (std::size_t i = 0; i < ALL_POSS_DIRECTIONS - 1; i++) {
  574. if (!neighborhood.at(i))
  575. prob_piece_dir.at(i) = 0;
  576. }
  577. // determine residual probability (R) by replacing 0 with (1 - overall sum)
  578. prob_piece_dir.at(remainPosIndex) = PROB_PIECES - std::accumulate(prob_piece_dir.begin(), prob_piece_dir.end(), 0);
  579. // <DEBUG_AREA>
  580. {
  581. assert(prob_piece_dir.at(remainPosIndex) <= 9 && prob_piece_dir.at(remainPosIndex) >= 6 && "Value not in Range!");
  582. assert(std::accumulate(prob_piece_dir.begin(), prob_piece_dir.end(), 0) == PROB_PIECES && "Does not sum up to 1!");
  583. }
  584. // </DEBUG_AREA>
  585. // generic output (for every view transition)
  586. for (std::size_t v = 0; v < viewTransition.size(); v++) {
  587. os << actionName << std::get<2>(viewTransition.at(v)) << moveGuard(agentIndex) << " x" << agentName << "=" << c.second << " & y" << agentName << "=" << c.first << std::get<0>(viewTransition.at(v));
  588. for (std::size_t i = 0; i < ALL_POSS_DIRECTIONS; i++) {
  589. os << (i == 0 ? " -> " : " + ") << prob_piece_dir.at(i) << "/" << PROB_PIECES << " : " << positionTransition.at(i) << std::get<1>(viewTransition.at(v)) << moveUpdate(agentIndex) << (i == ALL_POSS_DIRECTIONS - 1 ? ";\n" : "\n");
  590. }
  591. }
  592. return os;
  593. }
  594. std::ostream& PrismModulesPrinter::printSlipperyMove(std::ostream &os, const AgentName &agentName, const size_t &agentIndex, const coordinates &c, std::set<std::string> &slipperyActions, const std::array<bool, 8>& neighborhood, SlipperyType orientation) {
  595. constexpr std::size_t PROB_PIECES = 9, ALL_POSS_DIRECTIONS = 8;
  596. std::array<std::string, ALL_POSS_DIRECTIONS> positionTransition = {
  597. /* north */ "(y" + agentName + "'=y" + agentName + "-1)",
  598. /* north east */ "(x" + agentName + "'=x" + agentName + "+1) & (y" + agentName + "'=y" + agentName + "-1)",
  599. /* east */ "(x" + agentName + "'=x" + agentName + "+1)",
  600. /* east south */ "(x" + agentName + "'=x" + agentName + "+1) & (y" + agentName + "'=y" + agentName + "+1)",
  601. /* south */ "(y" + agentName + "'=y" + agentName + "+1)",
  602. /* south west */ "(x" + agentName + "'=x" + agentName + "-1) & (y" + agentName + "'=y" + agentName + "+1)",
  603. /* west */ "(x" + agentName + "'=x" + agentName + "-1)",
  604. /* north west */ "(x" + agentName + "'=x" + agentName + "-1) & (y" + agentName + "'=y" + agentName + "-1)"
  605. };
  606. // direction specifics
  607. std::size_t straightPosIndex;
  608. std::string actionName, specialTransition; // if straight ahead is blocked
  609. std::array<std::size_t, ALL_POSS_DIRECTIONS> prob_piece_dir; // { n, ne, w, se, s, sw, w, nw }
  610. switch (orientation)
  611. {
  612. case SlipperyType::North:
  613. actionName = "\t[" + agentName + "move_on_slip_north]";
  614. prob_piece_dir = { 0, 0, 1, 2, 0 /* <- R */, 2, 1, 0 };
  615. straightPosIndex = 4;
  616. specialTransition = "(y" + agentName + "'=y" + agentName + (!neighborhood.at(straightPosIndex) ? ")" : "+1)");
  617. break;
  618. case SlipperyType::South:
  619. actionName = "\t[" + agentName + "move_on_slip_south]";
  620. prob_piece_dir = { 0 /* <- R */, 2, 1, 0, 0, 0, 1, 2 };
  621. straightPosIndex = 0; // always north
  622. specialTransition = "(y" + agentName + "'=y" + agentName + (!neighborhood.at(straightPosIndex) ? ")" : "-1)");
  623. break;
  624. case SlipperyType::East:
  625. actionName = "\t[" + agentName + "move_on_slip_east]";
  626. prob_piece_dir = { 1, 0, 0, 0, 1, 2, 0 /* <- R */, 2 };
  627. straightPosIndex = 6;
  628. specialTransition = "(x" + agentName + "'=x" + agentName + (!neighborhood.at(straightPosIndex) ? ")" : "-1)");
  629. break;
  630. case SlipperyType::West:
  631. actionName = "\t[" + agentName + "move_on_slip_west]";
  632. prob_piece_dir = { 1, 2, 0 /* <- R */, 2, 1, 0, 0, 0 };
  633. straightPosIndex = 2;
  634. specialTransition = "(x" + agentName + "'=x" + agentName + (!neighborhood.at(straightPosIndex) ? ")" : "+1)");
  635. break;
  636. }
  637. slipperyActions.insert(actionName);
  638. // override probability to 0 if corresp. direction is blocked
  639. for (std::size_t i = 0; i < ALL_POSS_DIRECTIONS; i++) {
  640. if (!neighborhood.at(i))
  641. prob_piece_dir.at(i) = 0;
  642. }
  643. // determine residual probability (R) by replacing 0 with (1 - overall sum)
  644. if(enforceOneWays) {
  645. prob_piece_dir = {0,0,0,0,0,0,0,0};
  646. }
  647. prob_piece_dir.at(straightPosIndex) = PROB_PIECES - std::accumulate(prob_piece_dir.begin(), prob_piece_dir.end(), 0);
  648. // <DEBUG_AREA>
  649. {
  650. assert(prob_piece_dir.at(straightPosIndex) <= 9 && prob_piece_dir.at(straightPosIndex) >= 3 && "Value not in Range!");
  651. assert(std::accumulate(prob_piece_dir.begin(), prob_piece_dir.end(), 0) == PROB_PIECES && "Does not sum up to 1!");
  652. assert(orientation != SlipperyType::North || (prob_piece_dir.at(7) == 0 && prob_piece_dir.at(0) == 0 && prob_piece_dir.at(1) == 0 && "Slippery up should be impossible!"));
  653. assert(orientation != SlipperyType::South || (prob_piece_dir.at(3) == 0 && prob_piece_dir.at(4) == 0 && prob_piece_dir.at(5) == 0 && "Slippery down should be impossible!"));
  654. assert(orientation != SlipperyType::East || (prob_piece_dir.at(1) == 0 && prob_piece_dir.at(2) == 0 && prob_piece_dir.at(3) == 0 && "Slippery right should be impossible!"));
  655. assert(orientation != SlipperyType::West || (prob_piece_dir.at(5) == 0 && prob_piece_dir.at(6) == 0 && prob_piece_dir.at(7) == 0 && "Slippery left should be impossible!"));
  656. }
  657. // </DEBUG_AREA>
  658. // special case: straight forward is blocked (then remain in same position)
  659. positionTransition.at(straightPosIndex) = specialTransition;
  660. // generic output (for every view and every possible view direction)
  661. os << actionName << moveGuard(agentIndex) << " x" << agentName << "=" << c.second << " & y" << agentName << "=" << c.first << " & " << agentName << "SlipperyMoveForwardAllowed ";
  662. for (std::size_t i = 0; i < ALL_POSS_DIRECTIONS; i++) {
  663. os << (i == 0 ? " -> " : " + ") << prob_piece_dir.at(i) << "/" << PROB_PIECES << " : " << positionTransition.at(i) << moveUpdate(agentIndex) << (i == ALL_POSS_DIRECTIONS - 1 ? ";\n" : "\n");
  664. }
  665. return os;
  666. }
  667. std::ostream& PrismModulesPrinter::printEndmodule(std::ostream &os) {
  668. os << "endmodule\n";
  669. os << "\n";
  670. return os;
  671. }
  672. std::ostream& PrismModulesPrinter::printPlayerStruct(std::ostream &os, const AgentName &agentName, const bool agentWithView, const std::vector<float> &probabilities, const std::set<std::string> &slipperyActions) {
  673. os << "player " << agentName << "\n\t";
  674. bool first = true;
  675. std::list<std::string> allActions = { "_move_north", "_move_east", "_move_south", "_move_west" };
  676. std::list<std::string> movementActions = allActions;
  677. for(auto const& probability : probabilities) {
  678. std::string percentageString = std::to_string((int)(100 * probability));
  679. for(auto const& movement : movementActions) {
  680. allActions.push_back(movement + "_" + percentageString);
  681. }
  682. }
  683. if(agentWithView) {
  684. allActions.push_back("_turn_left");
  685. allActions.push_back("_turn_right");
  686. } else {
  687. allActions.push_back("_turns");
  688. }
  689. for(auto const& action : allActions) {
  690. if(first) first = false; else os << ", ";
  691. os << "[" << agentName << action << "]";
  692. }
  693. for(auto const& action : slipperyActions) {
  694. os << ", " << action;
  695. }
  696. os << "\nendplayer\n";
  697. return os;
  698. }
  699. std::ostream& PrismModulesPrinter::printGlobalMoveVariable(std::ostream &os, const size_t &numberOfPlayer) {
  700. os << "\nglobal move : [0.." << std::to_string(numberOfPlayer - 1) << "] init 0;\n\n";
  701. return os;
  702. }
  703. std::ostream& PrismModulesPrinter::printRewards(std::ostream &os, const AgentName &agentName, const std::map<coordinates, float> &stateRewards, const cells &lava, const cells &goals, const std::map<Color, cells> &backgroundTiles) {
  704. if(lava.size() != 0) {
  705. os << "rewards \"" << agentName << "SafetyNoBFS\"\n";
  706. os << "\t" <<agentName << "IsInLavaAndNotDone: -100;\n";
  707. os << "endrewards\n";
  708. }
  709. if (!goals.empty() || !lava.empty()) {
  710. os << "rewards \"" << agentName << "SafetyNoBFSAndGoal\"\n";
  711. if(goals.size() != 0) os << "\t" << agentName << "IsInGoalAndNotDone: 100;\n";
  712. if(lava.size() != 0) os << "\t" << agentName << "IsInLavaAndNotDone: -100;\n";
  713. os << "endrewards\n";
  714. }
  715. os << "rewards \"" << agentName << "Time\"\n";
  716. os << "\t!" << agentName << "IsInGoal : -1;\n";
  717. if(goals.size() != 0) os << "\t" << agentName << "IsInGoalAndNotDone: 100;\n";
  718. if(lava.size() != 0) os << "\t" << agentName << "IsInLavaAndNotDone: -100;\n";
  719. os << "endrewards\n";
  720. if(stateRewards.size() > 0) {
  721. os << "rewards \"" << agentName << "SafetyWithBFS\"\n";
  722. if(lava.size() != 0) os << "\t" << agentName << "IsInLavaAndNotDone: -100;\n";
  723. for(auto const [coordinates, reward] : stateRewards) {
  724. os << "\txAgent=" << coordinates.first << "&yAgent=" << coordinates.second << " : " << reward << ";\n";
  725. }
  726. os << "endrewards\n";
  727. }
  728. if(stateRewards.size() > 0) {
  729. os << "rewards \"" << agentName << "SafetyWithBFSAndGoal\"\n";
  730. if(goals.size() != 0) os << "\tAgentIsInGoalAndNotDone: 100;\n";
  731. if(lava.size() != 0) os << "\tAgentIsInLavaAndNotDone: -100;\n";
  732. for(auto const [coordinates, reward] : stateRewards) {
  733. os << "\txAgent=" << coordinates.first << "&yAgent=" << coordinates.second << " : " << reward << ";\n";
  734. }
  735. os << "endrewards\n";
  736. }
  737. for(auto const entry : backgroundTiles)
  738. {
  739. std::cout << getColor(entry.first) << " ";
  740. for(auto const cell : entry.second){
  741. std::cout << cell.getCoordinates().first << " " << cell.getCoordinates().second << std::endl;
  742. }
  743. }
  744. if(backgroundTiles.size() > 0) {
  745. os << "rewards \"TaxiReward\"\n";
  746. os << "\t!AgentIsInGoal : -1;\n";
  747. std::string allPassengersPickedUp = "";
  748. bool first = true;
  749. for(auto const [color, cells] : backgroundTiles) {
  750. if(cells.size() == 0) continue;
  751. if(first) first = false; else allPassengersPickedUp += "&";
  752. std::string c = getColor(color);
  753. c.at(0) = std::toupper(c.at(0));
  754. std::string visitedLabel = agentName + "_picked_up_" + c;
  755. allPassengersPickedUp += visitedLabel;
  756. os << "[" << agentName << "_pickup_" << c << "] true : 100;\n";
  757. }
  758. if(goals.size() != 0) os << "\tAgentIsInGoalAndNotDone & " << allPassengersPickedUp << " : 100;\n";
  759. if(goals.size() != 0) os << "\tAgentIsInGoalAndNotDone & !(" << allPassengersPickedUp << ") : -100;\n";
  760. os << "endrewards";
  761. }
  762. return os;
  763. }
  764. std::string PrismModulesPrinter::moveGuard(const size_t &agentIndex) {
  765. return isGame() ? " move=" + std::to_string(agentIndex) + " & " : " ";
  766. }
  767. std::string PrismModulesPrinter::moveUpdate(const size_t &agentIndex) {
  768. return isGame() ?
  769. (agentIndex == numberOfPlayer - 1) ?
  770. " & (move'=0) " :
  771. " & (move'=" + std::to_string(agentIndex + 1) + ") " :
  772. "";
  773. }
  774. std::string PrismModulesPrinter::viewVariable(const AgentName &agentName, const size_t &agentDirection, const bool agentWithView) {
  775. return agentWithView ? " view" + agentName + "=" + std::to_string(agentDirection) + " & " : " ";
  776. }
  777. bool PrismModulesPrinter::isGame() const {
  778. return modelType == ModelType::SMG;
  779. }
  780. }