/* $Id$ */

/*

 * This file is part of OpenTTD.

 * OpenTTD is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, version 2.

 * OpenTTD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.

 * See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with OpenTTD. If not, see <http://www.gnu.org/licenses/>.

 */

/** @file newgrf_station.cpp Functions for dealing with station classes and custom stations. */

#include "stdafx.h"

#include "variables.h"

#include "landscape.h"

#include "debug.h"

#include "station_base.h"

#include "waypoint_base.h"

#include "roadstop_base.h"

#include "newgrf_cargo.h"

#include "newgrf_commons.h"

#include "newgrf_station.h"

#include "newgrf_spritegroup.h"

#include "newgrf_sound.h"

#include "town.h"

#include "newgrf_town.h"

#include "gfx_func.h"

#include "date_func.h"

#include "company_func.h"

#include "animated_tile_func.h"

#include "functions.h"

#include "tunnelbridge_map.h"

#include "spritecache.h"

#include "table/strings.h"

static StationClass _station_classes[STAT_CLASS_MAX];

enum {

	MAX_SPECLIST = 255,

};

enum TriggerArea {

	TA_TILE,

	TA_PLATFORM,

	TA_WHOLE,

};

struct ETileArea : TileArea {

	ETileArea(const BaseStation *st, TileIndex tile, TriggerArea ta)

	{

		switch (ta) {

			default: NOT_REACHED();

			case TA_TILE:

				this->tile = tile;

				this->w    = 1;

				this->h    = 1;

				break;

			case TA_PLATFORM: {

				TileIndex start, end;

				Axis axis = GetRailStationAxis(tile);

				TileIndexDiff delta = TileOffsByDiagDir(AxisToDiagDir(axis));

				for (end = tile; IsRailStationTile(end + delta) && IsCompatibleTrainStationTile(tile, end + delta); end += delta) { /* Nothing */ }

				for (start = tile; IsRailStationTile(start - delta) && IsCompatibleTrainStationTile(tile, start - delta); start -= delta) { /* Nothing */ }

				this->tile = start;

				this->w = TileX(end) - TileX(start) + 1;

				this->h = TileY(end) - TileY(start) + 1;

				break;

			}

			case TA_WHOLE:

				st->GetTileArea(this, Station::IsExpected(st) ? STATION_RAIL : STATION_WAYPOINT);

				break;

		}

	}

};

/**

 * Reset station classes to their default state.

 * This includes initialising the Default and Waypoint classes with an empty

 * entry, for standard stations and waypoints.

 */

void ResetStationClasses()

{

	for (StationClassID i = STAT_CLASS_BEGIN; i < STAT_CLASS_MAX; i++) {

		_station_classes[i].id = 0;

		_station_classes[i].name = STR_EMPTY;

		_station_classes[i].stations = 0;

		free(_station_classes[i].spec);

		_station_classes[i].spec = NULL;

	}

	/* Set up initial data */

	_station_classes[0].id = 'DFLT';

	_station_classes[0].name = STR_STATION_CLASS_DFLT;

	_station_classes[0].stations = 1;

	_station_classes[0].spec = MallocT<StationSpec*>(1);

	_station_classes[0].spec[0] = NULL;

	_station_classes[1].id = 'WAYP';

	_station_classes[1].name = STR_STATION_CLASS_WAYP;

	_station_classes[1].stations = 1;

	_station_classes[1].spec = MallocT<StationSpec*>(1);

	_station_classes[1].spec[0] = NULL;

}

/**

 * Allocate a station class for the given class id.

 * @param cls A 32 bit value identifying the class.

 * @return Index into _station_classes of allocated class.

 */

StationClassID AllocateStationClass(uint32 cls)

{

	for (StationClassID i = STAT_CLASS_BEGIN; i < STAT_CLASS_MAX; i++) {

		if (_station_classes[i].id == cls) {

			/* ClassID is already allocated, so reuse it. */

			return i;

		} else if (_station_classes[i].id == 0) {

			/* This class is empty, so allocate it to the ClassID. */

			_station_classes[i].id = cls;

			return i;

		}

	}

	grfmsg(2, "StationClassAllocate: already allocated %d classes, using default", STAT_CLASS_MAX);

	return STAT_CLASS_DFLT;

}

