OpenTTD Source  20241108-master-g80f628063a
viewport_kdtree.h
1 /*
2  * This file is part of OpenTTD.
3  * 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.
4  * 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.
5  * 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/>.
6  */
7 
10 #ifndef VIEWPORT_KDTREE_H
11 #define VIEWPORT_KDTREE_H
12 
13 #include "core/kdtree.hpp"
14 #include "viewport_type.h"
15 #include "station_base.h"
16 #include "town_type.h"
17 #include "signs_base.h"
18 
20  enum ItemType : uint16_t {
21  VKI_STATION,
22  VKI_WAYPOINT,
23  VKI_TOWN,
24  VKI_SIGN,
25  };
26  ItemType type;
27  union {
28  StationID station;
29  TownID town;
30  SignID sign;
31  } id;
32  int32_t center;
33  int32_t top;
34 
35  bool operator== (const ViewportSignKdtreeItem &other) const
36  {
37  if (this->type != other.type) return false;
38  switch (this->type) {
39  case VKI_STATION:
40  case VKI_WAYPOINT:
41  return this->id.station == other.id.station;
42  case VKI_TOWN:
43  return this->id.town == other.id.town;
44  case VKI_SIGN:
45  return this->id.sign == other.id.sign;
46  default:
47  NOT_REACHED();
48  }
49  }
50 
51  bool operator< (const ViewportSignKdtreeItem &other) const
52  {
53  if (this->type != other.type) return this->type < other.type;
54  switch (this->type) {
55  case VKI_STATION:
56  case VKI_WAYPOINT:
57  return this->id.station < other.id.station;
58  case VKI_TOWN:
59  return this->id.town < other.id.town;
60  case VKI_SIGN:
61  return this->id.sign < other.id.sign;
62  default:
63  NOT_REACHED();
64  }
65  }
66 
67  static ViewportSignKdtreeItem MakeStation(StationID id);
68  static ViewportSignKdtreeItem MakeWaypoint(StationID id);
69  static ViewportSignKdtreeItem MakeTown(TownID id);
70  static ViewportSignKdtreeItem MakeSign(SignID id);
71 };
72 
73 inline int32_t Kdtree_ViewportSignXYFunc(const ViewportSignKdtreeItem &item, int dim)
74 {
75  return (dim == 0) ? item.center : item.top;
76 }
77 
78 typedef Kdtree<ViewportSignKdtreeItem, decltype(&Kdtree_ViewportSignXYFunc), int32_t, int32_t> ViewportSignKdtree;
79 extern ViewportSignKdtree _viewport_sign_kdtree;
80 
81 void RebuildViewportKdtree();
82 
83 #endif
K-dimensional tree, specialised for 2-dimensional space.
Definition: kdtree.hpp:35
K-d tree template specialised for 2-dimensional Manhattan geometry.
Base class for signs.
uint16_t SignID
The type of the IDs of signs.
Definition: signs_type.h:14
Base classes/functions for stations.
Types related to towns.
Types related to viewports.