1

/* $Id$ */

     2

     5

#ifndef  YAPF_NODE_HPP

     6

#define  YAPF_NODE_HPP

     7

     8

/** Yapf Node Key that evaluates hash from (and compares) tile & exit dir. */

     9

struct CYapfNodeKeyExitDir {

    10

	TileIndex      m_tile;

    11

	Trackdir       m_td;

    12

	DiagDirection  m_exitdir;

    13

    14

	FORCEINLINE void Set(TileIndex tile, Trackdir td)

    15

	{

    16

		m_tile = tile;

    17

		m_td = td;

    18

		m_exitdir = (m_td == INVALID_TRACKDIR) ? INVALID_DIAGDIR : TrackdirToExitdir(m_td);

    19

	}

    20

    21

	FORCEINLINE int CalcHash() const {return m_exitdir | (m_tile << 2);}

    22

	FORCEINLINE bool operator == (const CYapfNodeKeyExitDir& other) const {return (m_tile == other.m_tile) && (m_exitdir == other.m_exitdir);}

    23

};

    24

    25

struct CYapfNodeKeyTrackDir : public CYapfNodeKeyExitDir

    26

{

    27

	FORCEINLINE int CalcHash() const {return m_td | (m_tile << 4);}

    28

	FORCEINLINE bool operator == (const CYapfNodeKeyTrackDir& other) const {return (m_tile == other.m_tile) && (m_td == other.m_td);}

    29

};

    30

    31

/** Yapf Node base */

    32

template <class Tkey_, class Tnode>

    33

struct CYapfNodeT {

    34

	typedef Tkey_ Key;

    35

	typedef Tnode Node;

    36

    37

	Tkey_       m_key;

    38

	Node       *m_hash_next;

    39

	Node       *m_parent;

    40

	int         m_cost;

    41

	int         m_estimate;

    42

    43

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

    44

	{

    45

		m_key.Set(tile, td);

    46

		m_hash_next = NULL;

    47

		m_parent = parent;

    48

		m_cost = 0;

    49

		m_estimate = 0;

    50

	}

    51

    52

	FORCEINLINE Node* GetHashNext() {return m_hash_next;}

    53

	FORCEINLINE void SetHashNext(Node *pNext) {m_hash_next = pNext;}

    54

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

    55

	FORCEINLINE Trackdir GetTrackdir() const {return m_key.m_td;}

    56

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

    57

	FORCEINLINE int GetCost() {return m_cost;}

    58

	FORCEINLINE int GetCostEstimate() {return m_estimate;}

    59

	FORCEINLINE bool operator < (const Node& other) const {return m_estimate < other.m_estimate;}

    60

};

    61

    62

/** Yapf Node for ships */

    63

template <class Tkey_>

    64

struct CYapfShipNodeT

    65

	: CYapfNodeT<Tkey_, CYapfShipNodeT<Tkey_> >

    66

{

    67

    68

};

    69

    70

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

    71

typedef CYapfShipNodeT<CYapfNodeKeyExitDir>  CYapfShipNodeExitDir;

    72

typedef CYapfShipNodeT<CYapfNodeKeyTrackDir> CYapfShipNodeTrackDir;

    73

    74

// Default NodeList types

    75

typedef CNodeList_HashTableT<CYapfShipNodeExitDir , 14, 16> CShipNodeListExitDir;

    76

typedef CNodeList_HashTableT<CYapfShipNodeTrackDir, 16, 20> CShipNodeListTrackDir;

    77

    78

    79

#endif /* YAPF_NODE_HPP */