src/pathfinder/yapf/yapf_node.hpp

   1 /* $Id$ */
   2
   3 /*
   4  * This file is part of OpenTTD.
   5  * OpenTTD is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, version 2.
   6  * OpenTTD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
   7  * See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with OpenTTD. If not, see <http://www.gnu.org/licenses/>.
   8  */
   9
  10 /** @file yapf_node.hpp Node in the pathfinder's graph. */
  11
  12 #ifndef YAPF_NODE_HPP
  13 #define YAPF_NODE_HPP
  14
  15 #include "../pos.h"
  16
  17 /** Yapf Node Key base class. */
  18 struct CYapfNodeKey : PathPos {
  19         inline void Set(const PathPos &pos)
  20         {
  21                 PathPos::set(pos);
  22         }
  23
  24         void Dump(DumpTarget &dmp) const
  25         {
  26                 dmp.WriteTile("m_tile", tile);
  27                 dmp.WriteEnumT("m_td", td);
  28         }
  29 };
  30
  31 /** Yapf Node Key that evaluates hash from (and compares) tile & exit dir. */
  32 struct CYapfNodeKeyExitDir : public CYapfNodeKey
  33 {
  34         DiagDirection  exitdir;
  35
  36         inline void Set(const PathPos &pos)
  37         {
  38                 CYapfNodeKey::Set(pos);
  39                 exitdir = (pos.td == INVALID_TRACKDIR) ? INVALID_DIAGDIR : TrackdirToExitdir(pos.td);
  40         }
  41
  42         inline int CalcHash() const
  43         {
  44                 return exitdir | (tile << 2);
  45         }
  46
  47         inline bool operator == (const CYapfNodeKeyExitDir& other) const
  48         {
  49                 return PathTile::operator==(other) && (exitdir == other.exitdir);
  50         }
  51
  52         void Dump(DumpTarget &dmp) const
  53         {
  54                 CYapfNodeKey::Dump(dmp);
  55                 dmp.WriteEnumT("m_exitdir", exitdir);
  56         }
  57 };
  58
  59 /** Yapf Node Key that evaluates hash from (and compares) tile & track dir. */
  60 struct CYapfNodeKeyTrackDir : public CYapfNodeKey
  61 {
  62         inline int CalcHash() const
  63         {
  64                 return (in_wormhole() ? (td + 6) : td) | (tile << 4);
  65         }
  66
  67         inline bool operator == (const CYapfNodeKeyTrackDir& other) const
  68         {
  69                 return PathTile::operator==(other) && (td == other.td);
  70         }
  71 };
  72
  73 /** Yapf Node base */
  74 template <class Tkey_, class Tnode>
  75 struct CYapfNodeT : AstarNodeBase<Tnode> {
  76         typedef AstarNodeBase<Tnode> ABase;
  77         typedef Tkey_ Key;
  78         typedef Tnode Node;
  79
  80         Tkey_       m_key;
  81
  82         inline void Set(Node *parent, const PathPos &pos)
  83         {
  84                 ABase::Set (parent);
  85                 m_key.Set(pos);
  86         }
  87
  88         inline const PathPos& GetPos() const {return m_key;}
  89         inline const Tkey_& GetKey() const {return m_key;}
  90
  91         void Dump(DumpTarget &dmp) const
  92         {
  93                 ABase::Dump(dmp);
  94                 dmp.WriteStructT("m_key", &m_key);
  95         }
  96 };
  97
  98 #endif /* YAPF_NODE_HPP */