From 7cfad68508670563e310dc5250bc99cd64ddc478 Mon Sep 17 00:00:00 2001
From: michaelkirschner <mjk.kirschner@gmail.com>
Date: Sun, 13 Dec 2020 00:48:09 -0500
Subject: [PATCH] first commit  - it works

---
 TuneUp/ProfiledNodeViewModel.cs | 14 ++++++++++++++
 TuneUp/TuneUp.csproj            |  2 +-
 TuneUp/TuneUpWindow.xaml        | 25 +++++++++++++++++++++++++
 TuneUp/TuneUpWindow.xaml.cs     | 20 ++++++++++++++++++++
 TuneUp/TuneUpWindowViewModel.cs | 15 +++++++++++++++
 5 files changed, 75 insertions(+), 1 deletion(-)

diff --git a/TuneUp/ProfiledNodeViewModel.cs b/TuneUp/ProfiledNodeViewModel.cs
index 8c52c92..a5c00bd 100644
--- a/TuneUp/ProfiledNodeViewModel.cs
+++ b/TuneUp/ProfiledNodeViewModel.cs
@@ -130,6 +130,20 @@ public string StateDescription
             }
         }
 
+        private string callsiteData;
+        /// <summary>
+        /// A string representing the serialized trace data contained in this callsite.
+        /// </summary>
+        public string CallSiteData
+        {
+            get { return callsiteData; }
+            set
+            {
+                callsiteData = value;
+                RaisePropertyChanged(nameof(CallSiteData));
+            }
+        }
+
         internal NodeModel NodeModel { get; set; }
 
         #endregion
diff --git a/TuneUp/TuneUp.csproj b/TuneUp/TuneUp.csproj
index fcd97bc..dc5f950 100644
--- a/TuneUp/TuneUp.csproj
+++ b/TuneUp/TuneUp.csproj
@@ -93,7 +93,7 @@
   </Target>
   -->
   <PropertyGroup>
-    <DynamoVersion>2.5</DynamoVersion>
+    <DynamoVersion>2.11</DynamoVersion>
     <PackageName>TuneUp</PackageName>
     <PackageFolder>$(ProjectDir)dist\$(PackageName)\</PackageFolder>
     <BinFolder>$(PackageFolder)bin\</BinFolder>
diff --git a/TuneUp/TuneUpWindow.xaml b/TuneUp/TuneUpWindow.xaml
index 3316f72..4a46595 100644
--- a/TuneUp/TuneUpWindow.xaml
+++ b/TuneUp/TuneUpWindow.xaml
@@ -98,6 +98,15 @@
                 Padding="5,2,5,2">
                 Force Re-execute
             </Button>
+            <Button
+                Name="ClearTraceButton"
+                Width="Auto"
+                Height="Auto"
+                Margin="2,1,1,10"
+                Click="ClearTrace_Click"
+                Padding="5,2,5,2">
+                Clear Trace/ElementBinding Data
+            </Button>
             <Label Foreground="White">Total Graph Execution Time: </Label>
             <Label Foreground="White"
                    Name="TotalGraphExecutiontimeLabel"
@@ -189,6 +198,22 @@
                         </DataGridTextColumn.ElementStyle>
                     </DataGridTextColumn>
 
+                    <!-- callsite data -->
+                    <DataGridTextColumn 
+                    Header="callsite data" 
+                    Binding="{Binding CallSiteData}" 
+                    Foreground="#aaaaaa" 
+                    IsReadOnly="True" 
+                    Width="Auto">
+                        <DataGridTextColumn.ElementStyle>
+                            <Style TargetType="TextBlock" >
+                                <Setter Property="VerticalAlignment" Value="Center" />
+                                <Setter Property="TextBlock.TextWrapping" Value="Wrap" />
+                                <Setter Property="Margin" Value="10,0,10,0"/>
+                            </Style>
+                        </DataGridTextColumn.ElementStyle>
+                    </DataGridTextColumn>
+
                 </DataGrid.Columns>
             </DataGrid>
         </StackPanel>
diff --git a/TuneUp/TuneUpWindow.xaml.cs b/TuneUp/TuneUpWindow.xaml.cs
index 155c9b1..b0b49a8 100644
--- a/TuneUp/TuneUpWindow.xaml.cs
+++ b/TuneUp/TuneUpWindow.xaml.cs
@@ -4,8 +4,10 @@
 using System.Windows.Controls;
 using Dynamo.Extensions;
 using Dynamo.Graph.Nodes;
+using Dynamo.Graph.Workspaces;
 using Dynamo.Models;
 using Dynamo.Utilities;
+using Dynamo.ViewModels;
 using Dynamo.Wpf.Extensions;
 
 namespace TuneUp
@@ -90,5 +92,23 @@ private void RecomputeGraph_Click(object sender, RoutedEventArgs e)
         {
             (NodeAnalysisTable.DataContext as TuneUpWindowViewModel).ResetProfiling();
         }
+
+        private void ClearTrace_Click(object sender, RoutedEventArgs e)
+        {
+            var engine = (viewLoadedParams.CurrentWorkspaceModel as HomeWorkspaceModel).EngineController;
+            var callsites = engine.LiveRunnerRuntimeCore.RuntimeData.GetCallsitesForNodes(
+                viewLoadedParams.CurrentWorkspaceModel.Nodes.Select(x => x.GUID),
+                engine.LiveRunnerCore.DSExecutable);
+
+            foreach (var kvp in callsites)
+            {
+                foreach (var callsite in kvp.Value)
+                {
+                    callsite.TraceData.Clear();
+                }
+            }
+            (NodeAnalysisTable.DataContext as TuneUpWindowViewModel).GetCallSiteDataForCurrentNodesAndUpdateView();
+
+        }
     }
 }
diff --git a/TuneUp/TuneUpWindowViewModel.cs b/TuneUp/TuneUpWindowViewModel.cs
index 27ea38e..a1f6e1c 100644
--- a/TuneUp/TuneUpWindowViewModel.cs
+++ b/TuneUp/TuneUpWindowViewModel.cs
@@ -260,9 +260,24 @@ private void CurrentWorkspaceModel_EvaluationCompleted(object sender, Dynamo.Mod
                 ProfiledNodesCollection.SortDescriptions.Add(new SortDescription(nameof(ProfiledNodeViewModel.ExecutionOrderNumber), ListSortDirection.Descending));
                 if (ProfiledNodesCollection.View != null)
                     ProfiledNodesCollection.View.Refresh();
+                GetCallSiteDataForCurrentNodesAndUpdateView();
             });
         }
 
+        internal void GetCallSiteDataForCurrentNodesAndUpdateView()
+        {
+            foreach (var item in ProfiledNodes)
+            {
+                if (item.NodeModel != null)
+                {
+                    var engine = (viewLoadedParams.CurrentWorkspaceModel as HomeWorkspaceModel).EngineController;
+                    var callsites = engine.LiveRunnerRuntimeCore.RuntimeData.GetCallsitesForNodes(new Guid[] { (item.NodeModel.GUID) }, engine.LiveRunnerCore.DSExecutable).FirstOrDefault();
+                    item.CallSiteData = String.Join(Environment.NewLine, callsites.Value.Select(x => x.GetTraceDataToSave()));
+                }
+
+            }
+        }
+
         /// <summary>
         /// Update execution time rows. These rows are always removed and re-added after each run.
         /// May consider instead, always updating them in the future.