1

/* $Id$ */

     2

     3

/** @file station_newgrf.c Functions for dealing with station classes and custom stations. */

     4

     5

#include "stdafx.h"

     6

#include "openttd.h"

     7

#include "debug.h"

     8

#include "sprite.h"

     9

#include "station_newgrf.h"

    10

    11

static StationClass station_classes[STAT_CLASS_MAX];

    12

    13

/**

    14

 * Reset station classes to their default state.

    15

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

    16

 * entry, for standard stations and waypoints.

    17

 */

    18

void ResetStationClasses(void)

    19

{

    20

	StationClassID i;

    21

	for (i = 0; i < STAT_CLASS_MAX; i++) {

    22

		station_classes[i].id = 0;

    23

    24

		free(station_classes[i].name);

    25

		station_classes[i].name = NULL;

    26

    27

		station_classes[i].stations = 0;

    28

    29

		free(station_classes[i].spec);

    30

		station_classes[i].spec = NULL;

    31

	}

    32

    33

	// Set up initial data

    34

	station_classes[0].id = 'DFLT';

    35

	station_classes[0].name = strdup("Default");

    36

	station_classes[0].stations = 1;

    37

	station_classes[0].spec = malloc(sizeof(*station_classes[0].spec));

    38

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

    39

    40

	station_classes[1].id = 'WAYP';

    41

	station_classes[1].name = strdup("Waypoints");

    42

	station_classes[1].stations = 1;

    43

	station_classes[1].spec = malloc(sizeof(*station_classes[1].spec));

    44

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

    45

}

    46

    47

/**

    48

 * Allocate a station class for the given class id.

    49

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

    50

 * @return Index into station_classes of allocated class.

    51

 */

    52

StationClassID AllocateStationClass(uint32 class)

    53

{

    54

	StationClassID i;

    55

    56

	for (i = 0; i < STAT_CLASS_MAX; i++) {

    57

		if (station_classes[i].id == class) {

    58

			// ClassID is already allocated, so reuse it.

    59

			return i;

    60

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

    61

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

    62

			station_classes[i].id = class;

    63

			return i;

    64

		}

    65

	}

    66

    67

	DEBUG(grf, 2)("StationClassAllocate: Already allocated %d classes, using default.", STAT_CLASS_MAX);

    68

	return STAT_CLASS_DFLT;

    69

}

    70

    71

/**

    72

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

    73

 * @param sclass Index of the station class.

    74

 * @return Number of stations in the class.

    75

 */

    76

uint GetNumCustomStations(StationClassID sclass)

    77

{

    78

	assert(sclass < STAT_CLASS_MAX);

    79

	return station_classes[sclass].stations;

    80

}

    81

    82

/**

    83

 * Tie a station spec to its station class.

    84

 * @param spec The station spec.

    85

 */

    86

void SetCustomStation(StationSpec *spec)

    87

{

    88

	StationClass *station_class;

    89

	int i;

    90

    91

	assert(spec->sclass < STAT_CLASS_MAX);

    92

	station_class = &station_classes[spec->sclass];

    93

    94

	i = station_class->stations++;

    95

	station_class->spec = realloc(station_class->spec, station_class->stations * sizeof(*station_class->spec));

    96

    97

	station_class->spec[i] = spec;

    98

}

    99

   100

/**

   101

 * Retrieve a station spec from a class.

   102

 * @param sclass Index of the station class.

   103

 * @param station The station index with the class.

   104

 * @return The station spec.

   105

 */

   106

const StationSpec *GetCustomStation(StationClassID sclass, uint station)

   107

{

   108

	assert(sclass < STAT_CLASS_MAX);

   109

	if (station < station_classes[sclass].stations)

   110

		return station_classes[sclass].spec[station];

   111

   112

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

   113

	// probably was not loaded.

   114

	return NULL;

   115

}