commit
						85d0cbc81f
					
				 15 changed files with 2937 additions and 0 deletions
			
			
		- 
					34.gitignore
- 
					22CMakeLists.txt
- 
					171main.cpp
- 
					8util/CMakeLists.txt
- 
					229util/Grid.cpp
- 
					63util/Grid.h
- 
					102util/MinigridGrammar.h
- 
					32util/OptionParser.cpp
- 
					13util/OptionParser.h
- 
					681util/PrismModulesPrinter.cpp
- 
					91util/PrismModulesPrinter.h
- 
					15util/PrismPrinter.h
- 
					83util/cell.cpp
- 
					62util/cell.h
- 
					1331util/popl.hpp
| @ -0,0 +1,34 @@ | |||
| # Prerequisites | |||
| *.d | |||
| 
 | |||
| # Compiled Object files | |||
| *.slo | |||
| *.lo | |||
| *.o | |||
| *.obj | |||
| 
 | |||
| # Precompiled Headers | |||
| *.gch | |||
| *.pch | |||
| 
 | |||
| # Compiled Dynamic libraries | |||
| *.so | |||
| *.dylib | |||
| *.dll | |||
| 
 | |||
| # Fortran module files | |||
| *.mod | |||
| *.smod | |||
| 
 | |||
| # Compiled Static libraries | |||
| *.lai | |||
| *.la | |||
| *.a | |||
| *.lib | |||
| 
 | |||
| # Executables | |||
| *.exe | |||
| *.out | |||
| *.app | |||
| 
 | |||
