yapf_common.hpp

Go to the documentation of this file.

/*
 * This file is part of OpenTTD.
 * 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.
 * 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.
 * 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/>.
 */
 
#ifndef YAPF_COMMON_HPP
#define YAPF_COMMON_HPP
 
template <class Types>
class CYapfOriginTileT
{
public:
  typedef typename Types::Tpf Tpf;              
  typedef typename Types::NodeList::Titem Node; 
  typedef typename Node::Key Key;               
 
protected:
  TileIndex    m_orgTile;                       
  TrackdirBits m_orgTrackdirs;                  
 
  inline Tpf &Yapf()
  {
    return *static_cast<Tpf *>(this);
  }
 
public:
  void SetOrigin(TileIndex tile, TrackdirBits trackdirs)
  {
    m_orgTile = tile;
    m_orgTrackdirs = trackdirs;
  }
 
  void PfSetStartupNodes()
  {
    bool is_choice = (KillFirstBit(m_orgTrackdirs) != TRACKDIR_BIT_NONE);
    for (TrackdirBits tdb = m_orgTrackdirs; tdb != TRACKDIR_BIT_NONE; tdb = KillFirstBit(tdb)) {
      Trackdir td = (Trackdir)FindFirstBit(tdb);
      Node &n1 = Yapf().CreateNewNode();
      n1.Set(nullptr, m_orgTile, td, is_choice);
      Yapf().AddStartupNode(n1);
    }
  }
};
 
template <class Types>
class CYapfOriginTileTwoWayT
{
public:
  typedef typename Types::Tpf Tpf;              
  typedef typename Types::NodeList::Titem Node; 
  typedef typename Node::Key Key;               
 
protected:
  TileIndex   m_orgTile;                        
  Trackdir    m_orgTd;                          
  TileIndex   m_revTile;                        
  Trackdir    m_revTd;                          
  int         m_reverse_penalty;                
  bool        m_treat_first_red_two_way_signal_as_eol; 
 
  inline Tpf &Yapf()
  {
    return *static_cast<Tpf *>(this);
  }
 
public:
  void SetOrigin(TileIndex tile, Trackdir td, TileIndex tiler = INVALID_TILE, Trackdir tdr = INVALID_TRACKDIR, int reverse_penalty = 0, bool treat_first_red_two_way_signal_as_eol = true)
  {
    m_orgTile = tile;
    m_orgTd = td;
    m_revTile = tiler;
    m_revTd = tdr;
    m_reverse_penalty = reverse_penalty;
    m_treat_first_red_two_way_signal_as_eol = treat_first_red_two_way_signal_as_eol;
  }
 
  void PfSetStartupNodes()
  {
    if (m_orgTile != INVALID_TILE && m_orgTd != INVALID_TRACKDIR) {
      Node &n1 = Yapf().CreateNewNode();
      n1.Set(nullptr, m_orgTile, m_orgTd, false);
      Yapf().AddStartupNode(n1);
    }
    if (m_revTile != INVALID_TILE && m_revTd != INVALID_TRACKDIR) {
      Node &n2 = Yapf().CreateNewNode();
      n2.Set(nullptr, m_revTile, m_revTd, false);
      n2.m_cost = m_reverse_penalty;
      Yapf().AddStartupNode(n2);
    }
  }
 
  inline bool TreatFirstRedTwoWaySignalAsEOL()
  {
    return Yapf().PfGetSettings().rail_firstred_twoway_eol && m_treat_first_red_two_way_signal_as_eol;
  }
};
 
template <class Types>
class CYapfDestinationTileT
{
public:
  typedef typename Types::Tpf Tpf;              
  typedef typename Types::NodeList::Titem Node; 
  typedef typename Node::Key Key;               
 
protected:
  TileIndex    m_destTile;                      
  TrackdirBits m_destTrackdirs;                 
 
public:
  void SetDestination(TileIndex tile, TrackdirBits trackdirs)
  {
    m_destTile = tile;
    m_destTrackdirs = trackdirs;
  }
 
protected:
  Tpf &Yapf()
  {
    return *static_cast<Tpf *>(this);
  }
 
public:
  inline bool PfDetectDestination(Node &n)
  {
    return (n.m_key.m_tile == m_destTile) && HasTrackdir(m_destTrackdirs, n.GetTrackdir());
  }
 
  inline bool PfCalcEstimate(Node &n)
  {
    static const int dg_dir_to_x_offs[] = {-1, 0, 1, 0};
    static const int dg_dir_to_y_offs[] = {0, 1, 0, -1};
    if (PfDetectDestination(n)) {
      n.m_estimate = n.m_cost;
      return true;
    }
 
    TileIndex tile = n.GetTile();
    DiagDirection exitdir = TrackdirToExitdir(n.GetTrackdir());
    int x1 = 2 * TileX(tile) + dg_dir_to_x_offs[(int)exitdir];
    int y1 = 2 * TileY(tile) + dg_dir_to_y_offs[(int)exitdir];
    int x2 = 2 * TileX(m_destTile);
    int y2 = 2 * TileY(m_destTile);
    int dx = abs(x1 - x2);
    int dy = abs(y1 - y2);
    int dmin = std::min(dx, dy);
    int dxy = abs(dx - dy);
    int d = dmin * YAPF_TILE_CORNER_LENGTH + (dxy - 1) * (YAPF_TILE_LENGTH / 2);
    n.m_estimate = n.m_cost + d;
    assert(n.m_estimate >= n.m_parent->m_estimate);
    return true;
  }
};
 
template <class Ttypes>
class CYapfT
  : public Ttypes::PfBase         
  , public Ttypes::PfCost         
  , public Ttypes::PfCache        
  , public Ttypes::PfOrigin       
  , public Ttypes::PfDestination  
  , public Ttypes::PfFollow       
{
};
 
 
 
#endif /* YAPF_COMMON_HPP */