copick-utils 0.5.0__py3-none-any.whl → 0.6.0__py3-none-any.whl

copick_utils/io/readers.py

@@ -0,0 +1,132 @@
+import numpy as np
+
+def tomogram(run, 
+             voxel_size: float = 10, 
+             algorithm: str = 'wbp',
+             raise_error: bool = False):
+    
+    voxel_spacing_obj = run.get_voxel_spacing(voxel_size)
+
+    if voxel_spacing_obj is None:
+        # Query Avaiable Voxel Spacings
+        availableVoxelSpacings = [tomo.voxel_size for tomo in run.voxel_spacings]
+
+        # Report to the user which voxel spacings they can use 
+        message = (f"[Warning] No tomogram found for {run.name} with voxel size {voxel_size} and tomogram type {algorithm}"
+                   f"Available spacings are: {', '.join(map(str, availableVoxelSpacings))}" ) 
+        if raise_error:
+            raise ValueError(message)
+        else:
+            print(message)
+            return None
+    
+    tomogram = voxel_spacing_obj.get_tomogram(algorithm)
+    if tomogram is None:
+        # Get available algorithms
+        availableAlgorithms = [tomo.tomo_type for tomo in run.get_voxel_spacing(voxel_size).tomograms]
+        
+        # Report to the user which algorithms are available
+        message = (f"[Warning] No tomogram found for {run.name} with voxel size {voxel_size} and tomogram type {algorithm}"
+                   f"Available algorithms are: {', '.join(availableAlgorithms)}")
+        if raise_error:
+            raise ValueError(message)
+        else:
+            print(message)
+            return None
+
+    return tomogram.numpy()
+
+def segmentation(run, 
+                 voxel_spacing: float,
+                 segmentation_name: str, 
+                 session_id=None,
+                 user_id=None,
+                 raise_error = False):
+
+    seg = run.get_segmentations(name=segmentation_name, 
+                                session_id = session_id,
+                                user_id = user_id,
+                                voxel_size = voxel_spacing)
+
+    # No Segmentations Are Available, Result in Error
+    if len(seg) == 0:
+        # Get all available segmentations with their metadata
+        available_segs = run.get_segmentations(voxel_size=voxel_spacing)
+        seg_info = [(s.name, s.user_id, s.session_id) for s in available_segs]
+        
+        # Format the information for display
+        seg_details = [f"(name: {name}, user_id: {uid}, session_id: {sid})" 
+                      for name, uid, sid in seg_info]
+        
+        message = ( f'\nNo segmentation found matching:\n'
+                    f'  name: {segmentation_name}, user_id: {user_id}, session_id: {session_id}\n'
+                    f'Available segmentations in {run.name} are:\n  ' + 
+                    '\n  '.join(seg_details) )
+        if raise_error:
+            raise ValueError(message)
+        else:
+            print(message)
+            return None
+
+    # No Segmentations Are Available, Result in Error
+    if len(seg) > 1:
+        print(f'[Warning] More Than 1 Segmentation is Available for the Query Information. '
+              f'Available Segmentations are: {seg} '
+              f'Defaulting to Loading: {seg[0]}\n')
+    seg = seg[0]
+
+    return seg.numpy()
+
+def coordinates(run,                     # CoPick run object containing the segmentation data
+                name: str,               # Name of the object or protein for which coordinates are being extracted
+                user_id: str,            # Identifier of the user that generated the picks
+                session_id: str = None,  # Identifier of the session that generated the picks
+                voxel_size: float = 10,  # Voxel size of the tomogram, used for scaling the coordinates
+                raise_error: bool = False):
+                           
+    # Retrieve the pick points associated with the specified object and user ID
+    picks = run.get_picks(object_name=name, user_id=user_id, session_id=session_id)
+
+    if len(picks) == 0:
+        # Get all available segmentations with their metadata
+
+        available_picks = run.get_picks()
+        picks_info = [(s.pickable_object_name, s.user_id, s.session_id) for s in available_picks]
+        
+        # Format the information for display
+        picks_details = [f"(name: {name}, user_id: {uid}, session_id: {sid})" 
+                      for name, uid, sid in picks_info]
+        
+        message = ( f'\nNo picks found matching:\n'
+                    f'  name: {name}, user_id: {user_id}, session_id: {session_id}\n'
+                    f'Available picks are:\n  '  
+                    + '\n  '.join(picks_details) )
+        if raise_error:
+            raise ValueError(message)
+        else:
+            print(message)
+            return None
+    elif len(picks) > 1:
+        # Format pick information for display
+        picks_info = [(p.pickable_object_name, p.user_id, p.session_id) for p in picks]
+        picks_details = [f"(name: {name}, user_id: {uid}, session_id: {sid})" 
+                        for name, uid, sid in picks_info]
+
+        print(f'[Warning] More than 1 pick is available for the query information.'
+              f'\nAvailable picks are:\n  ' + 
+              '\n  '.join(picks_details) +
+              f'\nDefaulting to loading:\n {picks[0]}\n')
+    points = picks[0].points
+
+    # Initialize an array to store the coordinates
+    nPoints = len(picks[0].points)                      # Number of points retrieved
+    coordinates = np.zeros([len(picks[0].points), 3])   # Create an empty array to hold the (z, y, x) coordinates
+
+    # Iterate over all points and convert their locations to coordinates in voxel space
+    for ii in range(nPoints):
+        coordinates[ii,] = [points[ii].location.z / voxel_size,   # Scale z-coordinate by voxel size
+                            points[ii].location.y / voxel_size,   # Scale y-coordinate by voxel size
+                            points[ii].location.x / voxel_size]   # Scale x-coordinate by voxel size
+
+    # Return the array of coordinates
+    return coordinates

