custom label discretising

Chengxuan-Li · Feb 6, 2024 · 8aea404 · 8aea404
1 parent 62a11c5
commit 8aea404
Show file tree

Hide file tree

Showing 5 changed files with 35 additions and 10 deletions.
diff --git a/Legend.cs b/Legend.cs
@@ -27,7 +27,9 @@ public void SetDimStyle()
         public double Scale = 1.0;
         public double Height = 1.0;
         public int DisiredCount = 20;
-        public List<double> Labels => PrettyBreaks(MinValue, MaxValue, DisiredCount);
+        public bool AutomaticLabelling = true;
+        public List<double> AlternativeLabels;
+        public List<double> Labels => AutomaticLabelling? PrettyBreaks(MinValue, MaxValue, DisiredCount) : AlternativeLabels;
         public bool OnLeft = true;
         public List<double> Positions
         {

diff --git a/ProfileVisualiser.cs b/ProfileVisualiser.cs
@@ -24,6 +24,10 @@ public class ProfileVisualiser
 
         public Legend Legend;
 
+        public bool OverrideLegendMinMax = false;
+        public double OverrideLegendMin;
+        public double OverrideLegendMax;
+
         public double ActualMinValue => Math.Min(YValues.Min(), SuggestedMinValue);
         public double ActualMaxValue => Math.Max(YValues.Max(), SuggestedMaxValue);
 
@@ -73,6 +77,8 @@ public void LegendLeft()
                 Height = this.Height,
                 Plane = plane,
                 OnLeft = true,
+                AutomaticLabelling = !OverrideLegendMinMax,
+                AlternativeLabels = Legend.PrettyBreaks(OverrideLegendMin, OverrideLegendMax, 7),
             };
             return;
         }
@@ -93,6 +99,8 @@ public void LegendRight()
                 Height = this.Height,
                 Plane = plane,
                 OnLeft = false,
+                AutomaticLabelling = !OverrideLegendMinMax,
+                AlternativeLabels = Legend.PrettyBreaks(OverrideLegendMin, OverrideLegendMax, 7),
             };
             return;
         }

diff --git a/ProfileDisplay.cs → TemperatureProfileDisplay.cs b/ProfileDisplay.cs → TemperatureProfileDisplay.cs
@@ -6,17 +6,17 @@
 
 namespace WallSectionWidget
 {
-    public class ProfileDisplay : GH_Component
+    public class TemperatureProfileDisplay : GH_Component
     {
 
         public List<TextEntity> PreviewTexts = new List<TextEntity>();
 
         /// <summary>
         /// Initializes a new instance of the ProfileDisplay class.
         /// </summary>
-        public ProfileDisplay()
-          : base("ProfileDisplay", "PFDis",
-              "Display for a wall section numeric profile",
+        public TemperatureProfileDisplay()
+          : base("TemperatureProfileDisplay", "TPFDis",
+              "Display for a wall section temperature profile",
               "WallSectionWidget", "Display")
         {
         }
@@ -32,11 +32,14 @@ protected override void RegisterInputParams(GH_Component.GH_InputParamManager pM
             pManager.AddPlaneParameter("Plane", "Pl", "Plane", GH_ParamAccess.item);
             pManager.AddNumberParameter("Height", "H", "Height", GH_ParamAccess.item);
             pManager.AddNumberParameter("Scale", "S", "Custom scale factor", GH_ParamAccess.item);
+            pManager.AddNumberParameter("OverrideLegendMin", "OLMin", "Value used to override the legend minimum", GH_ParamAccess.item);
+            pManager.AddNumberParameter("OverrideLegendMax", "OLMax", "Value used to override the legend maximum", GH_ParamAccess.item);
             pManager[1].Optional = true;
             pManager[2].Optional = true;
             pManager[3].Optional = true;
             pManager[4].Optional = true;
-            //pManager[5].Optional = true;
+            pManager[5].Optional = true;
+            pManager[6].Optional = true;
         }
 
         /// <summary>
@@ -86,8 +89,20 @@ protected override void SolveInstance(IGH_DataAccess DA)
                 AddRuntimeMessage(GH_RuntimeMessageLevel.Error, "Input error: scale should be bigger than zero");
                 return;
             }
-
-            ProfileVisualiser vis = new ProfileVisualiser(model.Depths, model.DewPoints);
+            double overrideLegendMin = 0.0;
+            double overrideLegendMax = 30.0;
+            bool overrideLegendMinMax = false;
+            if (DA.GetData(5, ref overrideLegendMin) && DA.GetData(6, ref overrideLegendMax))
+            {
+                overrideLegendMinMax = true;
+            }
+            ProfileVisualiser vis = new ProfileVisualiser(model.Depths, model.Temperatures);
+            vis.OverrideLegendMinMax = overrideLegendMinMax;
+            if (overrideLegendMinMax)
+            {
+                vis.OverrideLegendMax = overrideLegendMax;
+                vis.OverrideLegendMin = overrideLegendMin;
+            }
             vis.Plane = plane;
             vis.Height = height;
             vis.Scale = scale;
@@ -117,7 +132,7 @@ protected override System.Drawing.Bitmap Icon
         {
             get
             {
-                //You can add image files to your project resources and access them like this:
+                // You can add image files to your project resources and access them like this:
                 // return Resources.IconForThisComponent;
                 return null;
             }

diff --git a/WallSectionWidget.csproj b/WallSectionWidget.csproj
@@ -71,7 +71,7 @@
     <Compile Include="GHIOParam.cs" />
     <Compile Include="ModelParameters.cs" />
     <Compile Include="Parameters.cs" />
-    <Compile Include="ProfileDisplay.cs" />
+    <Compile Include="TemperatureProfileDisplay.cs" />
     <Compile Include="ProfileVisualiser.cs" />
     <Compile Include="Properties\Resources.Designer.cs">
       <AutoGen>True</AutoGen>

diff --git a/tests/sample.gh b/tests/sample.gh