Initial implementation of new team settings format

lib/mapping/MapFormatJson.cpp

@@ -109,6 +109,14 @@ namespace TerrainDetail
 	};
 }
 
+namespace MapHeaderDetail
+{
+	static const std::array<std::string, PlayerColor::PLAYER_LIMIT_I>  playerColorNames =
+	{
+		"red", "blue", "tan", "green", "orange", "purple", "teal", "pink"
+	};
+}
+
 ///CMapFormatJson
 const int CMapFormatJson::VERSION_MAJOR = 1;
 const int CMapFormatJson::VERSION_MINOR = 0;
@@ -297,6 +305,7 @@ void CMapLoaderJson::readHeader()
 
 	readTriggeredEvents(header);
 	readPlayerInfo(header);
+	readTeams(header);
 	//TODO: readHeader
 }
 
@@ -319,7 +328,6 @@ void CMapLoaderJson::readPlayerInfo(const JsonNode & input)
 		else
 		{
 			readPlayerInfo(info, playerSrc);
-			vstd::amax(howManyTeams, info.team.getNum()+1);
 		}
 	}
 	mapHeader->howManyTeams = howManyTeams;
@@ -333,7 +341,7 @@ void CMapLoaderJson::readPlayerInfo(PlayerInfo& info, const JsonNode& input)
 	info.canComputerPlay = input["canComputerPlay"].Bool();
 	info.canHumanPlay = input["canHumanPlay"].Bool();
 
-	//plasedHeroes
+	//placedHeroes
 
 	//mainTown
 
@@ -346,10 +354,45 @@ void CMapLoaderJson::readPlayerInfo(PlayerInfo& info, const JsonNode& input)
 	//mainHeroPortrait
 
 	info.mainCustomHeroName = input["mainHeroName"].String();
-	
-	info.team = TeamID(input["team"].Float());
 }
 
+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(MapHeaderDetail::playerColorNames, playerData.String()));
+				if(player.isValidPlayer())
+				{
+					map->players[player.getNum()].team = TeamID(team);
+				}
+			}
+		}
+	}
+}
 
 void CMapLoaderJson::readTerrainTile(const std::string& src, TerrainTile& tile)
 {
@@ -476,7 +519,6 @@ void CMapLoaderJson::readTerrainLevel(const JsonNode& src, const int index)
 	}
 }
 
-
 void CMapLoaderJson::readTerrain()
 {
 	{
@@ -567,6 +609,8 @@ void CMapSaverJson::writeHeader()
 
 	writePlayerInfo(header);
 
+	writeTeams(header);
+
 	//todo:	allowedHeroes;
 	//todo: placeholdedHeroes;
 
@@ -609,8 +653,43 @@ void CMapSaverJson::writePlayerInfo(const PlayerInfo & info, JsonNode & output)
 	//mainHeroPortrait
 
 	output["mainHeroName"].String() =  info.mainCustomHeroName;
+}
 
-	output["team"].Float() = info.team.getNum();	
+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))
+			teamsData.at(team).insert(PlayerColor(idx));
+	}
+
+	//remove single-member teams
+	vstd::erase_if(teamsData, [](std::set<PlayerColor> & elem) -> bool
+	{
+		return elem.size() <= 1;
+	});
+
+	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)

lib/mapping/MapFormatJson.h

@@ -143,6 +143,8 @@ private:
 	 */
 	void readPlayerInfo(PlayerInfo & info, const JsonNode & input);
 	
+	void readTeams(const JsonNode & input);
+	
 	void readTerrainTile(const std::string & src, TerrainTile & tile);
 	
 	void readTerrainLevel(const JsonNode & src, const int index);	
@@ -191,6 +193,8 @@ private:
 	 */	
 	void writePlayerInfo(const PlayerInfo & info, JsonNode & output);
 	
+	void writeTeams(JsonNode & output);	
+
 	const std::string writeTerrainTile(const TerrainTile & tile);
 	
 	JsonNode writeTerrainLevel(const int index);