1

/* $Id$ */

     2

     3

#include "stdafx.h"

     4

#include "openttd.h"

     5

#include "variables.h"

     6

#include "macros.h"

     7

#include "pool.h"

     8

#include "newgrf_spritegroup.h"

     9

#include "date.h"

    10

    11

enum {

    12

	SPRITEGROUP_POOL_BLOCK_SIZE_BITS = 4, /* (1 << 4) == 16 items */

    13

	SPRITEGROUP_POOL_MAX_BLOCKS      = 8000,

    14

};

    15

    16

static uint _spritegroup_count = 0;

    17

static MemoryPool _spritegroup_pool;

    18

    19

void DestroySpriteGroup(SpriteGroup *group)

    20

{

    21

	/* Free dynamically allocated memory */

    22

	/* XXX Cast away the consts due to MSVC being buggy... */

    23

	switch (group->type) {

    24

		case SGT_REAL:

    25

			free((SpriteGroup**)group->g.real.loaded);

    26

			free((SpriteGroup**)group->g.real.loading);

    27

			break;

    28

    29

		case SGT_DETERMINISTIC:

    30

			free(group->g.determ.adjusts);

    31

			free(group->g.determ.ranges);

    32

			break;

    33

    34

		case SGT_RANDOMIZED:

    35

			free((SpriteGroup**)group->g.random.groups);

    36

			break;

    37

    38

		default:

    39

			break;

    40

	}

    41

}

    42

    43

static void SpriteGroupPoolCleanBlock(uint start_item, uint end_item)

    44

{

    45

	uint i;

    46

    47

	for (i = start_item; i <= end_item; i++) {

    48

		DestroySpriteGroup((SpriteGroup*)GetItemFromPool(&_spritegroup_pool, i));

    49

	}

    50

}

    51

    52

/* Initialize the SpriteGroup pool */

    53

static MemoryPool _spritegroup_pool = { "SpriteGr", SPRITEGROUP_POOL_MAX_BLOCKS, SPRITEGROUP_POOL_BLOCK_SIZE_BITS, sizeof(SpriteGroup), NULL, &SpriteGroupPoolCleanBlock, 0, 0, NULL };

    54

    55

    56

/* Allocate a new SpriteGroup */

    57

SpriteGroup *AllocateSpriteGroup(void)

    58

{

    59

	/* This is totally different to the other pool allocators, as we never remove an item from the pool. */

    60

	if (_spritegroup_count == _spritegroup_pool.total_items) {

    61

		if (!AddBlockToPool(&_spritegroup_pool)) return NULL;

    62

	}

    63

    64

	return (SpriteGroup*)GetItemFromPool(&_spritegroup_pool, _spritegroup_count++);

    65

}

    66

    67

    68

void InitializeSpriteGroupPool(void)

    69

{

    70

	CleanPool(&_spritegroup_pool);

    71

    72

	_spritegroup_count = 0;

    73

}

    74

    75

    76

static inline uint32 GetVariable(const ResolverObject *object, byte variable, byte parameter, bool *available)

    77

{

    78

	/* Return common variables */

    79

	switch (variable) {

    80

		case 0x00: return max(_date - DAYS_TILL_ORIGINAL_BASE_YEAR, 0);

    81

		case 0x01: return clamp(_cur_year, ORIGINAL_BASE_YEAR, ORIGINAL_MAX_YEAR) - ORIGINAL_BASE_YEAR;

    82

		case 0x02: return _cur_month;

    83

		case 0x03: return _opt.landscape;

    84

		case 0x09: return _date_fract;

    85

		case 0x0A: return _tick_counter;

    86

		case 0x0C: return object->callback;

    87

		case 0x10: return object->callback_param1;

    88

		case 0x11: return 0;

    89

		case 0x18: return object->callback_param2;

    90

		case 0x1A: return -1;

    91

		case 0x1B: return GB(_display_opt, 0, 6);

    92

		case 0x1C: return object->last_value;

    93

		case 0x20: return _opt.landscape == LT_HILLY ? _opt.snow_line : 0xFF;

    94

    95

		/* Not a common variable, so evalute the feature specific variables */

    96

		default: return object->GetVariable(object, variable, parameter, available);

    97

	}

    98

}

    99

   100

   101

/* Evaluate an adjustment for a variable of the given size. This is a bit of

   102

 * an unwieldy macro, but it saves triplicating the code. */

   103

#define BUILD_EVAL_ADJUST(size, usize) \

   104

static inline usize EvalAdjust_ ## size(const DeterministicSpriteGroupAdjust *adjust, usize last_value, int32 value) \

   105

