CoCalc -- xr_body_tracker.cpp

GitHub Repository: godotengine/godot
Path: blob/master/servers/xr/xr_body_tracker.cpp
¹⁰²⁷⁷ views
1
/**************************************************************************/
2
/*  xr_body_tracker.cpp                                                   */
3
/**************************************************************************/
4
/*                         This file is part of:                          */
5
/*                             GODOT ENGINE                               */
6
/*                        https://godotengine.org                         */
7
/**************************************************************************/
8
/* Copyright (c) 2014-present Godot Engine contributors (see AUTHORS.md). */
9
/* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur.                  */
10
/*                                                                        */
11
/* Permission is hereby granted, free of charge, to any person obtaining  */
12
/* a copy of this software and associated documentation files (the        */
13
/* "Software"), to deal in the Software without restriction, including    */
14
/* without limitation the rights to use, copy, modify, merge, publish,    */
15
/* distribute, sublicense, and/or sell copies of the Software, and to     */
16
/* permit persons to whom the Software is furnished to do so, subject to  */
17
/* the following conditions:                                              */
18
/*                                                                        */
19
/* The above copyright notice and this permission notice shall be         */
20
/* included in all copies or substantial portions of the Software.        */
21
/*                                                                        */
22
/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,        */
23
/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF     */
24
/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. */
25
/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY   */
26
/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,   */
27
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE      */
28
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.                 */
29
/**************************************************************************/
30

31
#include "xr_body_tracker.h"
32