| build/* | |||
| @ -0,0 +1,22 @@ | |||
| include(util/CMakeLists.txt) | |||
| 
 | |||
| set(CMAKE_CXX_STANDARD 20) | |||
| 
 | |||
| set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -D__FILENAME__='\"$(subst ${CMAKE_SOURCE_DIR}/,,$(abspath $<))\"'") | |||
| add_definitions(-DLOG_DEBUG) | |||
| 
 | |||
| cmake_minimum_required(VERSION 3.0...3.22) | |||
| 
 | |||
| set(CMAKE_BUILD_TYPE Debug) | |||
| 
 | |||
| project( | |||
|   Minigrid2PRISM | |||
|   VERSION 0.1 | |||
|   LANGUAGES CXX) | |||
| 
 | |||
| add_executable(main | |||
|                ${SRCS} | |||
|                main.cpp | |||
|                ) | |||
| 
 | |||
| target_link_libraries(main pthread) | |||
| @ -0,0 +1,171 @@ | |||
| #include "util/OptionParser.h"
 | |||
| #include "util/MinigridGrammar.h"
 | |||
| #include "util/Grid.h"
 | |||
| 
 | |||
| #include <iostream>
 | |||
| #include <fstream>
 | |||
| #include <filesystem>
 | |||
| #include <sstream>
 | |||
| 
 | |||
| std::vector<std::string> parseCommaSeparatedString(std::string const& str) { | |||
|   std::vector<std::string> result; | |||
|   std::stringstream stream(str); | |||
|   while(stream.good()) { | |||
|     std::string substr; | |||
|     getline(stream, substr, ','); | |||
|     substr.at(0) = std::toupper(substr.at(0)); | |||
|     result.push_back(substr); | |||
|   } | |||
|   return result; | |||
| } | |||
| 
 | |||
| struct printer { | |||
|     typedef boost::spirit::utf8_string string; | |||
| 
 | |||
|     void element(string const& tag, string const& value, int depth) const { | |||
|         for (int i = 0; i < (depth*4); ++i) std::cout << ' '; | |||
| 
 | |||
|         std::cout << "tag: " << tag; | |||
|         if (value != "") std::cout << ", value: " << value; | |||
|         std::cout << std::endl; | |||
|     } | |||
| }; | |||
| 
 | |||
| void print_info(boost::spirit::info const& what) { | |||
|   using boost::spirit::basic_info_walker; | |||
| 
 | |||
|   printer pr; | |||
|   basic_info_walker<printer> walker(pr, what.tag, 0); | |||
|   boost::apply_visitor(walker, what.value); | |||
| } | |||
| 
 | |||
| int main(int argc, char* argv[]) { | |||
|   popl::OptionParser optionParser("Allowed options"); | |||
| 
 | |||
|   auto helpOption = optionParser.add<popl::Switch>("h", "help", "Print this help message."); | |||
|   auto inputFilename = optionParser.add<popl::Value<std::string>>("i", "input-file", "Filename of the input file."); | |||
|   auto outputFilename = optionParser.add<popl::Value<std::string>>("o", "output-file", "Filename for the output file."); | |||
| 
 | |||
|   auto agentsToBeConsidered = optionParser.add<popl::Value<std::string>, popl::Attribute::optional>("a", "agents", "Which parsed agents should be considered in the output. WIP."); | |||
|   auto viewForAgents = optionParser.add<popl::Value<std::string>, popl::Attribute::optional>("v", "view", "Agents for which the 'view'('direction') variable should be included. WIP."); | |||
| 
 | |||
|   auto probabilisticBehaviourAgents = optionParser.add<popl::Value<std::string>, popl::Attribute::optional>("p", "prob-beh", "Agents for which we want to include probabilistic actions"); | |||
|   auto probabilities = optionParser.add<popl::Value<std::string>, popl::Attribute::optional>("q", "probs", "The probabilities for which probabilistic actions should be added. WIP"); | |||
| 
 | |||
|   auto enforceOneWays = optionParser.add<popl::Switch>("f", "force-oneways", "Enforce encoding of oneways. This entails that slippery tiles do not allow turning and have prob 1 to shift the agent by one and makes turning impossible if in a one tile wide section of the grid."); | |||
| 
 | |||
|   try { | |||
|     optionParser.parse(argc, argv); | |||
| 
 | |||
|     if(helpOption->count() > 0) { | |||
|       std::cout << optionParser << std::endl; | |||
|       return EXIT_SUCCESS; | |||
|     } | |||
|   } catch (const popl::invalid_option &e) { | |||
|     return io::printPoplException(e); | |||
|   } catch (const std::exception &e) { | |||
| 		std::cerr << "Exception: " << e.what() << "\n"; | |||
| 		return EXIT_FAILURE; | |||
| 	} | |||
| 
 | |||
|   GridOptions gridOptions = { {}, {} }; | |||
|   if(agentsToBeConsidered->is_set()) { | |||
|     gridOptions.agentsToBeConsidered = parseCommaSeparatedString(agentsToBeConsidered->value(0)); | |||
|   } | |||
|   if(viewForAgents->is_set()) { | |||
|     gridOptions.agentsWithView = parseCommaSeparatedString(viewForAgents->value(0)); | |||
|   } | |||
|   if(enforceOneWays->is_set()) { | |||
|     gridOptions.enforceOneWays = true; | |||
|   } | |||
|   if(probabilisticBehaviourAgents->is_set()) { | |||
|     gridOptions.agentsWithProbabilisticBehaviour = parseCommaSeparatedString(probabilisticBehaviourAgents->value(0)); | |||
|     for(auto const& a : gridOptions.agentsWithProbabilisticBehaviour) { | |||
|       std::cout << a << std::endl; | |||
|     } | |||
|     if(probabilities->is_set()) { | |||
|       std::vector<std::string> parsedStrings = parseCommaSeparatedString(probabilities->value(0)); | |||
| 
 | |||
|       std::transform(parsedStrings.begin(), parsedStrings.end(), std::back_inserter(gridOptions.probabilitiesForActions), [](const std::string& string) { | |||
|         return std::stof(string); | |||
|       }); | |||
|       for(auto const& a : gridOptions.probabilitiesForActions) { | |||
|         std::cout << a << std::endl; | |||
|       } | |||
|     } else { | |||
|       throw std::logic_error{ "When adding agents with probabilistic behaviour, you also need to specify a list of probabilities via --probs." }; | |||
|     } | |||
|   } | |||
| 
 | |||
| 
 | |||
|   std::fstream file {outputFilename->value(0), file.trunc | file.out}; | |||
|   std::fstream infile {inputFilename->value(0), infile.in}; | |||
|   std::string line, content, background, rewards; | |||
|   std::cout << "\n"; | |||
|   bool parsingBackground = false; | |||
|   bool parsingStateRewards = false; | |||
|   while (std::getline(infile, line) && !line.empty()) { | |||
|     if(line.at(0) == '-' && line.at(line.size() - 1) == '-' && parsingBackground) { | |||
|       parsingStateRewards = true; | |||
|       parsingBackground = false; | |||
|       continue; | |||
|     } else if(line.at(0) == '-' && line.at(line.size() - 1) == '-') { | |||
|       parsingBackground = true; | |||
|       continue; | |||
|     } | |||
|     if(!parsingBackground && !parsingStateRewards) { | |||
|       std::cout << "Reading   :\t" << line << "\n"; | |||
|       content += line + "\n"; | |||
|     } else if (parsingBackground) { | |||
|       std::cout << "Background:\t" << line << "\n"; | |||
|       background += line + "\n"; | |||
|     } else if(parsingStateRewards) { | |||
|       rewards += line + "\n"; | |||
|     } | |||
|   } | |||
|   std::cout << "\n"; | |||
| 
 | |||
| 
 | |||
|   pos_iterator_t contentFirst(content.begin()); | |||
|   pos_iterator_t contentIter = contentFirst; | |||
|   pos_iterator_t contentLast(content.end()); | |||
|   MinigridParser<pos_iterator_t> contentParser(contentFirst); | |||
|   pos_iterator_t backgroundFirst(background.begin()); | |||
|   pos_iterator_t backgroundIter = backgroundFirst; | |||
|   pos_iterator_t backgroundLast(background.end()); | |||
|   MinigridParser<pos_iterator_t> backgroundParser(backgroundFirst); | |||
| 
 | |||
|   cells contentCells; | |||
|   cells backgroundCells; | |||
|   std::map<coordinates, float> stateRewards; | |||
|   try { | |||
|     bool ok = phrase_parse(contentIter, contentLast, contentParser, qi::space, contentCells); | |||
|     // TODO if(background is not empty) {
 | |||
|     ok     &= phrase_parse(backgroundIter, backgroundLast, backgroundParser, qi::space, backgroundCells); | |||
|     // TODO }
 | |||
| 
 | |||
|     boost::escaped_list_separator<char> seps('\\', ';', '\n'); | |||
|     Tokenizer csvParser(rewards, seps); | |||
|     for(auto iter = csvParser.begin(); iter != csvParser.end(); ++iter) { | |||
|       int x = std::stoi(*iter); | |||
|       int y = std::stoi(*(++iter)); | |||
|       float reward = std::stof(*(++iter)); | |||
|       stateRewards[std::make_pair(x,y)] = reward; | |||
|     } | |||
|     if(ok) { | |||
|       Grid grid(contentCells, backgroundCells, gridOptions, stateRewards); | |||
|       //grid.printToPrism(std::cout, prism::ModelType::MDP);
 | |||
|       grid.printToPrism(file, prism::ModelType::MDP); | |||
|     } | |||
|   } catch(qi::expectation_failure<pos_iterator_t> const& e) { | |||
|     std::cout << "expected: "; print_info(e.what_); | |||
|     std::cout << "got: \"" << std::string(e.first, e.last) << '"' << std::endl; | |||
|     std::cout << "Expectation failure: " << e.what() << " at '" << std::string(e.first,e.last) << "'\n"; | |||
|   } catch(const std::exception& e) { | |||
|     std::cerr << "Exception '" << typeid(e).name() << "' caught:" << std::endl; | |||
|     std::cerr << "\t" << e.what() << std::endl; | |||
|     std::exit(EXIT_FAILURE); | |||
|   } | |||
| 
 | |||
|   return 0; | |||
| } | |||
| @ -0,0 +1,8 @@ | |||
| list(APPEND SRCS | |||
|   ${CMAKE_CURRENT_LIST_DIR}/cell.cpp | |||
|   ${CMAKE_CURRENT_LIST_DIR}/MinigridGrammar.h | |||
|   ${CMAKE_CURRENT_LIST_DIR}/Grid.cpp | |||
|   ${CMAKE_CURRENT_LIST_DIR}/PrismModulesPrinter.cpp | |||
|   ${CMAKE_CURRENT_LIST_DIR}/popl.hpp | |||
|   ${CMAKE_CURRENT_LIST_DIR}/OptionParser.cpp | |||
| ) | |||
| @ -0,0 +1,229 @@ | |||
| #include "Grid.h"
 | |||
| 
 | |||
| #include <algorithm>
 | |||
| 
 | |||
| Grid::Grid(cells gridCells, cells background, const GridOptions &gridOptions, const std::map<coordinates, float> &stateRewards) | |||
|   : allGridCells(gridCells), background(background), gridOptions(gridOptions), stateRewards(stateRewards) | |||
| { | |||
|   cell max = allGridCells.at(allGridCells.size() - 1); | |||
|   maxBoundaries = std::make_pair(max.row - 1, max.column - 1); | |||
|   std::copy_if(gridCells.begin(), gridCells.end(), std::back_inserter(walls), [](cell c) { | |||
|       return c.type == Type::Wall; | |||
|   }); | |||
|   std::copy_if(gridCells.begin(), gridCells.end(), std::back_inserter(lava), [](cell c) { | |||
|       return c.type == Type::Lava; | |||
|   }); | |||
|   std::copy_if(gridCells.begin(), gridCells.end(), std::back_inserter(floor), [](cell c) { | |||
|       return c.type == Type::Floor; | |||
|   }); | |||
|   std::copy_if(background.begin(), background.end(), std::back_inserter(slipperyNorth), [](cell c) { | |||
|       return c.type == Type::SlipperyNorth; | |||
|   }); | |||
|   std::copy_if(background.begin(), background.end(), std::back_inserter(slipperyEast), [](cell c) { | |||
|       return c.type == Type::SlipperyEast; | |||
|   }); | |||
|   std::copy_if(background.begin(), background.end(), std::back_inserter(slipperySouth), [](cell c) { | |||
|       return c.type == Type::SlipperySouth; | |||
|   }); | |||
|   std::copy_if(background.begin(), background.end(), std::back_inserter(slipperyWest), [](cell c) { | |||
|       return c.type == Type::SlipperyWest; | |||
|   }); | |||
|   std::copy_if(gridCells.begin(), gridCells.end(), std::back_inserter(lockedDoors), [](cell c) { | |||
|       return c.type == Type::LockedDoor; | |||
|   }); | |||
|   std::copy_if(gridCells.begin(), gridCells.end(), std::back_inserter(goals), [](cell c) { | |||
|       return c.type == Type::Goal; | |||
|   }); | |||
|   std::copy_if(gridCells.begin(), gridCells.end(), std::back_inserter(keys), [](cell c) { | |||
|       return c.type == Type::Key; | |||
|   }); | |||
|   std::copy_if(gridCells.begin(), gridCells.end(), std::back_inserter(boxes), [](cell c) { | |||
|       return c.type == Type::Box; | |||
|   }); | |||
|   agent = *std::find_if(gridCells.begin(), gridCells.end(), [](cell c) { | |||
|       return c.type == Type::Agent; | |||
|   }); | |||
|   std::copy_if(gridCells.begin(), gridCells.end(), std::back_inserter(adversaries), [](cell c) { | |||
|       return c.type == Type::Adversary; | |||
|   }); | |||
|   agentNameAndPositionMap.insert({ "Agent", agent.getCoordinates() }); | |||
|   for(auto const& adversary : adversaries) { | |||
|     std::string color = adversary.getColor(); | |||
|     color.at(0) = std::toupper(color.at(0)); | |||
|     try { | |||
|       if(gridOptions.agentsToBeConsidered.size() != 0 && std::find(gridOptions.agentsToBeConsidered.begin(), gridOptions.agentsToBeConsidered.end(), color) == gridOptions.agentsToBeConsidered.end()) continue; | |||
|       auto success = agentNameAndPositionMap.insert({ color, adversary.getCoordinates() }); | |||
|       if(!success.second) { | |||
|         throw std::logic_error("Agent with " + color + " already present\n"); | |||
|       } | |||
|     } catch(const std::logic_error& e) { | |||
|       std::cerr << "Expected agents colors to be different. Agent with color : '" << color << "' already present." << std::endl; | |||
|       throw; | |||
|     } | |||
|   } | |||
|   for(auto const& color : allColors) { | |||
|     cells cellsOfColor; | |||
|     std::copy_if(background.begin(), background.end(), std::back_inserter(cellsOfColor), [&color](cell c) { | |||
|         return c.type == Type::Floor && c.color == color; | |||
|     }); | |||
|     if(cellsOfColor.size() > 0) { | |||
|       backgroundTiles.emplace(color, cellsOfColor); | |||
|     } | |||
|   } | |||
| } | |||
| 
 | |||
| std::ostream& operator<<(std::ostream& os, const Grid& grid) { | |||
|   int lastRow = 1; | |||
|   for(auto const& cell : grid.allGridCells) { | |||
|     if(lastRow != cell.row) | |||
|       os << std::endl; | |||
|     os << static_cast<char>(cell.type) << static_cast<char>(cell.color); | |||
|     lastRow = cell.row; | |||
|   } | |||
|   return os; | |||
| } | |||
| 
 | |||
| cells Grid::getGridCells() { | |||
|   return allGridCells; | |||
| } | |||
| 
 | |||
| bool Grid::isBlocked(coordinates p) { | |||
|   return isWall(p) || isLockedDoor(p) || isKey(p); | |||
| } | |||
| 
 | |||
| bool Grid::isWall(coordinates p) { | |||
|   return std::find_if(walls.begin(), walls.end(), | |||
|       [p](cell cell) { | |||
|         return cell.row == p.first && cell.column == p.second; | |||
|       }) != walls.end(); | |||
| } | |||
| 
 | |||
| bool Grid::isLockedDoor(coordinates p) { | |||
|   return std::find_if(lockedDoors.begin(), lockedDoors.end(), | |||
|       [p](cell cell) { | |||
|         return cell.row == p.first && cell.column == p.second; | |||
|       }) != lockedDoors.end(); | |||
| } | |||
| 
 | |||
| bool Grid::isKey(coordinates p) { | |||
|   return std::find_if(keys.begin(), keys.end(), | |||
|       [p](cell cell) { | |||
|         return cell.row == p.first && cell.column == p.second; | |||
|       }) != keys.end(); | |||
| } | |||
| 
 | |||
| bool Grid::isBox(coordinates p) { | |||
|   return std::find_if(boxes.begin(), boxes.end(), | |||
|       [p](cell cell) { | |||
|         return cell.row == p.first && cell.column == p.second; | |||
|       }) != boxes.end(); | |||
| } | |||
| 
 | |||
| void Grid::printToPrism(std::ostream& os, const prism::ModelType& modelType) { | |||
|   cells northRestriction; | |||
|   cells eastRestriction; | |||
|   cells southRestriction; | |||
|   cells westRestriction; | |||
|   cells walkable = floor; | |||
|   walkable.insert(walkable.end(), goals.begin(), goals.end()); | |||
|   walkable.insert(walkable.end(), boxes.begin(), boxes.end()); | |||
|   walkable.push_back(agent); | |||
|   walkable.insert(walkable.end(), adversaries.begin(), adversaries.end()); | |||
|   walkable.insert(walkable.end(), lava.begin(), lava.end()); | |||
|   for(auto const& c : walkable) { | |||
|     if(isBlocked(c.getNorth())) northRestriction.push_back(c); | |||
|     if(isBlocked(c.getEast()))   eastRestriction.push_back(c); | |||
|     if(isBlocked(c.getSouth())) southRestriction.push_back(c); | |||
|     if(isBlocked(c.getWest()))   westRestriction.push_back(c); | |||
|   } | |||
| 
 | |||
|   prism::PrismModulesPrinter printer(modelType, agentNameAndPositionMap.size(), gridOptions.enforceOneWays); | |||
|   printer.printModel(os, modelType); | |||
|   if(modelType == prism::ModelType::SMG) { | |||
|     printer.printGlobalMoveVariable(os, agentNameAndPositionMap.size()); | |||
|   } | |||
|   for(auto const &backgroundTilesOfColor : backgroundTiles) { | |||
|     for(auto agentNameAndPosition = agentNameAndPositionMap.begin(); agentNameAndPosition != agentNameAndPositionMap.end(); ++agentNameAndPosition) { | |||
|       printer.printBackgroundLabels(os, agentNameAndPosition->first, backgroundTilesOfColor); | |||
|     } | |||
|   } | |||
|   cells noTurnFloor; | |||
|   if(gridOptions.enforceOneWays) { | |||
|     for(auto const& c : floor) { | |||
|       cell east =  c.getEast(allGridCells); | |||
|       cell south = c.getSouth(allGridCells); | |||
|       cell west =  c.getWest(allGridCells); | |||
|       cell north = c.getNorth(allGridCells); | |||
|       if( (east.type == Type::Wall && west.type == Type::Wall) or | |||
|           (north.type == Type::Wall && south.type == Type::Wall) ) { | |||
|         noTurnFloor.push_back(c); | |||
|       } | |||
|     } | |||
|   } | |||
| 
 | |||
|   for(auto agentNameAndPosition = agentNameAndPositionMap.begin(); agentNameAndPosition != agentNameAndPositionMap.end(); ++agentNameAndPosition) { | |||
|     printer.printFormulas(os, agentNameAndPosition->first, northRestriction, eastRestriction, southRestriction, westRestriction, { slipperyNorth, slipperyEast, slipperySouth, slipperyWest }, lava, walls, noTurnFloor, slipperyNorth, slipperyEast, slipperySouth, slipperyWest); | |||
|     printer.printGoalLabel(os, agentNameAndPosition->first, goals); | |||
|     printer.printKeysLabels(os, agentNameAndPosition->first, keys); | |||
|   } | |||
| 
 | |||
|   std::vector<AgentName> agentNames; | |||
|   std::transform(agentNameAndPositionMap.begin(), | |||
|                  agentNameAndPositionMap.end(), | |||
|                  std::back_inserter(agentNames), | |||
|                  [](const std::map<AgentNameAndPosition::first_type,AgentNameAndPosition::second_type>::value_type &pair){return pair.first;}); | |||
|   if(modelType == prism::ModelType::SMG) { | |||
|     printer.printCrashLabel(os, agentNames); | |||
|   } | |||
|   size_t agentIndex  = 0; | |||
|   for(auto agentNameAndPosition = agentNameAndPositionMap.begin(); agentNameAndPosition != agentNameAndPositionMap.end(); ++agentNameAndPosition, agentIndex++) { | |||
|     AgentName agentName = agentNameAndPosition->first; | |||
|     bool agentWithView = std::find(gridOptions.agentsWithView.begin(), gridOptions.agentsWithView.end(), agentName) != gridOptions.agentsWithView.end(); | |||
|     bool agentWithProbabilisticBehaviour = std::find(gridOptions.agentsWithProbabilisticBehaviour.begin(), gridOptions.agentsWithProbabilisticBehaviour.end(), agentName) != gridOptions.agentsWithProbabilisticBehaviour.end(); | |||
|     std::set<std::string> slipperyActions; | |||
|     printer.printInitStruct(os, agentName); | |||
|     if(agentWithProbabilisticBehaviour) printer.printModule(os, agentName, agentIndex, maxBoundaries, agentNameAndPosition->second, keys, backgroundTiles, agentWithView, gridOptions.probabilitiesForActions); | |||
|     else                                printer.printModule(os, agentName, agentIndex, maxBoundaries, agentNameAndPosition->second, keys, backgroundTiles, agentWithView); | |||
|     for(auto const& c : slipperyNorth) { | |||
|       printer.printSlipperyMove(os, agentName, agentIndex, c.getCoordinates(), slipperyActions, getWalkableDirOf8Neighborhood(c), prism::PrismModulesPrinter::SlipperyType::North); | |||
|       if(!gridOptions.enforceOneWays) printer.printSlipperyTurn(os, agentName, agentIndex, c.getCoordinates(), slipperyActions, getWalkableDirOf8Neighborhood(c), prism::PrismModulesPrinter::SlipperyType::North); | |||
| 
 | |||
|     } | |||
|     for(auto const& c : slipperyEast) { | |||
|       printer.printSlipperyMove(os, agentName, agentIndex, c.getCoordinates(), slipperyActions, getWalkableDirOf8Neighborhood(c), prism::PrismModulesPrinter::SlipperyType::East); | |||
|       if(!gridOptions.enforceOneWays) printer.printSlipperyTurn(os, agentName, agentIndex, c.getCoordinates(), slipperyActions, getWalkableDirOf8Neighborhood(c), prism::PrismModulesPrinter::SlipperyType::East); | |||
|     } | |||
|     for(auto const& c : slipperySouth) { | |||
|       printer.printSlipperyMove(os, agentName, agentIndex, c.getCoordinates(), slipperyActions, getWalkableDirOf8Neighborhood(c), prism::PrismModulesPrinter::SlipperyType::South); | |||
|       if(!gridOptions.enforceOneWays) printer.printSlipperyTurn(os, agentName, agentIndex, c.getCoordinates(), slipperyActions, getWalkableDirOf8Neighborhood(c), prism::PrismModulesPrinter::SlipperyType::South); | |||
|     } | |||
|     for(auto const& c : slipperyWest) { | |||
|       printer.printSlipperyMove(os, agentName, agentIndex, c.getCoordinates(), slipperyActions, getWalkableDirOf8Neighborhood(c), prism::PrismModulesPrinter::SlipperyType::West); | |||
|       if(!gridOptions.enforceOneWays) printer.printSlipperyTurn(os, agentName, agentIndex, c.getCoordinates(), slipperyActions, getWalkableDirOf8Neighborhood(c), prism::PrismModulesPrinter::SlipperyType::West); | |||
|     } | |||
| 
 | |||
|     printer.printEndmodule(os); | |||
|     if(modelType == prism::ModelType::SMG) { | |||
|       if(agentWithProbabilisticBehaviour) printer.printPlayerStruct(os, agentNameAndPosition->first, agentWithView, gridOptions.probabilitiesForActions, slipperyActions); | |||
|       else                                printer.printPlayerStruct(os, agentNameAndPosition->first, agentWithView, {}, slipperyActions); | |||
|     } | |||
|     //if(!stateRewards.empty()) {
 | |||
|     printer.printRewards(os, agentName, stateRewards, lava, goals, backgroundTiles); | |||
|     //}
 | |||
|   } | |||
| } | |||
| 
 | |||
| 
 | |||
| std::array<bool, 8> Grid::getWalkableDirOf8Neighborhood(cell c) /* const */ { | |||
|   return (std::array<bool, 8>) | |||
|          { | |||
|            !isBlocked(c.getNorth()), | |||
|            !isBlocked(c.getNorth(allGridCells).getEast()), | |||
|            !isBlocked(c.getEast()), | |||
|            !isBlocked(c.getSouth(allGridCells).getEast()), | |||
|            !isBlocked(c.getSouth()), | |||
|            !isBlocked(c.getSouth(allGridCells).getWest()), | |||
|            !isBlocked(c.getWest()), | |||
|            !isBlocked(c.getNorth(allGridCells).getWest()) | |||
|          }; | |||
| } | |||
| @ -0,0 +1,63 @@ | |||
| #pragma once | |||
| 
 | |||