/** Set the name of a custom station class */

void SetStationClassName(StationClassID sclass, StringID name)

{

	assert(sclass < STAT_CLASS_MAX);

	_station_classes[sclass].name = name;

}

/** Retrieve the name of a custom station class */

StringID GetStationClassName(StationClassID sclass)

{

	assert(sclass < STAT_CLASS_MAX);

	return _station_classes[sclass].name;

}

/**

 * Get the number of station classes in use.

 * @return Number of station classes.

 */

uint GetNumStationClasses()

{

	uint i;

	for (i = 0; i < STAT_CLASS_MAX && _station_classes[i].id != 0; i++) {}

	return i;

}

/**

 * Return the number of stations for the given station class.

 * @param sclass Index of the station class.

 * @return Number of stations in the class.

 */

uint GetNumCustomStations(StationClassID sclass)

{

	assert(sclass < STAT_CLASS_MAX);

	return _station_classes[sclass].stations;

}

/**

 * Tie a station spec to its station class.

 * @param statspec The station spec.

 */

void SetCustomStationSpec(StationSpec *statspec)

{

	StationClass *station_class;

	int i;

	/* If the station has already been allocated, don't reallocate it. */

	if (statspec->allocated) return;

	assert(statspec->sclass < STAT_CLASS_MAX);

	station_class = &_station_classes[statspec->sclass];

	i = station_class->stations++;

	station_class->spec = ReallocT(station_class->spec, station_class->stations);

	station_class->spec[i] = statspec;

	statspec->allocated = true;

}

/**

 * Retrieve a station spec from a class.

 * @param sclass Index of the station class.

 * @param station The station index with the class.

 * @return The station spec.

 */

const StationSpec *GetCustomStationSpec(StationClassID sclass, uint station)

{

	assert(sclass < STAT_CLASS_MAX);

	if (station < _station_classes[sclass].stations)

		return _station_classes[sclass].spec[station];

	/* If the custom station isn't defined any more, then the GRF file

	 * probably was not loaded. */

	return NULL;

}

/**

 * Retrieve a station spec by GRF location.

 * @param grfid    GRF ID of station spec.

 * @param localidx Index within GRF file of station spec.

 * @param index    Pointer to return the index of the station spec in its station class. If NULL then not used.

 * @return The station spec.

 */

const StationSpec *GetCustomStationSpecByGrf(uint32 grfid, byte localidx, int *index)

{

	uint j;

	for (StationClassID i = STAT_CLASS_BEGIN; i < STAT_CLASS_MAX; i++) {

		for (j = 0; j < _station_classes[i].stations; j++) {

			const StationSpec *statspec = _station_classes[i].spec[j];

			if (statspec == NULL) continue;

			if (statspec->grffile->grfid == grfid && statspec->localidx == localidx) {

				if (index != NULL) *index = j;

				return statspec;

			}

		}

	}

	return NULL;

}

/* Evaluate a tile's position within a station, and return the result a bitstuffed format.

 * if not centred: .TNLcCpP, if centred: .TNL..CP

 * T = Tile layout number (GetStationGfx), N = Number of platforms, L = Length of platforms

 * C = Current platform number from start, c = from end

 * P = Position along platform from start, p = from end

 * if centred, C/P start from the centre and c/p are not available.

 */

uint32 GetPlatformInfo(Axis axis, byte tile, int platforms, int length, int x, int y, bool centred)

{

	uint32 retval = 0;

	if (axis == AXIS_X) {

		Swap(platforms, length);

		Swap(x, y);

	}

	/* Limit our sizes to 4 bits */

	platforms = min(15, platforms);

	length    = min(15, length);

	x = min(15, x);

	y = min(15, y);

	if (centred) {

		x -= platforms / 2;

		y -= length / 2;

		SB(retval,  0, 4, y & 0xF);

		SB(retval,  4, 4, x & 0xF);

	} else {

		SB(retval,  0, 4, y);

		SB(retval,  4, 4, length - y - 1);

		SB(retval,  8, 4, x);

		SB(retval, 12, 4, platforms - x - 1);

	}

	SB(retval, 16, 4, length);