grid_map.cpp 33 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148
  1. /*************************************************************************/
  2. /* grid_map.cpp */
  3. /*************************************************************************/
  4. /* This file is part of: */
  5. /* GODOT ENGINE */
  6. /* https://godotengine.org */
  7. /*************************************************************************/
  8. /* Copyright (c) 2007-2019 Juan Linietsky, Ariel Manzur. */
  9. /* Copyright (c) 2014-2019 Godot Engine contributors (cf. AUTHORS.md) */
  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. #include "grid_map.h"
  31. #include "core/io/marshalls.h"
  32. #include "core/message_queue.h"
  33. #include "scene/3d/light.h"
  34. #include "scene/resources/mesh_library.h"
  35. #include "scene/resources/surface_tool.h"
  36. #include "scene/scene_string_names.h"
  37. #include "servers/visual_server.h"
  38. bool GridMap::_set(const StringName &p_name, const Variant &p_value) {
  39. String name = p_name;
  40. if (name == "data") {
  41. Dictionary d = p_value;
  42. if (d.has("cells")) {
  43. PoolVector<int> cells = d["cells"];
  44. int amount = cells.size();
  45. PoolVector<int>::Read r = cells.read();
  46. ERR_FAIL_COND_V(amount % 3, false); // not even
  47. cell_map.clear();
  48. for (int i = 0; i < amount / 3; i++) {
  49. IndexKey ik;
  50. ik.key = decode_uint64((const uint8_t *)&r[i * 3]);
  51. Cell cell;
  52. cell.cell = decode_uint32((const uint8_t *)&r[i * 3 + 2]);
  53. cell_map[ik] = cell;
  54. }
  55. }
  56. _recreate_octant_data();
  57. } else if (name == "baked_meshes") {
  58. clear_baked_meshes();
  59. Array meshes = p_value;
  60. for (int i = 0; i < meshes.size(); i++) {
  61. BakedMesh bm;
  62. bm.mesh = meshes[i];
  63. ERR_CONTINUE(!bm.mesh.is_valid());
  64. bm.instance = VS::get_singleton()->instance_create();
  65. VS::get_singleton()->get_singleton()->instance_set_base(bm.instance, bm.mesh->get_rid());
  66. VS::get_singleton()->instance_attach_object_instance_id(bm.instance, get_instance_id());
  67. if (is_inside_tree()) {
  68. VS::get_singleton()->instance_set_scenario(bm.instance, get_world()->get_scenario());
  69. VS::get_singleton()->instance_set_transform(bm.instance, get_global_transform());
  70. }
  71. baked_meshes.push_back(bm);
  72. }
  73. _recreate_octant_data();
  74. } else {
  75. return false;
  76. }
  77. return true;
  78. }
  79. bool GridMap::_get(const StringName &p_name, Variant &r_ret) const {
  80. String name = p_name;
  81. if (name == "data") {
  82. Dictionary d;
  83. PoolVector<int> cells;
  84. cells.resize(cell_map.size() * 3);
  85. {
  86. PoolVector<int>::Write w = cells.write();
  87. int i = 0;
  88. for (Map<IndexKey, Cell>::Element *E = cell_map.front(); E; E = E->next(), i++) {
  89. encode_uint64(E->key().key, (uint8_t *)&w[i * 3]);
  90. encode_uint32(E->get().cell, (uint8_t *)&w[i * 3 + 2]);
  91. }
  92. }
  93. d["cells"] = cells;
  94. r_ret = d;
  95. } else if (name == "baked_meshes") {
  96. Array ret;
  97. ret.resize(baked_meshes.size());
  98. for (int i = 0; i < baked_meshes.size(); i++) {
  99. ret.push_back(baked_meshes[i].mesh);
  100. }
  101. r_ret = ret;
  102. } else
  103. return false;
  104. return true;
  105. }
  106. void GridMap::_get_property_list(List<PropertyInfo> *p_list) const {
  107. if (baked_meshes.size()) {
  108. p_list->push_back(PropertyInfo(Variant::ARRAY, "baked_meshes", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_STORAGE));
  109. }
  110. p_list->push_back(PropertyInfo(Variant::DICTIONARY, "data", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_STORAGE));
  111. }
  112. void GridMap::set_collision_layer(uint32_t p_layer) {
  113. collision_layer = p_layer;
  114. _reset_physic_bodies_collision_filters();
  115. }
  116. uint32_t GridMap::get_collision_layer() const {
  117. return collision_layer;
  118. }
  119. void GridMap::set_collision_mask(uint32_t p_mask) {
  120. collision_mask = p_mask;
  121. _reset_physic_bodies_collision_filters();
  122. }
  123. uint32_t GridMap::get_collision_mask() const {
  124. return collision_mask;
  125. }
  126. void GridMap::set_collision_mask_bit(int p_bit, bool p_value) {
  127. uint32_t mask = get_collision_mask();
  128. if (p_value)
  129. mask |= 1 << p_bit;
  130. else
  131. mask &= ~(1 << p_bit);
  132. set_collision_mask(mask);
  133. }
  134. bool GridMap::get_collision_mask_bit(int p_bit) const {
  135. return get_collision_mask() & (1 << p_bit);
  136. }
  137. void GridMap::set_collision_layer_bit(int p_bit, bool p_value) {
  138. uint32_t mask = get_collision_layer();
  139. if (p_value)
  140. mask |= 1 << p_bit;
  141. else
  142. mask &= ~(1 << p_bit);
  143. set_collision_layer(mask);
  144. }
  145. bool GridMap::get_collision_layer_bit(int p_bit) const {
  146. return get_collision_layer() & (1 << p_bit);
  147. }
  148. #ifndef DISABLE_DEPRECATED
  149. void GridMap::set_theme(const Ref<MeshLibrary> &p_theme) {
  150. ERR_EXPLAIN("GridMap.theme/set_theme() is deprecated and will be removed in a future version. Use GridMap.mesh_library/set_mesh_library() instead.");
  151. WARN_DEPRECATED;
  152. set_mesh_library(p_theme);
  153. }
  154. Ref<MeshLibrary> GridMap::get_theme() const {
  155. ERR_EXPLAIN("GridMap.theme/get_theme() is deprecated and will be removed in a future version. Use GridMap.mesh_library/get_mesh_library() instead.");
  156. WARN_DEPRECATED;
  157. return get_mesh_library();
  158. }
  159. #endif // DISABLE_DEPRECATED
  160. void GridMap::set_mesh_library(const Ref<MeshLibrary> &p_mesh_library) {
  161. if (!mesh_library.is_null())
  162. mesh_library->unregister_owner(this);
  163. mesh_library = p_mesh_library;
  164. if (!mesh_library.is_null())
  165. mesh_library->register_owner(this);
  166. _recreate_octant_data();
  167. _change_notify("mesh_library");
  168. }
  169. Ref<MeshLibrary> GridMap::get_mesh_library() const {
  170. return mesh_library;
  171. }
  172. void GridMap::set_cell_size(const Vector3 &p_size) {
  173. ERR_FAIL_COND(p_size.x < 0.001 || p_size.y < 0.001 || p_size.z < 0.001);
  174. cell_size = p_size;
  175. _recreate_octant_data();
  176. }
  177. Vector3 GridMap::get_cell_size() const {
  178. return cell_size;
  179. }
  180. void GridMap::set_octant_size(int p_size) {
  181. ERR_FAIL_COND(p_size == 0);
  182. octant_size = p_size;
  183. _recreate_octant_data();
  184. }
  185. int GridMap::get_octant_size() const {
  186. return octant_size;
  187. }
  188. void GridMap::set_center_x(bool p_enable) {
  189. center_x = p_enable;
  190. _recreate_octant_data();
  191. }
  192. bool GridMap::get_center_x() const {
  193. return center_x;
  194. }
  195. void GridMap::set_center_y(bool p_enable) {
  196. center_y = p_enable;
  197. _recreate_octant_data();
  198. }
  199. bool GridMap::get_center_y() const {
  200. return center_y;
  201. }
  202. void GridMap::set_center_z(bool p_enable) {
  203. center_z = p_enable;
  204. _recreate_octant_data();
  205. }
  206. bool GridMap::get_center_z() const {
  207. return center_z;
  208. }
  209. void GridMap::set_cell_item(int p_x, int p_y, int p_z, int p_item, int p_rot) {
  210. if (baked_meshes.size() && !recreating_octants) {
  211. //if you set a cell item, baked meshes go good bye
  212. clear_baked_meshes();
  213. _recreate_octant_data();
  214. }
  215. ERR_FAIL_INDEX(ABS(p_x), 1 << 20);
  216. ERR_FAIL_INDEX(ABS(p_y), 1 << 20);
  217. ERR_FAIL_INDEX(ABS(p_z), 1 << 20);
  218. IndexKey key;
  219. key.x = p_x;
  220. key.y = p_y;
  221. key.z = p_z;
  222. OctantKey ok;
  223. ok.x = p_x / octant_size;
  224. ok.y = p_y / octant_size;
  225. ok.z = p_z / octant_size;
  226. if (p_item < 0) {
  227. //erase
  228. if (cell_map.has(key)) {
  229. OctantKey octantkey = ok;
  230. ERR_FAIL_COND(!octant_map.has(octantkey));
  231. Octant &g = *octant_map[octantkey];
  232. g.cells.erase(key);
  233. g.dirty = true;
  234. cell_map.erase(key);
  235. _queue_octants_dirty();
  236. }
  237. return;
  238. }
  239. OctantKey octantkey = ok;
  240. if (!octant_map.has(octantkey)) {
  241. //create octant because it does not exist
  242. Octant *g = memnew(Octant);
  243. g->dirty = true;
  244. g->static_body = PhysicsServer::get_singleton()->body_create(PhysicsServer::BODY_MODE_STATIC);
  245. PhysicsServer::get_singleton()->body_attach_object_instance_id(g->static_body, get_instance_id());
  246. PhysicsServer::get_singleton()->body_set_collision_layer(g->static_body, collision_layer);
  247. PhysicsServer::get_singleton()->body_set_collision_mask(g->static_body, collision_mask);
  248. SceneTree *st = SceneTree::get_singleton();
  249. if (st && st->is_debugging_collisions_hint()) {
  250. g->collision_debug = VisualServer::get_singleton()->mesh_create();
  251. g->collision_debug_instance = VisualServer::get_singleton()->instance_create();
  252. VisualServer::get_singleton()->instance_set_base(g->collision_debug_instance, g->collision_debug);
  253. }
  254. octant_map[octantkey] = g;
  255. if (is_inside_world()) {
  256. _octant_enter_world(octantkey);
  257. _octant_transform(octantkey);
  258. }
  259. }
  260. Octant &g = *octant_map[octantkey];
  261. g.cells.insert(key);
  262. g.dirty = true;
  263. _queue_octants_dirty();
  264. Cell c;
  265. c.item = p_item;
  266. c.rot = p_rot;
  267. cell_map[key] = c;
  268. }
  269. int GridMap::get_cell_item(int p_x, int p_y, int p_z) const {
  270. ERR_FAIL_INDEX_V(ABS(p_x), 1 << 20, INVALID_CELL_ITEM);
  271. ERR_FAIL_INDEX_V(ABS(p_y), 1 << 20, INVALID_CELL_ITEM);
  272. ERR_FAIL_INDEX_V(ABS(p_z), 1 << 20, INVALID_CELL_ITEM);
  273. IndexKey key;
  274. key.x = p_x;
  275. key.y = p_y;
  276. key.z = p_z;
  277. if (!cell_map.has(key))
  278. return INVALID_CELL_ITEM;
  279. return cell_map[key].item;
  280. }
  281. int GridMap::get_cell_item_orientation(int p_x, int p_y, int p_z) const {
  282. ERR_FAIL_INDEX_V(ABS(p_x), 1 << 20, -1);
  283. ERR_FAIL_INDEX_V(ABS(p_y), 1 << 20, -1);
  284. ERR_FAIL_INDEX_V(ABS(p_z), 1 << 20, -1);
  285. IndexKey key;
  286. key.x = p_x;
  287. key.y = p_y;
  288. key.z = p_z;
  289. if (!cell_map.has(key))
  290. return -1;
  291. return cell_map[key].rot;
  292. }
  293. Vector3 GridMap::world_to_map(const Vector3 &p_world_pos) const {
  294. Vector3 map_pos = p_world_pos / cell_size;
  295. map_pos.x = floor(map_pos.x);
  296. map_pos.y = floor(map_pos.y);
  297. map_pos.z = floor(map_pos.z);
  298. return map_pos;
  299. }
  300. Vector3 GridMap::map_to_world(int p_x, int p_y, int p_z) const {
  301. Vector3 offset = _get_offset();
  302. Vector3 world_pos(
  303. p_x * cell_size.x + offset.x,
  304. p_y * cell_size.y + offset.y,
  305. p_z * cell_size.z + offset.z);
  306. return world_pos;
  307. }
  308. void GridMap::_octant_transform(const OctantKey &p_key) {
  309. ERR_FAIL_COND(!octant_map.has(p_key));
  310. Octant &g = *octant_map[p_key];
  311. PhysicsServer::get_singleton()->body_set_state(g.static_body, PhysicsServer::BODY_STATE_TRANSFORM, get_global_transform());
  312. if (g.collision_debug_instance.is_valid()) {
  313. VS::get_singleton()->instance_set_transform(g.collision_debug_instance, get_global_transform());
  314. }
  315. for (int i = 0; i < g.multimesh_instances.size(); i++) {
  316. VS::get_singleton()->instance_set_transform(g.multimesh_instances[i].instance, get_global_transform());
  317. }
  318. }
  319. bool GridMap::_octant_update(const OctantKey &p_key) {
  320. ERR_FAIL_COND_V(!octant_map.has(p_key), false);
  321. Octant &g = *octant_map[p_key];
  322. if (!g.dirty)
  323. return false;
  324. //erase body shapes
  325. PhysicsServer::get_singleton()->body_clear_shapes(g.static_body);
  326. //erase body shapes debug
  327. if (g.collision_debug.is_valid()) {
  328. VS::get_singleton()->mesh_clear(g.collision_debug);
  329. }
  330. //erase navigation
  331. if (navigation) {
  332. for (Map<IndexKey, Octant::NavMesh>::Element *E = g.navmesh_ids.front(); E; E = E->next()) {
  333. navigation->navmesh_remove(E->get().id);
  334. }
  335. g.navmesh_ids.clear();
  336. }
  337. //erase multimeshes
  338. for (int i = 0; i < g.multimesh_instances.size(); i++) {
  339. VS::get_singleton()->free(g.multimesh_instances[i].instance);
  340. VS::get_singleton()->free(g.multimesh_instances[i].multimesh);
  341. }
  342. g.multimesh_instances.clear();
  343. if (g.cells.size() == 0) {
  344. //octant no longer needed
  345. _octant_clean_up(p_key);
  346. return true;
  347. }
  348. PoolVector<Vector3> col_debug;
  349. /*
  350. * foreach item in this octant,
  351. * set item's multimesh's instance count to number of cells which have this item
  352. * and set said multimesh bounding box to one containing all cells which have this item
  353. */
  354. Map<int, List<Pair<Transform, IndexKey> > > multimesh_items;
  355. for (Set<IndexKey>::Element *E = g.cells.front(); E; E = E->next()) {
  356. ERR_CONTINUE(!cell_map.has(E->get()));
  357. const Cell &c = cell_map[E->get()];
  358. if (!mesh_library.is_valid() || !mesh_library->has_item(c.item))
  359. continue;
  360. Vector3 cellpos = Vector3(E->get().x, E->get().y, E->get().z);
  361. Vector3 ofs = _get_offset();
  362. Transform xform;
  363. if (clip && ((clip_above && cellpos[clip_axis] > clip_floor) || (!clip_above && cellpos[clip_axis] < clip_floor))) {
  364. } else {
  365. }
  366. xform.basis.set_orthogonal_index(c.rot);
  367. xform.set_origin(cellpos * cell_size + ofs);
  368. xform.basis.scale(Vector3(cell_scale, cell_scale, cell_scale));
  369. if (baked_meshes.size() == 0) {
  370. if (mesh_library->get_item_mesh(c.item).is_valid()) {
  371. if (!multimesh_items.has(c.item)) {
  372. multimesh_items[c.item] = List<Pair<Transform, IndexKey> >();
  373. }
  374. Pair<Transform, IndexKey> p;
  375. p.first = xform;
  376. p.second = E->get();
  377. multimesh_items[c.item].push_back(p);
  378. }
  379. }
  380. Vector<MeshLibrary::ShapeData> shapes = mesh_library->get_item_shapes(c.item);
  381. // add the item's shape at given xform to octant's static_body
  382. for (int i = 0; i < shapes.size(); i++) {
  383. // add the item's shape
  384. if (!shapes[i].shape.is_valid())
  385. continue;
  386. PhysicsServer::get_singleton()->body_add_shape(g.static_body, shapes[i].shape->get_rid(), xform * shapes[i].local_transform);
  387. if (g.collision_debug.is_valid()) {
  388. shapes.write[i].shape->add_vertices_to_array(col_debug, xform * shapes[i].local_transform);
  389. }
  390. }
  391. // add the item's navmesh at given xform to GridMap's Navigation ancestor
  392. Ref<NavigationMesh> navmesh = mesh_library->get_item_navmesh(c.item);
  393. if (navmesh.is_valid()) {
  394. Octant::NavMesh nm;
  395. nm.xform = xform * mesh_library->get_item_navmesh_transform(c.item);
  396. if (navigation) {
  397. nm.id = navigation->navmesh_add(navmesh, xform, this);
  398. } else {
  399. nm.id = -1;
  400. }
  401. g.navmesh_ids[E->get()] = nm;
  402. }
  403. }
  404. //update multimeshes, only if not baked
  405. if (baked_meshes.size() == 0) {
  406. for (Map<int, List<Pair<Transform, IndexKey> > >::Element *E = multimesh_items.front(); E; E = E->next()) {
  407. Octant::MultimeshInstance mmi;
  408. RID mm = VS::get_singleton()->multimesh_create();
  409. VS::get_singleton()->multimesh_allocate(mm, E->get().size(), VS::MULTIMESH_TRANSFORM_3D, VS::MULTIMESH_COLOR_NONE);
  410. VS::get_singleton()->multimesh_set_mesh(mm, mesh_library->get_item_mesh(E->key())->get_rid());
  411. int idx = 0;
  412. for (List<Pair<Transform, IndexKey> >::Element *F = E->get().front(); F; F = F->next()) {
  413. VS::get_singleton()->multimesh_instance_set_transform(mm, idx, F->get().first);
  414. #ifdef TOOLS_ENABLED
  415. Octant::MultimeshInstance::Item it;
  416. it.index = idx;
  417. it.transform = F->get().first;
  418. it.key = F->get().second;
  419. mmi.items.push_back(it);
  420. #endif
  421. idx++;
  422. }
  423. RID instance = VS::get_singleton()->instance_create();
  424. VS::get_singleton()->instance_set_base(instance, mm);
  425. if (is_inside_tree()) {
  426. VS::get_singleton()->instance_set_scenario(instance, get_world()->get_scenario());
  427. VS::get_singleton()->instance_set_transform(instance, get_global_transform());
  428. }
  429. mmi.multimesh = mm;
  430. mmi.instance = instance;
  431. g.multimesh_instances.push_back(mmi);
  432. }
  433. }
  434. if (col_debug.size()) {
  435. Array arr;
  436. arr.resize(VS::ARRAY_MAX);
  437. arr[VS::ARRAY_VERTEX] = col_debug;
  438. VS::get_singleton()->mesh_add_surface_from_arrays(g.collision_debug, VS::PRIMITIVE_LINES, arr);
  439. SceneTree *st = SceneTree::get_singleton();
  440. if (st) {
  441. VS::get_singleton()->mesh_surface_set_material(g.collision_debug, 0, st->get_debug_collision_material()->get_rid());
  442. }
  443. }
  444. g.dirty = false;
  445. return false;
  446. }
  447. void GridMap::_reset_physic_bodies_collision_filters() {
  448. for (Map<OctantKey, Octant *>::Element *E = octant_map.front(); E; E = E->next()) {
  449. PhysicsServer::get_singleton()->body_set_collision_layer(E->get()->static_body, collision_layer);
  450. PhysicsServer::get_singleton()->body_set_collision_mask(E->get()->static_body, collision_mask);
  451. }
  452. }
  453. void GridMap::_octant_enter_world(const OctantKey &p_key) {
  454. ERR_FAIL_COND(!octant_map.has(p_key));
  455. Octant &g = *octant_map[p_key];
  456. PhysicsServer::get_singleton()->body_set_state(g.static_body, PhysicsServer::BODY_STATE_TRANSFORM, get_global_transform());
  457. PhysicsServer::get_singleton()->body_set_space(g.static_body, get_world()->get_space());
  458. if (g.collision_debug_instance.is_valid()) {
  459. VS::get_singleton()->instance_set_scenario(g.collision_debug_instance, get_world()->get_scenario());
  460. VS::get_singleton()->instance_set_transform(g.collision_debug_instance, get_global_transform());
  461. }
  462. for (int i = 0; i < g.multimesh_instances.size(); i++) {
  463. VS::get_singleton()->instance_set_scenario(g.multimesh_instances[i].instance, get_world()->get_scenario());
  464. VS::get_singleton()->instance_set_transform(g.multimesh_instances[i].instance, get_global_transform());
  465. }
  466. if (navigation && mesh_library.is_valid()) {
  467. for (Map<IndexKey, Octant::NavMesh>::Element *F = g.navmesh_ids.front(); F; F = F->next()) {
  468. if (cell_map.has(F->key()) && F->get().id < 0) {
  469. Ref<NavigationMesh> nm = mesh_library->get_item_navmesh(cell_map[F->key()].item);
  470. if (nm.is_valid()) {
  471. F->get().id = navigation->navmesh_add(nm, F->get().xform, this);
  472. }
  473. }
  474. }
  475. }
  476. }
  477. void GridMap::_octant_exit_world(const OctantKey &p_key) {
  478. ERR_FAIL_COND(!octant_map.has(p_key));
  479. Octant &g = *octant_map[p_key];
  480. PhysicsServer::get_singleton()->body_set_state(g.static_body, PhysicsServer::BODY_STATE_TRANSFORM, get_global_transform());
  481. PhysicsServer::get_singleton()->body_set_space(g.static_body, RID());
  482. if (g.collision_debug_instance.is_valid()) {
  483. VS::get_singleton()->instance_set_scenario(g.collision_debug_instance, RID());
  484. }
  485. for (int i = 0; i < g.multimesh_instances.size(); i++) {
  486. VS::get_singleton()->instance_set_scenario(g.multimesh_instances[i].instance, RID());
  487. }
  488. if (navigation) {
  489. for (Map<IndexKey, Octant::NavMesh>::Element *F = g.navmesh_ids.front(); F; F = F->next()) {
  490. if (F->get().id >= 0) {
  491. navigation->navmesh_remove(F->get().id);
  492. F->get().id = -1;
  493. }
  494. }
  495. }
  496. }
  497. void GridMap::_octant_clean_up(const OctantKey &p_key) {
  498. ERR_FAIL_COND(!octant_map.has(p_key));
  499. Octant &g = *octant_map[p_key];
  500. if (g.collision_debug.is_valid())
  501. VS::get_singleton()->free(g.collision_debug);
  502. if (g.collision_debug_instance.is_valid())
  503. VS::get_singleton()->free(g.collision_debug_instance);
  504. PhysicsServer::get_singleton()->free(g.static_body);
  505. //erase navigation
  506. if (navigation) {
  507. for (Map<IndexKey, Octant::NavMesh>::Element *E = g.navmesh_ids.front(); E; E = E->next()) {
  508. navigation->navmesh_remove(E->get().id);
  509. }
  510. g.navmesh_ids.clear();
  511. }
  512. //erase multimeshes
  513. for (int i = 0; i < g.multimesh_instances.size(); i++) {
  514. VS::get_singleton()->free(g.multimesh_instances[i].instance);
  515. VS::get_singleton()->free(g.multimesh_instances[i].multimesh);
  516. }
  517. g.multimesh_instances.clear();
  518. }
  519. void GridMap::_notification(int p_what) {
  520. switch (p_what) {
  521. case NOTIFICATION_ENTER_WORLD: {
  522. Spatial *c = this;
  523. while (c) {
  524. navigation = Object::cast_to<Navigation>(c);
  525. if (navigation) {
  526. break;
  527. }
  528. c = Object::cast_to<Spatial>(c->get_parent());
  529. }
  530. last_transform = get_global_transform();
  531. for (Map<OctantKey, Octant *>::Element *E = octant_map.front(); E; E = E->next()) {
  532. _octant_enter_world(E->key());
  533. }
  534. for (int i = 0; i < baked_meshes.size(); i++) {
  535. VS::get_singleton()->instance_set_scenario(baked_meshes[i].instance, get_world()->get_scenario());
  536. VS::get_singleton()->instance_set_transform(baked_meshes[i].instance, get_global_transform());
  537. }
  538. } break;
  539. case NOTIFICATION_TRANSFORM_CHANGED: {
  540. Transform new_xform = get_global_transform();
  541. if (new_xform == last_transform)
  542. break;
  543. //update run
  544. for (Map<OctantKey, Octant *>::Element *E = octant_map.front(); E; E = E->next()) {
  545. _octant_transform(E->key());
  546. }
  547. last_transform = new_xform;
  548. for (int i = 0; i < baked_meshes.size(); i++) {
  549. VS::get_singleton()->instance_set_transform(baked_meshes[i].instance, get_global_transform());
  550. }
  551. } break;
  552. case NOTIFICATION_EXIT_WORLD: {
  553. for (Map<OctantKey, Octant *>::Element *E = octant_map.front(); E; E = E->next()) {
  554. _octant_exit_world(E->key());
  555. }
  556. navigation = NULL;
  557. //_queue_octants_dirty(MAP_DIRTY_INSTANCES|MAP_DIRTY_TRANSFORMS);
  558. //_update_octants_callback();
  559. //_update_area_instances();
  560. for (int i = 0; i < baked_meshes.size(); i++) {
  561. VS::get_singleton()->instance_set_scenario(baked_meshes[i].instance, RID());
  562. }
  563. } break;
  564. case NOTIFICATION_VISIBILITY_CHANGED: {
  565. _update_visibility();
  566. } break;
  567. }
  568. }
  569. void GridMap::_update_visibility() {
  570. if (!is_inside_tree())
  571. return;
  572. _change_notify("visible");
  573. for (Map<OctantKey, Octant *>::Element *e = octant_map.front(); e; e = e->next()) {
  574. Octant *octant = e->value();
  575. for (int i = 0; i < octant->multimesh_instances.size(); i++) {
  576. const Octant::MultimeshInstance &mi = octant->multimesh_instances[i];
  577. VS::get_singleton()->instance_set_visible(mi.instance, is_visible());
  578. }
  579. }
  580. }
  581. void GridMap::_queue_octants_dirty() {
  582. if (awaiting_update)
  583. return;
  584. MessageQueue::get_singleton()->push_call(this, "_update_octants_callback");
  585. awaiting_update = true;
  586. }
  587. void GridMap::_recreate_octant_data() {
  588. recreating_octants = true;
  589. Map<IndexKey, Cell> cell_copy = cell_map;
  590. _clear_internal();
  591. for (Map<IndexKey, Cell>::Element *E = cell_copy.front(); E; E = E->next()) {
  592. set_cell_item(E->key().x, E->key().y, E->key().z, E->get().item, E->get().rot);
  593. }
  594. recreating_octants = false;
  595. }
  596. void GridMap::_clear_internal() {
  597. for (Map<OctantKey, Octant *>::Element *E = octant_map.front(); E; E = E->next()) {
  598. if (is_inside_world())
  599. _octant_exit_world(E->key());
  600. _octant_clean_up(E->key());
  601. memdelete(E->get());
  602. }
  603. octant_map.clear();
  604. cell_map.clear();
  605. }
  606. void GridMap::clear() {
  607. _clear_internal();
  608. clear_baked_meshes();
  609. }
  610. void GridMap::resource_changed(const RES &p_res) {
  611. _recreate_octant_data();
  612. }
  613. void GridMap::_update_octants_callback() {
  614. if (!awaiting_update)
  615. return;
  616. List<OctantKey> to_delete;
  617. for (Map<OctantKey, Octant *>::Element *E = octant_map.front(); E; E = E->next()) {
  618. if (_octant_update(E->key())) {
  619. to_delete.push_back(E->key());
  620. }
  621. }
  622. while (to_delete.front()) {
  623. octant_map.erase(to_delete.front()->get());
  624. to_delete.pop_back();
  625. }
  626. _update_visibility();
  627. awaiting_update = false;
  628. }
  629. void GridMap::_bind_methods() {
  630. ClassDB::bind_method(D_METHOD("set_collision_layer", "layer"), &GridMap::set_collision_layer);
  631. ClassDB::bind_method(D_METHOD("get_collision_layer"), &GridMap::get_collision_layer);
  632. ClassDB::bind_method(D_METHOD("set_collision_mask", "mask"), &GridMap::set_collision_mask);
  633. ClassDB::bind_method(D_METHOD("get_collision_mask"), &GridMap::get_collision_mask);
  634. ClassDB::bind_method(D_METHOD("set_collision_mask_bit", "bit", "value"), &GridMap::set_collision_mask_bit);
  635. ClassDB::bind_method(D_METHOD("get_collision_mask_bit", "bit"), &GridMap::get_collision_mask_bit);
  636. ClassDB::bind_method(D_METHOD("set_collision_layer_bit", "bit", "value"), &GridMap::set_collision_layer_bit);
  637. ClassDB::bind_method(D_METHOD("get_collision_layer_bit", "bit"), &GridMap::get_collision_layer_bit);
  638. #ifndef DISABLE_DEPRECATED
  639. ClassDB::bind_method(D_METHOD("set_theme", "theme"), &GridMap::set_theme);
  640. ClassDB::bind_method(D_METHOD("get_theme"), &GridMap::get_theme);
  641. #endif // DISABLE_DEPRECATED
  642. ClassDB::bind_method(D_METHOD("set_mesh_library", "mesh_library"), &GridMap::set_mesh_library);
  643. ClassDB::bind_method(D_METHOD("get_mesh_library"), &GridMap::get_mesh_library);
  644. ClassDB::bind_method(D_METHOD("set_cell_size", "size"), &GridMap::set_cell_size);
  645. ClassDB::bind_method(D_METHOD("get_cell_size"), &GridMap::get_cell_size);
  646. ClassDB::bind_method(D_METHOD("set_cell_scale", "scale"), &GridMap::set_cell_scale);
  647. ClassDB::bind_method(D_METHOD("get_cell_scale"), &GridMap::get_cell_scale);
  648. ClassDB::bind_method(D_METHOD("set_octant_size", "size"), &GridMap::set_octant_size);
  649. ClassDB::bind_method(D_METHOD("get_octant_size"), &GridMap::get_octant_size);
  650. ClassDB::bind_method(D_METHOD("set_cell_item", "x", "y", "z", "item", "orientation"), &GridMap::set_cell_item, DEFVAL(0));
  651. ClassDB::bind_method(D_METHOD("get_cell_item", "x", "y", "z"), &GridMap::get_cell_item);
  652. ClassDB::bind_method(D_METHOD("get_cell_item_orientation", "x", "y", "z"), &GridMap::get_cell_item_orientation);
  653. ClassDB::bind_method(D_METHOD("world_to_map", "pos"), &GridMap::world_to_map);
  654. ClassDB::bind_method(D_METHOD("map_to_world", "x", "y", "z"), &GridMap::map_to_world);
  655. ClassDB::bind_method(D_METHOD("_update_octants_callback"), &GridMap::_update_octants_callback);
  656. ClassDB::bind_method(D_METHOD("resource_changed", "resource"), &GridMap::resource_changed);
  657. ClassDB::bind_method(D_METHOD("set_center_x", "enable"), &GridMap::set_center_x);
  658. ClassDB::bind_method(D_METHOD("get_center_x"), &GridMap::get_center_x);
  659. ClassDB::bind_method(D_METHOD("set_center_y", "enable"), &GridMap::set_center_y);
  660. ClassDB::bind_method(D_METHOD("get_center_y"), &GridMap::get_center_y);
  661. ClassDB::bind_method(D_METHOD("set_center_z", "enable"), &GridMap::set_center_z);
  662. ClassDB::bind_method(D_METHOD("get_center_z"), &GridMap::get_center_z);
  663. ClassDB::bind_method(D_METHOD("set_clip", "enabled", "clipabove", "floor", "axis"), &GridMap::set_clip, DEFVAL(true), DEFVAL(0), DEFVAL(Vector3::AXIS_X));
  664. ClassDB::bind_method(D_METHOD("clear"), &GridMap::clear);
  665. ClassDB::bind_method(D_METHOD("get_used_cells"), &GridMap::get_used_cells);
  666. ClassDB::bind_method(D_METHOD("get_meshes"), &GridMap::get_meshes);
  667. ClassDB::bind_method(D_METHOD("get_bake_meshes"), &GridMap::get_bake_meshes);
  668. ClassDB::bind_method(D_METHOD("get_bake_mesh_instance", "idx"), &GridMap::get_bake_mesh_instance);
  669. ClassDB::bind_method(D_METHOD("clear_baked_meshes"), &GridMap::clear_baked_meshes);
  670. ClassDB::bind_method(D_METHOD("make_baked_meshes", "gen_lightmap_uv", "lightmap_uv_texel_size"), &GridMap::make_baked_meshes, DEFVAL(false), DEFVAL(0.1));
  671. #ifndef DISABLE_DEPRECATED
  672. ADD_PROPERTY(PropertyInfo(Variant::OBJECT, "theme", PROPERTY_HINT_RESOURCE_TYPE, "MeshLibrary", 0), "set_theme", "get_theme");
  673. #endif // DISABLE_DEPRECATED
  674. ADD_PROPERTY(PropertyInfo(Variant::OBJECT, "mesh_library", PROPERTY_HINT_RESOURCE_TYPE, "MeshLibrary"), "set_mesh_library", "get_mesh_library");
  675. ADD_GROUP("Cell", "cell_");
  676. ADD_PROPERTY(PropertyInfo(Variant::VECTOR3, "cell_size"), "set_cell_size", "get_cell_size");
  677. ADD_PROPERTY(PropertyInfo(Variant::INT, "cell_octant_size", PROPERTY_HINT_RANGE, "1,1024,1"), "set_octant_size", "get_octant_size");
  678. ADD_PROPERTY(PropertyInfo(Variant::BOOL, "cell_center_x"), "set_center_x", "get_center_x");
  679. ADD_PROPERTY(PropertyInfo(Variant::BOOL, "cell_center_y"), "set_center_y", "get_center_y");
  680. ADD_PROPERTY(PropertyInfo(Variant::BOOL, "cell_center_z"), "set_center_z", "get_center_z");
  681. ADD_PROPERTY(PropertyInfo(Variant::REAL, "cell_scale"), "set_cell_scale", "get_cell_scale");
  682. ADD_GROUP("Collision", "collision_");
  683. ADD_PROPERTY(PropertyInfo(Variant::INT, "collision_layer", PROPERTY_HINT_LAYERS_3D_PHYSICS), "set_collision_layer", "get_collision_layer");
  684. ADD_PROPERTY(PropertyInfo(Variant::INT, "collision_mask", PROPERTY_HINT_LAYERS_3D_PHYSICS), "set_collision_mask", "get_collision_mask");
  685. BIND_CONSTANT(INVALID_CELL_ITEM);
  686. }
  687. void GridMap::set_clip(bool p_enabled, bool p_clip_above, int p_floor, Vector3::Axis p_axis) {
  688. if (!p_enabled && !clip)
  689. return;
  690. if (clip && p_enabled && clip_floor == p_floor && p_clip_above == clip_above && p_axis == clip_axis)
  691. return;
  692. clip = p_enabled;
  693. clip_floor = p_floor;
  694. clip_axis = p_axis;
  695. clip_above = p_clip_above;
  696. //make it all update
  697. for (Map<OctantKey, Octant *>::Element *E = octant_map.front(); E; E = E->next()) {
  698. Octant *g = E->get();
  699. g->dirty = true;
  700. }
  701. awaiting_update = true;
  702. _update_octants_callback();
  703. }
  704. void GridMap::set_cell_scale(float p_scale) {
  705. cell_scale = p_scale;
  706. _recreate_octant_data();
  707. }
  708. float GridMap::get_cell_scale() const {
  709. return cell_scale;
  710. }
  711. Array GridMap::get_used_cells() const {
  712. Array a;
  713. a.resize(cell_map.size());
  714. int i = 0;
  715. for (Map<IndexKey, Cell>::Element *E = cell_map.front(); E; E = E->next()) {
  716. Vector3 p(E->key().x, E->key().y, E->key().z);
  717. a[i++] = p;
  718. }
  719. return a;
  720. }
  721. Array GridMap::get_meshes() {
  722. if (mesh_library.is_null())
  723. return Array();
  724. Vector3 ofs = _get_offset();
  725. Array meshes;
  726. for (Map<IndexKey, Cell>::Element *E = cell_map.front(); E; E = E->next()) {
  727. int id = E->get().item;
  728. if (!mesh_library->has_item(id))
  729. continue;
  730. Ref<Mesh> mesh = mesh_library->get_item_mesh(id);
  731. if (mesh.is_null())
  732. continue;
  733. IndexKey ik = E->key();
  734. Vector3 cellpos = Vector3(ik.x, ik.y, ik.z);
  735. Transform xform;
  736. xform.basis.set_orthogonal_index(E->get().rot);
  737. xform.set_origin(cellpos * cell_size + ofs);
  738. xform.basis.scale(Vector3(cell_scale, cell_scale, cell_scale));
  739. meshes.push_back(xform);
  740. meshes.push_back(mesh);
  741. }
  742. return meshes;
  743. }
  744. Vector3 GridMap::_get_offset() const {
  745. return Vector3(
  746. cell_size.x * 0.5 * int(center_x),
  747. cell_size.y * 0.5 * int(center_y),
  748. cell_size.z * 0.5 * int(center_z));
  749. }
  750. void GridMap::clear_baked_meshes() {
  751. for (int i = 0; i < baked_meshes.size(); i++) {
  752. VS::get_singleton()->free(baked_meshes[i].instance);
  753. }
  754. baked_meshes.clear();
  755. _recreate_octant_data();
  756. }
  757. void GridMap::make_baked_meshes(bool p_gen_lightmap_uv, float p_lightmap_uv_texel_size) {
  758. if (!mesh_library.is_valid())
  759. return;
  760. //generate
  761. Map<OctantKey, Map<Ref<Material>, Ref<SurfaceTool> > > surface_map;
  762. for (Map<IndexKey, Cell>::Element *E = cell_map.front(); E; E = E->next()) {
  763. IndexKey key = E->key();
  764. int item = E->get().item;
  765. if (!mesh_library->has_item(item))
  766. continue;
  767. Ref<Mesh> mesh = mesh_library->get_item_mesh(item);
  768. if (!mesh.is_valid())
  769. continue;
  770. Vector3 cellpos = Vector3(key.x, key.y, key.z);
  771. Vector3 ofs = _get_offset();
  772. Transform xform;
  773. xform.basis.set_orthogonal_index(E->get().rot);
  774. xform.set_origin(cellpos * cell_size + ofs);
  775. xform.basis.scale(Vector3(cell_scale, cell_scale, cell_scale));
  776. OctantKey ok;
  777. ok.x = key.x / octant_size;
  778. ok.y = key.y / octant_size;
  779. ok.z = key.z / octant_size;
  780. if (!surface_map.has(ok)) {
  781. surface_map[ok] = Map<Ref<Material>, Ref<SurfaceTool> >();
  782. }
  783. Map<Ref<Material>, Ref<SurfaceTool> > &mat_map = surface_map[ok];
  784. for (int i = 0; i < mesh->get_surface_count(); i++) {
  785. if (mesh->surface_get_primitive_type(i) != Mesh::PRIMITIVE_TRIANGLES)
  786. continue;
  787. Ref<Material> surf_mat = mesh->surface_get_material(i);
  788. if (!mat_map.has(surf_mat)) {
  789. Ref<SurfaceTool> st;
  790. st.instance();
  791. st->begin(Mesh::PRIMITIVE_TRIANGLES);
  792. st->set_material(surf_mat);
  793. mat_map[surf_mat] = st;
  794. }
  795. mat_map[surf_mat]->append_from(mesh, i, xform);
  796. }
  797. }
  798. for (Map<OctantKey, Map<Ref<Material>, Ref<SurfaceTool> > >::Element *E = surface_map.front(); E; E = E->next()) {
  799. Ref<ArrayMesh> mesh;
  800. mesh.instance();
  801. for (Map<Ref<Material>, Ref<SurfaceTool> >::Element *F = E->get().front(); F; F = F->next()) {
  802. F->get()->commit(mesh);
  803. }
  804. BakedMesh bm;
  805. bm.mesh = mesh;
  806. bm.instance = VS::get_singleton()->instance_create();
  807. VS::get_singleton()->get_singleton()->instance_set_base(bm.instance, bm.mesh->get_rid());
  808. VS::get_singleton()->instance_attach_object_instance_id(bm.instance, get_instance_id());
  809. if (is_inside_tree()) {
  810. VS::get_singleton()->instance_set_scenario(bm.instance, get_world()->get_scenario());
  811. VS::get_singleton()->instance_set_transform(bm.instance, get_global_transform());
  812. }
  813. if (p_gen_lightmap_uv) {
  814. mesh->lightmap_unwrap(get_global_transform(), p_lightmap_uv_texel_size);
  815. }
  816. baked_meshes.push_back(bm);
  817. }
  818. _recreate_octant_data();
  819. }
  820. Array GridMap::get_bake_meshes() {
  821. if (!baked_meshes.size()) {
  822. make_baked_meshes(true);
  823. }
  824. Array arr;
  825. for (int i = 0; i < baked_meshes.size(); i++) {
  826. arr.push_back(baked_meshes[i].mesh);
  827. arr.push_back(Transform());
  828. }
  829. return arr;
  830. }
  831. RID GridMap::get_bake_mesh_instance(int p_idx) {
  832. ERR_FAIL_INDEX_V(p_idx, baked_meshes.size(), RID());
  833. return baked_meshes[p_idx].instance;
  834. }
  835. GridMap::GridMap() {
  836. collision_layer = 1;
  837. collision_mask = 1;
  838. cell_size = Vector3(2, 2, 2);
  839. octant_size = 8;
  840. awaiting_update = false;
  841. _in_tree = false;
  842. center_x = true;
  843. center_y = true;
  844. center_z = true;
  845. clip = false;
  846. clip_floor = 0;
  847. clip_axis = Vector3::AXIS_Z;
  848. clip_above = true;
  849. cell_scale = 1.0;
  850. navigation = NULL;
  851. set_notify_transform(true);
  852. recreating_octants = false;
  853. }
  854. GridMap::~GridMap() {
  855. if (!mesh_library.is_null())
  856. mesh_library->unregister_owner(this);
  857. clear();
  858. }