{copick_utils-0.5.0.dist-info → copick_utils-0.6.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: copick-utils
-Version: 0.5.0
+Version: 0.6.0
 Summary: Utilities for copick
 License: MIT
 Author: Kyle Harrington
@@ -15,7 +15,6 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
-Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy
 Requires-Dist: copick (>=0.8.0)

{copick_utils-0.5.0.dist-info → copick_utils-0.6.0.dist-info}/RECORD

@@ -2,14 +2,14 @@ copick_utils/__about__.py,sha256=7D13PJEcpdhEa6RrlKLt7IEkoVVVGuzqHJb5MQxgLiI,135
 copick_utils/__init__.py,sha256=v-RIkEuGuAXivakLMrneraDQd7cWN7zsdGLmjwLtDDw,113
 copick_utils/features/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 copick_utils/features/skimage.py,sha256=t38jpu-ntC7Zw--1qSxWhGIg9e02RLhVBZEySIa5dQs,4036
+copick_utils/io/readers.py,sha256=ajqe64e5tp67iZoqSV6AsjpGX25Rdz_q0tX8qXVVp7g,5902
+copick_utils/io/writers.py,sha256=KhdZUUZrZR02bzS5SWM9q-u4y2s1qxcySwN3MN5j5YA,3055
 copick_utils/pickers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 copick_utils/pickers/grid_picker.py,sha256=NhFbWxMQREb0fLKTho4602yzH7zE6DCkJY94dgJ-gIQ,2353
 copick_utils/segmentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 copick_utils/segmentation/picks_from_segmentation.py,sha256=Ne_RYfnEFaV_qwNw4uOwPfLLgGxHMEQ2uecp9X2_EVc,2951
 copick_utils/segmentation/segmentation_from_picks.py,sha256=3oIlFWRAR904j5IpC5Fo8Y4gP9iL6qUiwybRgJvhzmc,6820
-copick_utils/writers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-copick_utils/writers/write.py,sha256=KhdZUUZrZR02bzS5SWM9q-u4y2s1qxcySwN3MN5j5YA,3055
-copick_utils-0.5.0.dist-info/LICENSE.txt,sha256=3UHKsYd99Gh_qf1a9s8G5sdKqafgbGs5WIMoeX0OcdY,1105
-copick_utils-0.5.0.dist-info/METADATA,sha256=zd_eTHyOWidHZvieUbSzRJMTX-Lft3lhZE5uJz3qajM,2792
-copick_utils-0.5.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-copick_utils-0.5.0.dist-info/RECORD,,
+copick_utils-0.6.0.dist-info/LICENSE.txt,sha256=3UHKsYd99Gh_qf1a9s8G5sdKqafgbGs5WIMoeX0OcdY,1105
+copick_utils-0.6.0.dist-info/METADATA,sha256=TwbWK0CmHk8pB7tI401f3khfIG8mMEtTENOg2b1brOY,2742
+copick_utils-0.6.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+copick_utils-0.6.0.dist-info/RECORD,,