1

/* $Id$ */

     2

     3

/** @file helpers.hpp */

     4

     5

#ifndef HELPERS_HPP

     6

#define HELPERS_HPP

     7

     8

#include "macros.h"

     9

    10

/** When allocating using malloc/calloc in C++ it is usually needed to cast the return value

    11

*  from void* to the proper pointer type. Another alternative would be MallocT<> as follows */

    12

template <typename T> FORCEINLINE T* MallocT(size_t num_elements)

    13

{

    14

	T *t_ptr = (T*)malloc(num_elements * sizeof(T));

    15

	return t_ptr;

    16

}

    17

/** When allocating using malloc/calloc in C++ it is usually needed to cast the return value

    18

*  from void* to the proper pointer type. Another alternative would be MallocT<> as follows */

    19

template <typename T> FORCEINLINE T* CallocT(size_t num_elements)

    20

{

    21

	T *t_ptr = (T*)calloc(num_elements, sizeof(T));

    22

	return t_ptr;

    23

}

    24

/** When allocating using malloc/calloc in C++ it is usually needed to cast the return value

    25

*  from void* to the proper pointer type. Another alternative would be MallocT<> as follows */

    26

template <typename T> FORCEINLINE T* ReallocT(T* t_ptr, size_t num_elements)

    27

{

    28

	t_ptr = (T*)realloc(t_ptr, num_elements * sizeof(T));

    29

	return t_ptr;

    30

}

    31

    32

    33

/** type safe swap operation */

    34

template<typename T> void Swap(T& a, T& b)

    35

{

    36

	T t = a;

    37

	a = b;

    38

	b = t;

    39

}

    40

    41

    42

/** returns the absolute value of (scalar) variable. @note assumes variable to be signed */

    43

template <typename T> static inline T myabs(T a) { return a < (T)0 ? -a : a; }

    44

/** returns the (absolute) difference between two (scalar) variables */

    45

template <typename T> static inline T delta(T a, T b) { return a < b ? b - a : a - b; }

    46

    47

/** Some enums need to have allowed incrementing (i.e. StationClassID) */

    48

#define DECLARE_POSTFIX_INCREMENT(type) \

    49

	FORCEINLINE type operator ++(type& e, int) \

    50

	{ \

    51

		type e_org = e; \

    52

		e = (type)((int)e + 1); \

    53

		return e_org; \

    54

	} \

    55

	FORCEINLINE type operator --(type& e, int) \

    56

	{ \

    57

		type e_org = e; \

    58

		e = (type)((int)e - 1); \

    59

		return e_org; \

    60

	}

    61

    62

    63

    64

/** Operators to allow to work with enum as with type safe bit set in C++ */

    65

# define DECLARE_ENUM_AS_BIT_SET(mask_t) \

    66

	FORCEINLINE mask_t operator | (mask_t m1, mask_t m2) {return (mask_t)((int)m1 | m2);} \

    67

	FORCEINLINE mask_t operator & (mask_t m1, mask_t m2) {return (mask_t)((int)m1 & m2);} \

    68

	FORCEINLINE mask_t operator ^ (mask_t m1, mask_t m2) {return (mask_t)((int)m1 ^ m2);} \

    69

	FORCEINLINE mask_t& operator |= (mask_t& m1, mask_t m2) {m1 = m1 | m2; return m1;} \

    70

	FORCEINLINE mask_t& operator &= (mask_t& m1, mask_t m2) {m1 = m1 & m2; return m1;} \

    71

	FORCEINLINE mask_t& operator ^= (mask_t& m1, mask_t m2) {m1 = m1 ^ m2; return m1;} \

    72

	FORCEINLINE mask_t operator ~(mask_t m) {return (mask_t)(~(int)m);}

    73

    74

    75

/** Informative template class exposing basic enumeration properties used by several

    76

 *  other templates below. Here we have only forward declaration. For each enum type

    77

 *  we will create specialization derived from MakeEnumPropsT<>.

    78

 *  i.e.:

    79

 *    template <> struct EnumPropsT<Track> : MakeEnumPropsT<Track, byte, TRACK_BEGIN, TRACK_END, INVALID_TRACK> {};

    80

 *  followed by:

    81

 *    typedef TinyEnumT<Track> TrackByte;

    82

 */

    83

template <typename Tenum_t> struct EnumPropsT;

    84

    85

