export csv with SaveFileDialog (#49)

DynamoDS · Jul 16, 2024 · ee93525 · ee93525
1 parent 5ec3826
commit ee93525
Show file tree

Hide file tree

Showing 3 changed files with 45 additions and 1 deletion.
diff --git a/TuneUp/TuneUpWindow.xaml b/TuneUp/TuneUpWindow.xaml
@@ -96,7 +96,17 @@
                 IsEnabled="{Binding  Path=IsRecomputeEnabled}"
                 Click="RecomputeGraph_Click"
                 Padding="5,2,5,2">
-                Force Re-execute
+                Run All
+            </Button>
+            <Button
+                Name="ExportTimes"
+                Width="Auto"
+                Height="Auto"
+                Margin="2,1,1,10"
+                Padding="5,2,5,2"
+                IsEnabled="{Binding  Path=IsRecomputeEnabled}"
+                Click="ExportTimes_Click">
+                Export
             </Button>
             <Label Foreground="White">Total Graph Execution Time: </Label>
             <Label Foreground="White"

diff --git a/TuneUp/TuneUpWindow.xaml.cs b/TuneUp/TuneUpWindow.xaml.cs
@@ -116,5 +116,10 @@ private void RecomputeGraph_Click(object sender, RoutedEventArgs e)
         {
             (NodeAnalysisTable.DataContext as TuneUpWindowViewModel).ResetProfiling();
         }
+
+        private void ExportTimes_Click(object sender, RoutedEventArgs e)
+        {
+            (NodeAnalysisTable.DataContext as TuneUpWindowViewModel).ExportToCsv();
+        }
     }
 }
diff --git a/TuneUp/TuneUpWindowViewModel.cs b/TuneUp/TuneUpWindowViewModel.cs
@@ -3,6 +3,7 @@
 using System.Collections.ObjectModel;
 using System.ComponentModel;
 using System.ComponentModel.DataAnnotations;
+using System.IO;
 using System.Linq;
 using System.Threading;
 using System.Windows.Data;
@@ -12,6 +13,7 @@
 using Dynamo.Graph.Workspaces;
 using Dynamo.ViewModels;
 using Dynamo.Wpf.Extensions;
+using Microsoft.Win32;
 
 namespace TuneUp
 {
@@ -448,5 +450,32 @@ public void Dispose()
         }
 
         #endregion
+
+        #region Export Node times
+
+        /// <summary>
+        /// Exports the ProfiledNodesCollection to a CSV file.
+        /// </summary>
+        public void ExportToCsv()
+        {
+            SaveFileDialog saveFileDialog = new SaveFileDialog();
+            //saveFileDialog.RestoreDirectory = false;
+            saveFileDialog.Filter = "CSV file (*.csv)|*.csv|All files (*.*)|*.*";
+
+            if (saveFileDialog.ShowDialog() == true)
+            {
+                using (var writer = new StreamWriter(saveFileDialog.FileName))
+                {
+                    writer.WriteLine("Execution Order,Name,Execution Time (ms)");
+
+                    foreach (ProfiledNodeViewModel node in ProfiledNodesCollection.View.Cast<ProfiledNodeViewModel>())
+                    {
+                        writer.WriteLine($"{node.ExecutionOrderNumber},{node.Name},{node.ExecutionMilliseconds}");
+                    }
+                }
+            }
+        }
+
+        #endregion
     }
 }