fix(viz): Add visualization steps

pollination · Dec 28, 2023 · e691280 · e691280
1 parent ca2db68
commit e691280
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 0 deletions.
diff --git a/pollination/annual_daylight/_post_process.py b/pollination/annual_daylight/_post_process.py
@@ -3,6 +3,7 @@
 from pollination_dsl.dag import Inputs, GroupedDAG, task, Outputs
 from pollination.honeybee_radiance_postprocess.grid import MergeFolderMetrics
 from pollination.honeybee_radiance_postprocess.post_process import GridSummaryMetrics
+from pollination.honeybee_display.translate import ModelToVis
 
 
 @dataclass
@@ -67,6 +68,23 @@ def grid_summary_metrics(
             }
         ]
 
+    @task(template=ModelToVis, needs=[restructure_metrics])
+    def create_vsf(
+        self, model=model, grid_data='metrics', active_grid_data='udi',
+        output_format='vsf'
+    ):
+        return [
+            {
+                'from': ModelToVis()._outputs.output_file,
+                'to': 'visualization.vsf'
+            }
+        ]
+
+    visualization = Outputs.file(
+        source='visualization.vsf',
+        description='Annual daylight result visualization in VisualizationSet format.'
+    )
+
     metrics = Outputs.folder(
         source='metrics', description='metrics folder.'
     )

diff --git a/pollination/annual_daylight/entry.py b/pollination/annual_daylight/entry.py
@@ -185,12 +185,21 @@ def post_process_annual_daylight(
                 'from': AnnualDaylightPostProcess()._outputs.metrics,
                 'to': 'metrics'
             },
+            {
+                'from': AnnualDaylightPostProcess()._outputs.visualization,
+                'to': 'visualization.vsf'
+            },
             {
                 'from': AnnualDaylightPostProcess()._outputs.grid_summary,
                 'to': 'grid_summary.csv'
             }
         ]
 
+    visualization = Outputs.file(
+        source='visualization.vsf',
+        description='Result visualization in VisualizationSet format.'
+    )
+
     metrics = Outputs.folder(
         source='metrics', description='Annual metrics folder.'
     )