fix(viz): Add visualization steps

pollination · Feb 17, 2024 · c13341d · c13341d
1 parent 78eddb8
commit c13341d
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 1 deletion.
diff --git a/pollination/direct_sun_hours/_postprocess.py b/pollination/direct_sun_hours/_postprocess.py
@@ -2,14 +2,21 @@
 from dataclasses import dataclass
 from pollination.honeybee_radiance.grid import MergeFolderData
 from pollination.honeybee_radiance_postprocess.grid import MergeFolderData as MergeFolderDataPostprocess
+from pollination.honeybee_radiance.post_process import DirectSunHoursVisMetadata
 from pollination.path.copy import CopyFile, CopyFileMultiple
+from pollination.honeybee_display.translate import ModelToVis
 
 
 @dataclass
 class DirectSunHoursPostprocess(GroupedDAG):
     """Post-process for direct sun hours."""
 
     # inputs
+    model = Inputs.file(
+        description='Input Honeybee model.',
+        extensions=['json', 'hbjson', 'pkl', 'hbpkl', 'zip']
+    )
+
     input_folder = Inputs.folder(
         description='Folder with initial results before redistributing the '
         'results to the original grids.'
@@ -97,6 +104,34 @@ def restructure_cumulative_results(
             }
         ]
 
+    @task(
+        template=DirectSunHoursVisMetadata,
+        needs=[restructure_cumulative_results]
+    )
+    def create_vis_metadata(self):
+        return [
+            {
+                'from': DirectSunHoursVisMetadata()._outputs.cfg_file,
+                'to': 'results/cumulative/vis_metadata.json'
+            }
+        ]
+
+    @task(template=ModelToVis, needs=[create_vis_metadata])
+    def create_vsf(
+        self, model=model, grid_data='results/cumulative', output_format='vsf'
+    ):
+        return [
+            {
+                'from': ModelToVis()._outputs.output_file,
+                'to': 'visualization.vsf'
+            }
+        ]
+
     results = Outputs.folder(
         source='results', description='results folder.'
     )
+
+    visualization = Outputs.file(
+        source='visualization.vsf',
+        description='Direct sun hours result visualization in VisualizationSet format.'
+    )
diff --git a/pollination/direct_sun_hours/entry.py b/pollination/direct_sun_hours/entry.py
@@ -137,16 +137,26 @@ def postprocess_direct_sun_hours(
         self, input_folder=prepare_folder_direct_sun_hours._outputs.initial_results,
         grids_info=prepare_folder_direct_sun_hours._outputs.resources,
         sun_up_hours=prepare_folder_direct_sun_hours._outputs.resources,
+        dist_info=prepare_folder_direct_sun_hours._outputs.resources,
         timestep=prepare_folder_direct_sun_hours._outputs.resources,
-        dist_info=prepare_folder_direct_sun_hours._outputs.resources
+        model=model
     ):
         return [
             {
                 'from': DirectSunHoursPostprocess()._outputs.results,
                 'to': 'results'
+            },
+            {
+                'from': DirectSunHoursPostprocess()._outputs.visualization,
+                'to': 'visualization.vsf'
             }
         ]
 
+    visualization = Outputs.file(
+        source='visualization.vsf',
+        description='Result visualization in VisualizationSet format.'
+    )
+
     direct_sun_hours = Outputs.folder(
         source='results/direct_sun_hours',
         description='Hourly results for direct sun hours.',