| #include <iostream> | |||
| #include <fstream> | |||
| #include <map> | |||
| #include <utility> | |||
| 
 | |||
| #include "MinigridGrammar.h" | |||
| #include "PrismModulesPrinter.h" | |||
| 
 | |||
| struct GridOptions { | |||
|   std::vector<AgentName> agentsToBeConsidered; | |||
|   std::vector<AgentName> agentsWithView; | |||
|   std::vector<AgentName> agentsWithProbabilisticBehaviour; | |||
|   std::vector<float>     probabilitiesForActions; | |||
|   bool                   enforceOneWays; | |||
| }; | |||
| 
 | |||
| class Grid { | |||
|   public: | |||
|     Grid(cells gridCells, cells background, const GridOptions &gridOptions, const std::map<coordinates, float> &stateRewards = {}); | |||
| 
 | |||
|     cells getGridCells(); | |||
| 
 | |||
|     bool isBlocked(coordinates p); | |||
|     bool isWall(coordinates p); | |||
|     bool isLockedDoor(coordinates p); | |||
|     bool isKey(coordinates p); | |||
|     bool isBox(coordinates p); | |||
|     void printToPrism(std::ostream &os, const prism::ModelType& modelType); | |||
| 
 | |||
|     std::array<bool, 8> getWalkableDirOf8Neighborhood(cell c); | |||
| 
 | |||
|     friend std::ostream& operator<<(std::ostream& os, const Grid &grid); | |||
| 
 | |||
|   private: | |||
|     GridOptions gridOptions; | |||
| 
 | |||
|     cells allGridCells; | |||
|     cells background; | |||
|     coordinates maxBoundaries; | |||
| 
 | |||
|     cell agent; | |||
|     cells adversaries; | |||
|     AgentNameAndPositionMap agentNameAndPositionMap; | |||
| 
 | |||
|     cells walls; | |||
|     cells floor; | |||
|     cells slipperyNorth; | |||
|     cells slipperyEast; | |||
|     cells slipperySouth; | |||
|     cells slipperyWest; | |||
|     cells lockedDoors; | |||
|     cells boxes; | |||
|     cells lava; | |||
| 
 | |||
|     cells goals; | |||
|     cells keys; | |||
| 
 | |||
|     std::map<Color, cells> backgroundTiles; | |||
| 
 | |||
|     std::map<coordinates, float> stateRewards; | |||
| }; | |||
| @ -0,0 +1,102 @@ | |||
| #pragma once | |||
| 
 | |||
| #include "cell.h" | |||
| 
 | |||
| #include <vector> | |||
| 
 | |||
| #include <boost/tokenizer.hpp> | |||
| #include <boost/fusion/adapted/struct.hpp> | |||
| #include <boost/spirit/include/qi.hpp> | |||
| #include <boost/spirit/include/phoenix.hpp> | |||
| #include <boost/spirit/include/phoenix_operator.hpp> | |||
| #include <boost/variant/recursive_wrapper.hpp> | |||
| #include <boost/spirit/include/support_line_pos_iterator.hpp> | |||
| 
 | |||
| namespace qi      = boost::spirit::qi; | |||
| namespace phoenix = boost::phoenix; | |||
| 
 | |||
| typedef boost::spirit::line_pos_iterator<std::string::const_iterator> pos_iterator_t; | |||
| 
 | |||
| typedef std::vector<cell> row; | |||
| typedef std::vector<cell> cells; | |||
| 
 | |||
| BOOST_FUSION_ADAPT_STRUCT( | |||
|     cell, | |||
|     (Type, type) | |||
|     (Color, color) | |||
| ) | |||
| 
 | |||
| template<typename It> | |||
| struct annotation_f { | |||
|     typedef void result_type; | |||
| 
 | |||
|     annotation_f(It first) : first(first) {} | |||
|     It const first; | |||
| 
 | |||
|     template<typename Val, typename First, typename Last> | |||
|     void operator()(Val& v, First f, Last l) const { | |||
|         do_annotate(v, f, l, first); | |||
|     } | |||
| 
 | |||
|   private: | |||
|     void static do_annotate(cell& c, It f, It l, It first) { | |||
|         c.row    = get_line(f) - 1; | |||
|         c.column = get_column(first, f) / 2; | |||
|     } | |||
|     static void do_annotate(...) { std::cerr << "(not having LocationInfo)\n"; } | |||
| }; | |||
| 
 | |||
| template <typename It> | |||
|     struct MinigridParser : qi::grammar<It, row()> | |||
| { | |||
|   MinigridParser(It first) : MinigridParser::base_type(row_), annotate(first) | |||
|   { | |||
|     using namespace qi; | |||
|     type_.add | |||
|       ("W", Type::Wall) | |||
|       (" ", Type::Floor) | |||
|       ("D", Type::Door) | |||
|       ("L", Type::LockedDoor) | |||
|       ("K", Type::Key) | |||
|       ("A", Type::Ball) | |||
|       ("B", Type::Box) | |||
|       ("G", Type::Goal) | |||
|       ("V", Type::Lava) | |||
|       ("n", Type::SlipperyNorth) | |||
|       ("e", Type::SlipperyEast) | |||
|       ("s", Type::SlipperySouth) | |||
|       ("w", Type::SlipperyWest) | |||
|       ("X", Type::Agent) | |||
|       ("Z", Type::Adversary); | |||
|     color_.add | |||
|       ("R", Color::Red) | |||
|       ("G", Color::Green) | |||
|       ("B", Color::Blue) | |||
|       ("P", Color::Purple) | |||
|       ("Y", Color::Yellow) | |||
|       (" ", Color::None); | |||
| 
 | |||
|     cell_ = type_ > color_; | |||
| 
 | |||
|     row_ = (cell_ % -qi::char_("\n")); | |||
| 
 | |||
|     auto set_location_info = annotate(_val, _1, _3); | |||
|     on_success(cell_, set_location_info); | |||
| 
 | |||
|     BOOST_SPIRIT_DEBUG_NODE(type_); | |||
|     BOOST_SPIRIT_DEBUG_NODE(color_); | |||
|     BOOST_SPIRIT_DEBUG_NODE(cell_); | |||
|   } | |||
| 
 | |||
|   private: | |||
|     phoenix::function<annotation_f<It>> annotate; | |||
| 
 | |||
|     qi::symbols<char, Type>  type_; | |||
|     qi::symbols<char, Color> color_; | |||
| 
 | |||
|     qi::rule<It, cell()> cell_; | |||
|     qi::rule<It, row()>  row_; | |||
| }; | |||
| 
 | |||
| typedef boost::tokenizer< boost::escaped_list_separator<char> , std::string::const_iterator, std::string> Tokenizer; | |||
| //std::ostream& operator<<(std::ostream& os, const row& r); | |||
| @ -0,0 +1,32 @@ | |||
| #include <iostream>
 | |||
| 
 | |||
| #include "popl.hpp"
 | |||
| #include "OptionParser.h"
 | |||
| 
 | |||
| namespace io { | |||
|   int printPoplException(const popl::invalid_option &e) { | |||
|     std::cerr << "Invalid Option Exception: " << e.what() << "\n"; | |||
| 		std::cerr << "error:  "; | |||
| 		if (e.error() == popl::invalid_option::Error::missing_argument) { | |||
| 			std::cerr << "missing_argument\n"; | |||
|     } else if (e.error() == popl::invalid_option::Error::invalid_argument) { | |||
| 			std::cerr << "invalid_argument\n"; | |||
|     } else if (e.error() == popl::invalid_option::Error::too_many_arguments) { | |||
| 			std::cerr << "too_many_arguments\n"; | |||
|     } else if (e.error() == popl::invalid_option::Error::missing_option) { | |||
| 			std::cerr << "missing_option\n"; | |||
|     } | |||
| 
 | |||
| 		if (e.error() == popl::invalid_option::Error::missing_option) { | |||
|       std::string option_name(e.option()->name(popl::OptionName::short_name, true)); | |||
| 			if (option_name.empty()) | |||
| 				option_name = e.option()->name(popl::OptionName::long_name, true); | |||
| 			std::cerr << "option: " << option_name << "\n"; | |||
| 		} | |||
| 		else { | |||
| 			std::cerr << "option: " << e.option()->name(e.what_name()) << "\n"; | |||
| 			std::cerr << "value:  " << e.value() << "\n"; | |||
| 		} | |||
| 		return EXIT_FAILURE; | |||
|   } | |||
| } | |||
| @ -0,0 +1,13 @@ | |||
| #pragma once | |||
| 
 | |||
| #include <iostream> | |||
| 
 | |||
| #include "popl.hpp" | |||
| 
 | |||
| #define INPUT_ERROR 1 | |||
| #define INPUT_OK 0 | |||
| 
 | |||
| namespace io { | |||
|   int printPoplException(const popl::invalid_option &e); | |||
| 
 | |||
| } | |||
| @ -0,0 +1,681 @@ | |||
| #include "PrismModulesPrinter.h"
 | |||
| 
 | |||
| #include <map>
 | |||
| #include <string>
 | |||
| 
 | |||
| namespace prism { | |||
| 
 | |||
|   PrismModulesPrinter::PrismModulesPrinter(const ModelType &modelType, const size_t &numberOfPlayer, const bool enforceOneWays) | |||
|     : modelType(modelType), numberOfPlayer(numberOfPlayer), enforceOneWays(enforceOneWays) { | |||
|   } | |||
| 
 | |||
|   std::ostream& PrismModulesPrinter::printModel(std::ostream &os, const ModelType &modelType) { | |||
|     switch(modelType) { | |||
|       case(ModelType::MDP): | |||
|         os << "mdp"; | |||
|         break; | |||
|       case(ModelType::SMG): | |||
|         os << "smg"; | |||
|         break; | |||
|     } | |||
|     os << "\n\n"; | |||
|     return os; | |||
|   } | |||
| 
 | |||
|   std::ostream& PrismModulesPrinter::printBackgroundLabels(std::ostream &os, const AgentName &agentName, const std::pair<Color, cells> &backgroundTiles) { | |||
|     if(backgroundTiles.second.size() == 0) return os; | |||
| 
 | |||
|     bool first = true; | |||
|     std::string color = getColor(backgroundTiles.first); | |||
|     color.at(0) = std::toupper(color.at(0)); | |||
|     os << "formula " << agentName << "On" << color << " = "; | |||
|     for(auto const& cell : backgroundTiles.second) { | |||
|       if(first) first = false; else os << " | "; | |||
|       os << "(x" << agentName << "=" << cell.column << "&y" << agentName << "=" << cell.row << ")"; | |||
|     } | |||
|     os << ";\n"; | |||
|     os << "label \"" << agentName << "On" << color << "\" = " << agentName << "On" << color << ";\n"; | |||
|     return os; | |||
|   } | |||
| 
 | |||
|   std::ostream& PrismModulesPrinter::printRestrictionFormula(std::ostream& os, const AgentName &agentName, const std::string &direction,  const cells &cells) { | |||
|     bool first = true; | |||
|     os << "formula " << agentName << "CannotMove" << direction << " = " ; | |||
|     for(auto const& cell : cells) { | |||
|       if(first) first = false; | |||
|       else os << " | "; | |||
|       os << "(x" << agentName << "=" << cell.column << "&y" << agentName << "=" << cell.row << ")"; | |||
|     } | |||
|     os << ";\n"; | |||
|     return os; | |||
|   } | |||
| 
 | |||
|   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) { | |||
|     if(std::find_if(slipperyCollection.cbegin(), slipperyCollection.cend(), [=](const std::reference_wrapper<cells>& c) { return !c.get().empty(); }) == slipperyCollection.cend()) { | |||
|       os << "formula " << agentName << "IsOnSlippery = false;\n"; | |||
|       return os; | |||
|     } | |||
| 
 | |||
|     bool first = true; | |||
|     os << "formula " << agentName << "IsOnSlippery = "; | |||
| 
 | |||
|     for (const auto& slippery: slipperyCollection) { | |||
|       for(const auto& cell : slippery.get()) { | |||
|         if(first) first = false; else os << " | "; | |||
|         os << "(x" << agentName << "=" << cell.column << "&y" << agentName << "=" << cell.row << ")"; | |||
|       } | |||
|     } | |||
|     os << ";\n"; | |||
|     if(enforceOneWays) { | |||
|       first = true; | |||
|       os << "formula " << agentName << "IsOnSlipperyNorth = "; | |||
| 
 | |||
|       for (const auto& cell: slipperyNorth) { | |||
|         if(first) first = false; else os << " | "; | |||
|         os << "(x" << agentName << "=" << cell.column << "&y" << agentName << "=" << cell.row << ")"; | |||
|       } | |||
|       os << ";\n"; | |||
|       first = true; | |||
|       os << "formula " << agentName << "IsOnSlipperyEast = "; | |||
| 
 | |||
|       for (const auto& cell: slipperyEast) { | |||
|         if(first) first = false; else os << " | "; | |||
|         os << "(x" << agentName << "=" << cell.column << "&y" << agentName << "=" << cell.row << ")"; | |||
|       } | |||
|       os << ";\n"; | |||
|       first = true; | |||
|       os << "formula " << agentName << "IsOnSlipperySouth = "; | |||
| 
 | |||
|       for (const auto& cell: slipperySouth) { | |||
|         if(first) first = false; else os << " | "; | |||
|         os << "(x" << agentName << "=" << cell.column << "&y" << agentName << "=" << cell.row << ")"; | |||
|       } | |||
|       os << ";\n"; | |||
|       first = true; | |||
|       os << "formula " << agentName << "IsOnSlipperyWest = "; | |||
|       for (const auto& cell: slipperyWest) { | |||
|         if(first) first = false; else os << " | "; | |||
|         os << "(x" << agentName << "=" << cell.column << "&y" << agentName << "=" << cell.row << ")"; | |||
|       } | |||
|       os << ";\n"; | |||
|     } | |||
|     return os; | |||
|   } | |||
| 
 | |||
|   std::ostream& PrismModulesPrinter::printIsInLavaFormula(std::ostream& os, const AgentName &agentName, const cells &lava) { | |||
|     if(lava.size() == 0) { | |||
|       os << "formula " << agentName << "IsInLava = false;\n"; | |||
|       return os; | |||
|     } | |||
|     bool first = true; | |||
|     os << "formula " << agentName << "IsInLava = "; | |||
|     for(auto const& cell : lava) { | |||
|       if(first) first = false; else os << " | "; | |||
|       os << "(x" << agentName << "=" << cell.column << "&y" << agentName << "=" << cell.row << ")"; | |||
|     } | |||
|     os << ";\n"; | |||
|     os << "formula " << agentName << "IsInLavaAndNotDone = " << agentName << "IsInLava & !" << agentName << "Done;\n"; | |||
|     os << "label \"" << agentName << "IsInLavaAndNotDone\" = " << agentName << "IsInLava & !" << agentName << "Done;\n"; | |||
|     return os; | |||
|   } | |||
| 
 | |||
|   std::ostream& PrismModulesPrinter::printTurningNotAllowedFormulas(std::ostream& os, const AgentName &agentName, const cells &noTurnFloor) { | |||
|     if( (!enforceOneWays or noTurnFloor.size() == 0) or (noTurnFloor.size() == 0) ) { | |||
|       os << "formula " << agentName << "CannotTurn = false;\n"; | |||
|       return os; | |||
|     } | |||
|     bool first = true; | |||
|     os << "formula " << agentName << "CannotTurn = "; | |||
|     for(auto const& cell : noTurnFloor) { | |||
|       if(first) first = false; else os << " | "; | |||
|       os << "(x" << agentName << "=" << cell.column << "&y" << agentName << "=" << cell.row << ")"; | |||
|     } | |||
|     os << " | " << agentName << "IsOnSlippery;\n"; | |||
|     return os; | |||
|   } | |||
| 
 | |||
|   std::ostream& PrismModulesPrinter::printIsFixedFormulas(std::ostream& os, const AgentName &agentName) { | |||
|     os << "formula " << agentName << "IsFixed = false;\n"; | |||
|     os << "formula " << agentName << "SlipperyTurnLeftAllowed = true;\n"; | |||
|     os << "formula " << agentName << "SlipperyTurnRightAllowed = true;\n"; | |||
|     os << "formula " << agentName << "SlipperyMoveForwardAllowed = true;\n"; | |||
|     os << "label \"FixedStates\" = " << agentName << "IsFixed | !" << agentName << "SlipperyTurnRightAllowed | !" << agentName << "SlipperyTurnLeftAllowed | !" << agentName << "SlipperyMoveForwardAllowed | " << agentName << "IsInGoal | " << agentName << "IsInLava"; | |||
|     if(enforceOneWays) { | |||
|       os << " | " << agentName << "CannotTurn"; | |||
|     } | |||
|     os << ";\n"; | |||
|     //os << "label \"FixedStates\" = " << agentName << "IsFixed | " << agentName << "IsOnSlippery  | " << agentName << "IsInGoal | " << agentName << "IsInLava;\n";
 | |||
| 
 | |||
|     return os; | |||
|   } | |||
| 
 | |||
| 
 | |||
|   std::ostream& PrismModulesPrinter::printWallFormula(std::ostream& os, const AgentName &agentName, const cells &walls) { | |||
|     os << "formula " << agentName << "IsOnWall = "; | |||
|     bool first = true; | |||
|     for(auto const& cell : walls) { | |||
|       if(first) first = false; else os << " | "; | |||
|       os << "(x" << agentName << "=" << cell.column << "&y" << agentName << "=" << cell.row << ")"; | |||
|     } | |||
|     os << ";\n"; | |||
|     return os; | |||
|   } | |||
| 
 | |||
