1

/* $Id$ */

     2

     5

#ifndef  YAPF_NODE_RAIL_HPP

     6

#define  YAPF_NODE_RAIL_HPP

     7

     8

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

     9

struct CYapfRailSegmentKey

    10

{

    11

	uint32    m_value;

    12

    13

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

    14

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

    15

    16

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

    17

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

    18

    19

	FORCEINLINE int32 CalcHash() const {return m_value;}

    20

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

    21

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

    22

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

    23

};

    24

    25

/** cached segment cost for rail YAPF */

    26

struct CYapfRailSegment

    27

{

    28

	typedef CYapfRailSegmentKey Key;

    29

    30

	CYapfRailSegmentKey    m_key;

    31

	TileIndex              m_last_tile;

    32

	Trackdir               m_last_td;

    33

	int                    m_cost;

    34

	TileIndex              m_last_signal_tile;

    35

	Trackdir               m_last_signal_td;

    36

	CYapfRailSegment*      m_hash_next;

    37

	union {

    38

		byte                 m_flags;

    39

		struct {

    40

			bool                   m_end_of_line : 1;

    41

		} flags_s;

    42

	} flags_u;

    43

	byte m_reserve[3];

    44

    45

	FORCEINLINE CYapfRailSegment(const CYapfRailSegmentKey& key)

    46

		: m_key(key)

    47

		, m_last_tile(INVALID_TILE)

    48

		, m_last_td(INVALID_TRACKDIR)

    49

		, m_cost(-1)

    50

		, m_last_signal_tile(INVALID_TILE)

    51

		, m_last_signal_td(INVALID_TRACKDIR)

    52

		, m_hash_next(NULL)

    53

	{

    54

		flags_u.m_flags = 0;

    55

	}

    56

    57

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

    58

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

    59

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

    60

	FORCEINLINE CYapfRailSegment* GetHashNext() {return m_hash_next;}

    61

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

    62

};

    63

    64

/** Yapf Node for rail YAPF */

    65

template <class Tkey_>

    66

struct CYapfRailNodeT

    67

	: CYapfNodeT<Tkey_, CYapfRailNodeT<Tkey_> >

    68

{

    69

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

    70

	typedef CYapfRailSegment CachedData;

    71

    72

	CYapfRailSegment *m_segment;

    73

	uint16            m_num_signals_passed;

    74

	union {

    75

		uint32          m_inherited_flags;

    76

		struct {

    77

			bool          m_targed_seen : 1;

    78

			bool          m_choice_seen : 1;

    79

			bool          m_last_signal_was_red : 1;

    80

		} flags_s;

    81

	} flags_u;

    82

	SignalType        m_last_red_signal_type;

    83

    84

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

    85

	{

    86

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

    87

		m_segment = NULL;

    88

		if (parent == NULL) {

    89

			m_num_signals_passed      = 0;

    90

			flags_u.m_inherited_flags = 0;

    91

			m_last_red_signal_type    = SIGTYPE_NORMAL;

    92

		} else {

    93

			m_num_signals_passed      = parent->m_num_signals_passed;

    94

			flags_u.m_inherited_flags = parent->flags_u.m_inherited_flags;

    95

			m_last_red_signal_type    = parent->m_last_red_signal_type;

    96

		}

    97

		flags_u.flags_s.m_choice_seen |= is_choice;

    98

	}

    99

   100

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

   101

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

   102

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

   103

};

   104

   105

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

   106

typedef CYapfRailNodeT<CYapfNodeKeyExitDir>  CYapfRailNodeExitDir;

   107

typedef CYapfRailNodeT<CYapfNodeKeyTrackDir> CYapfRailNodeTrackDir;

   108

   109

// Default NodeList types

   110

typedef CNodeList_HashTableT<CYapfRailNodeExitDir , 10, 12> CRailNodeListExitDir;

   111

typedef CNodeList_HashTableT<CYapfRailNodeTrackDir, 12, 16> CRailNodeListTrackDir;

   112

   113

   114

   115

#endif /* YAPF_NODE_RAIL_HPP */