EditorGraph.cpp 166 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317331833193320332133223323332433253326332733283329333033313332333333343335333633373338333933403341334233433344334533463347334833493350335133523353335433553356335733583359336033613362336333643365336633673368336933703371337233733374337533763377337833793380338133823383338433853386338733883389339033913392339333943395339633973398339934003401340234033404340534063407340834093410341134123413341434153416341734183419342034213422342334243425342634273428342934303431343234333434343534363437343834393440344134423443344434453446344734483449345034513452345334543455345634573458345934603461346234633464346534663467346834693470347134723473347434753476347734783479348034813482348334843485348634873488348934903491349234933494349534963497349834993500350135023503350435053506350735083509351035113512351335143515351635173518351935203521352235233524352535263527352835293530353135323533353435353536353735383539354035413542354335443545354635473548354935503551355235533554355535563557355835593560356135623563356435653566356735683569357035713572357335743575357635773578357935803581358235833584358535863587358835893590359135923593359435953596359735983599360036013602360336043605360636073608360936103611361236133614361536163617361836193620362136223623362436253626362736283629363036313632363336343635363636373638363936403641364236433644364536463647364836493650365136523653365436553656365736583659366036613662366336643665366636673668366936703671367236733674367536763677367836793680368136823683368436853686368736883689369036913692369336943695369636973698369937003701370237033704370537063707370837093710371137123713371437153716371737183719372037213722372337243725372637273728372937303731373237333734373537363737373837393740374137423743374437453746374737483749375037513752375337543755375637573758375937603761376237633764376537663767376837693770377137723773377437753776377737783779378037813782378337843785378637873788378937903791379237933794379537963797379837993800380138023803380438053806380738083809381038113812381338143815381638173818381938203821382238233824382538263827382838293830383138323833383438353836383738383839384038413842384338443845384638473848384938503851385238533854385538563857385838593860386138623863386438653866386738683869387038713872387338743875387638773878387938803881388238833884388538863887388838893890389138923893389438953896389738983899390039013902390339043905390639073908390939103911391239133914391539163917391839193920392139223923392439253926392739283929393039313932393339343935393639373938393939403941394239433944394539463947394839493950395139523953395439553956395739583959396039613962396339643965396639673968396939703971397239733974397539763977397839793980398139823983398439853986398739883989399039913992399339943995399639973998399940004001400240034004400540064007400840094010401140124013401440154016
  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. GraphCanvas::GraphCanvasRequestBus::BroadcastResult(dataDisplay, &GraphCanvas::GraphCanvasRequests::CreateAssetIdNodePropertyDisplay, static_cast<GraphCanvas::AssetIdDataInterface*>(dataInterface));
  1235. }
  1236. else if (slotType.IS_A(ScriptCanvas::Data::Type::BehaviorContextObject(ScriptCanvas::GraphScopedVariableId::TYPEINFO_Uuid())))
  1237. {
  1238. dataInterface = aznew ScriptCanvasGraphScopedVariableDataInterface(&m_variableDataModel, GetScriptCanvasId(), scriptCanvasNodeId, scriptCanvasSlotId);
  1239. GraphCanvas::GraphCanvasRequestBus::BroadcastResult(dataDisplay, &GraphCanvas::GraphCanvasRequests::CreateComboBoxNodePropertyDisplay, static_cast<GraphCanvas::ComboBoxDataInterface*>(dataInterface));
  1240. }
  1241. if (dataDisplay != nullptr)
  1242. {
  1243. return dataDisplay;
  1244. }
  1245. delete dataInterface;
  1246. }
  1247. return nullptr;
  1248. }
  1249. void EditorGraph::SignalDirty()
  1250. {
  1251. SourceHandle handle(m_owner, AZ::Uuid::CreateNull());
  1252. GeneralRequestBus::Broadcast(&GeneralRequests::SignalSceneDirty, handle);
  1253. }
  1254. void EditorGraph::HighlightNodesByType(const ScriptCanvas::NodeTypeIdentifier& nodeTypeIdentifier)
  1255. {
  1256. for (const auto& nodePair : GetNodeMapping())
  1257. {
  1258. if (nodePair.second->GetNodeType() == nodeTypeIdentifier)
  1259. {
  1260. HighlightScriptCanvasEntity(nodePair.first);
  1261. }
  1262. }
  1263. }
  1264. void EditorGraph::HighlightEBusNodes(const ScriptCanvas::EBusBusId& busId, const ScriptCanvas::EBusEventId& eventId)
  1265. {
  1266. ScriptCanvas::NodeTypeIdentifier ebusIdentifier = ScriptCanvas::NodeUtils::ConstructEBusIdentifier(busId);
  1267. for (const auto& nodePair : GetNodeMapping())
  1268. {
  1269. ScriptCanvas::Node* canvasNode = nodePair.second;
  1270. if (canvasNode->GetNodeType() == ebusIdentifier)
  1271. {
  1272. AZ::EntityId graphCanvasNodeId;
  1273. SceneMemberMappingRequestBus::EventResult(graphCanvasNodeId, canvasNode->GetEntityId(), &SceneMemberMappingRequests::GetGraphCanvasEntityId);
  1274. bool hasEvent = false;
  1275. EBusHandlerNodeDescriptorRequestBus::EventResult(hasEvent, graphCanvasNodeId, &EBusHandlerNodeDescriptorRequests::ContainsEvent, eventId);
  1276. if (hasEvent)
  1277. {
  1278. HighlightScriptCanvasEntity(canvasNode->GetEntityId());
  1279. }
  1280. }
  1281. }
  1282. }
  1283. void EditorGraph::HighlightScriptEventNodes(const ScriptCanvas::EBusBusId& busId, const ScriptCanvas::EBusEventId& eventId)
  1284. {
  1285. ScriptCanvas::NodeTypeIdentifier sendScriptEventIdentifier = ScriptCanvas::NodeUtils::ConstructSendScriptEventIdentifier(busId, eventId);
  1286. ScriptCanvas::NodeTypeIdentifier receiveScriptEventIdentifier = ScriptCanvas::NodeUtils::ConstructScriptEventIdentifier(busId);
  1287. for (AZ::Entity* entity : GetGraphData()->m_nodes)
  1288. {
  1289. ScriptCanvas::Node* canvasNode = AZ::EntityUtils::FindFirstDerivedComponent<ScriptCanvas::Node>(entity);
  1290. if (canvasNode->GetNodeType() == sendScriptEventIdentifier)
  1291. {
  1292. HighlightScriptCanvasEntity(entity->GetId());
  1293. }
  1294. else if (canvasNode->GetNodeType() == receiveScriptEventIdentifier)
  1295. {
  1296. AZ::EntityId graphCanvasNodeId;
  1297. SceneMemberMappingRequestBus::EventResult(graphCanvasNodeId, canvasNode->GetEntityId(), &SceneMemberMappingRequests::GetGraphCanvasEntityId);
  1298. bool hasEvent = false;
  1299. EBusHandlerNodeDescriptorRequestBus::EventResult(hasEvent, graphCanvasNodeId, &EBusHandlerNodeDescriptorRequests::ContainsEvent, eventId);
  1300. if (hasEvent)
  1301. {
  1302. HighlightScriptCanvasEntity(entity->GetId());
  1303. }
  1304. }
  1305. }
  1306. }
  1307. void EditorGraph::HighlightScriptCanvasEntity(const AZ::EntityId& scriptCanvasId)
  1308. {
  1309. GraphCanvas::SceneMemberGlowOutlineConfiguration glowConfiguration;
  1310. glowConfiguration.m_blurRadius = 0; // #17174 using blur degrades performance
  1311. glowConfiguration.m_pen = QPen();
  1312. glowConfiguration.m_pen.setBrush(QColor(243,129,29));
  1313. glowConfiguration.m_pen.setWidth(5);
  1314. SceneMemberMappingRequestBus::EventResult(glowConfiguration.m_sceneMember, scriptCanvasId, &SceneMemberMappingRequests::GetGraphCanvasEntityId);
  1315. glowConfiguration.m_pulseRate = AZStd::chrono::milliseconds(2500);
  1316. glowConfiguration.m_zValue = 0;
  1317. GraphCanvas::GraphicsEffectId graphicsEffectId;
  1318. GraphCanvas::SceneRequestBus::EventResult(graphicsEffectId, GetGraphCanvasGraphId(), &GraphCanvas::SceneRequests::CreateGlowOnSceneMember, glowConfiguration);
  1319. if (graphicsEffectId.IsValid())
  1320. {
  1321. m_highlights.insert(graphicsEffectId);
  1322. }
  1323. }
  1324. AZ::EntityId EditorGraph::FindGraphCanvasSlotId(const AZ::EntityId& graphCanvasNodeId, const ScriptCanvas::SlotId& slotId)
  1325. {
  1326. AZ::EntityId graphCanvasSlotId;
  1327. SlotMappingRequestBus::EventResult(graphCanvasSlotId, graphCanvasNodeId, &SlotMappingRequests::MapToGraphCanvasId, slotId);
  1328. if (!graphCanvasSlotId.IsValid())
  1329. {
  1330. // For the EBusHandler's I need to remap these to a different visual node.
  1331. // Since multiple GraphCanvas nodes depict a single ScriptCanvas EBus node.
  1332. if (EBusHandlerNodeDescriptorRequestBus::FindFirstHandler(graphCanvasNodeId) != nullptr)
  1333. {
  1334. GraphCanvas::Endpoint graphCanvasEventEndpoint;
  1335. EBusHandlerNodeDescriptorRequestBus::EventResult(graphCanvasEventEndpoint, graphCanvasNodeId, &EBusHandlerNodeDescriptorRequests::MapSlotToGraphCanvasEndpoint, slotId);
  1336. graphCanvasSlotId = graphCanvasEventEndpoint.GetSlotId();
  1337. }
  1338. }
  1339. return graphCanvasSlotId;
  1340. }
  1341. bool EditorGraph::ConfigureConnectionUserData(const ScriptCanvas::Endpoint& sourceEndpoint, const ScriptCanvas::Endpoint& targetEndpoint, GraphCanvas::ConnectionId connectionId)
  1342. {
  1343. bool isConfigured = true;
  1344. AZ::Entity* scConnectionEntity = nullptr;
  1345. FindConnection(scConnectionEntity, sourceEndpoint, targetEndpoint);
  1346. if (scConnectionEntity)
  1347. {
  1348. AZStd::any* connectionUserData = nullptr;
  1349. GraphCanvas::ConnectionRequestBus::EventResult(connectionUserData, connectionId, &GraphCanvas::ConnectionRequests::GetUserData);
  1350. if (connectionUserData)
  1351. {
  1352. *connectionUserData = scConnectionEntity->GetId();
  1353. SceneMemberMappingConfigurationRequestBus::Event(connectionId, &SceneMemberMappingConfigurationRequests::ConfigureMapping, scConnectionEntity->GetId());
  1354. }
  1355. }
  1356. else
  1357. {
  1358. isConfigured = false;
  1359. }
  1360. return isConfigured;
  1361. }
  1362. void EditorGraph::HandleQueuedUpdates()
  1363. {
  1364. bool signalDirty = false;
  1365. RequestPushPreventUndoStateUpdate();
  1366. AZStd::unordered_set< ScriptCanvas::Node* > newUpdates;
  1367. bool updatePropertyGrid = false;
  1368. for (const AZ::EntityId& queuedUpdate : m_queuedConvertingNodes)
  1369. {
  1370. bool isSelected = false;
  1371. GraphCanvas::SceneMemberUIRequestBus::EventResult(isSelected, queuedUpdate, &GraphCanvas::SceneMemberUIRequests::IsSelected);
  1372. if (isSelected)
  1373. {
  1374. updatePropertyGrid = true;
  1375. }
  1376. AZ::EntityId scriptCanvasNodeId = ConvertToScriptCanvasNodeId(queuedUpdate);
  1377. ScriptCanvas::Node* node = FindNode(scriptCanvasNodeId);
  1378. if (node->IsOutOfDate(GetVersion()))
  1379. {
  1380. if (OnVersionConversionBegin((*node)))
  1381. {
  1382. newUpdates.insert(node);
  1383. }
  1384. }
  1385. }
  1386. m_queuedConvertingNodes.clear();
  1387. AZStd::unordered_set<AZ::EntityId> deletedNodes;
  1388. for (ScriptCanvas::Node* node : newUpdates)
  1389. {
  1390. ScriptCanvas::UpdateResult updateResult = node->UpdateNode();
  1391. OnVersionConversionEnd((*node));
  1392. AZ::EntityId graphCanvasNodeId;
  1393. SceneMemberMappingRequestBus::EventResult(graphCanvasNodeId, node->GetEntityId(), &SceneMemberMappingRequests::GetGraphCanvasEntityId);
  1394. switch (updateResult)
  1395. {
  1396. case ScriptCanvas::UpdateResult::DeleteNode:
  1397. {
  1398. if (graphCanvasNodeId.IsValid())
  1399. {
  1400. deletedNodes.insert(graphCanvasNodeId);
  1401. }
  1402. signalDirty = true;
  1403. break;
  1404. }
  1405. default:
  1406. {
  1407. signalDirty = true;
  1408. break;
  1409. }
  1410. }
  1411. }
  1412. if (!deletedNodes.empty())
  1413. {
  1414. GraphCanvas::SceneRequestBus::Event(GetGraphCanvasGraphId(), &GraphCanvas::SceneRequests::Delete, deletedNodes);
  1415. }
  1416. RequestPopPreventUndoStateUpdate();
  1417. if (signalDirty)
  1418. {
  1419. SignalDirty();
  1420. }
  1421. if (updatePropertyGrid)
  1422. {
  1423. PropertyGridRequestBus::Broadcast(&PropertyGridRequests::RefreshPropertyGrid);
  1424. }
  1425. }
  1426. bool EditorGraph::IsNodeVersionConverting(const AZ::EntityId& graphCanvasNodeId) const
  1427. {
  1428. bool isConverting = false;
  1429. if (!m_convertingNodes.empty())
  1430. {
  1431. if (GraphCanvas::GraphUtils::IsNodeWrapped(graphCanvasNodeId))
  1432. {
  1433. AZ::EntityId parentId;
  1434. GraphCanvas::NodeRequestBus::EventResult(parentId, graphCanvasNodeId, &GraphCanvas::NodeRequests::GetWrappingNode);
  1435. if (m_convertingNodes.count(parentId) > 0)
  1436. {
  1437. isConverting = true;
  1438. }
  1439. }
  1440. else if (m_convertingNodes.count(graphCanvasNodeId) > 0)
  1441. {
  1442. isConverting = true;
  1443. }
  1444. }
  1445. return isConverting;
  1446. }
  1447. void EditorGraph::OnPreNodeDeleted(const AZ::EntityId& nodeId)
  1448. {
  1449. // If we are cdeleteing a HandlerEventNode we don't need to do anything since they are purely visual.
  1450. // And the underlying ScriptCanvas nodes will persist and maintain all of their state.
  1451. if (EBusHandlerEventNodeDescriptorRequestBus::FindFirstHandler(nodeId) == nullptr)
  1452. {
  1453. AZStd::any* userData = nullptr;
  1454. GraphCanvas::NodeRequestBus::EventResult(userData, nodeId, &GraphCanvas::NodeRequests::GetUserData);
  1455. if (userData && userData->is<AZ::EntityId>())
  1456. {
  1457. const AZ::EntityId* scriptCanvasNodeId = AZStd::any_cast<AZ::EntityId>(userData);
  1458. auto iter = m_graphCanvasSaveData.find((*scriptCanvasNodeId));
  1459. if (iter != m_graphCanvasSaveData.end())
  1460. {
  1461. delete iter->second;
  1462. m_graphCanvasSaveData.erase(iter);
  1463. }
  1464. }
  1465. AZStd::any* sourceUserData = nullptr;
  1466. GraphCanvas::NodeRequestBus::EventResult(sourceUserData, nodeId, &GraphCanvas::NodeRequests::GetUserData);
  1467. auto scriptCanvasNodeId = sourceUserData && sourceUserData->is<AZ::EntityId>() ? *AZStd::any_cast<AZ::EntityId>(sourceUserData) : AZ::EntityId();
  1468. if (RemoveNode(scriptCanvasNodeId))
  1469. {
  1470. AZ::ComponentApplicationBus::Broadcast(&AZ::ComponentApplicationRequests::DeleteEntity, scriptCanvasNodeId);
  1471. }
  1472. }
  1473. }
  1474. void EditorGraph::OnPreConnectionDeleted(const AZ::EntityId& connectionId)
  1475. {
  1476. AZStd::any* userData = nullptr;
  1477. GraphCanvas::ConnectionRequestBus::EventResult(userData, connectionId, &GraphCanvas::ConnectionRequests::GetUserData);
  1478. if (userData && userData->is<AZ::EntityId>())
  1479. {
  1480. const AZ::EntityId* scriptCanvasConnectionId = AZStd::any_cast<AZ::EntityId>(userData);
  1481. auto iter = m_graphCanvasSaveData.find((*scriptCanvasConnectionId));
  1482. if (iter != m_graphCanvasSaveData.end())
  1483. {
  1484. delete iter->second;
  1485. m_graphCanvasSaveData.erase(iter);
  1486. }
  1487. }
  1488. GraphCanvas::Endpoint sourceEndpoint;
  1489. GraphCanvas::ConnectionRequestBus::EventResult(sourceEndpoint, connectionId, &GraphCanvas::ConnectionRequests::GetSourceEndpoint);
  1490. ScriptCanvas::Endpoint scriptCanvasEndpoint = ConvertToScriptCanvasEndpoint(sourceEndpoint);
  1491. // Don't disconnect any connections if we are version converting a node involved
  1492. if (IsNodeVersionConverting(scriptCanvasEndpoint.GetNodeId()))
  1493. {
  1494. return;
  1495. }
  1496. GraphCanvas::Endpoint targetEndpoint;
  1497. GraphCanvas::ConnectionRequestBus::EventResult(targetEndpoint, connectionId, &GraphCanvas::ConnectionRequests::GetTargetEndpoint);
  1498. scriptCanvasEndpoint = ConvertToScriptCanvasEndpoint(targetEndpoint);
  1499. if (IsNodeVersionConverting(scriptCanvasEndpoint.GetNodeId()))
  1500. {
  1501. return;
  1502. }
  1503. DisconnectConnection(connectionId);
  1504. }
  1505. void EditorGraph::OnUnknownPaste([[maybe_unused]] const QPointF& scenePos)
  1506. {
  1507. GraphVariablesTableView::HandleVariablePaste(GetScriptCanvasId());
  1508. }
  1509. void EditorGraph::OnSelectionChanged()
  1510. {
  1511. ClearHighlights();
  1512. }
  1513. AZ::u32 EditorGraph::GetNewVariableCounter()
  1514. {
  1515. return ++m_variableCounter;
  1516. }
  1517. void EditorGraph::ReleaseVariableCounter(AZ::u32 variableCounter)
  1518. {
  1519. if (m_variableCounter == variableCounter)
  1520. {
  1521. --m_variableCounter;
  1522. }
  1523. }
  1524. void EditorGraph::RequestUndoPoint()
  1525. {
  1526. GeneralRequestBus::Broadcast(&GeneralRequests::PostUndoPoint, GetScriptCanvasId());
  1527. }
  1528. void EditorGraph::RequestPushPreventUndoStateUpdate()
  1529. {
  1530. GeneralRequestBus::Broadcast(&GeneralRequests::PushPreventUndoStateUpdate);
  1531. }
  1532. void EditorGraph::RequestPopPreventUndoStateUpdate()
  1533. {
  1534. GeneralRequestBus::Broadcast(&GeneralRequests::PopPreventUndoStateUpdate);
  1535. }
  1536. void EditorGraph::TriggerUndo()
  1537. {
  1538. GeneralRequestBus::Broadcast(&GeneralRequests::TriggerUndo);
  1539. }
  1540. void EditorGraph::TriggerRedo()
  1541. {
  1542. GeneralRequestBus::Broadcast(&GeneralRequests::TriggerRedo);
  1543. }
  1544. void EditorGraph::EnableNodes(const AZStd::unordered_set< GraphCanvas::NodeId >& nodeIds)
  1545. {
  1546. bool enabledNodes = false;
  1547. for (auto graphCanvasNodeId : nodeIds)
  1548. {
  1549. AZStd::any* nodeUserData = nullptr;
  1550. GraphCanvas::NodeRequestBus::EventResult(nodeUserData, graphCanvasNodeId, &GraphCanvas::NodeRequests::GetUserData);
  1551. if (auto* scNodeId = AZStd::any_cast<AZ::EntityId>(nodeUserData))
  1552. {
  1553. bool hasNonUserDisabledFlag = false;
  1554. ScriptCanvas::NodeRequestBus::EventResult(hasNonUserDisabledFlag, (*scNodeId), &ScriptCanvas::NodeRequests::HasNodeDisabledFlag, ScriptCanvas::NodeDisabledFlag::NonUser);
  1555. if (!hasNonUserDisabledFlag)
  1556. {
  1557. ScriptCanvas::NodeRequestBus::Event((*scNodeId), &ScriptCanvas::NodeRequests::RemoveNodeDisabledFlag, ScriptCanvas::NodeDisabledFlag::User);
  1558. enabledNodes = true;
  1559. }
  1560. }
  1561. }
  1562. if (enabledNodes)
  1563. {
  1564. GeneralRequestBus::Broadcast(&GeneralRequests::PostUndoPoint, GetScriptCanvasId());
  1565. }
  1566. }
  1567. void EditorGraph::DisableNodes(const AZStd::unordered_set< GraphCanvas::NodeId >& nodeIds)
  1568. {
  1569. bool disabledNodes = false;
  1570. for (auto graphCanvasNodeId : nodeIds)
  1571. {
  1572. AZStd::any* nodeUserData = nullptr;
  1573. GraphCanvas::NodeRequestBus::EventResult(nodeUserData, graphCanvasNodeId, &GraphCanvas::NodeRequests::GetUserData);
  1574. if (auto* scNodeId = AZStd::any_cast<AZ::EntityId>(nodeUserData))
  1575. {
  1576. ScriptCanvas::NodeRequestBus::Event((*scNodeId), &ScriptCanvas::NodeRequests::AddNodeDisabledFlag, ScriptCanvas::NodeDisabledFlag::User);
  1577. disabledNodes = true;
  1578. }
  1579. }
  1580. if (disabledNodes)
  1581. {
  1582. GeneralRequestBus::Broadcast(&GeneralRequests::PostUndoPoint, GetScriptCanvasId());
  1583. }
  1584. }
  1585. void EditorGraph::PostDeletionEvent()
  1586. {
  1587. GeneralRequestBus::Broadcast(&GeneralRequests::PostUndoPoint, GetScriptCanvasId());
  1588. // Work-around for a crash caused by the MainWindow::OnSystemTick not being handled before the ReflectedPropertyEditor's DoRefresh.
  1589. // This will force a refresh selection on any post-deletion events so that the DoRefresh will not crash on deleted objects
  1590. UIRequestBus::Broadcast(&UIRequests::RefreshSelection);
  1591. }
  1592. void EditorGraph::PostCreationEvent()
  1593. {
  1594. GeneralRequestBus::Broadcast(&GeneralRequests::PushPreventUndoStateUpdate);
  1595. if (m_wrapperNodeDropTarget.IsValid())
  1596. {
  1597. for (const AZ::EntityId& nodeId : m_lastGraphCanvasCreationGroup)
  1598. {
  1599. GraphCanvas::WrappedNodeConfiguration configuration;
  1600. GraphCanvas::WrapperNodeConfigurationRequestBus::EventResult(configuration, m_wrapperNodeDropTarget, &GraphCanvas::WrapperNodeConfigurationRequests::GetWrappedNodeConfiguration, nodeId);
  1601. GraphCanvas::WrapperNodeRequestBus::Event(m_wrapperNodeDropTarget, &GraphCanvas::WrapperNodeRequests::WrapNode, nodeId, configuration);
  1602. }
  1603. }
  1604. else
  1605. {
  1606. // List of nodes we want to delete, because they are invalid in our current context
  1607. AZStd::unordered_set<AZ::EntityId> invalidNodes;
  1608. // Three maps here.
  1609. // WrapperTypeMapping: Keeps track of which wrappers were created by wrapper type.
  1610. AZStd::unordered_map< AZ::Crc32, AZ::EntityId > wrapperTypeMapping;
  1611. // WrapperIdMapping: Keeps track of EntityId mappings for the Wrappers.
  1612. AZStd::unordered_map< AZ::EntityId, AZ::EntityId > wrapperIdMapping;
  1613. // RequiredWrappers: Keeps track of a map of all of the wrapper types required to be created, along with the nodes
  1614. // that wanted to create the nodes.
  1615. AZStd::unordered_multimap< AZ::Crc32, AZ::EntityId > requiredWrappersMapping;
  1616. // In general, we will only ever use 2 at once(in the case of a drag/drop: busType + eventWrapper)
  1617. // In the case of a paste: busIdWrappers + eventWrappers
  1618. // Logic is merged here just to try to reduce the duplicated logic, and because I can't really
  1619. // tell the difference between the two cases anyway.
  1620. //
  1621. // Idea here is to keep track of groupings so that when we paste, I can create the appropriate number
  1622. // of nodes and groupings within these nodes to create a proper duplicate. And when we drag and drop
  1623. // I want to merge as many wrapped elements onto a single node as I can.
  1624. //
  1625. // First step in this process is to sort our pasted nodes into EBus handlers and EBus events.
  1626. for (const AZ::EntityId& nodeId : m_lastGraphCanvasCreationGroup)
  1627. {
  1628. bool isExecutionNodeling = false;
  1629. NodeDescriptorRequestBus::EventResult(isExecutionNodeling, nodeId, &NodeDescriptorRequests::IsType, NodeDescriptorType::FunctionDefinitionNode);
  1630. if (isExecutionNodeling)
  1631. {
  1632. AZStd::any* userData = nullptr;
  1633. GraphCanvas::NodeRequestBus::EventResult(userData, nodeId, &GraphCanvas::NodeRequests::GetUserData);
  1634. AZ::EntityId scSourceNodeId = (userData && userData->is<AZ::EntityId>()) ? *AZStd::any_cast<AZ::EntityId>(userData) : AZ::EntityId();
  1635. ScriptCanvas::Nodes::Core::FunctionDefinitionNode* nodeling = azrtti_cast<ScriptCanvas::Nodes::Core::FunctionDefinitionNode*>(FindNode(scSourceNodeId));
  1636. if (nodeling)
  1637. {
  1638. nodeling->RemapId();
  1639. }
  1640. }
  1641. bool isFunctionNode = false;
  1642. NodeDescriptorRequestBus::EventResult(isFunctionNode, nodeId, &NodeDescriptorRequests::IsType, NodeDescriptorType::FunctionNode);
  1643. // Show all hidden slots on a paste, as a temporary fix until I can sort out what I want this to work like.
  1644. GraphCanvas::NodeRequestBus::Event(nodeId, &GraphCanvas::NodeRequests::ShowAllSlots);
  1645. if (GraphCanvas::WrapperNodeRequestBus::FindFirstHandler(nodeId) != nullptr)
  1646. {
  1647. wrapperIdMapping[nodeId] = nodeId;
  1648. AZ::Crc32 wrapperType;
  1649. GraphCanvas::WrapperNodeRequestBus::EventResult(wrapperType, nodeId, &GraphCanvas::WrapperNodeRequests::GetWrapperType);
  1650. if (wrapperType != AZ::Crc32())
  1651. {
  1652. auto mapIter = wrapperTypeMapping.find(wrapperType);
  1653. if (mapIter == wrapperTypeMapping.end())
  1654. {
  1655. wrapperTypeMapping[wrapperType] = nodeId;
  1656. }
  1657. }
  1658. }
  1659. if (GraphCanvas::ForcedWrappedNodeRequestBus::FindFirstHandler(nodeId) != nullptr)
  1660. {
  1661. bool isWrapped = false;
  1662. GraphCanvas::NodeRequestBus::EventResult(isWrapped, nodeId, &GraphCanvas::NodeRequests::IsWrapped);
  1663. if (!isWrapped)
  1664. {
  1665. AZ::Crc32 wrapperType;
  1666. GraphCanvas::ForcedWrappedNodeRequestBus::EventResult(wrapperType, nodeId, &GraphCanvas::ForcedWrappedNodeRequests::GetWrapperType);
  1667. if (wrapperType != AZ::Crc32())
  1668. {
  1669. requiredWrappersMapping.emplace(wrapperType,nodeId);
  1670. }
  1671. }
  1672. }
  1673. }
  1674. // Second step is to go through, and determine which usage case is valid so we know how to filter down our events.
  1675. // If we can't find a wrapper, or we can't create a handler for the wrapper. We need to delete it.
  1676. for (const auto& mapPair : requiredWrappersMapping)
  1677. {
  1678. AZ::EntityId wrapperNodeId;
  1679. // Look up in our previous group mapping to see if it belonged to a node previously
  1680. // (i.e. copy + pasted node).
  1681. AZ::EntityId previousGroupWrapperNodeId;
  1682. auto mapIter = m_wrappedNodeGroupings.find(mapPair.second);
  1683. if (mapIter != m_wrappedNodeGroupings.end())
  1684. {
  1685. previousGroupWrapperNodeId = mapIter->second;
  1686. auto busIter = wrapperIdMapping.find(previousGroupWrapperNodeId);
  1687. if (busIter != wrapperIdMapping.end())
  1688. {
  1689. wrapperNodeId = busIter->second;
  1690. }
  1691. }
  1692. // We may have already found our target node.
  1693. // If we have, bypass the creation step.
  1694. if (!wrapperNodeId.IsValid())
  1695. {
  1696. // If we haven't check if we match a type, or if our previous group wrapper node is valid.
  1697. // If we had a previous group. I need to create a wrapper for that group.
  1698. // If we didn't have a previous group, I want to just use the Bus name to find an appropriate grouping.
  1699. auto busIter = wrapperTypeMapping.find(mapPair.first);
  1700. if (busIter == wrapperTypeMapping.end() || previousGroupWrapperNodeId.IsValid())
  1701. {
  1702. AZ::EntityId forcedWrappedNodeId = mapPair.second;
  1703. AZ::Vector2 position;
  1704. GraphCanvas::GeometryRequestBus::EventResult(position, forcedWrappedNodeId, &GraphCanvas::GeometryRequests::GetPosition);
  1705. GraphCanvas::ForcedWrappedNodeRequestBus::EventResult(wrapperNodeId, forcedWrappedNodeId, &GraphCanvas::ForcedWrappedNodeRequests::CreateWrapperNode, GetGraphCanvasGraphId(), position);
  1706. if (wrapperNodeId.IsValid())
  1707. {
  1708. m_lastGraphCanvasCreationGroup.emplace_back(wrapperNodeId);
  1709. if (!previousGroupWrapperNodeId.IsValid())
  1710. {
  1711. wrapperTypeMapping.emplace(mapPair.first, wrapperNodeId);
  1712. }
  1713. else
  1714. {
  1715. wrapperIdMapping.emplace(previousGroupWrapperNodeId, wrapperNodeId);
  1716. }
  1717. }
  1718. else
  1719. {
  1720. AZ_Error("ScriptCanvas", false, "Failed to instantiate an Wrapper node with type: (%d)", mapPair.first);
  1721. invalidNodes.insert(mapPair.second);
  1722. continue;
  1723. }
  1724. }
  1725. else
  1726. {
  1727. wrapperNodeId = busIter->second;
  1728. }
  1729. }
  1730. GraphCanvas::WrappedNodeConfiguration configuration;
  1731. GraphCanvas::WrapperNodeConfigurationRequestBus::EventResult(configuration, wrapperNodeId, &GraphCanvas::WrapperNodeConfigurationRequests::GetWrappedNodeConfiguration, mapPair.second);
  1732. GraphCanvas::WrapperNodeRequestBus::Event(wrapperNodeId, &GraphCanvas::WrapperNodeRequests::WrapNode, mapPair.second, configuration);
  1733. }
  1734. GraphCanvas::SceneRequestBus::Event(GetGraphCanvasGraphId(), &GraphCanvas::SceneRequests::Delete, invalidNodes);
  1735. }
  1736. ScriptCanvas::Node::ExploredDynamicGroupCache exploredCache;
  1737. for (AZ::EntityId graphCanvasNodeId : m_lastGraphCanvasCreationGroup)
  1738. {
  1739. AZStd::any* userData = nullptr;
  1740. GraphCanvas::NodeRequestBus::EventResult(userData, graphCanvasNodeId, &GraphCanvas::NodeRequests::GetUserData);
  1741. AZ::EntityId scSourceNodeId = (userData && userData->is<AZ::EntityId>()) ? *AZStd::any_cast<AZ::EntityId>(userData) : AZ::EntityId();
  1742. if (scSourceNodeId.IsValid())
  1743. {
  1744. ScriptCanvas::Node* node = FindNode(scSourceNodeId);
  1745. if (node)
  1746. {
  1747. node->SanityCheckDynamicDisplay(exploredCache);
  1748. node->PostActivate();
  1749. }
  1750. }
  1751. OnSaveDataDirtied(graphCanvasNodeId);
  1752. Nodes::UpdateSlotDatumLabels(graphCanvasNodeId);
  1753. }
  1754. m_wrappedNodeGroupings.clear();
  1755. m_lastGraphCanvasCreationGroup.clear();
  1756. m_wrapperNodeDropTarget.SetInvalid();
  1757. GeneralRequestBus::Broadcast(&GeneralRequests::PopPreventUndoStateUpdate);
  1758. GeneralRequestBus::Broadcast(&GeneralRequests::PostUndoPoint, GetScriptCanvasId());
  1759. }
  1760. void EditorGraph::PostRestore(const UndoData&)
  1761. {
  1762. AZStd::vector<AZ::EntityId> graphCanvasNodeIds;
  1763. GraphCanvas::SceneRequestBus::EventResult(graphCanvasNodeIds, GetGraphCanvasGraphId(), &GraphCanvas::SceneRequests::GetNodes);
  1764. for (AZ::EntityId graphCanvasNodeId : graphCanvasNodeIds)
  1765. {
  1766. Nodes::UpdateSlotDatumLabels(graphCanvasNodeId);
  1767. }
  1768. GraphCanvas::ViewId viewId;
  1769. GraphCanvas::SceneRequestBus::EventResult(viewId, GetGraphCanvasGraphId(), &GraphCanvas::SceneRequests::GetViewId);
  1770. GraphCanvas::ViewRequestBus::Event(viewId, &GraphCanvas::ViewRequests::RefreshView);
  1771. }
  1772. void EditorGraph::OnPasteBegin()
  1773. {
  1774. GeneralRequestBus::Broadcast(&GeneralRequests::PushPreventUndoStateUpdate);
  1775. }
  1776. void EditorGraph::OnPasteEnd()
  1777. {
  1778. GeneralRequestBus::Broadcast(&GeneralRequests::PopPreventUndoStateUpdate);
  1779. GeneralRequestBus::Broadcast(&GeneralRequests::PostUndoPoint, GetScriptCanvasId());
  1780. }
  1781. void EditorGraph::OnGraphCanvasNodeCreated(const AZ::EntityId& nodeId)
  1782. {
  1783. m_lastGraphCanvasCreationGroup.emplace_back(nodeId);
  1784. }
  1785. void EditorGraph::ResetSlotToDefaultValue(const GraphCanvas::Endpoint& endpoint)
  1786. {
  1787. ScriptCanvas::Endpoint scEndpoint = ConvertToScriptCanvasEndpoint(endpoint);
  1788. ScriptCanvas::Node* canvasNode = FindNode(scEndpoint.GetNodeId());
  1789. if (canvasNode)
  1790. {
  1791. canvasNode->ResetSlotToDefaultValue(scEndpoint.GetSlotId());
  1792. }
  1793. }
  1794. void EditorGraph::ResetReference(const GraphCanvas::Endpoint& endpoint)
  1795. {
  1796. // ResetSlotToDefault deals with resetting the reference internal to the function call on the node.
  1797. ResetSlotToDefaultValue(endpoint);
  1798. }
  1799. void EditorGraph::ResetProperty(const GraphCanvas::NodeId& nodeId, const AZ::Crc32& propertyId)
  1800. {
  1801. AZ::EntityId scriptCanvasNodeId = ConvertToScriptCanvasNodeId(nodeId);
  1802. ScriptCanvas::Node* canvasNode = FindNode(scriptCanvasNodeId);
  1803. if (canvasNode)
  1804. {
  1805. canvasNode->ResetProperty(propertyId);
  1806. }
  1807. }
  1808. void EditorGraph::RemoveSlot(const GraphCanvas::Endpoint& endpoint)
  1809. {
  1810. ScriptCanvas::Endpoint scEndpoint = ConvertToScriptCanvasEndpoint(endpoint);
  1811. ScriptCanvas::Slot* slot = FindSlot(scEndpoint);
  1812. if (slot)
  1813. {
  1814. ScriptCanvas::GraphVariable* variable = slot->GetVariable();
  1815. if (variable && variable->GetScope() == ScriptCanvas::VariableFlags::Scope::Function)
  1816. {
  1817. bool success = false;
  1818. ScriptCanvas::GraphVariableManagerRequestBus::EventResult(success, GetScriptCanvasId(), &ScriptCanvas::GraphVariableManagerRequests::RemoveVariable, variable->GetVariableId());
  1819. if (!success)
  1820. {
  1821. AZ_Assert(success, "Failed to remove variable that corresponds to this slot");
  1822. }
  1823. }
  1824. }
  1825. ScriptCanvas::Node* canvasNode = FindNode(scEndpoint.GetNodeId());
  1826. if (canvasNode)
  1827. {
  1828. canvasNode->DeleteSlot(scEndpoint.GetSlotId());
  1829. }
  1830. }
  1831. bool EditorGraph::IsSlotRemovable(const GraphCanvas::Endpoint& endpoint) const
  1832. {
  1833. ScriptCanvas::Endpoint scEndpoint = ConvertToScriptCanvasEndpoint(endpoint);
  1834. ScriptCanvas::Node* canvasNode = FindNode(scEndpoint.GetNodeId());
  1835. if (canvasNode)
  1836. {
  1837. return canvasNode->CanDeleteSlot(scEndpoint.GetSlotId());
  1838. }
  1839. return false;
  1840. }
  1841. bool EditorGraph::ConvertSlotToReference(const GraphCanvas::Endpoint& endpoint, bool isNewSlot)
  1842. {
  1843. ScriptCanvas::Endpoint scEndpoint = ConvertToScriptCanvasEndpoint(endpoint);
  1844. ScriptCanvas::Node* canvasNode = FindNode(scEndpoint.GetNodeId());
  1845. if (canvasNode)
  1846. {
  1847. return canvasNode->ConvertSlotToReference(scEndpoint.GetSlotId(), isNewSlot);
  1848. }
  1849. return false;
  1850. }
  1851. bool EditorGraph::CanConvertSlotToReference(const GraphCanvas::Endpoint& endpoint, bool isNewSlot)
  1852. {
  1853. ScriptCanvas::Endpoint scEndpoint = ConvertToScriptCanvasEndpoint(endpoint);
  1854. ScriptCanvas::Node* canvasNode = FindNode(scEndpoint.GetNodeId());
  1855. if (canvasNode)
  1856. {
  1857. ScriptCanvas::Slot* slot = canvasNode->GetSlot(scEndpoint.GetSlotId());
  1858. if (slot)
  1859. {
  1860. return slot->CanConvertToReference(isNewSlot);
  1861. }
  1862. }
  1863. return false;
  1864. }
  1865. GraphCanvas::CanHandleMimeEventOutcome EditorGraph::CanHandleReferenceMimeEvent(const GraphCanvas::Endpoint& endpoint, const QMimeData* mimeData)
  1866. {
  1867. ScriptCanvas::Endpoint scEndpoint = ConvertToScriptCanvasEndpoint(endpoint);
  1868. ScriptCanvas::Node* canvasNode = FindNode(scEndpoint.GetNodeId());
  1869. if (canvasNode)
  1870. {
  1871. ScriptCanvas::Slot* slot = canvasNode->GetSlot(scEndpoint.GetSlotId());
  1872. if (slot->CanConvertToReference() || slot->IsVariableReference())
  1873. {
  1874. ScriptCanvas::VariableId variableId = GraphCanvas::QtMimeUtils::ExtractTypeFromMimeData<ScriptCanvas::VariableId>(mimeData, GraphCanvas::k_ReferenceMimeType);
  1875. ScriptCanvas::GraphVariable* variable = FindVariableById(variableId);
  1876. if (variable)
  1877. {
  1878. return canvasNode->SlotAcceptsType(scEndpoint.GetSlotId(), variable->GetDataType());
  1879. }
  1880. else
  1881. {
  1882. return AZ::Failure(AZStd::string("Unable to find variable"));
  1883. }
  1884. }
  1885. else
  1886. {
  1887. return AZ::Failure(AZStd::string("Unable to convert slot to Reference"));
  1888. }
  1889. }
  1890. return AZ::Failure(AZStd::string("Unable to find Node"));
  1891. }
  1892. bool EditorGraph::HandleReferenceMimeEvent(const GraphCanvas::Endpoint& endpoint, const QMimeData* mimeData)
  1893. {
  1894. bool handledEvent = false;
  1895. ScriptCanvas::Endpoint scEndpoint = ConvertToScriptCanvasEndpoint(endpoint);
  1896. ScriptCanvas::Node* canvasNode = FindNode(scEndpoint.GetNodeId());
  1897. if (canvasNode)
  1898. {
  1899. ScriptCanvas::Slot* slot = canvasNode->GetSlot(scEndpoint.GetSlotId());
  1900. if (slot->IsVariableReference())
  1901. {
  1902. ScriptCanvas::VariableId variableId = GraphCanvas::QtMimeUtils::ExtractTypeFromMimeData<ScriptCanvas::VariableId>(mimeData, GraphCanvas::k_ReferenceMimeType);
  1903. if (variableId.IsValid())
  1904. {
  1905. canvasNode->SetSlotVariableId(scEndpoint.GetSlotId(), variableId);
  1906. handledEvent = true;
  1907. }
  1908. }
  1909. }
  1910. return handledEvent;
  1911. }
  1912. bool EditorGraph::CanPromoteToVariable(const GraphCanvas::Endpoint& endpoint, [[maybe_unused]] bool isNewSlot) const
  1913. {
  1914. ScriptCanvas::Endpoint scriptCanvasEndpoint = ConvertToScriptCanvasEndpoint(endpoint);
  1915. auto activeSlot = FindSlot(scriptCanvasEndpoint);
  1916. if (activeSlot && !activeSlot->IsVariableReference() && activeSlot->CanConvertToReference())
  1917. {
  1918. if (!activeSlot->IsDynamicSlot() || activeSlot->HasDisplayType())
  1919. {
  1920. bool isValidVariableType = false;
  1921. VariablePaletteRequestBus::BroadcastResult(isValidVariableType, &VariablePaletteRequests::IsValidVariableType, activeSlot->GetDataType());
  1922. return isValidVariableType;
  1923. }
  1924. }
  1925. return false;
  1926. }
  1927. bool EditorGraph::PromoteToVariableAction(const GraphCanvas::Endpoint& endpoint, bool isNewSlot)
  1928. {
  1929. ScriptCanvas::Endpoint scriptCanvasEndpoint = ConvertToScriptCanvasEndpoint(endpoint);
  1930. auto activeNode = FindNode(scriptCanvasEndpoint.GetNodeId());
  1931. auto activeSlot = FindSlot(scriptCanvasEndpoint);
  1932. if (activeNode == nullptr || activeSlot == nullptr)
  1933. {
  1934. return false;
  1935. }
  1936. if (activeSlot->IsVariableReference())
  1937. {
  1938. return false;
  1939. }
  1940. if (activeSlot->IsDynamicSlot() && !activeSlot->HasDisplayType())
  1941. {
  1942. return false;
  1943. }
  1944. const ScriptCanvas::Datum* activeDatum = activeNode->FindDatum(scriptCanvasEndpoint.GetSlotId());
  1945. AZStd::string variableName = "";
  1946. int variableCounter = 0;
  1947. AZStd::string defaultName;
  1948. AZ::Outcome<void, ScriptCanvas::GraphVariableValidationErrorCode> hasValidDefault = AZ::Failure(ScriptCanvas::GraphVariableValidationErrorCode::Unknown);
  1949. do
  1950. {
  1951. variableCounter = GetNewVariableCounter();
  1952. defaultName = VariableDockWidget::ConstructDefaultVariableName(variableCounter);
  1953. ScriptCanvas::GraphVariableManagerRequestBus::EventResult(hasValidDefault, GetScriptCanvasId(), &ScriptCanvas::GraphVariableManagerRequests::IsNameValid, defaultName);
  1954. } while (!hasValidDefault);
  1955. bool nameAvailable = false;
  1956. QWidget* mainWindow = nullptr;
  1957. UIRequestBus::BroadcastResult(mainWindow, &UIRequests::GetMainWindow);
  1958. AZStd::string inBoxText = "";
  1959. // Special case to try re-using the slot name if this is on an execution nodeling, since the user had just
  1960. // given it a name already with the ShowVariableConfigurationWidget dialog
  1961. if (azrtti_istypeof<ScriptCanvas::Nodes::Core::FunctionDefinitionNode>(activeNode))
  1962. {
  1963. variableName = activeSlot->GetName();
  1964. ScriptCanvas::GraphVariableManagerRequestBus::EventResult(nameAvailable, GetScriptCanvasId(), &ScriptCanvas::GraphVariableManagerRequests::IsNameAvailable, variableName);
  1965. }
  1966. int nameCount = 0;
  1967. while (!nameAvailable)
  1968. {
  1969. if (nameCount == 0)
  1970. {
  1971. variableName.append(AZStd::string::format(" (%d)", ++nameCount));
  1972. }
  1973. else
  1974. {
  1975. AZ::StringFunc::Replace(variableName, AZStd::string::format("(%d)", nameCount-1).c_str(), AZStd::string::format("(%d)", nameCount).c_str());
  1976. ++nameCount;
  1977. }
  1978. ScriptCanvas::GraphVariableManagerRequestBus::EventResult(nameAvailable, GetScriptCanvasId(), &ScriptCanvas::GraphVariableManagerRequests::IsNameAvailable, variableName);
  1979. }
  1980. activeSlot->Rename(variableName);
  1981. ScriptCanvas::Datum variableDatum;
  1982. if (activeDatum)
  1983. {
  1984. variableDatum.ReconfigureDatumTo((*activeDatum));
  1985. }
  1986. else
  1987. {
  1988. variableDatum.SetType(activeSlot->GetDataType());
  1989. // BCO Objects are defaulted to a reference. Going to bypass them to avoid messing with variable
  1990. // defaults.
  1991. if (ScriptCanvas::Data::IsValueType(activeSlot->GetDataType()))
  1992. {
  1993. variableDatum.SetToDefaultValueOfType();
  1994. }
  1995. }
  1996. AZ::Outcome<ScriptCanvas::VariableId, AZStd::string> addOutcome;
  1997. ScriptCanvas::GraphVariableManagerRequestBus::EventResult(addOutcome, GetScriptCanvasId(), &ScriptCanvas::GraphVariableManagerRequests::AddVariable, variableName, variableDatum, true);
  1998. if (addOutcome.IsSuccess())
  1999. {
  2000. GraphCanvas::DataSlotRequestBus::Event(endpoint.GetSlotId(), &GraphCanvas::DataSlotRequests::ConvertToReference, isNewSlot);
  2001. activeSlot->SetVariableReference(addOutcome.GetValue());
  2002. }
  2003. return addOutcome.IsSuccess();
  2004. }
  2005. bool EditorGraph::SynchronizeReferences(const GraphCanvas::Endpoint& referenceSource, const GraphCanvas::Endpoint& referenceTarget)
  2006. {
  2007. ScriptCanvas::Endpoint scriptCanvasSourceEndpoint = ConvertToScriptCanvasEndpoint(referenceSource);
  2008. ScriptCanvas::Endpoint scriptCanvasTargetEndpoint = ConvertToScriptCanvasEndpoint(referenceTarget);
  2009. auto sourceSlot = FindSlot(scriptCanvasSourceEndpoint);
  2010. auto targetSlot = FindSlot(scriptCanvasTargetEndpoint);
  2011. if (sourceSlot == nullptr
  2012. || targetSlot == nullptr)
  2013. {
  2014. return false;
  2015. }
  2016. if (!sourceSlot->IsVariableReference())
  2017. {
  2018. return false;
  2019. }
  2020. if (sourceSlot->IsTypeMatchFor((*targetSlot)))
  2021. {
  2022. if (!targetSlot->IsVariableReference())
  2023. {
  2024. GraphCanvas::DataSlotRequestBus::Event(referenceTarget.GetSlotId(), &GraphCanvas::DataSlotRequests::ConvertToReference, false);
  2025. }
  2026. if (targetSlot->IsVariableReference())
  2027. {
  2028. ScriptCanvas::VariableId variableId = sourceSlot->GetVariableReference();
  2029. targetSlot->SetVariableReference(variableId);
  2030. return true;
  2031. }
  2032. }
  2033. return false;
  2034. }
  2035. bool EditorGraph::ConvertSlotToValue(const GraphCanvas::Endpoint& endpoint)
  2036. {
  2037. ScriptCanvas::Endpoint scEndpoint = ConvertToScriptCanvasEndpoint(endpoint);
  2038. ScriptCanvas::Node* canvasNode = FindNode(scEndpoint.GetNodeId());
  2039. if (canvasNode)
  2040. {
  2041. return canvasNode->ConvertSlotToValue(scEndpoint.GetSlotId());
  2042. }
  2043. return false;
  2044. }
  2045. bool EditorGraph::CanConvertSlotToValue(const GraphCanvas::Endpoint& endpoint)
  2046. {
  2047. ScriptCanvas::Endpoint scEndpoint = ConvertToScriptCanvasEndpoint(endpoint);
  2048. ScriptCanvas::Node* canvasNode = FindNode(scEndpoint.GetNodeId());
  2049. if (canvasNode)
  2050. {
  2051. ScriptCanvas::Slot* slot = canvasNode->GetSlot(scEndpoint.GetSlotId());
  2052. return slot && slot->CanConvertToValue();
  2053. }
  2054. return false;
  2055. }
  2056. GraphCanvas::CanHandleMimeEventOutcome EditorGraph::CanHandleValueMimeEvent(const GraphCanvas::Endpoint& endpoint, const QMimeData* mimeData)
  2057. {
  2058. AZ_UNUSED(endpoint);
  2059. AZ_UNUSED(mimeData);
  2060. AZ_Assert(false, "Unimplemented drag and drop flow");
  2061. return AZ::Failure(AZStd::string("Unimplemented drag and drop flow"));
  2062. }
  2063. bool EditorGraph::HandleValueMimeEvent(const GraphCanvas::Endpoint& endpoint, const QMimeData* mimeData)
  2064. {
  2065. AZ_UNUSED(endpoint);
  2066. AZ_UNUSED(mimeData);
  2067. return false;
  2068. }
  2069. GraphCanvas::SlotId EditorGraph::RequestExtension(const GraphCanvas::NodeId& nodeId, const GraphCanvas::ExtenderId& extenderId, GraphModelRequests::ExtensionRequestReason reason)
  2070. {
  2071. GraphCanvas::SlotId graphCanvasSlotId;
  2072. AZStd::any* nodeUserData = nullptr;
  2073. GraphCanvas::NodeRequestBus::EventResult(nodeUserData, nodeId, &GraphCanvas::NodeRequests::GetUserData);
  2074. if (auto* scNodeId = AZStd::any_cast<AZ::EntityId>(nodeUserData))
  2075. {
  2076. AZ::Entity* graphNodeEntity = nullptr;
  2077. AZ::ComponentApplicationBus::BroadcastResult(graphNodeEntity, &AZ::ComponentApplicationRequests::FindEntity, *scNodeId);
  2078. ScriptCanvas::Node* canvasNode = AZ::EntityUtils::FindFirstDerivedComponent<ScriptCanvas::Node>(graphNodeEntity);
  2079. if (canvasNode)
  2080. {
  2081. auto functionDefintionNode = azrtti_cast<ScriptCanvas::Nodes::Core::FunctionDefinitionNode*>(canvasNode);
  2082. if (functionDefintionNode && reason == GraphModelRequests::ExtensionRequestReason::ConnectionProposal)
  2083. {
  2084. }
  2085. else
  2086. {
  2087. ScriptCanvas::SlotId slotId = canvasNode->HandleExtension(extenderId);
  2088. if (slotId.IsValid())
  2089. {
  2090. SlotMappingRequestBus::EventResult(graphCanvasSlotId, nodeId, &SlotMappingRequests::MapToGraphCanvasId, slotId);
  2091. HandleFunctionDefinitionExtension(canvasNode, graphCanvasSlotId, nodeId);
  2092. }
  2093. }
  2094. }
  2095. }
  2096. return graphCanvasSlotId;
  2097. }
  2098. void EditorGraph::ExtensionCancelled(const GraphCanvas::NodeId& nodeId, const GraphCanvas::ExtenderId& extenderId)
  2099. {
  2100. AZ::EntityId scNodeId = ConvertToScriptCanvasNodeId(nodeId);
  2101. if (scNodeId.IsValid())
  2102. {
  2103. ScriptCanvas::Node* canvasNode = FindNode(scNodeId);
  2104. if (canvasNode)
  2105. {
  2106. canvasNode->ExtensionCancelled(extenderId);
  2107. }
  2108. }
  2109. }
  2110. void EditorGraph::FinalizeExtension(const GraphCanvas::NodeId& nodeId, const GraphCanvas::ExtenderId& extenderId)
  2111. {
  2112. AZ::EntityId scNodeId = ConvertToScriptCanvasNodeId(nodeId);
  2113. if (scNodeId.IsValid())
  2114. {
  2115. ScriptCanvas::Node* canvasNode = FindNode(scNodeId);
  2116. if (canvasNode)
  2117. {
  2118. canvasNode->FinalizeExtension(extenderId);
  2119. }
  2120. }
  2121. }
  2122. bool EditorGraph::ShouldWrapperAcceptDrop(const AZ::EntityId& wrapperNode, const QMimeData* mimeData) const
  2123. {
  2124. if (!mimeData->hasFormat(Widget::NodePaletteDockWidget::GetMimeType()))
  2125. {
  2126. return false;
  2127. }
  2128. // Deep mime inspection
  2129. QByteArray arrayData = mimeData->data(Widget::NodePaletteDockWidget::GetMimeType());
  2130. GraphCanvas::GraphCanvasMimeContainer mimeContainer;
  2131. if (!mimeContainer.FromBuffer(arrayData.constData(), arrayData.size()) || mimeContainer.m_mimeEvents.empty())
  2132. {
  2133. return false;
  2134. }
  2135. AZStd::string busName;
  2136. EBusHandlerNodeDescriptorRequestBus::EventResult(busName, wrapperNode, &EBusHandlerNodeDescriptorRequests::GetBusName);
  2137. for (GraphCanvas::GraphCanvasMimeEvent* mimeEvent : mimeContainer.m_mimeEvents)
  2138. {
  2139. CreateEBusHandlerEventMimeEvent* createEbusMethodEvent = azrtti_cast<CreateEBusHandlerEventMimeEvent*>(mimeEvent);
  2140. if (createEbusMethodEvent)
  2141. {
  2142. if (createEbusMethodEvent->GetBusName().compare(busName) != 0)
  2143. {
  2144. return false;
  2145. }
  2146. bool containsEvent = false;
  2147. EBusHandlerNodeDescriptorRequestBus::EventResult(containsEvent, wrapperNode, &EBusHandlerNodeDescriptorRequests::ContainsEvent, createEbusMethodEvent->GetEventId());
  2148. if (containsEvent)
  2149. {
  2150. return false;
  2151. }
  2152. }
  2153. else
  2154. {
  2155. return false;
  2156. }
  2157. }
  2158. return true;
  2159. }
  2160. void EditorGraph::AddWrapperDropTarget(const AZ::EntityId& wrapperNode)
  2161. {
  2162. if (!m_wrapperNodeDropTarget.IsValid())
  2163. {
  2164. m_wrapperNodeDropTarget = wrapperNode;
  2165. }
  2166. }
  2167. void EditorGraph::RemoveWrapperDropTarget(const AZ::EntityId& wrapperNode)
  2168. {
  2169. if (m_wrapperNodeDropTarget == wrapperNode)
  2170. {
  2171. m_wrapperNodeDropTarget.SetInvalid();
  2172. }
  2173. }
  2174. GraphCanvas::GraphId EditorGraph::GetGraphCanvasGraphId() const
  2175. {
  2176. if (m_saveFormatConverted)
  2177. {
  2178. if (m_graphCanvasSceneEntity)
  2179. {
  2180. return m_graphCanvasSceneEntity->GetId();
  2181. }
  2182. return AZ::EntityId();
  2183. }
  2184. else
  2185. {
  2186. return GetEntityId();
  2187. }
  2188. }
  2189. NodeIdPair EditorGraph::CreateCustomNode(const AZ::Uuid& typeId, const AZ::Vector2& position)
  2190. {
  2191. CreateCustomNodeMimeEvent mimeEvent(typeId);
  2192. AZ::Vector2 dropPosition = position;
  2193. if (mimeEvent.ExecuteEvent(position, dropPosition, GetGraphCanvasGraphId()))
  2194. {
  2195. return mimeEvent.GetCreatedPair();
  2196. }
  2197. return NodeIdPair();
  2198. }
  2199. void EditorGraph::AddCrcCache(const AZ::Crc32& crcValue, const AZStd::string& cacheString)
  2200. {
  2201. auto mapIter = m_crcCacheMap.find(crcValue);
  2202. if (mapIter == m_crcCacheMap.end())
  2203. {
  2204. m_crcCacheMap.emplace(crcValue, CRCCache(cacheString));
  2205. }
  2206. else
  2207. {
  2208. mapIter->second.m_cacheCount++;
  2209. }
  2210. }
  2211. void EditorGraph::RemoveCrcCache(const AZ::Crc32& crcValue)
  2212. {
  2213. auto mapIter = m_crcCacheMap.find(crcValue);
  2214. if (mapIter != m_crcCacheMap.end())
  2215. {
  2216. mapIter->second.m_cacheCount--;
  2217. if (mapIter->second.m_cacheCount <= 0)
  2218. {
  2219. m_crcCacheMap.erase(mapIter);
  2220. }
  2221. }
  2222. }
  2223. AZStd::string EditorGraph::DecodeCrc(const AZ::Crc32& crcValue)
  2224. {
  2225. auto mapIter = m_crcCacheMap.find(crcValue);
  2226. if (mapIter != m_crcCacheMap.end())
  2227. {
  2228. return mapIter->second.m_cacheValue;
  2229. }
  2230. return "";
  2231. }
  2232. void EditorGraph::ClearHighlights()
  2233. {
  2234. for (const GraphCanvas::GraphicsEffectId& effectId : m_highlights)
  2235. {
  2236. GraphCanvas::SceneRequestBus::Event(GetGraphCanvasGraphId(), &GraphCanvas::SceneRequests::CancelGraphicsEffect, effectId);
  2237. }
  2238. m_highlights.clear();
  2239. }
  2240. void EditorGraph::HighlightMembersFromTreeItem(const GraphCanvas::GraphCanvasTreeItem* treeItem)
  2241. {
  2242. ClearHighlights();
  2243. if (auto handleEbusEventTreeItem = azrtti_cast<const EBusHandleEventPaletteTreeItem*>(treeItem))
  2244. {
  2245. HighlightEBusNodes(handleEbusEventTreeItem->GetBusId(), handleEbusEventTreeItem->GetEventId());
  2246. }
  2247. else if (auto sendScriptEventTreeItem = azrtti_cast<const ScriptEventsEventNodePaletteTreeItem*>(treeItem))
  2248. {
  2249. HighlightScriptEventNodes(sendScriptEventTreeItem->GetBusIdentifier(), sendScriptEventTreeItem->GetEventIdentifier());
  2250. }
  2251. else
  2252. {
  2253. HighlightNodesByType(NodeIdentifierFactory::ConstructNodeIdentifier(treeItem));
  2254. }
  2255. }
  2256. void EditorGraph::HighlightVariables(const AZStd::unordered_set< ScriptCanvas::VariableId >& variableIds)
  2257. {
  2258. ClearHighlights();
  2259. for (auto nodeComponentPair : GetNodeMapping())
  2260. {
  2261. ScriptCanvas::Node* node = nodeComponentPair.second;
  2262. if (node->ContainsReferencesToVariables(variableIds))
  2263. {
  2264. HighlightScriptCanvasEntity(nodeComponentPair.first);
  2265. }
  2266. }
  2267. }
  2268. void EditorGraph::HighlightNodes(const AZStd::vector<NodeIdPair>& nodes)
  2269. {
  2270. ClearHighlights();
  2271. for (const NodeIdPair& nodeIdPair : nodes)
  2272. {
  2273. HighlightScriptCanvasEntity(nodeIdPair.m_scriptCanvasId);
  2274. }
  2275. }
  2276. void EditorGraph::RemoveUnusedVariables()
  2277. {
  2278. RequestPushPreventUndoStateUpdate();
  2279. auto variableData = GetVariableData();
  2280. auto variables = variableData->GetVariables();
  2281. AZStd::unordered_set<ScriptCanvas::VariableId> usedVariableIds;
  2282. for (auto nodePair : GetNodeMapping())
  2283. {
  2284. ScriptCanvas::Node* node = nodePair.second;
  2285. node->CollectVariableReferences(usedVariableIds);
  2286. }
  2287. AZStd::unordered_set<ScriptCanvas::VariableId> unusedVariables;
  2288. for (auto variableData2 : variables)
  2289. {
  2290. if (usedVariableIds.count(variableData2.first) == 0)
  2291. {
  2292. unusedVariables.insert(variableData2.first);
  2293. }
  2294. }
  2295. bool removedVariable = false;
  2296. for (ScriptCanvas::VariableId variableId : unusedVariables)
  2297. {
  2298. bool success = false;
  2299. ScriptCanvas::GraphVariableManagerRequestBus::EventResult(success, GetScriptCanvasId(), &ScriptCanvas::GraphVariableManagerRequests::RemoveVariable, variableId);
  2300. if (success)
  2301. {
  2302. removedVariable = true;
  2303. }
  2304. }
  2305. RequestPopPreventUndoStateUpdate();
  2306. if (removedVariable)
  2307. {
  2308. RequestUndoPoint();
  2309. }
  2310. }
  2311. bool EditorGraph::CanConvertVariableNodeToReference(const GraphCanvas::NodeId& nodeId)
  2312. {
  2313. AZ::EntityId scriptCanvasNodeId = ConvertToScriptCanvasNodeId(nodeId);
  2314. ScriptCanvas::VariableId variableId;
  2315. ScriptCanvas::VariableNodeRequestBus::EventResult(variableId, scriptCanvasNodeId, &ScriptCanvas::VariableNodeRequests::GetId);
  2316. ScriptCanvas::GraphVariable* variable = FindVariableById(variableId);
  2317. if (variable == nullptr)
  2318. {
  2319. return false;
  2320. }
  2321. AZStd::vector< GraphCanvas::SlotId > slotIds;
  2322. GraphCanvas::NodeRequestBus::EventResult(slotIds, nodeId, &GraphCanvas::NodeRequests::GetSlotIds);
  2323. for (const GraphCanvas::SlotId& slotId : slotIds)
  2324. {
  2325. GraphCanvas::SlotType slotType = GraphCanvas::SlotTypes::Invalid;
  2326. GraphCanvas::SlotRequestBus::EventResult(slotType, slotId, &GraphCanvas::SlotRequests::GetSlotType);
  2327. if (slotType == GraphCanvas::SlotTypes::ExecutionSlot)
  2328. {
  2329. continue;
  2330. }
  2331. GraphCanvas::Endpoint gcEndpoint(nodeId, slotId);
  2332. ScriptCanvas::Endpoint scEndpoint = ConvertToScriptCanvasEndpoint(gcEndpoint);
  2333. ScriptCanvas::Node* node = FindNode(scEndpoint.GetNodeId());
  2334. // We only care about the actual variable type for enabling/disabling the button.
  2335. // All other conditions will be handled in the conversion with user prompts.
  2336. if (node->SlotAcceptsType(scEndpoint.GetSlotId(), variable->GetDataType()))
  2337. {
  2338. AZStd::vector< GraphCanvas::ConnectionId > connectionIds;
  2339. GraphCanvas::SlotRequestBus::EventResult(connectionIds, slotId, &GraphCanvas::SlotRequests::GetConnections);
  2340. return !connectionIds.empty();
  2341. }
  2342. }
  2343. return false;
  2344. }
  2345. bool EditorGraph::ConvertVariableNodeToReference(const GraphCanvas::NodeId& nodeId)
  2346. {
  2347. AZ::EntityId scriptCanvasNodeId = ConvertToScriptCanvasNodeId(nodeId);
  2348. ScriptCanvas::VariableId variableId;
  2349. ScriptCanvas::VariableNodeRequestBus::EventResult(variableId, scriptCanvasNodeId, &ScriptCanvas::VariableNodeRequests::GetId);
  2350. ScriptCanvas::GraphVariable* variable = FindVariableById(variableId);
  2351. if (variable == nullptr)
  2352. {
  2353. return false;
  2354. }
  2355. AZStd::vector< GraphCanvas::SlotId > slotIds;
  2356. GraphCanvas::NodeRequestBus::EventResult(slotIds, nodeId, &GraphCanvas::NodeRequests::GetSlotIds);
  2357. AZStd::unordered_set< GraphCanvas::Endpoint > referencableEndpoints;
  2358. bool canDetachNode = true;
  2359. GraphCanvas::GraphId graphCanvasGraphId = GetGraphCanvasGraphId();
  2360. QMainWindow* mainWindow = nullptr;
  2361. UIRequestBus::BroadcastResult(mainWindow, &UIRequests::GetMainWindow);
  2362. for (const GraphCanvas::SlotId& slotId : slotIds)
  2363. {
  2364. GraphCanvas::SlotRequests* slotRequests = GraphCanvas::SlotRequestBus::FindFirstHandler(slotId);
  2365. if (slotRequests)
  2366. {
  2367. GraphCanvas::SlotType slotType = slotRequests->GetSlotType();
  2368. if (slotType == GraphCanvas::SlotTypes::DataSlot)
  2369. {
  2370. GraphCanvas::Endpoint currentEndpoint(nodeId, slotId);
  2371. // If we have a reference anywhere on us. We need to maintain this node, since it's not doing something
  2372. // we can merge out cleanly
  2373. ScriptCanvas::Endpoint scriptCanvasCurrentEndpoint = ConvertToScriptCanvasEndpoint(currentEndpoint);
  2374. ScriptCanvas::Slot* sourceSlot = FindSlot(scriptCanvasCurrentEndpoint);
  2375. if (sourceSlot->IsVariableReference())
  2376. {
  2377. canDetachNode = false;
  2378. }
  2379. auto connectionIds = slotRequests->GetConnections();
  2380. for (auto connectionId : connectionIds)
  2381. {
  2382. GraphCanvas::Endpoint otherEndpoint;
  2383. GraphCanvas::ConnectionRequestBus::EventResult(otherEndpoint, connectionId, &GraphCanvas::ConnectionRequests::FindOtherEndpoint, currentEndpoint);
  2384. ScriptCanvas::Endpoint scriptCanvasOtherEndpoint = ConvertToScriptCanvasEndpoint(otherEndpoint);
  2385. ScriptCanvas::Node* otherNode = FindNode(scriptCanvasOtherEndpoint.GetNodeId());
  2386. ScriptCanvas::Slot* otherSlot = FindSlot(scriptCanvasOtherEndpoint);
  2387. if (otherNode && otherSlot && otherNode->SlotAcceptsType(scriptCanvasOtherEndpoint.GetSlotId(), variable->GetDataType()))
  2388. {
  2389. AZStd::unordered_set< AZ::EntityId > deletedConnection = { connectionId };
  2390. GraphCanvas::SceneRequestBus::Event(graphCanvasGraphId, &GraphCanvas::SceneRequests::Delete, deletedConnection);
  2391. if (otherSlot->CanConvertToReference())
  2392. {
  2393. referencableEndpoints.insert(otherEndpoint);
  2394. }
  2395. else
  2396. {
  2397. // Try to resolve chained steps when we are going to end up being a 'Get' reference which is fine to convert to.
  2398. // Otherwise, if we chain to a 'set' reference, that might have unintended consequences, so we need to ignore that.
  2399. // 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.
  2400. if (sourceSlot->IsInput())
  2401. {
  2402. AZStd::vector< GraphCanvas::ConnectionId > chainedConnectionIds;
  2403. GraphCanvas::SlotRequestBus::EventResult(chainedConnectionIds, otherEndpoint.GetSlotId(), &GraphCanvas::SlotRequests::GetConnections);
  2404. for (auto chainedConnectionId : chainedConnectionIds)
  2405. {
  2406. GraphCanvas::Endpoint chainedEndpoint;
  2407. GraphCanvas::ConnectionRequestBus::EventResult(chainedEndpoint, chainedConnectionId, &GraphCanvas::ConnectionRequests::FindOtherEndpoint, otherEndpoint);
  2408. ScriptCanvas::Endpoint scriptCanvasChainedEndpoint = ConvertToScriptCanvasEndpoint(chainedEndpoint);
  2409. ScriptCanvas::Node* chainedNode = FindNode(scriptCanvasChainedEndpoint.GetNodeId());
  2410. ScriptCanvas::Slot* chainedSlot = FindSlot(scriptCanvasChainedEndpoint);
  2411. if (chainedNode && chainedSlot && chainedNode->SlotAcceptsType(scriptCanvasChainedEndpoint.GetSlotId(), variable->GetDataType()))
  2412. {
  2413. AZStd::unordered_set< AZ::EntityId > chainedDeletedConnection = { chainedConnectionId };
  2414. GraphCanvas::SceneRequestBus::Event(graphCanvasGraphId, &GraphCanvas::SceneRequests::Delete, chainedDeletedConnection);
  2415. if (chainedSlot->CanConvertToReference())
  2416. {
  2417. referencableEndpoints.insert(chainedEndpoint);
  2418. }
  2419. else
  2420. {
  2421. GraphCanvas::SlotRequestBus::Event(chainedEndpoint.GetSlotId(), &GraphCanvas::SlotRequests::CreateConnectionWithEndpoint, otherEndpoint);
  2422. }
  2423. }
  2424. }
  2425. }
  2426. if (otherSlot->CanConvertToReference())
  2427. {
  2428. referencableEndpoints.insert(otherEndpoint);
  2429. }
  2430. else
  2431. {
  2432. GraphCanvas::SlotRequestBus::Event(currentEndpoint.GetSlotId(), &GraphCanvas::SlotRequests::CreateConnectionWithEndpoint, otherEndpoint);
  2433. canDetachNode = false;
  2434. }
  2435. }
  2436. }
  2437. else
  2438. {
  2439. canDetachNode = false;
  2440. }
  2441. }
  2442. }
  2443. }
  2444. }
  2445. // Signal out on the graph that we did something to the node.
  2446. GraphCanvas::AnimatedPulseConfiguration animatedPulseConfig;
  2447. animatedPulseConfig.m_enableGradient = true;
  2448. if (canDetachNode)
  2449. {
  2450. animatedPulseConfig.m_drawColor = QColor(255, 0, 0);
  2451. }
  2452. else
  2453. {
  2454. animatedPulseConfig.m_drawColor = QColor(255, 255, 255);
  2455. }
  2456. animatedPulseConfig.m_durationSec = 0.25f;
  2457. GraphCanvas::SceneRequestBus::Event(GetGraphCanvasGraphId(), &GraphCanvas::SceneRequests::CreatePulseAroundSceneMember, nodeId, 4, animatedPulseConfig);
  2458. // If we can detach the node. All connections will be deleted, except for the ones we want to save.
  2459. if (canDetachNode)
  2460. {
  2461. GraphCanvas::NodeDetachConfig detachConfig(nodeId);
  2462. detachConfig.m_listingType = GraphCanvas::ListingType::InclusiveList;
  2463. detachConfig.m_typeListing.insert(GraphCanvas::SlotTypes::ExecutionSlot);
  2464. GraphCanvas::GraphUtils::DetachNodeAndStitchConnections(detachConfig);
  2465. AZStd::unordered_set<GraphCanvas::NodeId > nodeIds = { nodeId };
  2466. GraphCanvas::SceneRequestBus::Event(GetGraphCanvasGraphId(), &GraphCanvas::SceneRequests::Delete, nodeIds);
  2467. }
  2468. for (auto graphCanvasEndpoint : referencableEndpoints)
  2469. {
  2470. GraphCanvas::DataSlotRequestBus::Event(graphCanvasEndpoint.GetSlotId(), &GraphCanvas::DataSlotRequests::ConvertToReference, false);
  2471. ScriptCanvas::Endpoint scriptCanvasEndpoint = ConvertToScriptCanvasEndpoint(graphCanvasEndpoint);
  2472. ScriptCanvas::Slot* slot = FindSlot(scriptCanvasEndpoint);
  2473. if (slot && slot->IsVariableReference())
  2474. {
  2475. slot->SetVariableReference(variable->GetVariableId());
  2476. }
  2477. }
  2478. return true;
  2479. }
  2480. bool EditorGraph::ConvertReferenceToVariableNode([[maybe_unused]] const GraphCanvas::Endpoint& endpoint)
  2481. {
  2482. return false;
  2483. }
  2484. bool EditorGraph::OnVersionConversionBegin(ScriptCanvas::Node& scriptCanvasNode)
  2485. {
  2486. auto insertResult = m_convertingNodes.insert(scriptCanvasNode.GetEntityId());
  2487. if (!insertResult.second)
  2488. {
  2489. return false;
  2490. }
  2491. for (const ScriptCanvas::Slot& currentSlot : scriptCanvasNode.GetSlots())
  2492. {
  2493. m_versionedSlots.insert(AZStd::make_pair(scriptCanvasNode.GetEntityId(), currentSlot.GetId()));
  2494. }
  2495. EditorNodeNotificationBus::Event(scriptCanvasNode.GetEntityId(), &EditorNodeNotifications::OnVersionConversionBegin);
  2496. return true;
  2497. }
  2498. void EditorGraph::OnVersionConversionEnd(ScriptCanvas::Node& scriptCanvasNode)
  2499. {
  2500. EditorNodeNotificationBus::Event(scriptCanvasNode.GetEntityId(), &EditorNodeNotifications::OnVersionConversionEnd);
  2501. size_t removeCount = m_convertingNodes.count(scriptCanvasNode.GetEntityId());
  2502. if (removeCount > 0)
  2503. {
  2504. auto findResult = m_versionedSlots.equal_range(scriptCanvasNode.GetEntityId());
  2505. AZStd::unordered_set<ScriptCanvas::SlotId> previousSlots;
  2506. for (auto cacheIter = findResult.first; cacheIter != findResult.second; ++cacheIter)
  2507. {
  2508. previousSlots.insert(cacheIter->second);
  2509. }
  2510. AZStd::unordered_set< GraphCanvas::ConnectionId > deletedGraphCanvasConnections;
  2511. for (const ScriptCanvas::Slot& constantSlot : scriptCanvasNode.GetSlots())
  2512. {
  2513. ScriptCanvas::Slot* currentSlot = scriptCanvasNode.GetSlot(constantSlot.GetId());
  2514. if (!currentSlot)
  2515. {
  2516. AZ_Error("ScriptCanvas", false, "Missing slot from node %s after conversion ", scriptCanvasNode.GetDebugName().data());
  2517. continue;
  2518. }
  2519. ScriptCanvas::SlotId slotId = currentSlot->GetId();
  2520. size_t eraseCount = previousSlots.erase(slotId);
  2521. if (eraseCount == 0)
  2522. {
  2523. continue;
  2524. }
  2525. // Manage updating connections and remove invalid ones
  2526. ScriptCanvas::Endpoint endpoint = currentSlot->GetEndpoint();
  2527. GraphCanvas::Endpoint graphCanvasEndpoint = ConvertToGraphCanvasEndpoint(endpoint);
  2528. AZStd::vector< ScriptCanvas::Endpoint > connectedEndpoints = GetConnectedEndpoints(endpoint);
  2529. for (const ScriptCanvas::Endpoint& connectedEndpoint : connectedEndpoints)
  2530. {
  2531. if (IsNodeVersionConverting(connectedEndpoint.GetNodeId()))
  2532. {
  2533. continue;
  2534. }
  2535. bool allowConnection = CanConnectionExistBetween(connectedEndpoint, endpoint).IsSuccess();
  2536. bool deleteConnection = true;
  2537. if (graphCanvasEndpoint.IsValid())
  2538. {
  2539. GraphCanvas::Endpoint otherEndpoint = ConvertToGraphCanvasEndpoint(connectedEndpoint);
  2540. if (otherEndpoint.IsValid())
  2541. {
  2542. bool isConnected = false;
  2543. GraphCanvas::SlotRequestBus::EventResult(isConnected, graphCanvasEndpoint.GetSlotId(), &GraphCanvas::SlotRequests::IsConnectedTo, otherEndpoint);
  2544. if (isConnected)
  2545. {
  2546. deleteConnection = false;
  2547. // If Graph canvas is connected, but we need to kill the connection
  2548. // we'll let the Graph Canvas Deletion Update our internal state.
  2549. if (!allowConnection)
  2550. {
  2551. AZStd::unordered_set< GraphCanvas::Endpoint > searchEndpoints = { otherEndpoint };
  2552. GraphCanvas::SlotRequestBus::Event(graphCanvasEndpoint.GetSlotId(), &GraphCanvas::SlotRequests::FindConnectionsForEndpoints, searchEndpoints, deletedGraphCanvasConnections);
  2553. }
  2554. }
  2555. else if (allowConnection)
  2556. {
  2557. deleteConnection = false;
  2558. GraphCanvas::SlotRequestBus::Event(graphCanvasEndpoint.GetSlotId(), &GraphCanvas::SlotRequests::DisplayConnectionWithEndpoint, otherEndpoint);
  2559. }
  2560. }
  2561. }
  2562. if (deleteConnection)
  2563. {
  2564. AZ::Entity* connectionEntity = nullptr;
  2565. if (FindConnection(connectionEntity, endpoint, connectedEndpoint))
  2566. {
  2567. RemoveConnection(connectionEntity->GetId());
  2568. }
  2569. }
  2570. }
  2571. }
  2572. for (auto erasedSlot : previousSlots)
  2573. {
  2574. VersioningRemoveSlot(scriptCanvasNode, erasedSlot);
  2575. }
  2576. m_versionedSlots.erase(scriptCanvasNode.GetEntityId());
  2577. m_convertingNodes.erase(scriptCanvasNode.GetEntityId());
  2578. if (!deletedGraphCanvasConnections.empty())
  2579. {
  2580. GraphCanvas::SceneRequestBus::Event(GetGraphCanvasGraphId(), &GraphCanvas::SceneRequests::Delete, deletedGraphCanvasConnections);
  2581. }
  2582. AZStd::string updateString = scriptCanvasNode.GetUpdateString();
  2583. m_updateStrings.insert(updateString);
  2584. if (m_convertingNodes.empty())
  2585. {
  2586. DisplayUpdateToast();
  2587. }
  2588. }
  2589. }
  2590. AZStd::vector<NodeIdPair> EditorGraph::GetNodesOfType(const ScriptCanvas::NodeTypeIdentifier& nodeTypeIdentifier)
  2591. {
  2592. AZStd::vector<NodeIdPair> nodeIdPairs;
  2593. for (auto nodeMappingPair : GetNodeMapping())
  2594. {
  2595. ScriptCanvas::Node* canvasNode = nodeMappingPair.second;
  2596. AZ::EntityId nodeEntityId = canvasNode->GetEntityId();
  2597. if (canvasNode->GetNodeType() == nodeTypeIdentifier)
  2598. {
  2599. NodeIdPair nodeIdPair;
  2600. nodeIdPair.m_scriptCanvasId = nodeEntityId;
  2601. SceneMemberMappingRequestBus::EventResult(nodeIdPair.m_graphCanvasId, nodeEntityId, &SceneMemberMappingRequests::GetGraphCanvasEntityId);
  2602. nodeIdPairs.emplace_back(nodeIdPair);
  2603. }
  2604. else if (ScriptCanvas::Nodes::Core::EBusEventHandler* handlerNode = azrtti_cast<ScriptCanvas::Nodes::Core::EBusEventHandler*>(canvasNode))
  2605. {
  2606. ScriptCanvas::EBusBusId busId = handlerNode->GetEBusId();
  2607. for (auto eventPair : handlerNode->GetEvents())
  2608. {
  2609. ScriptCanvas::EBusEventId eventId = eventPair.second.m_eventId;
  2610. if (ScriptCanvas::NodeUtils::ConstructEBusEventReceiverIdentifier(busId, eventId) == nodeTypeIdentifier)
  2611. {
  2612. AZ::EntityId graphCanvasNodeId;
  2613. SceneMemberMappingRequestBus::EventResult(graphCanvasNodeId, canvasNode->GetEntityId(), &SceneMemberMappingRequests::GetGraphCanvasEntityId);
  2614. bool hasEvent = false;
  2615. EBusHandlerNodeDescriptorRequestBus::EventResult(hasEvent, graphCanvasNodeId, &EBusHandlerNodeDescriptorRequests::ContainsEvent, eventId);
  2616. if (hasEvent)
  2617. {
  2618. NodeIdPair nodeIdPair;
  2619. nodeIdPair.m_scriptCanvasId = nodeEntityId;
  2620. nodeIdPair.m_graphCanvasId = graphCanvasNodeId;
  2621. nodeIdPairs.emplace_back(nodeIdPair);
  2622. }
  2623. }
  2624. }
  2625. }
  2626. else if (ScriptCanvas::Nodes::Core::ReceiveScriptEvent* receiveScriptEvent = azrtti_cast<ScriptCanvas::Nodes::Core::ReceiveScriptEvent*>(canvasNode))
  2627. {
  2628. AZ::EntityId graphCanvasNodeId;
  2629. SceneMemberMappingRequestBus::EventResult(graphCanvasNodeId, canvasNode->GetEntityId(), &SceneMemberMappingRequests::GetGraphCanvasEntityId);
  2630. EBusHandlerNodeDescriptorRequests* ebusHandlerDescriptor = EBusHandlerNodeDescriptorRequestBus::FindFirstHandler(graphCanvasNodeId);
  2631. if (ebusHandlerDescriptor)
  2632. {
  2633. auto eventConfigurations = ebusHandlerDescriptor->GetEventConfigurations();
  2634. ScriptCanvas::EBusBusId busId = receiveScriptEvent->GetBusId();
  2635. for (auto eventConfiguration : eventConfigurations)
  2636. {
  2637. if (ScriptCanvas::NodeUtils::ConstructScriptEventReceiverIdentifier(busId, eventConfiguration.m_eventId) == nodeTypeIdentifier)
  2638. {
  2639. if (ebusHandlerDescriptor->ContainsEvent(eventConfiguration.m_eventId))
  2640. {
  2641. NodeIdPair nodeIdPair;
  2642. nodeIdPair.m_scriptCanvasId = nodeEntityId;
  2643. nodeIdPair.m_graphCanvasId = graphCanvasNodeId;
  2644. nodeIdPairs.emplace_back(nodeIdPair);
  2645. }
  2646. }
  2647. }
  2648. }
  2649. }
  2650. }
  2651. return nodeIdPairs;
  2652. }
  2653. AZStd::vector<NodeIdPair> EditorGraph::GetVariableNodes(const ScriptCanvas::VariableId& variableId)
  2654. {
  2655. AZStd::vector<NodeIdPair> variableNodes;
  2656. if (variableId.IsValid())
  2657. {
  2658. AZStd::unordered_set< ScriptCanvas::VariableId > variableIds = { variableId };
  2659. for (auto nodePairMapping : GetNodeMapping())
  2660. {
  2661. if (nodePairMapping.second->ContainsReferencesToVariables(variableIds))
  2662. {
  2663. NodeIdPair nodeIdPair;
  2664. nodeIdPair.m_scriptCanvasId = nodePairMapping.first;
  2665. SceneMemberMappingRequestBus::EventResult(nodeIdPair.m_graphCanvasId, nodePairMapping.first, &SceneMemberMappingRequests::GetGraphCanvasEntityId);
  2666. variableNodes.push_back(nodeIdPair);
  2667. }
  2668. }
  2669. }
  2670. return variableNodes;
  2671. }
  2672. void EditorGraph::QueueVersionUpdate(const AZ::EntityId& graphCanvasNodeId)
  2673. {
  2674. bool queueUpdate = m_queuedConvertingNodes.empty();
  2675. auto insertResult = m_queuedConvertingNodes.insert(graphCanvasNodeId);
  2676. if (insertResult.second && queueUpdate)
  2677. {
  2678. m_allowVersionUpdate = false;
  2679. AZ::SystemTickBus::Handler::BusConnect();
  2680. }
  2681. }
  2682. bool EditorGraph::CanExposeEndpoint(const GraphCanvas::Endpoint& endpoint)
  2683. {
  2684. bool isEnabled = false;
  2685. GraphCanvas::SlotType slotType;
  2686. GraphCanvas::SlotRequestBus::EventResult(slotType, endpoint.GetSlotId(), &GraphCanvas::SlotRequests::GetSlotType);
  2687. if (slotType == GraphCanvas::SlotTypes::DataSlot)
  2688. {
  2689. GraphCanvas::DataSlotType dataSlotType = GraphCanvas::DataSlotType::Unknown;
  2690. GraphCanvas::DataSlotRequestBus::EventResult(dataSlotType, endpoint.GetSlotId(), &GraphCanvas::DataSlotRequests::GetDataSlotType);
  2691. if (dataSlotType != GraphCanvas::DataSlotType::Value)
  2692. {
  2693. isEnabled = false;
  2694. }
  2695. bool hasConnections = false;
  2696. GraphCanvas::SlotRequestBus::EventResult(hasConnections, endpoint.GetSlotId(), &GraphCanvas::SlotRequests::HasConnections);
  2697. if (hasConnections)
  2698. {
  2699. isEnabled = false;
  2700. }
  2701. ScriptCanvas::Endpoint scEndpoint = ConvertToScriptCanvasEndpoint(endpoint);
  2702. ScriptCanvas::Slot* slot = FindSlot(scEndpoint);
  2703. // If we don't have a slot it likely means this is a remapped visual slot.
  2704. // So we don't want to perform many operations on it.
  2705. if (slot)
  2706. {
  2707. ScriptCanvas::Data::Type dataType = slot->GetDataType();
  2708. bool isValidVariableType = false;
  2709. VariablePaletteRequestBus::BroadcastResult(isValidVariableType, &VariablePaletteRequests::IsValidVariableType, dataType);
  2710. if (!isValidVariableType)
  2711. {
  2712. isEnabled = false;
  2713. }
  2714. }
  2715. else
  2716. {
  2717. isEnabled = false;
  2718. }
  2719. }
  2720. else
  2721. {
  2722. isEnabled = true;
  2723. }
  2724. bool isNodeling = false;
  2725. NodeDescriptorRequestBus::EventResult(isNodeling, endpoint.GetNodeId(), &NodeDescriptorRequests::IsType, NodeDescriptorType::FunctionDefinitionNode);
  2726. return isEnabled && !isNodeling;
  2727. }
  2728. ScriptCanvas::Endpoint EditorGraph::ConvertToScriptCanvasEndpoint(const GraphCanvas::Endpoint& endpoint) const
  2729. {
  2730. AZStd::any* userData = nullptr;
  2731. ScriptCanvas::Endpoint scriptCanvasEndpoint;
  2732. GraphCanvas::SlotRequestBus::EventResult(userData, endpoint.GetSlotId(), &GraphCanvas::SlotRequests::GetUserData);
  2733. ScriptCanvas::SlotId scSourceSlotId = (userData && userData->is<ScriptCanvas::SlotId>()) ? *AZStd::any_cast<ScriptCanvas::SlotId>(userData) : ScriptCanvas::SlotId();
  2734. userData = nullptr;
  2735. AZ::EntityId scriptCanvasNodeId = ConvertToScriptCanvasNodeId(endpoint.GetNodeId());
  2736. scriptCanvasEndpoint = ScriptCanvas::Endpoint(scriptCanvasNodeId, scSourceSlotId);
  2737. return scriptCanvasEndpoint;
  2738. }
  2739. GraphCanvas::Endpoint EditorGraph::ConvertToGraphCanvasEndpoint(const ScriptCanvas::Endpoint& endpoint) const
  2740. {
  2741. GraphCanvas::Endpoint graphCanvasEndpoint;
  2742. SlotMappingRequestBus::EventResult(graphCanvasEndpoint.m_slotId, endpoint.GetNodeId(), &SlotMappingRequests::MapToGraphCanvasId, endpoint.GetSlotId());
  2743. GraphCanvas::SlotRequestBus::EventResult(graphCanvasEndpoint.m_nodeId, graphCanvasEndpoint.GetSlotId(), &GraphCanvas::SlotRequests::GetNode);
  2744. return graphCanvasEndpoint;
  2745. }
  2746. void EditorGraph::OnSaveDataDirtied(const AZ::EntityId& savedElement)
  2747. {
  2748. // The EbusHandlerEvent's are a visual only representation of alternative data, and should not be saved.
  2749. if (EBusHandlerEventNodeDescriptorRequestBus::FindFirstHandler(savedElement) != nullptr
  2750. || m_ignoreSaveRequests)
  2751. {
  2752. return;
  2753. }
  2754. AZStd::any* userData = nullptr;
  2755. GraphCanvas::NodeRequestBus::EventResult(userData, savedElement, &GraphCanvas::NodeRequests::GetUserData);
  2756. if (userData && userData->is<AZ::EntityId>())
  2757. {
  2758. const AZ::EntityId* scriptCanvasNodeId = AZStd::any_cast<AZ::EntityId>(userData);
  2759. GraphCanvas::EntitySaveDataContainer* container = nullptr;
  2760. auto mapIter = m_graphCanvasSaveData.find((*scriptCanvasNodeId));
  2761. if (mapIter == m_graphCanvasSaveData.end())
  2762. {
  2763. container = aznew GraphCanvas::EntitySaveDataContainer();
  2764. m_graphCanvasSaveData[(*scriptCanvasNodeId)] = container;
  2765. }
  2766. else
  2767. {
  2768. container = mapIter->second;
  2769. }
  2770. GraphCanvas::EntitySaveDataRequestBus::Event(savedElement, &GraphCanvas::EntitySaveDataRequests::WriteSaveData, (*container));
  2771. }
  2772. else if (savedElement == GetGraphCanvasGraphId())
  2773. {
  2774. GraphCanvas::EntitySaveDataContainer* container = nullptr;
  2775. auto mapIter = m_graphCanvasSaveData.find(GetEntityId());
  2776. if (mapIter == m_graphCanvasSaveData.end())
  2777. {
  2778. container = aznew GraphCanvas::EntitySaveDataContainer();
  2779. m_graphCanvasSaveData[GetEntityId()] = container;
  2780. }
  2781. else
  2782. {
  2783. container = mapIter->second;
  2784. }
  2785. GraphCanvas::EntitySaveDataRequestBus::Event(savedElement, &GraphCanvas::EntitySaveDataRequests::WriteSaveData, (*container));
  2786. m_statisticsHelper.PopulateStatisticData(this);
  2787. }
  2788. }
  2789. bool EditorGraph::NeedsSaveConversion() const
  2790. {
  2791. return !m_saveFormatConverted;
  2792. }
  2793. void EditorGraph::ConvertSaveFormat()
  2794. {
  2795. if (!m_saveFormatConverted)
  2796. {
  2797. // Bit of a work around for not being able to clean this up in the actual save.
  2798. m_saveFormatConverted = true;
  2799. // SceneComponent
  2800. for (const AZ::Uuid& componentType : {
  2801. AZ::Uuid("{3F71486C-3D51-431F-B904-DA070C7A0238}"), // GraphCanvas::SceneComponent
  2802. AZ::Uuid("{486B009F-632B-44F6-81C2-3838746190AE}"), // ColorPaletteManagerComponent
  2803. AZ::Uuid("{A8F08DEA-0F42-4236-9E1E-B93C964B113F}"), // BookmarkManagerComponent
  2804. AZ::Uuid("{34B81206-2C69-4886-945B-4A9ECC0FDAEE}") // StyleSheet
  2805. }
  2806. )
  2807. {
  2808. AZ::Component* component = GetEntity()->FindComponent(componentType);
  2809. if (component)
  2810. {
  2811. if (GetEntity()->RemoveComponent(component))
  2812. {
  2813. delete component;
  2814. }
  2815. }
  2816. }
  2817. }
  2818. }
  2819. void EditorGraph::ConstructSaveData()
  2820. {
  2821. // Save out the SceneData
  2822. //
  2823. // For this one all of the GraphCanvas information lives on the same entity.
  2824. // So we need to use that key to look up everything
  2825. {
  2826. OnSaveDataDirtied(GetGraphCanvasGraphId());
  2827. }
  2828. AZStd::vector< AZ::EntityId > graphCanvasNodes;
  2829. GraphCanvas::SceneRequestBus::EventResult(graphCanvasNodes, GetGraphCanvasGraphId(), &GraphCanvas::SceneRequests::GetNodes);
  2830. for (const AZ::EntityId& graphCanvasNode : graphCanvasNodes)
  2831. {
  2832. OnSaveDataDirtied(graphCanvasNode);
  2833. }
  2834. }
  2835. void EditorGraph::OnToastInteraction()
  2836. {
  2837. const AzToolsFramework::ToastId* toastId = AzToolsFramework::ToastNotificationBus::GetCurrentBusId();
  2838. if (toastId)
  2839. {
  2840. NodeIdPair pair;
  2841. pair.m_scriptCanvasId = m_toastNodeIds[(*toastId)];
  2842. UnregisterToast((*toastId));
  2843. SceneMemberMappingRequestBus::EventResult(pair.m_graphCanvasId, pair.m_scriptCanvasId, &SceneMemberMappingRequests::GetGraphCanvasEntityId);
  2844. AZStd::vector<AZ::EntityId> focusElements = { pair.m_graphCanvasId };
  2845. m_focusHelper.Clear();
  2846. m_focusHelper.SetNodes(focusElements);
  2847. m_focusHelper.CycleToNextNode();
  2848. AZStd::vector< NodeIdPair > highlightPair = { pair };
  2849. HighlightNodes(highlightPair);
  2850. }
  2851. }
  2852. void EditorGraph::OnToastDismissed()
  2853. {
  2854. const AzToolsFramework::ToastId* toastId = AzToolsFramework::ToastNotificationBus::GetCurrentBusId();
  2855. if (toastId)
  2856. {
  2857. UnregisterToast((*toastId));
  2858. }
  2859. }
  2860. void EditorGraph::OnUndoRedoEnd()
  2861. {
  2862. for (const auto& nodePair : GetNodeMapping())
  2863. {
  2864. nodePair.second->SignalDeserialized();
  2865. }
  2866. }
  2867. void EditorGraph::ReportError(const ScriptCanvas::Node& node, const AZStd::string& errorSource, const AZStd::string& errorMessage)
  2868. {
  2869. AzQtComponents::ToastConfiguration toastConfiguration(AzQtComponents::ToastType::Error, errorSource.c_str(), errorMessage.c_str());
  2870. GraphCanvas::ViewId viewId;
  2871. GraphCanvas::SceneRequestBus::EventResult(viewId, GetGraphCanvasGraphId(), &GraphCanvas::SceneRequests::GetViewId);
  2872. AzToolsFramework::ToastId toastId;
  2873. GraphCanvas::ViewRequestBus::EventResult(toastId, viewId, &GraphCanvas::ViewRequests::ShowToastNotification, toastConfiguration);
  2874. AzToolsFramework::ToastNotificationBus::MultiHandler::BusConnect(toastId);
  2875. m_toastNodeIds[toastId] = node.GetEntityId();
  2876. }
  2877. void EditorGraph::UnregisterToast(const AzToolsFramework::ToastId& toastId)
  2878. {
  2879. AzToolsFramework::ToastNotificationBus::MultiHandler::BusDisconnect(toastId);
  2880. m_toastNodeIds.erase(toastId);
  2881. }
  2882. void EditorGraph::DisplayUpdateToast()
  2883. {
  2884. GraphCanvas::ViewId viewId;
  2885. GraphCanvas::SceneRequestBus::EventResult(viewId, GetGraphCanvasGraphId(), &GraphCanvas::SceneRequests::GetViewId);
  2886. if (viewId.IsValid() && !m_updateStrings.empty())
  2887. {
  2888. bool isVisible = false;
  2889. GraphCanvas::ViewRequestBus::EventResult(isVisible, viewId, &GraphCanvas::ViewRequests::IsShowing);
  2890. if (isVisible)
  2891. {
  2892. AZStd::string displayString;
  2893. for (const auto& updateData : m_updateStrings)
  2894. {
  2895. if (!displayString.empty())
  2896. {
  2897. displayString.append("\n");
  2898. }
  2899. displayString.append("- ");
  2900. displayString.append(updateData);
  2901. }
  2902. m_updateStrings.clear();
  2903. AzQtComponents::ToastConfiguration toastConfiguration(AzQtComponents::ToastType::Information, "Nodes Updates", displayString.c_str());
  2904. GraphCanvas::ViewRequestBus::Event(viewId, &GraphCanvas::ViewRequests::ShowToastNotification, toastConfiguration);
  2905. }
  2906. }
  2907. }
  2908. const GraphStatisticsHelper& EditorGraph::GetNodeUsageStatistics() const
  2909. {
  2910. return m_statisticsHelper;
  2911. }
  2912. void EditorGraph::CreateGraphCanvasScene()
  2913. {
  2914. if (!m_saveFormatConverted)
  2915. {
  2916. GraphCanvas::GraphId graphCanvasGraphId = GetGraphCanvasGraphId();
  2917. GraphCanvas::SceneNotificationBus::Handler::BusConnect(graphCanvasGraphId);
  2918. GraphCanvas::GraphModelRequestBus::Handler::BusConnect(graphCanvasGraphId);
  2919. GraphCanvas::SceneRequestBus::Event(graphCanvasGraphId, &GraphCanvas::SceneRequests::SetEditorId, ScriptCanvasEditor::AssetEditorId);
  2920. AZStd::any* userData = nullptr;
  2921. GraphCanvas::SceneRequestBus::EventResult(userData, graphCanvasGraphId, &GraphCanvas::SceneRequests::GetUserData);
  2922. if (userData)
  2923. {
  2924. (*userData) = GetScriptCanvasId();
  2925. }
  2926. }
  2927. else if (m_graphCanvasSceneEntity == nullptr)
  2928. {
  2929. GraphCanvas::GraphCanvasRequestBus::BroadcastResult(m_graphCanvasSceneEntity, &GraphCanvas::GraphCanvasRequests::CreateSceneAndActivate);
  2930. if (m_graphCanvasSceneEntity == nullptr)
  2931. {
  2932. return;
  2933. }
  2934. AZ::EntityId graphCanvasGraphId = GetGraphCanvasGraphId();
  2935. GraphCanvas::SceneRequestBus::Event(graphCanvasGraphId, &GraphCanvas::SceneRequests::SetEditorId, ScriptCanvasEditor::AssetEditorId);
  2936. DisplayGraphCanvasScene();
  2937. AZStd::any* userData = nullptr;
  2938. GraphCanvas::SceneRequestBus::EventResult(userData, graphCanvasGraphId, &GraphCanvas::SceneRequests::GetUserData);
  2939. if (userData)
  2940. {
  2941. (*userData) = GetScriptCanvasId();
  2942. }
  2943. }
  2944. m_focusHelper.SetActiveGraph(GetGraphCanvasGraphId());
  2945. }
  2946. bool EditorGraph::UpgradeGraph(SourceHandle source, UpgradeRequest upgradeRequest, const UpgradeGraphConfig& upgradeConfig)
  2947. {
  2948. m_upgradeSM.SetAsset(source);
  2949. m_upgradeSM.SetConfig(upgradeConfig);
  2950. if (upgradeRequest == UpgradeRequest::Forced || !GetVersion().IsLatest() || HasDeprecatedNode())
  2951. {
  2952. m_upgradeSM.Run(Start::StateID());
  2953. return true;
  2954. }
  2955. else
  2956. {
  2957. m_upgradeSM.Run(Skip::StateID());
  2958. return false;
  2959. }
  2960. }
  2961. void EditorGraph::ConnectGraphCanvasBuses()
  2962. {
  2963. GraphCanvas::GraphId graphCanvasGraphId = GetGraphCanvasGraphId();
  2964. GraphCanvas::GraphModelRequestBus::Handler::BusConnect(graphCanvasGraphId);
  2965. GraphCanvas::SceneNotificationBus::Handler::BusConnect(graphCanvasGraphId);
  2966. }
  2967. void EditorGraph::DisconnectGraphCanvasBuses()
  2968. {
  2969. GraphCanvas::GraphModelRequestBus::Handler::BusDisconnect();
  2970. GraphCanvas::SceneNotificationBus::Handler::BusDisconnect();
  2971. }
  2972. void EditorGraph::OnSystemTick()
  2973. {
  2974. if (!m_allowVersionUpdate)
  2975. {
  2976. m_allowVersionUpdate = true;
  2977. }
  2978. else
  2979. {
  2980. m_allowVersionUpdate = false;
  2981. AZ::SystemTickBus::Handler::BusDisconnect();
  2982. HandleQueuedUpdates();
  2983. }
  2984. }
  2985. void EditorGraph::DisplayGraphCanvasScene()
  2986. {
  2987. m_variableDataModel.Activate(GetScriptCanvasId());
  2988. RequestPushPreventUndoStateUpdate();
  2989. AZStd::unordered_map< AZ::EntityId, AZ::EntityId > scriptCanvasToGraphCanvasMapping;
  2990. bool graphNeedsDirtying = !GetVersion().IsLatest();
  2991. {
  2992. QScopedValueRollback<bool> ignoreRequests(m_ignoreSaveRequests, true);
  2993. GraphCanvas::GraphId graphCanvasGraphId = GetGraphCanvasGraphId();
  2994. GraphCanvas::GraphModelRequestBus::Handler::BusConnect(graphCanvasGraphId);
  2995. GraphCanvas::SceneNotificationBus::Handler::BusConnect(graphCanvasGraphId);
  2996. GraphCanvas::SceneRequestBus::Event(graphCanvasGraphId, &GraphCanvas::SceneRequests::SignalLoadStart);
  2997. for (auto& saveDataIter : m_graphCanvasSaveData)
  2998. {
  2999. GraphCanvas::EntitySaveDataRequestBus::Event(graphCanvasGraphId, &GraphCanvas::EntitySaveDataRequests::ReadSaveData, (*saveDataIter.second));
  3000. }
  3001. ScriptCanvas::NodeIdList nodeList = GetNodes();
  3002. AZStd::unordered_set<ScriptCanvas::Node*> outOfDateNodes;
  3003. AZStd::unordered_set<AZ::EntityId> deletedNodes;
  3004. AZStd::unordered_set<AZ::EntityId> assetSanitizationSet;
  3005. AZStd::unordered_set<ScriptCanvas::Node*> sanityCheckRequiredNodes;
  3006. ScriptCanvas::GraphUpdateReport graphUpdateReport;
  3007. for (const AZ::EntityId& scriptCanvasNodeId : nodeList)
  3008. {
  3009. assetSanitizationSet.insert(scriptCanvasNodeId);
  3010. ScriptCanvas::Node* scriptCanvasNode = FindNode(scriptCanvasNodeId);
  3011. if (scriptCanvasNode)
  3012. {
  3013. ScriptCanvas::NodeReplacementConfiguration nodeConfig;
  3014. auto replacementId = ScriptCanvasEditor::NodeReplacementSystem::GenerateReplacementId(scriptCanvasNode);
  3015. ScriptCanvasEditor::NodeReplacementRequestBus::BroadcastResult(
  3016. nodeConfig, &ScriptCanvasEditor::NodeReplacementRequestBus::Events::GetNodeReplacementConfiguration, replacementId);
  3017. if (!nodeConfig.IsValid())
  3018. {
  3019. nodeConfig = scriptCanvasNode->GetReplacementNodeConfiguration();
  3020. }
  3021. if (nodeConfig.IsValid() && !g_disableDeprecatedNodeUpdates)
  3022. {
  3023. ScriptCanvas::NodeUpdateReport nodeUpdateReport;
  3024. ScriptCanvasEditor::NodeReplacementRequestBus::BroadcastResult(nodeUpdateReport,
  3025. &ScriptCanvasEditor::NodeReplacementRequestBus::Events::ReplaceNodeByReplacementConfiguration,
  3026. GetScriptCanvasId(), scriptCanvasNode, nodeConfig);
  3027. if (!nodeUpdateReport.IsEmpty())
  3028. {
  3029. graphNeedsDirtying = true;
  3030. scriptCanvasNode = nodeUpdateReport.m_newNode;
  3031. m_updateStrings.insert(AZStd::string::format("Replaced node (%s)", scriptCanvasNode->GetNodeName().c_str()));
  3032. ScriptCanvas::MergeUpdateSlotReport(scriptCanvasNodeId, graphUpdateReport, nodeUpdateReport);
  3033. }
  3034. }
  3035. AZ::EntityId graphCanvasNodeId = Nodes::DisplayScriptCanvasNode(graphCanvasGraphId, scriptCanvasNode);
  3036. scriptCanvasToGraphCanvasMapping[scriptCanvasNodeId] = graphCanvasNodeId;
  3037. auto saveDataIter2 = m_graphCanvasSaveData.find(scriptCanvasNodeId);
  3038. if (saveDataIter2 != m_graphCanvasSaveData.end())
  3039. {
  3040. GraphCanvas::EntitySaveDataRequestBus::Event(graphCanvasNodeId, &GraphCanvas::EntitySaveDataRequests::ReadSaveData, (*saveDataIter2->second));
  3041. }
  3042. AZ::Vector2 position;
  3043. GraphCanvas::GeometryRequestBus::EventResult(position, graphCanvasNodeId, &GraphCanvas::GeometryRequests::GetPosition);
  3044. GraphCanvas::SceneRequestBus::Event(graphCanvasGraphId, &GraphCanvas::SceneRequests::AddNode, graphCanvasNodeId, position, false);
  3045. // If the node is deprecated, we want to stomp whatever style it had saved and apply the deprecated style
  3046. if (scriptCanvasNode->IsDeprecated())
  3047. {
  3048. GraphCanvas::NodeTitleRequestBus::Event(graphCanvasNodeId, &GraphCanvas::NodeTitleRequests::SetPaletteOverride, "DeprecatedNodeTitlePalette");
  3049. }
  3050. if (scriptCanvasNode->IsOutOfDate(GetVersion()))
  3051. {
  3052. OnVersionConversionBegin((*scriptCanvasNode));
  3053. outOfDateNodes.emplace(scriptCanvasNode);
  3054. }
  3055. if (scriptCanvasNode->IsSanityCheckRequired())
  3056. {
  3057. graphNeedsDirtying = true;
  3058. sanityCheckRequiredNodes.insert(scriptCanvasNode);
  3059. }
  3060. }
  3061. }
  3062. if (!graphUpdateReport.IsEmpty())
  3063. {
  3064. // currently, it is expected that there are no deleted old slots, those need manual correction
  3065. AZ_Error("ScriptCanvas", graphUpdateReport.m_deletedOldSlots.empty(), "Graph upgrade path: If old slots are deleted, manual upgrading is required");
  3066. UpdateConnectionStatus(*this, graphUpdateReport);
  3067. }
  3068. AZStd::unordered_set<AZ::EntityId> graphCanvasNodesToDelete;
  3069. for (auto scriptCanvasNode : outOfDateNodes)
  3070. {
  3071. auto graphCanvasNodeId = scriptCanvasToGraphCanvasMapping[scriptCanvasNode->GetEntityId()];
  3072. ScriptCanvas::UpdateResult updateResult = scriptCanvasNode->UpdateNode();
  3073. OnVersionConversionEnd((*scriptCanvasNode));
  3074. switch (updateResult)
  3075. {
  3076. case ScriptCanvas::UpdateResult::DeleteNode:
  3077. {
  3078. graphNeedsDirtying = true;
  3079. deletedNodes.insert(scriptCanvasNode->GetEntityId());
  3080. graphCanvasNodesToDelete.insert(graphCanvasNodeId);
  3081. break;
  3082. }
  3083. default:
  3084. {
  3085. graphNeedsDirtying = true;
  3086. break;
  3087. }
  3088. }
  3089. }
  3090. if (!graphCanvasNodesToDelete.empty())
  3091. {
  3092. GraphCanvas::SceneRequestBus::Event(graphCanvasGraphId, &GraphCanvas::SceneRequests::Delete, graphCanvasNodesToDelete);
  3093. }
  3094. AZStd::vector< AZ::EntityId > connectionIds = GetConnections();
  3095. for (const AZ::EntityId& connectionId : connectionIds)
  3096. {
  3097. ScriptCanvas::Endpoint scriptCanvasSourceEndpoint;
  3098. ScriptCanvas::Endpoint scriptCanvasTargetEndpoint;
  3099. ScriptCanvas::ConnectionRequestBus::EventResult(scriptCanvasSourceEndpoint, connectionId, &ScriptCanvas::ConnectionRequests::GetSourceEndpoint);
  3100. ScriptCanvas::ConnectionRequestBus::EventResult(scriptCanvasTargetEndpoint, connectionId, &ScriptCanvas::ConnectionRequests::GetTargetEndpoint);
  3101. ScriptCanvas::Slot* sourceSlot = FindSlot(scriptCanvasSourceEndpoint);
  3102. ScriptCanvas::Slot* targetSlot = FindSlot(scriptCanvasTargetEndpoint);
  3103. // Implicit connections don't have corresponding Graph Canvas entities
  3104. if ((sourceSlot && sourceSlot->CreatesImplicitConnections()) ||
  3105. (targetSlot && targetSlot->CreatesImplicitConnections()))
  3106. {
  3107. continue;
  3108. }
  3109. AZ::EntityId graphCanvasSourceNode;
  3110. auto scriptCanvasIter = scriptCanvasToGraphCanvasMapping.find(scriptCanvasSourceEndpoint.GetNodeId());
  3111. if (scriptCanvasIter != scriptCanvasToGraphCanvasMapping.end())
  3112. {
  3113. graphCanvasSourceNode = scriptCanvasIter->second;
  3114. }
  3115. else
  3116. {
  3117. AZ_Warning("ScriptCanvas", false, "Could not find ScriptCanvas Node with id %llu", static_cast<AZ::u64>(scriptCanvasSourceEndpoint.GetNodeId()));
  3118. }
  3119. AZ::EntityId graphCanvasSourceSlotId;
  3120. SlotMappingRequestBus::EventResult(graphCanvasSourceSlotId, graphCanvasSourceNode, &SlotMappingRequests::MapToGraphCanvasId, scriptCanvasSourceEndpoint.GetSlotId());
  3121. if (!graphCanvasSourceSlotId.IsValid())
  3122. {
  3123. // For the EBusHandler's I need to remap these to a different visual node.
  3124. // Since multiple GraphCanvas nodes depict a single ScriptCanvas EBus node.
  3125. if (EBusHandlerNodeDescriptorRequestBus::FindFirstHandler(graphCanvasSourceNode) != nullptr)
  3126. {
  3127. GraphCanvas::Endpoint graphCanvasEventEndpoint;
  3128. EBusHandlerNodeDescriptorRequestBus::EventResult(graphCanvasEventEndpoint, graphCanvasSourceNode, &EBusHandlerNodeDescriptorRequests::MapSlotToGraphCanvasEndpoint, scriptCanvasSourceEndpoint.GetSlotId());
  3129. graphCanvasSourceSlotId = graphCanvasEventEndpoint.GetSlotId();
  3130. }
  3131. if (!graphCanvasSourceSlotId.IsValid())
  3132. {
  3133. 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());
  3134. DisconnectById(connectionId);
  3135. continue;
  3136. }
  3137. }
  3138. GraphCanvas::Endpoint graphCanvasTargetEndpoint;
  3139. scriptCanvasIter = scriptCanvasToGraphCanvasMapping.find(scriptCanvasTargetEndpoint.GetNodeId());
  3140. if (scriptCanvasIter != scriptCanvasToGraphCanvasMapping.end())
  3141. {
  3142. graphCanvasTargetEndpoint.m_nodeId = scriptCanvasIter->second;
  3143. }
  3144. else
  3145. {
  3146. AZ_Warning("ScriptCanvas", false, "Could not find ScriptCanvas Node with id %llu", static_cast<AZ::u64>(scriptCanvasSourceEndpoint.GetNodeId()));
  3147. }
  3148. SlotMappingRequestBus::EventResult(graphCanvasTargetEndpoint.m_slotId, graphCanvasTargetEndpoint.GetNodeId(), &SlotMappingRequests::MapToGraphCanvasId, scriptCanvasTargetEndpoint.GetSlotId());
  3149. if (!graphCanvasTargetEndpoint.IsValid())
  3150. {
  3151. // For the EBusHandler's I need to remap these to a different visual node.
  3152. // Since multiple GraphCanvas nodes depict a single ScriptCanvas EBus node.
  3153. if (EBusHandlerNodeDescriptorRequestBus::FindFirstHandler(graphCanvasTargetEndpoint.GetNodeId()) != nullptr)
  3154. {
  3155. EBusHandlerNodeDescriptorRequestBus::EventResult(graphCanvasTargetEndpoint, graphCanvasTargetEndpoint.GetNodeId(), &EBusHandlerNodeDescriptorRequests::MapSlotToGraphCanvasEndpoint, scriptCanvasTargetEndpoint.GetSlotId());
  3156. }
  3157. if (!graphCanvasTargetEndpoint.IsValid())
  3158. {
  3159. 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());
  3160. DisconnectById(connectionId);
  3161. continue;
  3162. }
  3163. }
  3164. AZ::EntityId graphCanvasConnectionId;
  3165. GraphCanvas::SlotRequestBus::EventResult(graphCanvasConnectionId, graphCanvasSourceSlotId, &GraphCanvas::SlotRequests::DisplayConnectionWithEndpoint, graphCanvasTargetEndpoint);
  3166. if (graphCanvasConnectionId.IsValid())
  3167. {
  3168. AZStd::any* userData = nullptr;
  3169. GraphCanvas::ConnectionRequestBus::EventResult(userData, graphCanvasConnectionId, &GraphCanvas::ConnectionRequests::GetUserData);
  3170. if (userData)
  3171. {
  3172. (*userData) = connectionId;
  3173. SceneMemberMappingConfigurationRequestBus::Event(graphCanvasConnectionId, &SceneMemberMappingConfigurationRequests::ConfigureMapping, connectionId);
  3174. }
  3175. }
  3176. }
  3177. // Fix up leaked data elements
  3178. auto mapIter = m_graphCanvasSaveData.begin();
  3179. while (mapIter != m_graphCanvasSaveData.end())
  3180. {
  3181. // 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
  3182. // of old save data for no reason.
  3183. //
  3184. // Need to bypass our internal save data for graph canvas information
  3185. if (scriptCanvasToGraphCanvasMapping.find(mapIter->first) == scriptCanvasToGraphCanvasMapping.end()
  3186. && mapIter->first != GetEntityId())
  3187. {
  3188. delete mapIter->second;
  3189. mapIter = m_graphCanvasSaveData.erase(mapIter);
  3190. }
  3191. else
  3192. {
  3193. ++mapIter;
  3194. }
  3195. }
  3196. auto currentIter = GetGraphData()->m_scriptEventAssets.begin();
  3197. while (currentIter != GetGraphData()->m_scriptEventAssets.end())
  3198. {
  3199. if (assetSanitizationSet.find(currentIter->first) == assetSanitizationSet.end())
  3200. {
  3201. currentIter->second = ScriptEvents::ScriptEventsAssetPtr{};
  3202. currentIter = GetGraphData()->m_scriptEventAssets.erase(currentIter);
  3203. graphNeedsDirtying = true;
  3204. }
  3205. else
  3206. {
  3207. ++currentIter;
  3208. }
  3209. }
  3210. for (auto node : sanityCheckRequiredNodes)
  3211. {
  3212. if (node)
  3213. {
  3214. node->SanityCheckDynamicDisplay();
  3215. }
  3216. }
  3217. GraphCanvas::SceneRequestBus::Event(graphCanvasGraphId, &GraphCanvas::SceneRequests::SignalLoadEnd);
  3218. EditorGraphNotificationBus::Event(GetScriptCanvasId(), &EditorGraphNotifications::OnGraphCanvasSceneDisplayed);
  3219. }
  3220. GraphCanvas::SceneRequestBus::Event(GetGraphCanvasGraphId(), &GraphCanvas::SceneRequests::ProcessEnableDisableQueue);
  3221. if (m_graphCanvasSaveVersion != GraphCanvas::EntitySaveDataContainer::CurrentVersion)
  3222. {
  3223. for (auto saveDataPair : m_graphCanvasSaveData)
  3224. {
  3225. auto graphCanvasIter = scriptCanvasToGraphCanvasMapping.find(saveDataPair.first);
  3226. OnSaveDataDirtied(graphCanvasIter->second);
  3227. }
  3228. m_graphCanvasSaveVersion = GraphCanvas::EntitySaveDataContainer::CurrentVersion;
  3229. graphNeedsDirtying = true;
  3230. }
  3231. RequestPopPreventUndoStateUpdate();
  3232. if (graphNeedsDirtying)
  3233. {
  3234. SignalDirty();
  3235. }
  3236. MarkVersion();
  3237. }
  3238. void EditorGraph::OnGraphCanvasSceneVisible()
  3239. {
  3240. DisplayUpdateToast();
  3241. }
  3242. AZStd::unordered_map< AZ::EntityId, GraphCanvas::EntitySaveDataContainer* > EditorGraph::GetGraphCanvasSaveData()
  3243. {
  3244. return m_graphCanvasSaveData;
  3245. }
  3246. void EditorGraph::UpdateGraphCanvasSaveData(const AZStd::unordered_map< AZ::EntityId, GraphCanvas::EntitySaveDataContainer* >& saveData)
  3247. {
  3248. QScopedValueRollback<bool> ignoreRequests(m_ignoreSaveRequests, true);
  3249. GraphCanvas::GraphId graphCanvasGraphId = GetGraphCanvasGraphId();
  3250. GraphCanvas::SceneNotificationBus::Handler::BusDisconnect(graphCanvasGraphId);
  3251. GraphCanvas::GraphModelRequestBus::Handler::BusDisconnect(graphCanvasGraphId);
  3252. for (auto& entry : m_graphCanvasSaveData)
  3253. {
  3254. delete entry.second;
  3255. }
  3256. m_graphCanvasSaveData = saveData;
  3257. DisplayGraphCanvasScene();
  3258. }
  3259. void EditorGraph::ClearGraphCanvasScene()
  3260. {
  3261. GraphCanvas::GraphId graphCanvasGraphId = GetGraphCanvasGraphId();
  3262. RequestPushPreventUndoStateUpdate();
  3263. // Wipe out all of the Graph Canvas Visuals
  3264. GraphCanvas::SceneRequestBus::Event(graphCanvasGraphId, &GraphCanvas::SceneRequests::ClearScene);
  3265. RequestPopPreventUndoStateUpdate();
  3266. }
  3267. }