fix(postprocess): Add multiple functions for vis-metadata

pollination/honeybee_radiance/post_process.py

@@ -570,6 +570,34 @@ def create_point_in_time_config(self):
     cfg_file = Outputs.file(description='Output config file.', path='config.json')
 
 
+@dataclass
+class CumulativeRadiationVisMetadata(Function):
+    """Create a visualization metadata file for cumulative radiation."""
+
+    @command
+    def create_cumulative_radiation_vis_data(self):
+        return 'honeybee-radiance post-process cumulative-radiation-vis-metadata ' \
+            '--output-file vis_metadata.json'
+
+    # outputs
+    cfg_file = Outputs.file(
+        description='Output visualization metadata file.', path='vis_metadata.json')
+
+
+@dataclass
+class AverageIrradianceVisMetadata(Function):
+    """Create a visualization metadata file for average irradiance."""
+
+    @command
+    def create_average_irradiance_vis_data(self):
+        return 'honeybee-radiance post-process average-irradiance-vis-metadata ' \
+            '--output-file vis_metadata.json'
+
+    # outputs
+    cfg_file = Outputs.file(
+        description='Output visualization metadata file.', path='vis_metadata.json')
+
+
 @dataclass
 class CumulativeRadiationConfig(Function):
     """Create a config file for cumulative-radiation results."""
@@ -614,6 +642,20 @@ def create_point_in_time_config(self):
     cfg_file = Outputs.file(description='Output config file.', path='config.json')
 
 
+@dataclass
+class SkyViewVisMetadata(Function):
+    """Create a visualization metadata file for sky view."""
+
+    @command
+    def create_sky_view_vis_data(self):
+        return 'honeybee-radiance post-process sky-view-vis-metadata ' \
+            '--output-file vis_metadata.json'
+
+    # outputs
+    cfg_file = Outputs.file(
+        description='Output visualization metadata file.', path='vis_metadata.json')
+
+
 @dataclass
 class SkyViewConfig(Function):
     """Create a config file for sky-view results."""

tests/post_process_test.py

@@ -4,7 +4,9 @@
 from pollination.honeybee_radiance.post_process import ConvertToBinary, SumRow, \
     AnnualIrradianceMetrics, AnnualDaylightMetrics, LeedIlluminanceCredits, \
     SolarTrackingSynthesis, DaylightFactorConfig, DaylightFactorVisMetadata, \
-    ImagelessAnnualGlareVisMetadata, PointInTimeVisMetadata
+    ImagelessAnnualGlareVisMetadata, PointInTimeVisMetadata, \
+    CumulativeRadiationVisMetadata, AverageIrradianceVisMetadata, \
+    SkyViewVisMetadata
 
 from queenbee.plugin.function import Function
 
@@ -99,3 +101,51 @@ def test_point_in_time_vis_metadata():
     assert output_file.is_file()
 
     nukedir(folder)
+
+def test_cumulative_radiation_vis_metadata():
+    function = CumulativeRadiationVisMetadata()
+    qb_function = function.queenbee
+    assert qb_function.name == 'cumulative-radiation-vis-metadata'
+    assert isinstance(qb_function, Function)
+
+    inputs = {} # inputs is empty for this function
+    folder = Path('./tests/assets/temp')
+    if not folder.exists():
+        folder.mkdir(parents=True)
+    output_file = folder.joinpath('vis_metadata.json')
+    function._try(inputs, folder=folder)
+    assert output_file.is_file()
+
+    nukedir(folder)
+
+def test_average_irradiance_vis_metadata():
+    function = AverageIrradianceVisMetadata()
+    qb_function = function.queenbee
+    assert qb_function.name == 'average-irradiance-vis-metadata'
+    assert isinstance(qb_function, Function)
+
+    inputs = {} # inputs is empty for this function
+    folder = Path('./tests/assets/temp')
+    if not folder.exists():
+        folder.mkdir(parents=True)
+    output_file = folder.joinpath('vis_metadata.json')
+    function._try(inputs, folder=folder)
+    assert output_file.is_file()
+
+    nukedir(folder)
+
+def test_sky_view_vis_metadata():
+    function = SkyViewVisMetadata()
+    qb_function = function.queenbee
+    assert qb_function.name == 'sky-view-vis-metadata'
+    assert isinstance(qb_function, Function)
+
+    inputs = {} # inputs is empty for this function
+    folder = Path('./tests/assets/temp')
+    if not folder.exists():
+        folder.mkdir(parents=True)
+    output_file = folder.joinpath('vis_metadata.json')
+    function._try(inputs, folder=folder)
+    assert output_file.is_file()
+
+    nukedir(folder)