/** Helper template class that makes basic properties of given enumeration type visible

    86

 *  from outsize. It is used as base class of several EnumPropsT specializations each

    87

 *  dedicated to one of commonly used enumeration types.

    88

 *  @param Tenum_t enumeration type that you want to describe

    89

 *  @param Tstorage_t what storage type would be sufficient (i.e. byte)

    90

 *  @param Tbegin first valid value from the contiguous range (i.e. TRACK_BEGIN)

    91

 *  @param Tend one past the last valid value from the contiguous range (i.e. TRACK_END)

    92

 *  @param Tinvalid value used as invalid value marker (i.e. INVALID_TRACK)

    93

 */

    94

template <typename Tenum_t, typename Tstorage_t, Tenum_t Tbegin, Tenum_t Tend, Tenum_t Tinvalid>

    95

struct MakeEnumPropsT {

    96

	typedef Tenum_t type;                     ///< enum type (i.e. Trackdir)

    97

	typedef Tstorage_t storage;               ///< storage type (i.e. byte)

    98

	static const Tenum_t begin = Tbegin;      ///< lowest valid value (i.e. TRACKDIR_BEGIN)

    99

	static const Tenum_t end = Tend;          ///< one after the last valid value (i.e. TRACKDIR_END)

   100

	static const Tenum_t invalid = Tinvalid;  ///< what value is used as invalid value (i.e. INVALID_TRACKDIR)

   101

};

   102

   103

   104

   105

/** In some cases we use byte or uint16 to store values that are defined as enum. It is

   106

	*  necessary in order to control the sizeof() such values. Some compilers make enum

   107

	*  the same size as int (4 or 8 bytes instead of 1 or 2). As a consequence the strict

   108

	*  compiler type-checking causes errors like:

   109

	*     'HasPowerOnRail' : cannot convert parameter 1 from 'byte' to 'RailType' when

   110

	*  u->u.rail.railtype is passed as argument or type RailType. In such cases it is better

   111

	*  to teach the compiler that u->u.rail.railtype is to be treated as RailType. */

   112

template <typename Tenum_t> struct TinyEnumT;

   113

   114

/** The general declaration of TinyEnumT<> (above) */

   115

template <typename Tenum_t> struct TinyEnumT

   116

{

   117

	typedef Tenum_t enum_type;                      ///< expose our enumeration type (i.e. Trackdir) to outside

   118

	typedef EnumPropsT<Tenum_t> Props;              ///< make easier access to our enumeration propeties

   119

	typedef typename Props::storage storage_type;   ///< small storage type

   120

	static const enum_type begin = Props::begin;    ///< enum beginning (i.e. TRACKDIR_BEGIN)

   121

	static const enum_type end = Props::end;        ///< enum end (i.e. TRACKDIR_END)

   122

	static const enum_type invalid = Props::invalid;///< invalid value (i.e. INVALID_TRACKDIR)

   123

   124

	storage_type m_val;  ///< here we hold the actual value in small (i.e. byte) form

   125

   126

	/** Cast operator - invoked then the value is assigned to the Tenum_t type */

   127

	FORCEINLINE operator enum_type () const

   128

	{

   129

		return (enum_type)m_val;

   130

	}

   131

   132

	/** Assignment operator (from Tenum_t type) */

   133

	FORCEINLINE TinyEnumT& operator = (enum_type e)

   134

	{

   135

		m_val = (storage_type)e; return *this;

   136

	}

   137

   138

	/** postfix ++ operator on tiny type */

   139

	FORCEINLINE TinyEnumT operator ++ (int)

   140

	{

   141

		TinyEnumT org = *this;

   142

		if (++m_val >= end) m_val -= (storage_type)(end - begin);

   143

		return org;

   144

	}

   145

   146

	/** prefix ++ operator on tiny type */

   147

	FORCEINLINE TinyEnumT& operator ++ ()

   148

	{

   149

		if (++m_val >= end) m_val -= (storage_type)(end - begin);

   150

		return *this;

   151

	}

   152

};

   153

   154

template <typename T> void ClrBitT(T &t, int bit_index)

   155

{

   156

	t = (T)(t & ~((T)1 << bit_index));

   157

}

   158

   159

template <typename T> void SetBitT(T &t, int bit_index)

   160

{

   161

	t = (T)(t | ((T)1 << bit_index));

   162

}

   163

   164

template <typename T> void ToggleBitT(T &t, int bit_index)

   165