{ \

   106

	value >>= adjust->shift_num; \

   107

	value  &= adjust->and_mask; \

   108

\

   109

	if (adjust->type != DSGA_TYPE_NONE) value += (size)adjust->add_val; \

   110

\

   111

	switch (adjust->type) { \

   112

		case DSGA_TYPE_DIV:  value /= (size)adjust->divmod_val; break; \

   113

		case DSGA_TYPE_MOD:  value %= (usize)adjust->divmod_val; break; \

   114

		case DSGA_TYPE_NONE: break; \

   115

	} \

   116

\

   117

	/* Get our value to the correct range */ \

   118

	value = (usize)value; \

   119

\

   120

	switch (adjust->operation) { \

   121

		case DSGA_OP_ADD:  return last_value + value; \

   122

		case DSGA_OP_SUB:  return last_value - value; \

   123

		case DSGA_OP_SMIN: return min(last_value, value); \

   124

		case DSGA_OP_SMAX: return max(last_value, value); \

   125

		case DSGA_OP_UMIN: return min((usize)last_value, (usize)value); \

   126

		case DSGA_OP_UMAX: return max((usize)last_value, (usize)value); \

   127

		case DSGA_OP_SDIV: return last_value / value; \

   128

		case DSGA_OP_SMOD: return last_value % value; \

   129

		case DSGA_OP_UDIV: return (usize)last_value / (usize)value; \

   130

		case DSGA_OP_UMOD: return (usize)last_value % (usize)value; \

   131

		case DSGA_OP_MUL:  return last_value * value; \

   132

		case DSGA_OP_AND:  return last_value & value; \

   133

		case DSGA_OP_OR:   return last_value | value; \

   134

		case DSGA_OP_XOR:  return last_value ^ value; \

   135

		default:           return value; \

   136

	} \

   137

}

   138

   139

   140

BUILD_EVAL_ADJUST(int8, uint8)

   141

BUILD_EVAL_ADJUST(int16, uint16)

   142

BUILD_EVAL_ADJUST(int32, uint32)

   143

   144

   145

static inline const SpriteGroup *ResolveVariable(const SpriteGroup *group, ResolverObject *object)

   146

{

   147

	static SpriteGroup nvarzero;

   148

	int32 last_value = object->last_value;

   149

	int32 value = -1;

   150

	uint i;

   151

   152

	object->scope = group->g.determ.var_scope;

   153

   154

	for (i = 0; i < group->g.determ.num_adjusts; i++) {

   155

		DeterministicSpriteGroupAdjust *adjust = &group->g.determ.adjusts[i];

   156

   157

		/* Try to get the variable. We shall assume it is available, unless told otherwise. */

   158

		bool available = true;

   159

		value = GetVariable(object, adjust->variable, adjust->parameter, &available);

   160

   161

		if (!available) {

   162

			/* Unsupported property: skip further processing and return either

   163

			 * the group from the first range or the default group. */

   164

			return Resolve(group->g.determ.num_ranges > 0 ? group->g.determ.ranges[0].group : group->g.determ.default_group, object);

   165

		}

   166

   167

		switch (group->g.determ.size) {

   168

			case DSG_SIZE_BYTE:  value = EvalAdjust_int8(adjust, last_value, value); break;

   169

			case DSG_SIZE_WORD:  value = EvalAdjust_int16(adjust, last_value, value); break;

   170

			case DSG_SIZE_DWORD: value = EvalAdjust_int32(adjust, last_value, value); break;

   171

			default: NOT_REACHED(); break;

   172

		}

   173

		last_value = value;

   174

	}

   175

   176

	if (group->g.determ.num_ranges == 0) {

   177

		/* nvar == 0 is a special case -- we turn our value into a callback result */

   178

		nvarzero.type = SGT_CALLBACK;

   179

		nvarzero.g.callback.result = GB(value, 0, 15);

   180

		return &nvarzero;

   181

	}

   182

   183

	for (i = 0; i < group->g.determ.num_ranges; i++) {

   184

		if (group->g.determ.ranges[i].low <= (uint32)value && (uint32)value <= group->g.determ.ranges[i].high) {

   185

			return Resolve(group->g.determ.ranges[i].group, object);

   186

		}

   187

	}

   188

   189

	return Resolve(group->g.determ.default_group, object);

   190

}

   191

   192

   193

static inline const SpriteGroup *ResolveRandom(const SpriteGroup *group, ResolverObject *object)

   194

{

   195

	uint32 mask;

   196

	byte index;

   197

   198

	object->scope = group->g.random.var_scope;

   199

   200

	if (object->trigger != 0) {

   201

		/* Handle triggers */

   202

		/* Magic code that may or may not do the right things... */

   203

		byte waiting_triggers = object->GetTriggers(object);

   204

		byte match = group->g.random.triggers & (waiting_triggers | object->trigger);

   205

		bool res;

   206

   207

		res = (group->g.random.cmp_mode == RSG_CMP_ANY) ?

   208

			(match != 0) : (match == group->g.random.triggers);

   209

   210

		if (res) {

   211

			waiting_triggers &= ~match;

   212

			object->reseed |= (group->g.random.num_groups - 1) << group->g.random.lowest_randbit;

   213

		} else {

   214

			waiting_triggers |= object->trigger;

   215

		}

   216

   217

		object->SetTriggers(object, waiting_triggers);

   218

	}

   219

   220

	mask  = (group->g.random.num_groups - 1) << group->g.random.lowest_randbit;

   221

	index = (object->GetRandomBits(object) & mask) >> group->g.random.lowest_randbit;

   222

   223

	return Resolve(group->g.random.groups[index], object);

   224

}

   225

   226

   227

/* ResolverObject (re)entry point */

   228

const SpriteGroup *Resolve(const SpriteGroup *group, ResolverObject *object)

   229

{

   230

	/* We're called even if there is no group, so quietly return nothing */

   231

	if (group == NULL) return NULL;

   232

   233

	switch (group->type) {

   234

		case SGT_REAL:          return object->ResolveReal(object, group);

   235

		case SGT_DETERMINISTIC: return ResolveVariable(group, object);

   236

		case SGT_RANDOMIZED:    return ResolveRandom(group, object);

   237

		default:                return group;

   238

	}

   239

}