add print

Author: JosephBrunet

src/cardiotensor/visualization/fury_plotting_streamlines.py

@@ -185,14 +185,15 @@ def show_streamlines(
         if not streamlines_xyz:
             raise ValueError("❌ No streamlines intersect the crop box.")
 
+    print("Cropping applied")
 
     # --- Downsample and filter
+    print(f"Downsampling points by factor {downsample_factor}")
     downsampled_streamlines = []
     downsampled_colors = []
-    for sl in streamlines_xyz:
+    for sl, cl in zip(streamlines_xyz, color_values):
         ds_sl = downsample_streamline(sl, downsample_factor)
-        ds_cl = downsample_streamline(color_slice, downsample_factor)
-
+        ds_cl = downsample_streamline(cl, downsample_factor)
         if filter_min_len is None or len(ds_sl) >= filter_min_len:
             downsampled_streamlines.append(ds_sl)
             downsampled_colors.append(ds_cl)
@@ -205,6 +206,7 @@ def show_streamlines(
 
     # --- Subsample
     if subsample_factor > 1:
+        print(f"Subsample number of streamlines by factor {downsample_factor}")
         total = len(streamlines_xyz)
         selected_idx = sorted(random.sample(range(total), total // subsample_factor))
         streamlines_xyz = [streamlines_xyz[i] for i in selected_idx]

src/cardiotensor/visualization/streamlines.py

@@ -76,18 +76,21 @@ def visualize_streamlines(
     if not streamlines_file.exists():
         raise FileNotFoundError(f"❌ Streamlines file not found: {streamlines_file}")
 
+    print(f"Loading streamlines: {streamlines_file}")
     data = np.load(streamlines_file, allow_pickle=True)
     raw_streamlines = data.get("streamlines")
     if raw_streamlines is None:
         raise ValueError("'streamlines' array missing in .npz.")
 
     # Convert (z, y, x) to (x, y, z)
+    print("Convert (z, y, x) to (x, y, z)")
     streamlines_xyz = [
         np.array([(pt[2], pt[1], pt[0]) for pt in sl], dtype=np.float32)
         for sl in raw_streamlines.tolist()
     ]
 
     # Compute coloring
+    print("Compute coloring")
     if color_by == "elevation":
         color_values = compute_elevation_angles(streamlines_xyz)
     else:  # color_by == "ha"
@@ -101,6 +104,7 @@ def visualize_streamlines(
     # Default to helix_angle_cmap if no colormap is provided
     if colormap is None:
         colormap = helix_angle_cmap
+    
 
     # Render streamlines
     show_streamlines(