1

/* $Id$ */

     2

     3

/** @file yapf_node.hpp */

     4

     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

	void Dump(DumpTarget &dmp) const

    25

	{

    26

		dmp.WriteTile("m_tile", m_tile);

    27

		dmp.WriteEnumT("m_td", m_td);

    28

		dmp.WriteEnumT("m_exitdir", m_exitdir);

    29

	}

    30

};

    31

    32

struct CYapfNodeKeyTrackDir : public CYapfNodeKeyExitDir

    33

{

    34

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

    35

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

    36

};

    37

    38

/** Yapf Node base */

    39

template <class Tkey_, class Tnode>

    40

struct CYapfNodeT {

    41

	typedef Tkey_ Key;

    42

	typedef Tnode Node;

    43

    44

	Tkey_       m_key;

    45

	Node       *m_hash_next;

    46

	Node       *m_parent;

    47

	int         m_cost;

    48

	int         m_estimate;

    49

    50

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

    51

	{

    52

		m_key.Set(tile, td);

    53

		m_hash_next = NULL;

    54

		m_parent = parent;

    55

		m_cost = 0;

    56

		m_estimate = 0;

    57

	}

    58

    59

	FORCEINLINE Node* GetHashNext() {return m_hash_next;}

    60

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

    61

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

    62

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

    63

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

    64

	FORCEINLINE int GetCost() {return m_cost;}

    65

	FORCEINLINE int GetCostEstimate() {return m_estimate;}

    66

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

    67

    68

	void Dump(DumpTarget &dmp) const

    69

	{

    70

		dmp.WriteStructT("m_key", &m_key);

    71

		dmp.WriteStructT("m_parent", m_parent);

    72

		dmp.WriteLine("m_cost = %d", m_cost);

    73

		dmp.WriteLine("m_estimate = %d", m_estimate);

    74

	}

    75

};

    76

    77

/** Yapf Node for ships */

    78

template <class Tkey_>

    79

struct CYapfShipNodeT

    80

	: CYapfNodeT<Tkey_, CYapfShipNodeT<Tkey_> >

    81

{

    82

    83

};

    84

    85

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

    86

typedef CYapfShipNodeT<CYapfNodeKeyExitDir>  CYapfShipNodeExitDir;

    87

typedef CYapfShipNodeT<CYapfNodeKeyTrackDir> CYapfShipNodeTrackDir;

    88

    89

// Default NodeList types

    90

typedef CNodeList_HashTableT<CYapfShipNodeExitDir , 14, 16> CShipNodeListExitDir;

    91

typedef CNodeList_HashTableT<CYapfShipNodeTrackDir, 16, 20> CShipNodeListTrackDir;

    92

    93

    94

#endif /* YAPF_NODE_HPP */