33
void XRBodyTracker::_bind_methods() {
34
	ClassDB::bind_method(D_METHOD("set_has_tracking_data", "has_data"), &XRBodyTracker::set_has_tracking_data);
35
	ClassDB::bind_method(D_METHOD("get_has_tracking_data"), &XRBodyTracker::get_has_tracking_data);
36

37
	ClassDB::bind_method(D_METHOD("set_body_flags", "flags"), &XRBodyTracker::set_body_flags);
38
	ClassDB::bind_method(D_METHOD("get_body_flags"), &XRBodyTracker::get_body_flags);
39

40
	ClassDB::bind_method(D_METHOD("set_joint_flags", "joint", "flags"), &XRBodyTracker::set_joint_flags);
41
	ClassDB::bind_method(D_METHOD("get_joint_flags", "joint"), &XRBodyTracker::get_joint_flags);
42

43
	ClassDB::bind_method(D_METHOD("set_joint_transform", "joint", "transform"), &XRBodyTracker::set_joint_transform);
44
	ClassDB::bind_method(D_METHOD("get_joint_transform", "joint"), &XRBodyTracker::get_joint_transform);
45

46
	ADD_PROPERTY(PropertyInfo(Variant::BOOL, "has_tracking_data", PROPERTY_HINT_NONE), "set_has_tracking_data", "get_has_tracking_data");
47
	ADD_PROPERTY(PropertyInfo(Variant::INT, "body_flags", PROPERTY_HINT_FLAGS, "Upper Body,Lower Body,Hands"), "set_body_flags", "get_body_flags");
48

49
	BIND_BITFIELD_FLAG(BODY_FLAG_UPPER_BODY_SUPPORTED);
50
	BIND_BITFIELD_FLAG(BODY_FLAG_LOWER_BODY_SUPPORTED);
51
	BIND_BITFIELD_FLAG(BODY_FLAG_HANDS_SUPPORTED);
52

53
	BIND_ENUM_CONSTANT(JOINT_ROOT);
54
	BIND_ENUM_CONSTANT(JOINT_HIPS);
55
	BIND_ENUM_CONSTANT(JOINT_SPINE);
56
	BIND_ENUM_CONSTANT(JOINT_CHEST);
57
	BIND_ENUM_CONSTANT(JOINT_UPPER_CHEST);
58
	BIND_ENUM_CONSTANT(JOINT_NECK);
59
	BIND_ENUM_CONSTANT(JOINT_HEAD);
60
	BIND_ENUM_CONSTANT(JOINT_HEAD_TIP);
61
	BIND_ENUM_CONSTANT(JOINT_LEFT_SHOULDER);
62
	BIND_ENUM_CONSTANT(JOINT_LEFT_UPPER_ARM);
63
	BIND_ENUM_CONSTANT(JOINT_LEFT_LOWER_ARM);
64
	BIND_ENUM_CONSTANT(JOINT_RIGHT_SHOULDER);
65
	BIND_ENUM_CONSTANT(JOINT_RIGHT_UPPER_ARM);
66
	BIND_ENUM_CONSTANT(JOINT_RIGHT_LOWER_ARM);
67
	BIND_ENUM_CONSTANT(JOINT_LEFT_UPPER_LEG);
68
	BIND_ENUM_CONSTANT(JOINT_LEFT_LOWER_LEG);
69
	BIND_ENUM_CONSTANT(JOINT_LEFT_FOOT);
70
	BIND_ENUM_CONSTANT(JOINT_LEFT_TOES);
71
	BIND_ENUM_CONSTANT(JOINT_RIGHT_UPPER_LEG);
72
	BIND_ENUM_CONSTANT(JOINT_RIGHT_LOWER_LEG);
73
	BIND_ENUM_CONSTANT(JOINT_RIGHT_FOOT);
74
	BIND_ENUM_CONSTANT(JOINT_RIGHT_TOES);
75
	BIND_ENUM_CONSTANT(JOINT_LEFT_HAND);
76
	BIND_ENUM_CONSTANT(JOINT_LEFT_PALM);
77
	BIND_ENUM_CONSTANT(JOINT_LEFT_WRIST);
78
	BIND_ENUM_CONSTANT(JOINT_LEFT_THUMB_METACARPAL);
79
	BIND_ENUM_CONSTANT(JOINT_LEFT_THUMB_PHALANX_PROXIMAL);
80
	BIND_ENUM_CONSTANT(JOINT_LEFT_THUMB_PHALANX_DISTAL);
81
	BIND_ENUM_CONSTANT(JOINT_LEFT_THUMB_TIP);
82
	BIND_ENUM_CONSTANT(JOINT_LEFT_INDEX_FINGER_METACARPAL);
83
	BIND_ENUM_CONSTANT(JOINT_LEFT_INDEX_FINGER_PHALANX_PROXIMAL);
84
	BIND_ENUM_CONSTANT(JOINT_LEFT_INDEX_FINGER_PHALANX_INTERMEDIATE);
85
	BIND_ENUM_CONSTANT(JOINT_LEFT_INDEX_FINGER_PHALANX_DISTAL);
86
	BIND_ENUM_CONSTANT(JOINT_LEFT_INDEX_FINGER_TIP);
87
	BIND_ENUM_CONSTANT(JOINT_LEFT_MIDDLE_FINGER_METACARPAL);
88
	BIND_ENUM_CONSTANT(JOINT_LEFT_MIDDLE_FINGER_PHALANX_PROXIMAL);
89
	BIND_ENUM_CONSTANT(JOINT_LEFT_MIDDLE_FINGER_PHALANX_INTERMEDIATE);
90
	BIND_ENUM_CONSTANT(JOINT_LEFT_MIDDLE_FINGER_PHALANX_DISTAL);
91
	BIND_ENUM_CONSTANT(JOINT_LEFT_MIDDLE_FINGER_TIP);
92
	BIND_ENUM_CONSTANT(JOINT_LEFT_RING_FINGER_METACARPAL);
93
	BIND_ENUM_CONSTANT(JOINT_LEFT_RING_FINGER_PHALANX_PROXIMAL);
94
	BIND_ENUM_CONSTANT(JOINT_LEFT_RING_FINGER_PHALANX_INTERMEDIATE);
95
	BIND_ENUM_CONSTANT(JOINT_LEFT_RING_FINGER_PHALANX_DISTAL);
96
	BIND_ENUM_CONSTANT(JOINT_LEFT_RING_FINGER_TIP);
97
	BIND_ENUM_CONSTANT(JOINT_LEFT_PINKY_FINGER_METACARPAL);
98
	BIND_ENUM_CONSTANT(JOINT_LEFT_PINKY_FINGER_PHALANX_PROXIMAL);
99
	BIND_ENUM_CONSTANT(JOINT_LEFT_PINKY_FINGER_PHALANX_INTERMEDIATE);
100
	BIND_ENUM_CONSTANT(JOINT_LEFT_PINKY_FINGER_PHALANX_DISTAL);
101
	BIND_ENUM_CONSTANT(JOINT_LEFT_PINKY_FINGER_TIP);
102
	BIND_ENUM_CONSTANT(JOINT_RIGHT_HAND);
103
	BIND_ENUM_CONSTANT(JOINT_RIGHT_PALM);
104
	BIND_ENUM_CONSTANT(JOINT_RIGHT_WRIST);
105
	BIND_ENUM_CONSTANT(JOINT_RIGHT_THUMB_METACARPAL);
106
	BIND_ENUM_CONSTANT(JOINT_RIGHT_THUMB_PHALANX_PROXIMAL);
107
	BIND_ENUM_CONSTANT(JOINT_RIGHT_THUMB_PHALANX_DISTAL);
108
	BIND_ENUM_CONSTANT(JOINT_RIGHT_THUMB_TIP);
109
	BIND_ENUM_CONSTANT(JOINT_RIGHT_INDEX_FINGER_METACARPAL);
110
	BIND_ENUM_CONSTANT(JOINT_RIGHT_INDEX_FINGER_PHALANX_PROXIMAL);
111
	BIND_ENUM_CONSTANT(JOINT_RIGHT_INDEX_FINGER_PHALANX_INTERMEDIATE);
112
	BIND_ENUM_CONSTANT(JOINT_RIGHT_INDEX_FINGER_PHALANX_DISTAL);
113
	BIND_ENUM_CONSTANT(JOINT_RIGHT_INDEX_FINGER_TIP);
114
	BIND_ENUM_CONSTANT(JOINT_RIGHT_MIDDLE_FINGER_METACARPAL);
115
	BIND_ENUM_CONSTANT(JOINT_RIGHT_MIDDLE_FINGER_PHALANX_PROXIMAL);
116
	BIND_ENUM_CONSTANT(JOINT_RIGHT_MIDDLE_FINGER_PHALANX_INTERMEDIATE);
117
	BIND_ENUM_CONSTANT(JOINT_RIGHT_MIDDLE_FINGER_PHALANX_DISTAL);
118
	BIND_ENUM_CONSTANT(JOINT_RIGHT_MIDDLE_FINGER_TIP);
119
	BIND_ENUM_CONSTANT(JOINT_RIGHT_RING_FINGER_METACARPAL);
120
	BIND_ENUM_CONSTANT(JOINT_RIGHT_RING_FINGER_PHALANX_PROXIMAL);
121
	BIND_ENUM_CONSTANT(JOINT_RIGHT_RING_FINGER_PHALANX_INTERMEDIATE);
122
	BIND_ENUM_CONSTANT(JOINT_RIGHT_RING_FINGER_PHALANX_DISTAL);
123
	BIND_ENUM_CONSTANT(JOINT_RIGHT_RING_FINGER_TIP);
124
	BIND_ENUM_CONSTANT(JOINT_RIGHT_PINKY_FINGER_METACARPAL);
125
	BIND_ENUM_CONSTANT(JOINT_RIGHT_PINKY_FINGER_PHALANX_PROXIMAL);
126
	BIND_ENUM_CONSTANT(JOINT_RIGHT_PINKY_FINGER_PHALANX_INTERMEDIATE);
127
	BIND_ENUM_CONSTANT(JOINT_RIGHT_PINKY_FINGER_PHALANX_DISTAL);
128
	BIND_ENUM_CONSTANT(JOINT_RIGHT_PINKY_FINGER_TIP);
129
	BIND_ENUM_CONSTANT(JOINT_LOWER_CHEST);
130
	BIND_ENUM_CONSTANT(JOINT_LEFT_SCAPULA);
131
	BIND_ENUM_CONSTANT(JOINT_LEFT_WRIST_TWIST);
132
	BIND_ENUM_CONSTANT(JOINT_RIGHT_SCAPULA);
133
	BIND_ENUM_CONSTANT(JOINT_RIGHT_WRIST_TWIST);
134
	BIND_ENUM_CONSTANT(JOINT_LEFT_FOOT_TWIST);
135
	BIND_ENUM_CONSTANT(JOINT_LEFT_HEEL);
136
	BIND_ENUM_CONSTANT(JOINT_LEFT_MIDDLE_FOOT);
137
	BIND_ENUM_CONSTANT(JOINT_RIGHT_FOOT_TWIST);
138
	BIND_ENUM_CONSTANT(JOINT_RIGHT_HEEL);
139
	BIND_ENUM_CONSTANT(JOINT_RIGHT_MIDDLE_FOOT);
140
	BIND_ENUM_CONSTANT(JOINT_MAX);
141

142
	BIND_BITFIELD_FLAG(JOINT_FLAG_ORIENTATION_VALID);
143
	BIND_BITFIELD_FLAG(JOINT_FLAG_ORIENTATION_TRACKED);
144
	BIND_BITFIELD_FLAG(JOINT_FLAG_POSITION_VALID);
145
	BIND_BITFIELD_FLAG(JOINT_FLAG_POSITION_TRACKED);
146
}
147

