voxcity 1.0.13__py3-none-any.whl → 1.0.15__py3-none-any.whl

voxcity/simulator_gpu/solar/__init__.py

@@ -29,7 +29,17 @@ from .solar import (
     discretize_sky_directions
 )
 
-from .raytracing import RayTracer
+from .raytracing import (
+    RayTracer,
+    ray_aabb_intersect,
+    ray_voxel_first_hit,
+    ray_canopy_absorption,
+    ray_voxel_transmissivity,
+    ray_trace_to_target,
+    ray_point_to_point_transmissivity,
+    sample_hemisphere_direction,
+    hemisphere_solid_angle,
+)
 
 from .svf import SVFCalculator
 
@@ -70,6 +80,15 @@ from .sky import (
     TREGENZA_BAND_BOUNDARIES,
 )
 
+# Computation mask utilities
+from .mask import (
+    create_computation_mask,
+    draw_computation_mask,
+    get_mask_from_drawing,
+    visualize_computation_mask,
+    get_mask_info,
+)
+
 # VoxCity integration
 from .integration import (
     load_voxcity,
@@ -90,6 +109,11 @@ from .integration import (
     get_cumulative_building_solar_irradiance,
     get_global_solar_irradiance_using_epw,
     get_building_global_solar_irradiance_using_epw,
+    # Volumetric solar irradiance functions
+    get_volumetric_solar_irradiance_map,
+    get_cumulative_volumetric_solar_irradiance,
+    get_volumetric_solar_irradiance_using_epw,
+    clear_volumetric_flux_cache,
     save_irradiance_mesh,
     load_irradiance_mesh,
     # Temporal utilities
@@ -98,6 +122,7 @@ from .integration import (
     clear_radiation_model_cache,
     clear_building_radiation_model_cache,
     clear_all_radiation_caches,
+    clear_all_caches,
 )
 
 __version__ = "0.1.0"
@@ -115,6 +140,14 @@ __all__ = [
     'discretize_sky_directions',
     # Ray tracing
     'RayTracer',
+    'ray_aabb_intersect',
+    'ray_voxel_first_hit',
+    'ray_canopy_absorption',
+    'ray_voxel_transmissivity',
+    'ray_trace_to_target',
+    'ray_point_to_point_transmissivity',
+    'sample_hemisphere_direction',
+    'hemisphere_solid_angle',
     # SVF
     'SVFCalculator',
     # CSF
@@ -148,6 +181,12 @@ __all__ = [
     'visualize_sky_patches',
     'TREGENZA_BANDS',
     'TREGENZA_BAND_BOUNDARIES',
+    # Computation mask utilities
+    'create_computation_mask',
+    'draw_computation_mask',
+    'get_mask_from_drawing',
+    'visualize_computation_mask',
+    'get_mask_info',
     # VoxCity integration
     'load_voxcity',
     'convert_voxcity_to_domain',
@@ -167,6 +206,11 @@ __all__ = [
     'get_cumulative_building_solar_irradiance',
     'get_global_solar_irradiance_using_epw',
     'get_building_global_solar_irradiance_using_epw',
+    # Volumetric solar irradiance functions
+    'get_volumetric_solar_irradiance_map',
+    'get_cumulative_volumetric_solar_irradiance',
+    'get_volumetric_solar_irradiance_using_epw',
+    'clear_volumetric_flux_cache',
     'save_irradiance_mesh',
     'load_irradiance_mesh',
     # Cache management

voxcity/simulator_gpu/solar/domain.py

@@ -238,6 +238,63 @@ class Domain:
         self._set_lad_kernel(lad_array)
         self._update_plant_top()
     
+    def set_from_voxel_data(self, voxel_data: np.ndarray, tree_code: int = -2, solid_codes: Optional[list] = None):
+        """
+        Set domain from a 3D voxel data array.
+        
+        Args:
+            voxel_data: 3D numpy array with voxel class codes
+            tree_code: Class code for trees (default -2)
+            solid_codes: List of codes that are solid (default: all non-zero except tree_code)
+        """
+        if solid_codes is None:
+            # All non-zero codes except tree are solid
+            solid_codes = []
+        
+        self._set_from_voxel_data_kernel(voxel_data, tree_code)
+    
+    @ti.kernel
+    def _set_from_voxel_data_kernel(self, voxel_data: ti.types.ndarray(), tree_code: ti.i32):
+        for i, j, k in ti.ndrange(self.nx, self.ny, self.nz):
+            val = voxel_data[i, j, k]
+            if val == tree_code:
+                self.is_tree[i, j, k] = 1
+                self.is_solid[i, j, k] = 0
+            elif val != 0:
+                self.is_solid[i, j, k] = 1
+                self.is_tree[i, j, k] = 0
+            else:
+                self.is_solid[i, j, k] = 0
+                self.is_tree[i, j, k] = 0
+    
+    def add_tree_box(
+        self,
+        x_range: Tuple[int, int],
+        y_range: Tuple[int, int],
+        z_range: Tuple[int, int],
+        lad_value: float = 1.0
+    ):
+        """
+        Add a box-shaped tree canopy region to the domain.
+        
+        This is a simpler alternative to add_tree() for rectangular tree regions.
+        
+        Args:
+            x_range, y_range, z_range: Grid index ranges (start, end)
+            lad_value: Leaf Area Density value (m^2/m^3)
+        """
+        self._add_tree_box_kernel(x_range[0], x_range[1], y_range[0], y_range[1], 
+                                   z_range[0], z_range[1], lad_value)
+        self._update_plant_top()
+    
+    @ti.kernel
+    def _add_tree_box_kernel(self, i_min: ti.i32, i_max: ti.i32, j_min: ti.i32, 
+                              j_max: ti.i32, k_min: ti.i32, k_max: ti.i32, lad: ti.f32):
+        for i, j, k in ti.ndrange((i_min, i_max), (j_min, j_max), (k_min, k_max)):
+            if 0 <= i < self.nx and 0 <= j < self.ny and 0 <= k < self.nz:
+                self.is_tree[i, j, k] = 1
+                self.lad[i, j, k] = lad
+
     @ti.kernel
     def _set_lad_kernel(self, lad_array: ti.types.ndarray()):
         for i, j, k in self.lad: