123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892 |
- /*
- * MapFormatJson.cpp, part of VCMI engine
- *
- * Authors: listed in file AUTHORS in main folder
- *
- * License: GNU General Public License v2.0 or later
- * Full text of license available in license.txt file, in main folder
- *
- */
- #include "StdInc.h"
- #include "MapFormatJson.h"
- #include "../filesystem/CInputStream.h"
- #include "../filesystem/COutputStream.h"
- #include "CMap.h"
- #include "../CModHandler.h"
- #include "../CHeroHandler.h"
- #include "../CTownHandler.h"
- #include "../VCMI_Lib.h"
- #include "../mapObjects/ObjectTemplate.h"
- #include "../mapObjects/CObjectHandler.h"
- #include "../mapObjects/CObjectClassesHandler.h"
- #include "../mapObjects/CGHeroInstance.h"
- #include "../mapObjects/CGTownInstance.h"
- #include "../StringConstants.h"
- namespace TriggeredEventsDetail
- {
- static const std::array<std::string, 12> conditionNames =
- {
- "haveArtifact", "haveCreatures", "haveResources", "haveBuilding",
- "control", "destroy", "transport", "daysPassed",
- "isHuman", "daysWithoutTown", "standardWin", "constValue"
- };
- static const std::array<std::string, 2> typeNames = { "victory", "defeat" };
- static EventCondition JsonToCondition(const JsonNode & node)
- {
- EventCondition event;
- event.condition = EventCondition::EWinLoseType(vstd::find_pos(conditionNames, node.Vector()[0].String()));
- if (node.Vector().size() > 1)
- {
- const JsonNode & data = node.Vector()[1];
- if (data["type"].getType() == JsonNode::DATA_STRING)
- event.objectType = VLC->modh->identifiers.getIdentifier(data["type"]).get();
- if (data["type"].getType() == JsonNode::DATA_FLOAT)
- event.objectType = data["type"].Float();
- if (!data["value"].isNull())
- event.value = data["value"].Float();
- if (!data["position"].isNull())
- {
- auto & position = data["position"].Vector();
- event.position.x = position.at(0).Float();
- event.position.y = position.at(1).Float();
- event.position.z = position.at(2).Float();
- }
- }
- return event;
- }
- static JsonNode ConditionToJson(const EventCondition& event)
- {
- JsonNode json;
- JsonVector& asVector = json.Vector();
- JsonNode condition;
- condition.String() = conditionNames.at(event.condition);
- asVector.push_back(condition);
- JsonNode data;
- //todo: save identifier
- if(event.objectType != -1)
- data["type"].Float() = event.objectType;
- if(event.value != -1)
- data["value"].Float() = event.value;
- if(event.position != int3(-1,-1,-1))
- {
- auto & position = data["position"].Vector();
- position.resize(3);
- position[0].Float() = event.position.x;
- position[1].Float() = event.position.y;
- position[2].Float() = event.position.z;
- }
- if(!data.isNull())
- asVector.push_back(data);
- return std::move(json);
- }
- }//namespace TriggeredEventsDetail
- namespace TerrainDetail
- {
- static const std::array<std::string, 10> terrainCodes =
- {
- "dt", "sa", "gr", "sn", "sw", "rg", "sb", "lv", "wt", "rc"
- };
- static const std::array<std::string, 4> roadCodes =
- {
- "", "pd", "pg", "pc"
- };
- static const std::array<std::string, 5> riverCodes =
- {
- "", "rw", "ri", "rm", "rl"
- };
- static const std::array<char, 10> flipCodes =
- {
- '_', '-', '|', '+'
- };
- }
- static std::string tailString(const std::string & input, char separator)
- {
- std::string ret;
- size_t splitPos = input.find(separator);
- if (splitPos != std::string::npos)
- ret = input.substr(splitPos + 1);
- return ret;
- }
- static si32 extractNumber(const std::string & input, char separator)
- {
- std::string tmp = tailString(input, separator);
- return atoi(tmp.c_str());
- }
- ///CMapFormatJson
- const int CMapFormatJson::VERSION_MAJOR = 1;
- const int CMapFormatJson::VERSION_MINOR = 0;
- const std::string CMapFormatJson::HEADER_FILE_NAME = "header.json";
- const std::string CMapFormatJson::OBJECTS_FILE_NAME = "objects.json";
- void CMapFormatJson::readTriggeredEvents(const JsonNode & input)
- {
- mapHeader->victoryMessage = input["victoryString"].String();
- mapHeader->victoryIconIndex = input["victoryIconIndex"].Float();
- mapHeader->defeatMessage = input["defeatString"].String();
- mapHeader->defeatIconIndex = input["defeatIconIndex"].Float();
- mapHeader->triggeredEvents.clear();
- for (auto & entry : input["triggeredEvents"].Struct())
- {
- TriggeredEvent event;
- event.identifier = entry.first;
- readTriggeredEvent(event, entry.second);
- mapHeader->triggeredEvents.push_back(event);
- }
- }
- void CMapFormatJson::readTriggeredEvent(TriggeredEvent & event, const JsonNode & source)
- {
- using namespace TriggeredEventsDetail;
- event.onFulfill = source["message"].String();
- event.description = source["description"].String();
- event.effect.type = vstd::find_pos(typeNames, source["effect"]["type"].String());
- event.effect.toOtherMessage = source["effect"]["messageToSend"].String();
- event.trigger = EventExpression(source["condition"], JsonToCondition); // logical expression
- }
- void CMapFormatJson::writeTriggeredEvents(JsonNode& output)
- {
- output["victoryString"].String() = map->victoryMessage;
- output["victoryIconIndex"].Float() = map->victoryIconIndex;
- output["defeatString"].String() = map->defeatMessage;
- output["defeatIconIndex"].Float() = map->defeatIconIndex;
- JsonMap & triggeredEvents = output["triggeredEvents"].Struct();
- for(auto event : map->triggeredEvents)
- writeTriggeredEvent(event, triggeredEvents[event.identifier]);
- }
- void CMapFormatJson::writeTriggeredEvent(const TriggeredEvent& event, JsonNode& dest)
- {
- using namespace TriggeredEventsDetail;
- dest["message"].String() = event.onFulfill;
- dest["description"].String() = event.description;
- dest["effect"]["type"].String() = typeNames.at(size_t(event.effect.type));
- dest["effect"]["messageToSend"].String() = event.effect.toOtherMessage;
- dest["condition"] = event.trigger.toJson(ConditionToJson);
- }
- ///CMapPatcher
- CMapPatcher::CMapPatcher(JsonNode stream):
- input(stream)
- {
- }
- void CMapPatcher::patchMapHeader(std::unique_ptr<CMapHeader> & header)
- {
- header.swap(mapHeader);
- if (!input.isNull())
- readPatchData();
- header.swap(mapHeader);
- }
- void CMapPatcher::readPatchData()
- {
- readTriggeredEvents(input);
- }
- ///CMapFormatZip
- CMapFormatZip::CMapFormatZip(CInputOutputStream * stream):
- buffer(stream),
- ioApi(new CProxyIOApi(buffer))
- {
- }
- ///CMapLoaderJson
- CMapLoaderJson::CMapLoaderJson(CInputOutputStream * stream):
- CMapFormatZip(stream),
- loader("", "_", ioApi)
- {
- }
- si32 CMapLoaderJson::getIdentifier(const std::string& type, const std::string& name)
- {
- boost::optional<si32> res = VLC->modh->identifiers.getIdentifier("core", type, name, false);
- if(!res)
- {
- throw new std::runtime_error("Map load failed. Identifier not resolved.");
- }
- return res.get();
- }
- std::unique_ptr<CMap> CMapLoaderJson::loadMap()
- {
- map = new CMap();
- mapHeader = std::unique_ptr<CMapHeader>(dynamic_cast<CMapHeader *>(map));
- readMap();
- return std::unique_ptr<CMap>(dynamic_cast<CMap *>(mapHeader.release()));
- }
- std::unique_ptr<CMapHeader> CMapLoaderJson::loadMapHeader()
- {
- mapHeader.reset(new CMapHeader);
- readHeader();
- return std::move(mapHeader);
- }
- const JsonNode CMapLoaderJson::readJson(const std::string & archiveFilename)
- {
- ResourceID resource(archiveFilename, EResType::TEXT);
- if(!loader.existsResource(resource))
- throw new std::runtime_error(archiveFilename+" not found");
- auto data = loader.load(resource)->readAll();
- JsonNode res(reinterpret_cast<char*>(data.first.get()), data.second);
- return std::move(res);
- }
- void CMapLoaderJson::readMap()
- {
- readHeader();
- map->initTerrain();
- readTerrain();
- readObjects();
- // Calculate blocked / visitable positions
- for(auto & elem : map->objects)
- {
- map->addBlockVisTiles(elem);
- }
- map->calculateGuardingGreaturePositions();
- }
- void CMapLoaderJson::readHeader()
- {
- //do not use map field here, use only mapHeader
- const JsonNode header = readJson(HEADER_FILE_NAME);
- //TODO: read such data like map name & size
- //mapHeader->version = ??? //todo: new version field
- //todo: multilevel map load support
- const JsonNode levels = header["mapLevels"];
- mapHeader->height = levels["surface"]["height"].Float();
- mapHeader->width = levels["surface"]["width"].Float();
- mapHeader->twoLevel = !levels["underground"].isNull();
- mapHeader->name = header["name"].String();
- mapHeader->description = header["description"].String();
- //todo: support arbitrary percentage
- static const std::map<std::string, ui8> difficultyMap =
- {
- {"", 1},
- {"EASY", 0},
- {"NORMAL", 1},
- {"HARD", 2},
- {"EXPERT", 3},
- {"IMPOSSIBLE", 4}
- };
- mapHeader->difficulty = difficultyMap.at(header["difficulty"].String());
- mapHeader->levelLimit = header["levelLimit"].Float();
- // std::vector<bool> allowedHeroes;
- // std::vector<ui16> placeholdedHeroes;
- readTriggeredEvents(header);
- readPlayerInfo(header);
- readTeams(header);
- //TODO: readHeader
- }
- void CMapLoaderJson::readPlayerInfo(const JsonNode & input)
- {
- const JsonNode & src = input["players"];
- int howManyTeams = 0;
- for(int player = 0; player < PlayerColor::PLAYER_LIMIT_I; player++)
- {
- PlayerInfo & info = map->players.at(player);
- const JsonNode & playerSrc = src[GameConstants::PLAYER_COLOR_NAMES[player]];
- if(playerSrc.isNull())
- {
- info.canComputerPlay = false;
- info.canHumanPlay = false;
- }
- else
- {
- readPlayerInfo(info, playerSrc);
- }
- }
- mapHeader->howManyTeams = howManyTeams;
- }
- void CMapLoaderJson::readPlayerInfo(PlayerInfo& info, const JsonNode& input)
- {
- //allowed factions
- // info.isFactionRandom =
- info.canComputerPlay = true;
- info.canHumanPlay = input["canPlay"].String() != "AIOnly";
- //placedHeroes
- //mainTown
- info.generateHeroAtMainTown = input["generateHeroAtMainTown"].Bool();
- //mainHero
- //mainHeroPortrait
- //mainCustomHeroName
- }
- void CMapLoaderJson::readTeams(const JsonNode& input)
- {
- const JsonNode & src = input["teams"];
- if(src.getType() != JsonNode::DATA_VECTOR)
- {
- // No alliances
- if(src.getType() != JsonNode::DATA_NULL)
- logGlobal->errorStream() << "Invalid teams field type";
- mapHeader->howManyTeams = 0;
- for(int i = 0; i < PlayerColor::PLAYER_LIMIT_I; i++)
- {
- if(mapHeader->players[i].canComputerPlay || mapHeader->players[i].canHumanPlay)
- {
- mapHeader->players[i].team = TeamID(mapHeader->howManyTeams++);
- }
- }
- }
- else
- {
- const JsonVector & srcVector = src.Vector();
- mapHeader->howManyTeams = srcVector.size();
- for(int team = 0; team < mapHeader->howManyTeams; team++)
- {
- for(const JsonNode & playerData : srcVector[team].Vector())
- {
- PlayerColor player = PlayerColor(vstd::find_pos(GameConstants::PLAYER_COLOR_NAMES, playerData.String()));
- if(player.isValidPlayer())
- {
- if(map->players[player.getNum()].canAnyonePlay())
- {
- map->players[player.getNum()].team = TeamID(team);
- }
- }
- }
- }
- for(PlayerInfo & player : map->players)
- {
- if(player.canAnyonePlay() && player.team == TeamID::NO_TEAM)
- player.team = TeamID(mapHeader->howManyTeams++);
- }
- }
- }
- void CMapLoaderJson::readTerrainTile(const std::string& src, TerrainTile& tile)
- {
- using namespace TerrainDetail;
- {//terrain type
- const std::string typeCode = src.substr(0,2);
- int rawType = vstd::find_pos(terrainCodes, typeCode);
- if(rawType < 0)
- throw new std::runtime_error("Invalid terrain type code in "+src);
- tile.terType = ETerrainType(rawType);
- }
- int startPos = 2; //0+typeCode fixed length
- {//terrain view
- int pos = startPos;
- while(isdigit(src.at(pos)))
- pos++;
- int len = pos - startPos;
- if(len<=0)
- throw new std::runtime_error("Invalid terrain view in "+src);
- const std::string rawCode = src.substr(startPos,len);
- tile.terView = atoi(rawCode.c_str());
- startPos+=len;
- }
- {//terrain flip
- int terrainFlip = vstd::find_pos(flipCodes, src.at(startPos++));
- if(terrainFlip < 0)
- throw new std::runtime_error("Invalid terrain flip in "+src);
- else
- tile.extTileFlags = terrainFlip;
- }
- if(startPos >= src.size())
- return;
- bool hasRoad = true;
- {//road type
- const std::string typeCode = src.substr(startPos,2);
- startPos+=2;
- int rawType = vstd::find_pos(roadCodes, typeCode);
- if(rawType < 0)
- {
- rawType = vstd::find_pos(riverCodes, typeCode);
- if(rawType < 0)
- throw new std::runtime_error("Invalid river type in "+src);
- else
- {
- tile.riverType = ERiverType::ERiverType(rawType);
- hasRoad = false;
- }
- }
- else
- tile.roadType = ERoadType::ERoadType(rawType);
- }
- if(hasRoad)
- {//road dir
- int pos = startPos;
- while(isdigit(src.at(pos)))
- pos++;
- int len = pos - startPos;
- if(len<=0)
- throw new std::runtime_error("Invalid road dir in "+src);
- const std::string rawCode = src.substr(startPos,len);
- tile.roadDir = atoi(rawCode.c_str());
- startPos+=len;
- }
- if(hasRoad)
- {//road flip
- int flip = vstd::find_pos(flipCodes, src.at(startPos++));
- if(flip < 0)
- throw new std::runtime_error("Invalid road flip in "+src);
- else
- tile.extTileFlags |= (flip<<4);
- }
- if(startPos >= src.size())
- return;
- if(hasRoad)
- {//river type
- const std::string typeCode = src.substr(startPos,2);
- startPos+=2;
- int rawType = vstd::find_pos(riverCodes, typeCode);
- if(rawType < 0)
- throw new std::runtime_error("Invalid river type in "+src);
- tile.riverType = ERiverType::ERiverType(rawType);
- }
- {//river dir
- int pos = startPos;
- while(isdigit(src.at(pos)))
- pos++;
- int len = pos - startPos;
- if(len<=0)
- throw new std::runtime_error("Invalid river dir in "+src);
- const std::string rawCode = src.substr(startPos,len);
- tile.riverDir = atoi(rawCode.c_str());
- startPos+=len;
- }
- {//river flip
- int flip = vstd::find_pos(flipCodes, src.at(startPos++));
- if(flip < 0)
- throw new std::runtime_error("Invalid road flip in "+src);
- else
- tile.extTileFlags |= (flip<<2);
- }
- }
- void CMapLoaderJson::readTerrainLevel(const JsonNode& src, const int index)
- {
- int3 pos(0,0,index);
- const JsonVector & rows = src.Vector();
- if(rows.size() != map->height)
- throw new std::runtime_error("Invalid terrain data");
- for(pos.y = 0; pos.y < map->height; pos.y++)
- {
- const JsonVector & tiles = rows[pos.y].Vector();
- if(tiles.size() != map->width)
- throw new std::runtime_error("Invalid terrain data");
- for(pos.x = 0; pos.x < map->width; pos.x++)
- readTerrainTile(tiles[pos.x].String(), map->getTile(pos));
- }
- }
- void CMapLoaderJson::readTerrain()
- {
- {
- const JsonNode surface = readJson("surface_terrain.json");
- readTerrainLevel(surface, 0);
- }
- if(map->twoLevel)
- {
- const JsonNode underground = readJson("underground_terrain.json");
- readTerrainLevel(underground, 1);
- }
- }
- CMapLoaderJson::MapObjectLoader::MapObjectLoader(CMapLoaderJson * _owner, const JsonMap::value_type& json):
- owner(_owner), instance(nullptr),handler(nullptr),id(-1), jsonKey(json.first), configuration(json.second), internalId(extractNumber(jsonKey, '_'))
- {
- }
- void CMapLoaderJson::MapObjectLoader::construct()
- {
- //TODO:consider move to ObjectTypeHandler
- //find type handler
- std::string typeName = configuration["type"].String(), subTypeName = configuration["subType"].String();
- if(typeName.empty())
- {
- logGlobal->errorStream() << "Object type missing";
- return;
- }
- if(subTypeName.empty())
- {
- logGlobal->errorStream() << "Object subType missing";
- return;
- }
- //special case for grail
- if(typeName == "grail")
- {
- auto & pos = owner->map->grailPos;
- pos.x = configuration["x"].Float();
- pos.y = configuration["y"].Float();
- pos.z = configuration["l"].Float();
- owner->map->grailRadious = configuration["options"]["grailRadius"].Float();
- }
- handler = VLC->objtypeh->getHandlerFor(typeName, subTypeName);
- instance = handler->create(ObjectTemplate());
- instance->id = ObjectInstanceID(owner->map->objects.size());
- owner->map->objects.push_back(instance);
- }
- void CMapLoaderJson::MapObjectLoader::configure()
- {
- if(nullptr == instance)
- return;
- instance->readJson(configuration);
- if(instance->ID == Obj::TOWN)
- {
- owner->map->towns.push_back(static_cast<CGTownInstance *>(instance));
- }
- if(instance->ID == Obj::HERO)
- {
- logGlobal->debugStream() << "Hero: " << VLC->heroh->heroes[instance->subID]->name << " at " << instance->pos;
- owner->map->heroesOnMap.push_back(static_cast<CGHeroInstance *>(instance));
- }
- }
- void CMapLoaderJson::readObjects()
- {
- LOG_TRACE(logGlobal);
- std::vector<std::unique_ptr<MapObjectLoader>> loaders;//todo: optimize MapObjectLoader memory layout
- const JsonNode data = readJson(OBJECTS_FILE_NAME);
- //get raw data
- for(const auto & p : data.Struct())
- loaders.push_back(vstd::make_unique<MapObjectLoader>(this, p));
- auto sortInfos = [](const std::unique_ptr<MapObjectLoader> & lhs, const std::unique_ptr<MapObjectLoader> & rhs) -> bool
- {
- return lhs->internalId < rhs->internalId;
- };
- boost::sort(loaders, sortInfos);//this makes instance ids consistent after save-load, needed for testing
- //todo: use internalId in CMap
- for(auto & ptr : loaders)
- ptr->construct();
- //configure objects after all objects are constructed so we may resolve internal IDs even to actual pointers OTF
- for(auto & ptr : loaders)
- ptr->configure();
- }
- ///CMapSaverJson
- CMapSaverJson::CMapSaverJson(CInputOutputStream * stream):
- CMapFormatZip(stream),
- saver(ioApi, "_")
- {
- }
- CMapSaverJson::~CMapSaverJson()
- {
- }
- void CMapSaverJson::addToArchive(const JsonNode& data, const std::string& filename)
- {
- std::ostringstream out;
- out << data;
- out.flush();
- {
- auto s = out.str();
- std::unique_ptr<COutputStream> stream = saver.addFile(filename);
- if (stream->write((const ui8*)s.c_str(), s.size()) != s.size())
- throw new std::runtime_error("CMapSaverJson::saveHeader() zip compression failed.");
- }
- }
- void CMapSaverJson::saveMap(const std::unique_ptr<CMap>& map)
- {
- this->map = map.get();
- writeHeader();
- writeTerrain();
- writeObjects();
- }
- void CMapSaverJson::writeHeader()
- {
- JsonNode header;
- header["versionMajor"].Float() = VERSION_MAJOR;
- header["versionMinor"].Float() = VERSION_MINOR;
- //todo: multilevel map save support
- JsonNode & levels = header["mapLevels"];
- levels["surface"]["height"].Float() = map->height;
- levels["surface"]["width"].Float() = map->width;
- levels["surface"]["index"].Float() = 0;
- if(map->twoLevel)
- {
- levels["underground"]["height"].Float() = map->height;
- levels["underground"]["width"].Float() = map->width;
- levels["underground"]["index"].Float() = 1;
- }
- header["name"].String() = map->name;
- header["description"].String() = map->description;
- //todo: support arbitrary percentage
- static const std::map<ui8, std::string> difficultyMap =
- {
- {0, "EASY"},
- {1, "NORMAL"},
- {2, "HARD"},
- {3, "EXPERT"},
- {4, "IMPOSSIBLE"}
- };
- header["difficulty"].String() = difficultyMap.at(map->difficulty);
- header["levelLimit"].Float() = map->levelLimit;
- writeTriggeredEvents(header);
- writePlayerInfo(header);
- writeTeams(header);
- //todo: allowedHeroes;
- //todo: placeholdedHeroes;
- addToArchive(header, HEADER_FILE_NAME);
- }
- void CMapSaverJson::writePlayerInfo(JsonNode & output)
- {
- JsonNode & dest = output["players"];
- dest.setType(JsonNode::DATA_STRUCT);
- for(int player = 0; player < PlayerColor::PLAYER_LIMIT_I; player++)
- {
- const PlayerInfo & info = map->players[player];
- if(info.canAnyonePlay())
- writePlayerInfo(info, dest[GameConstants::PLAYER_COLOR_NAMES[player]]);
- }
- }
- void CMapSaverJson::writePlayerInfo(const PlayerInfo & info, JsonNode & output)
- {
- //allowed factions
- output["canPlay"].String() = info.canHumanPlay ? "PlayerOrAI" : "AIOnly";
- //mainTown
- output["generateHeroAtMainTown"].Bool() = info.generateHeroAtMainTown;
- //mainHero
- //towns
- //heroes
- }
- void CMapSaverJson::writeTeams(JsonNode& output)
- {
- JsonNode & dest = output["teams"];
- std::vector<std::set<PlayerColor>> teamsData;
- teamsData.resize(map->howManyTeams);
- //get raw data
- for(int idx = 0; idx < map->players.size(); idx++)
- {
- const PlayerInfo & player = map->players.at(idx);
- int team = player.team.getNum();
- if(vstd::isbetween(team, 0, map->howManyTeams-1) && player.canAnyonePlay())
- teamsData.at(team).insert(PlayerColor(idx));
- }
- //just an optimization but breaks test
- #if 0
- //remove single-member teams
- vstd::erase_if(teamsData, [](std::set<PlayerColor> & elem) -> bool
- {
- return elem.size() <= 1;
- });
- #endif
- //construct output
- dest.setType(JsonNode::DATA_VECTOR);
- for(const std::set<PlayerColor> & teamData : teamsData)
- {
- JsonNode team(JsonNode::DATA_VECTOR);
- for(const PlayerColor & player : teamData)
- {
- JsonNode member(JsonNode::DATA_STRING);
- member.String() = GameConstants::PLAYER_COLOR_NAMES[player.getNum()];
- team.Vector().push_back(std::move(member));
- }
- dest.Vector().push_back(std::move(team));
- }
- }
- const std::string CMapSaverJson::writeTerrainTile(const TerrainTile & tile)
- {
- using namespace TerrainDetail;
- std::ostringstream out;
- out.setf(std::ios::dec, std::ios::basefield);
- out.unsetf(std::ios::showbase);
- out << terrainCodes.at(int(tile.terType)) << (int)tile.terView << flipCodes[tile.extTileFlags % 4];
- if(tile.roadType != ERoadType::NO_ROAD)
- {
- out << roadCodes.at(int(tile.roadType)) << (int)tile.roadDir << flipCodes[(tile.extTileFlags >> 4) % 4];
- }
- if(tile.riverType != ERiverType::NO_RIVER)
- {
- out << riverCodes.at(int(tile.riverType)) << (int)tile.riverDir << flipCodes[(tile.extTileFlags >> 2) % 4];
- }
- return out.str();
- }
- JsonNode CMapSaverJson::writeTerrainLevel(const int index)
- {
- JsonNode data;
- int3 pos(0,0,index);
- data.Vector().resize(map->height);
- for(pos.y = 0; pos.y < map->height; pos.y++)
- {
- JsonNode & row = data.Vector()[pos.y];
- row.Vector().resize(map->width);
- for(pos.x = 0; pos.x < map->width; pos.x++)
- row.Vector()[pos.x].String() = writeTerrainTile(map->getTile(pos));
- }
- return std::move(data);
- }
- void CMapSaverJson::writeTerrain()
- {
- //todo: multilevel map save support
- JsonNode surface = writeTerrainLevel(0);
- addToArchive(surface, "surface_terrain.json");
- if(map->twoLevel)
- {
- JsonNode underground = writeTerrainLevel(1);
- addToArchive(underground, "underground_terrain.json");
- }
- }
- void CMapSaverJson::writeObjects()
- {
- JsonNode data(JsonNode::DATA_STRUCT);
- for(const CGObjectInstance * obj : map->objects)
- obj->writeJson(data[obj->getStringId()]);
- if(map->grailPos.valid())
- {
- JsonNode grail(JsonNode::DATA_STRUCT);
- grail["type"].String() = "grail";
- grail["x"].Float() = map->grailPos.x;
- grail["y"].Float() = map->grailPos.y;
- grail["l"].Float() = map->grailPos.z;
- grail["options"]["radius"].Float() = map->grailRadious;
- std::string grailId = boost::str(boost::format("grail_%d") % map->objects.size());
- data[grailId] = grail;
- }
- addToArchive(data, OBJECTS_FILE_NAME);
- }
|