148
void XRBodyTracker::set_tracker_type(XRServer::TrackerType p_type) {
149
	ERR_FAIL_COND_MSG(p_type != XRServer::TRACKER_BODY, "XRBodyTracker must be of type TRACKER_BODY.");
150
}
151

152
void XRBodyTracker::set_tracker_hand(const XRPositionalTracker::TrackerHand p_hand) {
153
	ERR_FAIL_COND_MSG(p_hand != XRPositionalTracker::TRACKER_HAND_UNKNOWN, "XRBodyTracker cannot specify hand.");
154
}
155

156
void XRBodyTracker::set_has_tracking_data(bool p_has_tracking_data) {
157
	has_tracking_data = p_has_tracking_data;
158
}
159

160
bool XRBodyTracker::get_has_tracking_data() const {
161
	return has_tracking_data;
162
}
163

164
void XRBodyTracker::set_body_flags(BitField<BodyFlags> p_body_flags) {
165
	body_flags = p_body_flags;
166
}
167

168
BitField<XRBodyTracker::BodyFlags> XRBodyTracker::get_body_flags() const {
169
	return body_flags;
170
}
171

172
void XRBodyTracker::set_joint_flags(Joint p_joint, BitField<JointFlags> p_flags) {
173
	ERR_FAIL_INDEX(p_joint, JOINT_MAX);
174
	joint_flags[p_joint] = p_flags;
175
}
176

177
BitField<XRBodyTracker::JointFlags> XRBodyTracker::get_joint_flags(Joint p_joint) const {
178
	ERR_FAIL_INDEX_V(p_joint, JOINT_MAX, BitField<JointFlags>());
179
	return joint_flags[p_joint];
180
}
181

182
void XRBodyTracker::set_joint_transform(Joint p_joint, const Transform3D &p_transform) {
183
	ERR_FAIL_INDEX(p_joint, JOINT_MAX);
184
	joint_transforms[p_joint] = p_transform;
185
}
186

187
Transform3D XRBodyTracker::get_joint_transform(Joint p_joint) const {
188
	ERR_FAIL_INDEX_V(p_joint, JOINT_MAX, Transform3D());
189
	return joint_transforms[p_joint];
190
}
191

192
XRBodyTracker::XRBodyTracker() {
193
	type = XRServer::TRACKER_BODY;
194
}
195

196
Product

Resources

Company
