mirror of
https://github.com/OpenTTD/OpenTTD.git
synced 2025-03-09 15:41:15 +00:00
fc8685d618
Kdtree uses a function pointer and incorrectly calls it a functor. The function pointer needs to be passed on instantiaton. Instead, use an actual functor. This simplifies instantiation.
49 lines
1.9 KiB
C++
49 lines
1.9 KiB
C++
/*
|
|
* 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/>.
|
|
*/
|
|
|
|
/** @file station_kdtree.h Declarations for accessing the k-d tree of stations */
|
|
|
|
#ifndef STATION_KDTREE_H
|
|
#define STATION_KDTREE_H
|
|
|
|
#include "core/kdtree.hpp"
|
|
#include "core/math_func.hpp"
|
|
#include "station_base.h"
|
|
#include "map_func.h"
|
|
|
|
struct Kdtree_StationXYFunc {
|
|
inline uint16_t operator()(StationID stid, int dim)
|
|
{
|
|
return (dim == 0) ? TileX(BaseStation::Get(stid)->xy) : TileY(BaseStation::Get(stid)->xy);
|
|
}
|
|
};
|
|
|
|
using StationKdtree = Kdtree<StationID, Kdtree_StationXYFunc, uint16_t, int>;
|
|
extern StationKdtree _station_kdtree;
|
|
|
|
/**
|
|
* Call a function on all stations whose sign is within a radius of a center tile.
|
|
* @param center Central tile to search around.
|
|
* @param radius Distance in both X and Y to search within.
|
|
* @param func The function to call, must take a single parameter which is Station*.
|
|
*/
|
|
template <typename Func>
|
|
void ForAllStationsRadius(TileIndex center, uint radius, Func func)
|
|
{
|
|
uint16_t x1, y1, x2, y2;
|
|
x1 = (uint16_t)std::max<int>(0, TileX(center) - radius);
|
|
x2 = (uint16_t)std::min<int>(TileX(center) + radius + 1, Map::SizeX());
|
|
y1 = (uint16_t)std::max<int>(0, TileY(center) - radius);
|
|
y2 = (uint16_t)std::min<int>(TileY(center) + radius + 1, Map::SizeY());
|
|
|
|
_station_kdtree.FindContained(x1, y1, x2, y2, [&](StationID id) {
|
|
func(Station::Get(id));
|
|
});
|
|
}
|
|
|
|
#endif
|