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
 
#include "../../core/bitmath_func.hpp"
#include "../../tile_type.h"
#include "../../track_type.h"
 
template <class Types>
class CYapfOriginTileT {
public:
  typedef typename Types::Tpf Tpf; 
  typedef typename Types::NodeList::Item Node; 
  typedef typename Node::Key Key; 
 
protected:
  TileIndex origin_tile; 
  TrackdirBits origin_trackdirs; 
 
  inline Tpf &Yapf()
  {
    return *static_cast<Tpf *>(this);
  }
 
public:
  void SetOrigin(TileIndex tile, TrackdirBits trackdirs)
  {
    this->origin_tile = tile;
    this->origin_trackdirs = trackdirs;
  }
 
  void PfSetStartupNodes()
  {
    bool is_choice = (KillFirstBit(this->origin_trackdirs) != TRACKDIR_BIT_NONE);
    for (TrackdirBits tdb = this->origin_trackdirs; tdb != TRACKDIR_BIT_NONE; tdb = KillFirstBit(tdb)) {
      Trackdir td = (Trackdir)FindFirstBit(tdb);
      Node &n1 = Yapf().CreateNewNode();
      n1.Set(nullptr, this->origin_tile, td, is_choice);
      Yapf().AddStartupNode(n1);
    }
  }
};
 
template <class Types>
class CYapfOriginTileTwoWayT {
public:
  typedef typename Types::Tpf Tpf; 
  typedef typename Types::NodeList::Item Node; 
  typedef typename Node::Key Key; 
 
protected:
  TileIndex origin_tile; 
  Trackdir origin_td; 
  TileIndex reverse_tile; 
  Trackdir reverse_td; 
  int reverse_penalty; 
 
  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)
  {
    this->origin_tile = tile;
    this->origin_td = td;
    this->reverse_tile = tiler;
    this->reverse_td = tdr;
    this->reverse_penalty = reverse_penalty;
  }
 
  void PfSetStartupNodes()
  {
    if (this->origin_tile != INVALID_TILE && this->origin_td != INVALID_TRACKDIR) {
      Node &n1 = Yapf().CreateNewNode();
      n1.Set(nullptr, this->origin_tile, this->origin_td, false);
      Yapf().AddStartupNode(n1);
    }
    if (this->reverse_tile != INVALID_TILE && this->reverse_td != INVALID_TRACKDIR) {
      Node &n2 = Yapf().CreateNewNode();
      n2.Set(nullptr, this->reverse_tile, this->reverse_td, false);
      n2.cost = this->reverse_penalty;
      Yapf().AddStartupNode(n2);
    }
  }
};
 
template <class Ttypes>
class CYapfT
  : public Ttypes::PfBase         
  , public Ttypes::PfCost         
  , public Ttypes::PfCache        
  , public Ttypes::PfOrigin       
  , public Ttypes::PfDestination  
  , public Ttypes::PfFollow       
{
};
 
inline int OctileDistanceCost(TileIndex start_tile, Trackdir start_td, TileIndex destination_tile)
{
  static constexpr int dg_dir_to_x_offs[] = {-1, 0, 1, 0};
  static constexpr int dg_dir_to_y_offs[] = {0, 1, 0, -1};
 
  const DiagDirection exitdir = TrackdirToExitdir(start_td);
 
  const int x1 = 2 * TileX(start_tile) + dg_dir_to_x_offs[static_cast<int>(exitdir)];
  const int y1 = 2 * TileY(start_tile) + dg_dir_to_y_offs[static_cast<int>(exitdir)];
  const int x2 = 2 * TileX(destination_tile);
  const int y2 = 2 * TileY(destination_tile);
  const int dx = abs(x1 - x2);
  const int dy = abs(y1 - y2);
  const int dmin = std::min(dx, dy);
  const int dxy = abs(dx - dy);
 
  return dmin * YAPF_TILE_CORNER_LENGTH + (dxy - 1) * (YAPF_TILE_LENGTH / 2);
}
 
#endif /* YAPF_COMMON_HPP */