summaryrefslogtreecommitdiff
path: root/core/math/a_star.h
blob: a0517b5941117365e92a0e213f50c7960b01ae36 (plain)
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
#ifndef ASTAR_H
#define ASTAR_H

#include "reference.h"
#include "self_list.h"

class AStar: public Reference {

	OBJ_TYPE(AStar,Reference)


	uint64_t pass;

	struct Point {

		SelfList<Point> list;

		int id;
		Vector3 pos;
		float weight_scale;
		uint64_t last_pass;

		Vector<Point*> neighbours;

		//used for pathfinding
		Point *prev_point;
		float distance;

		Point() : list(this) {}
	};

	Map<int,Point*> points;

	struct Segment {
		union {
			struct {
				int32_t from;
				int32_t to;
			};
			uint64_t key;
		};

		Point *from_point;
		Point *to_point;

		bool operator<(const Segment& p_s) const { return key<p_s.key; }
		Segment() { key=0; }
		Segment(int p_from,int p_to) {
			if (p_from > p_to) {
				SWAP(p_from,p_to);
			}

			from=p_from;
			to=p_to;
		}
	};


	Set<Segment> segments;

	bool _solve(Point *begin_point, Point *end_point);

protected:

	static void _bind_methods();
public:

	int get_available_point_id() const;

	void add_point(int p_id,const Vector3& p_pos,float p_weight_scale=1);
	Vector3 get_point_pos(int p_id) const;
	float get_point_weight_scale(int p_id) const;
	void remove_point(int p_id);

	void connect_points(int p_id,int p_with_id);
	void disconnect_points(int p_id,int p_with_id);
	bool are_points_connected(int p_id,int p_with_id) const;

	void clear();


	int get_closest_point(const Vector3& p_point) const;
	Vector3 get_closest_pos_in_segment(const Vector3& p_point) const;

	DVector<Vector3> get_point_path(int p_from_id, int p_to_id);
	DVector<int> get_id_path(int p_from_id, int p_to_id);

	AStar();
	~AStar();
};

#endif // ASTAR_H