476 lines
13 KiB
C++
476 lines
13 KiB
C++
/*************************************************************************/
|
|
/* a_star.cpp */
|
|
/*************************************************************************/
|
|
/* This file is part of: */
|
|
/* GODOT ENGINE */
|
|
/* https://godotengine.org */
|
|
/*************************************************************************/
|
|
/* Copyright (c) 2007-2019 Juan Linietsky, Ariel Manzur. */
|
|
/* Copyright (c) 2014-2019 Godot Engine contributors (cf. AUTHORS.md) */
|
|
/* */
|
|
/* Permission is hereby granted, free of charge, to any person obtaining */
|
|
/* a copy of this software and associated documentation files (the */
|
|
/* "Software"), to deal in the Software without restriction, including */
|
|
/* without limitation the rights to use, copy, modify, merge, publish, */
|
|
/* distribute, sublicense, and/or sell copies of the Software, and to */
|
|
/* permit persons to whom the Software is furnished to do so, subject to */
|
|
/* the following conditions: */
|
|
/* */
|
|
/* The above copyright notice and this permission notice shall be */
|
|
/* included in all copies or substantial portions of the Software. */
|
|
/* */
|
|
/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
|
|
/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
|
|
/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
|
|
/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
|
|
/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
|
|
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
|
|
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
|
|
/*************************************************************************/
|
|
|
|
#include "a_star.h"
|
|
|
|
#include "core/math/geometry.h"
|
|
#include "core/script_language.h"
|
|
#include "scene/scene_string_names.h"
|
|
|
|
int AStar::get_available_point_id() const {
|
|
|
|
if (points.empty()) {
|
|
return 1;
|
|
}
|
|
|
|
return points.back()->key() + 1;
|
|
}
|
|
|
|
void AStar::add_point(int p_id, const Vector3 &p_pos, real_t p_weight_scale) {
|
|
|
|
ERR_FAIL_COND(p_id < 0);
|
|
ERR_FAIL_COND(p_weight_scale < 1);
|
|
|
|
if (!points.has(p_id)) {
|
|
Point *pt = memnew(Point);
|
|
pt->id = p_id;
|
|
pt->pos = p_pos;
|
|
pt->weight_scale = p_weight_scale;
|
|
pt->prev_point = NULL;
|
|
pt->open_pass = 0;
|
|
pt->closed_pass = 0;
|
|
pt->enabled = true;
|
|
points[p_id] = pt;
|
|
} else {
|
|
points[p_id]->pos = p_pos;
|
|
points[p_id]->weight_scale = p_weight_scale;
|
|
}
|
|
}
|
|
|
|
Vector3 AStar::get_point_position(int p_id) const {
|
|
|
|
ERR_FAIL_COND_V(!points.has(p_id), Vector3());
|
|
|
|
return points[p_id]->pos;
|
|
}
|
|
|
|
void AStar::set_point_position(int p_id, const Vector3 &p_pos) {
|
|
|
|
ERR_FAIL_COND(!points.has(p_id));
|
|
|
|
points[p_id]->pos = p_pos;
|
|
}
|
|
|
|
real_t AStar::get_point_weight_scale(int p_id) const {
|
|
|
|
ERR_FAIL_COND_V(!points.has(p_id), 0);
|
|
|
|
return points[p_id]->weight_scale;
|
|
}
|
|
|
|
void AStar::set_point_weight_scale(int p_id, real_t p_weight_scale) {
|
|
|
|
ERR_FAIL_COND(!points.has(p_id));
|
|
ERR_FAIL_COND(p_weight_scale < 1);
|
|
|
|
points[p_id]->weight_scale = p_weight_scale;
|
|
}
|
|
|
|
void AStar::remove_point(int p_id) {
|
|
|
|
ERR_FAIL_COND(!points.has(p_id));
|
|
|
|
Point *p = points[p_id];
|
|
|
|
Map<int, Point *>::Element *PE = points.front();
|
|
while (PE) {
|
|
for (Set<Point *>::Element *E = PE->get()->neighbours.front(); E; E = E->next()) {
|
|
Segment s(p_id, E->get()->id);
|
|
segments.erase(s);
|
|
E->get()->neighbours.erase(p);
|
|
}
|
|
PE = PE->next();
|
|
}
|
|
|
|
memdelete(p);
|
|
points.erase(p_id);
|
|
}
|
|
|
|
void AStar::connect_points(int p_id, int p_with_id, bool bidirectional) {
|
|
|
|
ERR_FAIL_COND(!points.has(p_id));
|
|
ERR_FAIL_COND(!points.has(p_with_id));
|
|
ERR_FAIL_COND(p_id == p_with_id);
|
|
|
|
Point *a = points[p_id];
|
|
Point *b = points[p_with_id];
|
|
a->neighbours.insert(b);
|
|
|
|
if (bidirectional)
|
|
b->neighbours.insert(a);
|
|
|
|
Segment s(p_id, p_with_id);
|
|
if (s.from == p_id) {
|
|
s.from_point = a;
|
|
s.to_point = b;
|
|
} else {
|
|
s.from_point = b;
|
|
s.to_point = a;
|
|
}
|
|
|
|
segments.insert(s);
|
|
}
|
|
void AStar::disconnect_points(int p_id, int p_with_id) {
|
|
|
|
Segment s(p_id, p_with_id);
|
|
ERR_FAIL_COND(!segments.has(s));
|
|
|
|
segments.erase(s);
|
|
|
|
Point *a = points[p_id];
|
|
Point *b = points[p_with_id];
|
|
a->neighbours.erase(b);
|
|
b->neighbours.erase(a);
|
|
}
|
|
|
|
bool AStar::has_point(int p_id) const {
|
|
|
|
return points.has(p_id);
|
|
}
|
|
|
|
Array AStar::get_points() {
|
|
|
|
Array point_list;
|
|
|
|
for (const Map<int, Point *>::Element *E = points.front(); E; E = E->next()) {
|
|
point_list.push_back(E->key());
|
|
}
|
|
|
|
return point_list;
|
|
}
|
|
|
|
PoolVector<int> AStar::get_point_connections(int p_id) {
|
|
|
|
ERR_FAIL_COND_V(!points.has(p_id), PoolVector<int>());
|
|
|
|
PoolVector<int> point_list;
|
|
|
|
Point *p = points[p_id];
|
|
|
|
for (Set<Point *>::Element *E = p->neighbours.front(); E; E = E->next()) {
|
|
point_list.push_back(E->get()->id);
|
|
}
|
|
|
|
return point_list;
|
|
}
|
|
|
|
bool AStar::are_points_connected(int p_id, int p_with_id) const {
|
|
|
|
Segment s(p_id, p_with_id);
|
|
return segments.has(s);
|
|
}
|
|
|
|
void AStar::clear() {
|
|
|
|
for (const Map<int, Point *>::Element *E = points.front(); E; E = E->next()) {
|
|
|
|
memdelete(E->get());
|
|
}
|
|
segments.clear();
|
|
points.clear();
|
|
}
|
|
|
|
int AStar::get_closest_point(const Vector3 &p_point) const {
|
|
|
|
int closest_id = -1;
|
|
real_t closest_dist = 1e20;
|
|
|
|
for (const Map<int, Point *>::Element *E = points.front(); E; E = E->next()) {
|
|
|
|
real_t d = p_point.distance_squared_to(E->get()->pos);
|
|
if (closest_id < 0 || d < closest_dist) {
|
|
closest_dist = d;
|
|
closest_id = E->key();
|
|
}
|
|
}
|
|
|
|
return closest_id;
|
|
}
|
|
|
|
Vector3 AStar::get_closest_position_in_segment(const Vector3 &p_point) const {
|
|
|
|
real_t closest_dist = 1e20;
|
|
bool found = false;
|
|
Vector3 closest_point;
|
|
|
|
for (const Set<Segment>::Element *E = segments.front(); E; E = E->next()) {
|
|
|
|
Vector3 segment[2] = {
|
|
E->get().from_point->pos,
|
|
E->get().to_point->pos,
|
|
};
|
|
|
|
Vector3 p = Geometry::get_closest_point_to_segment(p_point, segment);
|
|
real_t d = p_point.distance_squared_to(p);
|
|
if (!found || d < closest_dist) {
|
|
|
|
closest_point = p;
|
|
closest_dist = d;
|
|
found = true;
|
|
}
|
|
}
|
|
|
|
return closest_point;
|
|
}
|
|
|
|
bool AStar::_solve(Point *begin_point, Point *end_point) {
|
|
|
|
pass++;
|
|
|
|
if (!end_point->enabled)
|
|
return false;
|
|
|
|
bool found_route = false;
|
|
|
|
Vector<Point *> open_list;
|
|
SortArray<Point *, SortPoints> sorter;
|
|
|
|
begin_point->g_score = 0;
|
|
begin_point->f_score = _estimate_cost(begin_point->id, end_point->id);
|
|
|
|
open_list.push_back(begin_point);
|
|
|
|
while (true) {
|
|
|
|
if (open_list.size() == 0) // No path found
|
|
break;
|
|
|
|
Point *p = open_list[0]; // The currently processed point
|
|
|
|
if (p == end_point) {
|
|
found_route = true;
|
|
break;
|
|
}
|
|
|
|
sorter.pop_heap(0, open_list.size(), open_list.ptrw()); // Remove the current point from the open list
|
|
open_list.remove(open_list.size() - 1);
|
|
p->closed_pass = pass; // Mark the point as closed
|
|
|
|
for (Set<Point *>::Element *E = p->neighbours.front(); E; E = E->next()) {
|
|
|
|
Point *e = E->get(); // The neighbour point
|
|
|
|
if (!e->enabled || e->closed_pass == pass)
|
|
continue;
|
|
|
|
real_t tentative_g_score = p->g_score + _compute_cost(p->id, e->id) * e->weight_scale;
|
|
|
|
bool new_point = false;
|
|
|
|
if (e->open_pass != pass) { // The point wasn't inside the open list
|
|
|
|
e->open_pass = pass;
|
|
open_list.push_back(e);
|
|
new_point = true;
|
|
} else if (tentative_g_score >= e->g_score) { // The new path is worse than the previous
|
|
|
|
continue;
|
|
}
|
|
|
|
e->prev_point = p;
|
|
e->g_score = tentative_g_score;
|
|
e->f_score = e->g_score + _estimate_cost(e->id, end_point->id);
|
|
|
|
if (new_point) // The position of the new points is already known
|
|
sorter.push_heap(0, open_list.size() - 1, 0, e, open_list.ptrw());
|
|
else
|
|
sorter.push_heap(0, open_list.find(e), 0, e, open_list.ptrw());
|
|
}
|
|
}
|
|
|
|
return found_route;
|
|
}
|
|
|
|
float AStar::_estimate_cost(int p_from_id, int p_to_id) {
|
|
|
|
if (get_script_instance() && get_script_instance()->has_method(SceneStringNames::get_singleton()->_estimate_cost))
|
|
return get_script_instance()->call(SceneStringNames::get_singleton()->_estimate_cost, p_from_id, p_to_id);
|
|
|
|
return points[p_from_id]->pos.distance_to(points[p_to_id]->pos);
|
|
}
|
|
|
|
float AStar::_compute_cost(int p_from_id, int p_to_id) {
|
|
|
|
if (get_script_instance() && get_script_instance()->has_method(SceneStringNames::get_singleton()->_compute_cost))
|
|
return get_script_instance()->call(SceneStringNames::get_singleton()->_compute_cost, p_from_id, p_to_id);
|
|
|
|
return points[p_from_id]->pos.distance_to(points[p_to_id]->pos);
|
|
}
|
|
|
|
PoolVector<Vector3> AStar::get_point_path(int p_from_id, int p_to_id) {
|
|
|
|
ERR_FAIL_COND_V(!points.has(p_from_id), PoolVector<Vector3>());
|
|
ERR_FAIL_COND_V(!points.has(p_to_id), PoolVector<Vector3>());
|
|
|
|
Point *a = points[p_from_id];
|
|
Point *b = points[p_to_id];
|
|
|
|
if (a == b) {
|
|
PoolVector<Vector3> ret;
|
|
ret.push_back(a->pos);
|
|
return ret;
|
|
}
|
|
|
|
Point *begin_point = a;
|
|
Point *end_point = b;
|
|
|
|
bool found_route = _solve(begin_point, end_point);
|
|
|
|
if (!found_route)
|
|
return PoolVector<Vector3>();
|
|
|
|
// Midpoints
|
|
Point *p = end_point;
|
|
int pc = 1; // Begin point
|
|
while (p != begin_point) {
|
|
pc++;
|
|
p = p->prev_point;
|
|
}
|
|
|
|
PoolVector<Vector3> path;
|
|
path.resize(pc);
|
|
|
|
{
|
|
PoolVector<Vector3>::Write w = path.write();
|
|
|
|
Point *p2 = end_point;
|
|
int idx = pc - 1;
|
|
while (p2 != begin_point) {
|
|
w[idx--] = p2->pos;
|
|
p2 = p2->prev_point;
|
|
}
|
|
|
|
w[0] = p2->pos; // Assign first
|
|
}
|
|
|
|
return path;
|
|
}
|
|
|
|
PoolVector<int> AStar::get_id_path(int p_from_id, int p_to_id) {
|
|
|
|
ERR_FAIL_COND_V(!points.has(p_from_id), PoolVector<int>());
|
|
ERR_FAIL_COND_V(!points.has(p_to_id), PoolVector<int>());
|
|
|
|
Point *a = points[p_from_id];
|
|
Point *b = points[p_to_id];
|
|
|
|
if (a == b) {
|
|
PoolVector<int> ret;
|
|
ret.push_back(a->id);
|
|
return ret;
|
|
}
|
|
|
|
Point *begin_point = a;
|
|
Point *end_point = b;
|
|
|
|
bool found_route = _solve(begin_point, end_point);
|
|
|
|
if (!found_route)
|
|
return PoolVector<int>();
|
|
|
|
// Midpoints
|
|
Point *p = end_point;
|
|
int pc = 1; // Begin point
|
|
while (p != begin_point) {
|
|
pc++;
|
|
p = p->prev_point;
|
|
}
|
|
|
|
PoolVector<int> path;
|
|
path.resize(pc);
|
|
|
|
{
|
|
PoolVector<int>::Write w = path.write();
|
|
|
|
p = end_point;
|
|
int idx = pc - 1;
|
|
while (p != begin_point) {
|
|
w[idx--] = p->id;
|
|
p = p->prev_point;
|
|
}
|
|
|
|
w[0] = p->id; // Assign first
|
|
}
|
|
|
|
return path;
|
|
}
|
|
|
|
void AStar::set_point_disabled(int p_id, bool p_disabled) {
|
|
points[p_id]->enabled = !p_disabled;
|
|
}
|
|
|
|
bool AStar::is_point_disabled(int p_id) const {
|
|
return !points[p_id]->enabled;
|
|
}
|
|
|
|
void AStar::_bind_methods() {
|
|
|
|
ClassDB::bind_method(D_METHOD("get_available_point_id"), &AStar::get_available_point_id);
|
|
ClassDB::bind_method(D_METHOD("add_point", "id", "position", "weight_scale"), &AStar::add_point, DEFVAL(1.0));
|
|
ClassDB::bind_method(D_METHOD("get_point_position", "id"), &AStar::get_point_position);
|
|
ClassDB::bind_method(D_METHOD("set_point_position", "id", "position"), &AStar::set_point_position);
|
|
ClassDB::bind_method(D_METHOD("get_point_weight_scale", "id"), &AStar::get_point_weight_scale);
|
|
ClassDB::bind_method(D_METHOD("set_point_weight_scale", "id", "weight_scale"), &AStar::set_point_weight_scale);
|
|
ClassDB::bind_method(D_METHOD("remove_point", "id"), &AStar::remove_point);
|
|
ClassDB::bind_method(D_METHOD("has_point", "id"), &AStar::has_point);
|
|
ClassDB::bind_method(D_METHOD("get_points"), &AStar::get_points);
|
|
|
|
ClassDB::bind_method(D_METHOD("set_point_disabled", "id", "disabled"), &AStar::set_point_disabled, DEFVAL(true));
|
|
ClassDB::bind_method(D_METHOD("is_point_disabled", "id"), &AStar::is_point_disabled);
|
|
|
|
ClassDB::bind_method(D_METHOD("get_point_connections", "id"), &AStar::get_point_connections);
|
|
|
|
ClassDB::bind_method(D_METHOD("connect_points", "id", "to_id", "bidirectional"), &AStar::connect_points, DEFVAL(true));
|
|
ClassDB::bind_method(D_METHOD("disconnect_points", "id", "to_id"), &AStar::disconnect_points);
|
|
ClassDB::bind_method(D_METHOD("are_points_connected", "id", "to_id"), &AStar::are_points_connected);
|
|
|
|
ClassDB::bind_method(D_METHOD("clear"), &AStar::clear);
|
|
|
|
ClassDB::bind_method(D_METHOD("get_closest_point", "to_position"), &AStar::get_closest_point);
|
|
ClassDB::bind_method(D_METHOD("get_closest_position_in_segment", "to_position"), &AStar::get_closest_position_in_segment);
|
|
|
|
ClassDB::bind_method(D_METHOD("get_point_path", "from_id", "to_id"), &AStar::get_point_path);
|
|
ClassDB::bind_method(D_METHOD("get_id_path", "from_id", "to_id"), &AStar::get_id_path);
|
|
|
|
BIND_VMETHOD(MethodInfo(Variant::REAL, "_estimate_cost", PropertyInfo(Variant::INT, "from_id"), PropertyInfo(Variant::INT, "to_id")));
|
|
BIND_VMETHOD(MethodInfo(Variant::REAL, "_compute_cost", PropertyInfo(Variant::INT, "from_id"), PropertyInfo(Variant::INT, "to_id")));
|
|
}
|
|
|
|
AStar::AStar() {
|
|
|
|
pass = 1;
|
|
}
|
|
|
|
AStar::~AStar() {
|
|
|
|
pass = 1;
|
|
clear();
|
|
}
|