Skip to content
This repository was archived by the owner on Feb 26, 2025. It is now read-only.

Commit a99ba5e

Browse files
authored
Add arg detection for overloading (#26)
1 parent 1a9e689 commit a99ba5e

File tree

1 file changed

+17
-6
lines changed

1 file changed

+17
-6
lines changed

atlas_placement_hints/compute_placement_hints.py

Lines changed: 17 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -70,12 +70,23 @@ def compute_placement_hints(
7070
least one distance-related problem.
7171
See distances.distance_to_meshes.report_distance_problems doc.
7272
"""
73-
distances_info = atlas.compute_distances_to_layer_boundaries(
74-
direction_vectors,
75-
flip_direction_vectors=flip_direction_vectors,
76-
has_hemispheres=has_hemispheres,
77-
thalamus_meshes_dir=thalamus_meshes_dir,
78-
)
73+
74+
if thalamus_meshes_dir == "":
75+
# Use either the AbstractLayeredAtlas or VoxelBasedLayeredAtlas version if
76+
# no meshes are provided
77+
distances_info = atlas.compute_distances_to_layer_boundaries(
78+
direction_vectors,
79+
flip_direction_vectors=flip_direction_vectors,
80+
has_hemispheres=has_hemispheres,
81+
)
82+
else:
83+
# Use the MeshBasedLayeredAtlas version if meshes are provided
84+
distances_info = atlas.compute_distances_to_layer_boundaries(
85+
direction_vectors,
86+
flip_direction_vectors=flip_direction_vectors,
87+
has_hemispheres=has_hemispheres,
88+
thalamus_meshes_dir=thalamus_meshes_dir,
89+
)
7990

8091
distances_to_meshes = distances_info["distances_to_layer_boundaries"]
8192
tolerance = 2.0 * atlas.region.voxel_dimensions[0]

0 commit comments

Comments
 (0)