fixed pattern2cloud

lyceanem/base_classes.py

@@ -444,6 +444,7 @@ def __init__(
             self.arbitary_pattern_format = arbitary_pattern_format
             antenna_pattern_import_implemented = False
             assert antenna_pattern_import_implemented
+            ## needs implementing
             self.import_pattern(file_location)
             self.field_radius = 1.0
         else:

lyceanem/electromagnetics/sources.py

@@ -68,6 +68,7 @@ def antenna_pattern_source(radius, import_antenna=False, antenna_file=None):
     if import_antenna:
         # import antenna pattern
         pattern = antenna_pattern()
+        ## import the pattern not implemented
         pattern.import_pattern(antenna_file)
         pattern.field_radius = radius
     else:

lyceanem/raycasting/rayfunctions.py

@@ -293,7 +293,20 @@ def integratedRaycaster(ray_index, scattering_points, environment_local):
 
     return final_index
 
-
+# @njit
+def patterntocloud(pattern_data, shell_coords, maxarea):
+    # takes the pattern_data and shell_coordinates, and creates an open3d point cloud based upon the data.
+    point_cloud = points2pointcloud(shell_coords)
+    points, elements = np.shape(pattern_data)
+    # normdata
+    viridis = cm.get_cmap("viridis", 40)
+    visible_elements = np.sum(pattern_data / maxarea, axis=1)
+    np_colors = viridis(visible_elements)
+    point_cloud.point_data["red"] = np_colors[:, 0]
+    point_cloud.point_data["green"] = np_colors[:, 1]
+    point_cloud.point_data["blue"] = np_colors[:, 2]
+
+    return point_cloud
 def visiblespace(
     source_coords,
     source_normals,
@@ -468,6 +481,7 @@ def patterncreator(az_range, elev_range, source_coords, pointingindex, hit_index
     return visible_patterns
 
 
+
 def quickpatterncreator(
     az_range, elev_range, source_coords, angles, vertex_area, hit_index
 ):