1

/* $Id$ */

     2

     3

#ifndef  YAPF_NODE_RAIL_HPP

     4

#define  YAPF_NODE_RAIL_HPP

     5

     6

/** key for cached segment cost for rail YAPF */

     7

struct CYapfRailSegmentKey

     8

{

     9

	uint32    m_value;

    10

    11

	FORCEINLINE CYapfRailSegmentKey(const CYapfRailSegmentKey& src) : m_value(src.m_value) {}

    12

	FORCEINLINE CYapfRailSegmentKey(const CYapfNodeKeyExitDir& node_key) {Set(node_key);}

    13

    14

	FORCEINLINE void Set(const CYapfRailSegmentKey& src) {m_value = src.m_value;}

    15

	FORCEINLINE void Set(const CYapfNodeKeyExitDir& node_key) {m_value = (((int)node_key.m_tile) << 2) | node_key.m_exitdir;}

    16

    17

	FORCEINLINE int32 CalcHash() const {return m_value;}

    18

	FORCEINLINE TileIndex GetTile() const {return (TileIndex)(m_value >> 2);}

    19

	FORCEINLINE DiagDirection GetExitDir() const {return (DiagDirection)(m_value & 3);}

    20

	FORCEINLINE bool operator == (const CYapfRailSegmentKey& other) const {return m_value == other.m_value;}

    21

};

    22

    23

/** cached segment cost for rail YAPF */

    24

struct CYapfRailSegment

    25

{

    26

	typedef CYapfRailSegmentKey Key;

    27

    28

	CYapfRailSegmentKey    m_key;

    29

	TileIndex              m_last_tile;

    30

	Trackdir               m_last_td;

    31

	int                    m_cost;

    32

	TileIndex              m_last_signal_tile;

    33

	Trackdir               m_last_signal_td;

    34

	CYapfRailSegment*      m_hash_next;

    35

	union {

    36

		byte                 m_flags;

    37

		struct {

    38

			bool                   m_end_of_line : 1;

    39

		} flags_s;

    40

	} flags_u;

    41

	byte m_reserve[3];

    42

    43

	FORCEINLINE CYapfRailSegment(const CYapfRailSegmentKey& key)

    44

		: m_key(key)

    45

		, m_last_tile(INVALID_TILE)

    46

		, m_last_td(INVALID_TRACKDIR)

    47

		, m_cost(-1)

    48

		, m_last_signal_tile(INVALID_TILE)

    49

		, m_last_signal_td(INVALID_TRACKDIR)

    50

		, m_hash_next(NULL)

    51

	{

    52

		flags_u.m_flags = 0;

    53

	}

    54

    55

	FORCEINLINE const Key& GetKey() const {return m_key;}

    56

	FORCEINLINE TileIndex GetTile() const {return m_key.GetTile();}

    57

	FORCEINLINE DiagDirection GetExitDir() const {return m_key.GetExitDir();}

    58

	FORCEINLINE CYapfRailSegment* GetHashNext() {return m_hash_next;}

    59

	FORCEINLINE void SetHashNext(CYapfRailSegment* next) {m_hash_next = next;}

    60

};

    61

    62

/** Yapf Node for rail YAPF */

    63

template <class Tkey_>

    64

struct CYapfRailNodeT

    65

	: CYapfNodeT<Tkey_, CYapfRailNodeT<Tkey_> >

    66

{

    67

	typedef CYapfNodeT<Tkey_, CYapfRailNodeT<Tkey_> > base;

    68

	typedef CYapfRailSegment CachedData;

    69

    70

	CYapfRailSegment *m_segment;

    71

	uint16            m_num_signals_passed;

    72

	union {

    73

		uint32          m_inherited_flags;

    74

		struct {

    75

			bool          m_targed_seen : 1;

    76

			bool          m_choice_seen : 1;

    77

			bool          m_last_signal_was_red : 1;

    78

		} flags_s;

    79

	} flags_u;

    80

	SignalType        m_last_red_signal_type;

    81

    82

	FORCEINLINE void Set(CYapfRailNodeT* parent, TileIndex tile, Trackdir td, bool is_choice)

    83

	{

    84

		base::Set(parent, tile, td, is_choice);

    85

		m_segment = NULL;

    86

		if (parent == NULL) {

    87

			m_num_signals_passed      = 0;

    88

			flags_u.m_inherited_flags = 0;

    89

			m_last_red_signal_type    = SIGTYPE_NORMAL;

    90

		} else {

    91

			m_num_signals_passed      = parent->m_num_signals_passed;

    92

			flags_u.m_inherited_flags = parent->flags_u.m_inherited_flags;

    93

			m_last_red_signal_type    = parent->m_last_red_signal_type;

    94

		}

    95

		flags_u.flags_s.m_choice_seen |= is_choice;

    96

	}

    97

    98

	FORCEINLINE TileIndex GetLastTile() const {assert(m_segment != NULL); return m_segment->m_last_tile;}

    99

	FORCEINLINE Trackdir GetLastTrackdir() const {assert(m_segment != NULL); return m_segment->m_last_td;}

   100

	FORCEINLINE void SetLastTileTrackdir(TileIndex tile, Trackdir td) {assert(m_segment != NULL); m_segment->m_last_tile = tile; m_segment->m_last_td = td;}

   101

};

   102

   103

// now define two major node types (that differ by key type)

   104

typedef CYapfRailNodeT<CYapfNodeKeyExitDir>  CYapfRailNodeExitDir;

   105

typedef CYapfRailNodeT<CYapfNodeKeyTrackDir> CYapfRailNodeTrackDir;

   106

   107

// Default NodeList types

   108

typedef CNodeList_HashTableT<CYapfRailNodeExitDir , 10, 12> CRailNodeListExitDir;

   109

typedef CNodeList_HashTableT<CYapfRailNodeTrackDir, 12, 16> CRailNodeListTrackDir;

   110

   111

   112

   113

#endif /* YAPF_NODE_RAIL_HPP */