EditorGraph.cpp 169 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715371637173718371937203721372237233724372537263727372837293730373137323733373437353736373737383739374037413742374337443745374637473748374937503751375237533754375537563757375837593760376137623763376437653766376737683769377037713772377337743775377637773778377937803781378237833784378537863787378837893790379137923793379437953796379737983799380038013802380338043805380638073808380938103811381238133814381538163817381838193820382138223823382438253826382738283829383038313832383338343835383638373838383938403841384238433844384538463847384838493850385138523853385438553856385738583859386038613862386338643865386638673868386938703871387238733874387538763877387838793880388138823883388438853886388738883889389038913892389338943895389638973898389939003901390239033904390539063907390839093910391139123913391439153916391739183919392039213922392339243925392639273928392939303931393239333934393539363937393839393940394139423943394439453946394739483949395039513952395339543955395639573958395939603961396239633964396539663967396839693970397139723973397439753976397739783979398039813982398339843985398639873988398939903991399239933994399539963997399839994000400140024003400440054006400740084009401040114012401340144015401640174018401940204021402240234024402540264027402840294030403140324033403440354036403740384039404040414042404340444045404640474048404940504051405240534054405540564057405840594060406140624063406440654066406740684069407040714072407340744075407640774078407940804081408240834084
  1. /*
  2. * Copyright (c) Contributors to the Open 3D Engine Project.
  3. * For complete copyright and license terms please see the LICENSE at the root of this distribution.
  4. *
  5. * SPDX-License-Identifier: Apache-2.0 OR MIT
  6. *
  7. */
  8. #include <AzCore/PlatformDef.h>
  9. AZ_PUSH_DISABLE_WARNING(4251 4800 4244, "-Wunknown-warning-option")
  10. #include <QScopedValueRollback>
  11. #include <QInputDialog>
  12. #include <QFile>
  13. #include <qmimedata.h>
  14. #include <QMessageBox>
  15. AZ_POP_DISABLE_WARNING
  16. #include <AzCore/Serialization/IdUtils.h>
  17. #include <AzCore/Asset/AssetManagerBus.h>
  18. #include <AzFramework/StringFunc/StringFunc.h>
  19. #include <AzQtComponents/Components/ToastNotification.h>
  20. #include <GraphCanvas/GraphCanvasBus.h>
  21. #include <GraphCanvas/Components/GridBus.h>
  22. #include <GraphCanvas/Components/Nodes/NodeBus.h>
  23. #include <GraphCanvas/Components/Slots/Data/DataSlotBus.h>
  24. #include <GraphCanvas/Components/GeometryBus.h>
  25. #include <GraphCanvas/Widgets/GraphCanvasMimeContainer.h>
  26. #include <GraphCanvas/Types/EntitySaveData.h>
  27. #include <GraphCanvas/Types/GraphCanvasGraphSerialization.h>
  28. #include <ScriptCanvas/Variable/GraphVariableManagerComponent.h>
  29. #include <ScriptCanvas/Components/EditorGraph.h>
  30. #include <ScriptCanvas/Bus/RequestBus.h>
  31. #include <ScriptCanvas/Bus/EditorScriptCanvasBus.h>
  32. #include <ScriptCanvas/Core/ConnectionBus.h>
  33. #include <ScriptCanvas/Core/GraphScopedTypes.h>
  34. #include <ScriptCanvas/Core/NodeBus.h>
  35. #include <ScriptCanvas/GraphCanvas/MappingBus.h>
  36. #include <ScriptCanvas/Libraries/Core/EBusEventHandler.h>
  37. #include <ScriptCanvas/Libraries/Core/FunctionDefinitionNode.h>
  38. #include <Editor/Include/ScriptCanvas/Components/NodeReplacementSystem.h>
  39. #include <Editor/Include/ScriptCanvas/GraphCanvas/MappingBus.h>
  40. #include <Editor/Include/ScriptCanvas/GraphCanvas/NodeDescriptorBus.h>
  41. #include <Editor/Nodes/NodeCreateUtils.h>
  42. #include <Editor/Nodes/NodeDisplayUtils.h>
  43. #include <Editor/GraphCanvas/GraphCanvasEditorNotificationBusId.h>
  44. #include <Editor/GraphCanvas/PropertySlotIds.h>
  45. #include <Editor/GraphCanvas/DataInterfaces/ScriptCanvasAssetIdDataInterface.h>
  46. #include <Editor/GraphCanvas/DataInterfaces/ScriptCanvasBoolDataInterface.h>
  47. #include <Editor/GraphCanvas/DataInterfaces/ScriptCanvasEntityIdDataInterface.h>
  48. #include <Editor/GraphCanvas/DataInterfaces/ScriptCanvasNumericDataInterface.h>
  49. #include <Editor/GraphCanvas/DataInterfaces/ScriptCanvasColorDataInterface.h>
  50. #include <Editor/GraphCanvas/DataInterfaces/ScriptCanvasCRCDataInterface.h>
  51. #include <Editor/GraphCanvas/DataInterfaces/ScriptCanvasStringDataInterface.h>
  52. #include <Editor/GraphCanvas/DataInterfaces/ScriptCanvasVectorDataInterface.h>
  53. #include <Editor/GraphCanvas/DataInterfaces/ScriptCanvasVariableDataInterface.h>
  54. #include <Editor/GraphCanvas/DataInterfaces/ScriptCanvasQuaternionDataInterface.h>
  55. #include <Editor/GraphCanvas/PropertyInterfaces/ScriptCanvasEnumComboBoxPropertyDataInterface.h>
  56. #include <Editor/GraphCanvas/PropertyInterfaces/ScriptCanvasStringPropertyDataInterface.h>
  57. #include <Editor/Translation/TranslationHelper.h>
  58. #include <Editor/View/Dialogs/SettingsDialog.h>
  59. #include <Editor/View/Widgets/ScriptCanvasNodePaletteDockWidget.h>
  60. #include <Editor/View/Widgets/NodePalette/EBusNodePaletteTreeItemTypes.h>
  61. #include <Editor/View/Widgets/NodePalette/ScriptEventsNodePaletteTreeItemTypes.h>
  62. #include <Editor/View/Widgets/NodePalette/GeneralNodePaletteTreeItemTypes.h>
  63. #include <Editor/View/Widgets/NodePalette/SpecializedNodePaletteTreeItemTypes.h>
  64. #include <Editor/View/Widgets/NodePalette/VariableNodePaletteTreeItemTypes.h>
  65. #include <Editor/View/Widgets/VariablePanel/GraphVariablesTableView.h>
  66. #include <Editor/View/Widgets/VariablePanel/VariableDockWidget.h>
  67. #include <Libraries/Core/GetVariable.h>
  68. #include <Libraries/Core/SetVariable.h>
  69. #include <Libraries/Core/Method.h>
  70. #include <Libraries/Core/MethodOverloaded.h>
  71. #include <Libraries/Core/EBusEventHandler.h>
  72. #include <Libraries/Core/ReceiveScriptEvent.h>
  73. #include <Libraries/Core/ScriptEventBase.h>
  74. #include <Libraries/Core/SendScriptEvent.h>
  75. #include <ScriptCanvas/Asset/RuntimeAsset.h>
  76. #include <ScriptCanvas/Core/Connection.h>
  77. #include <ScriptCanvas/Utils/NodeUtils.h>
  78. #include <ScriptCanvas/Variable/VariableBus.h>
  79. AZ_CVAR(bool, g_disableDeprecatedNodeUpdates, false, {}, AZ::ConsoleFunctorFlags::Null,
  80. "Disables automatic update attempts of deprecated nodes, so that graphs that require and update can be viewed in their original form");
  81. namespace EditorGraphCpp
  82. {
  83. enum Version
  84. {
  85. BeforeCovertedUnitTestNodes = 6,
  86. RemoveUnusedField,
  87. // label your entry above
  88. Current
  89. };
  90. }
  91. namespace ScriptCanvasEditor
  92. {
  93. EditorGraph::~EditorGraph()
  94. {
  95. for (auto& entry : m_graphCanvasSaveData)
  96. {
  97. delete entry.second;
  98. }
  99. m_graphCanvasSaveData.clear();
  100. delete m_graphCanvasSceneEntity;
  101. m_graphCanvasSceneEntity = nullptr;
  102. }
  103. static bool GraphVersionConverter(AZ::SerializeContext& context, AZ::SerializeContext::DataElementNode& rootDataElementNode)
  104. {
  105. // Version 0/1 graph will have their SaveFormatConverted flag flipped off
  106. if (rootDataElementNode.GetVersion() < 2)
  107. {
  108. rootDataElementNode.AddElementWithData(context, "m_saveFormatConverted", false);
  109. }
  110. if (rootDataElementNode.GetVersion() < 6)
  111. {
  112. rootDataElementNode.AddElementWithData(context, "GraphCanvasSaveVersion", GraphCanvas::EntitySaveDataContainer::NoVersion);
  113. }
  114. if (rootDataElementNode.GetVersion() < 7)
  115. {
  116. rootDataElementNode.RemoveElementByName(AZ_CRC_CE("m_pureDataNodesConvertedToVariables"));
  117. }
  118. // Always check and remove this unused field to keep asset clean
  119. if (rootDataElementNode.FindElement(AZ_CRC_CE("unitTestNodesConverted")) != -1)
  120. {
  121. rootDataElementNode.RemoveElementByName(AZ_CRC_CE("unitTestNodesConverted"));
  122. }
  123. return true;
  124. }
  125. void EditorGraph::Reflect(AZ::ReflectContext* context)
  126. {
  127. GraphStatisticsHelper::Reflect(context);
  128. auto serializeContext = azrtti_cast<AZ::SerializeContext*>(context);
  129. if (serializeContext)
  130. {
  131. serializeContext->Class<CRCCache>()
  132. ->Version(1)
  133. ->Field("String", &CRCCache::m_cacheValue)
  134. ->Field("Count", &CRCCache::m_cacheCount)
  135. ;
  136. serializeContext->Class<EditorGraph, ScriptCanvas::Graph>()
  137. ->Version(EditorGraphCpp::Version::Current, &GraphVersionConverter)
  138. ->Field("m_variableCounter", &EditorGraph::m_variableCounter)
  139. ->Field("m_saveFormatConverted", &EditorGraph::m_saveFormatConverted)
  140. ->Field("GraphCanvasData", &EditorGraph::m_graphCanvasSaveData)
  141. ->Field("CRCCacheMap", &EditorGraph::m_crcCacheMap)
  142. ->Field("StatisticsHelper", &EditorGraph::m_statisticsHelper)
  143. ->Field("GraphCanvasSaveVersion", &EditorGraph::m_graphCanvasSaveVersion)
  144. ;
  145. }
  146. }
  147. void EditorGraph::Activate()
  148. {
  149. const ScriptCanvas::ScriptCanvasId& scriptCanvasId = GetScriptCanvasId();
  150. // Overridden to prevent graph execution in the editor
  151. NodeCreationNotificationBus::Handler::BusConnect(scriptCanvasId);
  152. SceneCounterRequestBus::Handler::BusConnect(scriptCanvasId);
  153. EditorGraphRequestBus::Handler::BusConnect(scriptCanvasId);
  154. ScriptCanvas::GraphRequestBus::Handler::BusConnect(scriptCanvasId);
  155. ScriptCanvas::StatusRequestBus::Handler::BusConnect(scriptCanvasId);
  156. GraphItemCommandNotificationBus::Handler::BusConnect(scriptCanvasId);
  157. GeneralEditorNotificationBus::Handler::BusConnect(scriptCanvasId);
  158. ScriptCanvas::Graph::Activate();
  159. m_undoHelper.SetSource(this);
  160. }
  161. void EditorGraph::Deactivate()
  162. {
  163. GraphItemCommandNotificationBus::Handler::BusDisconnect();
  164. ScriptCanvas::GraphRequestBus::Handler::BusDisconnect();
  165. EditorGraphRequestBus::Handler::BusDisconnect();
  166. SceneCounterRequestBus::Handler::BusDisconnect();
  167. NodeCreationNotificationBus::Handler::BusDisconnect();
  168. AZ::SystemTickBus::Handler::BusDisconnect();
  169. GraphCanvas::SceneNotificationBus::Handler::BusDisconnect();
  170. GraphCanvas::GraphModelRequestBus::Handler::BusDisconnect();
  171. delete m_graphCanvasSceneEntity;
  172. m_graphCanvasSceneEntity = nullptr;
  173. }
  174. void EditorGraph::OnViewRegistered()
  175. {
  176. if (!m_saveFormatConverted)
  177. {
  178. ConstructSaveData();
  179. }
  180. }
  181. void EditorGraph::HandleFunctionDefinitionExtension(ScriptCanvas::Node* node, GraphCanvas::SlotId graphCanvasSlotId, const GraphCanvas::NodeId& nodeId)
  182. {
  183. // Special-case for the execution nodeling extensions, which are adding input/output data slots.
  184. // We want to automatically promote them to variables so that the user can refer to them more easily
  185. auto functionDefintionNode = azrtti_cast<ScriptCanvas::Nodes::Core::FunctionDefinitionNode*>(node);
  186. if (functionDefintionNode && graphCanvasSlotId.IsValid())
  187. {
  188. GraphCanvas::Endpoint endpoint;
  189. GraphCanvas::SlotRequestBus::EventResult(endpoint, graphCanvasSlotId, &GraphCanvas::SlotRequests::GetEndpoint);
  190. const ScriptCanvas::Endpoint scEndpoint = ConvertToScriptCanvasEndpoint(endpoint);
  191. if (scEndpoint.IsValid())
  192. {
  193. ScriptCanvas::Slot* slot = FindSlot(scEndpoint);
  194. if (slot)
  195. {
  196. AZ::Vector2 position;
  197. GraphCanvas::GeometryRequestBus::EventResult(position, nodeId, &GraphCanvas::GeometryRequests::GetPosition);
  198. // First we need to automatically display the ShowVariableConfigurationWidget dialog so the user
  199. // can assign a type and name to the slot they are adding
  200. VariablePaletteRequests::VariableConfigurationInput selectedSlotSetup;
  201. selectedSlotSetup.m_configurationVariableTitle = "Slot";
  202. selectedSlotSetup.m_createVariable = false;
  203. selectedSlotSetup.m_currentName = slot->GetName();
  204. selectedSlotSetup.m_currentType = slot->GetDataType();
  205. VariablePaletteRequests::VariableConfigurationOutput output;
  206. QPoint scenePoint(aznumeric_cast<int>(position.GetX()), aznumeric_cast<int>(position.GetY()));
  207. VariablePaletteRequestBus::BroadcastResult(output, &VariablePaletteRequests::ShowVariableConfigurationWidget
  208. , selectedSlotSetup, scenePoint);
  209. if (output.m_actionIsValid && output.m_type.IsValid())
  210. {
  211. if (slot)
  212. {
  213. slot->SetDisplayType(output.m_type);
  214. if (!output.m_name.empty())
  215. {
  216. slot->Rename(output.m_name);
  217. }
  218. }
  219. if (!node->GetGraph()->IsScriptEventExtension())
  220. {
  221. // Now that the slot has a valid type/name, we can actually promote it to a variable
  222. if (PromoteToVariableAction(endpoint, true))
  223. {
  224. ScriptCanvas::GraphVariable* variable = slot->GetVariable();
  225. if (variable)
  226. {
  227. if (variable->GetScope() != ScriptCanvas::VariableFlags::Scope::Function)
  228. {
  229. variable->SetScope(ScriptCanvas::VariableFlags::Scope::Function);
  230. }
  231. }
  232. }
  233. }
  234. }
  235. else
  236. {
  237. RemoveSlot(endpoint);
  238. }
  239. }
  240. }
  241. }
  242. }
  243. AZStd::optional<ScriptCanvas::NodeReplacementConfiguration> CreateVariableNodeThatRequiresUpdate
  244. ( ScriptCanvas::Node& node, const ScriptCanvas::VariableId& variableId, [[maybe_unused]] ScriptCanvas::ScriptCanvasId scriptCanvasGraphId)
  245. {
  246. if (auto getVarNode = azrtti_cast<ScriptCanvas::Nodes::Core::GetVariableNode*>(&node);
  247. getVarNode && getVarNode->GetId() == variableId)
  248. {
  249. ScriptCanvas::NodeReplacementConfiguration config;
  250. config.create = [variableId](const ScriptCanvas::Node& oldNode) ->ScriptCanvas::Node*
  251. {
  252. return Nodes::CreateGetVariableNodeResult(variableId, oldNode.GetOwningScriptCanvasId()).node;
  253. };
  254. config.m_type = azrtti_typeid<ScriptCanvas::Nodes::Core::GetVariableNode>();
  255. return config;
  256. }
  257. if (auto setVarNode = azrtti_cast<ScriptCanvas::Nodes::Core::SetVariableNode*>(&node);
  258. setVarNode && setVarNode->GetId() == variableId)
  259. {
  260. ScriptCanvas::NodeReplacementConfiguration config;
  261. config.create = [variableId](const ScriptCanvas::Node& oldNode) ->ScriptCanvas::Node*
  262. {
  263. return Nodes::CreateSetVariableNodeResult(variableId, oldNode.GetOwningScriptCanvasId()).node;
  264. };
  265. config.m_type = azrtti_typeid<ScriptCanvas::Nodes::Core::SetVariableNode>();
  266. return config;
  267. }
  268. return AZStd::nullopt;
  269. }
  270. void EditorGraph::RefreshVariableReferences(const ScriptCanvas::VariableId& variableId)
  271. {
  272. const auto variable = FindVariableById(variableId);
  273. if (!variable)
  274. {
  275. AZ_Warning("ScriptCanvas", false
  276. , "EditorGraph::::RefreshVariableReferences called with variableId that did not refer to a variable");
  277. return;
  278. }
  279. ReplacementInfoByNode replacementInfoByNewNode;
  280. AZStd::vector<ScriptCanvas::Node*> remainingNodes;
  281. {
  282. ReplacementInfoByNode replacementInfoByOldNode;
  283. auto nodeEntities = GetNodeEntities();
  284. AZStd::vector<ScriptCanvas::Node*> nodes;
  285. nodes.reserve(nodeEntities.size());
  286. // first discover all nodes that need a change
  287. // get their states, and cache their replacement configuration
  288. for (auto nodeEntity : nodeEntities)
  289. {
  290. if (auto node = FindNode(nodeEntity->GetId()))
  291. {
  292. if (auto configOptional = CreateVariableNodeThatRequiresUpdate(*node, variableId, GetScriptCanvasId()))
  293. {
  294. auto slotStateOutcome = GetSlotState(*node);
  295. if (!slotStateOutcome.IsSuccess())
  296. {
  297. AZ_Error("ScriptCanvas"
  298. , false
  299. , "Could not complete type change of variable. Failed to get slot state from to-be-replaced Node %s: "
  300. , node->GetNodeName().c_str()
  301. , slotStateOutcome.GetError().c_str());
  302. return;
  303. }
  304. ReplacementInfo info;
  305. info.config = AZStd::move(*configOptional);
  306. info.oldNodeId = node->GetEntityId();
  307. info.slotStates = slotStateOutcome.TakeValue();
  308. replacementInfoByOldNode[node->GetEntityId()] = AZStd::move(info);
  309. }
  310. nodes.push_back(node);
  311. }
  312. }
  313. // replace all required nodes, keep a list of the nodes that will remain
  314. for (auto node : nodes)
  315. {
  316. if (auto iter = replacementInfoByOldNode.find(node->GetEntityId()); iter != replacementInfoByOldNode.end())
  317. {
  318. const auto nodeName = node->GetNodeName();
  319. auto replaceOutcome = ReplaceLiveNode(*node, iter->second.config);
  320. if (replaceOutcome.IsSuccess())
  321. {
  322. auto newNode = replaceOutcome.GetValue();
  323. replacementInfoByNewNode[newNode->GetEntityId()] = iter->second;
  324. node = newNode;
  325. }
  326. else
  327. {
  328. AZ_Error("ScriptCanvas"
  329. , false
  330. , "Could not complete type change of variable. Failed to update node: %s - %s"
  331. , nodeName.c_str()
  332. , replaceOutcome.GetError().c_str());
  333. return;
  334. }
  335. }
  336. remainingNodes.push_back(node);
  337. }
  338. }
  339. const auto variableType = variable->GetDataType();
  340. ScriptCanvas::EndpointsResolved referenceSlots;
  341. for (auto node : remainingNodes)
  342. {
  343. // clear all required slots
  344. for (auto slot : node->ModAllSlots())
  345. {
  346. if (slot->IsData() && slot->IsVariableReference() && slot->GetVariableReference() == variableId)
  347. {
  348. if (!azrtti_cast<const ScriptCanvas::Nodes::Core::FunctionDefinitionNode*>(node))
  349. {
  350. const auto scriptCanvasEndpoint = ScriptCanvas::Endpoint(node->GetEntityId(), slot->GetId());
  351. const auto graphCanvasEndpoint = ConvertToGraphCanvasEndpoint(scriptCanvasEndpoint);
  352. bool convertedToValue = false;
  353. GraphCanvas::DataSlotRequestBus::EventResult
  354. ( convertedToValue
  355. , graphCanvasEndpoint.GetSlotId()
  356. , &GraphCanvas::DataSlotRequests::ConvertToValue);
  357. if (!convertedToValue)
  358. {
  359. AZ_Error("ScriptCanvas", false, "A reference to the type changed variable failed to convert back to value, future type checking will fail");
  360. }
  361. }
  362. referenceSlots.push_back({node, slot});
  363. }
  364. }
  365. }
  366. // clear display types
  367. for (auto& resolvedEndpoint : referenceSlots)
  368. {
  369. auto node = resolvedEndpoint.first;
  370. auto slot = const_cast<ScriptCanvas::Slot*>(resolvedEndpoint.second);
  371. if (!azrtti_cast<const ScriptCanvas::Nodes::Core::FunctionDefinitionNode*>(node))
  372. {
  373. slot->ClearDisplayType();
  374. }
  375. }
  376. // set all required slots back to references
  377. for (auto& resolvedEndpoint : referenceSlots)
  378. {
  379. auto node = resolvedEndpoint.first;
  380. auto slot = const_cast<ScriptCanvas::Slot*>(resolvedEndpoint.second);
  381. if (azrtti_cast<const ScriptCanvas::Nodes::Core::FunctionDefinitionNode*>(node))
  382. {
  383. slot->SetVariableReference(variableId, ScriptCanvas::Slot::IsVariableTypeChange::Yes);
  384. }
  385. else
  386. {
  387. if (node->SlotAcceptsType(slot->GetId(), variableType).IsSuccess())
  388. {
  389. slot->SetVariableReference(variableId, ScriptCanvas::Slot::IsVariableTypeChange::Yes);
  390. }
  391. else
  392. {
  393. AZ_Warning("SC", node->SlotAcceptsType(slot->GetId(), variableType).IsSuccess(), "");
  394. }
  395. }
  396. }
  397. // update slots (except for connection data) on replaced nodes
  398. for (auto& replaceInfo : replacementInfoByNewNode)
  399. {
  400. auto newNode = FindNode(replaceInfo.first);
  401. if (!newNode)
  402. {
  403. AZ_Error("ScriptCanvas", false, "Could not complete type change of variable. Failed to find new node just added to graph");
  404. return;
  405. }
  406. auto slotStateUpdateOutcome = UpdateSlotState(*newNode, replaceInfo.second.config, replaceInfo.second.slotStates, replacementInfoByNewNode, FixConnections::No);
  407. if (!slotStateUpdateOutcome.IsSuccess())
  408. {
  409. AZ_Error("ScriptCanvas"
  410. , false
  411. , "Could not complete type change of variable. Failed to relplace connections for %s: %s"
  412. , newNode->GetNodeName().c_str()
  413. , slotStateUpdateOutcome.GetError().c_str());
  414. return;
  415. }
  416. }
  417. // finally, replace all possible old connections that required replacement, due to one or both ndoes on the ends being replaced
  418. for (auto& replaceInfo : replacementInfoByNewNode)
  419. {
  420. auto newNode = FindNode(replaceInfo.first);
  421. if (!newNode)
  422. {
  423. AZ_Error("ScriptCanvas", false, "Could not complete type change of variable. Failed to find new node just added to graph");
  424. return;
  425. }
  426. auto slotStateUpdateOutcome = UpdateSlotState(*newNode, replaceInfo.second.config, replaceInfo.second.slotStates, replacementInfoByNewNode, FixConnections::Yes);
  427. if (!slotStateUpdateOutcome.IsSuccess())
  428. {
  429. AZ_Error("ScriptCanvas"
  430. , false
  431. , "Could not complete type change of variable. Failed to relplace connections for %s: %s"
  432. , newNode->GetNodeName().c_str()
  433. , slotStateUpdateOutcome.GetError().c_str());
  434. return;
  435. }
  436. }
  437. }
  438. AZ::Outcome<ScriptCanvas::Node*, AZStd::string> EditorGraph::ReplaceLiveNode
  439. ( ScriptCanvas::Node& oldNode
  440. , ScriptCanvas::NodeReplacementConfiguration& nodeConfig)
  441. {
  442. ScriptCanvas::Node* returnNode = nodeConfig.create ? nodeConfig.create(oldNode) : nullptr;
  443. if (!returnNode)
  444. {
  445. return AZ::Failure(AZStd::string("ReplaceLiveNode: Failure to create replacement node from replacement configuration create function"));
  446. }
  447. AZ::EntityId graphCanvasGraphId = GetGraphCanvasGraphId();
  448. AZ::EntityId oldNodeGraphCanvasId;
  449. SceneMemberMappingRequestBus::EventResult(oldNodeGraphCanvasId, oldNode.GetEntityId(), &SceneMemberMappingRequests::GetGraphCanvasEntityId);
  450. const auto wasDisabled = oldNode.GetNodeDisabledFlag();
  451. AZ::Vector2 position(0, 0);
  452. GraphCanvas::GeometryRequestBus::EventResult(position, oldNodeGraphCanvasId, &GraphCanvas::GeometryRequests::GetPosition);
  453. AZStd::unordered_set<AZ::EntityId> oldNodeGraphCanvasIds;
  454. oldNodeGraphCanvasIds.insert(oldNodeGraphCanvasId);
  455. GraphCanvas::SceneRequestBus::Event(graphCanvasGraphId, &GraphCanvas::SceneRequests::Delete, oldNodeGraphCanvasIds);
  456. // ScriptCanvas::Node& oldNode is now deleted
  457. AZ::EntityId newNodeGraphCanvasId;
  458. SceneMemberMappingRequestBus::EventResult(newNodeGraphCanvasId, returnNode->GetEntityId(), &SceneMemberMappingRequests::GetGraphCanvasEntityId);
  459. GraphCanvas::SceneRequestBus::Event(graphCanvasGraphId, &GraphCanvas::SceneRequests::AddNode, newNodeGraphCanvasId, position, false);
  460. returnNode->SetNodeDisabledFlag(wasDisabled);
  461. return AZ::Success(returnNode);
  462. }
  463. AZ::Outcome<EditorGraph::LiveSlotInfo, AZStd::string> EditorGraph::ConvertToLiveStateInfo
  464. ( const ScriptCanvas::Node& node
  465. , const ScriptCanvas::Slot& nodeSlot) const
  466. {
  467. LiveSlotInfo info;
  468. ScriptCanvas::SlotState& slotState = info.state;
  469. slotState.type = nodeSlot.GetType();
  470. slotState.name = nodeSlot.GetName();
  471. info.oldEndpoint = ScriptCanvas::Endpoint(node.GetEntityId(), nodeSlot.GetId());
  472. if (IsData(slotState.type))
  473. {
  474. slotState.value.SetType(nodeSlot.GetDataType());
  475. if (nodeSlot.IsVariableReference())
  476. {
  477. slotState.variableReference = nodeSlot.GetVariableReference();
  478. }
  479. else if (slotState.type == ScriptCanvas::CombinedSlotType::DataIn)
  480. {
  481. if (!nodeSlot.IsConnected())
  482. {
  483. if (auto datum = nodeSlot.FindDatum())
  484. {
  485. slotState.value.DeepCopyDatum(*datum);
  486. }
  487. else
  488. {
  489. return AZ::Failure(AZStd::string::format
  490. ( "EditorGraph::ConvertToLiveStateInfo Failed to copy over required value from Slot: %s"
  491. , slotState.name.c_str()));
  492. }
  493. }
  494. else
  495. {
  496. slotState.value.SetToDefaultValueOfType();
  497. }
  498. }
  499. info.isGetSetVariableDataSlot = node.GetVariableInputSlot() == &nodeSlot || node.GetVariableOutputSlot() == &nodeSlot;
  500. }
  501. auto endpointsResolved = node.GetConnectedNodes(nodeSlot);
  502. // all the connects will be deleted by the type the debug name functions need to be called, so supply them now
  503. for (const auto& connection : endpointsResolved)
  504. {
  505. info.connections.push_back(ScriptCanvas::Endpoint(connection.first->GetEntityId(), connection.second->GetId()));
  506. info.connectionNames.push_back(AZStd::string::format("%s-%s", connection.first->GetNodeName().c_str(), connection.second->GetName().c_str()));
  507. }
  508. return AZ::Success(info);
  509. }
  510. AZ::Outcome<EditorGraph::LiveSlotStates, AZStd::string> EditorGraph::GetSlotState(const ScriptCanvas::Node& node) const
  511. {
  512. EditorGraph::LiveSlotStates slotStates;
  513. auto nodeSlots = node.GetAllSlots();
  514. for (auto nodeSlot : nodeSlots)
  515. {
  516. if (!nodeSlot)
  517. {
  518. return AZ::Failure(AZStd::string("EditorGraph::GetSlotState null slot in Node %s list: ", node.GetNodeName().c_str()));
  519. }
  520. auto liveSlotInfoOutcome = ConvertToLiveStateInfo(node, *nodeSlot);
  521. if (!liveSlotInfoOutcome.IsSuccess())
  522. {
  523. return AZ::Failure(liveSlotInfoOutcome.TakeError());
  524. }
  525. slotStates.push_back(liveSlotInfoOutcome.TakeValue());
  526. }
  527. return AZ::Success(slotStates);
  528. }
  529. EditorGraph::LiveSlotInfo* EditorGraph::FindMatchingSlotState
  530. ( ScriptCanvas::Node& node
  531. , ScriptCanvas::Slot& slot
  532. , LiveSlotStates& slotState) const
  533. {
  534. const bool isGetSetVariableDataSlot = node.GetVariableInputSlot() == &slot || node.GetVariableOutputSlot() == &slot;
  535. const auto slotName = slot.GetName();
  536. const auto slotType = slot.GetType();
  537. auto iter = AZStd::find_if(slotState.begin(), slotState.end()
  538. , [&slotName, &slotType, &isGetSetVariableDataSlot](LiveSlotInfo& info)
  539. {
  540. return info.state.type == slotType
  541. && ((info.isGetSetVariableDataSlot && isGetSetVariableDataSlot)
  542. || info.state.name == slotName);
  543. });
  544. return iter != slotState.end() ? iter : nullptr;
  545. }
  546. AZ::Outcome<void, AZStd::string> EditorGraph::UpdateSlotConnections
  547. ( ScriptCanvas::Node& node
  548. , ScriptCanvas::Slot& slot
  549. , [[maybe_unused]] const ScriptCanvas::NodeReplacementConfiguration& nodeConfig
  550. , LiveSlotInfo& oldSlotInfo
  551. , const ReplacementInfoByNode& infoByNewNode)
  552. {
  553. const auto newSCEndpoint = ScriptCanvas::Endpoint(node.GetEntityId(), slot.GetId());
  554. const auto newGCEndpoint = ConvertToGraphCanvasEndpoint(newSCEndpoint);
  555. for (size_t i = 0; i != oldSlotInfo.connections.size(); ++i)
  556. {
  557. AZ::Entity* unusedEntity{};
  558. auto previouslyConnectedSCEndpoint = oldSlotInfo.connections[i];
  559. // if endpoint.GetNodeId() is in infoByNewNode, find a new endpoint for connection
  560. const auto previouslyConnectedNodeId = previouslyConnectedSCEndpoint.GetNodeId();
  561. auto oldNodeInfoIter = AZStd::find_if(infoByNewNode.begin(), infoByNewNode.end()
  562. , [previouslyConnectedNodeId](const auto& info)
  563. {
  564. return info.second.oldNodeId == previouslyConnectedNodeId;
  565. });
  566. if (oldNodeInfoIter != infoByNewNode.end())
  567. {
  568. // look through the previously connected, now deleted node's slot info list...
  569. const auto& connectionOldInfos = oldNodeInfoIter->second.slotStates;
  570. auto oldConnectionMatch = AZStd::find_if(connectionOldInfos.begin(), connectionOldInfos.end(), [&previouslyConnectedSCEndpoint](const auto& oldConnection)
  571. {
  572. // ...for an endpoint to this new node's replaced node's endpoint...
  573. return previouslyConnectedSCEndpoint == oldConnection.oldEndpoint;
  574. });
  575. // ...a match should be found...
  576. if (oldConnectionMatch != connectionOldInfos.end())
  577. {
  578. // ...so we take the NEW node-slot endpoint from the match.
  579. previouslyConnectedSCEndpoint = oldConnectionMatch->newEndpoint;
  580. }
  581. else
  582. {
  583. AZ_Error("ScriptCanvas", false, "failed to map old connection endpoint to new one");
  584. continue;
  585. }
  586. }
  587. // another replaced node may have already restored this connection
  588. if (!FindConnection(unusedEntity, newSCEndpoint, previouslyConnectedSCEndpoint))
  589. {
  590. const auto previouslyConnectedGCEndpoint = ConvertToGraphCanvasEndpoint(previouslyConnectedSCEndpoint);
  591. GraphCanvas::SlotRequestBus::Event
  592. ( newGCEndpoint.GetSlotId()
  593. , &GraphCanvas::SlotRequests::CreateConnectionWithEndpoint
  594. , previouslyConnectedGCEndpoint);
  595. if (!FindConnection(unusedEntity, newSCEndpoint, previouslyConnectedSCEndpoint))
  596. {
  597. // let me see it happen this time...
  598. GraphCanvas::SlotRequestBus::Event
  599. ( newGCEndpoint.GetSlotId()
  600. , &GraphCanvas::SlotRequests::CreateConnectionWithEndpoint
  601. , previouslyConnectedGCEndpoint);
  602. return AZ::Failure(AZStd::string::format
  603. ( "EditorGraph::UpdateSlotConnections Failed to restore connection between %s-%s and %s."
  604. , node.GetNodeName().c_str()
  605. , slot.GetName().c_str()
  606. , oldSlotInfo.connectionNames[i].c_str()));
  607. }
  608. }
  609. }
  610. return AZ::Success();
  611. }
  612. AZ::Outcome<void, AZStd::string> EditorGraph::UpdateSlotDatum
  613. ( ScriptCanvas::Node& node
  614. , ScriptCanvas::Slot& slot
  615. , LiveSlotInfo& slotInfo)
  616. {
  617. if (!IsData(slot.GetType()))
  618. {
  619. return AZ::Success();
  620. }
  621. if (!slotInfo.connections.empty())
  622. {
  623. return AZ::Success();
  624. }
  625. else if (slotInfo.state.variableReference.IsValid())
  626. {
  627. slot.SetVariableReference(slotInfo.state.variableReference);
  628. return AZ::Success();
  629. }
  630. else if (slotInfo.state.value.GetType() == slot.GetDataType())
  631. {
  632. ScriptCanvas::ModifiableDatumView view;
  633. if (slot.FindModifiableDatumView(view))
  634. {
  635. view.HardCopyDatum(slotInfo.state.value);
  636. return AZ::Success();
  637. }
  638. return AZ::Failure(AZStd::string::format("EditorGraph::UpdateSlotDatum Failed to find datum for %s-%s to copy over data from replaced node."
  639. , node.GetNodeName().c_str(), slot.GetName().c_str()));
  640. }
  641. return AZ::Success();
  642. }
  643. AZ::Outcome<void, AZStd::string> EditorGraph::UpdateSlotState
  644. ( ScriptCanvas::Node& node
  645. , ScriptCanvas::Slot& slot
  646. , const ScriptCanvas::NodeReplacementConfiguration& nodeConfig
  647. , LiveSlotStates& slotState
  648. , const ReplacementInfoByNode& infoByNewNode
  649. , FixConnections fixConnections)
  650. {
  651. // find a match for the slot in the slot state
  652. auto match = FindMatchingSlotState(node, slot, slotState);
  653. if (!match)
  654. {
  655. const auto msg = AZStd::string::format
  656. ( "EditorGraph::UpdateSlotState No previous slot match found for slot: %s-%s"
  657. , node.GetNodeName().c_str()
  658. , slot.GetName().c_str());
  659. AZ_Warning("ScriptCanvas", !nodeConfig.m_warnOnToleratedErrors, msg.c_str());
  660. if (nodeConfig.m_tolerateNoMatchingPreviousSlot)
  661. {
  662. return AZ::Success();
  663. }
  664. else
  665. {
  666. return AZ::Failure(msg);
  667. }
  668. }
  669. if (fixConnections == FixConnections::No)
  670. {
  671. AZ_Error("ScriptCanvas", !match->newEndpoint.IsValid(), "The matching slot state has already been initialized");
  672. match->newEndpoint = ScriptCanvas::Endpoint(node.GetEntityId(), slot.GetId());
  673. // update based on type / values
  674. auto updateDataOutcome = UpdateSlotDatum(node, slot, *match);
  675. if (!updateDataOutcome.IsSuccess())
  676. {
  677. const auto msg = AZStd::string::format
  678. ( "EditorGraph::UpdateSlotState Failed to datum: %s-%s, from previous slot"
  679. , node.GetNodeName().c_str()
  680. , slot.GetName().c_str());
  681. AZ_Warning("ScriptCanvas", !nodeConfig.m_warnOnToleratedErrors, msg.c_str());
  682. if (!nodeConfig.m_tolerateFailureToUpdateData)
  683. {
  684. return AZ::Failure(msg);
  685. }
  686. }
  687. }
  688. else
  689. {
  690. // replace old connections
  691. auto updateConnectionsOutcome = UpdateSlotConnections(node, slot, nodeConfig, *match, infoByNewNode);
  692. if (!updateConnectionsOutcome.IsSuccess())
  693. {
  694. const auto msg = AZStd::string::format
  695. ( "EditorGraph::UpdateSlotState Failed to update slot connections: %s-%s, from previous slot"
  696. , node.GetNodeName().c_str()
  697. , slot.GetName().c_str());
  698. AZ_Warning("ScriptCanvas", !nodeConfig.m_warnOnToleratedErrors, msg.c_str());
  699. if (nodeConfig.m_tolerateFailureToReplaceConnections)
  700. {
  701. return AZ::Failure(msg);
  702. }
  703. }
  704. }
  705. return AZ::Success();
  706. }
  707. AZ::Outcome<void, AZStd::string> EditorGraph::UpdateSlotState
  708. ( ScriptCanvas::Node& node
  709. , const ScriptCanvas::NodeReplacementConfiguration& nodeConfig
  710. , LiveSlotStates& slotState
  711. , const ReplacementInfoByNode& infoByNewNode
  712. , FixConnections fixConnections)
  713. {
  714. auto nodeSlots = node.ModAllSlots();
  715. for (auto nodeSlot : nodeSlots)
  716. {
  717. if (!nodeSlot)
  718. {
  719. return AZ::Failure(AZStd::string::format
  720. ( "EditorGraph::UpdateSlotState null slot in Node %s list: "
  721. , node.GetNodeName().c_str()));
  722. }
  723. auto slotOutcome = UpdateSlotState(node, *nodeSlot, nodeConfig, slotState, infoByNewNode, fixConnections);
  724. if (!slotOutcome.IsSuccess() && !nodeConfig.m_tolerateIndividualSlotUpdateFailures)
  725. {
  726. return AZ::Failure(AZStd::string::format
  727. ( "EditorGraph::UpdateSlotState Slot failed to update: %s-%s, %s"
  728. , node.GetNodeName().c_str()
  729. , nodeSlot->GetName().c_str()
  730. , slotOutcome.GetError().c_str()));
  731. }
  732. }
  733. return AZ::Success();
  734. }
  735. void EditorGraph::OnEntitiesSerialized(GraphCanvas::GraphSerialization& serializationTarget)
  736. {
  737. const GraphCanvas::GraphData& graphCanvasGraphData = serializationTarget.GetGraphData();
  738. AZStd::unordered_set<ScriptCanvas::VariableId> variableIds;
  739. AZStd::unordered_set< AZ::EntityId > forcedWrappedNodes;
  740. AZStd::unordered_set<AZ::Entity*> scriptCanvasEntities;
  741. for (const auto& node : graphCanvasGraphData.m_nodes)
  742. {
  743. // EBus Event nodes are purely visual, but require some user data manipulation in order to function correctly.
  744. // As such we don't want to copy over their script canvas user data, since it's not what was intended to be copied.
  745. if (EBusHandlerEventNodeDescriptorRequestBus::FindFirstHandler(node->GetId()) == nullptr)
  746. {
  747. AZStd::any* userData = nullptr;
  748. GraphCanvas::NodeRequestBus::EventResult(userData, node->GetId(), &GraphCanvas::NodeRequests::GetUserData);
  749. auto scriptCanvasNodeId = userData->is<AZ::EntityId>() ? *AZStd::any_cast<AZ::EntityId>(userData) : AZ::EntityId();
  750. AZ::Entity* scriptCanvasEntity{};
  751. AZ::ComponentApplicationBus::BroadcastResult(scriptCanvasEntity, &AZ::ComponentApplicationRequests::FindEntity, scriptCanvasNodeId);
  752. if (scriptCanvasEntity)
  753. {
  754. scriptCanvasEntities.emplace(scriptCanvasEntity);
  755. ScriptCanvas::Node* nodeComponent = FindNode(scriptCanvasEntity->GetId());
  756. if (nodeComponent)
  757. {
  758. for (const auto& slot : nodeComponent->GetSlots())
  759. {
  760. if (slot.IsVariableReference())
  761. {
  762. variableIds.insert(slot.GetVariableReference());
  763. }
  764. }
  765. }
  766. }
  767. if (GraphCanvas::ForcedWrappedNodeRequestBus::FindFirstHandler(node->GetId()) != nullptr)
  768. {
  769. forcedWrappedNodes.insert(node->GetId());
  770. }
  771. }
  772. else
  773. {
  774. forcedWrappedNodes.insert(node->GetId());
  775. }
  776. }
  777. if (!variableIds.empty())
  778. {
  779. auto& userDataMapRef = serializationTarget.GetUserDataMapRef();
  780. auto mapIter = userDataMapRef.find(ScriptCanvas::CopiedVariableData::k_variableKey);
  781. ScriptCanvas::GraphVariableMapping* variableConfigurations = nullptr;
  782. if (mapIter == userDataMapRef.end())
  783. {
  784. ScriptCanvas::CopiedVariableData variableData;
  785. auto insertResult = userDataMapRef.emplace(ScriptCanvas::CopiedVariableData::k_variableKey, variableData);
  786. ScriptCanvas::CopiedVariableData* copiedVariableData = AZStd::any_cast<ScriptCanvas::CopiedVariableData>(&insertResult.first->second);
  787. variableConfigurations = (&copiedVariableData->m_variableMapping);
  788. }
  789. else
  790. {
  791. ScriptCanvas::CopiedVariableData* copiedVariableData = AZStd::any_cast<ScriptCanvas::CopiedVariableData>(&mapIter->second);
  792. variableConfigurations = (&copiedVariableData->m_variableMapping);
  793. }
  794. for (const auto& variableId : variableIds)
  795. {
  796. if (variableConfigurations->find(variableId) == variableConfigurations->end())
  797. {
  798. ScriptCanvas::ScriptCanvasId scriptCanvasId;
  799. GeneralRequestBus::BroadcastResult(scriptCanvasId, &GeneralRequests::GetActiveScriptCanvasId);
  800. ScriptCanvas::GraphVariable* configuration = nullptr;
  801. ScriptCanvas::GraphVariableManagerRequestBus::EventResult(configuration, scriptCanvasId, &ScriptCanvas::GraphVariableManagerRequests::FindVariableById, variableId);
  802. if (configuration)
  803. {
  804. variableConfigurations->emplace(variableId, (*configuration));
  805. }
  806. }
  807. }
  808. }
  809. for (const auto& connection : graphCanvasGraphData.m_connections)
  810. {
  811. AZStd::any* userData = nullptr;
  812. GraphCanvas::ConnectionRequestBus::EventResult(userData, connection->GetId(), &GraphCanvas::ConnectionRequests::GetUserData);
  813. auto scriptCanvasConnectionId = userData->is<AZ::EntityId>() ? *AZStd::any_cast<AZ::EntityId>(userData) : AZ::EntityId();
  814. AZ::Entity* scriptCanvasEntity{};
  815. AZ::ComponentApplicationBus::BroadcastResult(scriptCanvasEntity, &AZ::ComponentApplicationRequests::FindEntity, scriptCanvasConnectionId);
  816. if (scriptCanvasEntity)
  817. {
  818. scriptCanvasEntities.emplace(scriptCanvasEntity);
  819. }
  820. }
  821. auto& userDataMap = serializationTarget.GetUserDataMapRef();
  822. AZStd::unordered_set<AZ::Entity*> graphData = CopyItems(scriptCanvasEntities);
  823. userDataMap.emplace(EditorGraph::GetMimeType(), graphData);
  824. if (!forcedWrappedNodes.empty())
  825. {
  826. // Keep track of which ebus methods were grouped together when we serialized them out.
  827. // This is so when we recreate them, we can create the appropriate number of
  828. // EBus wrappers and put the correct methods into each.
  829. WrappedNodeGroupingMap forcedWrappedNodeGroupings;
  830. for (const AZ::EntityId& wrappedNode : forcedWrappedNodes)
  831. {
  832. AZ::EntityId wrapperNode;
  833. GraphCanvas::NodeRequestBus::EventResult(wrapperNode, wrappedNode, &GraphCanvas::NodeRequests::GetWrappingNode);
  834. if (wrapperNode.IsValid())
  835. {
  836. forcedWrappedNodeGroupings.emplace(wrappedNode, wrapperNode);
  837. }
  838. }
  839. userDataMap.emplace(EditorGraph::GetWrappedNodeGroupingMimeType(), forcedWrappedNodeGroupings);
  840. }
  841. }
  842. void EditorGraph::OnEntitiesDeserialized(const GraphCanvas::GraphSerialization& serializationSource)
  843. {
  844. const auto& userDataMap = serializationSource.GetUserDataMapRef();
  845. auto userDataIt = userDataMap.find(EditorGraph::GetMimeType());
  846. if (userDataIt != userDataMap.end())
  847. {
  848. auto graphEntities(AZStd::any_cast<AZStd::unordered_set<AZ::Entity*>>(&userDataIt->second));
  849. if (graphEntities)
  850. {
  851. AddItems(*graphEntities);
  852. const ScriptCanvas::GraphVariableMapping* variableMapping = nullptr;
  853. userDataIt = userDataMap.find(ScriptCanvas::CopiedVariableData::k_variableKey);
  854. if (userDataIt != userDataMap.end())
  855. {
  856. const ScriptCanvas::CopiedVariableData* copiedVariableData = AZStd::any_cast<ScriptCanvas::CopiedVariableData>(&userDataIt->second);
  857. variableMapping = (&copiedVariableData->m_variableMapping);
  858. }
  859. if (variableMapping)
  860. {
  861. for (AZ::Entity* entity : (*graphEntities))
  862. {
  863. ScriptCanvas::Node* node = FindNode(entity->GetId());
  864. if (node)
  865. {
  866. for (const auto& slot : node->GetSlots())
  867. {
  868. if (slot.IsVariableReference())
  869. {
  870. ScriptCanvas::VariableId originalId = slot.GetVariableReference();
  871. ScriptCanvas::GraphVariable* variable = FindVariableById(originalId);
  872. if (variable == nullptr)
  873. {
  874. auto variableIter = variableMapping->find(originalId);
  875. if (variableIter != variableMapping->end())
  876. {
  877. const ScriptCanvas::GraphVariable& variableConfiguration = variableIter->second;
  878. AZ::Outcome<ScriptCanvas::VariableId, AZStd::string> remapVariableOutcome = AZ::Failure(AZStd::string());
  879. ScriptCanvas::GraphVariableManagerRequestBus::EventResult(remapVariableOutcome, GetScriptCanvasId(), &ScriptCanvas::GraphVariableManagerRequests::RemapVariable, variableConfiguration);
  880. if (remapVariableOutcome)
  881. {
  882. node->SetSlotVariableId(slot.GetId(), remapVariableOutcome.GetValue());
  883. }
  884. else
  885. {
  886. node->ClearSlotVariableId(slot.GetId());
  887. }
  888. }
  889. else
  890. {
  891. node->ClearSlotVariableId(slot.GetId());
  892. }
  893. }
  894. }
  895. }
  896. }
  897. }
  898. }
  899. }
  900. }
  901. userDataIt = userDataMap.find(EditorGraph::GetWrappedNodeGroupingMimeType());
  902. if (userDataIt != userDataMap.end())
  903. {
  904. // Serialization system handled remapping this map data so we can just insert them into our map.
  905. const WrappedNodeGroupingMap* wrappedNodeGroupings = AZStd::any_cast<WrappedNodeGroupingMap>(&userDataIt->second);
  906. m_wrappedNodeGroupings.insert(wrappedNodeGroupings->begin(), wrappedNodeGroupings->end());
  907. }
  908. const GraphCanvas::GraphData& sceneData = serializationSource.GetGraphData();
  909. for (auto nodeEntity : sceneData.m_nodes)
  910. {
  911. NodeCreationNotificationBus::Event(GetScriptCanvasId(), &NodeCreationNotifications::OnGraphCanvasNodeCreated, nodeEntity->GetId());
  912. }
  913. }
  914. void EditorGraph::DisconnectConnection(const GraphCanvas::ConnectionId& connectionId)
  915. {
  916. AZStd::any* connectionUserData = nullptr;
  917. GraphCanvas::ConnectionRequestBus::EventResult(connectionUserData, connectionId, &GraphCanvas::ConnectionRequests::GetUserData);
  918. auto scConnectionId = connectionUserData && connectionUserData->is<AZ::EntityId>()
  919. ? *AZStd::any_cast<AZ::EntityId>(connectionUserData)
  920. : AZ::EntityId();
  921. if (auto connectionComponent = AZ::EntityUtils::FindFirstDerivedComponent<ScriptCanvas::Connection>(scConnectionId))
  922. {
  923. ScriptCanvas::Endpoint sourceEndpoint = connectionComponent->GetSourceEndpoint();
  924. ScriptCanvas::Endpoint targetEndpoint = connectionComponent->GetTargetEndpoint();
  925. ScriptCanvas::GraphNotificationBus::Event
  926. ( GetScriptCanvasId()
  927. , &ScriptCanvas::GraphNotifications::OnDisonnectionComplete
  928. , connectionId);
  929. DisconnectById(scConnectionId);
  930. UpdateCorrespondingImplicitConnection(sourceEndpoint, targetEndpoint);
  931. }
  932. }
  933. ScriptCanvas::DataPtr EditorGraph::Create()
  934. {
  935. if (AZ::Entity* entity = aznew AZ::Entity("Script Canvas Graph"))
  936. {
  937. auto graph = entity->CreateComponent<ScriptCanvasEditor::EditorGraph>();
  938. entity->CreateComponent<EditorGraphVariableManagerComponent>(graph->GetScriptCanvasId());
  939. if (ScriptCanvas::DataPtr data = aznew ScriptCanvas::ScriptCanvasData())
  940. {
  941. data->m_scriptCanvasEntity.reset(entity);
  942. graph->MarkOwnership(*data);
  943. graph->MarkVersion();
  944. entity->Init();
  945. entity->Activate();
  946. return data;
  947. }
  948. }
  949. return nullptr;
  950. }
  951. bool EditorGraph::CreateConnection(const GraphCanvas::ConnectionId& connectionId, const GraphCanvas::Endpoint& sourcePoint, const GraphCanvas::Endpoint& targetPoint)
  952. {
  953. if (!sourcePoint.IsValid() || !targetPoint.IsValid())
  954. {
  955. return false;
  956. }
  957. DisconnectConnection(connectionId);
  958. bool scConnected = false;
  959. ScriptCanvas::Endpoint scSourceEndpoint = ConvertToScriptCanvasEndpoint(sourcePoint);
  960. ScriptCanvas::Endpoint scTargetEndpoint = ConvertToScriptCanvasEndpoint(targetPoint);
  961. scConnected = ConnectByEndpoint(scSourceEndpoint, scTargetEndpoint);
  962. if (scConnected)
  963. {
  964. scConnected = ConfigureConnectionUserData(scSourceEndpoint, scTargetEndpoint, connectionId);
  965. }
  966. if (scConnected)
  967. {
  968. ScriptCanvas::GraphNotificationBus::Event(GetScriptCanvasId(), &ScriptCanvas::GraphNotifications::OnConnectionComplete, connectionId);
  969. }
  970. UpdateCorrespondingImplicitConnection(scSourceEndpoint, scTargetEndpoint);
  971. return scConnected;
  972. }
  973. void EditorGraph::UpdateCorrespondingImplicitConnection(const ScriptCanvas::Endpoint& sourceEndpoint, const ScriptCanvas::Endpoint& targetEndpoint)
  974. {
  975. auto sourceSlot = FindSlot(sourceEndpoint);
  976. auto targetSlot = FindSlot(targetEndpoint);
  977. if (!sourceSlot || !targetSlot)
  978. {
  979. return;
  980. }
  981. if (sourceSlot->IsExecution() || targetSlot->IsExecution())
  982. {
  983. return;
  984. }
  985. const ScriptCanvas::Node* sourceNode = FindNode(sourceEndpoint.GetNodeId());
  986. const ScriptCanvas::Node* targetNode = FindNode(targetEndpoint.GetNodeId());
  987. if (!sourceNode || !targetNode)
  988. {
  989. return;
  990. }
  991. const ScriptCanvas::Slot* sourceNodeExecutionSlot = sourceNode->GetCorrespondingExecutionSlot(FindSlot(sourceEndpoint));
  992. const ScriptCanvas::Slot* targetNodeExecutionSlot = targetNode->GetCorrespondingExecutionSlot(FindSlot(targetEndpoint));
  993. if (!sourceNodeExecutionSlot || !targetNodeExecutionSlot)
  994. {
  995. return;
  996. }
  997. // If either the source or target slot execution slots on these nodes are implicit, then check if the implicit connection should be updated
  998. if (sourceNodeExecutionSlot->CreatesImplicitConnections() || targetNodeExecutionSlot->CreatesImplicitConnections())
  999. {
  1000. const ScriptCanvas::Endpoint& implicitSourceEndpoint = sourceNodeExecutionSlot->GetEndpoint();
  1001. const ScriptCanvas::Endpoint& implicitTargetEndpoint = targetNodeExecutionSlot->GetEndpoint();
  1002. // If a connection exists between the provided endpoints, try to create an implicit connection
  1003. if (FindConnection(sourceEndpoint, targetEndpoint))
  1004. {
  1005. if (!FindConnection(implicitSourceEndpoint, implicitTargetEndpoint))
  1006. {
  1007. ConnectByEndpoint(implicitSourceEndpoint, implicitTargetEndpoint);
  1008. }
  1009. }
  1010. // If a connection doesn't exist between the provided endpoints, check if the implicit connection should be removed
  1011. else
  1012. {
  1013. AZStd::vector<const ScriptCanvas::Slot*> sourceNodeDataSlots = sourceNode->GetCorrespondingDataSlots(sourceNodeExecutionSlot);
  1014. AZStd::vector<const ScriptCanvas::Slot*> targetNodeDataSlots = targetNode->GetCorrespondingDataSlots(targetNodeExecutionSlot);
  1015. int numDataConnectionsBetween = 0;
  1016. // Count the number of data connections between the nodes
  1017. for (const ScriptCanvas::Slot* sourceDataSlot : sourceNodeDataSlots)
  1018. {
  1019. for (const ScriptCanvas::Slot* targetDataSlot : targetNodeDataSlots)
  1020. {
  1021. if (FindConnection(sourceDataSlot->GetEndpoint(), targetDataSlot->GetEndpoint()))
  1022. {
  1023. numDataConnectionsBetween++;
  1024. }
  1025. }
  1026. }
  1027. // If there are no connections, remove the implicit execution connection
  1028. if (numDataConnectionsBetween == 0)
  1029. {
  1030. DisconnectByEndpoint(implicitSourceEndpoint, implicitTargetEndpoint);
  1031. }
  1032. }
  1033. }
  1034. }
  1035. bool EditorGraph::IsValidConnection(const GraphCanvas::Endpoint& sourcePoint, const GraphCanvas::Endpoint& targetPoint) const
  1036. {
  1037. ScriptCanvas::Endpoint scSourceEndpoint = ConvertToScriptCanvasEndpoint(sourcePoint);
  1038. ScriptCanvas::Endpoint scTargetEndpoint = ConvertToScriptCanvasEndpoint(targetPoint);
  1039. return CanCreateConnectionBetween(scSourceEndpoint, scTargetEndpoint).IsSuccess();
  1040. }
  1041. AZStd::string EditorGraph::GetDataTypeString(const AZ::Uuid&)
  1042. {
  1043. // This is used by the default tooltip setting in GraphCanvas, returning an empty string
  1044. // in order for tooltips to be fully controlled by ScriptCanvas
  1045. return {};
  1046. }
  1047. void EditorGraph::OnRemoveUnusedNodes()
  1048. {
  1049. }
  1050. void EditorGraph::OnRemoveUnusedElements()
  1051. {
  1052. RemoveUnusedVariables();
  1053. }
  1054. bool EditorGraph::AllowReset(const GraphCanvas::Endpoint& endpoint) const
  1055. {
  1056. ScriptCanvas::Endpoint scEndpoint = ConvertToScriptCanvasEndpoint(endpoint);
  1057. ScriptCanvas::Node* node = FindNode(scEndpoint.GetNodeId());
  1058. if (node)
  1059. {
  1060. const ScriptCanvas::Slot* slot = node->GetSlot(scEndpoint.GetSlotId());
  1061. if (slot)
  1062. {
  1063. if (slot->IsVariableReference())
  1064. {
  1065. return !slot->IsUserAdded();
  1066. }
  1067. else
  1068. {
  1069. const ScriptCanvas::Datum* datum = node->FindDatum(scEndpoint.GetSlotId());
  1070. if (datum)
  1071. {
  1072. // BCO's create a reference when set to default. Going to bypass them for now.
  1073. return ScriptCanvas::Data::IsValueType(datum->GetType());
  1074. }
  1075. }
  1076. }
  1077. }
  1078. return false;
  1079. }
  1080. GraphCanvas::NodePropertyDisplay* EditorGraph::CreateDataSlotPropertyDisplay(const AZ::Uuid& dataType, const GraphCanvas::NodeId& nodeId, const GraphCanvas::SlotId& slotId) const
  1081. {
  1082. (void)dataType;
  1083. AZStd::any* nodeUserData = nullptr;
  1084. GraphCanvas::NodeRequestBus::EventResult(nodeUserData, nodeId, &GraphCanvas::NodeRequests::GetUserData);
  1085. auto scriptCanvasNodeId = nodeUserData && nodeUserData->is<AZ::EntityId>() ? *AZStd::any_cast<AZ::EntityId>(nodeUserData) : AZ::EntityId();
  1086. AZStd::any* slotUserData = nullptr;
  1087. GraphCanvas::SlotRequestBus::EventResult(slotUserData, slotId, &GraphCanvas::SlotRequests::GetUserData);
  1088. auto scriptCanvasSlotId = slotUserData && slotUserData->is<ScriptCanvas::SlotId>() ? *AZStd::any_cast<ScriptCanvas::SlotId>(slotUserData) : ScriptCanvas::SlotId();
  1089. return CreateDisplayPropertyForSlot(scriptCanvasNodeId, scriptCanvasSlotId);
  1090. }
  1091. GraphCanvas::NodePropertyDisplay* EditorGraph::CreatePropertySlotPropertyDisplay(const AZ::Crc32& propertyId, const GraphCanvas::NodeId& nodeId, const GraphCanvas::NodeId& slotId) const
  1092. {
  1093. (void)slotId;
  1094. AZStd::any* nodeUserData = nullptr;
  1095. GraphCanvas::NodeRequestBus::EventResult(nodeUserData, nodeId, &GraphCanvas::NodeRequests::GetUserData);
  1096. auto scriptCanvasNodeId = nodeUserData && nodeUserData->is<AZ::EntityId>() ? *AZStd::any_cast<AZ::EntityId>(nodeUserData) : AZ::EntityId();
  1097. ScriptCanvas::Node* node = AZ::EntityUtils::FindFirstDerivedComponent<ScriptCanvas::Node>(scriptCanvasNodeId);
  1098. if (node)
  1099. {
  1100. ScriptCanvas::NodePropertyInterface* propertyInterface = node->GetPropertyInterface(propertyId);
  1101. if (propertyInterface)
  1102. {
  1103. GraphCanvas::DataInterface* dataInterface = nullptr;
  1104. GraphCanvas::NodePropertyDisplay* dataDisplay = nullptr;
  1105. if (azrtti_cast<ScriptCanvas::ComboBoxPropertyInterface*>(propertyInterface))
  1106. {
  1107. GraphCanvas::ComboBoxDataInterface* comboBoxInterface = nullptr;
  1108. if (propertyInterface->GetDataType() == ScriptCanvas::Data::Type::BehaviorContextObject(ScriptCanvas::EnumComboBoxNodePropertyInterface::k_EnumUUID))
  1109. {
  1110. comboBoxInterface = aznew ScriptCanvasEnumComboBoxPropertyDataInterface(scriptCanvasNodeId, static_cast<ScriptCanvas::EnumComboBoxNodePropertyInterface*>(propertyInterface));
  1111. }
  1112. if (comboBoxInterface)
  1113. {
  1114. dataInterface = comboBoxInterface;
  1115. GraphCanvas::GraphCanvasRequestBus::BroadcastResult(dataDisplay, &GraphCanvas::GraphCanvasRequests::CreateComboBoxNodePropertyDisplay, comboBoxInterface);
  1116. }
  1117. }
  1118. else
  1119. {
  1120. switch (propertyInterface->GetDataType().GetType())
  1121. {
  1122. case ScriptCanvas::Data::eType::String:
  1123. dataInterface = aznew ScriptCanvasStringPropertyDataInterface(scriptCanvasNodeId, static_cast<ScriptCanvas::TypedNodePropertyInterface<ScriptCanvas::Data::StringType>*>(propertyInterface));
  1124. GraphCanvas::GraphCanvasRequestBus::BroadcastResult(dataDisplay, &GraphCanvas::GraphCanvasRequests::CreateStringNodePropertyDisplay, static_cast<GraphCanvas::StringDataInterface*>(dataInterface));
  1125. break;
  1126. default:
  1127. break;
  1128. }
  1129. }
  1130. if (dataDisplay != nullptr)
  1131. {
  1132. return dataDisplay;
  1133. }
  1134. delete dataInterface;
  1135. }
  1136. }
  1137. return nullptr;
  1138. }
  1139. AZ::EntityId EditorGraph::ConvertToScriptCanvasNodeId(const GraphCanvas::NodeId& nodeId) const
  1140. {
  1141. AZStd::any* userData = nullptr;
  1142. GraphCanvas::NodeRequestBus::EventResult(userData, nodeId, &GraphCanvas::NodeRequests::GetUserData);
  1143. return (userData && userData->is<AZ::EntityId>()) ? *AZStd::any_cast<AZ::EntityId>(userData) : AZ::EntityId();
  1144. }
  1145. GraphCanvas::NodePropertyDisplay* EditorGraph::CreateDisplayPropertyForSlot(const AZ::EntityId& scriptCanvasNodeId, const ScriptCanvas::SlotId& scriptCanvasSlotId) const
  1146. {
  1147. ScriptCanvas::Slot* slot = nullptr;
  1148. ScriptCanvas::NodeRequestBus::EventResult(slot, scriptCanvasNodeId, &ScriptCanvas::NodeRequests::GetSlot, scriptCanvasSlotId);
  1149. if (slot == nullptr || slot->IsUserAdded())
  1150. {
  1151. return nullptr;
  1152. }
  1153. if (slot->IsVariableReference())
  1154. {
  1155. ScriptCanvasVariableReferenceDataInterface* dataInterface = aznew ScriptCanvasVariableReferenceDataInterface(&m_variableDataModel, GetScriptCanvasId(), scriptCanvasNodeId, scriptCanvasSlotId);
  1156. GraphCanvas::NodePropertyDisplay* dataDisplay = nullptr;
  1157. GraphCanvas::GraphCanvasRequestBus::BroadcastResult(dataDisplay, &GraphCanvas::GraphCanvasRequests::CreateComboBoxNodePropertyDisplay, dataInterface);
  1158. if (dataDisplay)
  1159. {
  1160. return dataDisplay;
  1161. }
  1162. delete dataInterface;
  1163. return nullptr;
  1164. }
  1165. if (!slot->CanHaveInputField())
  1166. {
  1167. return nullptr;
  1168. }
  1169. // ScriptCanvas has access to better typing information regarding the slots than is exposed to GraphCanvas.
  1170. // So let ScriptCanvas check the types based on it's own information rather than relying on the information passed back from GraphCanvas.
  1171. ScriptCanvas::Data::Type slotType = slot->GetDataType();
  1172. GraphCanvas::DataInterface* dataInterface = nullptr;
  1173. {
  1174. GraphCanvas::NodePropertyDisplay* dataDisplay = nullptr;
  1175. if (slotType.IS_A(ScriptCanvas::Data::Type::Boolean()))
  1176. {
  1177. dataInterface = aznew ScriptCanvasBoolDataInterface(scriptCanvasNodeId, scriptCanvasSlotId);
  1178. GraphCanvas::GraphCanvasRequestBus::BroadcastResult(dataDisplay, &GraphCanvas::GraphCanvasRequests::CreateBooleanNodePropertyDisplay, static_cast<ScriptCanvasBoolDataInterface*>(dataInterface));
  1179. }
  1180. else if (slotType.IS_A(ScriptCanvas::Data::Type::Number()))
  1181. {
  1182. dataInterface = aznew ScriptCanvasNumericDataInterface(scriptCanvasNodeId, scriptCanvasSlotId);
  1183. GraphCanvas::GraphCanvasRequestBus::BroadcastResult(dataDisplay, &GraphCanvas::GraphCanvasRequests::CreateNumericNodePropertyDisplay, static_cast<ScriptCanvasNumericDataInterface*>(dataInterface));
  1184. }
  1185. else if (slotType.IS_A(ScriptCanvas::Data::Type::String()))
  1186. {
  1187. dataInterface = aznew ScriptCanvasStringDataInterface(scriptCanvasNodeId, scriptCanvasSlotId);
  1188. GraphCanvas::GraphCanvasRequestBus::BroadcastResult(dataDisplay, &GraphCanvas::GraphCanvasRequests::CreateStringNodePropertyDisplay, static_cast<ScriptCanvasStringDataInterface*>(dataInterface));
  1189. }
  1190. else if (slotType.IS_A(ScriptCanvas::Data::Type::EntityID()))
  1191. {
  1192. dataInterface = aznew ScriptCanvasEntityIdDataInterface(scriptCanvasNodeId, scriptCanvasSlotId);
  1193. GraphCanvas::GraphCanvasRequestBus::BroadcastResult(dataDisplay, &GraphCanvas::GraphCanvasRequests::CreateEntityIdNodePropertyDisplay, static_cast<ScriptCanvasEntityIdDataInterface*>(dataInterface));
  1194. }
  1195. else if (slotType.IS_A(ScriptCanvas::Data::Type::BehaviorContextObject(AZ::Vector3::TYPEINFO_Uuid()))
  1196. || slotType.IS_A(ScriptCanvas::Data::Type::Vector3()))
  1197. {
  1198. dataInterface = aznew ScriptCanvasVectorDataInterface<AZ::Vector3, 3>(scriptCanvasNodeId, scriptCanvasSlotId);
  1199. GraphCanvas::GraphCanvasRequestBus::BroadcastResult(dataDisplay, &GraphCanvas::GraphCanvasRequests::CreateVectorNodePropertyDisplay, static_cast<GraphCanvas::VectorDataInterface*>(dataInterface));
  1200. }
  1201. else if (slotType.IS_A(ScriptCanvas::Data::Type::BehaviorContextObject(AZ::Vector2::TYPEINFO_Uuid()))
  1202. || slotType.IS_A(ScriptCanvas::Data::Type::Vector2()))
  1203. {
  1204. dataInterface = aznew ScriptCanvasVectorDataInterface<AZ::Vector2, 2>(scriptCanvasNodeId, scriptCanvasSlotId);
  1205. GraphCanvas::GraphCanvasRequestBus::BroadcastResult(dataDisplay, &GraphCanvas::GraphCanvasRequests::CreateVectorNodePropertyDisplay, static_cast<GraphCanvas::VectorDataInterface*>(dataInterface));
  1206. }
  1207. else if (slotType.IS_A(ScriptCanvas::Data::Type::BehaviorContextObject(AZ::Vector4::TYPEINFO_Uuid()))
  1208. || slotType.IS_A(ScriptCanvas::Data::Type::Vector4()))
  1209. {
  1210. dataInterface = aznew ScriptCanvasVectorDataInterface<AZ::Vector4, 4>(scriptCanvasNodeId, scriptCanvasSlotId);
  1211. GraphCanvas::GraphCanvasRequestBus::BroadcastResult(dataDisplay, &GraphCanvas::GraphCanvasRequests::CreateVectorNodePropertyDisplay, static_cast<GraphCanvas::VectorDataInterface*>(dataInterface));
  1212. }
  1213. else if (slotType.IS_A(ScriptCanvas::Data::Type::BehaviorContextObject(AZ::Quaternion::TYPEINFO_Uuid()))
  1214. || slotType.IS_A(ScriptCanvas::Data::Type::Quaternion()))
  1215. {
  1216. dataInterface = aznew ScriptCanvasQuaternionDataInterface(scriptCanvasNodeId, scriptCanvasSlotId);
  1217. GraphCanvas::GraphCanvasRequestBus::BroadcastResult(dataDisplay, &GraphCanvas::GraphCanvasRequests::CreateVectorNodePropertyDisplay, static_cast<GraphCanvas::VectorDataInterface*>(dataInterface));
  1218. }
  1219. else if (slotType.IS_A(ScriptCanvas::Data::Type::BehaviorContextObject(AZ::Color::TYPEINFO_Uuid()))
  1220. || slotType.IS_A(ScriptCanvas::Data::Type::Color()))
  1221. {
  1222. dataInterface = aznew ScriptCanvasColorDataInterface(scriptCanvasNodeId, scriptCanvasSlotId);
  1223. GraphCanvas::GraphCanvasRequestBus::BroadcastResult(dataDisplay, &GraphCanvas::GraphCanvasRequests::CreateVectorNodePropertyDisplay, static_cast<GraphCanvas::VectorDataInterface*>(dataInterface));
  1224. }
  1225. else if (slotType.IS_A(ScriptCanvas::Data::Type::CRC()))
  1226. {
  1227. dataInterface = aznew ScriptCanvasCRCDataInterface(scriptCanvasNodeId, scriptCanvasSlotId);
  1228. GraphCanvas::GraphCanvasRequestBus::BroadcastResult(dataDisplay, &GraphCanvas::GraphCanvasRequests::CreateStringNodePropertyDisplay, static_cast<GraphCanvas::StringDataInterface*>(dataInterface));
  1229. }
  1230. else if (slotType.IS_A(ScriptCanvas::Data::Type::AssetId()))
  1231. {
  1232. dataInterface = aznew ScriptCanvasAssetIdDataInterface(scriptCanvasNodeId, scriptCanvasSlotId);
  1233. if (ScriptCanvas::Nodes::Core::Method* method = azrtti_cast<ScriptCanvas::Nodes::Core::Method*>(slot->GetNode()))
  1234. {
  1235. // Try to find the AssetType attribute
  1236. if (AZ::Attribute* assetTypeAttribute = FindAttribute(AZ::Script::Attributes::AssetType, method->GetMethod()->m_attributes))
  1237. {
  1238. AZ::AttributeReader attributeReader(nullptr, assetTypeAttribute);
  1239. AZ::Data::AssetType assetType;
  1240. attributeReader.Read<AZ::Data::AssetType>(assetType);
  1241. ScriptCanvasAssetIdDataInterface* assetIdinterface = static_cast<ScriptCanvasAssetIdDataInterface*>(dataInterface);
  1242. assetIdinterface->SetAssetType(assetType);
  1243. }
  1244. if (AZ::Attribute* sourceAssetFilterAttribute = FindAttribute(AZ::Edit::Attributes::SourceAssetFilterPattern, method->GetMethod()->m_attributes))
  1245. {
  1246. AZ::AttributeReader attributeReader(nullptr, sourceAssetFilterAttribute);
  1247. AZStd::string filterPattern;
  1248. attributeReader.Read<AZStd::string>(filterPattern);
  1249. ScriptCanvasAssetIdDataInterface* assetIdinterface = static_cast<ScriptCanvasAssetIdDataInterface*>(dataInterface);
  1250. assetIdinterface->SetStringFilter(filterPattern);
  1251. }
  1252. }
  1253. GraphCanvas::GraphCanvasRequestBus::BroadcastResult(dataDisplay, &GraphCanvas::GraphCanvasRequests::CreateAssetIdNodePropertyDisplay, static_cast<GraphCanvas::AssetIdDataInterface*>(dataInterface));
  1254. }
  1255. else if (slotType.IS_A(ScriptCanvas::Data::Type::BehaviorContextObject(ScriptCanvas::GraphScopedVariableId::TYPEINFO_Uuid())))
  1256. {
  1257. dataInterface = aznew ScriptCanvasGraphScopedVariableDataInterface(&m_variableDataModel, GetScriptCanvasId(), scriptCanvasNodeId, scriptCanvasSlotId);
  1258. GraphCanvas::GraphCanvasRequestBus::BroadcastResult(dataDisplay, &GraphCanvas::GraphCanvasRequests::CreateComboBoxNodePropertyDisplay, static_cast<GraphCanvas::ComboBoxDataInterface*>(dataInterface));
  1259. }
  1260. if (dataDisplay != nullptr)
  1261. {
  1262. return dataDisplay;
  1263. }
  1264. delete dataInterface;
  1265. }
  1266. return nullptr;
  1267. }
  1268. void EditorGraph::SignalDirty()
  1269. {
  1270. SourceHandle handle(m_owner, AZ::Uuid::CreateNull());
  1271. GeneralRequestBus::Broadcast(&GeneralRequests::SignalSceneDirty, handle);
  1272. }
  1273. void EditorGraph::HighlightNodesByType(const ScriptCanvas::NodeTypeIdentifier& nodeTypeIdentifier)
  1274. {
  1275. for (const auto& nodePair : GetNodeMapping())
  1276. {
  1277. if (nodePair.second->GetNodeType() == nodeTypeIdentifier)
  1278. {
  1279. HighlightScriptCanvasEntity(nodePair.first);
  1280. }
  1281. }
  1282. }
  1283. void EditorGraph::HighlightEBusNodes(const ScriptCanvas::EBusBusId& busId, const ScriptCanvas::EBusEventId& eventId)
  1284. {
  1285. ScriptCanvas::NodeTypeIdentifier ebusIdentifier = ScriptCanvas::NodeUtils::ConstructEBusIdentifier(busId);
  1286. for (const auto& nodePair : GetNodeMapping())
  1287. {
  1288. ScriptCanvas::Node* canvasNode = nodePair.second;
  1289. if (canvasNode->GetNodeType() == ebusIdentifier)
  1290. {
  1291. AZ::EntityId graphCanvasNodeId;
  1292. SceneMemberMappingRequestBus::EventResult(graphCanvasNodeId, canvasNode->GetEntityId(), &SceneMemberMappingRequests::GetGraphCanvasEntityId);
  1293. bool hasEvent = false;
  1294. EBusHandlerNodeDescriptorRequestBus::EventResult(hasEvent, graphCanvasNodeId, &EBusHandlerNodeDescriptorRequests::ContainsEvent, eventId);
  1295. if (hasEvent)
  1296. {
  1297. HighlightScriptCanvasEntity(canvasNode->GetEntityId());
  1298. }
  1299. }
  1300. }
  1301. }
  1302. void EditorGraph::HighlightScriptEventNodes(const ScriptCanvas::EBusBusId& busId, const ScriptCanvas::EBusEventId& eventId)
  1303. {
  1304. ScriptCanvas::NodeTypeIdentifier sendScriptEventIdentifier = ScriptCanvas::NodeUtils::ConstructSendScriptEventIdentifier(busId, eventId);
  1305. ScriptCanvas::NodeTypeIdentifier receiveScriptEventIdentifier = ScriptCanvas::NodeUtils::ConstructScriptEventIdentifier(busId);
  1306. for (AZ::Entity* entity : GetGraphData()->m_nodes)
  1307. {
  1308. ScriptCanvas::Node* canvasNode = AZ::EntityUtils::FindFirstDerivedComponent<ScriptCanvas::Node>(entity);
  1309. if (canvasNode->GetNodeType() == sendScriptEventIdentifier)
  1310. {
  1311. HighlightScriptCanvasEntity(entity->GetId());
  1312. }
  1313. else if (canvasNode->GetNodeType() == receiveScriptEventIdentifier)
  1314. {
  1315. AZ::EntityId graphCanvasNodeId;
  1316. SceneMemberMappingRequestBus::EventResult(graphCanvasNodeId, canvasNode->GetEntityId(), &SceneMemberMappingRequests::GetGraphCanvasEntityId);
  1317. bool hasEvent = false;
  1318. EBusHandlerNodeDescriptorRequestBus::EventResult(hasEvent, graphCanvasNodeId, &EBusHandlerNodeDescriptorRequests::ContainsEvent, eventId);
  1319. if (hasEvent)
  1320. {
  1321. HighlightScriptCanvasEntity(entity->GetId());
  1322. }
  1323. }
  1324. }
  1325. }
  1326. void EditorGraph::HighlightScriptCanvasEntity(const AZ::EntityId& scriptCanvasId)
  1327. {
  1328. GraphCanvas::SceneMemberGlowOutlineConfiguration glowConfiguration;
  1329. glowConfiguration.m_blurRadius = 0; // #17174 using blur degrades performance
  1330. glowConfiguration.m_pen = QPen();
  1331. glowConfiguration.m_pen.setBrush(QColor(243,129,29));
  1332. glowConfiguration.m_pen.setWidth(5);
  1333. SceneMemberMappingRequestBus::EventResult(glowConfiguration.m_sceneMember, scriptCanvasId, &SceneMemberMappingRequests::GetGraphCanvasEntityId);
  1334. glowConfiguration.m_pulseRate = AZStd::chrono::milliseconds(2500);
  1335. glowConfiguration.m_zValue = 0;
  1336. GraphCanvas::GraphicsEffectId graphicsEffectId;
  1337. GraphCanvas::SceneRequestBus::EventResult(graphicsEffectId, GetGraphCanvasGraphId(), &GraphCanvas::SceneRequests::CreateGlowOnSceneMember, glowConfiguration);
  1338. if (graphicsEffectId.IsValid())
  1339. {
  1340. m_highlights.insert(graphicsEffectId);
  1341. }
  1342. }
  1343. AZ::EntityId EditorGraph::FindGraphCanvasSlotId(const AZ::EntityId& graphCanvasNodeId, const ScriptCanvas::SlotId& slotId)
  1344. {
  1345. AZ::EntityId graphCanvasSlotId;
  1346. SlotMappingRequestBus::EventResult(graphCanvasSlotId, graphCanvasNodeId, &SlotMappingRequests::MapToGraphCanvasId, slotId);
  1347. if (!graphCanvasSlotId.IsValid())
  1348. {
  1349. // For the EBusHandler's I need to remap these to a different visual node.
  1350. // Since multiple GraphCanvas nodes depict a single ScriptCanvas EBus node.
  1351. if (EBusHandlerNodeDescriptorRequestBus::FindFirstHandler(graphCanvasNodeId) != nullptr)
  1352. {
  1353. GraphCanvas::Endpoint graphCanvasEventEndpoint;
  1354. EBusHandlerNodeDescriptorRequestBus::EventResult(graphCanvasEventEndpoint, graphCanvasNodeId, &EBusHandlerNodeDescriptorRequests::MapSlotToGraphCanvasEndpoint, slotId);
  1355. graphCanvasSlotId = graphCanvasEventEndpoint.GetSlotId();
  1356. }
  1357. }
  1358. return graphCanvasSlotId;
  1359. }
  1360. bool EditorGraph::ConfigureConnectionUserData(const ScriptCanvas::Endpoint& sourceEndpoint, const ScriptCanvas::Endpoint& targetEndpoint, GraphCanvas::ConnectionId connectionId)
  1361. {
  1362. bool isConfigured = true;
  1363. AZ::Entity* scConnectionEntity = nullptr;
  1364. FindConnection(scConnectionEntity, sourceEndpoint, targetEndpoint);
  1365. if (scConnectionEntity)
  1366. {
  1367. AZStd::any* connectionUserData = nullptr;
  1368. GraphCanvas::ConnectionRequestBus::EventResult(connectionUserData, connectionId, &GraphCanvas::ConnectionRequests::GetUserData);
  1369. if (connectionUserData)
  1370. {
  1371. *connectionUserData = scConnectionEntity->GetId();
  1372. SceneMemberMappingConfigurationRequestBus::Event(connectionId, &SceneMemberMappingConfigurationRequests::ConfigureMapping, scConnectionEntity->GetId());
  1373. }
  1374. }
  1375. else
  1376. {
  1377. isConfigured = false;
  1378. }
  1379. return isConfigured;
  1380. }
  1381. void EditorGraph::HandleQueuedUpdates()
  1382. {
  1383. bool signalDirty = false;
  1384. RequestPushPreventUndoStateUpdate();
  1385. AZStd::unordered_set< ScriptCanvas::Node* > newUpdates;
  1386. bool updatePropertyGrid = false;
  1387. for (const AZ::EntityId& queuedUpdate : m_queuedConvertingNodes)
  1388. {
  1389. bool isSelected = false;
  1390. GraphCanvas::SceneMemberUIRequestBus::EventResult(isSelected, queuedUpdate, &GraphCanvas::SceneMemberUIRequests::IsSelected);
  1391. if (isSelected)
  1392. {
  1393. updatePropertyGrid = true;
  1394. }
  1395. AZ::EntityId scriptCanvasNodeId = ConvertToScriptCanvasNodeId(queuedUpdate);
  1396. ScriptCanvas::Node* node = FindNode(scriptCanvasNodeId);
  1397. if (node->IsOutOfDate(GetVersion()))
  1398. {
  1399. if (OnVersionConversionBegin((*node)))
  1400. {
  1401. newUpdates.insert(node);
  1402. }
  1403. }
  1404. }
  1405. m_queuedConvertingNodes.clear();
  1406. AZStd::unordered_set<AZ::EntityId> deletedNodes;
  1407. for (ScriptCanvas::Node* node : newUpdates)
  1408. {
  1409. ScriptCanvas::UpdateResult updateResult = node->UpdateNode();
  1410. OnVersionConversionEnd((*node));
  1411. AZ::EntityId graphCanvasNodeId;
  1412. SceneMemberMappingRequestBus::EventResult(graphCanvasNodeId, node->GetEntityId(), &SceneMemberMappingRequests::GetGraphCanvasEntityId);
  1413. switch (updateResult)
  1414. {
  1415. case ScriptCanvas::UpdateResult::DeleteNode:
  1416. {
  1417. if (graphCanvasNodeId.IsValid())
  1418. {
  1419. deletedNodes.insert(graphCanvasNodeId);
  1420. }
  1421. signalDirty = true;
  1422. break;
  1423. }
  1424. default:
  1425. {
  1426. signalDirty = true;
  1427. break;
  1428. }
  1429. }
  1430. }
  1431. if (!deletedNodes.empty())
  1432. {
  1433. GraphCanvas::SceneRequestBus::Event(GetGraphCanvasGraphId(), &GraphCanvas::SceneRequests::Delete, deletedNodes);
  1434. }
  1435. RequestPopPreventUndoStateUpdate();
  1436. if (signalDirty)
  1437. {
  1438. SignalDirty();
  1439. }
  1440. if (updatePropertyGrid)
  1441. {
  1442. PropertyGridRequestBus::Broadcast(&PropertyGridRequests::RefreshPropertyGrid);
  1443. }
  1444. }
  1445. bool EditorGraph::IsNodeVersionConverting(const AZ::EntityId& graphCanvasNodeId) const
  1446. {
  1447. bool isConverting = false;
  1448. if (!m_convertingNodes.empty())
  1449. {
  1450. if (GraphCanvas::GraphUtils::IsNodeWrapped(graphCanvasNodeId))
  1451. {
  1452. AZ::EntityId parentId;
  1453. GraphCanvas::NodeRequestBus::EventResult(parentId, graphCanvasNodeId, &GraphCanvas::NodeRequests::GetWrappingNode);
  1454. if (m_convertingNodes.count(parentId) > 0)
  1455. {
  1456. isConverting = true;
  1457. }
  1458. }
  1459. else if (m_convertingNodes.count(graphCanvasNodeId) > 0)
  1460. {
  1461. isConverting = true;
  1462. }
  1463. }
  1464. return isConverting;
  1465. }
  1466. void EditorGraph::OnPreNodeDeleted(const AZ::EntityId& nodeId)
  1467. {
  1468. // If we are cdeleteing a HandlerEventNode we don't need to do anything since they are purely visual.
  1469. // And the underlying ScriptCanvas nodes will persist and maintain all of their state.
  1470. if (EBusHandlerEventNodeDescriptorRequestBus::FindFirstHandler(nodeId) == nullptr)
  1471. {
  1472. AZStd::any* userData = nullptr;
  1473. GraphCanvas::NodeRequestBus::EventResult(userData, nodeId, &GraphCanvas::NodeRequests::GetUserData);
  1474. if (userData && userData->is<AZ::EntityId>())
  1475. {
  1476. const AZ::EntityId* scriptCanvasNodeId = AZStd::any_cast<AZ::EntityId>(userData);
  1477. auto iter = m_graphCanvasSaveData.find((*scriptCanvasNodeId));
  1478. if (iter != m_graphCanvasSaveData.end())
  1479. {
  1480. delete iter->second;
  1481. m_graphCanvasSaveData.erase(iter);
  1482. }
  1483. }
  1484. AZStd::any* sourceUserData = nullptr;
  1485. GraphCanvas::NodeRequestBus::EventResult(sourceUserData, nodeId, &GraphCanvas::NodeRequests::GetUserData);
  1486. auto scriptCanvasNodeId = sourceUserData && sourceUserData->is<AZ::EntityId>() ? *AZStd::any_cast<AZ::EntityId>(sourceUserData) : AZ::EntityId();
  1487. if (RemoveNode(scriptCanvasNodeId))
  1488. {
  1489. AZ::ComponentApplicationBus::Broadcast(&AZ::ComponentApplicationRequests::DeleteEntity, scriptCanvasNodeId);
  1490. }
  1491. }
  1492. }
  1493. void EditorGraph::OnPreConnectionDeleted(const AZ::EntityId& connectionId)
  1494. {
  1495. AZStd::any* userData = nullptr;
  1496. GraphCanvas::ConnectionRequestBus::EventResult(userData, connectionId, &GraphCanvas::ConnectionRequests::GetUserData);
  1497. if (userData && userData->is<AZ::EntityId>())
  1498. {
  1499. const AZ::EntityId* scriptCanvasConnectionId = AZStd::any_cast<AZ::EntityId>(userData);
  1500. auto iter = m_graphCanvasSaveData.find((*scriptCanvasConnectionId));
  1501. if (iter != m_graphCanvasSaveData.end())
  1502. {
  1503. delete iter->second;
  1504. m_graphCanvasSaveData.erase(iter);
  1505. }
  1506. }
  1507. GraphCanvas::Endpoint sourceEndpoint;
  1508. GraphCanvas::ConnectionRequestBus::EventResult(sourceEndpoint, connectionId, &GraphCanvas::ConnectionRequests::GetSourceEndpoint);
  1509. ScriptCanvas::Endpoint scriptCanvasEndpoint = ConvertToScriptCanvasEndpoint(sourceEndpoint);
  1510. // Don't disconnect any connections if we are version converting a node involved
  1511. if (IsNodeVersionConverting(scriptCanvasEndpoint.GetNodeId()))
  1512. {
  1513. return;
  1514. }
  1515. GraphCanvas::Endpoint targetEndpoint;
  1516. GraphCanvas::ConnectionRequestBus::EventResult(targetEndpoint, connectionId, &GraphCanvas::ConnectionRequests::GetTargetEndpoint);
  1517. scriptCanvasEndpoint = ConvertToScriptCanvasEndpoint(targetEndpoint);
  1518. if (IsNodeVersionConverting(scriptCanvasEndpoint.GetNodeId()))
  1519. {
  1520. return;
  1521. }
  1522. DisconnectConnection(connectionId);
  1523. }
  1524. void EditorGraph::OnUnknownPaste([[maybe_unused]] const QPointF& scenePos)
  1525. {
  1526. GraphVariablesTableView::HandleVariablePaste(GetScriptCanvasId());
  1527. }
  1528. void EditorGraph::OnSelectionChanged()
  1529. {
  1530. ClearHighlights();
  1531. }
  1532. AZ::u32 EditorGraph::GetNewVariableCounter()
  1533. {
  1534. return ++m_variableCounter;
  1535. }
  1536. void EditorGraph::ReleaseVariableCounter(AZ::u32 variableCounter)
  1537. {
  1538. if (m_variableCounter == variableCounter)
  1539. {
  1540. --m_variableCounter;
  1541. }
  1542. }
  1543. void EditorGraph::RequestUndoPoint()
  1544. {
  1545. GeneralRequestBus::Broadcast(&GeneralRequests::PostUndoPoint, GetScriptCanvasId());
  1546. }
  1547. void EditorGraph::RequestPushPreventUndoStateUpdate()
  1548. {
  1549. GeneralRequestBus::Broadcast(&GeneralRequests::PushPreventUndoStateUpdate);
  1550. }
  1551. void EditorGraph::RequestPopPreventUndoStateUpdate()
  1552. {
  1553. GeneralRequestBus::Broadcast(&GeneralRequests::PopPreventUndoStateUpdate);
  1554. }
  1555. void EditorGraph::TriggerUndo()
  1556. {
  1557. GeneralRequestBus::Broadcast(&GeneralRequests::TriggerUndo);
  1558. }
  1559. void EditorGraph::TriggerRedo()
  1560. {
  1561. GeneralRequestBus::Broadcast(&GeneralRequests::TriggerRedo);
  1562. }
  1563. void EditorGraph::EnableNodes(const AZStd::unordered_set< GraphCanvas::NodeId >& nodeIds)
  1564. {
  1565. bool enabledNodes = false;
  1566. for (auto graphCanvasNodeId : nodeIds)
  1567. {
  1568. AZStd::any* nodeUserData = nullptr;
  1569. GraphCanvas::NodeRequestBus::EventResult(nodeUserData, graphCanvasNodeId, &GraphCanvas::NodeRequests::GetUserData);
  1570. if (auto* scNodeId = AZStd::any_cast<AZ::EntityId>(nodeUserData))
  1571. {
  1572. bool hasNonUserDisabledFlag = false;
  1573. ScriptCanvas::NodeRequestBus::EventResult(hasNonUserDisabledFlag, (*scNodeId), &ScriptCanvas::NodeRequests::HasNodeDisabledFlag, ScriptCanvas::NodeDisabledFlag::NonUser);
  1574. if (!hasNonUserDisabledFlag)
  1575. {
  1576. ScriptCanvas::NodeRequestBus::Event((*scNodeId), &ScriptCanvas::NodeRequests::RemoveNodeDisabledFlag, ScriptCanvas::NodeDisabledFlag::User);
  1577. enabledNodes = true;
  1578. }
  1579. }
  1580. }
  1581. if (enabledNodes)
  1582. {
  1583. GeneralRequestBus::Broadcast(&GeneralRequests::PostUndoPoint, GetScriptCanvasId());
  1584. }
  1585. }
  1586. void EditorGraph::DisableNodes(const AZStd::unordered_set< GraphCanvas::NodeId >& nodeIds)
  1587. {
  1588. bool disabledNodes = false;
  1589. for (auto graphCanvasNodeId : nodeIds)
  1590. {
  1591. AZStd::any* nodeUserData = nullptr;
  1592. GraphCanvas::NodeRequestBus::EventResult(nodeUserData, graphCanvasNodeId, &GraphCanvas::NodeRequests::GetUserData);
  1593. if (auto* scNodeId = AZStd::any_cast<AZ::EntityId>(nodeUserData))
  1594. {
  1595. ScriptCanvas::NodeRequestBus::Event((*scNodeId), &ScriptCanvas::NodeRequests::AddNodeDisabledFlag, ScriptCanvas::NodeDisabledFlag::User);
  1596. disabledNodes = true;
  1597. }
  1598. }
  1599. if (disabledNodes)
  1600. {
  1601. GeneralRequestBus::Broadcast(&GeneralRequests::PostUndoPoint, GetScriptCanvasId());
  1602. }
  1603. }
  1604. void EditorGraph::PostDeletionEvent()
  1605. {
  1606. GeneralRequestBus::Broadcast(&GeneralRequests::PostUndoPoint, GetScriptCanvasId());
  1607. // Work-around for a crash caused by the MainWindow::OnSystemTick not being handled before the ReflectedPropertyEditor's DoRefresh.
  1608. // This will force a refresh selection on any post-deletion events so that the DoRefresh will not crash on deleted objects
  1609. UIRequestBus::Broadcast(&UIRequests::RefreshSelection);
  1610. }
  1611. void EditorGraph::PostCreationEvent()
  1612. {
  1613. GeneralRequestBus::Broadcast(&GeneralRequests::PushPreventUndoStateUpdate);
  1614. if (m_wrapperNodeDropTarget.IsValid())
  1615. {
  1616. for (const AZ::EntityId& nodeId : m_lastGraphCanvasCreationGroup)
  1617. {
  1618. GraphCanvas::WrappedNodeConfiguration configuration;
  1619. GraphCanvas::WrapperNodeConfigurationRequestBus::EventResult(configuration, m_wrapperNodeDropTarget, &GraphCanvas::WrapperNodeConfigurationRequests::GetWrappedNodeConfiguration, nodeId);
  1620. GraphCanvas::WrapperNodeRequestBus::Event(m_wrapperNodeDropTarget, &GraphCanvas::WrapperNodeRequests::WrapNode, nodeId, configuration);
  1621. }
  1622. }
  1623. else
  1624. {
  1625. // List of nodes we want to delete, because they are invalid in our current context
  1626. AZStd::unordered_set<AZ::EntityId> invalidNodes;
  1627. // Three maps here.
  1628. // WrapperTypeMapping: Keeps track of which wrappers were created by wrapper type.
  1629. AZStd::unordered_map< AZ::Crc32, AZ::EntityId > wrapperTypeMapping;
  1630. // WrapperIdMapping: Keeps track of EntityId mappings for the Wrappers.
  1631. AZStd::unordered_map< AZ::EntityId, AZ::EntityId > wrapperIdMapping;
  1632. // RequiredWrappers: Keeps track of a map of all of the wrapper types required to be created, along with the nodes
  1633. // that wanted to create the nodes.
  1634. AZStd::unordered_multimap< AZ::Crc32, AZ::EntityId > requiredWrappersMapping;
  1635. // In general, we will only ever use 2 at once(in the case of a drag/drop: busType + eventWrapper)
  1636. // In the case of a paste: busIdWrappers + eventWrappers
  1637. // Logic is merged here just to try to reduce the duplicated logic, and because I can't really
  1638. // tell the difference between the two cases anyway.
  1639. //
  1640. // Idea here is to keep track of groupings so that when we paste, I can create the appropriate number
  1641. // of nodes and groupings within these nodes to create a proper duplicate. And when we drag and drop
  1642. // I want to merge as many wrapped elements onto a single node as I can.
  1643. //
  1644. // First step in this process is to sort our pasted nodes into EBus handlers and EBus events.
  1645. for (const AZ::EntityId& nodeId : m_lastGraphCanvasCreationGroup)
  1646. {
  1647. bool isExecutionNodeling = false;
  1648. NodeDescriptorRequestBus::EventResult(isExecutionNodeling, nodeId, &NodeDescriptorRequests::IsType, NodeDescriptorType::FunctionDefinitionNode);
  1649. if (isExecutionNodeling)
  1650. {
  1651. AZStd::any* userData = nullptr;
  1652. GraphCanvas::NodeRequestBus::EventResult(userData, nodeId, &GraphCanvas::NodeRequests::GetUserData);
  1653. AZ::EntityId scSourceNodeId = (userData && userData->is<AZ::EntityId>()) ? *AZStd::any_cast<AZ::EntityId>(userData) : AZ::EntityId();
  1654. ScriptCanvas::Nodes::Core::FunctionDefinitionNode* nodeling = azrtti_cast<ScriptCanvas::Nodes::Core::FunctionDefinitionNode*>(FindNode(scSourceNodeId));
  1655. if (nodeling)
  1656. {
  1657. nodeling->RemapId();
  1658. }
  1659. }
  1660. bool isFunctionNode = false;
  1661. NodeDescriptorRequestBus::EventResult(isFunctionNode, nodeId, &NodeDescriptorRequests::IsType, NodeDescriptorType::FunctionNode);
  1662. // Show all hidden slots on a paste, as a temporary fix until I can sort out what I want this to work like.
  1663. GraphCanvas::NodeRequestBus::Event(nodeId, &GraphCanvas::NodeRequests::ShowAllSlots);
  1664. if (GraphCanvas::WrapperNodeRequestBus::FindFirstHandler(nodeId) != nullptr)
  1665. {
  1666. wrapperIdMapping[nodeId] = nodeId;
  1667. AZ::Crc32 wrapperType;
  1668. GraphCanvas::WrapperNodeRequestBus::EventResult(wrapperType, nodeId, &GraphCanvas::WrapperNodeRequests::GetWrapperType);
  1669. if (wrapperType != AZ::Crc32())
  1670. {
  1671. auto mapIter = wrapperTypeMapping.find(wrapperType);
  1672. if (mapIter == wrapperTypeMapping.end())
  1673. {
  1674. wrapperTypeMapping[wrapperType] = nodeId;
  1675. }
  1676. }
  1677. }
  1678. if (GraphCanvas::ForcedWrappedNodeRequestBus::FindFirstHandler(nodeId) != nullptr)
  1679. {
  1680. bool isWrapped = false;
  1681. GraphCanvas::NodeRequestBus::EventResult(isWrapped, nodeId, &GraphCanvas::NodeRequests::IsWrapped);
  1682. if (!isWrapped)
  1683. {
  1684. AZ::Crc32 wrapperType;
  1685. GraphCanvas::ForcedWrappedNodeRequestBus::EventResult(wrapperType, nodeId, &GraphCanvas::ForcedWrappedNodeRequests::GetWrapperType);
  1686. if (wrapperType != AZ::Crc32())
  1687. {
  1688. requiredWrappersMapping.emplace(wrapperType,nodeId);
  1689. }
  1690. }
  1691. }
  1692. }
  1693. // Second step is to go through, and determine which usage case is valid so we know how to filter down our events.
  1694. // If we can't find a wrapper, or we can't create a handler for the wrapper. We need to delete it.
  1695. for (const auto& mapPair : requiredWrappersMapping)
  1696. {
  1697. AZ::EntityId wrapperNodeId;
  1698. // Look up in our previous group mapping to see if it belonged to a node previously
  1699. // (i.e. copy + pasted node).
  1700. AZ::EntityId previousGroupWrapperNodeId;
  1701. auto mapIter = m_wrappedNodeGroupings.find(mapPair.second);
  1702. if (mapIter != m_wrappedNodeGroupings.end())
  1703. {
  1704. previousGroupWrapperNodeId = mapIter->second;
  1705. auto busIter = wrapperIdMapping.find(previousGroupWrapperNodeId);
  1706. if (busIter != wrapperIdMapping.end())
  1707. {
  1708. wrapperNodeId = busIter->second;
  1709. }
  1710. }
  1711. // We may have already found our target node.
  1712. // If we have, bypass the creation step.
  1713. if (!wrapperNodeId.IsValid())
  1714. {
  1715. // If we haven't check if we match a type, or if our previous group wrapper node is valid.
  1716. // If we had a previous group. I need to create a wrapper for that group.
  1717. // If we didn't have a previous group, I want to just use the Bus name to find an appropriate grouping.
  1718. auto busIter = wrapperTypeMapping.find(mapPair.first);
  1719. if (busIter == wrapperTypeMapping.end() || previousGroupWrapperNodeId.IsValid())
  1720. {
  1721. AZ::EntityId forcedWrappedNodeId = mapPair.second;
  1722. AZ::Vector2 position;
  1723. GraphCanvas::GeometryRequestBus::EventResult(position, forcedWrappedNodeId, &GraphCanvas::GeometryRequests::GetPosition);
  1724. GraphCanvas::ForcedWrappedNodeRequestBus::EventResult(wrapperNodeId, forcedWrappedNodeId, &GraphCanvas::ForcedWrappedNodeRequests::CreateWrapperNode, GetGraphCanvasGraphId(), position);
  1725. if (wrapperNodeId.IsValid())
  1726. {
  1727. m_lastGraphCanvasCreationGroup.emplace_back(wrapperNodeId);
  1728. if (!previousGroupWrapperNodeId.IsValid())
  1729. {
  1730. wrapperTypeMapping.emplace(mapPair.first, wrapperNodeId);
  1731. }
  1732. else
  1733. {
  1734. wrapperIdMapping.emplace(previousGroupWrapperNodeId, wrapperNodeId);
  1735. }
  1736. }
  1737. else
  1738. {
  1739. AZ_Error("ScriptCanvas", false, "Failed to instantiate an Wrapper node with type: (%d)", mapPair.first);
  1740. invalidNodes.insert(mapPair.second);
  1741. continue;
  1742. }
  1743. }
  1744. else
  1745. {
  1746. wrapperNodeId = busIter->second;
  1747. }
  1748. }
  1749. GraphCanvas::WrappedNodeConfiguration configuration;
  1750. GraphCanvas::WrapperNodeConfigurationRequestBus::EventResult(configuration, wrapperNodeId, &GraphCanvas::WrapperNodeConfigurationRequests::GetWrappedNodeConfiguration, mapPair.second);
  1751. GraphCanvas::WrapperNodeRequestBus::Event(wrapperNodeId, &GraphCanvas::WrapperNodeRequests::WrapNode, mapPair.second, configuration);
  1752. }
  1753. GraphCanvas::SceneRequestBus::Event(GetGraphCanvasGraphId(), &GraphCanvas::SceneRequests::Delete, invalidNodes);
  1754. }
  1755. ScriptCanvas::Node::ExploredDynamicGroupCache exploredCache;
  1756. for (AZ::EntityId graphCanvasNodeId : m_lastGraphCanvasCreationGroup)
  1757. {
  1758. AZStd::any* userData = nullptr;
  1759. GraphCanvas::NodeRequestBus::EventResult(userData, graphCanvasNodeId, &GraphCanvas::NodeRequests::GetUserData);
  1760. AZ::EntityId scSourceNodeId = (userData && userData->is<AZ::EntityId>()) ? *AZStd::any_cast<AZ::EntityId>(userData) : AZ::EntityId();
  1761. if (scSourceNodeId.IsValid())
  1762. {
  1763. ScriptCanvas::Node* node = FindNode(scSourceNodeId);
  1764. if (node)
  1765. {
  1766. node->SanityCheckDynamicDisplay(exploredCache);
  1767. node->PostActivate();
  1768. }
  1769. }
  1770. OnSaveDataDirtied(graphCanvasNodeId);
  1771. Nodes::UpdateSlotDatumLabels(graphCanvasNodeId);
  1772. }
  1773. m_wrappedNodeGroupings.clear();
  1774. m_lastGraphCanvasCreationGroup.clear();
  1775. m_wrapperNodeDropTarget.SetInvalid();
  1776. GeneralRequestBus::Broadcast(&GeneralRequests::PopPreventUndoStateUpdate);
  1777. GeneralRequestBus::Broadcast(&GeneralRequests::PostUndoPoint, GetScriptCanvasId());
  1778. }
  1779. void EditorGraph::PostRestore(const UndoData&)
  1780. {
  1781. AZStd::vector<AZ::EntityId> graphCanvasNodeIds;
  1782. GraphCanvas::SceneRequestBus::EventResult(graphCanvasNodeIds, GetGraphCanvasGraphId(), &GraphCanvas::SceneRequests::GetNodes);
  1783. for (AZ::EntityId graphCanvasNodeId : graphCanvasNodeIds)
  1784. {
  1785. Nodes::UpdateSlotDatumLabels(graphCanvasNodeId);
  1786. }
  1787. GraphCanvas::ViewId viewId;
  1788. GraphCanvas::SceneRequestBus::EventResult(viewId, GetGraphCanvasGraphId(), &GraphCanvas::SceneRequests::GetViewId);
  1789. GraphCanvas::ViewRequestBus::Event(viewId, &GraphCanvas::ViewRequests::RefreshView);
  1790. }
  1791. void EditorGraph::OnPasteBegin()
  1792. {
  1793. GeneralRequestBus::Broadcast(&GeneralRequests::PushPreventUndoStateUpdate);
  1794. }
  1795. void EditorGraph::OnPasteEnd()
  1796. {
  1797. GeneralRequestBus::Broadcast(&GeneralRequests::PopPreventUndoStateUpdate);
  1798. GeneralRequestBus::Broadcast(&GeneralRequests::PostUndoPoint, GetScriptCanvasId());
  1799. }
  1800. void EditorGraph::OnGraphCanvasNodeCreated(const AZ::EntityId& nodeId)
  1801. {
  1802. m_lastGraphCanvasCreationGroup.emplace_back(nodeId);
  1803. }
  1804. void EditorGraph::ResetSlotToDefaultValue(const GraphCanvas::Endpoint& endpoint)
  1805. {
  1806. ScriptCanvas::Endpoint scEndpoint = ConvertToScriptCanvasEndpoint(endpoint);
  1807. ScriptCanvas::Node* canvasNode = FindNode(scEndpoint.GetNodeId());
  1808. if (canvasNode)
  1809. {
  1810. canvasNode->ResetSlotToDefaultValue(scEndpoint.GetSlotId());
  1811. }
  1812. }
  1813. void EditorGraph::ResetReference(const GraphCanvas::Endpoint& endpoint)
  1814. {
  1815. // ResetSlotToDefault deals with resetting the reference internal to the function call on the node.
  1816. ResetSlotToDefaultValue(endpoint);
  1817. }
  1818. void EditorGraph::ResetProperty(const GraphCanvas::NodeId& nodeId, const AZ::Crc32& propertyId)
  1819. {
  1820. AZ::EntityId scriptCanvasNodeId = ConvertToScriptCanvasNodeId(nodeId);
  1821. ScriptCanvas::Node* canvasNode = FindNode(scriptCanvasNodeId);
  1822. if (canvasNode)
  1823. {
  1824. canvasNode->ResetProperty(propertyId);
  1825. }
  1826. }
  1827. void EditorGraph::RemoveSlot(const GraphCanvas::Endpoint& endpoint)
  1828. {
  1829. ScriptCanvas::Endpoint scEndpoint = ConvertToScriptCanvasEndpoint(endpoint);
  1830. ScriptCanvas::Slot* slot = FindSlot(scEndpoint);
  1831. if (slot)
  1832. {
  1833. ScriptCanvas::GraphVariable* variable = slot->GetVariable();
  1834. if (variable && variable->GetScope() == ScriptCanvas::VariableFlags::Scope::Function)
  1835. {
  1836. bool success = false;
  1837. ScriptCanvas::GraphVariableManagerRequestBus::EventResult(success, GetScriptCanvasId(), &ScriptCanvas::GraphVariableManagerRequests::RemoveVariable, variable->GetVariableId());
  1838. if (!success)
  1839. {
  1840. AZ_Assert(success, "Failed to remove variable that corresponds to this slot");
  1841. }
  1842. }
  1843. }
  1844. ScriptCanvas::Node* canvasNode = FindNode(scEndpoint.GetNodeId());
  1845. if (canvasNode)
  1846. {
  1847. canvasNode->DeleteSlot(scEndpoint.GetSlotId());
  1848. }
  1849. }
  1850. bool EditorGraph::IsSlotRemovable(const GraphCanvas::Endpoint& endpoint) const
  1851. {
  1852. ScriptCanvas::Endpoint scEndpoint = ConvertToScriptCanvasEndpoint(endpoint);
  1853. ScriptCanvas::Node* canvasNode = FindNode(scEndpoint.GetNodeId());
  1854. if (canvasNode)
  1855. {
  1856. return canvasNode->CanDeleteSlot(scEndpoint.GetSlotId());
  1857. }
  1858. return false;
  1859. }
  1860. bool EditorGraph::ConvertSlotToReference(const GraphCanvas::Endpoint& endpoint, bool isNewSlot)
  1861. {
  1862. ScriptCanvas::Endpoint scEndpoint = ConvertToScriptCanvasEndpoint(endpoint);
  1863. ScriptCanvas::Node* canvasNode = FindNode(scEndpoint.GetNodeId());
  1864. if (canvasNode)
  1865. {
  1866. return canvasNode->ConvertSlotToReference(scEndpoint.GetSlotId(), isNewSlot);
  1867. }
  1868. return false;
  1869. }
  1870. bool EditorGraph::CanConvertSlotToReference(const GraphCanvas::Endpoint& endpoint, bool isNewSlot)
  1871. {
  1872. ScriptCanvas::Endpoint scEndpoint = ConvertToScriptCanvasEndpoint(endpoint);
  1873. ScriptCanvas::Node* canvasNode = FindNode(scEndpoint.GetNodeId());
  1874. if (canvasNode)
  1875. {
  1876. ScriptCanvas::Slot* slot = canvasNode->GetSlot(scEndpoint.GetSlotId());
  1877. if (slot)
  1878. {
  1879. return slot->CanConvertToReference(isNewSlot);
  1880. }
  1881. }
  1882. return false;
  1883. }
  1884. GraphCanvas::CanHandleMimeEventOutcome EditorGraph::CanHandleReferenceMimeEvent(const GraphCanvas::Endpoint& endpoint, const QMimeData* mimeData)
  1885. {
  1886. ScriptCanvas::Endpoint scEndpoint = ConvertToScriptCanvasEndpoint(endpoint);
  1887. ScriptCanvas::Node* canvasNode = FindNode(scEndpoint.GetNodeId());
  1888. if (canvasNode)
  1889. {
  1890. ScriptCanvas::Slot* slot = canvasNode->GetSlot(scEndpoint.GetSlotId());
  1891. if (slot->CanConvertToReference() || slot->IsVariableReference())
  1892. {
  1893. ScriptCanvas::VariableId variableId = GraphCanvas::QtMimeUtils::ExtractTypeFromMimeData<ScriptCanvas::VariableId>(mimeData, GraphCanvas::k_ReferenceMimeType);
  1894. ScriptCanvas::GraphVariable* variable = FindVariableById(variableId);
  1895. if (variable)
  1896. {
  1897. return canvasNode->SlotAcceptsType(scEndpoint.GetSlotId(), variable->GetDataType());
  1898. }
  1899. else
  1900. {
  1901. return AZ::Failure(AZStd::string("Unable to find variable"));
  1902. }
  1903. }
  1904. else
  1905. {
  1906. return AZ::Failure(AZStd::string("Unable to convert slot to Reference"));
  1907. }
  1908. }
  1909. return AZ::Failure(AZStd::string("Unable to find Node"));
  1910. }
  1911. bool EditorGraph::HandleReferenceMimeEvent(const GraphCanvas::Endpoint& endpoint, const QMimeData* mimeData)
  1912. {
  1913. bool handledEvent = false;
  1914. ScriptCanvas::Endpoint scEndpoint = ConvertToScriptCanvasEndpoint(endpoint);
  1915. ScriptCanvas::Node* canvasNode = FindNode(scEndpoint.GetNodeId());
  1916. if (canvasNode)
  1917. {
  1918. ScriptCanvas::Slot* slot = canvasNode->GetSlot(scEndpoint.GetSlotId());
  1919. if (slot->IsVariableReference())
  1920. {
  1921. ScriptCanvas::VariableId variableId = GraphCanvas::QtMimeUtils::ExtractTypeFromMimeData<ScriptCanvas::VariableId>(mimeData, GraphCanvas::k_ReferenceMimeType);
  1922. if (variableId.IsValid())
  1923. {
  1924. canvasNode->SetSlotVariableId(scEndpoint.GetSlotId(), variableId);
  1925. handledEvent = true;
  1926. }
  1927. }
  1928. }
  1929. return handledEvent;
  1930. }
  1931. bool EditorGraph::CanPromoteToVariable(const GraphCanvas::Endpoint& endpoint, [[maybe_unused]] bool isNewSlot) const
  1932. {
  1933. ScriptCanvas::Endpoint scriptCanvasEndpoint = ConvertToScriptCanvasEndpoint(endpoint);
  1934. auto activeSlot = FindSlot(scriptCanvasEndpoint);
  1935. if (activeSlot && !activeSlot->IsVariableReference() && activeSlot->CanConvertToReference())
  1936. {
  1937. if (!activeSlot->IsDynamicSlot() || activeSlot->HasDisplayType())
  1938. {
  1939. bool isValidVariableType = false;
  1940. VariablePaletteRequestBus::BroadcastResult(isValidVariableType, &VariablePaletteRequests::IsValidVariableType, activeSlot->GetDataType());
  1941. return isValidVariableType;
  1942. }
  1943. }
  1944. return false;
  1945. }
  1946. bool EditorGraph::PromoteToVariableAction(const GraphCanvas::Endpoint& endpoint, bool isNewSlot)
  1947. {
  1948. ScriptCanvas::Endpoint scriptCanvasEndpoint = ConvertToScriptCanvasEndpoint(endpoint);
  1949. auto activeNode = FindNode(scriptCanvasEndpoint.GetNodeId());
  1950. auto activeSlot = FindSlot(scriptCanvasEndpoint);
  1951. if (activeNode == nullptr || activeSlot == nullptr)
  1952. {
  1953. return false;
  1954. }
  1955. if (activeSlot->IsVariableReference())
  1956. {
  1957. return false;
  1958. }
  1959. if (activeSlot->IsDynamicSlot() && !activeSlot->HasDisplayType())
  1960. {
  1961. return false;
  1962. }
  1963. const ScriptCanvas::Datum* activeDatum = activeNode->FindDatum(scriptCanvasEndpoint.GetSlotId());
  1964. AZStd::string variableName = "";
  1965. int variableCounter = 0;
  1966. AZStd::string defaultName;
  1967. AZ::Outcome<void, ScriptCanvas::GraphVariableValidationErrorCode> hasValidDefault = AZ::Failure(ScriptCanvas::GraphVariableValidationErrorCode::Unknown);
  1968. do
  1969. {
  1970. variableCounter = GetNewVariableCounter();
  1971. defaultName = VariableDockWidget::ConstructDefaultVariableName(variableCounter);
  1972. ScriptCanvas::GraphVariableManagerRequestBus::EventResult(hasValidDefault, GetScriptCanvasId(), &ScriptCanvas::GraphVariableManagerRequests::IsNameValid, defaultName);
  1973. } while (!hasValidDefault);
  1974. bool nameAvailable = hasValidDefault.IsSuccess();
  1975. if (nameAvailable)
  1976. {
  1977. variableName = defaultName;
  1978. }
  1979. QWidget* mainWindow = nullptr;
  1980. UIRequestBus::BroadcastResult(mainWindow, &UIRequests::GetMainWindow);
  1981. AZStd::string inBoxText = "";
  1982. // Special case to try re-using the slot name if this is on an execution nodeling, since the user had just
  1983. // given it a name already with the ShowVariableConfigurationWidget dialog
  1984. if (azrtti_istypeof<ScriptCanvas::Nodes::Core::FunctionDefinitionNode>(activeNode))
  1985. {
  1986. variableName = activeSlot->GetName();
  1987. ScriptCanvas::GraphVariableManagerRequestBus::EventResult(nameAvailable, GetScriptCanvasId(), &ScriptCanvas::GraphVariableManagerRequests::IsNameAvailable, variableName);
  1988. }
  1989. int nameCount = 0;
  1990. while (!nameAvailable)
  1991. {
  1992. if (nameCount == 0)
  1993. {
  1994. variableName.append(AZStd::string::format(" (%d)", ++nameCount));
  1995. }
  1996. else
  1997. {
  1998. AZ::StringFunc::Replace(variableName, AZStd::string::format("(%d)", nameCount-1).c_str(), AZStd::string::format("(%d)", nameCount).c_str());
  1999. ++nameCount;
  2000. }
  2001. ScriptCanvas::GraphVariableManagerRequestBus::EventResult(nameAvailable, GetScriptCanvasId(), &ScriptCanvas::GraphVariableManagerRequests::IsNameAvailable, variableName);
  2002. }
  2003. activeSlot->Rename(variableName);
  2004. ScriptCanvas::Datum variableDatum;
  2005. if (activeDatum)
  2006. {
  2007. variableDatum.ReconfigureDatumTo((*activeDatum));
  2008. }
  2009. else
  2010. {
  2011. variableDatum.SetType(activeSlot->GetDataType());
  2012. // BCO Objects are defaulted to a reference. Going to bypass them to avoid messing with variable
  2013. // defaults.
  2014. if (ScriptCanvas::Data::IsValueType(activeSlot->GetDataType()))
  2015. {
  2016. variableDatum.SetToDefaultValueOfType();
  2017. }
  2018. }
  2019. AZ::Outcome<ScriptCanvas::VariableId, AZStd::string> addOutcome;
  2020. constexpr bool functionScope = false; // Promoted variables are used as references, thus they need to be a member variable
  2021. ScriptCanvas::GraphVariableManagerRequestBus::EventResult(addOutcome, GetScriptCanvasId(), &ScriptCanvas::GraphVariableManagerRequests::AddVariable, variableName, variableDatum, functionScope);
  2022. if (addOutcome.IsSuccess())
  2023. {
  2024. GraphCanvas::DataSlotRequestBus::Event(endpoint.GetSlotId(), &GraphCanvas::DataSlotRequests::ConvertToReference, isNewSlot);
  2025. activeSlot->SetVariableReference(addOutcome.GetValue());
  2026. }
  2027. return addOutcome.IsSuccess();
  2028. }
  2029. bool EditorGraph::SynchronizeReferences(const GraphCanvas::Endpoint& referenceSource, const GraphCanvas::Endpoint& referenceTarget)
  2030. {
  2031. ScriptCanvas::Endpoint scriptCanvasSourceEndpoint = ConvertToScriptCanvasEndpoint(referenceSource);
  2032. ScriptCanvas::Endpoint scriptCanvasTargetEndpoint = ConvertToScriptCanvasEndpoint(referenceTarget);
  2033. auto sourceSlot = FindSlot(scriptCanvasSourceEndpoint);
  2034. auto targetSlot = FindSlot(scriptCanvasTargetEndpoint);
  2035. if (sourceSlot == nullptr
  2036. || targetSlot == nullptr)
  2037. {
  2038. return false;
  2039. }
  2040. if (!sourceSlot->IsVariableReference())
  2041. {
  2042. return false;
  2043. }
  2044. if (sourceSlot->IsTypeMatchFor((*targetSlot)))
  2045. {
  2046. if (!targetSlot->IsVariableReference())
  2047. {
  2048. GraphCanvas::DataSlotRequestBus::Event(referenceTarget.GetSlotId(), &GraphCanvas::DataSlotRequests::ConvertToReference, false);
  2049. }
  2050. if (targetSlot->IsVariableReference())
  2051. {
  2052. ScriptCanvas::VariableId variableId = sourceSlot->GetVariableReference();
  2053. targetSlot->SetVariableReference(variableId);
  2054. return true;
  2055. }
  2056. }
  2057. return false;
  2058. }
  2059. bool EditorGraph::ConvertSlotToValue(const GraphCanvas::Endpoint& endpoint)
  2060. {
  2061. ScriptCanvas::Endpoint scEndpoint = ConvertToScriptCanvasEndpoint(endpoint);
  2062. ScriptCanvas::Node* canvasNode = FindNode(scEndpoint.GetNodeId());
  2063. if (canvasNode)
  2064. {
  2065. return canvasNode->ConvertSlotToValue(scEndpoint.GetSlotId());
  2066. }
  2067. return false;
  2068. }
  2069. bool EditorGraph::CanConvertSlotToValue(const GraphCanvas::Endpoint& endpoint)
  2070. {
  2071. ScriptCanvas::Endpoint scEndpoint = ConvertToScriptCanvasEndpoint(endpoint);
  2072. ScriptCanvas::Node* canvasNode = FindNode(scEndpoint.GetNodeId());
  2073. if (canvasNode)
  2074. {
  2075. ScriptCanvas::Slot* slot = canvasNode->GetSlot(scEndpoint.GetSlotId());
  2076. return slot && slot->CanConvertToValue();
  2077. }
  2078. return false;
  2079. }
  2080. GraphCanvas::CanHandleMimeEventOutcome EditorGraph::CanHandleValueMimeEvent(const GraphCanvas::Endpoint& endpoint, const QMimeData* mimeData)
  2081. {
  2082. AZ_UNUSED(endpoint);
  2083. AZ_UNUSED(mimeData);
  2084. AZ_Assert(false, "Unimplemented drag and drop flow");
  2085. return AZ::Failure(AZStd::string("Unimplemented drag and drop flow"));
  2086. }
  2087. bool EditorGraph::HandleValueMimeEvent(const GraphCanvas::Endpoint& endpoint, const QMimeData* mimeData)
  2088. {
  2089. AZ_UNUSED(endpoint);
  2090. AZ_UNUSED(mimeData);
  2091. return false;
  2092. }
  2093. GraphCanvas::SlotId EditorGraph::RequestExtension(const GraphCanvas::NodeId& nodeId, const GraphCanvas::ExtenderId& extenderId, GraphModelRequests::ExtensionRequestReason reason)
  2094. {
  2095. GraphCanvas::SlotId graphCanvasSlotId;
  2096. AZStd::any* nodeUserData = nullptr;
  2097. GraphCanvas::NodeRequestBus::EventResult(nodeUserData, nodeId, &GraphCanvas::NodeRequests::GetUserData);
  2098. if (auto* scNodeId = AZStd::any_cast<AZ::EntityId>(nodeUserData))
  2099. {
  2100. AZ::Entity* graphNodeEntity = nullptr;
  2101. AZ::ComponentApplicationBus::BroadcastResult(graphNodeEntity, &AZ::ComponentApplicationRequests::FindEntity, *scNodeId);
  2102. ScriptCanvas::Node* canvasNode = AZ::EntityUtils::FindFirstDerivedComponent<ScriptCanvas::Node>(graphNodeEntity);
  2103. if (canvasNode)
  2104. {
  2105. auto functionDefintionNode = azrtti_cast<ScriptCanvas::Nodes::Core::FunctionDefinitionNode*>(canvasNode);
  2106. if (functionDefintionNode && reason == GraphModelRequests::ExtensionRequestReason::ConnectionProposal)
  2107. {
  2108. }
  2109. else
  2110. {
  2111. ScriptCanvas::SlotId slotId = canvasNode->HandleExtension(extenderId);
  2112. if (slotId.IsValid())
  2113. {
  2114. SlotMappingRequestBus::EventResult(graphCanvasSlotId, nodeId, &SlotMappingRequests::MapToGraphCanvasId, slotId);
  2115. HandleFunctionDefinitionExtension(canvasNode, graphCanvasSlotId, nodeId);
  2116. }
  2117. }
  2118. }
  2119. }
  2120. return graphCanvasSlotId;
  2121. }
  2122. void EditorGraph::ExtensionCancelled(const GraphCanvas::NodeId& nodeId, const GraphCanvas::ExtenderId& extenderId)
  2123. {
  2124. AZ::EntityId scNodeId = ConvertToScriptCanvasNodeId(nodeId);
  2125. if (scNodeId.IsValid())
  2126. {
  2127. ScriptCanvas::Node* canvasNode = FindNode(scNodeId);
  2128. if (canvasNode)
  2129. {
  2130. canvasNode->ExtensionCancelled(extenderId);
  2131. }
  2132. }
  2133. }
  2134. void EditorGraph::FinalizeExtension(const GraphCanvas::NodeId& nodeId, const GraphCanvas::ExtenderId& extenderId)
  2135. {
  2136. AZ::EntityId scNodeId = ConvertToScriptCanvasNodeId(nodeId);
  2137. if (scNodeId.IsValid())
  2138. {
  2139. ScriptCanvas::Node* canvasNode = FindNode(scNodeId);
  2140. if (canvasNode)
  2141. {
  2142. canvasNode->FinalizeExtension(extenderId);
  2143. }
  2144. }
  2145. }
  2146. bool EditorGraph::ShouldWrapperAcceptDrop(const AZ::EntityId& wrapperNode, const QMimeData* mimeData) const
  2147. {
  2148. if (!mimeData->hasFormat(Widget::NodePaletteDockWidget::GetMimeType()))
  2149. {
  2150. return false;
  2151. }
  2152. // Deep mime inspection
  2153. QByteArray arrayData = mimeData->data(Widget::NodePaletteDockWidget::GetMimeType());
  2154. GraphCanvas::GraphCanvasMimeContainer mimeContainer;
  2155. if (!mimeContainer.FromBuffer(arrayData.constData(), arrayData.size()) || mimeContainer.m_mimeEvents.empty())
  2156. {
  2157. return false;
  2158. }
  2159. AZStd::string busName;
  2160. EBusHandlerNodeDescriptorRequestBus::EventResult(busName, wrapperNode, &EBusHandlerNodeDescriptorRequests::GetBusName);
  2161. for (GraphCanvas::GraphCanvasMimeEvent* mimeEvent : mimeContainer.m_mimeEvents)
  2162. {
  2163. CreateEBusHandlerEventMimeEvent* createEbusMethodEvent = azrtti_cast<CreateEBusHandlerEventMimeEvent*>(mimeEvent);
  2164. if (createEbusMethodEvent)
  2165. {
  2166. if (createEbusMethodEvent->GetBusName().compare(busName) != 0)
  2167. {
  2168. return false;
  2169. }
  2170. bool containsEvent = false;
  2171. EBusHandlerNodeDescriptorRequestBus::EventResult(containsEvent, wrapperNode, &EBusHandlerNodeDescriptorRequests::ContainsEvent, createEbusMethodEvent->GetEventId());
  2172. if (containsEvent)
  2173. {
  2174. return false;
  2175. }
  2176. }
  2177. else
  2178. {
  2179. return false;
  2180. }
  2181. }
  2182. return true;
  2183. }
  2184. void EditorGraph::AddWrapperDropTarget(const AZ::EntityId& wrapperNode)
  2185. {
  2186. if (!m_wrapperNodeDropTarget.IsValid())
  2187. {
  2188. m_wrapperNodeDropTarget = wrapperNode;
  2189. }
  2190. }
  2191. void EditorGraph::RemoveWrapperDropTarget(const AZ::EntityId& wrapperNode)
  2192. {
  2193. if (m_wrapperNodeDropTarget == wrapperNode)
  2194. {
  2195. m_wrapperNodeDropTarget.SetInvalid();
  2196. }
  2197. }
  2198. GraphCanvas::GraphId EditorGraph::GetGraphCanvasGraphId() const
  2199. {
  2200. if (m_saveFormatConverted)
  2201. {
  2202. if (m_graphCanvasSceneEntity)
  2203. {
  2204. return m_graphCanvasSceneEntity->GetId();
  2205. }
  2206. return AZ::EntityId();
  2207. }
  2208. else
  2209. {
  2210. return GetEntityId();
  2211. }
  2212. }
  2213. NodeIdPair EditorGraph::CreateCustomNode(const AZ::Uuid& typeId, const AZ::Vector2& position)
  2214. {
  2215. CreateCustomNodeMimeEvent mimeEvent(typeId);
  2216. AZ::Vector2 dropPosition = position;
  2217. if (mimeEvent.ExecuteEvent(position, dropPosition, GetGraphCanvasGraphId()))
  2218. {
  2219. return mimeEvent.GetCreatedPair();
  2220. }
  2221. return NodeIdPair();
  2222. }
  2223. void EditorGraph::AddCrcCache(const AZ::Crc32& crcValue, const AZStd::string& cacheString)
  2224. {
  2225. auto mapIter = m_crcCacheMap.find(crcValue);
  2226. if (mapIter == m_crcCacheMap.end())
  2227. {
  2228. m_crcCacheMap.emplace(crcValue, CRCCache(cacheString));
  2229. }
  2230. else
  2231. {
  2232. mapIter->second.m_cacheCount++;
  2233. }
  2234. }
  2235. void EditorGraph::RemoveCrcCache(const AZ::Crc32& crcValue)
  2236. {
  2237. auto mapIter = m_crcCacheMap.find(crcValue);
  2238. if (mapIter != m_crcCacheMap.end())
  2239. {
  2240. mapIter->second.m_cacheCount--;
  2241. if (mapIter->second.m_cacheCount <= 0)
  2242. {
  2243. m_crcCacheMap.erase(mapIter);
  2244. }
  2245. }
  2246. }
  2247. AZStd::string EditorGraph::DecodeCrc(const AZ::Crc32& crcValue)
  2248. {
  2249. auto mapIter = m_crcCacheMap.find(crcValue);
  2250. if (mapIter != m_crcCacheMap.end())
  2251. {
  2252. return mapIter->second.m_cacheValue;
  2253. }
  2254. return "";
  2255. }
  2256. void EditorGraph::ClearHighlights()
  2257. {
  2258. for (const GraphCanvas::GraphicsEffectId& effectId : m_highlights)
  2259. {
  2260. GraphCanvas::SceneRequestBus::Event(GetGraphCanvasGraphId(), &GraphCanvas::SceneRequests::CancelGraphicsEffect, effectId);
  2261. }
  2262. m_highlights.clear();
  2263. }
  2264. void EditorGraph::HighlightMembersFromTreeItem(const GraphCanvas::GraphCanvasTreeItem* treeItem)
  2265. {
  2266. ClearHighlights();
  2267. if (auto handleEbusEventTreeItem = azrtti_cast<const EBusHandleEventPaletteTreeItem*>(treeItem))
  2268. {
  2269. HighlightEBusNodes(handleEbusEventTreeItem->GetBusId(), handleEbusEventTreeItem->GetEventId());
  2270. }
  2271. else if (auto sendScriptEventTreeItem = azrtti_cast<const ScriptEventsEventNodePaletteTreeItem*>(treeItem))
  2272. {
  2273. HighlightScriptEventNodes(sendScriptEventTreeItem->GetBusIdentifier(), sendScriptEventTreeItem->GetEventIdentifier());
  2274. }
  2275. else
  2276. {
  2277. HighlightNodesByType(NodeIdentifierFactory::ConstructNodeIdentifier(treeItem));
  2278. }
  2279. }
  2280. void EditorGraph::HighlightVariables(const AZStd::unordered_set< ScriptCanvas::VariableId >& variableIds)
  2281. {
  2282. ClearHighlights();
  2283. for (auto nodeComponentPair : GetNodeMapping())
  2284. {
  2285. ScriptCanvas::Node* node = nodeComponentPair.second;
  2286. if (node->ContainsReferencesToVariables(variableIds))
  2287. {
  2288. HighlightScriptCanvasEntity(nodeComponentPair.first);
  2289. }
  2290. }
  2291. }
  2292. void EditorGraph::HighlightNodes(const AZStd::vector<NodeIdPair>& nodes)
  2293. {
  2294. ClearHighlights();
  2295. for (const NodeIdPair& nodeIdPair : nodes)
  2296. {
  2297. HighlightScriptCanvasEntity(nodeIdPair.m_scriptCanvasId);
  2298. }
  2299. }
  2300. void EditorGraph::RemoveUnusedVariables()
  2301. {
  2302. RequestPushPreventUndoStateUpdate();
  2303. auto variableData = GetVariableData();
  2304. auto variables = variableData->GetVariables();
  2305. AZStd::unordered_set<ScriptCanvas::VariableId> usedVariableIds;
  2306. for (auto nodePair : GetNodeMapping())
  2307. {
  2308. ScriptCanvas::Node* node = nodePair.second;
  2309. node->CollectVariableReferences(usedVariableIds);
  2310. }
  2311. AZStd::unordered_set<ScriptCanvas::VariableId> unusedVariables;
  2312. for (auto variableData2 : variables)
  2313. {
  2314. if (usedVariableIds.count(variableData2.first) == 0)
  2315. {
  2316. unusedVariables.insert(variableData2.first);
  2317. }
  2318. }
  2319. bool removedVariable = false;
  2320. for (ScriptCanvas::VariableId variableId : unusedVariables)
  2321. {
  2322. bool success = false;
  2323. ScriptCanvas::GraphVariableManagerRequestBus::EventResult(success, GetScriptCanvasId(), &ScriptCanvas::GraphVariableManagerRequests::RemoveVariable, variableId);
  2324. if (success)
  2325. {
  2326. removedVariable = true;
  2327. }
  2328. }
  2329. RequestPopPreventUndoStateUpdate();
  2330. if (removedVariable)
  2331. {
  2332. RequestUndoPoint();
  2333. }
  2334. }
  2335. bool EditorGraph::CanConvertVariableNodeToReference(const GraphCanvas::NodeId& nodeId)
  2336. {
  2337. AZ::EntityId scriptCanvasNodeId = ConvertToScriptCanvasNodeId(nodeId);
  2338. ScriptCanvas::VariableId variableId;
  2339. ScriptCanvas::VariableNodeRequestBus::EventResult(variableId, scriptCanvasNodeId, &ScriptCanvas::VariableNodeRequests::GetId);
  2340. ScriptCanvas::GraphVariable* variable = FindVariableById(variableId);
  2341. if (variable == nullptr)
  2342. {
  2343. return false;
  2344. }
  2345. AZStd::vector< GraphCanvas::SlotId > slotIds;
  2346. GraphCanvas::NodeRequestBus::EventResult(slotIds, nodeId, &GraphCanvas::NodeRequests::GetSlotIds);
  2347. for (const GraphCanvas::SlotId& slotId : slotIds)
  2348. {
  2349. GraphCanvas::SlotType slotType = GraphCanvas::SlotTypes::Invalid;
  2350. GraphCanvas::SlotRequestBus::EventResult(slotType, slotId, &GraphCanvas::SlotRequests::GetSlotType);
  2351. if (slotType == GraphCanvas::SlotTypes::ExecutionSlot)
  2352. {
  2353. continue;
  2354. }
  2355. GraphCanvas::Endpoint gcEndpoint(nodeId, slotId);
  2356. ScriptCanvas::Endpoint scEndpoint = ConvertToScriptCanvasEndpoint(gcEndpoint);
  2357. ScriptCanvas::Node* node = FindNode(scEndpoint.GetNodeId());
  2358. // We only care about the actual variable type for enabling/disabling the button.
  2359. // All other conditions will be handled in the conversion with user prompts.
  2360. if (node->SlotAcceptsType(scEndpoint.GetSlotId(), variable->GetDataType()))
  2361. {
  2362. AZStd::vector< GraphCanvas::ConnectionId > connectionIds;
  2363. GraphCanvas::SlotRequestBus::EventResult(connectionIds, slotId, &GraphCanvas::SlotRequests::GetConnections);
  2364. return !connectionIds.empty();
  2365. }
  2366. }
  2367. return false;
  2368. }
  2369. bool EditorGraph::ConvertVariableNodeToReference(const GraphCanvas::NodeId& nodeId)
  2370. {
  2371. AZ::EntityId scriptCanvasNodeId = ConvertToScriptCanvasNodeId(nodeId);
  2372. ScriptCanvas::VariableId variableId;
  2373. ScriptCanvas::VariableNodeRequestBus::EventResult(variableId, scriptCanvasNodeId, &ScriptCanvas::VariableNodeRequests::GetId);
  2374. ScriptCanvas::GraphVariable* variable = FindVariableById(variableId);
  2375. if (variable == nullptr)
  2376. {
  2377. return false;
  2378. }
  2379. AZStd::vector< GraphCanvas::SlotId > slotIds;
  2380. GraphCanvas::NodeRequestBus::EventResult(slotIds, nodeId, &GraphCanvas::NodeRequests::GetSlotIds);
  2381. AZStd::unordered_set< GraphCanvas::Endpoint > referencableEndpoints;
  2382. bool canDetachNode = true;
  2383. GraphCanvas::GraphId graphCanvasGraphId = GetGraphCanvasGraphId();
  2384. QMainWindow* mainWindow = nullptr;
  2385. UIRequestBus::BroadcastResult(mainWindow, &UIRequests::GetMainWindow);
  2386. for (const GraphCanvas::SlotId& slotId : slotIds)
  2387. {
  2388. GraphCanvas::SlotRequests* slotRequests = GraphCanvas::SlotRequestBus::FindFirstHandler(slotId);
  2389. if (slotRequests)
  2390. {
  2391. GraphCanvas::SlotType slotType = slotRequests->GetSlotType();
  2392. if (slotType == GraphCanvas::SlotTypes::DataSlot)
  2393. {
  2394. GraphCanvas::Endpoint currentEndpoint(nodeId, slotId);
  2395. // If we have a reference anywhere on us. We need to maintain this node, since it's not doing something
  2396. // we can merge out cleanly
  2397. ScriptCanvas::Endpoint scriptCanvasCurrentEndpoint = ConvertToScriptCanvasEndpoint(currentEndpoint);
  2398. ScriptCanvas::Slot* sourceSlot = FindSlot(scriptCanvasCurrentEndpoint);
  2399. if (sourceSlot->IsVariableReference())
  2400. {
  2401. canDetachNode = false;
  2402. }
  2403. auto connectionIds = slotRequests->GetConnections();
  2404. for (auto connectionId : connectionIds)
  2405. {
  2406. GraphCanvas::Endpoint otherEndpoint;
  2407. GraphCanvas::ConnectionRequestBus::EventResult(otherEndpoint, connectionId, &GraphCanvas::ConnectionRequests::FindOtherEndpoint, currentEndpoint);
  2408. ScriptCanvas::Endpoint scriptCanvasOtherEndpoint = ConvertToScriptCanvasEndpoint(otherEndpoint);
  2409. ScriptCanvas::Node* otherNode = FindNode(scriptCanvasOtherEndpoint.GetNodeId());
  2410. ScriptCanvas::Slot* otherSlot = FindSlot(scriptCanvasOtherEndpoint);
  2411. if (otherNode && otherSlot && otherNode->SlotAcceptsType(scriptCanvasOtherEndpoint.GetSlotId(), variable->GetDataType()))
  2412. {
  2413. AZStd::unordered_set< AZ::EntityId > deletedConnection = { connectionId };
  2414. GraphCanvas::SceneRequestBus::Event(graphCanvasGraphId, &GraphCanvas::SceneRequests::Delete, deletedConnection);
  2415. if (otherSlot->CanConvertToReference())
  2416. {
  2417. referencableEndpoints.insert(otherEndpoint);
  2418. }
  2419. else
  2420. {
  2421. // Try to resolve chained steps when we are going to end up being a 'Get' reference which is fine to convert to.
  2422. // Otherwise, if we chain to a 'set' reference, that might have unintended consequences, so we need to ignore that.
  2423. // Because we will double invert, check our source if we are an input. We can chain. If we are an output, we don't want to chain.
  2424. if (sourceSlot->IsInput())
  2425. {
  2426. AZStd::vector< GraphCanvas::ConnectionId > chainedConnectionIds;
  2427. GraphCanvas::SlotRequestBus::EventResult(chainedConnectionIds, otherEndpoint.GetSlotId(), &GraphCanvas::SlotRequests::GetConnections);
  2428. for (auto chainedConnectionId : chainedConnectionIds)
  2429. {
  2430. GraphCanvas::Endpoint chainedEndpoint;
  2431. GraphCanvas::ConnectionRequestBus::EventResult(chainedEndpoint, chainedConnectionId, &GraphCanvas::ConnectionRequests::FindOtherEndpoint, otherEndpoint);
  2432. ScriptCanvas::Endpoint scriptCanvasChainedEndpoint = ConvertToScriptCanvasEndpoint(chainedEndpoint);
  2433. ScriptCanvas::Node* chainedNode = FindNode(scriptCanvasChainedEndpoint.GetNodeId());
  2434. ScriptCanvas::Slot* chainedSlot = FindSlot(scriptCanvasChainedEndpoint);
  2435. if (chainedNode && chainedSlot && chainedNode->SlotAcceptsType(scriptCanvasChainedEndpoint.GetSlotId(), variable->GetDataType()))
  2436. {
  2437. AZStd::unordered_set< AZ::EntityId > chainedDeletedConnection = { chainedConnectionId };
  2438. GraphCanvas::SceneRequestBus::Event(graphCanvasGraphId, &GraphCanvas::SceneRequests::Delete, chainedDeletedConnection);
  2439. if (chainedSlot->CanConvertToReference())
  2440. {
  2441. referencableEndpoints.insert(chainedEndpoint);
  2442. }
  2443. else
  2444. {
  2445. GraphCanvas::SlotRequestBus::Event(chainedEndpoint.GetSlotId(), &GraphCanvas::SlotRequests::CreateConnectionWithEndpoint, otherEndpoint);
  2446. }
  2447. }
  2448. }
  2449. }
  2450. if (otherSlot->CanConvertToReference())
  2451. {
  2452. referencableEndpoints.insert(otherEndpoint);
  2453. }
  2454. else
  2455. {
  2456. GraphCanvas::SlotRequestBus::Event(currentEndpoint.GetSlotId(), &GraphCanvas::SlotRequests::CreateConnectionWithEndpoint, otherEndpoint);
  2457. canDetachNode = false;
  2458. }
  2459. }
  2460. }
  2461. else
  2462. {
  2463. canDetachNode = false;
  2464. }
  2465. }
  2466. }
  2467. }
  2468. }
  2469. // Signal out on the graph that we did something to the node.
  2470. GraphCanvas::AnimatedPulseConfiguration animatedPulseConfig;
  2471. animatedPulseConfig.m_enableGradient = true;
  2472. if (canDetachNode)
  2473. {
  2474. animatedPulseConfig.m_drawColor = QColor(255, 0, 0);
  2475. }
  2476. else
  2477. {
  2478. animatedPulseConfig.m_drawColor = QColor(255, 255, 255);
  2479. }
  2480. animatedPulseConfig.m_durationSec = 0.25f;
  2481. GraphCanvas::SceneRequestBus::Event(GetGraphCanvasGraphId(), &GraphCanvas::SceneRequests::CreatePulseAroundSceneMember, nodeId, 4, animatedPulseConfig);
  2482. // If we can detach the node. All connections will be deleted, except for the ones we want to save.
  2483. if (canDetachNode)
  2484. {
  2485. GraphCanvas::NodeDetachConfig detachConfig(nodeId);
  2486. detachConfig.m_listingType = GraphCanvas::ListingType::InclusiveList;
  2487. detachConfig.m_typeListing.insert(GraphCanvas::SlotTypes::ExecutionSlot);
  2488. GraphCanvas::GraphUtils::DetachNodeAndStitchConnections(detachConfig);
  2489. AZStd::unordered_set<GraphCanvas::NodeId > nodeIds = { nodeId };
  2490. GraphCanvas::SceneRequestBus::Event(GetGraphCanvasGraphId(), &GraphCanvas::SceneRequests::Delete, nodeIds);
  2491. }
  2492. for (auto graphCanvasEndpoint : referencableEndpoints)
  2493. {
  2494. GraphCanvas::DataSlotRequestBus::Event(graphCanvasEndpoint.GetSlotId(), &GraphCanvas::DataSlotRequests::ConvertToReference, false);
  2495. ScriptCanvas::Endpoint scriptCanvasEndpoint = ConvertToScriptCanvasEndpoint(graphCanvasEndpoint);
  2496. ScriptCanvas::Slot* slot = FindSlot(scriptCanvasEndpoint);
  2497. if (slot && slot->IsVariableReference())
  2498. {
  2499. slot->SetVariableReference(variable->GetVariableId());
  2500. }
  2501. }
  2502. return true;
  2503. }
  2504. bool EditorGraph::ConvertReferenceToVariableNode([[maybe_unused]] const GraphCanvas::Endpoint& endpoint)
  2505. {
  2506. return false;
  2507. }
  2508. bool EditorGraph::OnVersionConversionBegin(ScriptCanvas::Node& scriptCanvasNode)
  2509. {
  2510. auto insertResult = m_convertingNodes.insert(scriptCanvasNode.GetEntityId());
  2511. if (!insertResult.second)
  2512. {
  2513. return false;
  2514. }
  2515. for (const ScriptCanvas::Slot& currentSlot : scriptCanvasNode.GetSlots())
  2516. {
  2517. m_versionedSlots.insert(AZStd::make_pair(scriptCanvasNode.GetEntityId(), currentSlot.GetId()));
  2518. }
  2519. EditorNodeNotificationBus::Event(scriptCanvasNode.GetEntityId(), &EditorNodeNotifications::OnVersionConversionBegin);
  2520. return true;
  2521. }
  2522. void EditorGraph::OnVersionConversionEnd(ScriptCanvas::Node& scriptCanvasNode)
  2523. {
  2524. EditorNodeNotificationBus::Event(scriptCanvasNode.GetEntityId(), &EditorNodeNotifications::OnVersionConversionEnd);
  2525. size_t removeCount = m_convertingNodes.count(scriptCanvasNode.GetEntityId());
  2526. if (removeCount > 0)
  2527. {
  2528. auto findResult = m_versionedSlots.equal_range(scriptCanvasNode.GetEntityId());
  2529. AZStd::unordered_set<ScriptCanvas::SlotId> previousSlots;
  2530. for (auto cacheIter = findResult.first; cacheIter != findResult.second; ++cacheIter)
  2531. {
  2532. previousSlots.insert(cacheIter->second);
  2533. }
  2534. AZStd::unordered_set< GraphCanvas::ConnectionId > deletedGraphCanvasConnections;
  2535. for (const ScriptCanvas::Slot& constantSlot : scriptCanvasNode.GetSlots())
  2536. {
  2537. ScriptCanvas::Slot* currentSlot = scriptCanvasNode.GetSlot(constantSlot.GetId());
  2538. if (!currentSlot)
  2539. {
  2540. AZ_Error("ScriptCanvas", false, "Missing slot from node %s after conversion ", scriptCanvasNode.GetDebugName().data());
  2541. continue;
  2542. }
  2543. ScriptCanvas::SlotId slotId = currentSlot->GetId();
  2544. size_t eraseCount = previousSlots.erase(slotId);
  2545. if (eraseCount == 0)
  2546. {
  2547. continue;
  2548. }
  2549. // Manage updating connections and remove invalid ones
  2550. ScriptCanvas::Endpoint endpoint = currentSlot->GetEndpoint();
  2551. GraphCanvas::Endpoint graphCanvasEndpoint = ConvertToGraphCanvasEndpoint(endpoint);
  2552. AZStd::vector< ScriptCanvas::Endpoint > connectedEndpoints = GetConnectedEndpoints(endpoint);
  2553. for (const ScriptCanvas::Endpoint& connectedEndpoint : connectedEndpoints)
  2554. {
  2555. if (IsNodeVersionConverting(connectedEndpoint.GetNodeId()))
  2556. {
  2557. continue;
  2558. }
  2559. bool allowConnection = CanConnectionExistBetween(connectedEndpoint, endpoint).IsSuccess();
  2560. bool deleteConnection = true;
  2561. if (graphCanvasEndpoint.IsValid())
  2562. {
  2563. GraphCanvas::Endpoint otherEndpoint = ConvertToGraphCanvasEndpoint(connectedEndpoint);
  2564. if (otherEndpoint.IsValid())
  2565. {
  2566. bool isConnected = false;
  2567. GraphCanvas::SlotRequestBus::EventResult(isConnected, graphCanvasEndpoint.GetSlotId(), &GraphCanvas::SlotRequests::IsConnectedTo, otherEndpoint);
  2568. if (isConnected)
  2569. {
  2570. deleteConnection = false;
  2571. // If Graph canvas is connected, but we need to kill the connection
  2572. // we'll let the Graph Canvas Deletion Update our internal state.
  2573. if (!allowConnection)
  2574. {
  2575. AZStd::unordered_set< GraphCanvas::Endpoint > searchEndpoints = { otherEndpoint };
  2576. GraphCanvas::SlotRequestBus::Event(graphCanvasEndpoint.GetSlotId(), &GraphCanvas::SlotRequests::FindConnectionsForEndpoints, searchEndpoints, deletedGraphCanvasConnections);
  2577. }
  2578. }
  2579. else if (allowConnection)
  2580. {
  2581. deleteConnection = false;
  2582. GraphCanvas::SlotRequestBus::Event(graphCanvasEndpoint.GetSlotId(), &GraphCanvas::SlotRequests::DisplayConnectionWithEndpoint, otherEndpoint);
  2583. }
  2584. }
  2585. }
  2586. if (deleteConnection)
  2587. {
  2588. AZ::Entity* connectionEntity = nullptr;
  2589. if (FindConnection(connectionEntity, endpoint, connectedEndpoint))
  2590. {
  2591. RemoveConnection(connectionEntity->GetId());
  2592. }
  2593. }
  2594. }
  2595. }
  2596. for (auto erasedSlot : previousSlots)
  2597. {
  2598. VersioningRemoveSlot(scriptCanvasNode, erasedSlot);
  2599. }
  2600. m_versionedSlots.erase(scriptCanvasNode.GetEntityId());
  2601. m_convertingNodes.erase(scriptCanvasNode.GetEntityId());
  2602. if (!deletedGraphCanvasConnections.empty())
  2603. {
  2604. GraphCanvas::SceneRequestBus::Event(GetGraphCanvasGraphId(), &GraphCanvas::SceneRequests::Delete, deletedGraphCanvasConnections);
  2605. }
  2606. AZStd::string updateString = scriptCanvasNode.GetUpdateString();
  2607. m_updateStrings.insert(updateString);
  2608. if (m_convertingNodes.empty())
  2609. {
  2610. DisplayUpdateToast();
  2611. }
  2612. }
  2613. }
  2614. AZStd::vector<NodeIdPair> EditorGraph::GetNodesOfType(const ScriptCanvas::NodeTypeIdentifier& nodeTypeIdentifier)
  2615. {
  2616. AZStd::vector<NodeIdPair> nodeIdPairs;
  2617. for (auto nodeMappingPair : GetNodeMapping())
  2618. {
  2619. ScriptCanvas::Node* canvasNode = nodeMappingPair.second;
  2620. AZ::EntityId nodeEntityId = canvasNode->GetEntityId();
  2621. if (canvasNode->GetNodeType() == nodeTypeIdentifier)
  2622. {
  2623. NodeIdPair nodeIdPair;
  2624. nodeIdPair.m_scriptCanvasId = nodeEntityId;
  2625. SceneMemberMappingRequestBus::EventResult(nodeIdPair.m_graphCanvasId, nodeEntityId, &SceneMemberMappingRequests::GetGraphCanvasEntityId);
  2626. nodeIdPairs.emplace_back(nodeIdPair);
  2627. }
  2628. else if (ScriptCanvas::Nodes::Core::EBusEventHandler* handlerNode = azrtti_cast<ScriptCanvas::Nodes::Core::EBusEventHandler*>(canvasNode))
  2629. {
  2630. ScriptCanvas::EBusBusId busId = handlerNode->GetEBusId();
  2631. for (auto eventPair : handlerNode->GetEvents())
  2632. {
  2633. ScriptCanvas::EBusEventId eventId = eventPair.second.m_eventId;
  2634. if (ScriptCanvas::NodeUtils::ConstructEBusEventReceiverIdentifier(busId, eventId) == nodeTypeIdentifier)
  2635. {
  2636. AZ::EntityId graphCanvasNodeId;
  2637. SceneMemberMappingRequestBus::EventResult(graphCanvasNodeId, canvasNode->GetEntityId(), &SceneMemberMappingRequests::GetGraphCanvasEntityId);
  2638. bool hasEvent = false;
  2639. EBusHandlerNodeDescriptorRequestBus::EventResult(hasEvent, graphCanvasNodeId, &EBusHandlerNodeDescriptorRequests::ContainsEvent, eventId);
  2640. if (hasEvent)
  2641. {
  2642. NodeIdPair nodeIdPair;
  2643. nodeIdPair.m_scriptCanvasId = nodeEntityId;
  2644. nodeIdPair.m_graphCanvasId = graphCanvasNodeId;
  2645. nodeIdPairs.emplace_back(nodeIdPair);
  2646. }
  2647. }
  2648. }
  2649. }
  2650. else if (ScriptCanvas::Nodes::Core::ReceiveScriptEvent* receiveScriptEvent = azrtti_cast<ScriptCanvas::Nodes::Core::ReceiveScriptEvent*>(canvasNode))
  2651. {
  2652. AZ::EntityId graphCanvasNodeId;
  2653. SceneMemberMappingRequestBus::EventResult(graphCanvasNodeId, canvasNode->GetEntityId(), &SceneMemberMappingRequests::GetGraphCanvasEntityId);
  2654. EBusHandlerNodeDescriptorRequests* ebusHandlerDescriptor = EBusHandlerNodeDescriptorRequestBus::FindFirstHandler(graphCanvasNodeId);
  2655. if (ebusHandlerDescriptor)
  2656. {
  2657. auto eventConfigurations = ebusHandlerDescriptor->GetEventConfigurations();
  2658. ScriptCanvas::EBusBusId busId = receiveScriptEvent->GetBusId();
  2659. for (auto eventConfiguration : eventConfigurations)
  2660. {
  2661. if (ScriptCanvas::NodeUtils::ConstructScriptEventReceiverIdentifier(busId, eventConfiguration.m_eventId) == nodeTypeIdentifier)
  2662. {
  2663. if (ebusHandlerDescriptor->ContainsEvent(eventConfiguration.m_eventId))
  2664. {
  2665. NodeIdPair nodeIdPair;
  2666. nodeIdPair.m_scriptCanvasId = nodeEntityId;
  2667. nodeIdPair.m_graphCanvasId = graphCanvasNodeId;
  2668. nodeIdPairs.emplace_back(nodeIdPair);
  2669. }
  2670. }
  2671. }
  2672. }
  2673. }
  2674. }
  2675. return nodeIdPairs;
  2676. }
  2677. AZStd::vector<NodeIdPair> EditorGraph::GetVariableNodes(const ScriptCanvas::VariableId& variableId)
  2678. {
  2679. AZStd::vector<NodeIdPair> variableNodes;
  2680. if (variableId.IsValid())
  2681. {
  2682. AZStd::unordered_set< ScriptCanvas::VariableId > variableIds = { variableId };
  2683. for (auto nodePairMapping : GetNodeMapping())
  2684. {
  2685. if (nodePairMapping.second->ContainsReferencesToVariables(variableIds))
  2686. {
  2687. NodeIdPair nodeIdPair;
  2688. nodeIdPair.m_scriptCanvasId = nodePairMapping.first;
  2689. SceneMemberMappingRequestBus::EventResult(nodeIdPair.m_graphCanvasId, nodePairMapping.first, &SceneMemberMappingRequests::GetGraphCanvasEntityId);
  2690. variableNodes.push_back(nodeIdPair);
  2691. }
  2692. }
  2693. }
  2694. return variableNodes;
  2695. }
  2696. void EditorGraph::QueueVersionUpdate(const AZ::EntityId& graphCanvasNodeId)
  2697. {
  2698. bool queueUpdate = m_queuedConvertingNodes.empty();
  2699. auto insertResult = m_queuedConvertingNodes.insert(graphCanvasNodeId);
  2700. if (insertResult.second && queueUpdate)
  2701. {
  2702. m_allowVersionUpdate = false;
  2703. AZ::SystemTickBus::Handler::BusConnect();
  2704. }
  2705. }
  2706. bool EditorGraph::CanExposeEndpoint(const GraphCanvas::Endpoint& endpoint)
  2707. {
  2708. bool isEnabled = false;
  2709. GraphCanvas::SlotType slotType;
  2710. GraphCanvas::SlotRequestBus::EventResult(slotType, endpoint.GetSlotId(), &GraphCanvas::SlotRequests::GetSlotType);
  2711. if (slotType == GraphCanvas::SlotTypes::DataSlot)
  2712. {
  2713. GraphCanvas::DataSlotType dataSlotType = GraphCanvas::DataSlotType::Unknown;
  2714. GraphCanvas::DataSlotRequestBus::EventResult(dataSlotType, endpoint.GetSlotId(), &GraphCanvas::DataSlotRequests::GetDataSlotType);
  2715. if (dataSlotType != GraphCanvas::DataSlotType::Value)
  2716. {
  2717. isEnabled = false;
  2718. }
  2719. bool hasConnections = false;
  2720. GraphCanvas::SlotRequestBus::EventResult(hasConnections, endpoint.GetSlotId(), &GraphCanvas::SlotRequests::HasConnections);
  2721. if (hasConnections)
  2722. {
  2723. isEnabled = false;
  2724. }
  2725. ScriptCanvas::Endpoint scEndpoint = ConvertToScriptCanvasEndpoint(endpoint);
  2726. ScriptCanvas::Slot* slot = FindSlot(scEndpoint);
  2727. // If we don't have a slot it likely means this is a remapped visual slot.
  2728. // So we don't want to perform many operations on it.
  2729. if (slot)
  2730. {
  2731. ScriptCanvas::Data::Type dataType = slot->GetDataType();
  2732. bool isValidVariableType = false;
  2733. VariablePaletteRequestBus::BroadcastResult(isValidVariableType, &VariablePaletteRequests::IsValidVariableType, dataType);
  2734. if (!isValidVariableType)
  2735. {
  2736. isEnabled = false;
  2737. }
  2738. }
  2739. else
  2740. {
  2741. isEnabled = false;
  2742. }
  2743. }
  2744. else
  2745. {
  2746. isEnabled = true;
  2747. }
  2748. bool isNodeling = false;
  2749. NodeDescriptorRequestBus::EventResult(isNodeling, endpoint.GetNodeId(), &NodeDescriptorRequests::IsType, NodeDescriptorType::FunctionDefinitionNode);
  2750. return isEnabled && !isNodeling;
  2751. }
  2752. ScriptCanvas::Endpoint EditorGraph::ConvertToScriptCanvasEndpoint(const GraphCanvas::Endpoint& endpoint) const
  2753. {
  2754. AZStd::any* userData = nullptr;
  2755. ScriptCanvas::Endpoint scriptCanvasEndpoint;
  2756. GraphCanvas::SlotRequestBus::EventResult(userData, endpoint.GetSlotId(), &GraphCanvas::SlotRequests::GetUserData);
  2757. ScriptCanvas::SlotId scSourceSlotId = (userData && userData->is<ScriptCanvas::SlotId>()) ? *AZStd::any_cast<ScriptCanvas::SlotId>(userData) : ScriptCanvas::SlotId();
  2758. userData = nullptr;
  2759. AZ::EntityId scriptCanvasNodeId = ConvertToScriptCanvasNodeId(endpoint.GetNodeId());
  2760. scriptCanvasEndpoint = ScriptCanvas::Endpoint(scriptCanvasNodeId, scSourceSlotId);
  2761. return scriptCanvasEndpoint;
  2762. }
  2763. GraphCanvas::Endpoint EditorGraph::ConvertToGraphCanvasEndpoint(const ScriptCanvas::Endpoint& endpoint) const
  2764. {
  2765. GraphCanvas::Endpoint graphCanvasEndpoint;
  2766. SlotMappingRequestBus::EventResult(graphCanvasEndpoint.m_slotId, endpoint.GetNodeId(), &SlotMappingRequests::MapToGraphCanvasId, endpoint.GetSlotId());
  2767. GraphCanvas::SlotRequestBus::EventResult(graphCanvasEndpoint.m_nodeId, graphCanvasEndpoint.GetSlotId(), &GraphCanvas::SlotRequests::GetNode);
  2768. return graphCanvasEndpoint;
  2769. }
  2770. void EditorGraph::SetOriginalToNewIdsMap(const AZStd::unordered_map<AZ::EntityId, AZ::EntityId>& originalIdToNewIds)
  2771. {
  2772. m_originalIdToNewIds = originalIdToNewIds;
  2773. }
  2774. void EditorGraph::GetOriginalToNewIdsMap(AZStd::unordered_map<AZ::EntityId, AZ::EntityId>& originalIdToNewIdsOut) const
  2775. {
  2776. originalIdToNewIdsOut = m_originalIdToNewIds;
  2777. }
  2778. AZ::EntityId EditorGraph::FindNewIdFromOriginal(const AZ::EntityId& originalId) const
  2779. {
  2780. if (m_originalIdToNewIds.empty())
  2781. {
  2782. return originalId;
  2783. }
  2784. if (!m_originalIdToNewIds.contains(originalId))
  2785. {
  2786. return AZ::EntityId();
  2787. }
  2788. return m_originalIdToNewIds.at(originalId);
  2789. }
  2790. AZ::EntityId EditorGraph::FindOriginalIdFromNew(const AZ::EntityId& newId) const
  2791. {
  2792. if (m_originalIdToNewIds.empty())
  2793. {
  2794. return newId;
  2795. }
  2796. const auto it = m_originalIdToNewIds.find(newId);
  2797. if (it == m_originalIdToNewIds.end())
  2798. {
  2799. return AZ::EntityId();
  2800. }
  2801. return it->first;
  2802. }
  2803. void EditorGraph::OnSaveDataDirtied(const AZ::EntityId& savedElement)
  2804. {
  2805. // The EbusHandlerEvent's are a visual only representation of alternative data, and should not be saved.
  2806. if (EBusHandlerEventNodeDescriptorRequestBus::FindFirstHandler(savedElement) != nullptr
  2807. || m_ignoreSaveRequests)
  2808. {
  2809. return;
  2810. }
  2811. AZStd::any* userData = nullptr;
  2812. GraphCanvas::NodeRequestBus::EventResult(userData, savedElement, &GraphCanvas::NodeRequests::GetUserData);
  2813. if (userData && userData->is<AZ::EntityId>())
  2814. {
  2815. const AZ::EntityId* scriptCanvasNodeId = AZStd::any_cast<AZ::EntityId>(userData);
  2816. GraphCanvas::EntitySaveDataContainer* container = nullptr;
  2817. auto mapIter = m_graphCanvasSaveData.find((*scriptCanvasNodeId));
  2818. if (mapIter == m_graphCanvasSaveData.end())
  2819. {
  2820. container = aznew GraphCanvas::EntitySaveDataContainer();
  2821. m_graphCanvasSaveData[(*scriptCanvasNodeId)] = container;
  2822. }
  2823. else
  2824. {
  2825. container = mapIter->second;
  2826. }
  2827. GraphCanvas::EntitySaveDataRequestBus::Event(savedElement, &GraphCanvas::EntitySaveDataRequests::WriteSaveData, (*container));
  2828. }
  2829. else if (savedElement == GetGraphCanvasGraphId())
  2830. {
  2831. GraphCanvas::EntitySaveDataContainer* container = nullptr;
  2832. auto mapIter = m_graphCanvasSaveData.find(GetEntityId());
  2833. if (mapIter == m_graphCanvasSaveData.end())
  2834. {
  2835. container = aznew GraphCanvas::EntitySaveDataContainer();
  2836. m_graphCanvasSaveData[GetEntityId()] = container;
  2837. }
  2838. else
  2839. {
  2840. container = mapIter->second;
  2841. }
  2842. GraphCanvas::EntitySaveDataRequestBus::Event(savedElement, &GraphCanvas::EntitySaveDataRequests::WriteSaveData, (*container));
  2843. m_statisticsHelper.PopulateStatisticData(this);
  2844. }
  2845. }
  2846. bool EditorGraph::NeedsSaveConversion() const
  2847. {
  2848. return !m_saveFormatConverted;
  2849. }
  2850. void EditorGraph::ConvertSaveFormat()
  2851. {
  2852. if (!m_saveFormatConverted)
  2853. {
  2854. // Bit of a work around for not being able to clean this up in the actual save.
  2855. m_saveFormatConverted = true;
  2856. // SceneComponent
  2857. for (const AZ::Uuid& componentType : {
  2858. AZ::Uuid("{3F71486C-3D51-431F-B904-DA070C7A0238}"), // GraphCanvas::SceneComponent
  2859. AZ::Uuid("{486B009F-632B-44F6-81C2-3838746190AE}"), // ColorPaletteManagerComponent
  2860. AZ::Uuid("{A8F08DEA-0F42-4236-9E1E-B93C964B113F}"), // BookmarkManagerComponent
  2861. AZ::Uuid("{34B81206-2C69-4886-945B-4A9ECC0FDAEE}") // StyleSheet
  2862. }
  2863. )
  2864. {
  2865. AZ::Component* component = GetEntity()->FindComponent(componentType);
  2866. if (component)
  2867. {
  2868. if (GetEntity()->RemoveComponent(component))
  2869. {
  2870. delete component;
  2871. }
  2872. }
  2873. }
  2874. }
  2875. }
  2876. void EditorGraph::ConstructSaveData()
  2877. {
  2878. // Save out the SceneData
  2879. //
  2880. // For this one all of the GraphCanvas information lives on the same entity.
  2881. // So we need to use that key to look up everything
  2882. {
  2883. OnSaveDataDirtied(GetGraphCanvasGraphId());
  2884. }
  2885. AZStd::vector< AZ::EntityId > graphCanvasNodes;
  2886. GraphCanvas::SceneRequestBus::EventResult(graphCanvasNodes, GetGraphCanvasGraphId(), &GraphCanvas::SceneRequests::GetNodes);
  2887. for (const AZ::EntityId& graphCanvasNode : graphCanvasNodes)
  2888. {
  2889. OnSaveDataDirtied(graphCanvasNode);
  2890. }
  2891. }
  2892. void EditorGraph::OnToastInteraction()
  2893. {
  2894. const AzToolsFramework::ToastId* toastId = AzToolsFramework::ToastNotificationBus::GetCurrentBusId();
  2895. if (toastId)
  2896. {
  2897. NodeIdPair pair;
  2898. pair.m_scriptCanvasId = m_toastNodeIds[(*toastId)];
  2899. UnregisterToast((*toastId));
  2900. SceneMemberMappingRequestBus::EventResult(pair.m_graphCanvasId, pair.m_scriptCanvasId, &SceneMemberMappingRequests::GetGraphCanvasEntityId);
  2901. AZStd::vector<AZ::EntityId> focusElements = { pair.m_graphCanvasId };
  2902. m_focusHelper.Clear();
  2903. m_focusHelper.SetNodes(focusElements);
  2904. m_focusHelper.CycleToNextNode();
  2905. AZStd::vector< NodeIdPair > highlightPair = { pair };
  2906. HighlightNodes(highlightPair);
  2907. }
  2908. }
  2909. void EditorGraph::OnToastDismissed()
  2910. {
  2911. const AzToolsFramework::ToastId* toastId = AzToolsFramework::ToastNotificationBus::GetCurrentBusId();
  2912. if (toastId)
  2913. {
  2914. UnregisterToast((*toastId));
  2915. }
  2916. }
  2917. void EditorGraph::OnUndoRedoEnd()
  2918. {
  2919. for (const auto& nodePair : GetNodeMapping())
  2920. {
  2921. nodePair.second->SignalDeserialized();
  2922. }
  2923. }
  2924. void EditorGraph::ReportError(const ScriptCanvas::Node& node, const AZStd::string& errorSource, const AZStd::string& errorMessage)
  2925. {
  2926. AzQtComponents::ToastConfiguration toastConfiguration(AzQtComponents::ToastType::Error, errorSource.c_str(), errorMessage.c_str());
  2927. GraphCanvas::ViewId viewId;
  2928. GraphCanvas::SceneRequestBus::EventResult(viewId, GetGraphCanvasGraphId(), &GraphCanvas::SceneRequests::GetViewId);
  2929. AzToolsFramework::ToastId toastId;
  2930. GraphCanvas::ViewRequestBus::EventResult(toastId, viewId, &GraphCanvas::ViewRequests::ShowToastNotification, toastConfiguration);
  2931. AzToolsFramework::ToastNotificationBus::MultiHandler::BusConnect(toastId);
  2932. m_toastNodeIds[toastId] = node.GetEntityId();
  2933. }
  2934. void EditorGraph::UnregisterToast(const AzToolsFramework::ToastId& toastId)
  2935. {
  2936. AzToolsFramework::ToastNotificationBus::MultiHandler::BusDisconnect(toastId);
  2937. m_toastNodeIds.erase(toastId);
  2938. }
  2939. void EditorGraph::DisplayUpdateToast()
  2940. {
  2941. GraphCanvas::ViewId viewId;
  2942. GraphCanvas::SceneRequestBus::EventResult(viewId, GetGraphCanvasGraphId(), &GraphCanvas::SceneRequests::GetViewId);
  2943. if (viewId.IsValid() && !m_updateStrings.empty())
  2944. {
  2945. bool isVisible = false;
  2946. GraphCanvas::ViewRequestBus::EventResult(isVisible, viewId, &GraphCanvas::ViewRequests::IsShowing);
  2947. if (isVisible)
  2948. {
  2949. AZStd::string displayString;
  2950. for (const auto& updateData : m_updateStrings)
  2951. {
  2952. if (!displayString.empty())
  2953. {
  2954. displayString.append("\n");
  2955. }
  2956. displayString.append("- ");
  2957. displayString.append(updateData);
  2958. }
  2959. m_updateStrings.clear();
  2960. AzQtComponents::ToastConfiguration toastConfiguration(AzQtComponents::ToastType::Information, "Nodes Updates", displayString.c_str());
  2961. GraphCanvas::ViewRequestBus::Event(viewId, &GraphCanvas::ViewRequests::ShowToastNotification, toastConfiguration);
  2962. }
  2963. }
  2964. }
  2965. const GraphStatisticsHelper& EditorGraph::GetNodeUsageStatistics() const
  2966. {
  2967. return m_statisticsHelper;
  2968. }
  2969. void EditorGraph::CreateGraphCanvasScene()
  2970. {
  2971. if (!m_saveFormatConverted)
  2972. {
  2973. GraphCanvas::GraphId graphCanvasGraphId = GetGraphCanvasGraphId();
  2974. GraphCanvas::SceneNotificationBus::Handler::BusConnect(graphCanvasGraphId);
  2975. GraphCanvas::GraphModelRequestBus::Handler::BusConnect(graphCanvasGraphId);
  2976. GraphCanvas::SceneRequestBus::Event(graphCanvasGraphId, &GraphCanvas::SceneRequests::SetEditorId, ScriptCanvasEditor::AssetEditorId);
  2977. AZStd::any* userData = nullptr;
  2978. GraphCanvas::SceneRequestBus::EventResult(userData, graphCanvasGraphId, &GraphCanvas::SceneRequests::GetUserData);
  2979. if (userData)
  2980. {
  2981. (*userData) = GetScriptCanvasId();
  2982. }
  2983. }
  2984. else if (m_graphCanvasSceneEntity == nullptr)
  2985. {
  2986. GraphCanvas::GraphCanvasRequestBus::BroadcastResult(m_graphCanvasSceneEntity, &GraphCanvas::GraphCanvasRequests::CreateSceneAndActivate);
  2987. if (m_graphCanvasSceneEntity == nullptr)
  2988. {
  2989. return;
  2990. }
  2991. AZ::EntityId graphCanvasGraphId = GetGraphCanvasGraphId();
  2992. GraphCanvas::SceneRequestBus::Event(graphCanvasGraphId, &GraphCanvas::SceneRequests::SetEditorId, ScriptCanvasEditor::AssetEditorId);
  2993. DisplayGraphCanvasScene();
  2994. AZStd::any* userData = nullptr;
  2995. GraphCanvas::SceneRequestBus::EventResult(userData, graphCanvasGraphId, &GraphCanvas::SceneRequests::GetUserData);
  2996. if (userData)
  2997. {
  2998. (*userData) = GetScriptCanvasId();
  2999. }
  3000. }
  3001. m_focusHelper.SetActiveGraph(GetGraphCanvasGraphId());
  3002. }
  3003. bool EditorGraph::UpgradeGraph(SourceHandle source, UpgradeRequest upgradeRequest, const UpgradeGraphConfig& upgradeConfig)
  3004. {
  3005. m_upgradeSM.SetAsset(source);
  3006. m_upgradeSM.SetConfig(upgradeConfig);
  3007. if (upgradeRequest == UpgradeRequest::Forced || !GetVersion().IsLatest() || HasDeprecatedNode())
  3008. {
  3009. m_upgradeSM.Run(Start::StateID());
  3010. return true;
  3011. }
  3012. else
  3013. {
  3014. m_upgradeSM.Run(Skip::StateID());
  3015. return false;
  3016. }
  3017. }
  3018. void EditorGraph::ConnectGraphCanvasBuses()
  3019. {
  3020. GraphCanvas::GraphId graphCanvasGraphId = GetGraphCanvasGraphId();
  3021. GraphCanvas::GraphModelRequestBus::Handler::BusConnect(graphCanvasGraphId);
  3022. GraphCanvas::SceneNotificationBus::Handler::BusConnect(graphCanvasGraphId);
  3023. }
  3024. void EditorGraph::DisconnectGraphCanvasBuses()
  3025. {
  3026. GraphCanvas::GraphModelRequestBus::Handler::BusDisconnect();
  3027. GraphCanvas::SceneNotificationBus::Handler::BusDisconnect();
  3028. }
  3029. void EditorGraph::OnSystemTick()
  3030. {
  3031. if (!m_allowVersionUpdate)
  3032. {
  3033. m_allowVersionUpdate = true;
  3034. }
  3035. else
  3036. {
  3037. m_allowVersionUpdate = false;
  3038. AZ::SystemTickBus::Handler::BusDisconnect();
  3039. HandleQueuedUpdates();
  3040. }
  3041. }
  3042. void EditorGraph::DisplayGraphCanvasScene()
  3043. {
  3044. m_variableDataModel.Activate(GetScriptCanvasId());
  3045. RequestPushPreventUndoStateUpdate();
  3046. AZStd::unordered_map< AZ::EntityId, AZ::EntityId > scriptCanvasToGraphCanvasMapping;
  3047. bool graphNeedsDirtying = !GetVersion().IsLatest();
  3048. {
  3049. QScopedValueRollback<bool> ignoreRequests(m_ignoreSaveRequests, true);
  3050. GraphCanvas::GraphId graphCanvasGraphId = GetGraphCanvasGraphId();
  3051. GraphCanvas::GraphModelRequestBus::Handler::BusConnect(graphCanvasGraphId);
  3052. GraphCanvas::SceneNotificationBus::Handler::BusConnect(graphCanvasGraphId);
  3053. GraphCanvas::SceneRequestBus::Event(graphCanvasGraphId, &GraphCanvas::SceneRequests::SignalLoadStart);
  3054. for (auto& saveDataIter : m_graphCanvasSaveData)
  3055. {
  3056. GraphCanvas::EntitySaveDataRequestBus::Event(graphCanvasGraphId, &GraphCanvas::EntitySaveDataRequests::ReadSaveData, (*saveDataIter.second));
  3057. }
  3058. ScriptCanvas::NodeIdList nodeList = GetNodes();
  3059. AZStd::unordered_set<ScriptCanvas::Node*> outOfDateNodes;
  3060. AZStd::unordered_set<AZ::EntityId> deletedNodes;
  3061. AZStd::unordered_set<AZ::EntityId> assetSanitizationSet;
  3062. AZStd::unordered_set<ScriptCanvas::Node*> sanityCheckRequiredNodes;
  3063. ScriptCanvas::GraphUpdateReport graphUpdateReport;
  3064. for (const AZ::EntityId& scriptCanvasNodeId : nodeList)
  3065. {
  3066. assetSanitizationSet.insert(scriptCanvasNodeId);
  3067. ScriptCanvas::Node* scriptCanvasNode = FindNode(scriptCanvasNodeId);
  3068. if (scriptCanvasNode)
  3069. {
  3070. ScriptCanvas::NodeReplacementConfiguration nodeConfig;
  3071. auto replacementId = ScriptCanvasEditor::NodeReplacementSystem::GenerateReplacementId(scriptCanvasNode);
  3072. ScriptCanvasEditor::NodeReplacementRequestBus::BroadcastResult(
  3073. nodeConfig, &ScriptCanvasEditor::NodeReplacementRequestBus::Events::GetNodeReplacementConfiguration, replacementId);
  3074. if (!nodeConfig.IsValid())
  3075. {
  3076. nodeConfig = scriptCanvasNode->GetReplacementNodeConfiguration();
  3077. }
  3078. if (nodeConfig.IsValid() && !g_disableDeprecatedNodeUpdates)
  3079. {
  3080. ScriptCanvas::NodeUpdateReport nodeUpdateReport;
  3081. ScriptCanvasEditor::NodeReplacementRequestBus::BroadcastResult(nodeUpdateReport,
  3082. &ScriptCanvasEditor::NodeReplacementRequestBus::Events::ReplaceNodeByReplacementConfiguration,
  3083. GetScriptCanvasId(), scriptCanvasNode, nodeConfig);
  3084. if (!nodeUpdateReport.IsEmpty())
  3085. {
  3086. graphNeedsDirtying = true;
  3087. scriptCanvasNode = nodeUpdateReport.m_newNode;
  3088. m_updateStrings.insert(AZStd::string::format("Replaced node (%s)", scriptCanvasNode->GetNodeName().c_str()));
  3089. ScriptCanvas::MergeUpdateSlotReport(scriptCanvasNodeId, graphUpdateReport, nodeUpdateReport);
  3090. }
  3091. }
  3092. AZ::EntityId graphCanvasNodeId = Nodes::DisplayScriptCanvasNode(graphCanvasGraphId, scriptCanvasNode);
  3093. scriptCanvasToGraphCanvasMapping[scriptCanvasNodeId] = graphCanvasNodeId;
  3094. auto saveDataIter2 = m_graphCanvasSaveData.find(scriptCanvasNodeId);
  3095. if (saveDataIter2 != m_graphCanvasSaveData.end())
  3096. {
  3097. GraphCanvas::EntitySaveDataRequestBus::Event(graphCanvasNodeId, &GraphCanvas::EntitySaveDataRequests::ReadSaveData, (*saveDataIter2->second));
  3098. }
  3099. AZ::Vector2 position;
  3100. GraphCanvas::GeometryRequestBus::EventResult(position, graphCanvasNodeId, &GraphCanvas::GeometryRequests::GetPosition);
  3101. GraphCanvas::SceneRequestBus::Event(graphCanvasGraphId, &GraphCanvas::SceneRequests::AddNode, graphCanvasNodeId, position, false);
  3102. // If the node is deprecated, we want to stomp whatever style it had saved and apply the deprecated style
  3103. if (scriptCanvasNode->IsDeprecated())
  3104. {
  3105. GraphCanvas::NodeTitleRequestBus::Event(graphCanvasNodeId, &GraphCanvas::NodeTitleRequests::SetPaletteOverride, "DeprecatedNodeTitlePalette");
  3106. }
  3107. if (scriptCanvasNode->IsOutOfDate(GetVersion()))
  3108. {
  3109. OnVersionConversionBegin((*scriptCanvasNode));
  3110. outOfDateNodes.emplace(scriptCanvasNode);
  3111. }
  3112. if (scriptCanvasNode->IsSanityCheckRequired())
  3113. {
  3114. graphNeedsDirtying = true;
  3115. sanityCheckRequiredNodes.insert(scriptCanvasNode);
  3116. }
  3117. }
  3118. }
  3119. if (!graphUpdateReport.IsEmpty())
  3120. {
  3121. // currently, it is expected that there are no deleted old slots, those need manual correction
  3122. AZ_Error("ScriptCanvas", graphUpdateReport.m_deletedOldSlots.empty(), "Graph upgrade path: If old slots are deleted, manual upgrading is required");
  3123. UpdateConnectionStatus(*this, graphUpdateReport);
  3124. }
  3125. AZStd::unordered_set<AZ::EntityId> graphCanvasNodesToDelete;
  3126. for (auto scriptCanvasNode : outOfDateNodes)
  3127. {
  3128. auto graphCanvasNodeId = scriptCanvasToGraphCanvasMapping[scriptCanvasNode->GetEntityId()];
  3129. ScriptCanvas::UpdateResult updateResult = scriptCanvasNode->UpdateNode();
  3130. OnVersionConversionEnd((*scriptCanvasNode));
  3131. switch (updateResult)
  3132. {
  3133. case ScriptCanvas::UpdateResult::DeleteNode:
  3134. {
  3135. graphNeedsDirtying = true;
  3136. deletedNodes.insert(scriptCanvasNode->GetEntityId());
  3137. graphCanvasNodesToDelete.insert(graphCanvasNodeId);
  3138. break;
  3139. }
  3140. default:
  3141. {
  3142. graphNeedsDirtying = true;
  3143. break;
  3144. }
  3145. }
  3146. }
  3147. if (!graphCanvasNodesToDelete.empty())
  3148. {
  3149. GraphCanvas::SceneRequestBus::Event(graphCanvasGraphId, &GraphCanvas::SceneRequests::Delete, graphCanvasNodesToDelete);
  3150. }
  3151. AZStd::vector< AZ::EntityId > connectionIds = GetConnections();
  3152. for (const AZ::EntityId& connectionId : connectionIds)
  3153. {
  3154. ScriptCanvas::Endpoint scriptCanvasSourceEndpoint;
  3155. ScriptCanvas::Endpoint scriptCanvasTargetEndpoint;
  3156. ScriptCanvas::ConnectionRequestBus::EventResult(scriptCanvasSourceEndpoint, connectionId, &ScriptCanvas::ConnectionRequests::GetSourceEndpoint);
  3157. ScriptCanvas::ConnectionRequestBus::EventResult(scriptCanvasTargetEndpoint, connectionId, &ScriptCanvas::ConnectionRequests::GetTargetEndpoint);
  3158. ScriptCanvas::Slot* sourceSlot = FindSlot(scriptCanvasSourceEndpoint);
  3159. ScriptCanvas::Slot* targetSlot = FindSlot(scriptCanvasTargetEndpoint);
  3160. // Implicit connections don't have corresponding Graph Canvas entities
  3161. if ((sourceSlot && sourceSlot->CreatesImplicitConnections()) ||
  3162. (targetSlot && targetSlot->CreatesImplicitConnections()))
  3163. {
  3164. continue;
  3165. }
  3166. AZ::EntityId graphCanvasSourceNode;
  3167. auto scriptCanvasIter = scriptCanvasToGraphCanvasMapping.find(scriptCanvasSourceEndpoint.GetNodeId());
  3168. if (scriptCanvasIter != scriptCanvasToGraphCanvasMapping.end())
  3169. {
  3170. graphCanvasSourceNode = scriptCanvasIter->second;
  3171. }
  3172. else
  3173. {
  3174. AZ_Warning("ScriptCanvas", false, "Could not find ScriptCanvas Node with id %llu", static_cast<AZ::u64>(scriptCanvasSourceEndpoint.GetNodeId()));
  3175. }
  3176. AZ::EntityId graphCanvasSourceSlotId;
  3177. SlotMappingRequestBus::EventResult(graphCanvasSourceSlotId, graphCanvasSourceNode, &SlotMappingRequests::MapToGraphCanvasId, scriptCanvasSourceEndpoint.GetSlotId());
  3178. if (!graphCanvasSourceSlotId.IsValid())
  3179. {
  3180. // For the EBusHandler's I need to remap these to a different visual node.
  3181. // Since multiple GraphCanvas nodes depict a single ScriptCanvas EBus node.
  3182. if (EBusHandlerNodeDescriptorRequestBus::FindFirstHandler(graphCanvasSourceNode) != nullptr)
  3183. {
  3184. GraphCanvas::Endpoint graphCanvasEventEndpoint;
  3185. EBusHandlerNodeDescriptorRequestBus::EventResult(graphCanvasEventEndpoint, graphCanvasSourceNode, &EBusHandlerNodeDescriptorRequests::MapSlotToGraphCanvasEndpoint, scriptCanvasSourceEndpoint.GetSlotId());
  3186. graphCanvasSourceSlotId = graphCanvasEventEndpoint.GetSlotId();
  3187. }
  3188. if (!graphCanvasSourceSlotId.IsValid())
  3189. {
  3190. AZ_Warning("ScriptCanvas", deletedNodes.count(scriptCanvasSourceEndpoint.GetNodeId()) > 0, "Could not create connection(%s) for Node(%s).", connectionId.ToString().c_str(), scriptCanvasSourceEndpoint.GetNodeId().ToString().c_str());
  3191. DisconnectById(connectionId);
  3192. continue;
  3193. }
  3194. }
  3195. GraphCanvas::Endpoint graphCanvasTargetEndpoint;
  3196. scriptCanvasIter = scriptCanvasToGraphCanvasMapping.find(scriptCanvasTargetEndpoint.GetNodeId());
  3197. if (scriptCanvasIter != scriptCanvasToGraphCanvasMapping.end())
  3198. {
  3199. graphCanvasTargetEndpoint.m_nodeId = scriptCanvasIter->second;
  3200. }
  3201. else
  3202. {
  3203. AZ_Warning("ScriptCanvas", false, "Could not find ScriptCanvas Node with id %llu", static_cast<AZ::u64>(scriptCanvasSourceEndpoint.GetNodeId()));
  3204. }
  3205. SlotMappingRequestBus::EventResult(graphCanvasTargetEndpoint.m_slotId, graphCanvasTargetEndpoint.GetNodeId(), &SlotMappingRequests::MapToGraphCanvasId, scriptCanvasTargetEndpoint.GetSlotId());
  3206. if (!graphCanvasTargetEndpoint.IsValid())
  3207. {
  3208. // For the EBusHandler's I need to remap these to a different visual node.
  3209. // Since multiple GraphCanvas nodes depict a single ScriptCanvas EBus node.
  3210. if (EBusHandlerNodeDescriptorRequestBus::FindFirstHandler(graphCanvasTargetEndpoint.GetNodeId()) != nullptr)
  3211. {
  3212. EBusHandlerNodeDescriptorRequestBus::EventResult(graphCanvasTargetEndpoint, graphCanvasTargetEndpoint.GetNodeId(), &EBusHandlerNodeDescriptorRequests::MapSlotToGraphCanvasEndpoint, scriptCanvasTargetEndpoint.GetSlotId());
  3213. }
  3214. if (!graphCanvasTargetEndpoint.IsValid())
  3215. {
  3216. AZ_Warning("ScriptCanvas", deletedNodes.count(scriptCanvasTargetEndpoint.GetNodeId()) > 0, "Could not create connection(%s) for Node(%s).", connectionId.ToString().c_str(), scriptCanvasTargetEndpoint.GetNodeId().ToString().c_str());
  3217. DisconnectById(connectionId);
  3218. continue;
  3219. }
  3220. }
  3221. AZ::EntityId graphCanvasConnectionId;
  3222. GraphCanvas::SlotRequestBus::EventResult(graphCanvasConnectionId, graphCanvasSourceSlotId, &GraphCanvas::SlotRequests::DisplayConnectionWithEndpoint, graphCanvasTargetEndpoint);
  3223. if (graphCanvasConnectionId.IsValid())
  3224. {
  3225. AZStd::any* userData = nullptr;
  3226. GraphCanvas::ConnectionRequestBus::EventResult(userData, graphCanvasConnectionId, &GraphCanvas::ConnectionRequests::GetUserData);
  3227. if (userData)
  3228. {
  3229. (*userData) = connectionId;
  3230. SceneMemberMappingConfigurationRequestBus::Event(graphCanvasConnectionId, &SceneMemberMappingConfigurationRequests::ConfigureMapping, connectionId);
  3231. }
  3232. }
  3233. }
  3234. // Fix up leaked data elements
  3235. auto mapIter = m_graphCanvasSaveData.begin();
  3236. while (mapIter != m_graphCanvasSaveData.end())
  3237. {
  3238. // Deleted using the wrong id, which orphaned the SaveData. For now we want to go through and sanitize our save data to avoid keeping around a bunch
  3239. // of old save data for no reason.
  3240. //
  3241. // Need to bypass our internal save data for graph canvas information
  3242. if (scriptCanvasToGraphCanvasMapping.find(mapIter->first) == scriptCanvasToGraphCanvasMapping.end()
  3243. && mapIter->first != GetEntityId())
  3244. {
  3245. delete mapIter->second;
  3246. mapIter = m_graphCanvasSaveData.erase(mapIter);
  3247. }
  3248. else
  3249. {
  3250. ++mapIter;
  3251. }
  3252. }
  3253. auto currentIter = GetGraphData()->m_scriptEventAssets.begin();
  3254. while (currentIter != GetGraphData()->m_scriptEventAssets.end())
  3255. {
  3256. if (assetSanitizationSet.find(currentIter->first) == assetSanitizationSet.end())
  3257. {
  3258. currentIter->second = ScriptEvents::ScriptEventsAssetPtr{};
  3259. currentIter = GetGraphData()->m_scriptEventAssets.erase(currentIter);
  3260. graphNeedsDirtying = true;
  3261. }
  3262. else
  3263. {
  3264. ++currentIter;
  3265. }
  3266. }
  3267. for (auto node : sanityCheckRequiredNodes)
  3268. {
  3269. if (node)
  3270. {
  3271. node->SanityCheckDynamicDisplay();
  3272. }
  3273. }
  3274. GraphCanvas::SceneRequestBus::Event(graphCanvasGraphId, &GraphCanvas::SceneRequests::SignalLoadEnd);
  3275. EditorGraphNotificationBus::Event(GetScriptCanvasId(), &EditorGraphNotifications::OnGraphCanvasSceneDisplayed);
  3276. }
  3277. GraphCanvas::SceneRequestBus::Event(GetGraphCanvasGraphId(), &GraphCanvas::SceneRequests::ProcessEnableDisableQueue);
  3278. if (m_graphCanvasSaveVersion != GraphCanvas::EntitySaveDataContainer::CurrentVersion)
  3279. {
  3280. for (auto saveDataPair : m_graphCanvasSaveData)
  3281. {
  3282. auto graphCanvasIter = scriptCanvasToGraphCanvasMapping.find(saveDataPair.first);
  3283. OnSaveDataDirtied(graphCanvasIter->second);
  3284. }
  3285. m_graphCanvasSaveVersion = GraphCanvas::EntitySaveDataContainer::CurrentVersion;
  3286. graphNeedsDirtying = true;
  3287. }
  3288. RequestPopPreventUndoStateUpdate();
  3289. if (graphNeedsDirtying)
  3290. {
  3291. SignalDirty();
  3292. }
  3293. MarkVersion();
  3294. }
  3295. void EditorGraph::OnGraphCanvasSceneVisible()
  3296. {
  3297. DisplayUpdateToast();
  3298. }
  3299. AZStd::unordered_map< AZ::EntityId, GraphCanvas::EntitySaveDataContainer* > EditorGraph::GetGraphCanvasSaveData()
  3300. {
  3301. return m_graphCanvasSaveData;
  3302. }
  3303. void EditorGraph::UpdateGraphCanvasSaveData(const AZStd::unordered_map< AZ::EntityId, GraphCanvas::EntitySaveDataContainer* >& saveData)
  3304. {
  3305. QScopedValueRollback<bool> ignoreRequests(m_ignoreSaveRequests, true);
  3306. GraphCanvas::GraphId graphCanvasGraphId = GetGraphCanvasGraphId();
  3307. GraphCanvas::SceneNotificationBus::Handler::BusDisconnect(graphCanvasGraphId);
  3308. GraphCanvas::GraphModelRequestBus::Handler::BusDisconnect(graphCanvasGraphId);
  3309. for (auto& entry : m_graphCanvasSaveData)
  3310. {
  3311. delete entry.second;
  3312. }
  3313. m_graphCanvasSaveData = saveData;
  3314. DisplayGraphCanvasScene();
  3315. }
  3316. void EditorGraph::ClearGraphCanvasScene()
  3317. {
  3318. GraphCanvas::GraphId graphCanvasGraphId = GetGraphCanvasGraphId();
  3319. RequestPushPreventUndoStateUpdate();
  3320. // Wipe out all of the Graph Canvas Visuals
  3321. GraphCanvas::SceneRequestBus::Event(graphCanvasGraphId, &GraphCanvas::SceneRequests::ClearScene);
  3322. RequestPopPreventUndoStateUpdate();
  3323. }
  3324. }