|   std::ostream& PrismModulesPrinter::printFormulas(std::ostream& os, | |||
|                               const AgentName &agentName, | |||
|                               const cells &restrictionNorth, | |||
|                               const cells &restrictionEast, | |||
|                               const cells &restrictionSouth, | |||
|                               const cells &restrictionWest, | |||
|                               const std::vector<std::reference_wrapper<cells>> &slipperyCollection, | |||
|                               const cells &lava, | |||
|                               const cells &walls, | |||
|                               const cells &noTurnFloor, | |||
|                               const cells &slipperyNorth, | |||
|                               const cells &slipperyEast, | |||
|                               const cells &slipperySouth, | |||
|                               const cells &slipperyWest) { | |||
|     printRestrictionFormula(os, agentName, "North", restrictionNorth); | |||
|     printRestrictionFormula(os, agentName, "East ", restrictionEast); | |||
|     printRestrictionFormula(os, agentName, "South", restrictionSouth); | |||
|     printRestrictionFormula(os, agentName, "West ", restrictionWest); | |||
|     printIsOnSlipperyFormula(os, agentName, slipperyCollection, slipperyNorth, slipperyEast, slipperySouth, slipperyWest); | |||
|     printIsInLavaFormula(os, agentName, lava); | |||
|     printWallFormula(os, agentName, walls); | |||
|     printTurningNotAllowedFormulas(os, agentName, noTurnFloor); | |||
|     printIsFixedFormulas(os, agentName); | |||
|     os << "\n"; | |||
|     return os; | |||
|   } | |||
| 
 | |||
|   std::ostream& PrismModulesPrinter::printGoalLabel(std::ostream& os, const AgentName &agentName, const cells &goals) { | |||
|     if(goals.size() == 0) { | |||
|       os << "formula " << agentName << "IsInGoal = false;\n"; | |||
|       return os; | |||
|     } | |||
| 
 | |||
|     bool first = true; | |||
|     os << "formula " << agentName << "IsInGoal = "; | |||
|     for(auto const& cell : goals) { | |||
|       if(first) first = false; else os << " | "; | |||
|       os << "(x" << agentName << "=" << cell.column << "&y" << agentName << "=" << cell.row << ")"; | |||
|     } | |||
|     os << ";\n"; | |||
|     os << "formula " << agentName << "IsInGoalAndNotDone = " << agentName << "IsInGoal & !" << agentName << "Done;\n"; | |||
|     os << "label \"" << agentName << "IsInGoalAndNotDone\" = " << agentName << "IsInGoal & !" << agentName << "Done;\n"; | |||
|     return os; | |||
|   } | |||
| 
 | |||
|   std::ostream& PrismModulesPrinter::printCrashLabel(std::ostream &os, const std::vector<AgentName> agentNames) { | |||
|     os << "label crash = "; | |||
|     bool first = true; | |||
|     for(auto const& agentName : agentNames) { | |||
|       if(agentName == "Agent") continue; | |||
|       if(first) first = false; else os << " | "; | |||
|       os << "(xAgent=x" << agentName << ")&(yAgent=y" << agentName << ")"; | |||
|     } | |||
|     os << ";\n\n"; | |||
|     return os; | |||
|   } | |||
| 
 | |||
|   std::ostream& PrismModulesPrinter::printAvoidanceLabel(std::ostream &os, const std::vector<AgentName> agentNames, const int &distance) { | |||
|     os << "label avoidance = "; | |||
|     bool first = true; | |||
|     for(auto const& agentName : agentNames) { | |||
|       if(agentName == "Agent") continue; | |||
|       if(first) first = false; else os << " | "; | |||
|       os << "max(xAgent-x" << agentName << ",x" << agentName << "-xAgent)+"; | |||
|       os << "max(yAgent-y" << agentName << ",y" << agentName << "-yAgent) "; | |||
|     } | |||
|     os << ";\n\n"; | |||
|     return os; | |||
|   } | |||
| 
 | |||
|   // TODO this does not account for multiple agents yet, i.e. key can be picked up multiple times
 | |||
|   std::ostream& PrismModulesPrinter::printKeysLabels(std::ostream& os, const AgentName &agentName, const cells &keys) { | |||
|     if(keys.size() == 0) return os; | |||
| 
 | |||
|     for(auto const& key : keys) { | |||
|       int xKey = key.getCoordinates().first; | |||
|       int yKey = key.getCoordinates().second; | |||
|       std::string keyColor = key.getColor(); | |||
|       os << "label \"" << agentName << "PickedUp" << keyColor << "Key\" = " << agentName << "_has_" << keyColor << "_key = true;\n"; | |||
|       os << "formula " << agentName << "CanPickUp" << keyColor << "Key = "; | |||
|       os << "((x" << agentName << "-1 = " << xKey << "&y" << agentName << "   = " << yKey << "&view" << agentName << " = 1) |"; | |||
|       os << " (x" << agentName << "+1 = " << xKey << "&y" << agentName << "   = " << yKey << "&view" << agentName << " = 3) |"; | |||
|       os << " (x" << agentName << "   = " << xKey << "&y" << agentName << "-1 = " << yKey << "&view" << agentName << " = 0) |"; | |||
|       os << " (x" << agentName << "   = " << xKey << "&y" << agentName << "+1 = " << yKey << "&view" << agentName << " = 2) ) &"; | |||
|       os << "!" << agentName << "_has_" << keyColor << "_key;"; | |||
|     } | |||
|     os << "\n"; | |||
|     return os; | |||
|   } | |||
| 
 | |||
|   std::ostream& PrismModulesPrinter::printBooleansForKeys(std::ostream &os, const AgentName &agentName, const cells &keys) { | |||
|     for(auto const& key : keys) { | |||
|       os << "\t" << agentName << "_has_"<< key.getColor() << "_key : bool init false;\n"; | |||
|     } | |||
|     os << "\n"; | |||
|     return os; | |||
|   } | |||
| 
 | |||
|   std::ostream& PrismModulesPrinter::printBooleansForBackground(std::ostream &os, const AgentName &agentName, const std::map<Color, cells> &backgroundTiles) { | |||
|     for(auto const& [color, cells] : backgroundTiles) { | |||
|       if(cells.size() == 0) continue; | |||
|       std::string c = getColor(color); | |||
|       c.at(0) = std::toupper(c.at(0)); | |||
|       os << "\t" << agentName << "_picked_up_" << c << " : bool init false;\n"; | |||
|     } | |||
|     os << "\n"; | |||
|     return os; | |||
|   } | |||
| 
 | |||
|   std::ostream& PrismModulesPrinter::printActionsForKeys(std::ostream &os, const AgentName &agentName, const cells &keys) { | |||
|     for(auto const& key : keys) { | |||
|       std::string keyColor = key.getColor(); | |||
|       os << "\t[pickup_" << keyColor << "_key] " << agentName << "CanPickUp" << keyColor << "Key -> "; | |||
|       os << "(" << agentName << "_has_" << keyColor << "_key'=true);"; | |||
|     } | |||
|     os << "\n"; | |||
|     return os; | |||
|   } | |||
| 
 | |||
|   std::ostream& PrismModulesPrinter::printActionsForBackground(std::ostream &os, const AgentName &agentName, const std::map<Color, cells> &backgroundTiles) { | |||
|     for(auto const& [color, cells] : backgroundTiles) { | |||
|       if(cells.size() == 0) continue; | |||
|       std::string c = getColor(color); | |||
|       c.at(0) = std::toupper(c.at(0)); | |||
|       os << "\t[" << agentName << "_pickup_" << c << "] " << agentName << "On" << c << " & !" << agentName << "_picked_up_" << c << " -> "; | |||
|       os << "(" << agentName << "_picked_up_" << c << "'=true);\n"; | |||
|     } | |||
|     os << "\n"; | |||
|     return os; | |||
|   } | |||
| 
 | |||
|   std::ostream& PrismModulesPrinter::printInitStruct(std::ostream &os, const AgentName &agentName) { | |||
|     os << "init\n"; | |||
|     os << "\t(!AgentIsInGoal & !AgentIsInLava & !AgentDone & !AgentIsOnWall)"; | |||
|     if(enforceOneWays) { | |||
|       os << " & ( !AgentCannotTurn ) "; | |||
|     } else { | |||
|       os << " & ( !AgentIsOnSlippery ) "; | |||
|     } | |||
|     os << "\nendinit\n\n"; | |||
| 
 | |||
|     return os; | |||
|   } | |||
| 
 | |||
|   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) { | |||
|     os << "module " << agentName << "\n"; | |||
|     os << "\tx" << agentName << " : [1.." << boundaries.second  << "];\n"; | |||
|     os << "\ty" << agentName << " : [1.." << boundaries.first << "];\n"; | |||
|     printBooleansForKeys(os, agentName, keys); | |||
|     printBooleansForBackground(os, agentName, backgroundTiles); | |||
|     os << "\t" << agentName << "Done : bool;\n"; | |||
|     if(agentWithView) { | |||
|       os << "\tview" << agentName << " : [0..3];\n"; | |||
|     os << "\n"; | |||
|     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"; | |||
|     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"; | |||
|     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"; | |||
|       if(enforceOneWays) { | |||
|         os << "\t[" << agentName << "_stuck]        !" << agentName << "IsFixed & " << agentName << "CannotTurn & view" << agentName << " = 0 & " << agentName << "CannotMoveEast -> true;\n"; | |||
|         os << "\t[" << agentName << "_stuck]        !" << agentName << "IsFixed & " << agentName << "CannotTurn & view" << agentName << " = 1 & " << agentName << "CannotMoveSouth -> true;\n"; | |||
|         os << "\t[" << agentName << "_stuck]        !" << agentName << "IsFixed & " << agentName << "CannotTurn & view" << agentName << " = 2 & " << agentName << "CannotMoveWest -> true;\n"; | |||
|         os << "\t[" << agentName << "_stuck]        !" << agentName << "IsFixed & " << agentName << "CannotTurn & view" << agentName << " = 3 & " << agentName << "CannotMoveNorth -> true;\n"; | |||
|       } | |||
|     } else { | |||
|       os << "\t[" << agentName << "_turns]  " << " !" << agentName << "CannotTurn & " << " !" << agentName << "IsFixed & " <<  moveGuard(agentIndex) << " true -> (x" << agentName << "'=x" << agentName << ")" << moveUpdate(agentIndex) << ";\n"; | |||
|     } | |||
|     printActionsForKeys(os, agentName, keys); | |||
|     printActionsForBackground(os, agentName, backgroundTiles); | |||
|     os << "\n"; | |||
|     printMovementActions(os, agentName, agentIndex, agentWithView); | |||
|     for(auto const& probability : probabilities) { | |||
|       printMovementActions(os, agentName, agentIndex, agentWithView, probability); | |||
|     } | |||
|     printDoneActions(os, agentName, agentIndex); | |||
|     os << "\n"; | |||
|     return os; | |||
|   } | |||
| 
 | |||
