fix(extra): Add option to filter viewport of visualization set legend

ladybug_grasshopper/json/LB_Preview_VisualizationSet.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.8.1", 
+  "version": "1.8.2", 
   "nickname": "VisSet", 
   "outputs": [
     [
@@ -40,6 +40,13 @@
       "description": "Optional text or an integer to select a specific data set from analysis\ngeometries within the Visualization Set. Note that this input only has\nmeaning for Visualization Sets that contain multiple data sets assigned\nto the same geometry. When using an integer, this will refer to the\nindex of the data set to be visualized (starting with 0). When using\ntext, this will refer to the name of the data type for the data set\nto be displayed.", 
       "type": "string", 
       "default": null
+    }, 
+    {
+      "access": "item", 
+      "name": "viewport_", 
+      "description": "Text for the name of the Rhino viewport to which the 2D screen-oriented\nlegend will be rendered. If unspecified, the 2D legend will be rendered\nin all viewports. Acceptable inputs include:\n-\nPerspective\nTop\nBottom\nLeft\nRight\nFront\nBack\nany view name that has been saved within the Rhino file", 
+      "type": "string", 
+      "default": null
     }
   ], 
   "subcategory": "4 :: Extra", 

ladybug_grasshopper/src/LB Preview VisualizationSet.py

@@ -37,6 +37,18 @@
             index of the data set to be visualized (starting with 0). When using
             text, this will refer to the name of the data type for the data set
             to be displayed.
+        viewport_: Text for the name of the Rhino viewport to which the 2D screen-oriented
+            legend will be rendered. If unspecified, the 2D legend will be rendered
+            in all viewports. Acceptable inputs include:
+                -
+                Perspective
+                Top
+                Bottom
+                Left
+                Right
+                Front
+                Back
+                any view name that has been saved within the Rhino file
 
     Returns:
         vs: A VisualizationSet object that can be baked into the Rhino document by
@@ -56,11 +68,12 @@ def __init__(self):
         super(MyComponent,self).__init__()
         self.vis_con = None
         self.vs_goo = None
+        self.viewport = None
 
-    def RunScript(self, _vis_set, legend_par_, leg_par2d_, data_set_):
+    def RunScript(self, _vis_set, legend_par_, leg_par2d_, data_set_, viewport_):
         ghenv.Component.Name = 'LB Preview VisualizationSet'
         ghenv.Component.NickName = 'VisSet'
-        ghenv.Component.Message = '1.8.1'
+        ghenv.Component.Message = '1.8.2'
         ghenv.Component.Category = 'Ladybug'
         ghenv.Component.SubCategory = '4 :: Extra'
         ghenv.Component.AdditionalHelpFromDocStrings = '1'
@@ -165,9 +178,11 @@ def RunScript(self, _vis_set, legend_par_, leg_par2d_, data_set_):
                                     geo.active_data = i
             self.vis_con = VisualizationSetConverter(vis_set_obj, leg3d, leg2d)
             self.vs_goo = VisSetGoo(vis_set_obj)
+            self.viewport = viewport_
         else:
             self.vis_con = None
             self.vs_goo = None
+            self.viewport = None
 
         # return the bake-able version of the visualization set 
         return self.vs_goo
@@ -209,10 +224,12 @@ def DrawViewportWires(self, args):
                     display.DrawCone(*draw_args)
                 for draw_args in self.vis_con.draw_cylinder:
                     display.DrawCylinder(*draw_args)
-                for draw_args in self.vis_con.draw_2d_text:
-                    display.Draw2dText(*draw_args)
-                for draw_args in self.vis_con.draw_sprite:
-                    display.DrawSprite(*draw_args)
+                if self.viewport is None or \
+                        self.viewport.lower() == args.Viewport.Name.lower():
+                    for draw_args in self.vis_con.draw_2d_text:
+                        display.Draw2dText(*draw_args)
+                    for draw_args in self.vis_con.draw_sprite:
+                        display.DrawSprite(*draw_args)
         except Exception, e:
             System.Windows.Forms.MessageBox.Show(str(e), "script error")