1

/* $Id$ */

     2

     3

/** @file cargotype.cpp */

     4

     5

#include "stdafx.h"

     6

#include "openttd.h"

     7

#include "newgrf_cargo.h"

     8

#include "cargotype.h"

     9

#include "core/bitmath_func.hpp"

    10

    11

#include "table/sprites.h"

    12

#include "table/strings.h"

    13

#include "table/cargo_const.h"

    14

    15

CargoSpec _cargo[NUM_CARGO];

    16

    17

static const byte INVALID_CARGO = 0xFF;

    18

    19

/* Bitmask of cargo types available */

    20

uint32 _cargo_mask;

    21

    22

    23

void SetupCargoForClimate(LandscapeID l)

    24

{

    25

	assert(l < lengthof(_default_climate_cargo));

    26

    27

	/* Reset and disable all cargo types */

    28

	memset(_cargo, 0, sizeof(_cargo));

    29

	for (CargoID i = 0; i < lengthof(_cargo); i++) _cargo[i].bitnum = INVALID_CARGO;

    30

    31

	_cargo_mask = 0;

    32

    33

	for (CargoID i = 0; i < lengthof(_default_climate_cargo[l]); i++) {

    34

		CargoLabel cl = _default_climate_cargo[l][i];

    35

    36

		/* Bzzt: check if cl is just an index into the cargo table */

    37

		if (cl < lengthof(_default_cargo)) {

    38

			/* Copy the indexed cargo */

    39

			_cargo[i] = _default_cargo[cl];

    40

			if (_cargo[i].bitnum != INVALID_CARGO) SetBit(_cargo_mask, i);

    41

			continue;

    42

		}

    43

    44

		/* Loop through each of the default cargo types to see if

    45

		 * the label matches */

    46

		for (uint j = 0; j < lengthof(_default_cargo); j++) {

    47

			if (_default_cargo[j].label == cl) {

    48

				_cargo[i] = _default_cargo[j];

    49

    50

				/* Populate the available cargo mask */

    51

				SetBit(_cargo_mask, i);

    52

				break;

    53

			}

    54

		}

    55

	}

    56

}

    57

    58

    59

const CargoSpec *GetCargo(CargoID c)

    60

{

    61

	assert(c < lengthof(_cargo));

    62

	return &_cargo[c];

    63

}

    64

    65

    66

bool CargoSpec::IsValid() const

    67

{

    68

	return bitnum != INVALID_CARGO;

    69

}

    70

    71

    72

CargoID GetCargoIDByLabel(CargoLabel cl)

    73

{

    74

	for (CargoID c = 0; c < lengthof(_cargo); c++) {

    75

		if (_cargo[c].bitnum == INVALID_CARGO) continue;

    76

		if (_cargo[c].label == cl) return c;

    77

	}

    78

    79

	/* No matching label was found, so it is invalid */

    80

	return CT_INVALID;

    81

}

    82

    83

    84

/** Find the CargoID of a 'bitnum' value.

    85

 * @param bitnum 'bitnum' to find.

    86

 * @return First CargoID with the given bitnum, or CT_INVALID if not found.

    87

 */

    88

CargoID GetCargoIDByBitnum(uint8 bitnum)

    89

{

    90

	if (bitnum == INVALID_CARGO) return CT_INVALID;

    91

    92

	for (CargoID c = 0; c < lengthof(_cargo); c++) {

    93

		if (_cargo[c].bitnum == bitnum) return c;

    94

	}

    95

    96

	/* No matching label was found, so it is invalid */

    97

	return CT_INVALID;

    98

}

    99