|   std::ostream& PrismModulesPrinter::printMovementActions(std::ostream &os, const AgentName &agentName, const size_t &agentIndex, const bool agentWithView, const float &probability) { | |||
|     if(probability >= 1) { | |||
|       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"; | |||
|       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"; | |||
|       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"; | |||
|       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"; | |||
|     } else { | |||
|       std::string probabilityString = std::to_string(probability); | |||
|       std::string percentageString = std::to_string((int)(100 * probability)); | |||
|       std::string complementProbabilityString = std::to_string(1 - probability); | |||
|       os << "\t[" << agentName << "_move_north_" << percentageString << "] "; | |||
|       os << moveGuard(agentIndex) << viewVariable(agentName, 3, agentWithView) << " !" << agentName << "IsFixed & " << " !" << agentName << "IsOnSlippery & !" << agentName << "IsInLava & !" << agentName << "CannotMoveNorth -> "; | |||
|       os << probabilityString << ": (y" << agentName << "'=y" << agentName << "-1)" << moveUpdate(agentIndex) << " + "; | |||
|       os << complementProbabilityString << ": (y" << agentName << "'=y" << agentName << ") " << moveUpdate(agentIndex) << ";\n"; | |||
| 
 | |||
|       os << "\t[" << agentName << "_move_east_" << percentageString << "] "; | |||
|       os << moveGuard(agentIndex) << viewVariable(agentName, 0, agentWithView) << " !" << agentName << "IsFixed & " << " !" << agentName << "IsOnSlippery & !" << agentName << "IsInLava & !" << agentName << "CannotMoveEast  -> "; | |||
|       os << probabilityString << ": (x" << agentName << "'=x" << agentName << "+1)" << moveUpdate(agentIndex) << " + "; | |||
|       os << complementProbabilityString << ": (x" << agentName << "'=x" << agentName << ") " << moveUpdate(agentIndex) << ";\n"; | |||
| 
 | |||
|       os << "\t[" << agentName << "_move_south_" << percentageString << "] "; | |||
|       os << moveGuard(agentIndex) << viewVariable(agentName, 1, agentWithView) << " !" << agentName << "IsFixed & " << " !" << agentName << "IsOnSlippery & !" << agentName << "IsInLava & !" << agentName << "CannotMoveSouth -> "; | |||
|       os << probabilityString << ": (y" << agentName << "'=y" << agentName << "+1)" << moveUpdate(agentIndex) << " + "; | |||
|       os << complementProbabilityString << ": (y" << agentName << "'=y" << agentName << ") " << moveUpdate(agentIndex) << ";\n"; | |||
| 
 | |||
|       os << "\t[" << agentName << "_move_west_" << percentageString << "] "; | |||
|       os << moveGuard(agentIndex) << viewVariable(agentName, 2, agentWithView) << " !" << agentName << "IsFixed & " << " !" << agentName << "IsOnSlippery & !" << agentName << "IsInLava & !" << agentName << "CannotMoveWest  -> "; | |||
|       os << probabilityString << ": (x" << agentName << "'=x" << agentName << "-1)" << moveUpdate(agentIndex) << " + "; | |||
|       os << complementProbabilityString << ": (x" << agentName << "'=x" << agentName << ") " << moveUpdate(agentIndex) << ";\n"; | |||
|     } | |||
|     return os; | |||
|   } | |||
| 
 | |||
|   std::ostream& PrismModulesPrinter::printDoneActions(std::ostream &os, const AgentName &agentName, const size_t &agentIndex) { | |||
|     os << "\t[" << agentName << "_done]" << moveGuard(agentIndex) << agentName << "IsInGoal | " << agentName << "IsInLava -> (" << agentName << "Done'=true);\n"; | |||
|     return os; | |||
|   } | |||
| 
 | |||
|   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) { | |||
|     constexpr std::size_t PROB_PIECES = 9, ALL_POSS_DIRECTIONS = 9; | |||
| 
 | |||
|     std::array<std::string, ALL_POSS_DIRECTIONS> positionTransition = { | |||
|       /* north      */                                                "(y" + agentName + "'=y" + agentName + "-1)", | |||
|       /* north east */   "(x" + agentName + "'=x" + agentName + "+1) & (y" + agentName + "'=y" + agentName + "-1)", | |||
|       /* east       */   "(x" + agentName + "'=x" + agentName + "+1)", | |||
|       /* east south */   "(x" + agentName + "'=x" + agentName + "+1) & (y" + agentName + "'=y" + agentName + "+1)", | |||
|       /* south      */                                                "(y" + agentName + "'=y" + agentName + "+1)", | |||
|       /* south west */   "(x" + agentName + "'=x" + agentName + "-1) & (y" + agentName + "'=y" + agentName + "+1)", | |||
|       /* west       */   "(x" + agentName + "'=x" + agentName + "-1)", | |||
|       /* north west */   "(x" + agentName + "'=x" + agentName + "-1) & (y" + agentName + "'=y" + agentName + "-1)", | |||
|       /* own position */ "(x" + agentName + "'=x" + agentName + ") & (y" + agentName + "'=y" + agentName + ")" | |||
|     }; | |||
| 
 | |||
|     // view transition appdx in form (guard, update part)
 | |||
|     // IMPORTANT: No mod() usage for turn left due to bug in mod() function for decrement
 | |||
| 
 | |||
|     std::array<std::pair<std::string, std::string>, 3> viewTransition = { | |||
|         /* turn to right */ std::make_pair(" & " + agentName + "SlipperyTurnRightAllowed ",                           " & (view" + agentName + "'=mod(view" + agentName + " + 1, 4))"), | |||
|         /* turn to left  */ std::make_pair(" & " + agentName + "SlipperyTurnLeftAllowed & view" + agentName + ">0", " & (view" + agentName + "'=view" + agentName + " - 1)"), | |||
|         /* turn to left  */ std::make_pair(" & " + agentName + "SlipperyTurnLeftAllowed & view" + agentName + "=0", " & (view" + agentName + "'=3)") | |||
|     }; | |||
| 
 | |||
|     // direction specifics
 | |||
| 
 | |||
|     std::string actionName; | |||
|     std::size_t remainPosIndex = 8; | |||
|     std::array<std::size_t, ALL_POSS_DIRECTIONS> prob_piece_dir; // { n, ne, w, se, s, sw, w, nw, CURRENT POS }
 | |||
| 
 | |||
|     switch (orientation) | |||
|     { | |||
|       case SlipperyType::North: | |||
|         actionName = "\t[" + agentName + "turn_at_slip_north]"; | |||
|         prob_piece_dir = { 0, 0, 0, 1, 1, 1, 0, 0, 0 /* <- R */ }; | |||
|         break; | |||
| 
 | |||
|       case SlipperyType::South: | |||
|         actionName = "\t[" + agentName + "turn_at_slip_south]"; | |||
|         prob_piece_dir = { 1, 1, 0, 0, 0, 0, 0, 1, 0 /* <- R */ }; | |||
|         break; | |||
| 
 | |||
|       case SlipperyType::East: | |||
|         actionName = "\t[" + agentName + "turn_at_slip_east]"; | |||
|         prob_piece_dir = { 0, 0, 0, 0, 0, 1, 1, 1, 0 /* <- R */ }; | |||
|         break; | |||
| 
 | |||
|       case SlipperyType::West: | |||
|         actionName = "\t[" + agentName + "turn_at_slip_west]"; | |||
|         prob_piece_dir = { 0, 1, 1, 1, 0, 0, 0, 0, 0 /* <- R */ }; | |||
|         break; | |||
|     } | |||
| 
 | |||
|     slipperyActions.insert(actionName); | |||
| 
 | |||
|     // override probability to 0 if corresp. direction is blocked
 | |||
| 
 | |||
|     for (std::size_t i = 0; i < ALL_POSS_DIRECTIONS - 1; i++) { | |||
|       if (!neighborhood.at(i)) | |||
|         prob_piece_dir.at(i) = 0; | |||
|     } | |||
| 
 | |||
|     // determine residual probability (R) by replacing 0 with (1 - overall sum)
 | |||
| 
 | |||
|     prob_piece_dir.at(remainPosIndex) = PROB_PIECES - std::accumulate(prob_piece_dir.begin(), prob_piece_dir.end(), 0); | |||
| 
 | |||
|     // <DEBUG_AREA>
 | |||
|     { | |||
|       assert(prob_piece_dir.at(remainPosIndex) <= 9 && prob_piece_dir.at(remainPosIndex) >= 6 && "Value not in Range!"); | |||
|       assert(std::accumulate(prob_piece_dir.begin(), prob_piece_dir.end(), 0) == PROB_PIECES && "Does not sum up to 1!"); | |||
|     } | |||
|     // </DEBUG_AREA>
 | |||
| 
 | |||
| 
 | |||
|     // generic output (for every view transition)
 | |||
|     for (std::size_t v = 0; v < viewTransition.size(); v++) { | |||
| 
 | |||
|         os << actionName << moveGuard(agentIndex) << " x" << agentName << "=" << c.second << " & y" << agentName << "=" << c.first << viewTransition.at(v).first; | |||
|         for (std::size_t i = 0; i < ALL_POSS_DIRECTIONS; i++) { | |||
|           os << (i == 0 ? " -> " : " + ") << prob_piece_dir.at(i) << "/" << PROB_PIECES << " : " << positionTransition.at(i) << viewTransition.at(v).second << moveUpdate(agentIndex) << (i == ALL_POSS_DIRECTIONS - 1 ? ";\n" : "\n"); | |||
|         } | |||
|     } | |||
| 
 | |||
|     return os; | |||
|   } | |||
| 
 | |||
