3
0

EditorGraph.cpp 168 KB

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