stats computed, but very unstable

nodepen · Jul 23, 2023 · f08194e · f08194e
1 parent 2a18ec3
commit f08194e
Showing 1 changed file with 42 additions and 1 deletion.
diff --git a/packages/converters/models/NodePenDataTree.cs b/packages/converters/models/NodePenDataTree.cs
@@ -1,6 +1,7 @@
 using Speckle.Core.Models;
 using System;
 using System.Collections.Generic;
+using System.Linq;
 using Speckle.Newtonsoft.Json;
 
 namespace NodePen.Converters
@@ -41,7 +42,8 @@ public void ComputeStats()
             Stats = new NodePenDataTreeStats();
 
             ComputeStructure();
-            // TODO: Compute other stats
+            ComputeTotalCounts();
+            ComputeTypes();
         }
 
         private void ComputeStructure()
@@ -72,6 +74,45 @@ private void ComputeStructure()
             Stats.TreeStructure = "tree";
         }
 
+        private void ComputeTotalCounts()
+        {
+            var valueCounts = new List<int>();
+
+            foreach (var branch in Branches)
+            {
+                valueCounts.Add(branch.Values.Count);
+            }
+
+            if (valueCounts.Count == 0)
+            {
+                Stats.BranchCount = Branches.Count;
+                Stats.ValueCount = 0;
+                return;
+            }
+
+            Stats.BranchCount = Branches.Count;
+            Stats.BranchValueCountDomain = new List<int>{
+              valueCounts.Min(),
+              valueCounts.Max()
+            };
+            Stats.ValueCount = valueCounts.Sum();
+        }
+
+        private void ComputeTypes()
+        {
+            var types = new HashSet<string>();
+
+            foreach (var branch in Branches)
+            {
+                foreach (var value in branch.Values)
+                {
+                    types.Add(value.Type.ToLower());
+                }
+            }
+
+            Stats.ValueTypes = types.ToList();
+        }
+
     }
 
     /// <summary>