|   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) { | |||
|     constexpr std::size_t PROB_PIECES = 9, ALL_POSS_DIRECTIONS = 8; | |||
| 
 | |||
|     std::array<std::string, ALL_POSS_DIRECTIONS> positionTransition = { | |||
|       /* north      */                                              "(y" + agentName + "'=y" + agentName + "-1)", | |||
|       /* north east */ "(x" + agentName + "'=x" + agentName + "+1) & (y" + agentName + "'=y" + agentName + "-1)", | |||
|       /* east       */ "(x" + agentName + "'=x" + agentName + "+1)", | |||
|       /* east south */ "(x" + agentName + "'=x" + agentName + "+1) & (y" + agentName + "'=y" + agentName + "+1)", | |||
|       /* south      */                                              "(y" + agentName + "'=y" + agentName + "+1)", | |||
|       /* south west */ "(x" + agentName + "'=x" + agentName + "-1) & (y" + agentName + "'=y" + agentName + "+1)", | |||
|       /* west       */ "(x" + agentName + "'=x" + agentName + "-1)", | |||
|       /* north west */ "(x" + agentName + "'=x" + agentName + "-1) & (y" + agentName + "'=y" + agentName + "-1)" | |||
|     }; | |||
| 
 | |||
|     // direction specifics
 | |||
| 
 | |||
|     std::size_t straightPosIndex; | |||
|     std::string actionName, specialTransition; // if straight ahead is blocked
 | |||
|     std::array<std::size_t, ALL_POSS_DIRECTIONS> prob_piece_dir; // { n, ne, w, se, s, sw, w, nw }
 | |||
| 
 | |||
|     switch (orientation) | |||
|     { | |||
|       case SlipperyType::North: | |||
|         actionName = "\t[" + agentName + "move_on_slip_north]"; | |||
|         prob_piece_dir = { 0, 0, 1, 2, 0 /* <- R */, 2, 1, 0 }; | |||
|         straightPosIndex = 4; | |||
|         specialTransition = "(y" + agentName + "'=y" + agentName + (!neighborhood.at(straightPosIndex) ? ")" : "+1)"); | |||
|         break; | |||
| 
 | |||
|       case SlipperyType::South: | |||
|         actionName = "\t[" + agentName + "move_on_slip_south]"; | |||
|         prob_piece_dir = { 0 /* <- R */, 2, 1, 0, 0, 0, 1, 2 }; | |||
|         straightPosIndex = 0; // always north
 | |||
|         specialTransition = "(y" + agentName + "'=y" + agentName + (!neighborhood.at(straightPosIndex) ? ")" : "-1)"); | |||
|         break; | |||
| 
 | |||
|       case SlipperyType::East: | |||
|         actionName = "\t[" + agentName + "move_on_slip_east]"; | |||
|         prob_piece_dir = { 1, 0, 0, 0, 1, 2, 0 /* <- R */, 2 }; | |||
|         straightPosIndex = 6; | |||
|         specialTransition = "(x" + agentName + "'=x" + agentName + (!neighborhood.at(straightPosIndex) ? ")" : "-1)"); | |||
|         break; | |||
| 
 | |||
|       case SlipperyType::West: | |||
|         actionName = "\t[" + agentName + "move_on_slip_west]"; | |||
|         prob_piece_dir = { 1, 2, 0 /* <- R */, 2, 1, 0, 0, 0 }; | |||
|         straightPosIndex = 2; | |||
|         specialTransition = "(x" + agentName + "'=x" + agentName + (!neighborhood.at(straightPosIndex) ? ")" : "+1)"); | |||
|         break; | |||
|     } | |||
| 
 | |||
|     slipperyActions.insert(actionName); | |||
| 
 | |||
|     // override probability to 0 if corresp. direction is blocked
 | |||
| 
 | |||
|     for (std::size_t i = 0; i < ALL_POSS_DIRECTIONS; i++) { | |||
|       if (!neighborhood.at(i)) | |||
|         prob_piece_dir.at(i) = 0; | |||
|     } | |||
| 
 | |||
|     // determine residual probability (R) by replacing 0 with (1 - overall sum)
 | |||
|     if(enforceOneWays) { | |||
|       prob_piece_dir = {0,0,0,0,0,0,0,0}; | |||
|     } | |||
|     prob_piece_dir.at(straightPosIndex) = PROB_PIECES - std::accumulate(prob_piece_dir.begin(), prob_piece_dir.end(), 0); | |||
| 
 | |||
|     // <DEBUG_AREA>
 | |||
|     { | |||
|       assert(prob_piece_dir.at(straightPosIndex) <= 9 && prob_piece_dir.at(straightPosIndex) >= 3 && "Value not in Range!"); | |||
|       assert(std::accumulate(prob_piece_dir.begin(), prob_piece_dir.end(), 0) == PROB_PIECES && "Does not sum up to 1!"); | |||
|       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!")); | |||
|       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!")); | |||
|       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!")); | |||
|       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!")); | |||
|     } | |||
|     // </DEBUG_AREA>
 | |||
| 
 | |||
|     // special case: straight forward is blocked (then remain in same position)
 | |||
| 
 | |||
|     positionTransition.at(straightPosIndex) = specialTransition; | |||
| 
 | |||
|     // generic output (for every view and every possible view direction)
 | |||
| 
 | |||
|     os << actionName << moveGuard(agentIndex) << " x" << agentName << "=" << c.second << " & y" << agentName << "=" << c.first << " & " << agentName << "SlipperyMoveForwardAllowed "; | |||
| 
 | |||
|     for (std::size_t i = 0; i < ALL_POSS_DIRECTIONS; i++) { | |||
|       os << (i == 0 ? " -> " : " + ") << prob_piece_dir.at(i) << "/" << PROB_PIECES << " : " << positionTransition.at(i) << moveUpdate(agentIndex) << (i == ALL_POSS_DIRECTIONS - 1 ? ";\n" : "\n"); | |||
|     } | |||
| 
 | |||
|     return os; | |||
|   } | |||
| 
 | |||
|   std::ostream& PrismModulesPrinter::printEndmodule(std::ostream &os) { | |||
|     os << "endmodule\n"; | |||
|     os << "\n"; | |||
|     return os; | |||
|   } | |||
| 
 | |||
|   std::ostream& PrismModulesPrinter::printPlayerStruct(std::ostream &os, const AgentName &agentName, const bool agentWithView, const std::vector<float> &probabilities, const std::set<std::string> &slipperyActions) { | |||
|     os << "player " << agentName << "\n\t"; | |||
|     bool first = true; | |||
|     std::list<std::string> allActions = { "_move_north", "_move_east", "_move_south", "_move_west" }; | |||
|     std::list<std::string> movementActions = allActions; | |||
|     for(auto const& probability : probabilities) { | |||
|       std::string percentageString = std::to_string((int)(100 * probability)); | |||
|       for(auto const& movement : movementActions) { | |||
|         allActions.push_back(movement + "_" + percentageString); | |||
|       } | |||
|     } | |||
|     if(agentWithView) { | |||
|       allActions.push_back("_turn_left"); | |||
|       allActions.push_back("_turn_right"); | |||
|     } else { | |||
|       allActions.push_back("_turns"); | |||
|     } | |||
| 
 | |||
|     for(auto const& action : allActions) { | |||
|       if(first) first = false; else os << ", "; | |||
|       os << "[" << agentName << action << "]"; | |||
|     } | |||
|     for(auto const& action : slipperyActions) { | |||
|       os << ", " << action; | |||
|     } | |||
|     os << "\nendplayer\n"; | |||
|     return os; | |||
|   } | |||
| 
 | |||
|   std::ostream& PrismModulesPrinter::printGlobalMoveVariable(std::ostream &os, const size_t &numberOfPlayer) { | |||
|     os << "\nglobal move : [0.." << std::to_string(numberOfPlayer - 1) << "] init 0;\n\n"; | |||
|     return os; | |||
|   } | |||
| 
 | |||
|   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) { | |||
|     if(lava.size() != 0) { | |||
|       os << "rewards \"SafetyNoBFS\"\n"; | |||
|       os << "\tAgentIsInLavaAndNotDone: -100;\n"; | |||
|       os << "endrewards\n"; | |||
|     } | |||
| 
 | |||
|     os << "rewards \"SafetyNoBFSAndGoal\"\n"; | |||
|     if(goals.size() != 0) os << "\tAgentIsInGoalAndNotDone:  100;\n"; | |||
|     if(lava.size() != 0) os << "\tAgentIsInLavaAndNotDone: -100;\n"; | |||
|     os << "endrewards\n"; | |||
| 
 | |||
|     os << "rewards \"Time\"\n"; | |||
|     os << "\t!AgentIsInGoal : -1;\n"; | |||
|     if(goals.size() != 0) os << "\tAgentIsInGoalAndNotDone:  100;\n"; | |||
|     if(lava.size() != 0) os << "\tAgentIsInLavaAndNotDone: -100;\n"; | |||
|     os << "endrewards\n"; | |||
| 
 | |||
|     if(stateRewards.size() > 0) { | |||
|       os << "rewards \"SafetyWithBFS\"\n"; | |||
|       if(lava.size() != 0) os << "\tAgentIsInLavaAndNotDone: -100;\n"; | |||
|       for(auto const [coordinates, reward] : stateRewards) { | |||
|         os << "\txAgent=" << coordinates.first << "&yAgent=" << coordinates.second << " : " << reward << ";\n"; | |||
|       } | |||
|       os << "endrewards\n"; | |||
|     } | |||
|     if(stateRewards.size() > 0) { | |||
|       os << "rewards \"SafetyWithBFSAndGoal\"\n"; | |||
|       if(goals.size() != 0) os << "\tAgentIsInGoalAndNotDone:  100;\n"; | |||
|       if(lava.size() != 0) os << "\tAgentIsInLavaAndNotDone: -100;\n"; | |||
|       for(auto const [coordinates, reward] : stateRewards) { | |||
|         os << "\txAgent=" << coordinates.first << "&yAgent=" << coordinates.second << " : " << reward << ";\n"; | |||
|       } | |||
|       os << "endrewards\n"; | |||
|     } | |||
| 
 | |||
|     for(auto const entry : backgroundTiles) | |||
|     { | |||
|       std::cout << getColor(entry.first) << " "; | |||
|       for(auto const cell : entry.second){ | |||
|         std::cout << cell.getCoordinates().first << " " << cell.getCoordinates().second << std::endl; | |||
|       } | |||
|     } | |||
|     if(backgroundTiles.size() > 0) { | |||
|       os << "rewards \"TaxiReward\"\n"; | |||
|       os << "\t!AgentIsInGoal : -1;\n"; | |||
|       std::string allPassengersPickedUp = ""; | |||
|       bool first = true; | |||
|       for(auto const [color, cells] : backgroundTiles) { | |||
|         if(cells.size() == 0) continue; | |||
|         if(first) first = false; else allPassengersPickedUp += "&"; | |||
|         std::string c = getColor(color); | |||
|         c.at(0) = std::toupper(c.at(0)); | |||
|         std::string visitedLabel = agentName + "_picked_up_" + c; | |||
|         allPassengersPickedUp += visitedLabel; | |||
|         os << "[" << agentName << "_pickup_" << c << "] true : 100;\n"; | |||
|       } | |||
|       if(goals.size() != 0) os << "\tAgentIsInGoalAndNotDone & " << allPassengersPickedUp << " :  100;\n"; | |||
|       if(goals.size() != 0) os << "\tAgentIsInGoalAndNotDone & !(" << allPassengersPickedUp << ") :  -100;\n"; | |||
|       os << "endrewards"; | |||
|     } | |||
|     return os; | |||
|   } | |||
| 
 | |||
|   std::string PrismModulesPrinter::moveGuard(const size_t &agentIndex) { | |||
|     return isGame() ? " move=" + std::to_string(agentIndex) + " & " : " "; | |||
|   } | |||
| 
 | |||
|   std::string PrismModulesPrinter::moveUpdate(const size_t &agentIndex) { | |||
|     return isGame() ? | |||
|              (agentIndex == numberOfPlayer - 1) ? | |||
|                " & (move'=0) " : | |||
|                " & (move'=" + std::to_string(agentIndex + 1) + ") " : | |||
|                ""; | |||
|   } | |||
| 
 | |||
|   std::string PrismModulesPrinter::viewVariable(const AgentName &agentName, const size_t &agentDirection, const bool agentWithView) { | |||
|     return agentWithView ? " view" + agentName + "=" + std::to_string(agentDirection) + " & " : " "; | |||
|   } | |||
| 
 | |||
|   bool PrismModulesPrinter::isGame() const { | |||
|     return modelType == ModelType::SMG; | |||
|   } | |||
| } | |||
| @ -0,0 +1,91 @@ | |||
| #pragma once | |||
| 
 | |||
| #include <iostream> | |||
| #include <functional> | |||
| #include "MinigridGrammar.h" | |||
| #include "PrismPrinter.h" | |||
| 
 | |||
