spatial_editor_plugin.cpp 138 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715371637173718371937203721372237233724372537263727372837293730373137323733373437353736373737383739374037413742374337443745374637473748374937503751375237533754375537563757375837593760376137623763376437653766376737683769377037713772377337743775377637773778377937803781378237833784378537863787378837893790379137923793379437953796379737983799380038013802380338043805380638073808380938103811381238133814381538163817381838193820382138223823382438253826382738283829383038313832383338343835383638373838383938403841384238433844384538463847384838493850385138523853385438553856385738583859386038613862386338643865386638673868386938703871387238733874387538763877387838793880388138823883388438853886388738883889389038913892389338943895389638973898389939003901390239033904390539063907390839093910391139123913391439153916391739183919392039213922392339243925392639273928392939303931393239333934393539363937393839393940394139423943394439453946394739483949395039513952395339543955395639573958395939603961396239633964396539663967396839693970397139723973397439753976397739783979398039813982398339843985398639873988398939903991399239933994399539963997399839994000400140024003400440054006400740084009401040114012401340144015401640174018401940204021402240234024402540264027402840294030403140324033403440354036403740384039404040414042404340444045404640474048404940504051405240534054405540564057405840594060406140624063406440654066406740684069407040714072407340744075407640774078407940804081408240834084408540864087408840894090409140924093409440954096409740984099410041014102410341044105410641074108410941104111411241134114411541164117411841194120412141224123412441254126412741284129413041314132413341344135413641374138413941404141414241434144414541464147414841494150415141524153415441554156415741584159416041614162416341644165416641674168416941704171417241734174417541764177417841794180418141824183418441854186418741884189419041914192419341944195419641974198419942004201420242034204420542064207420842094210421142124213421442154216421742184219422042214222422342244225422642274228422942304231423242334234423542364237423842394240424142424243424442454246424742484249425042514252425342544255425642574258425942604261426242634264426542664267426842694270427142724273427442754276427742784279428042814282
  1. /*************************************************************************/
  2. /* spatial_editor_plugin.cpp */
  3. /*************************************************************************/
  4. /* This file is part of: */
  5. /* GODOT ENGINE */
  6. /* https://godotengine.org */
  7. /*************************************************************************/
  8. /* Copyright (c) 2007-2017 Juan Linietsky, Ariel Manzur. */
  9. /* Copyright (c) 2014-2017 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 "spatial_editor_plugin.h"
  31. #include "camera_matrix.h"
  32. #include "core/os/input.h"
  33. #include "editor/animation_editor.h"
  34. #include "editor/editor_node.h"
  35. #include "editor/editor_settings.h"
  36. #include "editor/plugins/animation_player_editor_plugin.h"
  37. #include "editor/spatial_editor_gizmos.h"
  38. #include "os/keyboard.h"
  39. #include "print_string.h"
  40. #include "project_settings.h"
  41. #include "scene/3d/camera.h"
  42. #include "scene/3d/visual_instance.h"
  43. #include "scene/resources/surface_tool.h"
  44. #include "sort.h"
  45. #define DISTANCE_DEFAULT 4
  46. #define GIZMO_ARROW_SIZE 0.3
  47. #define GIZMO_RING_HALF_WIDTH 0.1
  48. //#define GIZMO_SCALE_DEFAULT 0.28
  49. #define GIZMO_SCALE_DEFAULT 0.15
  50. #define GIZMO_PLANE_SIZE 0.2
  51. #define GIZMO_PLANE_DST 0.3
  52. #define GIZMO_CIRCLE_SIZE 0.9
  53. #define ZOOM_MIN_DISTANCE 0.001
  54. #define ZOOM_MULTIPLIER 1.08
  55. #define ZOOM_INDICATOR_DELAY_S 1.5
  56. #define FREELOOK_MIN_SPEED 0.1
  57. #define MIN_Z 0.01
  58. #define MAX_Z 10000
  59. #define MIN_FOV 0.01
  60. #define MAX_FOV 179
  61. void SpatialEditorViewport::_update_camera(float p_interp_delta) {
  62. if (orthogonal) {
  63. //camera->set_orthogonal(size.width*cursor.distance,get_znear(),get_zfar());
  64. camera->set_orthogonal(2 * cursor.distance, 0.1, 8192);
  65. } else
  66. camera->set_perspective(get_fov(), get_znear(), get_zfar());
  67. Transform new_transform = to_camera_transform(cursor);
  68. Transform old_transform = camera->get_global_transform();
  69. Transform transform;
  70. bool disable_interp = (Input::get_singleton()->get_mouse_button_mask() & (2 | 4)) || Input::get_singleton()->is_key_pressed(KEY_SHIFT) || Input::get_singleton()->is_key_pressed(KEY_ALT) || Input::get_singleton()->is_key_pressed(KEY_CONTROL);
  71. if (p_interp_delta && !disable_interp) {
  72. //interpolate
  73. float interp_speed = 14; //maybe should be made configuration
  74. transform = old_transform.interpolate_with(new_transform, MIN(1.0, p_interp_delta * interp_speed));
  75. } else {
  76. transform = new_transform;
  77. }
  78. float tolerance = 0.0001;
  79. bool equal = true;
  80. for (int i = 0; i < 3; i++) {
  81. if (transform.basis[i].distance_to(old_transform.basis[i]) > tolerance) {
  82. equal = false;
  83. break;
  84. }
  85. }
  86. if (equal && transform.origin.distance_to(old_transform.origin) > tolerance) {
  87. equal = false;
  88. }
  89. if (equal) {
  90. transform = new_transform;
  91. }
  92. if (!equal || p_interp_delta == 0) {
  93. //print_line(transform);
  94. camera->set_global_transform(transform);
  95. update_transform_gizmo_view();
  96. }
  97. }
  98. Transform SpatialEditorViewport::to_camera_transform(const Cursor &p_cursor) const {
  99. Transform camera_transform;
  100. camera_transform.translate(p_cursor.pos);
  101. camera_transform.basis.rotate(Vector3(1, 0, 0), -p_cursor.x_rot);
  102. camera_transform.basis.rotate(Vector3(0, 1, 0), -p_cursor.y_rot);
  103. if (orthogonal)
  104. camera_transform.translate(0, 0, 4096);
  105. else
  106. camera_transform.translate(0, 0, p_cursor.distance);
  107. return camera_transform;
  108. }
  109. String SpatialEditorGizmo::get_handle_name(int p_idx) const {
  110. if (get_script_instance() && get_script_instance()->has_method("get_handle_name"))
  111. return get_script_instance()->call("get_handle_name", p_idx);
  112. return "";
  113. }
  114. Variant SpatialEditorGizmo::get_handle_value(int p_idx) const {
  115. if (get_script_instance() && get_script_instance()->has_method("get_handle_value"))
  116. return get_script_instance()->call("get_handle_value", p_idx);
  117. return Variant();
  118. }
  119. void SpatialEditorGizmo::set_handle(int p_idx, Camera *p_camera, const Point2 &p_point) {
  120. if (get_script_instance() && get_script_instance()->has_method("set_handle"))
  121. get_script_instance()->call("set_handle", p_idx, p_camera, p_point);
  122. }
  123. void SpatialEditorGizmo::commit_handle(int p_idx, const Variant &p_restore, bool p_cancel) {
  124. if (get_script_instance() && get_script_instance()->has_method("commit_handle"))
  125. get_script_instance()->call("commit_handle", p_idx, p_restore, p_cancel);
  126. }
  127. bool SpatialEditorGizmo::intersect_frustum(const Camera *p_camera, const Vector<Plane> &p_frustum) {
  128. return false;
  129. }
  130. bool SpatialEditorGizmo::intersect_ray(const Camera *p_camera, const Point2 &p_point, Vector3 &r_pos, Vector3 &r_normal, int *r_gizmo_handle, bool p_sec_first) {
  131. return false;
  132. }
  133. SpatialEditorGizmo::SpatialEditorGizmo() {
  134. selected = false;
  135. }
  136. int SpatialEditorViewport::get_selected_count() const {
  137. Map<Node *, Object *> &selection = editor_selection->get_selection();
  138. int count = 0;
  139. for (Map<Node *, Object *>::Element *E = selection.front(); E; E = E->next()) {
  140. Spatial *sp = Object::cast_to<Spatial>(E->key());
  141. if (!sp)
  142. continue;
  143. SpatialEditorSelectedItem *se = editor_selection->get_node_editor_data<SpatialEditorSelectedItem>(sp);
  144. if (!se)
  145. continue;
  146. count++;
  147. }
  148. return count;
  149. }
  150. float SpatialEditorViewport::get_znear() const {
  151. return CLAMP(spatial_editor->get_znear(), MIN_Z, MAX_Z);
  152. }
  153. float SpatialEditorViewport::get_zfar() const {
  154. return CLAMP(spatial_editor->get_zfar(), MIN_Z, MAX_Z);
  155. }
  156. float SpatialEditorViewport::get_fov() const {
  157. return CLAMP(spatial_editor->get_fov(), MIN_FOV, MAX_FOV);
  158. }
  159. Transform SpatialEditorViewport::_get_camera_transform() const {
  160. return camera->get_global_transform();
  161. }
  162. Vector3 SpatialEditorViewport::_get_camera_pos() const {
  163. return _get_camera_transform().origin;
  164. }
  165. Point2 SpatialEditorViewport::_point_to_screen(const Vector3 &p_point) {
  166. return camera->unproject_position(p_point);
  167. }
  168. Vector3 SpatialEditorViewport::_get_ray_pos(const Vector2 &p_pos) const {
  169. return camera->project_ray_origin(p_pos);
  170. }
  171. Vector3 SpatialEditorViewport::_get_camera_normal() const {
  172. return -_get_camera_transform().basis.get_axis(2);
  173. }
  174. Vector3 SpatialEditorViewport::_get_ray(const Vector2 &p_pos) {
  175. return camera->project_ray_normal(p_pos);
  176. }
  177. /*
  178. void SpatialEditorViewport::_clear_id(Spatial *p_node) {
  179. editor_selection->remove_node(p_node);
  180. }
  181. */
  182. void SpatialEditorViewport::_clear_selected() {
  183. editor_selection->clear();
  184. }
  185. void SpatialEditorViewport::_select_clicked(bool p_append, bool p_single) {
  186. if (!clicked)
  187. return;
  188. Spatial *sp = Object::cast_to<Spatial>(ObjectDB::get_instance(clicked));
  189. if (!sp)
  190. return;
  191. _select(sp, clicked_wants_append, true);
  192. }
  193. void SpatialEditorViewport::_select(Spatial *p_node, bool p_append, bool p_single) {
  194. if (!p_append) {
  195. // should not modify the selection..
  196. editor_selection->clear();
  197. editor_selection->add_node(p_node);
  198. } else {
  199. if (editor_selection->is_selected(p_node) && p_single) {
  200. //erase
  201. editor_selection->remove_node(p_node);
  202. } else {
  203. editor_selection->add_node(p_node);
  204. }
  205. }
  206. }
  207. ObjectID SpatialEditorViewport::_select_ray(const Point2 &p_pos, bool p_append, bool &r_includes_current, int *r_gizmo_handle, bool p_alt_select) {
  208. if (r_gizmo_handle)
  209. *r_gizmo_handle = -1;
  210. Vector3 ray = _get_ray(p_pos);
  211. Vector3 pos = _get_ray_pos(p_pos);
  212. Vector<ObjectID> instances = VisualServer::get_singleton()->instances_cull_ray(pos, ray, get_tree()->get_root()->get_world()->get_scenario());
  213. Set<Ref<SpatialEditorGizmo> > found_gizmos;
  214. Node *edited_scene = get_tree()->get_edited_scene_root();
  215. ObjectID closest = 0;
  216. Spatial *item = NULL;
  217. float closest_dist = 1e20;
  218. int selected_handle = -1;
  219. for (int i = 0; i < instances.size(); i++) {
  220. Spatial *spat = Object::cast_to<Spatial>(ObjectDB::get_instance(instances[i]));
  221. if (!spat)
  222. continue;
  223. Ref<SpatialEditorGizmo> seg = spat->get_gizmo();
  224. if ((!seg.is_valid()) || found_gizmos.has(seg)) {
  225. continue;
  226. }
  227. found_gizmos.insert(seg);
  228. Vector3 point;
  229. Vector3 normal;
  230. int handle = -1;
  231. bool inters = seg->intersect_ray(camera, p_pos, point, normal, NULL, p_alt_select);
  232. if (!inters)
  233. continue;
  234. float dist = pos.distance_to(point);
  235. if (dist < 0)
  236. continue;
  237. if (dist < closest_dist) {
  238. //make sure that whathever is selected is editable
  239. while (spat && spat != edited_scene && spat->get_owner() != edited_scene && !edited_scene->is_editable_instance(spat->get_owner())) {
  240. spat = Object::cast_to<Spatial>(spat->get_owner());
  241. }
  242. if (spat) {
  243. item = spat;
  244. closest = spat->get_instance_id();
  245. closest_dist = dist;
  246. selected_handle = handle;
  247. } else {
  248. ERR_PRINT("Bug?");
  249. }
  250. }
  251. // if (editor_selection->is_selected(spat))
  252. // r_includes_current=true;
  253. }
  254. if (!item)
  255. return 0;
  256. if (!editor_selection->is_selected(item) || (r_gizmo_handle && selected_handle >= 0)) {
  257. if (r_gizmo_handle)
  258. *r_gizmo_handle = selected_handle;
  259. }
  260. return closest;
  261. }
  262. void SpatialEditorViewport::_find_items_at_pos(const Point2 &p_pos, bool &r_includes_current, Vector<_RayResult> &results, bool p_alt_select) {
  263. Vector3 ray = _get_ray(p_pos);
  264. Vector3 pos = _get_ray_pos(p_pos);
  265. Vector<ObjectID> instances = VisualServer::get_singleton()->instances_cull_ray(pos, ray, get_tree()->get_root()->get_world()->get_scenario());
  266. Set<Ref<SpatialEditorGizmo> > found_gizmos;
  267. r_includes_current = false;
  268. for (int i = 0; i < instances.size(); i++) {
  269. Spatial *spat = Object::cast_to<Spatial>(ObjectDB::get_instance(instances[i]));
  270. if (!spat)
  271. continue;
  272. Ref<SpatialEditorGizmo> seg = spat->get_gizmo();
  273. if (!seg.is_valid())
  274. continue;
  275. if (found_gizmos.has(seg))
  276. continue;
  277. found_gizmos.insert(seg);
  278. Vector3 point;
  279. Vector3 normal;
  280. int handle = -1;
  281. bool inters = seg->intersect_ray(camera, p_pos, point, normal, NULL, p_alt_select);
  282. if (!inters)
  283. continue;
  284. float dist = pos.distance_to(point);
  285. if (dist < 0)
  286. continue;
  287. if (editor_selection->is_selected(spat))
  288. r_includes_current = true;
  289. _RayResult res;
  290. res.item = spat;
  291. res.depth = dist;
  292. res.handle = handle;
  293. results.push_back(res);
  294. }
  295. if (results.empty())
  296. return;
  297. results.sort();
  298. }
  299. Vector3 SpatialEditorViewport::_get_screen_to_space(const Vector3 &p_vector3) {
  300. CameraMatrix cm;
  301. cm.set_perspective(get_fov(), get_size().aspect(), get_znear(), get_zfar());
  302. float screen_w, screen_h;
  303. cm.get_viewport_size(screen_w, screen_h);
  304. Transform camera_transform;
  305. camera_transform.translate(cursor.pos);
  306. camera_transform.basis.rotate(Vector3(1, 0, 0), -cursor.x_rot);
  307. camera_transform.basis.rotate(Vector3(0, 1, 0), -cursor.y_rot);
  308. camera_transform.translate(0, 0, cursor.distance);
  309. return camera_transform.xform(Vector3(((p_vector3.x / get_size().width) * 2.0 - 1.0) * screen_w, ((1.0 - (p_vector3.y / get_size().height)) * 2.0 - 1.0) * screen_h, -get_znear()));
  310. }
  311. void SpatialEditorViewport::_select_region() {
  312. if (cursor.region_begin == cursor.region_end)
  313. return; //nothing really
  314. Vector3 box[4] = {
  315. Vector3(
  316. MIN(cursor.region_begin.x, cursor.region_end.x),
  317. MIN(cursor.region_begin.y, cursor.region_end.y),
  318. 0),
  319. Vector3(
  320. MAX(cursor.region_begin.x, cursor.region_end.x),
  321. MIN(cursor.region_begin.y, cursor.region_end.y),
  322. 0),
  323. Vector3(
  324. MAX(cursor.region_begin.x, cursor.region_end.x),
  325. MAX(cursor.region_begin.y, cursor.region_end.y),
  326. 0),
  327. Vector3(
  328. MIN(cursor.region_begin.x, cursor.region_end.x),
  329. MAX(cursor.region_begin.y, cursor.region_end.y),
  330. 0)
  331. };
  332. Vector<Plane> frustum;
  333. Vector3 cam_pos = _get_camera_pos();
  334. Set<Ref<SpatialEditorGizmo> > found_gizmos;
  335. for (int i = 0; i < 4; i++) {
  336. Vector3 a = _get_screen_to_space(box[i]);
  337. Vector3 b = _get_screen_to_space(box[(i + 1) % 4]);
  338. frustum.push_back(Plane(a, b, cam_pos));
  339. }
  340. Plane near(cam_pos, -_get_camera_normal());
  341. near.d -= get_znear();
  342. frustum.push_back(near);
  343. Plane far = -near;
  344. far.d += 500.0;
  345. frustum.push_back(far);
  346. Vector<ObjectID> instances = VisualServer::get_singleton()->instances_cull_convex(frustum, get_tree()->get_root()->get_world()->get_scenario());
  347. for (int i = 0; i < instances.size(); i++) {
  348. Spatial *sp = Object::cast_to<Spatial>(ObjectDB::get_instance(instances[i]));
  349. if (!sp)
  350. continue;
  351. Ref<SpatialEditorGizmo> seg = sp->get_gizmo();
  352. if (!seg.is_valid())
  353. continue;
  354. if (found_gizmos.has(seg))
  355. continue;
  356. if (seg->intersect_frustum(camera, frustum))
  357. _select(sp, true, false);
  358. }
  359. }
  360. void SpatialEditorViewport::_update_name() {
  361. String ortho = orthogonal ? TTR("Orthogonal") : TTR("Perspective");
  362. if (name != "")
  363. view_menu->set_text("[ " + name + " " + ortho + " ]");
  364. else
  365. view_menu->set_text("[ " + ortho + " ]");
  366. }
  367. void SpatialEditorViewport::_compute_edit(const Point2 &p_point) {
  368. _edit.click_ray = _get_ray(Vector2(p_point.x, p_point.y));
  369. _edit.click_ray_pos = _get_ray_pos(Vector2(p_point.x, p_point.y));
  370. _edit.plane = TRANSFORM_VIEW;
  371. spatial_editor->update_transform_gizmo();
  372. _edit.center = spatial_editor->get_gizmo_transform().origin;
  373. List<Node *> &selection = editor_selection->get_selected_node_list();
  374. //Vector3 center;
  375. //int nc=0;
  376. for (List<Node *>::Element *E = selection.front(); E; E = E->next()) {
  377. Spatial *sp = Object::cast_to<Spatial>(E->get());
  378. if (!sp)
  379. continue;
  380. SpatialEditorSelectedItem *se = editor_selection->get_node_editor_data<SpatialEditorSelectedItem>(sp);
  381. if (!se)
  382. continue;
  383. se->original = se->sp->get_global_transform();
  384. //center+=se->original.origin;
  385. //nc++;
  386. }
  387. /*
  388. if (nc)
  389. _edit.center=center/float(nc);
  390. */
  391. }
  392. static int _get_key_modifier_setting(const String &p_property) {
  393. switch (EditorSettings::get_singleton()->get(p_property).operator int()) {
  394. case 0: return 0;
  395. case 1: return KEY_SHIFT;
  396. case 2: return KEY_ALT;
  397. case 3: return KEY_META;
  398. case 4: return KEY_CONTROL;
  399. }
  400. return 0;
  401. }
  402. static int _get_key_modifier(Ref<InputEventWithModifiers> e) {
  403. if (e->get_shift())
  404. return KEY_SHIFT;
  405. if (e->get_alt())
  406. return KEY_ALT;
  407. if (e->get_control())
  408. return KEY_CONTROL;
  409. if (e->get_metakey())
  410. return KEY_META;
  411. return 0;
  412. }
  413. bool SpatialEditorViewport::_gizmo_select(const Vector2 &p_screenpos, bool p_highlight_only) {
  414. if (!spatial_editor->is_gizmo_visible())
  415. return false;
  416. if (get_selected_count() == 0) {
  417. if (p_highlight_only)
  418. spatial_editor->select_gizmo_highlight_axis(-1);
  419. return false;
  420. }
  421. Vector3 ray_pos = _get_ray_pos(Vector2(p_screenpos.x, p_screenpos.y));
  422. Vector3 ray = _get_ray(Vector2(p_screenpos.x, p_screenpos.y));
  423. Transform gt = spatial_editor->get_gizmo_transform();
  424. float gs = gizmo_scale;
  425. if (spatial_editor->get_tool_mode() == SpatialEditor::TOOL_MODE_SELECT || spatial_editor->get_tool_mode() == SpatialEditor::TOOL_MODE_MOVE) {
  426. int col_axis = -1;
  427. float col_d = 1e20;
  428. for (int i = 0; i < 3; i++) {
  429. Vector3 grabber_pos = gt.origin + gt.basis.get_axis(i) * gs;
  430. float grabber_radius = gs * GIZMO_ARROW_SIZE;
  431. Vector3 r;
  432. if (Geometry::segment_intersects_sphere(ray_pos, ray_pos + ray * 10000.0, grabber_pos, grabber_radius, &r)) {
  433. float d = r.distance_to(ray_pos);
  434. if (d < col_d) {
  435. col_d = d;
  436. col_axis = i;
  437. }
  438. }
  439. }
  440. bool is_plane_translate = false;
  441. // second try
  442. if (col_axis == -1) {
  443. col_d = 1e20;
  444. for (int i = 0; i < 3; i++) {
  445. Vector3 ivec2 = gt.basis.get_axis((i + 1) % 3).normalized();
  446. Vector3 ivec3 = gt.basis.get_axis((i + 2) % 3).normalized();
  447. Vector3 grabber_pos = gt.origin + (ivec2 + ivec3) * gs * (GIZMO_PLANE_SIZE + GIZMO_PLANE_DST);
  448. Vector3 r;
  449. Plane plane(gt.origin, gt.basis.get_axis(i).normalized());
  450. if (plane.intersects_ray(ray_pos, ray, &r)) {
  451. float dist = r.distance_to(grabber_pos);
  452. if (dist < (gs * GIZMO_PLANE_SIZE)) {
  453. float d = ray_pos.distance_to(r);
  454. if (d < col_d) {
  455. col_d = d;
  456. col_axis = i;
  457. is_plane_translate = true;
  458. }
  459. }
  460. }
  461. }
  462. }
  463. if (col_axis != -1) {
  464. if (p_highlight_only) {
  465. spatial_editor->select_gizmo_highlight_axis(col_axis + (is_plane_translate ? 6 : 0));
  466. } else {
  467. //handle plane translate
  468. _edit.mode = TRANSFORM_TRANSLATE;
  469. _compute_edit(Point2(p_screenpos.x, p_screenpos.y));
  470. _edit.plane = TransformPlane(TRANSFORM_X_AXIS + col_axis + (is_plane_translate ? 3 : 0));
  471. }
  472. return true;
  473. }
  474. }
  475. if (spatial_editor->get_tool_mode() == SpatialEditor::TOOL_MODE_SELECT || spatial_editor->get_tool_mode() == SpatialEditor::TOOL_MODE_ROTATE) {
  476. int col_axis = -1;
  477. float col_d = 1e20;
  478. for (int i = 0; i < 3; i++) {
  479. Plane plane(gt.origin, gt.basis.get_axis(i).normalized());
  480. Vector3 r;
  481. if (!plane.intersects_ray(ray_pos, ray, &r))
  482. continue;
  483. float dist = r.distance_to(gt.origin);
  484. if (dist > gs * (GIZMO_CIRCLE_SIZE - GIZMO_RING_HALF_WIDTH) && dist < gs * (GIZMO_CIRCLE_SIZE + GIZMO_RING_HALF_WIDTH)) {
  485. float d = ray_pos.distance_to(r);
  486. if (d < col_d) {
  487. col_d = d;
  488. col_axis = i;
  489. }
  490. }
  491. }
  492. if (col_axis != -1) {
  493. if (p_highlight_only) {
  494. spatial_editor->select_gizmo_highlight_axis(col_axis + 3);
  495. } else {
  496. //handle rotate
  497. _edit.mode = TRANSFORM_ROTATE;
  498. _compute_edit(Point2(p_screenpos.x, p_screenpos.y));
  499. _edit.plane = TransformPlane(TRANSFORM_X_AXIS + col_axis);
  500. }
  501. return true;
  502. }
  503. }
  504. if (p_highlight_only)
  505. spatial_editor->select_gizmo_highlight_axis(-1);
  506. return false;
  507. }
  508. void SpatialEditorViewport::_smouseenter() {
  509. if (!surface->has_focus() && (!get_focus_owner() || !get_focus_owner()->is_text_field()))
  510. surface->grab_focus();
  511. }
  512. void SpatialEditorViewport::_list_select(Ref<InputEventMouseButton> b) {
  513. _find_items_at_pos(b->get_position(), clicked_includes_current, selection_results, b->get_shift());
  514. Node *scene = editor->get_edited_scene();
  515. for (int i = 0; i < selection_results.size(); i++) {
  516. Spatial *item = selection_results[i].item;
  517. if (item != scene && item->get_owner() != scene && !scene->is_editable_instance(item->get_owner())) {
  518. //invalid result
  519. selection_results.remove(i);
  520. i--;
  521. }
  522. }
  523. clicked_wants_append = b->get_shift();
  524. if (selection_results.size() == 1) {
  525. clicked = selection_results[0].item->get_instance_id();
  526. selection_results.clear();
  527. if (clicked) {
  528. _select_clicked(clicked_wants_append, true);
  529. clicked = 0;
  530. }
  531. } else if (!selection_results.empty()) {
  532. NodePath root_path = get_tree()->get_edited_scene_root()->get_path();
  533. StringName root_name = root_path.get_name(root_path.get_name_count() - 1);
  534. for (int i = 0; i < selection_results.size(); i++) {
  535. Spatial *spat = selection_results[i].item;
  536. Ref<Texture> icon;
  537. if (spat->has_meta("_editor_icon"))
  538. icon = spat->get_meta("_editor_icon");
  539. else
  540. icon = get_icon(has_icon(spat->get_class(), "EditorIcons") ? spat->get_class() : String("Object"), "EditorIcons");
  541. String node_path = "/" + root_name + "/" + root_path.rel_path_to(spat->get_path());
  542. selection_menu->add_item(spat->get_name());
  543. selection_menu->set_item_icon(i, icon);
  544. selection_menu->set_item_metadata(i, node_path);
  545. selection_menu->set_item_tooltip(i, String(spat->get_name()) + "\nType: " + spat->get_class() + "\nPath: " + node_path);
  546. }
  547. selection_menu->set_global_position(b->get_global_position());
  548. selection_menu->popup();
  549. selection_menu->call_deferred("grab_click_focus");
  550. selection_menu->set_invalidate_click_until_motion();
  551. }
  552. }
  553. void SpatialEditorViewport::_sinput(const Ref<InputEvent> &p_event) {
  554. if (previewing)
  555. return; //do NONE
  556. {
  557. EditorNode *en = editor;
  558. EditorPluginList *force_input_forwarding_list = en->get_editor_plugins_force_input_forwarding();
  559. if (!force_input_forwarding_list->empty()) {
  560. bool discard = force_input_forwarding_list->forward_spatial_gui_input(camera, p_event, true);
  561. if (discard)
  562. return;
  563. }
  564. }
  565. {
  566. EditorNode *en = editor;
  567. EditorPluginList *over_plugin_list = en->get_editor_plugins_over();
  568. if (!over_plugin_list->empty()) {
  569. bool discard = over_plugin_list->forward_spatial_gui_input(camera, p_event, false);
  570. if (discard)
  571. return;
  572. }
  573. }
  574. Ref<InputEventMouseButton> b = p_event;
  575. if (b.is_valid()) {
  576. switch (b->get_button_index()) {
  577. case BUTTON_WHEEL_UP: {
  578. scale_cursor_distance(is_freelook_active() ? ZOOM_MULTIPLIER : 1.0 / ZOOM_MULTIPLIER);
  579. } break;
  580. case BUTTON_WHEEL_DOWN: {
  581. scale_cursor_distance(is_freelook_active() ? 1.0 / ZOOM_MULTIPLIER : ZOOM_MULTIPLIER);
  582. } break;
  583. case BUTTON_RIGHT: {
  584. NavigationScheme nav_scheme = (NavigationScheme)EditorSettings::get_singleton()->get("editors/3d/navigation_scheme").operator int();
  585. if (b->is_pressed() && _edit.gizmo.is_valid()) {
  586. //restore
  587. _edit.gizmo->commit_handle(_edit.gizmo_handle, _edit.gizmo_initial_value, true);
  588. _edit.gizmo = Ref<SpatialEditorGizmo>();
  589. }
  590. if (_edit.mode == TRANSFORM_NONE && b->is_pressed()) {
  591. if (b->get_alt()) {
  592. if (nav_scheme == NAVIGATION_MAYA)
  593. break;
  594. _list_select(b);
  595. return;
  596. }
  597. }
  598. if (_edit.mode != TRANSFORM_NONE && b->is_pressed()) {
  599. //cancel motion
  600. _edit.mode = TRANSFORM_NONE;
  601. //_validate_selection();
  602. List<Node *> &selection = editor_selection->get_selected_node_list();
  603. for (List<Node *>::Element *E = selection.front(); E; E = E->next()) {
  604. Spatial *sp = Object::cast_to<Spatial>(E->get());
  605. if (!sp)
  606. continue;
  607. SpatialEditorSelectedItem *se = editor_selection->get_node_editor_data<SpatialEditorSelectedItem>(sp);
  608. if (!se)
  609. continue;
  610. sp->set_global_transform(se->original);
  611. }
  612. surface->update();
  613. //VisualServer::get_singleton()->poly_clear(indicators);
  614. set_message(TTR("Transform Aborted."), 3);
  615. }
  616. if (b->is_pressed()) {
  617. int mod = _get_key_modifier(b);
  618. if (mod == _get_key_modifier_setting("editors/3d/freelook_activation_modifier")) {
  619. freelook_active = true;
  620. }
  621. } else {
  622. freelook_active = false;
  623. }
  624. if (freelook_active && !surface->has_focus()) {
  625. // Focus usually doesn't trigger on right-click, but in case of freelook it should,
  626. // otherwise using keyboard navigation would misbehave
  627. surface->grab_focus();
  628. }
  629. } break;
  630. case BUTTON_MIDDLE: {
  631. if (b->is_pressed() && _edit.mode != TRANSFORM_NONE) {
  632. switch (_edit.plane) {
  633. case TRANSFORM_VIEW: {
  634. _edit.plane = TRANSFORM_X_AXIS;
  635. set_message(TTR("X-Axis Transform."), 2);
  636. name = "";
  637. _update_name();
  638. } break;
  639. case TRANSFORM_X_AXIS: {
  640. _edit.plane = TRANSFORM_Y_AXIS;
  641. set_message(TTR("Y-Axis Transform."), 2);
  642. } break;
  643. case TRANSFORM_Y_AXIS: {
  644. _edit.plane = TRANSFORM_Z_AXIS;
  645. set_message(TTR("Z-Axis Transform."), 2);
  646. } break;
  647. case TRANSFORM_Z_AXIS: {
  648. _edit.plane = TRANSFORM_VIEW;
  649. set_message(TTR("View Plane Transform."), 2);
  650. } break;
  651. }
  652. }
  653. } break;
  654. case BUTTON_LEFT: {
  655. if (b->is_pressed()) {
  656. NavigationScheme nav_scheme = (NavigationScheme)EditorSettings::get_singleton()->get("editors/3d/navigation_scheme").operator int();
  657. if ((nav_scheme == NAVIGATION_MAYA || nav_scheme == NAVIGATION_MODO) && b->get_alt()) {
  658. break;
  659. }
  660. if (spatial_editor->get_tool_mode() == SpatialEditor::TOOL_MODE_LIST_SELECT) {
  661. _list_select(b);
  662. break;
  663. }
  664. _edit.mouse_pos = b->get_position();
  665. _edit.snap = false;
  666. _edit.mode = TRANSFORM_NONE;
  667. //gizmo has priority over everything
  668. bool can_select_gizmos = true;
  669. {
  670. int idx = view_menu->get_popup()->get_item_index(VIEW_GIZMOS);
  671. can_select_gizmos = view_menu->get_popup()->is_item_checked(idx);
  672. }
  673. if (can_select_gizmos && spatial_editor->get_selected()) {
  674. Ref<SpatialEditorGizmo> seg = spatial_editor->get_selected()->get_gizmo();
  675. if (seg.is_valid()) {
  676. int handle = -1;
  677. Vector3 point;
  678. Vector3 normal;
  679. bool inters = seg->intersect_ray(camera, _edit.mouse_pos, point, normal, &handle, b->get_shift());
  680. if (inters && handle != -1) {
  681. _edit.gizmo = seg;
  682. _edit.gizmo_handle = handle;
  683. //_edit.gizmo_initial_pos=seg->get_handle_pos(gizmo_handle);
  684. _edit.gizmo_initial_value = seg->get_handle_value(handle);
  685. break;
  686. }
  687. }
  688. }
  689. if (_gizmo_select(_edit.mouse_pos))
  690. break;
  691. clicked = 0;
  692. clicked_includes_current = false;
  693. if ((spatial_editor->get_tool_mode() == SpatialEditor::TOOL_MODE_SELECT && b->get_control()) || spatial_editor->get_tool_mode() == SpatialEditor::TOOL_MODE_ROTATE) {
  694. /* HANDLE ROTATION */
  695. if (get_selected_count() == 0)
  696. break; //bye
  697. //handle rotate
  698. _edit.mode = TRANSFORM_ROTATE;
  699. _compute_edit(b->get_position());
  700. break;
  701. }
  702. if (spatial_editor->get_tool_mode() == SpatialEditor::TOOL_MODE_MOVE) {
  703. if (get_selected_count() == 0)
  704. break; //bye
  705. //handle translate
  706. _edit.mode = TRANSFORM_TRANSLATE;
  707. _compute_edit(b->get_position());
  708. break;
  709. }
  710. if (spatial_editor->get_tool_mode() == SpatialEditor::TOOL_MODE_SCALE) {
  711. if (get_selected_count() == 0)
  712. break; //bye
  713. //handle rotate
  714. _edit.mode = TRANSFORM_SCALE;
  715. _compute_edit(b->get_position());
  716. break;
  717. }
  718. // todo scale
  719. int gizmo_handle = -1;
  720. clicked = _select_ray(b->get_position(), b->get_shift(), clicked_includes_current, &gizmo_handle, b->get_shift());
  721. //clicking is always deferred to either move or release
  722. clicked_wants_append = b->get_shift();
  723. if (!clicked) {
  724. if (!clicked_wants_append)
  725. _clear_selected();
  726. //default to regionselect
  727. cursor.region_select = true;
  728. cursor.region_begin = b->get_position();
  729. cursor.region_end = b->get_position();
  730. }
  731. if (clicked && gizmo_handle >= 0) {
  732. Spatial *spa = Object::cast_to<Spatial>(ObjectDB::get_instance(clicked));
  733. if (spa) {
  734. Ref<SpatialEditorGizmo> seg = spa->get_gizmo();
  735. if (seg.is_valid()) {
  736. _edit.gizmo = seg;
  737. _edit.gizmo_handle = gizmo_handle;
  738. //_edit.gizmo_initial_pos=seg->get_handle_pos(gizmo_handle);
  739. _edit.gizmo_initial_value = seg->get_handle_value(gizmo_handle);
  740. //print_line("GIZMO: "+itos(gizmo_handle)+" FROMPOS: "+_edit.orig_gizmo_pos);
  741. break;
  742. }
  743. }
  744. //_compute_edit(Point2(b.x,b.y)); //in case a motion happens..
  745. }
  746. surface->update();
  747. } else {
  748. if (_edit.gizmo.is_valid()) {
  749. _edit.gizmo->commit_handle(_edit.gizmo_handle, _edit.gizmo_initial_value, false);
  750. _edit.gizmo = Ref<SpatialEditorGizmo>();
  751. break;
  752. }
  753. if (clicked) {
  754. _select_clicked(clicked_wants_append, true);
  755. //clickd processing was deferred
  756. clicked = 0;
  757. }
  758. if (cursor.region_select) {
  759. _select_region();
  760. cursor.region_select = false;
  761. surface->update();
  762. }
  763. if (_edit.mode != TRANSFORM_NONE) {
  764. static const char *_transform_name[4] = { "None", "Rotate", "Translate", "Scale" };
  765. undo_redo->create_action(_transform_name[_edit.mode]);
  766. List<Node *> &selection = editor_selection->get_selected_node_list();
  767. for (List<Node *>::Element *E = selection.front(); E; E = E->next()) {
  768. Spatial *sp = Object::cast_to<Spatial>(E->get());
  769. if (!sp)
  770. continue;
  771. SpatialEditorSelectedItem *se = editor_selection->get_node_editor_data<SpatialEditorSelectedItem>(sp);
  772. if (!se)
  773. continue;
  774. undo_redo->add_do_method(sp, "set_global_transform", sp->get_global_transform());
  775. undo_redo->add_undo_method(sp, "set_global_transform", se->original);
  776. }
  777. undo_redo->commit_action();
  778. _edit.mode = TRANSFORM_NONE;
  779. //VisualServer::get_singleton()->poly_clear(indicators);
  780. set_message("");
  781. }
  782. surface->update();
  783. }
  784. } break;
  785. }
  786. }
  787. Ref<InputEventMouseMotion> m = p_event;
  788. if (m.is_valid()) {
  789. _edit.mouse_pos = m->get_position();
  790. if (spatial_editor->get_selected()) {
  791. Ref<SpatialEditorGizmo> seg = spatial_editor->get_selected()->get_gizmo();
  792. if (seg.is_valid()) {
  793. int selected_handle = -1;
  794. int handle = -1;
  795. Vector3 point;
  796. Vector3 normal;
  797. bool inters = seg->intersect_ray(camera, _edit.mouse_pos, point, normal, &handle, false);
  798. if (inters && handle != -1) {
  799. selected_handle = handle;
  800. }
  801. if (selected_handle != spatial_editor->get_over_gizmo_handle()) {
  802. spatial_editor->set_over_gizmo_handle(selected_handle);
  803. spatial_editor->get_selected()->update_gizmo();
  804. if (selected_handle != -1)
  805. spatial_editor->select_gizmo_highlight_axis(-1);
  806. }
  807. }
  808. }
  809. if (spatial_editor->get_over_gizmo_handle() == -1 && !(m->get_button_mask() & 1) && !_edit.gizmo.is_valid()) {
  810. _gizmo_select(_edit.mouse_pos, true);
  811. }
  812. NavigationScheme nav_scheme = (NavigationScheme)EditorSettings::get_singleton()->get("editors/3d/navigation_scheme").operator int();
  813. NavigationMode nav_mode = NAVIGATION_NONE;
  814. if (_edit.gizmo.is_valid()) {
  815. _edit.gizmo->set_handle(_edit.gizmo_handle, camera, m->get_position());
  816. Variant v = _edit.gizmo->get_handle_value(_edit.gizmo_handle);
  817. String n = _edit.gizmo->get_handle_name(_edit.gizmo_handle);
  818. set_message(n + ": " + String(v));
  819. } else if (m->get_button_mask() & BUTTON_MASK_LEFT) {
  820. if (nav_scheme == NAVIGATION_MAYA && m->get_alt()) {
  821. nav_mode = NAVIGATION_ORBIT;
  822. } else if (nav_scheme == NAVIGATION_MODO && m->get_alt() && m->get_shift()) {
  823. nav_mode = NAVIGATION_PAN;
  824. } else if (nav_scheme == NAVIGATION_MODO && m->get_alt() && m->get_control()) {
  825. nav_mode = NAVIGATION_ZOOM;
  826. } else if (nav_scheme == NAVIGATION_MODO && m->get_alt()) {
  827. nav_mode = NAVIGATION_ORBIT;
  828. } else {
  829. if (clicked) {
  830. if (!clicked_includes_current) {
  831. _select_clicked(clicked_wants_append, true);
  832. //clickd processing was deferred
  833. }
  834. _compute_edit(_edit.mouse_pos);
  835. clicked = 0;
  836. _edit.mode = TRANSFORM_TRANSLATE;
  837. }
  838. if (cursor.region_select && nav_mode == NAVIGATION_NONE) {
  839. cursor.region_end = m->get_position();
  840. surface->update();
  841. return;
  842. }
  843. if (_edit.mode == TRANSFORM_NONE && nav_mode == NAVIGATION_NONE)
  844. return;
  845. Vector3 ray_pos = _get_ray_pos(m->get_position());
  846. Vector3 ray = _get_ray(m->get_position());
  847. switch (_edit.mode) {
  848. case TRANSFORM_SCALE: {
  849. Plane plane = Plane(_edit.center, _get_camera_normal());
  850. Vector3 intersection;
  851. if (!plane.intersects_ray(ray_pos, ray, &intersection))
  852. break;
  853. Vector3 click;
  854. if (!plane.intersects_ray(_edit.click_ray_pos, _edit.click_ray, &click))
  855. break;
  856. float center_click_dist = click.distance_to(_edit.center);
  857. float center_inters_dist = intersection.distance_to(_edit.center);
  858. if (center_click_dist == 0)
  859. break;
  860. float scale = (center_inters_dist / center_click_dist) * 100.0;
  861. if (_edit.snap || spatial_editor->is_snap_enabled()) {
  862. scale = Math::stepify(scale, spatial_editor->get_scale_snap());
  863. }
  864. set_message(vformat(TTR("Scaling to %s%%."), String::num(scale, 1)));
  865. scale /= 100.0;
  866. Transform r;
  867. r.basis.scale(Vector3(scale, scale, scale));
  868. List<Node *> &selection = editor_selection->get_selected_node_list();
  869. for (List<Node *>::Element *E = selection.front(); E; E = E->next()) {
  870. Spatial *sp = Object::cast_to<Spatial>(E->get());
  871. if (!sp)
  872. continue;
  873. SpatialEditorSelectedItem *se = editor_selection->get_node_editor_data<SpatialEditorSelectedItem>(sp);
  874. if (!se)
  875. continue;
  876. Transform original = se->original;
  877. Transform base = Transform(Basis(), _edit.center);
  878. Transform t = base * (r * (base.inverse() * original));
  879. sp->set_global_transform(t);
  880. }
  881. surface->update();
  882. } break;
  883. case TRANSFORM_TRANSLATE: {
  884. Vector3 motion_mask;
  885. Plane plane;
  886. bool plane_mv;
  887. switch (_edit.plane) {
  888. case TRANSFORM_VIEW:
  889. motion_mask = Vector3(0, 0, 0);
  890. plane = Plane(_edit.center, _get_camera_normal());
  891. break;
  892. case TRANSFORM_X_AXIS:
  893. motion_mask = spatial_editor->get_gizmo_transform().basis.get_axis(0);
  894. plane = Plane(_edit.center, motion_mask.cross(motion_mask.cross(_get_camera_normal())).normalized());
  895. break;
  896. case TRANSFORM_Y_AXIS:
  897. motion_mask = spatial_editor->get_gizmo_transform().basis.get_axis(1);
  898. plane = Plane(_edit.center, motion_mask.cross(motion_mask.cross(_get_camera_normal())).normalized());
  899. break;
  900. case TRANSFORM_Z_AXIS:
  901. motion_mask = spatial_editor->get_gizmo_transform().basis.get_axis(2);
  902. plane = Plane(_edit.center, motion_mask.cross(motion_mask.cross(_get_camera_normal())).normalized());
  903. break;
  904. case TRANSFORM_YZ:
  905. motion_mask = spatial_editor->get_gizmo_transform().basis.get_axis(2) + spatial_editor->get_gizmo_transform().basis.get_axis(1);
  906. plane = Plane(_edit.center, spatial_editor->get_gizmo_transform().basis.get_axis(0));
  907. plane_mv = true;
  908. break;
  909. case TRANSFORM_XZ:
  910. motion_mask = spatial_editor->get_gizmo_transform().basis.get_axis(2) + spatial_editor->get_gizmo_transform().basis.get_axis(0);
  911. plane = Plane(_edit.center, spatial_editor->get_gizmo_transform().basis.get_axis(1));
  912. plane_mv = true;
  913. break;
  914. case TRANSFORM_XY:
  915. motion_mask = spatial_editor->get_gizmo_transform().basis.get_axis(0) + spatial_editor->get_gizmo_transform().basis.get_axis(1);
  916. plane = Plane(_edit.center, spatial_editor->get_gizmo_transform().basis.get_axis(2));
  917. plane_mv = true;
  918. break;
  919. }
  920. Vector3 intersection;
  921. if (!plane.intersects_ray(ray_pos, ray, &intersection))
  922. break;
  923. Vector3 click;
  924. if (!plane.intersects_ray(_edit.click_ray_pos, _edit.click_ray, &click))
  925. break;
  926. //_validate_selection();
  927. Vector3 motion = intersection - click;
  928. if (motion_mask != Vector3()) {
  929. if (plane_mv)
  930. motion *= motion_mask;
  931. else
  932. motion = motion_mask.dot(motion) * motion_mask;
  933. }
  934. //set_message("Translating: "+motion);
  935. List<Node *> &selection = editor_selection->get_selected_node_list();
  936. float snap = 0;
  937. if (_edit.snap || spatial_editor->is_snap_enabled()) {
  938. snap = spatial_editor->get_translate_snap();
  939. bool local_coords = spatial_editor->are_local_coords_enabled();
  940. if (local_coords) {
  941. bool multiple = false;
  942. Spatial *node = NULL;
  943. for (List<Node *>::Element *E = selection.front(); E; E = E->next()) {
  944. Spatial *sp = Object::cast_to<Spatial>(E->get());
  945. if (!sp) {
  946. continue;
  947. }
  948. if (node) {
  949. multiple = true;
  950. break;
  951. } else {
  952. node = sp;
  953. }
  954. }
  955. if (multiple) {
  956. motion.snap(Vector3(snap, snap, snap));
  957. } else {
  958. Basis b = node->get_global_transform().basis.orthonormalized();
  959. Vector3 local_motion = b.inverse().xform(motion);
  960. local_motion.snap(Vector3(snap, snap, snap));
  961. motion = b.xform(local_motion);
  962. }
  963. } else {
  964. motion.snap(Vector3(snap, snap, snap));
  965. }
  966. }
  967. for (List<Node *>::Element *E = selection.front(); E; E = E->next()) {
  968. Spatial *sp = Object::cast_to<Spatial>(E->get());
  969. if (!sp) {
  970. continue;
  971. }
  972. SpatialEditorSelectedItem *se = editor_selection->get_node_editor_data<SpatialEditorSelectedItem>(sp);
  973. if (!se) {
  974. continue;
  975. }
  976. Transform t = se->original;
  977. t.origin += motion;
  978. sp->set_global_transform(t);
  979. }
  980. } break;
  981. case TRANSFORM_ROTATE: {
  982. Plane plane;
  983. switch (_edit.plane) {
  984. case TRANSFORM_VIEW:
  985. plane = Plane(_edit.center, _get_camera_normal());
  986. break;
  987. case TRANSFORM_X_AXIS:
  988. plane = Plane(_edit.center, spatial_editor->get_gizmo_transform().basis.get_axis(0));
  989. break;
  990. case TRANSFORM_Y_AXIS:
  991. plane = Plane(_edit.center, spatial_editor->get_gizmo_transform().basis.get_axis(1));
  992. break;
  993. case TRANSFORM_Z_AXIS:
  994. plane = Plane(_edit.center, spatial_editor->get_gizmo_transform().basis.get_axis(2));
  995. break;
  996. }
  997. Vector3 intersection;
  998. if (!plane.intersects_ray(ray_pos, ray, &intersection))
  999. break;
  1000. Vector3 click;
  1001. if (!plane.intersects_ray(_edit.click_ray_pos, _edit.click_ray, &click))
  1002. break;
  1003. Vector3 y_axis = (click - _edit.center).normalized();
  1004. Vector3 x_axis = plane.normal.cross(y_axis).normalized();
  1005. float angle = Math::atan2(x_axis.dot(intersection - _edit.center), y_axis.dot(intersection - _edit.center));
  1006. if (_edit.snap || spatial_editor->is_snap_enabled()) {
  1007. float snap = spatial_editor->get_rotate_snap();
  1008. if (snap) {
  1009. angle = Math::rad2deg(angle) + snap * 0.5; //else it wont reach +180
  1010. angle -= Math::fmod(angle, snap);
  1011. set_message(vformat(TTR("Rotating %s degrees."), rtos(angle)));
  1012. angle = Math::deg2rad(angle);
  1013. } else
  1014. set_message(vformat(TTR("Rotating %s degrees."), rtos(Math::rad2deg(angle))));
  1015. } else {
  1016. set_message(vformat(TTR("Rotating %s degrees."), rtos(Math::rad2deg(angle))));
  1017. }
  1018. Transform r;
  1019. r.basis.rotate(plane.normal, angle);
  1020. List<Node *> &selection = editor_selection->get_selected_node_list();
  1021. for (List<Node *>::Element *E = selection.front(); E; E = E->next()) {
  1022. Spatial *sp = Object::cast_to<Spatial>(E->get());
  1023. if (!sp)
  1024. continue;
  1025. SpatialEditorSelectedItem *se = editor_selection->get_node_editor_data<SpatialEditorSelectedItem>(sp);
  1026. if (!se)
  1027. continue;
  1028. Transform original = se->original;
  1029. Transform base = Transform(Basis(), _edit.center);
  1030. Transform t = base * r * base.inverse() * original;
  1031. sp->set_global_transform(t);
  1032. }
  1033. surface->update();
  1034. /*
  1035. VisualServer::get_singleton()->poly_clear(indicators);
  1036. Vector<Vector3> points;
  1037. Vector<Vector3> empty;
  1038. Vector<Color> colors;
  1039. points.push_back(intersection);
  1040. points.push_back(_edit.original.origin);
  1041. colors.push_back( Color(255,155,100) );
  1042. colors.push_back( Color(255,155,100) );
  1043. VisualServer::get_singleton()->poly_add_primitive(indicators,points,empty,colors,empty);
  1044. */
  1045. } break;
  1046. default: {}
  1047. }
  1048. }
  1049. } else if (m->get_button_mask() & BUTTON_MASK_RIGHT) {
  1050. if (nav_scheme == NAVIGATION_MAYA && m->get_alt()) {
  1051. nav_mode = NAVIGATION_ZOOM;
  1052. } else if (freelook_active) {
  1053. nav_mode = NAVIGATION_LOOK;
  1054. }
  1055. } else if (m->get_button_mask() & BUTTON_MASK_MIDDLE) {
  1056. if (nav_scheme == NAVIGATION_GODOT) {
  1057. int mod = _get_key_modifier(m);
  1058. if (mod == _get_key_modifier_setting("editors/3d/pan_modifier"))
  1059. nav_mode = NAVIGATION_PAN;
  1060. else if (mod == _get_key_modifier_setting("editors/3d/zoom_modifier"))
  1061. nav_mode = NAVIGATION_ZOOM;
  1062. else if (mod == _get_key_modifier_setting("editors/3d/orbit_modifier"))
  1063. nav_mode = NAVIGATION_ORBIT;
  1064. } else if (nav_scheme == NAVIGATION_MAYA) {
  1065. if (m->get_alt())
  1066. nav_mode = NAVIGATION_PAN;
  1067. }
  1068. } else if (EditorSettings::get_singleton()->get("editors/3d/emulate_3_button_mouse")) {
  1069. // Handle trackpad (no external mouse) use case
  1070. int mod = _get_key_modifier(m);
  1071. if (mod) {
  1072. if (mod == _get_key_modifier_setting("editors/3d/pan_modifier"))
  1073. nav_mode = NAVIGATION_PAN;
  1074. else if (mod == _get_key_modifier_setting("editors/3d/zoom_modifier"))
  1075. nav_mode = NAVIGATION_ZOOM;
  1076. else if (mod == _get_key_modifier_setting("editors/3d/orbit_modifier"))
  1077. nav_mode = NAVIGATION_ORBIT;
  1078. }
  1079. }
  1080. switch (nav_mode) {
  1081. case NAVIGATION_PAN: {
  1082. real_t pan_speed = 1 / 150.0;
  1083. int pan_speed_modifier = 10;
  1084. if (nav_scheme == NAVIGATION_MAYA && m->get_shift())
  1085. pan_speed *= pan_speed_modifier;
  1086. Point2i relative = _get_warped_mouse_motion(m);
  1087. Transform camera_transform;
  1088. camera_transform.translate(cursor.pos);
  1089. camera_transform.basis.rotate(Vector3(1, 0, 0), -cursor.x_rot);
  1090. camera_transform.basis.rotate(Vector3(0, 1, 0), -cursor.y_rot);
  1091. Vector3 translation(-relative.x * pan_speed, relative.y * pan_speed, 0);
  1092. translation *= cursor.distance / DISTANCE_DEFAULT;
  1093. camera_transform.translate(translation);
  1094. cursor.pos = camera_transform.origin;
  1095. } break;
  1096. case NAVIGATION_ZOOM: {
  1097. real_t zoom_speed = 1 / 80.0;
  1098. int zoom_speed_modifier = 10;
  1099. if (nav_scheme == NAVIGATION_MAYA && m->get_shift())
  1100. zoom_speed *= zoom_speed_modifier;
  1101. NavigationZoomStyle zoom_style = (NavigationZoomStyle)EditorSettings::get_singleton()->get("editors/3d/zoom_style").operator int();
  1102. if (zoom_style == NAVIGATION_ZOOM_HORIZONTAL) {
  1103. if (m->get_relative().x > 0)
  1104. scale_cursor_distance(1 - m->get_relative().x * zoom_speed);
  1105. else if (m->get_relative().x < 0)
  1106. scale_cursor_distance(1.0 / (1 + m->get_relative().x * zoom_speed));
  1107. } else {
  1108. if (m->get_relative().y > 0)
  1109. scale_cursor_distance(1 + m->get_relative().y * zoom_speed);
  1110. else if (m->get_relative().y < 0)
  1111. scale_cursor_distance(1.0 / (1 - m->get_relative().y * zoom_speed));
  1112. }
  1113. } break;
  1114. case NAVIGATION_ORBIT: {
  1115. Point2i relative = _get_warped_mouse_motion(m);
  1116. real_t degrees_per_pixel = EditorSettings::get_singleton()->get("editors/3d/orbit_sensitivity");
  1117. real_t radians_per_pixel = Math::deg2rad(degrees_per_pixel);
  1118. cursor.x_rot += relative.y * radians_per_pixel;
  1119. cursor.y_rot += relative.x * radians_per_pixel;
  1120. if (cursor.x_rot > Math_PI / 2.0)
  1121. cursor.x_rot = Math_PI / 2.0;
  1122. if (cursor.x_rot < -Math_PI / 2.0)
  1123. cursor.x_rot = -Math_PI / 2.0;
  1124. name = "";
  1125. _update_name();
  1126. } break;
  1127. case NAVIGATION_LOOK: {
  1128. // Freelook only works properly in perspective.
  1129. // It technically works too in ortho, but it's awful for a user due to fov being near zero
  1130. if (!orthogonal) {
  1131. Point2i relative = _get_warped_mouse_motion(m);
  1132. real_t degrees_per_pixel = EditorSettings::get_singleton()->get("editors/3d/orbit_sensitivity");
  1133. real_t radians_per_pixel = Math::deg2rad(degrees_per_pixel);
  1134. cursor.x_rot += relative.y * radians_per_pixel;
  1135. cursor.y_rot += relative.x * radians_per_pixel;
  1136. if (cursor.x_rot > Math_PI / 2.0)
  1137. cursor.x_rot = Math_PI / 2.0;
  1138. if (cursor.x_rot < -Math_PI / 2.0)
  1139. cursor.x_rot = -Math_PI / 2.0;
  1140. // Look is like Orbit, except the cursor translates, not the camera
  1141. Transform camera_transform = to_camera_transform(cursor);
  1142. Vector3 pos = camera_transform.xform(Vector3(0, 0, 0));
  1143. Vector3 diff = camera->get_translation() - pos;
  1144. cursor.pos += diff;
  1145. name = "";
  1146. _update_name();
  1147. }
  1148. } break;
  1149. default: {}
  1150. }
  1151. }
  1152. Ref<InputEventKey> k = p_event;
  1153. if (k.is_valid()) {
  1154. if (!k->is_pressed())
  1155. return;
  1156. if (ED_IS_SHORTCUT("spatial_editor/snap", p_event)) {
  1157. if (_edit.mode != TRANSFORM_NONE) {
  1158. _edit.snap = true;
  1159. }
  1160. }
  1161. if (ED_IS_SHORTCUT("spatial_editor/bottom_view", p_event)) {
  1162. cursor.y_rot = 0;
  1163. cursor.x_rot = -Math_PI / 2.0;
  1164. set_message(TTR("Bottom View."), 2);
  1165. name = TTR("Bottom");
  1166. _update_name();
  1167. }
  1168. if (ED_IS_SHORTCUT("spatial_editor/top_view", p_event)) {
  1169. cursor.y_rot = 0;
  1170. cursor.x_rot = Math_PI / 2.0;
  1171. set_message(TTR("Top View."), 2);
  1172. name = TTR("Top");
  1173. _update_name();
  1174. }
  1175. if (ED_IS_SHORTCUT("spatial_editor/rear_view", p_event)) {
  1176. cursor.x_rot = 0;
  1177. cursor.y_rot = Math_PI;
  1178. set_message(TTR("Rear View."), 2);
  1179. name = TTR("Rear");
  1180. _update_name();
  1181. }
  1182. if (ED_IS_SHORTCUT("spatial_editor/front_view", p_event)) {
  1183. cursor.x_rot = 0;
  1184. cursor.y_rot = 0;
  1185. set_message(TTR("Front View."), 2);
  1186. name = TTR("Front");
  1187. _update_name();
  1188. }
  1189. if (ED_IS_SHORTCUT("spatial_editor/left_view", p_event)) {
  1190. cursor.x_rot = 0;
  1191. cursor.y_rot = Math_PI / 2.0;
  1192. set_message(TTR("Left View."), 2);
  1193. name = TTR("Left");
  1194. _update_name();
  1195. }
  1196. if (ED_IS_SHORTCUT("spatial_editor/right_view", p_event)) {
  1197. cursor.x_rot = 0;
  1198. cursor.y_rot = -Math_PI / 2.0;
  1199. set_message(TTR("Right View."), 2);
  1200. name = TTR("Right");
  1201. _update_name();
  1202. }
  1203. if (ED_IS_SHORTCUT("spatial_editor/switch_perspective_orthogonal", p_event)) {
  1204. _menu_option(orthogonal ? VIEW_PERSPECTIVE : VIEW_ORTHOGONAL);
  1205. _update_name();
  1206. }
  1207. if (ED_IS_SHORTCUT("spatial_editor/insert_anim_key", p_event)) {
  1208. if (!get_selected_count() || _edit.mode != TRANSFORM_NONE)
  1209. return;
  1210. if (!AnimationPlayerEditor::singleton->get_key_editor()->has_keying()) {
  1211. set_message(TTR("Keying is disabled (no key inserted)."));
  1212. return;
  1213. }
  1214. List<Node *> &selection = editor_selection->get_selected_node_list();
  1215. for (List<Node *>::Element *E = selection.front(); E; E = E->next()) {
  1216. Spatial *sp = Object::cast_to<Spatial>(E->get());
  1217. if (!sp)
  1218. continue;
  1219. emit_signal("transform_key_request", sp, "", sp->get_transform());
  1220. }
  1221. set_message(TTR("Animation Key Inserted."));
  1222. }
  1223. if (k->get_scancode() == KEY_SPACE) {
  1224. if (!k->is_pressed()) emit_signal("toggle_maximize_view", this);
  1225. }
  1226. }
  1227. }
  1228. void SpatialEditorViewport::scale_cursor_distance(real_t scale) {
  1229. // Prevents zero distance which would short-circuit any scaling
  1230. if (cursor.distance < ZOOM_MIN_DISTANCE)
  1231. cursor.distance = ZOOM_MIN_DISTANCE;
  1232. real_t prev_distance = cursor.distance;
  1233. cursor.distance *= scale;
  1234. if (cursor.distance < ZOOM_MIN_DISTANCE)
  1235. cursor.distance = ZOOM_MIN_DISTANCE;
  1236. if (is_freelook_active()) {
  1237. // In freelook mode, cursor reference is reversed so it needs to be adjusted
  1238. Vector3 forward = camera->get_transform().basis.xform(Vector3(0, 0, -1));
  1239. cursor.pos += (cursor.distance - prev_distance) * forward;
  1240. }
  1241. zoom_indicator_delay = ZOOM_INDICATOR_DELAY_S;
  1242. surface->update();
  1243. }
  1244. Point2i SpatialEditorViewport::_get_warped_mouse_motion(const Ref<InputEventMouseMotion> &p_ev_mouse_motion) const {
  1245. Point2i relative;
  1246. if (bool(EditorSettings::get_singleton()->get("editors/3d/warped_mouse_panning"))) {
  1247. relative = Input::get_singleton()->warp_mouse_motion(p_ev_mouse_motion, surface->get_global_rect());
  1248. } else {
  1249. relative = p_ev_mouse_motion->get_relative();
  1250. }
  1251. return relative;
  1252. }
  1253. void SpatialEditorViewport::_update_freelook(real_t delta) {
  1254. if (!is_freelook_active()) {
  1255. freelook_velocity = Vector3();
  1256. return;
  1257. }
  1258. Vector3 forward = camera->get_transform().basis.xform(Vector3(0, 0, -1));
  1259. Vector3 right = camera->get_transform().basis.xform(Vector3(1, 0, 0));
  1260. Vector3 up = camera->get_transform().basis.xform(Vector3(0, 1, 0));
  1261. int key_left = Object::cast_to<InputEventKey>(ED_GET_SHORTCUT("spatial_editor/freelook_left")->get_shortcut().ptr())->get_scancode();
  1262. int key_right = Object::cast_to<InputEventKey>(ED_GET_SHORTCUT("spatial_editor/freelook_right")->get_shortcut().ptr())->get_scancode();
  1263. int key_forward = Object::cast_to<InputEventKey>(ED_GET_SHORTCUT("spatial_editor/freelook_forward")->get_shortcut().ptr())->get_scancode();
  1264. int key_backwards = Object::cast_to<InputEventKey>(ED_GET_SHORTCUT("spatial_editor/freelook_backwards")->get_shortcut().ptr())->get_scancode();
  1265. int key_up = Object::cast_to<InputEventKey>(ED_GET_SHORTCUT("spatial_editor/freelook_up")->get_shortcut().ptr())->get_scancode();
  1266. int key_down = Object::cast_to<InputEventKey>(ED_GET_SHORTCUT("spatial_editor/freelook_down")->get_shortcut().ptr())->get_scancode();
  1267. int key_speed_modifier = Object::cast_to<InputEventKey>(ED_GET_SHORTCUT("spatial_editor/freelook_speed_modifier")->get_shortcut().ptr())->get_scancode();
  1268. Vector3 direction;
  1269. bool pressed = false;
  1270. bool speed_modifier = false;
  1271. const Input &input = *Input::get_singleton();
  1272. if (input.is_key_pressed(key_left)) {
  1273. direction -= right;
  1274. pressed = true;
  1275. }
  1276. if (input.is_key_pressed(key_right)) {
  1277. direction += right;
  1278. pressed = true;
  1279. }
  1280. if (input.is_key_pressed(key_forward)) {
  1281. direction += forward;
  1282. pressed = true;
  1283. }
  1284. if (input.is_key_pressed(key_backwards)) {
  1285. direction -= forward;
  1286. pressed = true;
  1287. }
  1288. if (input.is_key_pressed(key_up)) {
  1289. direction += up;
  1290. pressed = true;
  1291. }
  1292. if (input.is_key_pressed(key_down)) {
  1293. direction -= up;
  1294. pressed = true;
  1295. }
  1296. if (input.is_key_pressed(key_speed_modifier)) {
  1297. speed_modifier = true;
  1298. }
  1299. const EditorSettings &s = *EditorSettings::get_singleton();
  1300. real_t inertia = s.get("editors/3d/freelook_inertia");
  1301. if (inertia < 0)
  1302. inertia = 0;
  1303. const real_t base_speed = s.get("editors/3d/freelook_base_speed");
  1304. const real_t modifier_speed_factor = s.get("editors/3d/freelook_modifier_speed_factor");
  1305. real_t speed = base_speed * cursor.distance;
  1306. if (speed_modifier)
  1307. speed *= modifier_speed_factor;
  1308. Vector3 instant_velocity = direction * speed;
  1309. // Higher inertia should increase "lag" (lerp with factor between 0 and 1)
  1310. // Inertia of zero should produce instant movement (lerp with factor of 1)
  1311. // Takes reference of 60fps for units, so that inertia of 1 gives approximate lerp factor of 0.5
  1312. real_t factor = 1.0 / (1.0 + inertia * delta * 60.f);
  1313. freelook_velocity = freelook_velocity.linear_interpolate(instant_velocity, CLAMP(factor, 0, 1));
  1314. cursor.pos += freelook_velocity * delta;
  1315. }
  1316. void SpatialEditorViewport::set_message(String p_message, float p_time) {
  1317. message = p_message;
  1318. message_time = p_time;
  1319. }
  1320. void SpatialEditorViewport::_notification(int p_what) {
  1321. if (p_what == NOTIFICATION_VISIBILITY_CHANGED) {
  1322. bool visible = is_visible_in_tree();
  1323. set_process(visible);
  1324. if (visible)
  1325. _update_camera(0);
  1326. call_deferred("update_transform_gizmo_view");
  1327. }
  1328. if (p_what == NOTIFICATION_RESIZED) {
  1329. call_deferred("update_transform_gizmo_view");
  1330. }
  1331. if (p_what == NOTIFICATION_PROCESS) {
  1332. //force editr camera
  1333. /*
  1334. current_camera=get_root_node()->get_current_camera();
  1335. if (current_camera!=camera) {
  1336. }
  1337. */
  1338. real_t delta = get_tree()->get_idle_process_time();
  1339. if (zoom_indicator_delay > 0) {
  1340. zoom_indicator_delay -= delta;
  1341. if (zoom_indicator_delay <= 0) {
  1342. surface->update();
  1343. }
  1344. }
  1345. _update_freelook(delta);
  1346. _update_camera(get_process_delta_time());
  1347. Map<Node *, Object *> &selection = editor_selection->get_selection();
  1348. bool changed = false;
  1349. bool exist = false;
  1350. for (Map<Node *, Object *>::Element *E = selection.front(); E; E = E->next()) {
  1351. Spatial *sp = Object::cast_to<Spatial>(E->key());
  1352. if (!sp)
  1353. continue;
  1354. SpatialEditorSelectedItem *se = editor_selection->get_node_editor_data<SpatialEditorSelectedItem>(sp);
  1355. if (!se)
  1356. continue;
  1357. VisualInstance *vi = Object::cast_to<VisualInstance>(sp);
  1358. if (se->aabb.has_no_surface()) {
  1359. se->aabb = vi ? vi->get_aabb() : Rect3(Vector3(-0.2, -0.2, -0.2), Vector3(0.4, 0.4, 0.4));
  1360. }
  1361. Transform t = sp->get_global_transform();
  1362. t.translate(se->aabb.position);
  1363. // apply AABB scaling before item's global transform
  1364. Basis aabb_s;
  1365. aabb_s.scale(se->aabb.size);
  1366. t.basis = t.basis * aabb_s;
  1367. exist = true;
  1368. if (se->last_xform == t)
  1369. continue;
  1370. changed = true;
  1371. se->last_xform = t;
  1372. VisualServer::get_singleton()->instance_set_transform(se->sbox_instance, t);
  1373. }
  1374. if (changed || (spatial_editor->is_gizmo_visible() && !exist)) {
  1375. spatial_editor->update_transform_gizmo();
  1376. }
  1377. if (message_time > 0) {
  1378. if (message != last_message) {
  1379. surface->update();
  1380. last_message = message;
  1381. }
  1382. message_time -= get_fixed_process_delta_time();
  1383. if (message_time < 0)
  1384. surface->update();
  1385. }
  1386. //update shadow atlas if changed
  1387. int shadowmap_size = ProjectSettings::get_singleton()->get("rendering/quality/shadow_atlas/size");
  1388. int atlas_q0 = ProjectSettings::get_singleton()->get("rendering/quality/shadow_atlas/quadrant_0_subdiv");
  1389. int atlas_q1 = ProjectSettings::get_singleton()->get("rendering/quality/shadow_atlas/quadrant_1_subdiv");
  1390. int atlas_q2 = ProjectSettings::get_singleton()->get("rendering/quality/shadow_atlas/quadrant_2_subdiv");
  1391. int atlas_q3 = ProjectSettings::get_singleton()->get("rendering/quality/shadow_atlas/quadrant_3_subdiv");
  1392. viewport->set_shadow_atlas_size(shadowmap_size);
  1393. viewport->set_shadow_atlas_quadrant_subdiv(0, Viewport::ShadowAtlasQuadrantSubdiv(atlas_q0));
  1394. viewport->set_shadow_atlas_quadrant_subdiv(1, Viewport::ShadowAtlasQuadrantSubdiv(atlas_q1));
  1395. viewport->set_shadow_atlas_quadrant_subdiv(2, Viewport::ShadowAtlasQuadrantSubdiv(atlas_q2));
  1396. viewport->set_shadow_atlas_quadrant_subdiv(3, Viewport::ShadowAtlasQuadrantSubdiv(atlas_q3));
  1397. //update msaa if changed
  1398. int msaa_mode = ProjectSettings::get_singleton()->get("rendering/quality/filters/msaa");
  1399. viewport->set_msaa(Viewport::MSAA(msaa_mode));
  1400. bool hdr = ProjectSettings::get_singleton()->get("rendering/quality/depth/hdr");
  1401. viewport->set_hdr(hdr);
  1402. bool show_info = view_menu->get_popup()->is_item_checked(view_menu->get_popup()->get_item_index(VIEW_INFORMATION));
  1403. if (show_info != info->is_visible()) {
  1404. if (show_info)
  1405. info->show();
  1406. else
  1407. info->hide();
  1408. }
  1409. if (show_info) {
  1410. String text;
  1411. text += TTR("Objects Drawn") + ": " + itos(viewport->get_render_info(Viewport::RENDER_INFO_OBJECTS_IN_FRAME)) + "\n";
  1412. text += TTR("Material Changes") + ": " + itos(viewport->get_render_info(Viewport::RENDER_INFO_MATERIAL_CHANGES_IN_FRAME)) + "\n";
  1413. text += TTR("Shader Changes") + ": " + itos(viewport->get_render_info(Viewport::RENDER_INFO_SHADER_CHANGES_IN_FRAME)) + "\n";
  1414. text += TTR("Surface Changes") + ": " + itos(viewport->get_render_info(Viewport::RENDER_INFO_SURFACE_CHANGES_IN_FRAME)) + "\n";
  1415. text += TTR("Draw Calls") + ": " + itos(viewport->get_render_info(Viewport::RENDER_INFO_DRAW_CALLS_IN_FRAME)) + "\n";
  1416. text += TTR("Vertices") + ": " + itos(viewport->get_render_info(Viewport::RENDER_INFO_VERTICES_IN_FRAME));
  1417. if (info_label->get_text() != text || surface->get_size() != prev_size) {
  1418. info_label->set_text(text);
  1419. Size2 ms = info->get_minimum_size();
  1420. info->set_position(surface->get_size() - ms - Vector2(20, 20) * EDSCALE);
  1421. }
  1422. }
  1423. prev_size = surface->get_size();
  1424. }
  1425. if (p_what == NOTIFICATION_ENTER_TREE) {
  1426. surface->connect("draw", this, "_draw");
  1427. surface->connect("gui_input", this, "_sinput");
  1428. surface->connect("mouse_entered", this, "_smouseenter");
  1429. info->add_style_override("panel", get_stylebox("panel", "Panel"));
  1430. preview_camera->set_icon(get_icon("Camera", "EditorIcons"));
  1431. _init_gizmo_instance(index);
  1432. }
  1433. if (p_what == NOTIFICATION_EXIT_TREE) {
  1434. _finish_gizmo_instances();
  1435. }
  1436. if (p_what == NOTIFICATION_MOUSE_ENTER) {
  1437. }
  1438. if (p_what == NOTIFICATION_DRAW) {
  1439. }
  1440. }
  1441. // TODO That should be part of the drawing API...
  1442. static void stroke_rect(CanvasItem *ci, Rect2 rect, Color color, real_t width = 1.0) {
  1443. // a---b
  1444. // | |
  1445. // c---d
  1446. Vector2 a(rect.position);
  1447. Vector2 b(rect.position.x + rect.size.x, rect.position.y);
  1448. Vector2 c(rect.position.x, rect.position.y + rect.size.y);
  1449. Vector2 d(rect.position + rect.size);
  1450. ci->draw_line(a, b, color, width);
  1451. ci->draw_line(b, d, color, width);
  1452. ci->draw_line(d, c, color, width);
  1453. ci->draw_line(c, a, color, width);
  1454. }
  1455. void SpatialEditorViewport::_draw() {
  1456. if (surface->has_focus()) {
  1457. Size2 size = surface->get_size();
  1458. Rect2 r = Rect2(Point2(), size);
  1459. get_stylebox("Focus", "EditorStyles")->draw(surface->get_canvas_item(), r);
  1460. }
  1461. RID ci = surface->get_canvas_item();
  1462. if (cursor.region_select) {
  1463. VisualServer::get_singleton()->canvas_item_add_rect(ci, Rect2(cursor.region_begin, cursor.region_end - cursor.region_begin), Color(0.7, 0.7, 1.0, 0.3));
  1464. }
  1465. if (message_time > 0) {
  1466. Ref<Font> font = get_font("font", "Label");
  1467. Point2 msgpos = Point2(5, get_size().y - 20);
  1468. font->draw(ci, msgpos + Point2(1, 1), message, Color(0, 0, 0, 0.8));
  1469. font->draw(ci, msgpos + Point2(-1, -1), message, Color(0, 0, 0, 0.8));
  1470. font->draw(ci, msgpos, message, Color(1, 1, 1, 1));
  1471. }
  1472. if (_edit.mode == TRANSFORM_ROTATE) {
  1473. Point2 center = _point_to_screen(_edit.center);
  1474. VisualServer::get_singleton()->canvas_item_add_line(ci, _edit.mouse_pos, center, Color(0.4, 0.7, 1.0, 0.8));
  1475. }
  1476. if (previewing) {
  1477. Size2 ss = Size2(ProjectSettings::get_singleton()->get("display/window/size/width"), ProjectSettings::get_singleton()->get("display/window/size/height"));
  1478. float aspect = ss.aspect();
  1479. Size2 s = get_size();
  1480. Rect2 draw_rect;
  1481. switch (previewing->get_keep_aspect_mode()) {
  1482. case Camera::KEEP_WIDTH: {
  1483. draw_rect.size = Size2(s.width, s.width / aspect);
  1484. draw_rect.position.x = 0;
  1485. draw_rect.position.y = (s.height - draw_rect.size.y) * 0.5;
  1486. } break;
  1487. case Camera::KEEP_HEIGHT: {
  1488. draw_rect.size = Size2(s.height * aspect, s.height);
  1489. draw_rect.position.y = 0;
  1490. draw_rect.position.x = (s.width - draw_rect.size.x) * 0.5;
  1491. } break;
  1492. }
  1493. draw_rect = Rect2(Vector2(), s).clip(draw_rect);
  1494. stroke_rect(surface, draw_rect, Color(0.6, 0.6, 0.1, 0.5), 2.0);
  1495. } else {
  1496. if (zoom_indicator_delay > 0.0) {
  1497. // Show indicative zoom factor
  1498. real_t min_distance = ZOOM_MIN_DISTANCE; // TODO Why not pick znear to limit zoom?
  1499. real_t max_distance = camera->get_zfar();
  1500. real_t scale_length = (max_distance - min_distance);
  1501. if (Math::abs(scale_length) > CMP_EPSILON) {
  1502. real_t logscale_t = 1.0 - Math::log(1 + cursor.distance - min_distance) / Math::log(1 + scale_length);
  1503. // There is no real maximum distance so that factor can become negative,
  1504. // Let's make it look asymptotic instead (will decrease slower and slower).
  1505. if (logscale_t < 0.25)
  1506. logscale_t = 0.25 * Math::exp(4.0 * logscale_t - 1.0);
  1507. Vector2 surface_size = surface->get_size();
  1508. real_t h = surface_size.y / 2.0;
  1509. real_t y = (surface_size.y - h) / 2.0;
  1510. Rect2 r(10, y, 6, h);
  1511. real_t sy = r.size.y * logscale_t;
  1512. surface->draw_rect(r, Color(1, 1, 1, 0.2));
  1513. surface->draw_rect(Rect2(r.position.x, r.position.y + r.size.y - sy, r.size.x, sy), Color(1, 1, 1, 0.6));
  1514. stroke_rect(surface, r.grow(1), Color(0, 0, 0, 0.7));
  1515. }
  1516. }
  1517. }
  1518. }
  1519. void SpatialEditorViewport::_menu_option(int p_option) {
  1520. switch (p_option) {
  1521. case VIEW_TOP: {
  1522. cursor.x_rot = Math_PI / 2.0;
  1523. cursor.y_rot = 0;
  1524. name = TTR("Top");
  1525. _update_name();
  1526. } break;
  1527. case VIEW_BOTTOM: {
  1528. cursor.x_rot = -Math_PI / 2.0;
  1529. cursor.y_rot = 0;
  1530. name = TTR("Bottom");
  1531. _update_name();
  1532. } break;
  1533. case VIEW_LEFT: {
  1534. cursor.y_rot = Math_PI / 2.0;
  1535. cursor.x_rot = 0;
  1536. name = TTR("Left");
  1537. _update_name();
  1538. } break;
  1539. case VIEW_RIGHT: {
  1540. cursor.y_rot = -Math_PI / 2.0;
  1541. cursor.x_rot = 0;
  1542. name = TTR("Right");
  1543. _update_name();
  1544. } break;
  1545. case VIEW_FRONT: {
  1546. cursor.y_rot = 0;
  1547. cursor.x_rot = 0;
  1548. name = TTR("Front");
  1549. _update_name();
  1550. } break;
  1551. case VIEW_REAR: {
  1552. cursor.y_rot = Math_PI;
  1553. cursor.x_rot = 0;
  1554. name = TTR("Rear");
  1555. _update_name();
  1556. } break;
  1557. case VIEW_CENTER_TO_ORIGIN: {
  1558. cursor.pos = Vector3(0, 0, 0);
  1559. } break;
  1560. case VIEW_CENTER_TO_SELECTION: {
  1561. focus_selection();
  1562. } break;
  1563. case VIEW_ALIGN_SELECTION_WITH_VIEW: {
  1564. if (!get_selected_count())
  1565. break;
  1566. Transform camera_transform = camera->get_global_transform();
  1567. List<Node *> &selection = editor_selection->get_selected_node_list();
  1568. undo_redo->create_action(TTR("Align with view"));
  1569. for (List<Node *>::Element *E = selection.front(); E; E = E->next()) {
  1570. Spatial *sp = Object::cast_to<Spatial>(E->get());
  1571. if (!sp)
  1572. continue;
  1573. SpatialEditorSelectedItem *se = editor_selection->get_node_editor_data<SpatialEditorSelectedItem>(sp);
  1574. if (!se)
  1575. continue;
  1576. Transform xform = camera_transform;
  1577. xform.scale_basis(sp->get_scale());
  1578. undo_redo->add_do_method(sp, "set_global_transform", xform);
  1579. undo_redo->add_undo_method(sp, "set_global_transform", sp->get_global_transform());
  1580. }
  1581. undo_redo->commit_action();
  1582. } break;
  1583. case VIEW_ENVIRONMENT: {
  1584. int idx = view_menu->get_popup()->get_item_index(VIEW_ENVIRONMENT);
  1585. bool current = view_menu->get_popup()->is_item_checked(idx);
  1586. current = !current;
  1587. if (current) {
  1588. camera->set_environment(RES());
  1589. } else {
  1590. camera->set_environment(SpatialEditor::get_singleton()->get_viewport_environment());
  1591. }
  1592. view_menu->get_popup()->set_item_checked(idx, current);
  1593. } break;
  1594. case VIEW_PERSPECTIVE: {
  1595. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(VIEW_PERSPECTIVE), true);
  1596. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(VIEW_ORTHOGONAL), false);
  1597. orthogonal = false;
  1598. call_deferred("update_transform_gizmo_view");
  1599. _update_name();
  1600. } break;
  1601. case VIEW_ORTHOGONAL: {
  1602. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(VIEW_PERSPECTIVE), false);
  1603. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(VIEW_ORTHOGONAL), true);
  1604. orthogonal = true;
  1605. call_deferred("update_transform_gizmo_view");
  1606. _update_name();
  1607. } break;
  1608. case VIEW_AUDIO_LISTENER: {
  1609. int idx = view_menu->get_popup()->get_item_index(VIEW_AUDIO_LISTENER);
  1610. bool current = view_menu->get_popup()->is_item_checked(idx);
  1611. current = !current;
  1612. viewport->set_as_audio_listener(current);
  1613. view_menu->get_popup()->set_item_checked(idx, current);
  1614. } break;
  1615. case VIEW_AUDIO_DOPPLER: {
  1616. int idx = view_menu->get_popup()->get_item_index(VIEW_AUDIO_DOPPLER);
  1617. bool current = view_menu->get_popup()->is_item_checked(idx);
  1618. current = !current;
  1619. camera->set_doppler_tracking(current ? Camera::DOPPLER_TRACKING_IDLE_STEP : Camera::DOPPLER_TRACKING_DISABLED);
  1620. view_menu->get_popup()->set_item_checked(idx, current);
  1621. } break;
  1622. case VIEW_GIZMOS: {
  1623. int idx = view_menu->get_popup()->get_item_index(VIEW_GIZMOS);
  1624. bool current = view_menu->get_popup()->is_item_checked(idx);
  1625. current = !current;
  1626. if (current)
  1627. camera->set_cull_mask(((1 << 20) - 1) | (1 << (GIZMO_BASE_LAYER + index)) | (1 << GIZMO_EDIT_LAYER) | (1 << GIZMO_GRID_LAYER));
  1628. else
  1629. camera->set_cull_mask(((1 << 20) - 1) | (1 << (GIZMO_BASE_LAYER + index)) | (1 << GIZMO_GRID_LAYER));
  1630. view_menu->get_popup()->set_item_checked(idx, current);
  1631. } break;
  1632. case VIEW_INFORMATION: {
  1633. int idx = view_menu->get_popup()->get_item_index(VIEW_INFORMATION);
  1634. bool current = view_menu->get_popup()->is_item_checked(idx);
  1635. view_menu->get_popup()->set_item_checked(idx, !current);
  1636. } break;
  1637. case VIEW_DISPLAY_NORMAL: {
  1638. viewport->set_debug_draw(Viewport::DEBUG_DRAW_DISABLED);
  1639. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(VIEW_DISPLAY_NORMAL), true);
  1640. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(VIEW_DISPLAY_WIREFRAME), false);
  1641. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(VIEW_DISPLAY_OVERDRAW), false);
  1642. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(VIEW_DISPLAY_SHADELESS), false);
  1643. } break;
  1644. case VIEW_DISPLAY_WIREFRAME: {
  1645. viewport->set_debug_draw(Viewport::DEBUG_DRAW_WIREFRAME);
  1646. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(VIEW_DISPLAY_NORMAL), false);
  1647. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(VIEW_DISPLAY_WIREFRAME), true);
  1648. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(VIEW_DISPLAY_OVERDRAW), false);
  1649. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(VIEW_DISPLAY_SHADELESS), false);
  1650. } break;
  1651. case VIEW_DISPLAY_OVERDRAW: {
  1652. viewport->set_debug_draw(Viewport::DEBUG_DRAW_OVERDRAW);
  1653. VisualServer::get_singleton()->scenario_set_debug(get_tree()->get_root()->get_world()->get_scenario(), VisualServer::SCENARIO_DEBUG_OVERDRAW);
  1654. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(VIEW_DISPLAY_NORMAL), false);
  1655. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(VIEW_DISPLAY_WIREFRAME), false);
  1656. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(VIEW_DISPLAY_OVERDRAW), true);
  1657. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(VIEW_DISPLAY_SHADELESS), false);
  1658. } break;
  1659. case VIEW_DISPLAY_SHADELESS: {
  1660. viewport->set_debug_draw(Viewport::DEBUG_DRAW_UNSHADED);
  1661. VisualServer::get_singleton()->scenario_set_debug(get_tree()->get_root()->get_world()->get_scenario(), VisualServer::SCENARIO_DEBUG_SHADELESS);
  1662. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(VIEW_DISPLAY_NORMAL), false);
  1663. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(VIEW_DISPLAY_WIREFRAME), false);
  1664. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(VIEW_DISPLAY_OVERDRAW), false);
  1665. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(VIEW_DISPLAY_SHADELESS), true);
  1666. } break;
  1667. }
  1668. }
  1669. void SpatialEditorViewport::_preview_exited_scene() {
  1670. preview_camera->set_pressed(false);
  1671. _toggle_camera_preview(false);
  1672. view_menu->show();
  1673. }
  1674. void SpatialEditorViewport::_init_gizmo_instance(int p_idx) {
  1675. uint32_t layer = 1 << (GIZMO_BASE_LAYER + p_idx); //|(1<<GIZMO_GRID_LAYER);
  1676. for (int i = 0; i < 3; i++) {
  1677. move_gizmo_instance[i] = VS::get_singleton()->instance_create();
  1678. VS::get_singleton()->instance_set_base(move_gizmo_instance[i], spatial_editor->get_move_gizmo(i)->get_rid());
  1679. VS::get_singleton()->instance_set_scenario(move_gizmo_instance[i], get_tree()->get_root()->get_world()->get_scenario());
  1680. VS::get_singleton()->instance_set_visible(move_gizmo_instance[i], false);
  1681. //VS::get_singleton()->instance_geometry_set_flag(move_gizmo_instance[i],VS::INSTANCE_FLAG_DEPH_SCALE,true);
  1682. VS::get_singleton()->instance_geometry_set_cast_shadows_setting(move_gizmo_instance[i], VS::SHADOW_CASTING_SETTING_OFF);
  1683. VS::get_singleton()->instance_set_layer_mask(move_gizmo_instance[i], layer);
  1684. move_plane_gizmo_instance[i] = VS::get_singleton()->instance_create();
  1685. VS::get_singleton()->instance_set_base(move_plane_gizmo_instance[i], spatial_editor->get_move_plane_gizmo(i)->get_rid());
  1686. VS::get_singleton()->instance_set_scenario(move_plane_gizmo_instance[i], get_tree()->get_root()->get_world()->get_scenario());
  1687. VS::get_singleton()->instance_set_visible(move_plane_gizmo_instance[i], false);
  1688. //VS::get_singleton()->instance_geometry_set_flag(move_plane_gizmo_instance[i],VS::INSTANCE_FLAG_DEPH_SCALE,true);
  1689. VS::get_singleton()->instance_geometry_set_cast_shadows_setting(move_plane_gizmo_instance[i], VS::SHADOW_CASTING_SETTING_OFF);
  1690. VS::get_singleton()->instance_set_layer_mask(move_plane_gizmo_instance[i], layer);
  1691. rotate_gizmo_instance[i] = VS::get_singleton()->instance_create();
  1692. VS::get_singleton()->instance_set_base(rotate_gizmo_instance[i], spatial_editor->get_rotate_gizmo(i)->get_rid());
  1693. VS::get_singleton()->instance_set_scenario(rotate_gizmo_instance[i], get_tree()->get_root()->get_world()->get_scenario());
  1694. VS::get_singleton()->instance_set_visible(rotate_gizmo_instance[i], false);
  1695. //VS::get_singleton()->instance_geometry_set_flag(rotate_gizmo_instance[i],VS::INSTANCE_FLAG_DEPH_SCALE,true);
  1696. VS::get_singleton()->instance_geometry_set_cast_shadows_setting(rotate_gizmo_instance[i], VS::SHADOW_CASTING_SETTING_OFF);
  1697. VS::get_singleton()->instance_set_layer_mask(rotate_gizmo_instance[i], layer);
  1698. }
  1699. }
  1700. void SpatialEditorViewport::_finish_gizmo_instances() {
  1701. for (int i = 0; i < 3; i++) {
  1702. VS::get_singleton()->free(move_gizmo_instance[i]);
  1703. VS::get_singleton()->free(move_plane_gizmo_instance[i]);
  1704. VS::get_singleton()->free(rotate_gizmo_instance[i]);
  1705. }
  1706. }
  1707. void SpatialEditorViewport::_toggle_camera_preview(bool p_activate) {
  1708. ERR_FAIL_COND(p_activate && !preview);
  1709. ERR_FAIL_COND(!p_activate && !previewing);
  1710. if (!p_activate) {
  1711. previewing->disconnect("tree_exited", this, "_preview_exited_scene");
  1712. previewing = NULL;
  1713. VS::get_singleton()->viewport_attach_camera(viewport->get_viewport_rid(), camera->get_camera()); //restore
  1714. if (!preview)
  1715. preview_camera->hide();
  1716. view_menu->show();
  1717. surface->update();
  1718. } else {
  1719. previewing = preview;
  1720. previewing->connect("tree_exited", this, "_preview_exited_scene");
  1721. VS::get_singleton()->viewport_attach_camera(viewport->get_viewport_rid(), preview->get_camera()); //replace
  1722. view_menu->hide();
  1723. surface->update();
  1724. }
  1725. }
  1726. void SpatialEditorViewport::_selection_result_pressed(int p_result) {
  1727. if (selection_results.size() <= p_result)
  1728. return;
  1729. clicked = selection_results[p_result].item->get_instance_id();
  1730. if (clicked) {
  1731. _select_clicked(clicked_wants_append, true);
  1732. clicked = 0;
  1733. }
  1734. }
  1735. void SpatialEditorViewport::_selection_menu_hide() {
  1736. selection_results.clear();
  1737. selection_menu->clear();
  1738. selection_menu->set_size(Vector2(0, 0));
  1739. }
  1740. void SpatialEditorViewport::set_can_preview(Camera *p_preview) {
  1741. preview = p_preview;
  1742. if (!preview_camera->is_pressed()) {
  1743. if (p_preview) {
  1744. preview_camera->show();
  1745. } else {
  1746. preview_camera->hide();
  1747. }
  1748. }
  1749. }
  1750. void SpatialEditorViewport::update_transform_gizmo_view() {
  1751. if (!is_visible_in_tree())
  1752. return;
  1753. Transform xform = spatial_editor->get_gizmo_transform();
  1754. Transform camera_xform = camera->get_transform();
  1755. Vector3 camz = -camera_xform.get_basis().get_axis(2).normalized();
  1756. Vector3 camy = -camera_xform.get_basis().get_axis(1).normalized();
  1757. Plane p(camera_xform.origin, camz);
  1758. float gizmo_d = Math::abs(p.distance_to(xform.origin));
  1759. float d0 = camera->unproject_position(camera_xform.origin + camz * gizmo_d).y;
  1760. float d1 = camera->unproject_position(camera_xform.origin + camz * gizmo_d + camy).y;
  1761. float dd = Math::abs(d0 - d1);
  1762. if (dd == 0)
  1763. dd = 0.0001;
  1764. float gsize = EditorSettings::get_singleton()->get("editors/3d/manipulator_gizmo_size");
  1765. gizmo_scale = (gsize / Math::abs(dd));
  1766. Vector3 scale = Vector3(1, 1, 1) * gizmo_scale;
  1767. xform.basis.scale(scale);
  1768. //xform.basis.scale(GIZMO_SCALE_DEFAULT*Vector3(1,1,1));
  1769. for (int i = 0; i < 3; i++) {
  1770. VisualServer::get_singleton()->instance_set_transform(move_gizmo_instance[i], xform);
  1771. VisualServer::get_singleton()->instance_set_visible(move_gizmo_instance[i], spatial_editor->is_gizmo_visible() && (spatial_editor->get_tool_mode() == SpatialEditor::TOOL_MODE_SELECT || spatial_editor->get_tool_mode() == SpatialEditor::TOOL_MODE_MOVE));
  1772. VisualServer::get_singleton()->instance_set_transform(move_plane_gizmo_instance[i], xform);
  1773. VisualServer::get_singleton()->instance_set_visible(move_plane_gizmo_instance[i], spatial_editor->is_gizmo_visible() && (spatial_editor->get_tool_mode() == SpatialEditor::TOOL_MODE_SELECT || spatial_editor->get_tool_mode() == SpatialEditor::TOOL_MODE_MOVE));
  1774. VisualServer::get_singleton()->instance_set_transform(rotate_gizmo_instance[i], xform);
  1775. VisualServer::get_singleton()->instance_set_visible(rotate_gizmo_instance[i], spatial_editor->is_gizmo_visible() && (spatial_editor->get_tool_mode() == SpatialEditor::TOOL_MODE_SELECT || spatial_editor->get_tool_mode() == SpatialEditor::TOOL_MODE_ROTATE));
  1776. }
  1777. }
  1778. void SpatialEditorViewport::set_state(const Dictionary &p_state) {
  1779. cursor.pos = p_state["pos"];
  1780. cursor.x_rot = p_state["x_rot"];
  1781. cursor.y_rot = p_state["y_rot"];
  1782. cursor.distance = p_state["distance"];
  1783. bool env = p_state["use_environment"];
  1784. bool orth = p_state["use_orthogonal"];
  1785. if (orth)
  1786. _menu_option(VIEW_ORTHOGONAL);
  1787. else
  1788. _menu_option(VIEW_PERSPECTIVE);
  1789. if (env != camera->get_environment().is_valid())
  1790. _menu_option(VIEW_ENVIRONMENT);
  1791. if (p_state.has("listener")) {
  1792. bool listener = p_state["listener"];
  1793. int idx = view_menu->get_popup()->get_item_index(VIEW_AUDIO_LISTENER);
  1794. viewport->set_as_audio_listener(listener);
  1795. view_menu->get_popup()->set_item_checked(idx, listener);
  1796. }
  1797. if (p_state.has("doppler")) {
  1798. bool doppler = p_state["doppler"];
  1799. int idx = view_menu->get_popup()->get_item_index(VIEW_AUDIO_DOPPLER);
  1800. camera->set_doppler_tracking(doppler ? Camera::DOPPLER_TRACKING_IDLE_STEP : Camera::DOPPLER_TRACKING_DISABLED);
  1801. view_menu->get_popup()->set_item_checked(idx, doppler);
  1802. }
  1803. if (p_state.has("previewing")) {
  1804. Node *pv = EditorNode::get_singleton()->get_edited_scene()->get_node(p_state["previewing"]);
  1805. if (Object::cast_to<Camera>(pv)) {
  1806. previewing = Object::cast_to<Camera>(pv);
  1807. previewing->connect("tree_exited", this, "_preview_exited_scene");
  1808. VS::get_singleton()->viewport_attach_camera(viewport->get_viewport_rid(), previewing->get_camera()); //replace
  1809. view_menu->hide();
  1810. surface->update();
  1811. preview_camera->set_pressed(true);
  1812. preview_camera->show();
  1813. }
  1814. }
  1815. }
  1816. Dictionary SpatialEditorViewport::get_state() const {
  1817. Dictionary d;
  1818. d["pos"] = cursor.pos;
  1819. d["x_rot"] = cursor.x_rot;
  1820. d["y_rot"] = cursor.y_rot;
  1821. d["distance"] = cursor.distance;
  1822. d["use_environment"] = camera->get_environment().is_valid();
  1823. d["use_orthogonal"] = camera->get_projection() == Camera::PROJECTION_ORTHOGONAL;
  1824. d["listener"] = viewport->is_audio_listener();
  1825. if (previewing) {
  1826. d["previewing"] = EditorNode::get_singleton()->get_edited_scene()->get_path_to(previewing);
  1827. }
  1828. return d;
  1829. }
  1830. void SpatialEditorViewport::_bind_methods() {
  1831. ClassDB::bind_method(D_METHOD("_draw"), &SpatialEditorViewport::_draw);
  1832. ClassDB::bind_method(D_METHOD("_smouseenter"), &SpatialEditorViewport::_smouseenter);
  1833. ClassDB::bind_method(D_METHOD("_sinput"), &SpatialEditorViewport::_sinput);
  1834. ClassDB::bind_method(D_METHOD("_menu_option"), &SpatialEditorViewport::_menu_option);
  1835. ClassDB::bind_method(D_METHOD("_toggle_camera_preview"), &SpatialEditorViewport::_toggle_camera_preview);
  1836. ClassDB::bind_method(D_METHOD("_preview_exited_scene"), &SpatialEditorViewport::_preview_exited_scene);
  1837. ClassDB::bind_method(D_METHOD("update_transform_gizmo_view"), &SpatialEditorViewport::update_transform_gizmo_view);
  1838. ClassDB::bind_method(D_METHOD("_selection_result_pressed"), &SpatialEditorViewport::_selection_result_pressed);
  1839. ClassDB::bind_method(D_METHOD("_selection_menu_hide"), &SpatialEditorViewport::_selection_menu_hide);
  1840. ADD_SIGNAL(MethodInfo("toggle_maximize_view", PropertyInfo(Variant::OBJECT, "viewport")));
  1841. }
  1842. void SpatialEditorViewport::reset() {
  1843. orthogonal = false;
  1844. message_time = 0;
  1845. message = "";
  1846. last_message = "";
  1847. name = "";
  1848. cursor.x_rot = 0.5;
  1849. cursor.y_rot = 0.5;
  1850. cursor.distance = 4;
  1851. cursor.region_select = false;
  1852. _update_name();
  1853. }
  1854. void SpatialEditorViewport::focus_selection() {
  1855. if (!get_selected_count())
  1856. return;
  1857. Vector3 center;
  1858. int count = 0;
  1859. List<Node *> &selection = editor_selection->get_selected_node_list();
  1860. for (List<Node *>::Element *E = selection.front(); E; E = E->next()) {
  1861. Spatial *sp = Object::cast_to<Spatial>(E->get());
  1862. if (!sp)
  1863. continue;
  1864. SpatialEditorSelectedItem *se = editor_selection->get_node_editor_data<SpatialEditorSelectedItem>(sp);
  1865. if (!se)
  1866. continue;
  1867. center += sp->get_global_transform().origin;
  1868. count++;
  1869. }
  1870. if (count != 0) {
  1871. center /= float(count);
  1872. }
  1873. cursor.pos = center;
  1874. }
  1875. SpatialEditorViewport::SpatialEditorViewport(SpatialEditor *p_spatial_editor, EditorNode *p_editor, int p_index) {
  1876. _edit.mode = TRANSFORM_NONE;
  1877. _edit.plane = TRANSFORM_VIEW;
  1878. _edit.edited_gizmo = 0;
  1879. _edit.snap = 1;
  1880. _edit.gizmo_handle = 0;
  1881. index = p_index;
  1882. editor = p_editor;
  1883. editor_selection = editor->get_editor_selection();
  1884. undo_redo = editor->get_undo_redo();
  1885. clicked = 0;
  1886. clicked_includes_current = false;
  1887. orthogonal = false;
  1888. message_time = 0;
  1889. zoom_indicator_delay = 0.0;
  1890. spatial_editor = p_spatial_editor;
  1891. ViewportContainer *c = memnew(ViewportContainer);
  1892. c->set_stretch(true);
  1893. add_child(c);
  1894. c->set_area_as_parent_rect();
  1895. viewport = memnew(Viewport);
  1896. viewport->set_disable_input(true);
  1897. c->add_child(viewport);
  1898. surface = memnew(Control);
  1899. add_child(surface);
  1900. surface->set_area_as_parent_rect();
  1901. surface->set_clip_contents(true);
  1902. camera = memnew(Camera);
  1903. camera->set_disable_gizmo(true);
  1904. camera->set_cull_mask(((1 << 20) - 1) | (1 << (GIZMO_BASE_LAYER + p_index)) | (1 << GIZMO_EDIT_LAYER) | (1 << GIZMO_GRID_LAYER));
  1905. //camera->set_environment(SpatialEditor::get_singleton()->get_viewport_environment());
  1906. viewport->add_child(camera);
  1907. camera->make_current();
  1908. surface->set_focus_mode(FOCUS_ALL);
  1909. view_menu = memnew(MenuButton);
  1910. surface->add_child(view_menu);
  1911. view_menu->set_position(Point2(4, 4) * EDSCALE);
  1912. view_menu->set_self_modulate(Color(1, 1, 1, 0.5));
  1913. view_menu->get_popup()->add_shortcut(ED_GET_SHORTCUT("spatial_editor/top_view"), VIEW_TOP);
  1914. view_menu->get_popup()->add_shortcut(ED_GET_SHORTCUT("spatial_editor/bottom_view"), VIEW_BOTTOM);
  1915. view_menu->get_popup()->add_shortcut(ED_GET_SHORTCUT("spatial_editor/left_view"), VIEW_LEFT);
  1916. view_menu->get_popup()->add_shortcut(ED_GET_SHORTCUT("spatial_editor/right_view"), VIEW_RIGHT);
  1917. view_menu->get_popup()->add_shortcut(ED_GET_SHORTCUT("spatial_editor/front_view"), VIEW_FRONT);
  1918. view_menu->get_popup()->add_shortcut(ED_GET_SHORTCUT("spatial_editor/rear_view"), VIEW_REAR);
  1919. view_menu->get_popup()->add_separator();
  1920. view_menu->get_popup()->add_check_item(TTR("Perspective") + " (" + ED_GET_SHORTCUT("spatial_editor/switch_perspective_orthogonal")->get_as_text() + ")", VIEW_PERSPECTIVE);
  1921. view_menu->get_popup()->add_check_item(TTR("Orthogonal") + " (" + ED_GET_SHORTCUT("spatial_editor/switch_perspective_orthogonal")->get_as_text() + ")", VIEW_ORTHOGONAL);
  1922. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(VIEW_PERSPECTIVE), true);
  1923. view_menu->get_popup()->add_separator();
  1924. view_menu->get_popup()->add_check_shortcut(ED_SHORTCUT("spatial_editor/view_display_normal", TTR("Display Normal")), VIEW_DISPLAY_NORMAL);
  1925. view_menu->get_popup()->add_check_shortcut(ED_SHORTCUT("spatial_editor/view_display_wireframe", TTR("Display Wireframe")), VIEW_DISPLAY_WIREFRAME);
  1926. view_menu->get_popup()->add_check_shortcut(ED_SHORTCUT("spatial_editor/view_display_overdraw", TTR("Display Overdraw")), VIEW_DISPLAY_OVERDRAW);
  1927. view_menu->get_popup()->add_check_shortcut(ED_SHORTCUT("spatial_editor/view_display_unshaded", TTR("Display Unshaded")), VIEW_DISPLAY_SHADELESS);
  1928. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(VIEW_DISPLAY_NORMAL), true);
  1929. view_menu->get_popup()->add_separator();
  1930. view_menu->get_popup()->add_check_shortcut(ED_SHORTCUT("spatial_editor/view_environment", TTR("View Environment")), VIEW_ENVIRONMENT);
  1931. view_menu->get_popup()->add_check_shortcut(ED_SHORTCUT("spatial_editor/view_gizmos", TTR("View Gizmos")), VIEW_GIZMOS);
  1932. view_menu->get_popup()->add_check_shortcut(ED_SHORTCUT("spatial_editor/view_information", TTR("View Information")), VIEW_INFORMATION);
  1933. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(VIEW_ENVIRONMENT), true);
  1934. view_menu->get_popup()->add_separator();
  1935. view_menu->get_popup()->add_check_shortcut(ED_SHORTCUT("spatial_editor/view_audio_listener", TTR("Audio Listener")), VIEW_AUDIO_LISTENER);
  1936. view_menu->get_popup()->add_check_shortcut(ED_SHORTCUT("spatial_editor/view_audio_doppler", TTR("Doppler Enable")), VIEW_AUDIO_DOPPLER);
  1937. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(VIEW_GIZMOS), true);
  1938. view_menu->get_popup()->add_separator();
  1939. view_menu->get_popup()->add_shortcut(ED_GET_SHORTCUT("spatial_editor/focus_origin"), VIEW_CENTER_TO_ORIGIN);
  1940. view_menu->get_popup()->add_shortcut(ED_GET_SHORTCUT("spatial_editor/focus_selection"), VIEW_CENTER_TO_SELECTION);
  1941. view_menu->get_popup()->add_shortcut(ED_GET_SHORTCUT("spatial_editor/align_selection_with_view"), VIEW_ALIGN_SELECTION_WITH_VIEW);
  1942. view_menu->get_popup()->connect("id_pressed", this, "_menu_option");
  1943. ED_SHORTCUT("spatial_editor/freelook_left", TTR("Freelook Left"), KEY_A);
  1944. ED_SHORTCUT("spatial_editor/freelook_right", TTR("Freelook Right"), KEY_D);
  1945. ED_SHORTCUT("spatial_editor/freelook_forward", TTR("Freelook Forward"), KEY_W);
  1946. ED_SHORTCUT("spatial_editor/freelook_backwards", TTR("Freelook Backwards"), KEY_S);
  1947. ED_SHORTCUT("spatial_editor/freelook_up", TTR("Freelook Up"), KEY_Q);
  1948. ED_SHORTCUT("spatial_editor/freelook_down", TTR("Freelook Down"), KEY_E);
  1949. ED_SHORTCUT("spatial_editor/freelook_speed_modifier", TTR("Freelook Speed Modifier"), KEY_SHIFT);
  1950. preview_camera = memnew(Button);
  1951. preview_camera->set_toggle_mode(true);
  1952. preview_camera->set_anchor_and_margin(MARGIN_LEFT, ANCHOR_END, -90 * EDSCALE);
  1953. preview_camera->set_anchor_and_margin(MARGIN_TOP, ANCHOR_BEGIN, 10 * EDSCALE);
  1954. preview_camera->set_text(TTR("preview"));
  1955. surface->add_child(preview_camera);
  1956. preview_camera->hide();
  1957. preview_camera->connect("toggled", this, "_toggle_camera_preview");
  1958. previewing = NULL;
  1959. preview = NULL;
  1960. gizmo_scale = 1.0;
  1961. info = memnew(PanelContainer);
  1962. info->set_self_modulate(Color(1, 1, 1, 0.4));
  1963. surface->add_child(info);
  1964. info_label = memnew(Label);
  1965. info->add_child(info_label);
  1966. info->hide();
  1967. freelook_active = false;
  1968. selection_menu = memnew(PopupMenu);
  1969. add_child(selection_menu);
  1970. selection_menu->set_custom_minimum_size(Size2(100, 0) * EDSCALE);
  1971. selection_menu->connect("id_pressed", this, "_selection_result_pressed");
  1972. selection_menu->connect("popup_hide", this, "_selection_menu_hide");
  1973. if (p_index == 0) {
  1974. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(VIEW_AUDIO_LISTENER), true);
  1975. viewport->set_as_audio_listener(true);
  1976. }
  1977. name = "";
  1978. _update_name();
  1979. EditorSettings::get_singleton()->connect("settings_changed", this, "update_transform_gizmo_view");
  1980. }
  1981. //////////////////////////////////////////////////////////////
  1982. void SpatialEditorViewportContainer::_gui_input(const Ref<InputEvent> &p_event) {
  1983. Ref<InputEventMouseButton> mb = p_event;
  1984. if (mb.is_valid() && mb->is_pressed() && mb->get_button_index() == BUTTON_LEFT) {
  1985. Vector2 size = get_size();
  1986. int h_sep = get_constant("separation", "HSplitContainer");
  1987. int v_sep = get_constant("separation", "VSplitContainer");
  1988. int mid_w = size.width * ratio_h;
  1989. int mid_h = size.height * ratio_v;
  1990. dragging_h = mb->get_position().x > (mid_w - h_sep / 2) && mb->get_position().x < (mid_w + h_sep / 2);
  1991. dragging_v = mb->get_position().y > (mid_h - v_sep / 2) && mb->get_position().y < (mid_h + v_sep / 2);
  1992. drag_begin_pos = mb->get_position();
  1993. drag_begin_ratio.x = ratio_h;
  1994. drag_begin_ratio.y = ratio_v;
  1995. switch (view) {
  1996. case VIEW_USE_1_VIEWPORT: {
  1997. dragging_h = false;
  1998. dragging_v = false;
  1999. } break;
  2000. case VIEW_USE_2_VIEWPORTS: {
  2001. dragging_h = false;
  2002. } break;
  2003. case VIEW_USE_2_VIEWPORTS_ALT: {
  2004. dragging_v = false;
  2005. } break;
  2006. case VIEW_USE_3_VIEWPORTS: {
  2007. if (dragging_v)
  2008. dragging_h = false;
  2009. else
  2010. dragging_v = false;
  2011. } break;
  2012. case VIEW_USE_3_VIEWPORTS_ALT: {
  2013. if (dragging_h)
  2014. dragging_v = false;
  2015. else
  2016. dragging_h = false;
  2017. } break;
  2018. case VIEW_USE_4_VIEWPORTS: {
  2019. } break;
  2020. }
  2021. }
  2022. if (mb.is_valid() && !mb->is_pressed() && mb->get_button_index() == BUTTON_LEFT) {
  2023. dragging_h = false;
  2024. dragging_v = false;
  2025. }
  2026. Ref<InputEventMouseMotion> mm = p_event;
  2027. if (mm.is_valid() && (dragging_h || dragging_v)) {
  2028. if (dragging_h) {
  2029. float new_ratio = drag_begin_ratio.x + (mm->get_position().x - drag_begin_pos.x) / get_size().width;
  2030. new_ratio = CLAMP(new_ratio, 40 / get_size().width, (get_size().width - 40) / get_size().width);
  2031. ratio_h = new_ratio;
  2032. queue_sort();
  2033. update();
  2034. }
  2035. if (dragging_v) {
  2036. float new_ratio = drag_begin_ratio.y + (mm->get_position().y - drag_begin_pos.y) / get_size().height;
  2037. new_ratio = CLAMP(new_ratio, 40 / get_size().height, (get_size().height - 40) / get_size().height);
  2038. ratio_v = new_ratio;
  2039. queue_sort();
  2040. update();
  2041. }
  2042. }
  2043. }
  2044. void SpatialEditorViewportContainer::_notification(int p_what) {
  2045. if (p_what == NOTIFICATION_MOUSE_ENTER || p_what == NOTIFICATION_MOUSE_EXIT) {
  2046. mouseover = (p_what == NOTIFICATION_MOUSE_ENTER);
  2047. update();
  2048. }
  2049. if (p_what == NOTIFICATION_DRAW && mouseover) {
  2050. Ref<Texture> h_grabber = get_icon("grabber", "HSplitContainer");
  2051. Ref<Texture> v_grabber = get_icon("grabber", "VSplitContainer");
  2052. Vector2 size = get_size();
  2053. int h_sep = get_constant("separation", "HSplitContainer");
  2054. int v_sep = get_constant("separation", "VSplitContainer");
  2055. int mid_w = size.width * ratio_h;
  2056. int mid_h = size.height * ratio_v;
  2057. int size_left = mid_w - h_sep / 2;
  2058. int size_bottom = size.height - mid_h - v_sep / 2;
  2059. switch (view) {
  2060. case VIEW_USE_1_VIEWPORT: {
  2061. //nothing to show
  2062. } break;
  2063. case VIEW_USE_2_VIEWPORTS: {
  2064. draw_texture(v_grabber, Vector2((size.width - v_grabber->get_width()) / 2, mid_h - v_grabber->get_height() / 2));
  2065. } break;
  2066. case VIEW_USE_2_VIEWPORTS_ALT: {
  2067. draw_texture(h_grabber, Vector2(mid_w - h_grabber->get_width() / 2, (size.height - h_grabber->get_height()) / 2));
  2068. } break;
  2069. case VIEW_USE_3_VIEWPORTS: {
  2070. draw_texture(v_grabber, Vector2((size.width - v_grabber->get_width()) / 2, mid_h - v_grabber->get_height() / 2));
  2071. draw_texture(h_grabber, Vector2(mid_w - h_grabber->get_width() / 2, mid_h + v_grabber->get_height() / 2 + (size_bottom - h_grabber->get_height()) / 2));
  2072. } break;
  2073. case VIEW_USE_3_VIEWPORTS_ALT: {
  2074. draw_texture(v_grabber, Vector2((size_left - v_grabber->get_width()) / 2, mid_h - v_grabber->get_height() / 2));
  2075. draw_texture(h_grabber, Vector2(mid_w - h_grabber->get_width() / 2, (size.height - h_grabber->get_height()) / 2));
  2076. } break;
  2077. case VIEW_USE_4_VIEWPORTS: {
  2078. Vector2 half(mid_w, mid_h);
  2079. draw_texture(v_grabber, half - v_grabber->get_size() / 2.0);
  2080. draw_texture(h_grabber, half - h_grabber->get_size() / 2.0);
  2081. } break;
  2082. }
  2083. }
  2084. if (p_what == NOTIFICATION_SORT_CHILDREN) {
  2085. SpatialEditorViewport *viewports[4];
  2086. int vc = 0;
  2087. for (int i = 0; i < get_child_count(); i++) {
  2088. viewports[vc] = Object::cast_to<SpatialEditorViewport>(get_child(i));
  2089. if (viewports[vc]) {
  2090. vc++;
  2091. }
  2092. }
  2093. ERR_FAIL_COND(vc != 4);
  2094. Size2 size = get_size();
  2095. if (size.x < 10 || size.y < 10) {
  2096. for (int i = 0; i < 4; i++) {
  2097. viewports[i]->hide();
  2098. }
  2099. return;
  2100. }
  2101. int h_sep = get_constant("separation", "HSplitContainer");
  2102. int v_sep = get_constant("separation", "VSplitContainer");
  2103. int mid_w = size.width * ratio_h;
  2104. int mid_h = size.height * ratio_v;
  2105. int size_left = mid_w - h_sep / 2;
  2106. int size_right = size.width - mid_w - h_sep / 2;
  2107. int size_top = mid_h - v_sep / 2;
  2108. int size_bottom = size.height - mid_h - v_sep / 2;
  2109. switch (view) {
  2110. case VIEW_USE_1_VIEWPORT: {
  2111. for (int i = 1; i < 4; i++) {
  2112. viewports[i]->hide();
  2113. }
  2114. fit_child_in_rect(viewports[0], Rect2(Vector2(), size));
  2115. } break;
  2116. case VIEW_USE_2_VIEWPORTS: {
  2117. for (int i = 1; i < 4; i++) {
  2118. if (i == 1 || i == 3)
  2119. viewports[i]->hide();
  2120. else
  2121. viewports[i]->show();
  2122. }
  2123. fit_child_in_rect(viewports[0], Rect2(Vector2(), Vector2(size.width, size_top)));
  2124. fit_child_in_rect(viewports[2], Rect2(Vector2(0, mid_h + v_sep / 2), Vector2(size.width, size_bottom)));
  2125. } break;
  2126. case VIEW_USE_2_VIEWPORTS_ALT: {
  2127. for (int i = 1; i < 4; i++) {
  2128. if (i == 1 || i == 3)
  2129. viewports[i]->hide();
  2130. else
  2131. viewports[i]->show();
  2132. }
  2133. fit_child_in_rect(viewports[0], Rect2(Vector2(), Vector2(size_left, size.height)));
  2134. fit_child_in_rect(viewports[2], Rect2(Vector2(mid_w + h_sep / 2, 0), Vector2(size_right, size.height)));
  2135. } break;
  2136. case VIEW_USE_3_VIEWPORTS: {
  2137. for (int i = 1; i < 4; i++) {
  2138. if (i == 1)
  2139. viewports[i]->hide();
  2140. else
  2141. viewports[i]->show();
  2142. }
  2143. fit_child_in_rect(viewports[0], Rect2(Vector2(), Vector2(size.width, size_top)));
  2144. fit_child_in_rect(viewports[2], Rect2(Vector2(0, mid_h + v_sep / 2), Vector2(size_left, size_bottom)));
  2145. fit_child_in_rect(viewports[3], Rect2(Vector2(mid_w + h_sep / 2, mid_h + v_sep / 2), Vector2(size_right, size_bottom)));
  2146. } break;
  2147. case VIEW_USE_3_VIEWPORTS_ALT: {
  2148. for (int i = 1; i < 4; i++) {
  2149. if (i == 1)
  2150. viewports[i]->hide();
  2151. else
  2152. viewports[i]->show();
  2153. }
  2154. fit_child_in_rect(viewports[0], Rect2(Vector2(), Vector2(size_left, size_top)));
  2155. fit_child_in_rect(viewports[2], Rect2(Vector2(0, mid_h + v_sep / 2), Vector2(size_left, size_bottom)));
  2156. fit_child_in_rect(viewports[3], Rect2(Vector2(mid_w + h_sep / 2, 0), Vector2(size_right, size.height)));
  2157. } break;
  2158. case VIEW_USE_4_VIEWPORTS: {
  2159. for (int i = 1; i < 4; i++) {
  2160. viewports[i]->show();
  2161. }
  2162. fit_child_in_rect(viewports[0], Rect2(Vector2(), Vector2(size_left, size_top)));
  2163. fit_child_in_rect(viewports[1], Rect2(Vector2(mid_w + h_sep / 2, 0), Vector2(size_right, size_top)));
  2164. fit_child_in_rect(viewports[2], Rect2(Vector2(0, mid_h + v_sep / 2), Vector2(size_left, size_bottom)));
  2165. fit_child_in_rect(viewports[3], Rect2(Vector2(mid_w + h_sep / 2, mid_h + v_sep / 2), Vector2(size_right, size_bottom)));
  2166. } break;
  2167. }
  2168. }
  2169. }
  2170. void SpatialEditorViewportContainer::set_view(View p_view) {
  2171. view = p_view;
  2172. queue_sort();
  2173. }
  2174. SpatialEditorViewportContainer::View SpatialEditorViewportContainer::get_view() {
  2175. return view;
  2176. }
  2177. void SpatialEditorViewportContainer::_bind_methods() {
  2178. ClassDB::bind_method("_gui_input", &SpatialEditorViewportContainer::_gui_input);
  2179. }
  2180. SpatialEditorViewportContainer::SpatialEditorViewportContainer() {
  2181. view = VIEW_USE_1_VIEWPORT;
  2182. mouseover = false;
  2183. ratio_h = 0.5;
  2184. ratio_v = 0.5;
  2185. dragging_v = false;
  2186. dragging_h = false;
  2187. }
  2188. ///////////////////////////////////////////////////////////////////
  2189. SpatialEditor *SpatialEditor::singleton = NULL;
  2190. SpatialEditorSelectedItem::~SpatialEditorSelectedItem() {
  2191. if (sbox_instance.is_valid())
  2192. VisualServer::get_singleton()->free(sbox_instance);
  2193. }
  2194. void SpatialEditor::select_gizmo_highlight_axis(int p_axis) {
  2195. for (int i = 0; i < 3; i++) {
  2196. move_gizmo[i]->surface_set_material(0, i == p_axis ? gizmo_hl : gizmo_color[i]);
  2197. move_plane_gizmo[i]->surface_set_material(0, (i + 6) == p_axis ? gizmo_hl : plane_gizmo_color[i]);
  2198. rotate_gizmo[i]->surface_set_material(0, (i + 3) == p_axis ? gizmo_hl : gizmo_color[i]);
  2199. }
  2200. }
  2201. void SpatialEditor::update_transform_gizmo() {
  2202. List<Node *> &selection = editor_selection->get_selected_node_list();
  2203. Rect3 center;
  2204. bool first = true;
  2205. Basis gizmo_basis;
  2206. bool local_gizmo_coords = transform_menu->get_popup()->is_item_checked(transform_menu->get_popup()->get_item_index(MENU_TRANSFORM_LOCAL_COORDS));
  2207. for (List<Node *>::Element *E = selection.front(); E; E = E->next()) {
  2208. Spatial *sp = Object::cast_to<Spatial>(E->get());
  2209. if (!sp)
  2210. continue;
  2211. SpatialEditorSelectedItem *se = editor_selection->get_node_editor_data<SpatialEditorSelectedItem>(sp);
  2212. if (!se)
  2213. continue;
  2214. Transform xf = se->sp->get_global_transform();
  2215. if (first) {
  2216. center.position = xf.origin;
  2217. first = false;
  2218. if (local_gizmo_coords) {
  2219. gizmo_basis = xf.basis;
  2220. gizmo_basis.orthonormalize();
  2221. }
  2222. } else {
  2223. center.expand_to(xf.origin);
  2224. gizmo_basis = Basis();
  2225. }
  2226. //count++;
  2227. }
  2228. Vector3 pcenter = center.position + center.size * 0.5;
  2229. gizmo.visible = !first;
  2230. gizmo.transform.origin = pcenter;
  2231. gizmo.transform.basis = gizmo_basis;
  2232. for (int i = 0; i < VIEWPORTS_COUNT; i++) {
  2233. viewports[i]->update_transform_gizmo_view();
  2234. }
  2235. }
  2236. Object *SpatialEditor::_get_editor_data(Object *p_what) {
  2237. Spatial *sp = Object::cast_to<Spatial>(p_what);
  2238. if (!sp)
  2239. return NULL;
  2240. SpatialEditorSelectedItem *si = memnew(SpatialEditorSelectedItem);
  2241. si->sp = sp;
  2242. si->sbox_instance = VisualServer::get_singleton()->instance_create2(selection_box->get_rid(), sp->get_world()->get_scenario());
  2243. VS::get_singleton()->instance_geometry_set_cast_shadows_setting(si->sbox_instance, VS::SHADOW_CASTING_SETTING_OFF);
  2244. if (Engine::get_singleton()->is_editor_hint())
  2245. editor->call("edit_node", sp);
  2246. return si;
  2247. }
  2248. void SpatialEditor::_generate_selection_box() {
  2249. Rect3 aabb(Vector3(), Vector3(1, 1, 1));
  2250. aabb.grow_by(aabb.get_longest_axis_size() / 20.0);
  2251. Ref<SurfaceTool> st = memnew(SurfaceTool);
  2252. st->begin(Mesh::PRIMITIVE_LINES);
  2253. for (int i = 0; i < 12; i++) {
  2254. Vector3 a, b;
  2255. aabb.get_edge(i, a, b);
  2256. /*Vector<Vector3> points;
  2257. Vector<Color> colors;
  2258. points.push_back(a);
  2259. points.push_back(b);*/
  2260. st->add_color(Color(1.0, 1.0, 0.8, 0.8));
  2261. st->add_vertex(a);
  2262. st->add_color(Color(1.0, 1.0, 0.8, 0.4));
  2263. st->add_vertex(a.linear_interpolate(b, 0.2));
  2264. st->add_color(Color(1.0, 1.0, 0.8, 0.4));
  2265. st->add_vertex(a.linear_interpolate(b, 0.8));
  2266. st->add_color(Color(1.0, 1.0, 0.8, 0.8));
  2267. st->add_vertex(b);
  2268. }
  2269. Ref<SpatialMaterial> mat = memnew(SpatialMaterial);
  2270. mat->set_flag(SpatialMaterial::FLAG_UNSHADED, true);
  2271. mat->set_albedo(Color(1, 1, 1));
  2272. mat->set_feature(SpatialMaterial::FEATURE_TRANSPARENT, true);
  2273. mat->set_flag(SpatialMaterial::FLAG_ALBEDO_FROM_VERTEX_COLOR, true);
  2274. mat->set_flag(SpatialMaterial::FLAG_SRGB_VERTEX_COLOR, true);
  2275. st->set_material(mat);
  2276. selection_box = st->commit();
  2277. }
  2278. Dictionary SpatialEditor::get_state() const {
  2279. Dictionary d;
  2280. d["snap_enabled"] = snap_enabled;
  2281. d["translate_snap"] = get_translate_snap();
  2282. d["rotate_snap"] = get_rotate_snap();
  2283. d["scale_snap"] = get_scale_snap();
  2284. int local_coords_index = transform_menu->get_popup()->get_item_index(MENU_TRANSFORM_LOCAL_COORDS);
  2285. d["local_coords"] = transform_menu->get_popup()->is_item_checked(local_coords_index);
  2286. int vc = 0;
  2287. if (view_menu->get_popup()->is_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_1_VIEWPORT)))
  2288. vc = 1;
  2289. else if (view_menu->get_popup()->is_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_2_VIEWPORTS)))
  2290. vc = 2;
  2291. else if (view_menu->get_popup()->is_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_3_VIEWPORTS)))
  2292. vc = 3;
  2293. else if (view_menu->get_popup()->is_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_4_VIEWPORTS)))
  2294. vc = 4;
  2295. else if (view_menu->get_popup()->is_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_2_VIEWPORTS_ALT)))
  2296. vc = 5;
  2297. else if (view_menu->get_popup()->is_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_3_VIEWPORTS_ALT)))
  2298. vc = 6;
  2299. d["viewport_mode"] = vc;
  2300. Array vpdata;
  2301. for (int i = 0; i < 4; i++) {
  2302. vpdata.push_back(viewports[i]->get_state());
  2303. }
  2304. d["viewports"] = vpdata;
  2305. d["show_grid"] = view_menu->get_popup()->is_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_GRID));
  2306. d["show_origin"] = view_menu->get_popup()->is_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_ORIGIN));
  2307. d["fov"] = get_fov();
  2308. d["znear"] = get_znear();
  2309. d["zfar"] = get_zfar();
  2310. return d;
  2311. }
  2312. void SpatialEditor::set_state(const Dictionary &p_state) {
  2313. Dictionary d = p_state;
  2314. if (d.has("snap_enabled")) {
  2315. snap_enabled = d["snap_enabled"];
  2316. int snap_enabled_idx = transform_menu->get_popup()->get_item_index(MENU_TRANSFORM_USE_SNAP);
  2317. transform_menu->get_popup()->set_item_checked(snap_enabled_idx, snap_enabled);
  2318. }
  2319. if (d.has("translate_snap"))
  2320. snap_translate->set_text(d["translate_snap"]);
  2321. if (d.has("rotate_snap"))
  2322. snap_rotate->set_text(d["rotate_snap"]);
  2323. if (d.has("scale_snap"))
  2324. snap_scale->set_text(d["scale_snap"]);
  2325. if (d.has("local_coords")) {
  2326. int local_coords_idx = transform_menu->get_popup()->get_item_index(MENU_TRANSFORM_LOCAL_COORDS);
  2327. transform_menu->get_popup()->set_item_checked(local_coords_idx, d["local_coords"]);
  2328. update_transform_gizmo();
  2329. }
  2330. if (d.has("viewport_mode")) {
  2331. int vc = d["viewport_mode"];
  2332. if (vc == 1)
  2333. _menu_item_pressed(MENU_VIEW_USE_1_VIEWPORT);
  2334. else if (vc == 2)
  2335. _menu_item_pressed(MENU_VIEW_USE_2_VIEWPORTS);
  2336. else if (vc == 3)
  2337. _menu_item_pressed(MENU_VIEW_USE_3_VIEWPORTS);
  2338. else if (vc == 4)
  2339. _menu_item_pressed(MENU_VIEW_USE_4_VIEWPORTS);
  2340. else if (vc == 5)
  2341. _menu_item_pressed(MENU_VIEW_USE_2_VIEWPORTS_ALT);
  2342. else if (vc == 6)
  2343. _menu_item_pressed(MENU_VIEW_USE_3_VIEWPORTS_ALT);
  2344. }
  2345. if (d.has("viewports")) {
  2346. Array vp = d["viewports"];
  2347. ERR_FAIL_COND(vp.size() > 4);
  2348. for (int i = 0; i < VIEWPORTS_COUNT; i++) {
  2349. viewports[i]->set_state(vp[i]);
  2350. }
  2351. }
  2352. if (d.has("zfar"))
  2353. settings_zfar->set_value(float(d["zfar"]));
  2354. if (d.has("znear"))
  2355. settings_znear->set_value(float(d["znear"]));
  2356. if (d.has("fov"))
  2357. settings_fov->set_value(float(d["fov"]));
  2358. if (d.has("default_srgb")) {
  2359. bool use = d["default_srgb"];
  2360. //viewport_environment->set_enable_fx(Environment::FX_SRGB,use);
  2361. //view_menu->get_popup()->set_item_checked( view_menu->get_popup()->get_item_index(MENU_VIEW_USE_DEFAULT_SRGB), use );
  2362. }
  2363. if (d.has("show_grid")) {
  2364. bool use = d["show_grid"];
  2365. if (use != view_menu->get_popup()->is_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_GRID))) {
  2366. _menu_item_pressed(MENU_VIEW_GRID);
  2367. }
  2368. }
  2369. if (d.has("show_origin")) {
  2370. bool use = d["show_origin"];
  2371. if (use != view_menu->get_popup()->is_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_ORIGIN))) {
  2372. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_ORIGIN), use);
  2373. VisualServer::get_singleton()->instance_set_visible(origin_instance, use);
  2374. }
  2375. }
  2376. }
  2377. void SpatialEditor::edit(Spatial *p_spatial) {
  2378. if (p_spatial != selected) {
  2379. if (selected) {
  2380. Ref<SpatialEditorGizmo> seg = selected->get_gizmo();
  2381. if (seg.is_valid()) {
  2382. seg->set_selected(false);
  2383. selected->update_gizmo();
  2384. }
  2385. }
  2386. selected = p_spatial;
  2387. over_gizmo_handle = -1;
  2388. if (selected) {
  2389. Ref<SpatialEditorGizmo> seg = selected->get_gizmo();
  2390. if (seg.is_valid()) {
  2391. seg->set_selected(true);
  2392. selected->update_gizmo();
  2393. }
  2394. }
  2395. }
  2396. /*
  2397. if (p_spatial) {
  2398. _validate_selection();
  2399. if (selected.has(p_spatial->get_instance_id()) && selected.size()==1)
  2400. return;
  2401. _select(p_spatial->get_instance_id(),false,true);
  2402. // should become the selection
  2403. }
  2404. */
  2405. }
  2406. void SpatialEditor::_xform_dialog_action() {
  2407. Transform t;
  2408. //translation
  2409. Vector3 scale;
  2410. Vector3 rotate;
  2411. Vector3 translate;
  2412. for (int i = 0; i < 3; i++) {
  2413. translate[i] = xform_translate[i]->get_text().to_double();
  2414. rotate[i] = Math::deg2rad(xform_rotate[i]->get_text().to_double());
  2415. scale[i] = xform_scale[i]->get_text().to_double();
  2416. }
  2417. t.basis.scale(scale);
  2418. t.basis.rotate(rotate);
  2419. t.origin = translate;
  2420. undo_redo->create_action(TTR("XForm Dialog"));
  2421. List<Node *> &selection = editor_selection->get_selected_node_list();
  2422. for (List<Node *>::Element *E = selection.front(); E; E = E->next()) {
  2423. Spatial *sp = Object::cast_to<Spatial>(E->get());
  2424. if (!sp)
  2425. continue;
  2426. SpatialEditorSelectedItem *se = editor_selection->get_node_editor_data<SpatialEditorSelectedItem>(sp);
  2427. if (!se)
  2428. continue;
  2429. bool post = xform_type->get_selected() > 0;
  2430. Transform tr = sp->get_global_transform();
  2431. if (post)
  2432. tr = tr * t;
  2433. else {
  2434. tr.basis = t.basis * tr.basis;
  2435. tr.origin += t.origin;
  2436. }
  2437. undo_redo->add_do_method(sp, "set_global_transform", tr);
  2438. undo_redo->add_undo_method(sp, "set_global_transform", sp->get_global_transform());
  2439. }
  2440. undo_redo->commit_action();
  2441. }
  2442. void SpatialEditor::_menu_item_pressed(int p_option) {
  2443. switch (p_option) {
  2444. case MENU_TOOL_SELECT:
  2445. case MENU_TOOL_MOVE:
  2446. case MENU_TOOL_ROTATE:
  2447. case MENU_TOOL_SCALE:
  2448. case MENU_TOOL_LIST_SELECT: {
  2449. for (int i = 0; i < TOOL_MAX; i++)
  2450. tool_button[i]->set_pressed(i == p_option);
  2451. tool_mode = (ToolMode)p_option;
  2452. //static const char *_mode[]={"Selection Mode.","Translation Mode.","Rotation Mode.","Scale Mode.","List Selection Mode."};
  2453. //set_message(_mode[p_option],3);
  2454. update_transform_gizmo();
  2455. } break;
  2456. case MENU_TRANSFORM_USE_SNAP: {
  2457. bool is_checked = transform_menu->get_popup()->is_item_checked(transform_menu->get_popup()->get_item_index(p_option));
  2458. snap_enabled = !is_checked;
  2459. transform_menu->get_popup()->set_item_checked(transform_menu->get_popup()->get_item_index(p_option), snap_enabled);
  2460. } break;
  2461. case MENU_TRANSFORM_CONFIGURE_SNAP: {
  2462. snap_dialog->popup_centered(Size2(200, 180));
  2463. } break;
  2464. case MENU_TRANSFORM_LOCAL_COORDS: {
  2465. bool is_checked = transform_menu->get_popup()->is_item_checked(transform_menu->get_popup()->get_item_index(p_option));
  2466. transform_menu->get_popup()->set_item_checked(transform_menu->get_popup()->get_item_index(p_option), !is_checked);
  2467. update_transform_gizmo();
  2468. } break;
  2469. case MENU_TRANSFORM_DIALOG: {
  2470. for (int i = 0; i < 3; i++) {
  2471. xform_translate[i]->set_text("0");
  2472. xform_rotate[i]->set_text("0");
  2473. xform_scale[i]->set_text("1");
  2474. }
  2475. xform_dialog->popup_centered(Size2(320, 240) * EDSCALE);
  2476. } break;
  2477. case MENU_VIEW_USE_1_VIEWPORT: {
  2478. viewport_base->set_view(SpatialEditorViewportContainer::VIEW_USE_1_VIEWPORT);
  2479. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_1_VIEWPORT), true);
  2480. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_2_VIEWPORTS), false);
  2481. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_3_VIEWPORTS), false);
  2482. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_4_VIEWPORTS), false);
  2483. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_2_VIEWPORTS_ALT), false);
  2484. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_3_VIEWPORTS_ALT), false);
  2485. } break;
  2486. case MENU_VIEW_USE_2_VIEWPORTS: {
  2487. viewport_base->set_view(SpatialEditorViewportContainer::VIEW_USE_2_VIEWPORTS);
  2488. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_1_VIEWPORT), false);
  2489. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_2_VIEWPORTS), true);
  2490. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_3_VIEWPORTS), false);
  2491. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_4_VIEWPORTS), false);
  2492. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_2_VIEWPORTS_ALT), false);
  2493. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_3_VIEWPORTS_ALT), false);
  2494. } break;
  2495. case MENU_VIEW_USE_2_VIEWPORTS_ALT: {
  2496. viewport_base->set_view(SpatialEditorViewportContainer::VIEW_USE_2_VIEWPORTS_ALT);
  2497. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_1_VIEWPORT), false);
  2498. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_2_VIEWPORTS), false);
  2499. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_3_VIEWPORTS), false);
  2500. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_4_VIEWPORTS), false);
  2501. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_2_VIEWPORTS_ALT), true);
  2502. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_3_VIEWPORTS_ALT), false);
  2503. } break;
  2504. case MENU_VIEW_USE_3_VIEWPORTS: {
  2505. viewport_base->set_view(SpatialEditorViewportContainer::VIEW_USE_3_VIEWPORTS);
  2506. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_1_VIEWPORT), false);
  2507. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_2_VIEWPORTS), false);
  2508. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_3_VIEWPORTS), true);
  2509. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_4_VIEWPORTS), false);
  2510. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_2_VIEWPORTS_ALT), false);
  2511. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_3_VIEWPORTS_ALT), false);
  2512. } break;
  2513. case MENU_VIEW_USE_3_VIEWPORTS_ALT: {
  2514. viewport_base->set_view(SpatialEditorViewportContainer::VIEW_USE_3_VIEWPORTS_ALT);
  2515. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_1_VIEWPORT), false);
  2516. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_2_VIEWPORTS), false);
  2517. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_3_VIEWPORTS), false);
  2518. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_4_VIEWPORTS), false);
  2519. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_2_VIEWPORTS_ALT), false);
  2520. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_3_VIEWPORTS_ALT), true);
  2521. } break;
  2522. case MENU_VIEW_USE_4_VIEWPORTS: {
  2523. viewport_base->set_view(SpatialEditorViewportContainer::VIEW_USE_4_VIEWPORTS);
  2524. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_1_VIEWPORT), false);
  2525. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_2_VIEWPORTS), false);
  2526. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_3_VIEWPORTS), false);
  2527. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_4_VIEWPORTS), true);
  2528. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_2_VIEWPORTS_ALT), false);
  2529. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_3_VIEWPORTS_ALT), false);
  2530. } break;
  2531. case MENU_VIEW_ORIGIN: {
  2532. bool is_checked = view_menu->get_popup()->is_item_checked(view_menu->get_popup()->get_item_index(p_option));
  2533. is_checked = !is_checked;
  2534. VisualServer::get_singleton()->instance_set_visible(origin_instance, is_checked);
  2535. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(p_option), is_checked);
  2536. } break;
  2537. case MENU_VIEW_GRID: {
  2538. bool is_checked = view_menu->get_popup()->is_item_checked(view_menu->get_popup()->get_item_index(p_option));
  2539. grid_enabled = !is_checked;
  2540. for (int i = 0; i < 3; ++i) {
  2541. if (grid_enable[i]) {
  2542. VisualServer::get_singleton()->instance_set_visible(grid_instance[i], grid_enabled);
  2543. grid_visible[i] = grid_enabled;
  2544. }
  2545. }
  2546. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(p_option), grid_enabled);
  2547. } break;
  2548. case MENU_VIEW_CAMERA_SETTINGS: {
  2549. settings_dialog->popup_centered(settings_vbc->get_combined_minimum_size() + Size2(50, 50));
  2550. } break;
  2551. }
  2552. }
  2553. void SpatialEditor::_init_indicators() {
  2554. //RID mat = VisualServer::get_singleton()->fixed_material_create();
  2555. ///VisualServer::get_singleton()->fixed_material_set_flag(mat, VisualServer::FIXED_MATERIAL_FLAG_USE_ALPHA,true);
  2556. //VisualServer::get_singleton()->fixed_material_set_flag(mat, VisualServer::FIXED_MATERIAL_FLAG_USE_COLOR_ARRAY,true);
  2557. {
  2558. indicator_mat.instance();
  2559. indicator_mat->set_flag(SpatialMaterial::FLAG_UNSHADED, true);
  2560. //indicator_mat->set_flag(SpatialMaterial::FLAG_ONTOP,true);
  2561. indicator_mat->set_flag(SpatialMaterial::FLAG_ALBEDO_FROM_VERTEX_COLOR, true);
  2562. indicator_mat->set_flag(SpatialMaterial::FLAG_SRGB_VERTEX_COLOR, true);
  2563. indicator_mat->set_feature(SpatialMaterial::FEATURE_TRANSPARENT, true);
  2564. PoolVector<Color> grid_colors[3];
  2565. PoolVector<Vector3> grid_points[3];
  2566. Vector<Color> origin_colors;
  2567. Vector<Vector3> origin_points;
  2568. Color grid_color = EditorSettings::get_singleton()->get("editors/3d/grid_color");
  2569. for (int i = 0; i < 3; i++) {
  2570. Vector3 axis;
  2571. axis[i] = 1;
  2572. Vector3 axis_n1;
  2573. axis_n1[(i + 1) % 3] = 1;
  2574. Vector3 axis_n2;
  2575. axis_n2[(i + 2) % 3] = 1;
  2576. origin_colors.push_back(Color(axis.x, axis.y, axis.z));
  2577. origin_colors.push_back(Color(axis.x, axis.y, axis.z));
  2578. origin_points.push_back(axis * 4096);
  2579. origin_points.push_back(axis * -4096);
  2580. #define ORIGIN_GRID_SIZE 25
  2581. for (int j = -ORIGIN_GRID_SIZE; j <= ORIGIN_GRID_SIZE; j++) {
  2582. grid_colors[i].push_back(grid_color);
  2583. grid_colors[i].push_back(grid_color);
  2584. grid_colors[i].push_back(grid_color);
  2585. grid_colors[i].push_back(grid_color);
  2586. grid_points[i].push_back(axis_n1 * ORIGIN_GRID_SIZE + axis_n2 * j);
  2587. grid_points[i].push_back(-axis_n1 * ORIGIN_GRID_SIZE + axis_n2 * j);
  2588. grid_points[i].push_back(axis_n2 * ORIGIN_GRID_SIZE + axis_n1 * j);
  2589. grid_points[i].push_back(-axis_n2 * ORIGIN_GRID_SIZE + axis_n1 * j);
  2590. }
  2591. grid[i] = VisualServer::get_singleton()->mesh_create();
  2592. Array d;
  2593. d.resize(VS::ARRAY_MAX);
  2594. d[VisualServer::ARRAY_VERTEX] = grid_points[i];
  2595. d[VisualServer::ARRAY_COLOR] = grid_colors[i];
  2596. VisualServer::get_singleton()->mesh_add_surface_from_arrays(grid[i], VisualServer::PRIMITIVE_LINES, d);
  2597. VisualServer::get_singleton()->mesh_surface_set_material(grid[i], 0, indicator_mat->get_rid());
  2598. grid_instance[i] = VisualServer::get_singleton()->instance_create2(grid[i], get_tree()->get_root()->get_world()->get_scenario());
  2599. grid_visible[i] = false;
  2600. grid_enable[i] = false;
  2601. VisualServer::get_singleton()->instance_set_visible(grid_instance[i], false);
  2602. VisualServer::get_singleton()->instance_geometry_set_cast_shadows_setting(grid_instance[i], VS::SHADOW_CASTING_SETTING_OFF);
  2603. VS::get_singleton()->instance_set_layer_mask(grid_instance[i], 1 << SpatialEditorViewport::GIZMO_GRID_LAYER);
  2604. }
  2605. origin = VisualServer::get_singleton()->mesh_create();
  2606. Array d;
  2607. d.resize(VS::ARRAY_MAX);
  2608. d[VisualServer::ARRAY_VERTEX] = origin_points;
  2609. d[VisualServer::ARRAY_COLOR] = origin_colors;
  2610. VisualServer::get_singleton()->mesh_add_surface_from_arrays(origin, VisualServer::PRIMITIVE_LINES, d);
  2611. VisualServer::get_singleton()->mesh_surface_set_material(origin, 0, indicator_mat->get_rid());
  2612. //origin = VisualServer::get_singleton()->poly_create();
  2613. //VisualServer::get_singleton()->poly_add_primitive(origin,origin_points,Vector<Vector3>(),origin_colors,Vector<Vector3>());
  2614. //VisualServer::get_singleton()->poly_set_material(origin,indicator_mat,true);
  2615. origin_instance = VisualServer::get_singleton()->instance_create2(origin, get_tree()->get_root()->get_world()->get_scenario());
  2616. VS::get_singleton()->instance_set_layer_mask(origin_instance, 1 << SpatialEditorViewport::GIZMO_GRID_LAYER);
  2617. VisualServer::get_singleton()->instance_geometry_set_cast_shadows_setting(origin_instance, VS::SHADOW_CASTING_SETTING_OFF);
  2618. VisualServer::get_singleton()->instance_set_visible(grid_instance[1], true);
  2619. grid_enable[1] = true;
  2620. grid_visible[1] = true;
  2621. grid_enabled = true;
  2622. last_grid_snap = 1;
  2623. }
  2624. {
  2625. cursor_mesh = VisualServer::get_singleton()->mesh_create();
  2626. PoolVector<Vector3> cursor_points;
  2627. float cs = 0.25;
  2628. cursor_points.push_back(Vector3(+cs, 0, 0));
  2629. cursor_points.push_back(Vector3(-cs, 0, 0));
  2630. cursor_points.push_back(Vector3(0, +cs, 0));
  2631. cursor_points.push_back(Vector3(0, -cs, 0));
  2632. cursor_points.push_back(Vector3(0, 0, +cs));
  2633. cursor_points.push_back(Vector3(0, 0, -cs));
  2634. cursor_material.instance();
  2635. cursor_material->set_albedo(Color(0, 1, 1));
  2636. cursor_material->set_flag(SpatialMaterial::FLAG_UNSHADED, true);
  2637. Array d;
  2638. d.resize(VS::ARRAY_MAX);
  2639. d[VS::ARRAY_VERTEX] = cursor_points;
  2640. VisualServer::get_singleton()->mesh_add_surface_from_arrays(cursor_mesh, VS::PRIMITIVE_LINES, d);
  2641. VisualServer::get_singleton()->mesh_surface_set_material(cursor_mesh, 0, cursor_material->get_rid());
  2642. cursor_instance = VisualServer::get_singleton()->instance_create2(cursor_mesh, get_tree()->get_root()->get_world()->get_scenario());
  2643. VS::get_singleton()->instance_set_layer_mask(cursor_instance, 1 << SpatialEditorViewport::GIZMO_GRID_LAYER);
  2644. VisualServer::get_singleton()->instance_geometry_set_cast_shadows_setting(cursor_instance, VS::SHADOW_CASTING_SETTING_OFF);
  2645. }
  2646. {
  2647. //move gizmo
  2648. float gizmo_alph = EditorSettings::get_singleton()->get("editors/3d/manipulator_gizmo_opacity");
  2649. gizmo_hl = Ref<SpatialMaterial>(memnew(SpatialMaterial));
  2650. gizmo_hl->set_flag(SpatialMaterial::FLAG_UNSHADED, true);
  2651. gizmo_hl->set_on_top_of_alpha();
  2652. gizmo_hl->set_feature(SpatialMaterial::FEATURE_TRANSPARENT, true);
  2653. gizmo_hl->set_albedo(Color(1, 1, 1, gizmo_alph + 0.2f));
  2654. gizmo_hl->set_cull_mode(SpatialMaterial::CULL_DISABLED);
  2655. for (int i = 0; i < 3; i++) {
  2656. move_gizmo[i] = Ref<ArrayMesh>(memnew(ArrayMesh));
  2657. move_plane_gizmo[i] = Ref<ArrayMesh>(memnew(ArrayMesh));
  2658. rotate_gizmo[i] = Ref<ArrayMesh>(memnew(ArrayMesh));
  2659. Ref<SpatialMaterial> mat = memnew(SpatialMaterial);
  2660. mat->set_flag(SpatialMaterial::FLAG_UNSHADED, true);
  2661. mat->set_on_top_of_alpha();
  2662. mat->set_feature(SpatialMaterial::FEATURE_TRANSPARENT, true);
  2663. Color col;
  2664. col[i] = 1.0;
  2665. col.a = gizmo_alph;
  2666. mat->set_albedo(col);
  2667. gizmo_color[i] = mat;
  2668. Vector3 ivec;
  2669. ivec[i] = 1;
  2670. Vector3 nivec;
  2671. nivec[(i + 1) % 3] = 1;
  2672. nivec[(i + 2) % 3] = 1;
  2673. Vector3 ivec2;
  2674. ivec2[(i + 1) % 3] = 1;
  2675. Vector3 ivec3;
  2676. ivec3[(i + 2) % 3] = 1;
  2677. {
  2678. Ref<SurfaceTool> surftool = memnew(SurfaceTool);
  2679. surftool->begin(Mesh::PRIMITIVE_TRIANGLES);
  2680. //translate
  2681. const int arrow_points = 5;
  2682. Vector3 arrow[5] = {
  2683. nivec * 0.0 + ivec * 0.0,
  2684. nivec * 0.01 + ivec * 0.0,
  2685. nivec * 0.01 + ivec * 1.0,
  2686. nivec * 0.1 + ivec * 1.0,
  2687. nivec * 0.0 + ivec * (1 + GIZMO_ARROW_SIZE),
  2688. };
  2689. int arrow_sides = 6;
  2690. for (int k = 0; k < 7; k++) {
  2691. Basis ma(ivec, Math_PI * 2 * float(k) / arrow_sides);
  2692. Basis mb(ivec, Math_PI * 2 * float(k + 1) / arrow_sides);
  2693. for (int j = 0; j < arrow_points - 1; j++) {
  2694. Vector3 points[4] = {
  2695. ma.xform(arrow[j]),
  2696. mb.xform(arrow[j]),
  2697. mb.xform(arrow[j + 1]),
  2698. ma.xform(arrow[j + 1]),
  2699. };
  2700. surftool->add_vertex(points[0]);
  2701. surftool->add_vertex(points[1]);
  2702. surftool->add_vertex(points[2]);
  2703. surftool->add_vertex(points[0]);
  2704. surftool->add_vertex(points[2]);
  2705. surftool->add_vertex(points[3]);
  2706. }
  2707. }
  2708. surftool->set_material(mat);
  2709. surftool->commit(move_gizmo[i]);
  2710. }
  2711. // plane translation
  2712. {
  2713. Ref<SurfaceTool> surftool = memnew(SurfaceTool);
  2714. surftool->begin(Mesh::PRIMITIVE_TRIANGLES);
  2715. Vector3 vec = ivec2 - ivec3;
  2716. Vector3 plane[4] = {
  2717. vec * GIZMO_PLANE_DST,
  2718. vec * GIZMO_PLANE_DST + ivec2 * GIZMO_PLANE_SIZE,
  2719. vec * (GIZMO_PLANE_DST + GIZMO_PLANE_SIZE),
  2720. vec * GIZMO_PLANE_DST - ivec3 * GIZMO_PLANE_SIZE
  2721. };
  2722. Basis ma(ivec, Math_PI / 2);
  2723. Vector3 points[4] = {
  2724. ma.xform(plane[0]),
  2725. ma.xform(plane[1]),
  2726. ma.xform(plane[2]),
  2727. ma.xform(plane[3]),
  2728. };
  2729. surftool->add_vertex(points[0]);
  2730. surftool->add_vertex(points[1]);
  2731. surftool->add_vertex(points[2]);
  2732. surftool->add_vertex(points[0]);
  2733. surftool->add_vertex(points[2]);
  2734. surftool->add_vertex(points[3]);
  2735. Ref<SpatialMaterial> plane_mat = memnew(SpatialMaterial);
  2736. plane_mat->set_flag(SpatialMaterial::FLAG_UNSHADED, true);
  2737. plane_mat->set_on_top_of_alpha();
  2738. plane_mat->set_feature(SpatialMaterial::FEATURE_TRANSPARENT, true);
  2739. plane_mat->set_cull_mode(SpatialMaterial::CULL_DISABLED);
  2740. Color col;
  2741. col[i] = 1.0;
  2742. col.a = gizmo_alph;
  2743. plane_mat->set_albedo(col);
  2744. plane_gizmo_color[i] = plane_mat; // needed, so we can draw planes from both sides
  2745. surftool->set_material(plane_mat);
  2746. surftool->commit(move_plane_gizmo[i]);
  2747. }
  2748. {
  2749. Ref<SurfaceTool> surftool = memnew(SurfaceTool);
  2750. surftool->begin(Mesh::PRIMITIVE_TRIANGLES);
  2751. Vector3 circle[5] = {
  2752. ivec * 0.02 + ivec2 * 0.02 + ivec2 * GIZMO_CIRCLE_SIZE,
  2753. ivec * -0.02 + ivec2 * 0.02 + ivec2 * GIZMO_CIRCLE_SIZE,
  2754. ivec * -0.02 + ivec2 * -0.02 + ivec2 * GIZMO_CIRCLE_SIZE,
  2755. ivec * 0.02 + ivec2 * -0.02 + ivec2 * GIZMO_CIRCLE_SIZE,
  2756. ivec * 0.02 + ivec2 * 0.02 + ivec2 * GIZMO_CIRCLE_SIZE,
  2757. };
  2758. for (int k = 0; k < 33; k++) {
  2759. Basis ma(ivec, Math_PI * 2 * float(k) / 32);
  2760. Basis mb(ivec, Math_PI * 2 * float(k + 1) / 32);
  2761. for (int j = 0; j < 4; j++) {
  2762. Vector3 points[4] = {
  2763. ma.xform(circle[j]),
  2764. mb.xform(circle[j]),
  2765. mb.xform(circle[j + 1]),
  2766. ma.xform(circle[j + 1]),
  2767. };
  2768. surftool->add_vertex(points[0]);
  2769. surftool->add_vertex(points[1]);
  2770. surftool->add_vertex(points[2]);
  2771. surftool->add_vertex(points[0]);
  2772. surftool->add_vertex(points[2]);
  2773. surftool->add_vertex(points[3]);
  2774. }
  2775. }
  2776. surftool->set_material(mat);
  2777. surftool->commit(rotate_gizmo[i]);
  2778. }
  2779. }
  2780. }
  2781. /*for(int i=0;i<4;i++) {
  2782. viewports[i]->init_gizmo_instance(i);
  2783. }*/
  2784. _generate_selection_box();
  2785. //Object::cast_to<EditorNode>(get_scene()->get_root_node())->get_scene_root()->add_child(camera);
  2786. //current_camera=camera;
  2787. }
  2788. void SpatialEditor::_finish_indicators() {
  2789. VisualServer::get_singleton()->free(origin_instance);
  2790. VisualServer::get_singleton()->free(origin);
  2791. for (int i = 0; i < 3; i++) {
  2792. VisualServer::get_singleton()->free(grid_instance[i]);
  2793. VisualServer::get_singleton()->free(grid[i]);
  2794. }
  2795. //VisualServer::get_singleton()->free(poly);
  2796. //VisualServer::get_singleton()->free(indicators_instance);
  2797. //VisualServer::get_singleton()->free(indicators);
  2798. VisualServer::get_singleton()->free(cursor_instance);
  2799. VisualServer::get_singleton()->free(cursor_mesh);
  2800. }
  2801. bool SpatialEditor::is_any_freelook_active() const {
  2802. for (unsigned int i = 0; i < VIEWPORTS_COUNT; ++i) {
  2803. if (viewports[i]->is_freelook_active())
  2804. return true;
  2805. }
  2806. return false;
  2807. }
  2808. void SpatialEditor::_unhandled_key_input(Ref<InputEvent> p_event) {
  2809. if (!is_visible_in_tree() || get_viewport()->gui_has_modal_stack())
  2810. return;
  2811. Ref<InputEventKey> k = p_event;
  2812. if (k.is_valid()) {
  2813. // Note: need to check is_echo because first person movement keys might still be held
  2814. if (!is_any_freelook_active() && !p_event->is_echo()) {
  2815. if (!k->is_pressed())
  2816. return;
  2817. if (ED_IS_SHORTCUT("spatial_editor/tool_select", p_event))
  2818. _menu_item_pressed(MENU_TOOL_SELECT);
  2819. else if (ED_IS_SHORTCUT("spatial_editor/tool_move", p_event))
  2820. _menu_item_pressed(MENU_TOOL_MOVE);
  2821. else if (ED_IS_SHORTCUT("spatial_editor/tool_rotate", p_event))
  2822. _menu_item_pressed(MENU_TOOL_ROTATE);
  2823. else if (ED_IS_SHORTCUT("spatial_editor/tool_scale", p_event))
  2824. _menu_item_pressed(MENU_TOOL_SCALE);
  2825. }
  2826. }
  2827. }
  2828. void SpatialEditor::_notification(int p_what) {
  2829. if (p_what == NOTIFICATION_READY) {
  2830. tool_button[SpatialEditor::TOOL_MODE_SELECT]->set_icon(get_icon("ToolSelect", "EditorIcons"));
  2831. tool_button[SpatialEditor::TOOL_MODE_MOVE]->set_icon(get_icon("ToolMove", "EditorIcons"));
  2832. tool_button[SpatialEditor::TOOL_MODE_ROTATE]->set_icon(get_icon("ToolRotate", "EditorIcons"));
  2833. tool_button[SpatialEditor::TOOL_MODE_SCALE]->set_icon(get_icon("ToolScale", "EditorIcons"));
  2834. tool_button[SpatialEditor::TOOL_MODE_LIST_SELECT]->set_icon(get_icon("ListSelect", "EditorIcons"));
  2835. view_menu->get_popup()->set_item_icon(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_1_VIEWPORT), get_icon("Panels1", "EditorIcons"));
  2836. view_menu->get_popup()->set_item_icon(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_2_VIEWPORTS), get_icon("Panels2", "EditorIcons"));
  2837. view_menu->get_popup()->set_item_icon(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_2_VIEWPORTS_ALT), get_icon("Panels2Alt", "EditorIcons"));
  2838. view_menu->get_popup()->set_item_icon(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_3_VIEWPORTS), get_icon("Panels3", "EditorIcons"));
  2839. view_menu->get_popup()->set_item_icon(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_3_VIEWPORTS_ALT), get_icon("Panels3Alt", "EditorIcons"));
  2840. view_menu->get_popup()->set_item_icon(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_4_VIEWPORTS), get_icon("Panels4", "EditorIcons"));
  2841. _menu_item_pressed(MENU_VIEW_USE_1_VIEWPORT);
  2842. get_tree()->connect("node_removed", this, "_node_removed");
  2843. }
  2844. if (p_what == NOTIFICATION_ENTER_TREE) {
  2845. gizmos = memnew(SpatialEditorGizmos);
  2846. _init_indicators();
  2847. }
  2848. if (p_what == NOTIFICATION_EXIT_TREE) {
  2849. _finish_indicators();
  2850. memdelete(gizmos);
  2851. }
  2852. if (p_what == EditorSettings::NOTIFICATION_EDITOR_SETTINGS_CHANGED) {
  2853. tool_button[SpatialEditor::TOOL_MODE_SELECT]->set_icon(get_icon("ToolSelect", "EditorIcons"));
  2854. tool_button[SpatialEditor::TOOL_MODE_MOVE]->set_icon(get_icon("ToolMove", "EditorIcons"));
  2855. tool_button[SpatialEditor::TOOL_MODE_ROTATE]->set_icon(get_icon("ToolRotate", "EditorIcons"));
  2856. tool_button[SpatialEditor::TOOL_MODE_SCALE]->set_icon(get_icon("ToolScale", "EditorIcons"));
  2857. tool_button[SpatialEditor::TOOL_MODE_LIST_SELECT]->set_icon(get_icon("ListSelect", "EditorIcons"));
  2858. view_menu->get_popup()->set_item_icon(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_1_VIEWPORT), get_icon("Panels1", "EditorIcons"));
  2859. view_menu->get_popup()->set_item_icon(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_2_VIEWPORTS), get_icon("Panels2", "EditorIcons"));
  2860. view_menu->get_popup()->set_item_icon(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_2_VIEWPORTS_ALT), get_icon("Panels2Alt", "EditorIcons"));
  2861. view_menu->get_popup()->set_item_icon(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_3_VIEWPORTS), get_icon("Panels3", "EditorIcons"));
  2862. view_menu->get_popup()->set_item_icon(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_3_VIEWPORTS_ALT), get_icon("Panels3Alt", "EditorIcons"));
  2863. view_menu->get_popup()->set_item_icon(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_4_VIEWPORTS), get_icon("Panels4", "EditorIcons"));
  2864. }
  2865. }
  2866. void SpatialEditor::add_control_to_menu_panel(Control *p_control) {
  2867. hbc_menu->add_child(p_control);
  2868. }
  2869. void SpatialEditor::set_can_preview(Camera *p_preview) {
  2870. for (int i = 0; i < 4; i++) {
  2871. viewports[i]->set_can_preview(p_preview);
  2872. }
  2873. }
  2874. VSplitContainer *SpatialEditor::get_shader_split() {
  2875. return shader_split;
  2876. }
  2877. HSplitContainer *SpatialEditor::get_palette_split() {
  2878. return palette_split;
  2879. }
  2880. void SpatialEditor::_request_gizmo(Object *p_obj) {
  2881. Spatial *sp = Object::cast_to<Spatial>(p_obj);
  2882. if (!sp)
  2883. return;
  2884. if (editor->get_edited_scene() && (sp == editor->get_edited_scene() || (sp->get_owner() && editor->get_edited_scene()->is_a_parent_of(sp)))) {
  2885. Ref<SpatialEditorGizmo> seg;
  2886. for (int i = 0; i < EditorNode::get_singleton()->get_editor_data().get_editor_plugin_count(); i++) {
  2887. seg = EditorNode::get_singleton()->get_editor_data().get_editor_plugin(i)->create_spatial_gizmo(sp);
  2888. if (seg.is_valid())
  2889. break;
  2890. }
  2891. if (!seg.is_valid()) {
  2892. seg = gizmos->get_gizmo(sp);
  2893. }
  2894. if (seg.is_valid()) {
  2895. sp->set_gizmo(seg);
  2896. }
  2897. if (seg.is_valid() && sp == selected) {
  2898. seg->set_selected(true);
  2899. selected->update_gizmo();
  2900. }
  2901. }
  2902. }
  2903. void SpatialEditor::_toggle_maximize_view(Object *p_viewport) {
  2904. if (!p_viewport) return;
  2905. SpatialEditorViewport *current_viewport = Object::cast_to<SpatialEditorViewport>(p_viewport);
  2906. if (!current_viewport) return;
  2907. int index = -1;
  2908. bool maximized = false;
  2909. for (int i = 0; i < 4; i++) {
  2910. if (viewports[i] == current_viewport) {
  2911. index = i;
  2912. if (current_viewport->get_global_rect() == viewport_base->get_global_rect())
  2913. maximized = true;
  2914. break;
  2915. }
  2916. }
  2917. if (index == -1) return;
  2918. if (!maximized) {
  2919. for (int i = 0; i < VIEWPORTS_COUNT; i++) {
  2920. if (i == index)
  2921. viewports[i]->set_area_as_parent_rect();
  2922. else
  2923. viewports[i]->hide();
  2924. }
  2925. } else {
  2926. for (int i = 0; i < VIEWPORTS_COUNT; i++)
  2927. viewports[i]->show();
  2928. if (view_menu->get_popup()->is_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_1_VIEWPORT)))
  2929. _menu_item_pressed(MENU_VIEW_USE_1_VIEWPORT);
  2930. else if (view_menu->get_popup()->is_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_2_VIEWPORTS)))
  2931. _menu_item_pressed(MENU_VIEW_USE_2_VIEWPORTS);
  2932. else if (view_menu->get_popup()->is_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_2_VIEWPORTS_ALT)))
  2933. _menu_item_pressed(MENU_VIEW_USE_2_VIEWPORTS_ALT);
  2934. else if (view_menu->get_popup()->is_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_3_VIEWPORTS)))
  2935. _menu_item_pressed(MENU_VIEW_USE_3_VIEWPORTS);
  2936. else if (view_menu->get_popup()->is_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_3_VIEWPORTS_ALT)))
  2937. _menu_item_pressed(MENU_VIEW_USE_3_VIEWPORTS_ALT);
  2938. else if (view_menu->get_popup()->is_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_USE_4_VIEWPORTS)))
  2939. _menu_item_pressed(MENU_VIEW_USE_4_VIEWPORTS);
  2940. }
  2941. }
  2942. void SpatialEditor::_node_removed(Node *p_node) {
  2943. if (p_node == selected)
  2944. selected = NULL;
  2945. }
  2946. void SpatialEditor::_bind_methods() {
  2947. //ClassDB::bind_method("_gui_input",&SpatialEditor::_gui_input);
  2948. ClassDB::bind_method("_unhandled_key_input", &SpatialEditor::_unhandled_key_input);
  2949. ClassDB::bind_method("_node_removed", &SpatialEditor::_node_removed);
  2950. ClassDB::bind_method("_menu_item_pressed", &SpatialEditor::_menu_item_pressed);
  2951. ClassDB::bind_method("_xform_dialog_action", &SpatialEditor::_xform_dialog_action);
  2952. ClassDB::bind_method("_get_editor_data", &SpatialEditor::_get_editor_data);
  2953. ClassDB::bind_method("_request_gizmo", &SpatialEditor::_request_gizmo);
  2954. ClassDB::bind_method("_toggle_maximize_view", &SpatialEditor::_toggle_maximize_view);
  2955. ADD_SIGNAL(MethodInfo("transform_key_request"));
  2956. }
  2957. void SpatialEditor::clear() {
  2958. settings_fov->set_value(EDITOR_DEF("editors/3d/default_fov", 55.0));
  2959. settings_znear->set_value(EDITOR_DEF("editors/3d/default_z_near", 0.1));
  2960. settings_zfar->set_value(EDITOR_DEF("editors/3d/default_z_far", 1500.0));
  2961. for (int i = 0; i < VIEWPORTS_COUNT; i++) {
  2962. viewports[i]->reset();
  2963. }
  2964. VisualServer::get_singleton()->instance_set_visible(origin_instance, true);
  2965. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_ORIGIN), true);
  2966. for (int i = 0; i < 3; ++i) {
  2967. if (grid_enable[i]) {
  2968. VisualServer::get_singleton()->instance_set_visible(grid_instance[i], true);
  2969. grid_visible[i] = true;
  2970. }
  2971. }
  2972. for (int i = 0; i < VIEWPORTS_COUNT; i++) {
  2973. viewports[i]->view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(SpatialEditorViewport::VIEW_AUDIO_LISTENER), i == 0);
  2974. viewports[i]->viewport->set_as_audio_listener(i == 0);
  2975. }
  2976. view_menu->get_popup()->set_item_checked(view_menu->get_popup()->get_item_index(MENU_VIEW_GRID), true);
  2977. }
  2978. SpatialEditor::SpatialEditor(EditorNode *p_editor) {
  2979. gizmo.visible = true;
  2980. gizmo.scale = 1.0;
  2981. viewport_environment = Ref<Environment>(memnew(Environment));
  2982. undo_redo = p_editor->get_undo_redo();
  2983. VBoxContainer *vbc = this;
  2984. custom_camera = NULL;
  2985. singleton = this;
  2986. editor = p_editor;
  2987. editor_selection = editor->get_editor_selection();
  2988. editor_selection->add_editor_plugin(this);
  2989. snap_enabled = false;
  2990. tool_mode = TOOL_MODE_SELECT;
  2991. //set_focus_mode(FOCUS_ALL);
  2992. hbc_menu = memnew(HBoxContainer);
  2993. vbc->add_child(hbc_menu);
  2994. Vector<Variant> button_binds;
  2995. button_binds.resize(1);
  2996. tool_button[TOOL_MODE_SELECT] = memnew(ToolButton);
  2997. hbc_menu->add_child(tool_button[TOOL_MODE_SELECT]);
  2998. tool_button[TOOL_MODE_SELECT]->set_toggle_mode(true);
  2999. tool_button[TOOL_MODE_SELECT]->set_flat(true);
  3000. tool_button[TOOL_MODE_SELECT]->set_pressed(true);
  3001. button_binds[0] = MENU_TOOL_SELECT;
  3002. tool_button[TOOL_MODE_SELECT]->connect("pressed", this, "_menu_item_pressed", button_binds);
  3003. tool_button[TOOL_MODE_SELECT]->set_tooltip(TTR("Select Mode (Q)\n") + keycode_get_string(KEY_MASK_CMD) + TTR("Drag: Rotate\nAlt+Drag: Move\nAlt+RMB: Depth list selection"));
  3004. tool_button[TOOL_MODE_MOVE] = memnew(ToolButton);
  3005. hbc_menu->add_child(tool_button[TOOL_MODE_MOVE]);
  3006. tool_button[TOOL_MODE_MOVE]->set_toggle_mode(true);
  3007. tool_button[TOOL_MODE_MOVE]->set_flat(true);
  3008. button_binds[0] = MENU_TOOL_MOVE;
  3009. tool_button[TOOL_MODE_MOVE]->connect("pressed", this, "_menu_item_pressed", button_binds);
  3010. tool_button[TOOL_MODE_MOVE]->set_tooltip(TTR("Move Mode (W)"));
  3011. tool_button[TOOL_MODE_ROTATE] = memnew(ToolButton);
  3012. hbc_menu->add_child(tool_button[TOOL_MODE_ROTATE]);
  3013. tool_button[TOOL_MODE_ROTATE]->set_toggle_mode(true);
  3014. tool_button[TOOL_MODE_ROTATE]->set_flat(true);
  3015. button_binds[0] = MENU_TOOL_ROTATE;
  3016. tool_button[TOOL_MODE_ROTATE]->connect("pressed", this, "_menu_item_pressed", button_binds);
  3017. tool_button[TOOL_MODE_ROTATE]->set_tooltip(TTR("Rotate Mode (E)"));
  3018. tool_button[TOOL_MODE_SCALE] = memnew(ToolButton);
  3019. hbc_menu->add_child(tool_button[TOOL_MODE_SCALE]);
  3020. tool_button[TOOL_MODE_SCALE]->set_toggle_mode(true);
  3021. tool_button[TOOL_MODE_SCALE]->set_flat(true);
  3022. button_binds[0] = MENU_TOOL_SCALE;
  3023. tool_button[TOOL_MODE_SCALE]->connect("pressed", this, "_menu_item_pressed", button_binds);
  3024. tool_button[TOOL_MODE_SCALE]->set_tooltip(TTR("Scale Mode (R)"));
  3025. VSeparator *vs = memnew(VSeparator);
  3026. hbc_menu->add_child(vs);
  3027. tool_button[TOOL_MODE_LIST_SELECT] = memnew(ToolButton);
  3028. hbc_menu->add_child(tool_button[TOOL_MODE_LIST_SELECT]);
  3029. tool_button[TOOL_MODE_LIST_SELECT]->set_toggle_mode(true);
  3030. tool_button[TOOL_MODE_LIST_SELECT]->set_flat(true);
  3031. button_binds[0] = MENU_TOOL_LIST_SELECT;
  3032. tool_button[TOOL_MODE_LIST_SELECT]->connect("pressed", this, "_menu_item_pressed", button_binds);
  3033. tool_button[TOOL_MODE_LIST_SELECT]->set_tooltip(TTR("Show a list of all objects at the position clicked\n(same as Alt+RMB in select mode)."));
  3034. vs = memnew(VSeparator);
  3035. hbc_menu->add_child(vs);
  3036. ED_SHORTCUT("spatial_editor/bottom_view", TTR("Bottom View"), KEY_MASK_ALT + KEY_KP_7);
  3037. ED_SHORTCUT("spatial_editor/top_view", TTR("Top View"), KEY_KP_7);
  3038. ED_SHORTCUT("spatial_editor/rear_view", TTR("Rear View"), KEY_MASK_ALT + KEY_KP_1);
  3039. ED_SHORTCUT("spatial_editor/front_view", TTR("Front View"), KEY_KP_1);
  3040. ED_SHORTCUT("spatial_editor/left_view", TTR("Left View"), KEY_MASK_ALT + KEY_KP_3);
  3041. ED_SHORTCUT("spatial_editor/right_view", TTR("Right View"), KEY_KP_3);
  3042. ED_SHORTCUT("spatial_editor/switch_perspective_orthogonal", TTR("Switch Perspective/Orthogonal view"), KEY_KP_5);
  3043. ED_SHORTCUT("spatial_editor/snap", TTR("Snap"), KEY_S);
  3044. ED_SHORTCUT("spatial_editor/insert_anim_key", TTR("Insert Animation Key"), KEY_K);
  3045. ED_SHORTCUT("spatial_editor/focus_origin", TTR("Focus Origin"), KEY_O);
  3046. ED_SHORTCUT("spatial_editor/focus_selection", TTR("Focus Selection"), KEY_F);
  3047. ED_SHORTCUT("spatial_editor/align_selection_with_view", TTR("Align Selection With View"), KEY_MASK_ALT + KEY_MASK_CMD + KEY_F);
  3048. ED_SHORTCUT("spatial_editor/tool_select", TTR("Tool Select"), KEY_Q);
  3049. ED_SHORTCUT("spatial_editor/tool_move", TTR("Tool Move"), KEY_W);
  3050. ED_SHORTCUT("spatial_editor/tool_rotate", TTR("Tool Rotate"), KEY_E);
  3051. ED_SHORTCUT("spatial_editor/tool_scale", TTR("Tool Scale"), KEY_R);
  3052. ED_SHORTCUT("spatial_editor/display_wireframe", TTR("Display Wireframe"), KEY_Z);
  3053. PopupMenu *p;
  3054. transform_menu = memnew(MenuButton);
  3055. transform_menu->set_text(TTR("Transform"));
  3056. hbc_menu->add_child(transform_menu);
  3057. p = transform_menu->get_popup();
  3058. p->add_check_shortcut(ED_SHORTCUT("spatial_editor/use_snap", TTR("Use Snap")), MENU_TRANSFORM_USE_SNAP);
  3059. p->add_shortcut(ED_SHORTCUT("spatial_editor/configure_snap", TTR("Configure Snap..")), MENU_TRANSFORM_CONFIGURE_SNAP);
  3060. p->add_separator();
  3061. p->add_check_shortcut(ED_SHORTCUT("spatial_editor/local_coords", TTR("Local Coords")), MENU_TRANSFORM_LOCAL_COORDS);
  3062. //p->set_item_checked(p->get_item_count()-1,true);
  3063. p->add_separator();
  3064. p->add_shortcut(ED_SHORTCUT("spatial_editor/transform_dialog", TTR("Transform Dialog..")), MENU_TRANSFORM_DIALOG);
  3065. p->connect("id_pressed", this, "_menu_item_pressed");
  3066. view_menu = memnew(MenuButton);
  3067. view_menu->set_text(TTR("View"));
  3068. view_menu->set_position(Point2(212, 0));
  3069. hbc_menu->add_child(view_menu);
  3070. p = view_menu->get_popup();
  3071. p->add_check_shortcut(ED_SHORTCUT("spatial_editor/1_viewport", TTR("1 Viewport"), KEY_MASK_CMD + KEY_1), MENU_VIEW_USE_1_VIEWPORT);
  3072. p->add_check_shortcut(ED_SHORTCUT("spatial_editor/2_viewports", TTR("2 Viewports"), KEY_MASK_CMD + KEY_2), MENU_VIEW_USE_2_VIEWPORTS);
  3073. p->add_check_shortcut(ED_SHORTCUT("spatial_editor/2_viewports_alt", TTR("2 Viewports (Alt)"), KEY_MASK_ALT + KEY_MASK_CMD + KEY_2), MENU_VIEW_USE_2_VIEWPORTS_ALT);
  3074. p->add_check_shortcut(ED_SHORTCUT("spatial_editor/3_viewports", TTR("3 Viewports"), KEY_MASK_CMD + KEY_3), MENU_VIEW_USE_3_VIEWPORTS);
  3075. p->add_check_shortcut(ED_SHORTCUT("spatial_editor/3_viewports_alt", TTR("3 Viewports (Alt)"), KEY_MASK_ALT + KEY_MASK_CMD + KEY_3), MENU_VIEW_USE_3_VIEWPORTS_ALT);
  3076. p->add_check_shortcut(ED_SHORTCUT("spatial_editor/4_viewports", TTR("4 Viewports"), KEY_MASK_CMD + KEY_4), MENU_VIEW_USE_4_VIEWPORTS);
  3077. p->add_separator();
  3078. p->add_check_shortcut(ED_SHORTCUT("spatial_editor/view_origin", TTR("View Origin")), MENU_VIEW_ORIGIN);
  3079. p->add_check_shortcut(ED_SHORTCUT("spatial_editor/view_grid", TTR("View Grid")), MENU_VIEW_GRID);
  3080. p->add_separator();
  3081. p->add_shortcut(ED_SHORTCUT("spatial_editor/settings", TTR("Settings")), MENU_VIEW_CAMERA_SETTINGS);
  3082. p->set_item_checked(p->get_item_index(MENU_VIEW_ORIGIN), true);
  3083. p->set_item_checked(p->get_item_index(MENU_VIEW_GRID), true);
  3084. p->connect("id_pressed", this, "_menu_item_pressed");
  3085. /* REST OF MENU */
  3086. palette_split = memnew(HSplitContainer);
  3087. palette_split->set_v_size_flags(SIZE_EXPAND_FILL);
  3088. vbc->add_child(palette_split);
  3089. shader_split = memnew(VSplitContainer);
  3090. shader_split->set_h_size_flags(SIZE_EXPAND_FILL);
  3091. palette_split->add_child(shader_split);
  3092. viewport_base = memnew(SpatialEditorViewportContainer);
  3093. shader_split->add_child(viewport_base);
  3094. viewport_base->set_v_size_flags(SIZE_EXPAND_FILL);
  3095. for (int i = 0; i < VIEWPORTS_COUNT; i++) {
  3096. viewports[i] = memnew(SpatialEditorViewport(this, editor, i));
  3097. viewports[i]->connect("toggle_maximize_view", this, "_toggle_maximize_view");
  3098. viewport_base->add_child(viewports[i]);
  3099. }
  3100. //vbc->add_child(viewport_base);
  3101. /* SNAP DIALOG */
  3102. snap_dialog = memnew(ConfirmationDialog);
  3103. snap_dialog->set_title(TTR("Snap Settings"));
  3104. add_child(snap_dialog);
  3105. VBoxContainer *snap_dialog_vbc = memnew(VBoxContainer);
  3106. snap_dialog->add_child(snap_dialog_vbc);
  3107. //snap_dialog->set_child_rect(snap_dialog_vbc);
  3108. snap_translate = memnew(LineEdit);
  3109. snap_translate->set_text("1");
  3110. snap_dialog_vbc->add_margin_child(TTR("Translate Snap:"), snap_translate);
  3111. snap_rotate = memnew(LineEdit);
  3112. snap_rotate->set_text("5");
  3113. snap_dialog_vbc->add_margin_child(TTR("Rotate Snap (deg.):"), snap_rotate);
  3114. snap_scale = memnew(LineEdit);
  3115. snap_scale->set_text("5");
  3116. snap_dialog_vbc->add_margin_child(TTR("Scale Snap (%):"), snap_scale);
  3117. /* SETTINGS DIALOG */
  3118. settings_dialog = memnew(ConfirmationDialog);
  3119. settings_dialog->set_title(TTR("Viewport Settings"));
  3120. add_child(settings_dialog);
  3121. settings_vbc = memnew(VBoxContainer);
  3122. settings_vbc->set_custom_minimum_size(Size2(200, 0) * EDSCALE);
  3123. settings_dialog->add_child(settings_vbc);
  3124. //settings_dialog->set_child_rect(settings_vbc);
  3125. settings_fov = memnew(SpinBox);
  3126. settings_fov->set_max(MAX_FOV);
  3127. settings_fov->set_min(MIN_FOV);
  3128. settings_fov->set_step(0.01);
  3129. settings_fov->set_value(EDITOR_DEF("editors/3d/default_fov", 55.0));
  3130. settings_vbc->add_margin_child(TTR("Perspective FOV (deg.):"), settings_fov);
  3131. settings_znear = memnew(SpinBox);
  3132. settings_znear->set_max(MAX_Z);
  3133. settings_znear->set_min(MIN_Z);
  3134. settings_znear->set_step(0.01);
  3135. settings_znear->set_value(EDITOR_DEF("editors/3d/default_z_near", 0.1));
  3136. settings_vbc->add_margin_child(TTR("View Z-Near:"), settings_znear);
  3137. settings_zfar = memnew(SpinBox);
  3138. settings_zfar->set_max(MAX_Z);
  3139. settings_zfar->set_min(MIN_Z);
  3140. settings_zfar->set_step(0.01);
  3141. settings_zfar->set_value(EDITOR_DEF("editors/3d/default_z_far", 1500));
  3142. settings_vbc->add_margin_child(TTR("View Z-Far:"), settings_zfar);
  3143. //settings_dialog->get_cancel()->hide();
  3144. /* XFORM DIALOG */
  3145. xform_dialog = memnew(ConfirmationDialog);
  3146. xform_dialog->set_title(TTR("Transform Change"));
  3147. add_child(xform_dialog);
  3148. VBoxContainer *xform_vbc = memnew(VBoxContainer);
  3149. xform_dialog->add_child(xform_vbc);
  3150. Label *l = memnew(Label);
  3151. l->set_text(TTR("Translate:"));
  3152. xform_vbc->add_child(l);
  3153. HBoxContainer *xform_hbc = memnew(HBoxContainer);
  3154. xform_vbc->add_child(xform_hbc);
  3155. for (int i = 0; i < 3; i++) {
  3156. xform_translate[i] = memnew(LineEdit);
  3157. xform_translate[i]->set_h_size_flags(SIZE_EXPAND_FILL);
  3158. xform_hbc->add_child(xform_translate[i]);
  3159. }
  3160. l = memnew(Label);
  3161. l->set_text(TTR("Rotate (deg.):"));
  3162. xform_vbc->add_child(l);
  3163. xform_hbc = memnew(HBoxContainer);
  3164. xform_vbc->add_child(xform_hbc);
  3165. for (int i = 0; i < 3; i++) {
  3166. xform_rotate[i] = memnew(LineEdit);
  3167. xform_rotate[i]->set_h_size_flags(SIZE_EXPAND_FILL);
  3168. xform_hbc->add_child(xform_rotate[i]);
  3169. }
  3170. l = memnew(Label);
  3171. l->set_text(TTR("Scale (ratio):"));
  3172. xform_vbc->add_child(l);
  3173. xform_hbc = memnew(HBoxContainer);
  3174. xform_vbc->add_child(xform_hbc);
  3175. for (int i = 0; i < 3; i++) {
  3176. xform_scale[i] = memnew(LineEdit);
  3177. xform_scale[i]->set_h_size_flags(SIZE_EXPAND_FILL);
  3178. xform_hbc->add_child(xform_scale[i]);
  3179. }
  3180. l = memnew(Label);
  3181. l->set_text(TTR("Transform Type"));
  3182. xform_vbc->add_child(l);
  3183. xform_type = memnew(OptionButton);
  3184. xform_type->set_h_size_flags(SIZE_EXPAND_FILL);
  3185. xform_type->add_item(TTR("Pre"));
  3186. xform_type->add_item(TTR("Post"));
  3187. xform_vbc->add_child(xform_type);
  3188. xform_dialog->connect("confirmed", this, "_xform_dialog_action");
  3189. scenario_debug = VisualServer::SCENARIO_DEBUG_DISABLED;
  3190. selected = NULL;
  3191. set_process_unhandled_key_input(true);
  3192. add_to_group("_spatial_editor_group");
  3193. EDITOR_DEF("editors/3d/manipulator_gizmo_size", 80);
  3194. EditorSettings::get_singleton()->add_property_hint(PropertyInfo(Variant::INT, "editors/3d/manipulator_gizmo_size", PROPERTY_HINT_RANGE, "16,1024,1"));
  3195. EDITOR_DEF("editors/3d/manipulator_gizmo_opacity", 0.2);
  3196. over_gizmo_handle = -1;
  3197. }
  3198. SpatialEditor::~SpatialEditor() {
  3199. }
  3200. void SpatialEditorPlugin::make_visible(bool p_visible) {
  3201. if (p_visible) {
  3202. spatial_editor->show();
  3203. spatial_editor->set_process(true);
  3204. //VisualServer::get_singleton()->viewport_set_hide_scenario(editor->get_scene_root()->get_viewport(),false);
  3205. spatial_editor->grab_focus();
  3206. } else {
  3207. spatial_editor->hide();
  3208. spatial_editor->set_process(false);
  3209. //VisualServer::get_singleton()->viewport_set_hide_scenario(editor->get_scene_root()->get_viewport(),true);
  3210. }
  3211. }
  3212. void SpatialEditorPlugin::edit(Object *p_object) {
  3213. spatial_editor->edit(Object::cast_to<Spatial>(p_object));
  3214. }
  3215. bool SpatialEditorPlugin::handles(Object *p_object) const {
  3216. return p_object->is_class("Spatial");
  3217. }
  3218. Dictionary SpatialEditorPlugin::get_state() const {
  3219. return spatial_editor->get_state();
  3220. }
  3221. void SpatialEditorPlugin::set_state(const Dictionary &p_state) {
  3222. spatial_editor->set_state(p_state);
  3223. }
  3224. void SpatialEditor::snap_cursor_to_plane(const Plane &p_plane) {
  3225. //cursor.pos=p_plane.project(cursor.pos);
  3226. }
  3227. void SpatialEditorPlugin::_bind_methods() {
  3228. ClassDB::bind_method("snap_cursor_to_plane", &SpatialEditorPlugin::snap_cursor_to_plane);
  3229. }
  3230. void SpatialEditorPlugin::snap_cursor_to_plane(const Plane &p_plane) {
  3231. spatial_editor->snap_cursor_to_plane(p_plane);
  3232. }
  3233. SpatialEditorPlugin::SpatialEditorPlugin(EditorNode *p_node) {
  3234. editor = p_node;
  3235. spatial_editor = memnew(SpatialEditor(p_node));
  3236. spatial_editor->set_v_size_flags(Control::SIZE_EXPAND_FILL);
  3237. editor->get_viewport()->add_child(spatial_editor);
  3238. //spatial_editor->set_area_as_parent_rect();
  3239. spatial_editor->hide();
  3240. spatial_editor->connect("transform_key_request", editor, "_transform_keyed");
  3241. //spatial_editor->set_process(true);
  3242. }
  3243. SpatialEditorPlugin::~SpatialEditorPlugin() {
  3244. }