forked from Mudlet/Mudlet
-
Notifications
You must be signed in to change notification settings - Fork 0
/
TAstar.h
120 lines (99 loc) · 3.78 KB
/
TAstar.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
#ifndef MUDLET_TASTAR_H
#define MUDLET_TASTAR_H
/***************************************************************************
* Copyright (C) 2010-2011 by Heiko Koehn - KoehnHeiko@googlemail.com *
* Copyright (C) 2014 by Ahmed Charles - acharles@outlook.com *
* Copyright (C) 2015, 2020, 2022 by Stephen Lyons *
* - slysven@virginmedia.com *
* *
* This program 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; either version 2 of the License, or *
* (at your option) any later version. *
* *
* This program 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 this program; if not, write to the *
* Free Software Foundation, Inc., *
* 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. *
***************************************************************************/
#include "TRoom.h"
#ifndef Q_MOC_RUN
#include "pre_guard.h"
#include <boost/graph/adjacency_list.hpp>
#include <boost/graph/astar_search.hpp>
#include <boost/graph/graphviz.hpp>
#include "post_guard.h"
#endif
#include "pre_guard.h"
#include <QDebug>
#include <QString>
#include "post_guard.h"
#include <math.h> // for sqrt
class TRoom;
using namespace boost;
// auxiliary types
struct location
{
int id; // Typically 4 bytes
TRoom* pR; // 4 or 8 bytes? - so may have reduced size from 20 to 8 or 12 plus padding...?
};
typedef float cost;
// Used to record edge details and to deduplicate parallel ones:
struct route
{
float cost; // Needed during establishing the best parallel edge
quint8 direction; // Use DIR_xxx values to code exit direction
QString specialExitName; // If direction is DIR_OTHER then this is needed
};
// euclidean distance heuristic
template <class Graph, class CostType, class LocMap>
class distance_heuristic : public boost::astar_heuristic<Graph, CostType>
{
public:
typedef typename boost::graph_traits<Graph>::vertex_descriptor Vertex;
distance_heuristic(LocMap l, Vertex goal)
: m_location(l)
, m_goal(goal)
{}
CostType operator()(Vertex u)
{
if (m_location[m_goal].pR->getArea() != m_location[u].pR->getArea()) {
return 1;
}
CostType dx = m_location[m_goal].pR->x() - m_location[u].pR->x();
CostType dy = m_location[m_goal].pR->y() - m_location[u].pR->y();
CostType dz = m_location[m_goal].pR->z() - m_location[u].pR->z();
return std::sqrt(dx * dx + dy * dy + dz * dz);
}
private:
LocMap m_location;
Vertex m_goal;
};
// exception for termination
struct found_goal
{
};
// visitor that terminates when we find the goal
template <class Vertex>
class astar_goal_visitor : public boost::default_astar_visitor
{
public:
explicit astar_goal_visitor(Vertex goal)
: m_goal(goal)
{}
template <class Graph>
void examine_vertex(Vertex u, Graph& g) {
Q_UNUSED(g)
if (u == m_goal) {
throw found_goal();
}
}
private:
Vertex m_goal;
};
#endif // MUDLET_TASTAR_H