| namespace prism { | |||
|   class PrismModulesPrinter { | |||
|     public: | |||
|       PrismModulesPrinter(const ModelType &modelType, const size_t &numberOfPlayer, const bool enforceOneWays = false); | |||
| 
 | |||
|       std::ostream& printRestrictionFormula(std::ostream& os, const AgentName &agentName, const std::string &direction, const cells &cells); | |||
|       std::ostream& 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); | |||
|       std::ostream& printGoalLabel(std::ostream& os, const AgentName&agentName, const cells &goals); | |||
|       std::ostream& printCrashLabel(std::ostream &os, const std::vector<AgentName> agentNames); | |||
|       std::ostream& printAvoidanceLabel(std::ostream &os, const std::vector<AgentName> agentNames, const int &distance); | |||
|       std::ostream& printKeysLabels(std::ostream& os, const AgentName&agentName, const cells &keys); | |||
|       std::ostream& printBackgroundLabels(std::ostream &os, const AgentName &agentName, const std::pair<Color, cells> &backgroundTiles); | |||
|       std::ostream& printIsInLavaFormula(std::ostream& os, const AgentName &agentName, const cells &lava); | |||
|       std::ostream& printIsFixedFormulas(std::ostream& os, const AgentName &agentName); | |||
|       std::ostream& printTurningNotAllowedFormulas(std::ostream& os, const AgentName &agentName, const cells &floor); | |||
|       std::ostream& printWallFormula(std::ostream& os, const AgentName &agentName, const cells &walls); | |||
|       std::ostream& printFormulas(std::ostream& os, | |||
|                                   const AgentName&agentName, | |||
|                                   const cells &restrictionNorth, | |||
|                                   const cells &restrictionEast, | |||
|                                   const cells &restrictionSouth, | |||
|                                   const cells &restrictionWest, | |||
|                                   const std::vector<std::reference_wrapper<cells>> &slipperyCollection, | |||
|                                   const cells &lava, | |||
|                                   const cells &walls, | |||
|                                   const cells &noTurnFloor, | |||
|                                   const cells &slipperyNorth, | |||
|                                   const cells &slipperyEast, | |||
|                                   const cells &slipperySouth, | |||
|                                   const cells &slipperyWest); | |||
| 
 | |||
|        /* | |||
|         * Representation for Slippery Tile. | |||
|         *  -) North: Slips from North to South | |||
|         *  -) East: Slips from East to West | |||
|         *  -) South: Slips from South to North | |||
|         *  -) West: Slips from West to East | |||
|         */ | |||
|       enum class SlipperyType { North, East, South, West }; | |||
| 
 | |||
|       /* | |||
|        * Prints Slippery on move action. | |||
|        * | |||
|        * @param neighborhood: Information of wall-blocks in 8-neighborhood { n, nw, e, se, s, sw, w, nw }. If entry is false, then corresponding neighboorhood position is a wall. | |||
|        * @param orientation: Information of slippery type (either north, south, east, west). | |||
|        */ | |||
|       std::ostream& 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); | |||
| 
 | |||
|       /* | |||
|        * Prints Slippery on turn action. | |||
|        * | |||
|        * @param neighborhood: Information of wall-blocks in 8-neighborhood { n, nw, e, se, s, sw, w, nw }. If entry is false, then corresponding neighboorhood position is a wall. | |||
|        * @param orientation: Information of slippery type (either north, south, east, west). | |||
|        */ | |||
|       std::ostream& 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); | |||
| 
 | |||
|       std::ostream& printModel(std::ostream &os, const ModelType &modelType); | |||
|       std::ostream& printBooleansForKeys(std::ostream &os, const AgentName &agentName, const cells &keys); | |||
|       std::ostream& printActionsForKeys(std::ostream &os, const AgentName &agentName, const cells &keys); | |||
|       std::ostream& printBooleansForBackground(std::ostream &os, const AgentName &agentName, const std::map<Color, cells> &backgroundTiles); | |||
|       std::ostream& printActionsForBackground(std::ostream &os, const AgentName &agentName, const std::map<Color, cells> &backgroundTiles); | |||
|       std::ostream& printInitStruct(std::ostream &os, const AgentName &agentName); | |||
|       std::ostream& 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 = {}); | |||
|       std::ostream& printMovementActions(std::ostream &os, const AgentName &agentName, const size_t &agentIndex, const bool agentWithView, const float &probability = 1.0); | |||
|       std::ostream& printDoneActions(std::ostream &os, const AgentName &agentName, const size_t &agentIndex); | |||
|       std::ostream& printEndmodule(std::ostream &os); | |||
|       std::ostream& printPlayerStruct(std::ostream &os, const AgentName &agentName, const bool agentWithView, const std::vector<float> &probabilities = {}, const std::set<std::string> &slipperyActions = {}); | |||
|       std::ostream& printGlobalMoveVariable(std::ostream &os, const size_t &numberOfPlayer); | |||
| 
 | |||
|       std::ostream& 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); | |||
| 
 | |||
|       std::string moveGuard(const size_t &agentIndex); | |||
|       std::string moveUpdate(const size_t &agentIndex); | |||
| 
 | |||
|       std::string viewVariable(const AgentName &agentName, const size_t &agentDirection, const bool agentWithView); | |||
| 
 | |||
|       bool isGame() const; | |||
|     private: | |||
| 
 | |||
|       ModelType const& modelType; | |||
|       const size_t numberOfPlayer; | |||
|       bool enforceOneWays; | |||
|   }; | |||
| } | |||
| @ -0,0 +1,15 @@ | |||
| #pragma once | |||
| 
 | |||
| #include <string> | |||
| 
 | |||
| #include "cell.h" | |||
| 
 | |||
| typedef std::string AgentName; | |||
| typedef std::pair<std::string, coordinates> AgentNameAndPosition; | |||
| typedef std::map<AgentNameAndPosition::first_type, AgentNameAndPosition::second_type> AgentNameAndPositionMap; | |||
| 
 | |||
| namespace prism { | |||
|   enum class ModelType { | |||
|     MDP, SMG | |||
|   }; | |||
| } | |||
| @ -0,0 +1,83 @@ | |||
| #include "cell.h"
 | |||
| 
 | |||
| #include <stdexcept>
 | |||
| 
 | |||
| std::ostream &operator<<(std::ostream &os, const cell &c) { | |||
|   os << static_cast<char>(c.type) << static_cast<char>(c.color); | |||
|   os <<  " at (" << c.row << "," << c.column << ")"; | |||
|   return os; | |||
| } | |||
| 
 | |||
| cell cell::getNorth(const std::vector<cell> &grid) const { | |||
|   auto north = std::find_if(grid.begin(), grid.end(), [this](const cell &c) { | |||
|         return this->row - 1 == c.row && this->column == c.column; | |||
|       }); | |||
|   if(north == grid.end()) { | |||
|     throw std::logic_error{ "Cannot get cell north of (" + std::to_string(row) + "," + std::to_string(column) + ")"}; | |||
|     std::exit(EXIT_FAILURE); | |||
|   } | |||
|   return *north; | |||
| } | |||
| 
 | |||
| cell cell::getEast(const std::vector<cell> &grid) const { | |||
|   auto east = std::find_if(grid.begin(), grid.end(), [this](const cell &c) { | |||
|         return this->row == c.row && this->column + 1 == c.column; | |||
|       }); | |||
|   if(east == grid.end()) { | |||
|     throw std::logic_error{ "Cannot get cell east of (" + std::to_string(row) + "," + std::to_string(column) + ")"}; | |||
|     std::exit(EXIT_FAILURE); | |||
|   } | |||
|   return *east; | |||
| } | |||
| 
 | |||
| cell cell::getSouth(const std::vector<cell> &grid) const { | |||
|   auto south = std::find_if(grid.begin(), grid.end(), [this](const cell &c) { | |||
|         return this->row + 1 == c.row && this->column == c.column; | |||
|       }); | |||
|   if(south == grid.end()) { | |||
|     throw std::logic_error{ "Cannot get cell south of (" + std::to_string(row) + "," + std::to_string(column) + ")"}; | |||
|     std::exit(EXIT_FAILURE); | |||
|   } | |||
|   return *south; | |||
| } | |||
| 
 | |||
| cell cell::getWest(const std::vector<cell> &grid) const { | |||
|   auto west = std::find_if(grid.begin(), grid.end(), [this](const cell &c) { | |||
|         return this->row == c.row && this->column - 1 == c.column; | |||
|       }); | |||
|   if(west == grid.end()) { | |||
|     throw std::logic_error{ "Cannot get cell west of (" + std::to_string(row) + "," + std::to_string(column) + ")"}; | |||
|     std::exit(EXIT_FAILURE); | |||
|   } | |||
|   return *west; | |||
| } | |||
| 
 | |||
| coordinates cell::getCoordinates() const { | |||
|   return std::make_pair(row, column); | |||
| } | |||
| 
 | |||
| std::string cell::getColor() const { | |||
|   switch(color) { | |||
|     case Color::Red:    return "red"; | |||
|     case Color::Green:  return "green"; | |||
|     case Color::Blue:   return "blue"; | |||
|     case Color::Purple: return "purple"; | |||
|     case Color::Yellow: return "yellow"; | |||
|     case Color::None:   return "transparent"; | |||
|     default: return ""; | |||
|     //case Color::Grey   = 'G',
 | |||
|   } | |||
| } | |||
| 
 | |||
| std::string getColor(Color color) { | |||
|   switch(color) { | |||
|     case Color::Red:    return "red"; | |||
|     case Color::Green:  return "green"; | |||
|     case Color::Blue:   return "blue"; | |||
|     case Color::Purple: return "purple"; | |||
|     case Color::Yellow: return "yellow"; | |||
|     case Color::None:   return "transparent"; | |||
|     default: return ""; | |||
|     //case Color::Grey   = 'G',
 | |||
|   } | |||
| } | |||
| @ -0,0 +1,62 @@ | |||
| #pragma once | |||
| 
 | |||
| #include <iostream> | |||
| #include <utility> | |||
| #include <vector> | |||
| 
 | |||
| typedef std::pair<int, int> coordinates; | |||
| 
 | |||
| class Grid; | |||
| 
 | |||
| enum class Type : char { | |||
|   Wall       = 'W', | |||
|   Floor      = ' ', | |||
|   Door       = 'D', | |||
|   LockedDoor = 'L', | |||
|   Key        = 'K', | |||
|   Ball       = 'A', | |||
|   Box        = 'B', | |||
|   Goal       = 'G', | |||
|   Lava       = 'V', | |||
|   Agent      = 'X', | |||
|   Adversary  = 'Z', | |||
|   SlipperyNorth = 'n', | |||
|   SlipperySouth = 's', | |||
|   SlipperyEast  = 'e', | |||
|   SlipperyWest  = 'w' | |||
| }; | |||
| enum class Color : char { | |||
|   Red    = 'R', | |||
|   Green  = 'G', | |||
|   Blue   = 'B', | |||
|   Purple = 'P', | |||
|   Yellow = 'Y', | |||
|   //Grey   = 'G', | |||
|   None   = ' ' | |||
| }; | |||
| 
 | |||
| constexpr std::initializer_list<Color> allColors = {Color::Red, Color::Green, Color::Blue, Color::Purple, Color::Yellow}; | |||
| std::string getColor(Color color); | |||
| 
 | |||
| class cell { | |||
|   public: | |||
|     coordinates getNorth() const { return std::make_pair(row - 1, column); } | |||
|     coordinates getSouth() const { return std::make_pair(row + 1, column); } | |||
|     coordinates getEast()  const { return std::make_pair(row, column + 1); } | |||
|     coordinates getWest()  const { return std::make_pair(row, column - 1); } | |||
| 
 | |||
|     cell getNorth(const std::vector<cell> &grid) const; | |||
|     cell getEast(const std::vector<cell> &grid) const; | |||
|     cell getSouth(const std::vector<cell> &grid) const; | |||
|     cell getWest(const std::vector<cell> &grid) const; | |||
| 
 | |||
|     friend std::ostream& operator<<(std::ostream& os, const cell& cell); | |||
| 
 | |||
|     coordinates getCoordinates() const; | |||
|     std::string getColor() const; | |||
| 
 | |||
|     int row; | |||
|     int column; | |||
|     Type type; | |||
|     Color color; | |||
| }; | |||
						
							
						
						
							1331
	
						
						util/popl.hpp
						
							File diff suppressed because it is too large
							
							
								
									View File
								
							
						
					
				File diff suppressed because it is too large
							
							
								
									View File
								
							
						
						Write
						Preview
					
					
					Loading…
					
					Cancel
						Save
					
		Reference in new issue