{

   166

	t = (T)(t ^ ((T)1 << bit_index));

   167

}

   168

   169

/**

   170

 * Overflow safe template for integers, i.e. integers that will never overflow

   171

 * you multiply the maximum value with 2, or add 2, or substract somethng from

   172

 * the minimum value, etc.

   173

 * @param T     the type these integers are stored with.

   174

 * @param T_MAX the maximum value for the integers.

   175

 * @param T_MIN the minimum value for the integers.

   176

 */

   177

template <class T, T T_MAX, T T_MIN>

   178

class OverflowSafeInt

   179

{

   180

private:

   181

	/** The non-overflow safe backend to store the value in. */

   182

	T m_value;

   183

public:

   184

	OverflowSafeInt() : m_value(0) { }

   185

   186

	OverflowSafeInt(const OverflowSafeInt& other) { this->m_value = other.m_value; }

   187

	OverflowSafeInt(const int64 int_)             { this->m_value = int_; }

   188

   189

	FORCEINLINE OverflowSafeInt& operator = (const OverflowSafeInt& other) { this->m_value = other.m_value; return *this; }

   190

   191

	FORCEINLINE OverflowSafeInt operator - () const { return OverflowSafeInt(-this->m_value); }

   192

   193

	/**

   194

	 * Safe implementation of addition.

   195

	 * @param other the amount to add

   196

	 * @note when the addition would yield more than T_MAX (or less than T_MIN),

   197

	 *       it will be T_MAX (respectively T_MIN).

   198

	 */

   199

	FORCEINLINE OverflowSafeInt& operator += (const OverflowSafeInt& other)

   200

	{

   201

		if ((T_MAX - myabs(other.m_value)) < myabs(this->m_value) &&

   202

				(this->m_value < 0) == (other.m_value < 0)) {

   203

			this->m_value = (this->m_value < 0) ? T_MIN : T_MAX ;

   204

		} else {

   205

			this->m_value += other.m_value;

   206

		}

   207

		return *this;

   208

	}

   209

   210

	/* Operators for addition and substraction */

   211

	FORCEINLINE OverflowSafeInt  operator +  (const OverflowSafeInt& other) const { OverflowSafeInt result = *this; result += other; return result; }

   212

	FORCEINLINE OverflowSafeInt  operator +  (const int              other) const { OverflowSafeInt result = *this; result += (int64)other; return result; }

   213

	FORCEINLINE OverflowSafeInt  operator +  (const uint             other) const { OverflowSafeInt result = *this; result += (int64)other; return result; }

   214

	FORCEINLINE OverflowSafeInt& operator -= (const OverflowSafeInt& other)       { return *this += (-other); }

   215

	FORCEINLINE OverflowSafeInt  operator -  (const OverflowSafeInt& other) const { OverflowSafeInt result = *this; result -= other; return result; }

   216

	FORCEINLINE OverflowSafeInt  operator -  (const int              other) const { OverflowSafeInt result = *this; result -= (int64)other; return result; }

   217

	FORCEINLINE OverflowSafeInt  operator -  (const uint             other) const { OverflowSafeInt result = *this; result -= (int64)other; return result; }

   218

   219

	FORCEINLINE OverflowSafeInt& operator ++ () { return *this += 1; }

   220

	FORCEINLINE OverflowSafeInt& operator -- () { return *this += -1; }

   221

	FORCEINLINE OverflowSafeInt operator ++ (int) { OverflowSafeInt org = *this; *this += 1; return org; }

   222

	FORCEINLINE OverflowSafeInt operator -- (int) { OverflowSafeInt org = *this; *this += -1; return org; }

   223

   224

	/**

   225

	 * Safe implementation of multiplication.

   226

	 * @param factor the factor to multiply this with.

   227

	 * @note when the multiplication would yield more than T_MAX (or less than T_MIN),

   228

	 *       it will be T_MAX (respectively T_MIN).

   229

	 */

   230

	FORCEINLINE OverflowSafeInt& operator *= (const int factor)

   231

	{

   232

		if (factor != 0 && (T_MAX / myabs(factor)) < myabs(this->m_value)) {

   233

			 this->m_value = ((this->m_value < 0) == (factor < 0)) ? T_MAX : T_MIN ;

   234

		} else {

   235

			this->m_value *= factor ;

   236

		}

   237

		return *this;

   238

	}

   239

   240

	/* Operators for multiplication */

   241

	FORCEINLINE OverflowSafeInt operator * (const int64  factor) const { OverflowSafeInt result = *this; result *= factor; return result; }

   242

	FORCEINLINE OverflowSafeInt operator * (const int    factor) const { OverflowSafeInt result = *this; result *= (int64)factor; return result; }

   243

	FORCEINLINE OverflowSafeInt operator * (const uint   factor) const { OverflowSafeInt result = *this; result *= (int64)factor; return result; }

   244

	FORCEINLINE OverflowSafeInt operator * (const uint16 factor) const { OverflowSafeInt result = *this; result *= (int64)factor; return result; }

   245

	FORCEINLINE OverflowSafeInt operator * (const byte   factor) const { OverflowSafeInt result = *this; result *= (int64)factor; return result; }

   246

   247

	/* Operators for division */

   248

	FORCEINLINE OverflowSafeInt& operator /= (const int              divisor)       { this->m_value /= divisor; return *this; }

   249

	FORCEINLINE OverflowSafeInt  operator /  (const OverflowSafeInt& divisor) const { OverflowSafeInt result = *this; result /= divisor.m_value; return result; }

   250

	FORCEINLINE OverflowSafeInt  operator /  (const int              divisor) const { OverflowSafeInt result = *this; result /= divisor; return result; }

   251

	FORCEINLINE OverflowSafeInt  operator /  (const uint             divisor) const { OverflowSafeInt result = *this; result /= (int)divisor; return result; }

   252

   253

	/* Operators for modulo */

   254

	FORCEINLINE OverflowSafeInt& operator %= (const int  divisor)       { this->m_value %= divisor; return *this; }

   255

	FORCEINLINE OverflowSafeInt  operator %  (const int  divisor) const { OverflowSafeInt result = *this; result %= divisor; return result; }

   256

   257

	/* Operators for shifting */

   258

	FORCEINLINE OverflowSafeInt& operator <<= (const int shift)       { this->m_value <<= shift; return *this; }

   259

	FORCEINLINE OverflowSafeInt  operator <<  (const int shift) const { OverflowSafeInt result = *this; result <<= shift; return result; }

   260

	FORCEINLINE OverflowSafeInt& operator >>= (const int shift)       { this->m_value >>= shift; return *this; }

   261

	FORCEINLINE OverflowSafeInt  operator >>  (const int shift) const { OverflowSafeInt result = *this; result >>= shift; return result; }

   262

   263

	/* Operators for (in)equality when comparing overflow safe ints */

   264

	FORCEINLINE bool operator == (const OverflowSafeInt& other) const { return this->m_value == other.m_value; }

   265

	FORCEINLINE bool operator != (const OverflowSafeInt& other) const { return !(*this == other); }

   266

	FORCEINLINE bool operator >  (const OverflowSafeInt& other) const { return this->m_value > other.m_value; }

   267

	FORCEINLINE bool operator >= (const OverflowSafeInt& other) const { return this->m_value >= other.m_value; }

   268

	FORCEINLINE bool operator <  (const OverflowSafeInt& other) const { return !(*this >= other); }

   269

	FORCEINLINE bool operator <= (const OverflowSafeInt& other) const { return !(*this > other); }

   270

   271

	/* Operators for (in)equality when comparing non-overflow safe ints */

   272

	FORCEINLINE bool operator == (const int other) const { return this->m_value == other; }

   273

	FORCEINLINE bool operator != (const int other) const { return !(*this == other); }

   274

	FORCEINLINE bool operator >  (const int other) const { return this->m_value > other; }

   275

	FORCEINLINE bool operator >= (const int other) const { return this->m_value >= other; }

   276

	FORCEINLINE bool operator <  (const int other) const { return !(*this >= other); }

   277

	FORCEINLINE bool operator <= (const int other) const { return !(*this > other); }

   278

   279

	FORCEINLINE operator int64 () const { return this->m_value; }

   280

};

   281

   282

