wesnoth/map.cpp

/* $Id$ */
/*
   Copyright (C) 2003 by David White <davidnwhite@optusnet.com.au>
   Part of the Battle for Wesnoth Project http://wesnoth.whitevine.net

   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License.
   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY.

   See the COPYING file for more details.
*/
#include "map.hpp"

#include <algorithm>
#include <cassert>
#include <cctype>
#include <cstdlib>
#include <iostream>
#include <sstream>

gamemap::location gamemap::location::null_location;

const std::string& gamemap::terrain_name(gamemap::TERRAIN terrain) const
{
	static const std::string default_val;
	const std::map<TERRAIN,terrain_type>::const_iterator i =
	                                      letterToTerrain_.find(terrain);
	if(i == letterToTerrain_.end())
		return default_val;
	else
		return i->second.name();
}

const std::string& gamemap::underlying_terrain_name(gamemap::TERRAIN terrain) const
{
	static const std::string default_val;
	const std::map<TERRAIN,terrain_type>::const_iterator i =
	                                       letterToTerrain_.find(terrain);
	if(i == letterToTerrain_.end()) {
		return default_val;
	} else {
		if(i->second.is_alias()) {
			//we could call underlying_terrain_name, but that could allow
			//infinite recursion with bad data files, so we call terrain_name
			//to be safe
			return terrain_name(i->second.type());
		} else {
			return i->second.name();
		}
	}
}

gamemap::TERRAIN gamemap::underlying_terrain(TERRAIN terrain) const
{
	const std::map<TERRAIN,terrain_type>::const_iterator i =
	                                       letterToTerrain_.find(terrain);
	if(i == letterToTerrain_.end()) {
		return terrain;
	} else {
		return i->second.type();
	}
}

gamemap::location::location(config& cfg) : x(-1), y(-1)
{
	const std::string& xstr = cfg.values["x"];
	const std::string& ystr = cfg.values["y"];

	//the co-ordinates in config files will be 1-based, while we
	//want them as 0-based
	if(xstr.empty() == false)
		x = atoi(xstr.c_str()) - 1;

	if(ystr.empty() == false)
		y = atoi(ystr.c_str()) - 1;
}

bool gamemap::location::operator==(const gamemap::location& a) const
{
	return x == a.x && y == a.y;
}

bool gamemap::location::operator!=(const gamemap::location& a) const
{
	return !operator==(a);
}

bool gamemap::location::operator<(const gamemap::location& a) const
{
	return x < a.x || x == a.x && y < a.y;
}

gamemap::location gamemap::location::get_direction(
                                     gamemap::location::DIRECTION dir) const
{
	switch(dir) {
		case NORTH:      return gamemap::location(x,y-1);
		case NORTH_EAST: return gamemap::location(x+1,y-((x%2) == 0 ? 1:0));
		case SOUTH_EAST: return gamemap::location(x+1,y+((x%2) == 0 ? 0:1));
		case SOUTH:      return gamemap::location(x,y+1);
		case SOUTH_WEST: return gamemap::location(x-1,y+((x%2) == 0 ? 0:1));
		case NORTH_WEST: return gamemap::location(x-1,y+((x%2) == 0 ? 1:0));
		default:
			assert(false);
			return gamemap::location();
	}
}

gamemap::gamemap(config& cfg, const std::string& data) : tiles_(1)
{
	std::vector<config*>& terrains = cfg.children["terrain"];
	create_terrain_maps(terrains,terrainPrecedence_,letterToTerrain_,terrain_);

	int x = 0, y = 0;
	for(std::string::const_iterator i = data.begin(); i != data.end(); ++i) {
		char c = *i;
		if(c == '\n') {
			tiles_.push_back(std::vector<TERRAIN>());
			y = 0;
			++x;
		} else {
			if(letterToTerrain_.count(c) == 0) {
				if(isdigit(*i)) {
					startingPositions_[c - '0'] = location(x,y);
					c = CASTLE;
				} else {
					std::cerr << "Illegal character in map: '" << c << "'\n";
					throw incorrect_format_exception("Illegal character");
				}
			}

			if(c == TOWER) {
				towers_.push_back(location(x,y));
			}
			
			tiles_.back().push_back(c);

			++y;
		}
	}

	if(tiles_.empty())
		throw incorrect_format_exception("empty map");

	for(int n = 0; n != tiles_.size(); ++n) {
		if(tiles_[n].size() != this->y()) {
			tiles_.erase(tiles_.begin()+n);
			--n;
		}
	}
}

std::string gamemap::write() const
{
	std::stringstream str;
	for(int i = 0; i != x(); ++i) {
		for(int j = 0; j != y(); ++j) {
			int n;
			for(n = 0; n != 10; ++n) {
				if(startingPositions_[n] == location(i,j))
					break;
			}

			if(n < 10)
				str << n;
			else
				str << tiles_[i][j];
		}

		str << "\n";
	}

	return str.str();
}

int gamemap::x() const { return tiles_.size(); }
int gamemap::y() const { return tiles_[0].size(); }

const std::vector<gamemap::TERRAIN>& gamemap::operator[](int index) const
{
	return tiles_[index];
}

const gamemap::location& gamemap::starting_position(int n) const
{
	return startingPositions_[n];
}

int gamemap::num_starting_positions() const
{
	return sizeof(startingPositions_)/sizeof(*startingPositions_);
}

bool gamemap::is_starting_position(const gamemap::location& loc) const
{
	const gamemap::location* const end
	                  = startingPositions_+num_starting_positions();
	return std::find(startingPositions_,end,loc) != end;
}

const terrain_type& gamemap::get_terrain_info(TERRAIN terrain) const
{
	static const terrain_type default_terrain;
	const std::map<TERRAIN,terrain_type>::const_iterator i =
	                                letterToTerrain_.find(terrain);
	if(i != letterToTerrain_.end())
		return i->second;
	else
		return default_terrain;
}

const std::vector<gamemap::TERRAIN>& gamemap::get_terrain_precedence() const
{
	return terrainPrecedence_;
}

void gamemap::set_terrain(const gamemap::location& loc, gamemap::TERRAIN ter)
{
	if(!on_board(loc))
		return;

	tiles_[loc.x][loc.y] = ter;
}