/* Sometimes we got int64 operator OverflowSafeInt instead of vice versa. Handle that properly */

   283

template <class T, int64 T_MAX, int64 T_MIN> FORCEINLINE OverflowSafeInt<T, T_MAX, T_MIN> operator + (int64 a, OverflowSafeInt<T, T_MAX, T_MIN> b) { return b + a; }

   284

template <class T, int64 T_MAX, int64 T_MIN> FORCEINLINE OverflowSafeInt<T, T_MAX, T_MIN> operator - (int64 a, OverflowSafeInt<T, T_MAX, T_MIN> b) { return -b + a; }

   285

template <class T, int64 T_MAX, int64 T_MIN> FORCEINLINE OverflowSafeInt<T, T_MAX, T_MIN> operator * (int64 a, OverflowSafeInt<T, T_MAX, T_MIN> b) { return b * a; }

   286

template <class T, int64 T_MAX, int64 T_MIN> FORCEINLINE OverflowSafeInt<T, T_MAX, T_MIN> operator / (int64 a, OverflowSafeInt<T, T_MAX, T_MIN> b) { return (OverflowSafeInt<T, T_MAX, T_MIN>)a / (int)b; }

   287

   288

/* Sometimes we got int operator OverflowSafeInt instead of vice versa. Handle that properly */

   289

template <class T, int64 T_MAX, int64 T_MIN> FORCEINLINE OverflowSafeInt<T, T_MAX, T_MIN> operator + (int   a, OverflowSafeInt<T, T_MAX, T_MIN> b) { return b + a; }

   290

template <class T, int64 T_MAX, int64 T_MIN> FORCEINLINE OverflowSafeInt<T, T_MAX, T_MIN> operator - (int   a, OverflowSafeInt<T, T_MAX, T_MIN> b) { return -b + a; }

   291

template <class T, int64 T_MAX, int64 T_MIN> FORCEINLINE OverflowSafeInt<T, T_MAX, T_MIN> operator * (int   a, OverflowSafeInt<T, T_MAX, T_MIN> b) { return b * a; }

   292

template <class T, int64 T_MAX, int64 T_MIN> FORCEINLINE OverflowSafeInt<T, T_MAX, T_MIN> operator / (int   a, OverflowSafeInt<T, T_MAX, T_MIN> b) { return (OverflowSafeInt<T, T_MAX, T_MIN>)a / (int)b; }

   293

   294

/* Sometimes we got uint operator OverflowSafeInt instead of vice versa. Handle that properly */

   295

template <class T, int64 T_MAX, int64 T_MIN> FORCEINLINE OverflowSafeInt<T, T_MAX, T_MIN> operator + (uint  a, OverflowSafeInt<T, T_MAX, T_MIN> b) { return b + a; }

   296

template <class T, int64 T_MAX, int64 T_MIN> FORCEINLINE OverflowSafeInt<T, T_MAX, T_MIN> operator - (uint  a, OverflowSafeInt<T, T_MAX, T_MIN> b) { return -b + a; }

   297

template <class T, int64 T_MAX, int64 T_MIN> FORCEINLINE OverflowSafeInt<T, T_MAX, T_MIN> operator * (uint  a, OverflowSafeInt<T, T_MAX, T_MIN> b) { return b * a; }

   298

template <class T, int64 T_MAX, int64 T_MIN> FORCEINLINE OverflowSafeInt<T, T_MAX, T_MIN> operator / (uint  a, OverflowSafeInt<T, T_MAX, T_MIN> b) { return (OverflowSafeInt<T, T_MAX, T_MIN>)a / (int)b; }

   299

   300

/* Sometimes we got byte operator OverflowSafeInt instead of vice versa. Handle that properly */

   301

template <class T, int64 T_MAX, int64 T_MIN> FORCEINLINE OverflowSafeInt<T, T_MAX, T_MIN> operator + (byte  a, OverflowSafeInt<T, T_MAX, T_MIN> b) { return b + a; }

   302

template <class T, int64 T_MAX, int64 T_MIN> FORCEINLINE OverflowSafeInt<T, T_MAX, T_MIN> operator - (byte  a, OverflowSafeInt<T, T_MAX, T_MIN> b) { return -b + a; }

   303

template <class T, int64 T_MAX, int64 T_MIN> FORCEINLINE OverflowSafeInt<T, T_MAX, T_MIN> operator * (byte  a, OverflowSafeInt<T, T_MAX, T_MIN> b) { return b * a; }

   304

template <class T, int64 T_MAX, int64 T_MIN> FORCEINLINE OverflowSafeInt<T, T_MAX, T_MIN> operator / (byte  a, OverflowSafeInt<T, T_MAX, T_MIN> b) { return (OverflowSafeInt<T, T_MAX, T_MIN>)a / (int)b; }

   305

   306

#endif /* HELPERS_HPP */