change to warn for multi channel multi plane

src/roiextractors/extractors/suite2p/suite2psegmentationextractor.py

@@ -30,7 +30,8 @@ class Suite2pSegmentationExtractor(SegmentationExtractor):
     @classmethod
     def get_available_channels(cls, folder_path: PathType):
         """Get the available channel names from the folder paths produced by Suite2p.
-        outputs
+
+        Parameters
         ----------
         file_path : PathType
             Path to Suite2p output path.
@@ -43,7 +44,7 @@ def get_available_channels(cls, folder_path: PathType):
         plane_names = cls.get_available_planes(folder_path=folder_path)
 
         channel_names = ["chan1"]
-        file_path = folder_path / plane_names[0] / "F_chan2.npy"
+        file_path = Path(folder_path) / plane_names[0] / "F_chan2.npy"
 
         if os.path.isfile(file_path):
             channel_names.append("chan2")
@@ -53,7 +54,8 @@ def get_available_channels(cls, folder_path: PathType):
     @classmethod
     def get_available_planes(cls, folder_path: PathType):
         """Get the available plane names from the folder produced by Suite2p.
-        outputs
+
+        Parameters
         ----------
         file_path : PathType
             Path to Suite2p output path.
@@ -109,22 +111,35 @@ def __init__(
         channel_names = self.get_available_channels(folder_path=folder_path)
         if channel_name is None:
             if len(channel_names) > 1:
-                raise ValueError(
+                # For backward compatibility maybe it is better to warn first
+                warn(
                     "More than one channel is detected! Please specify which channel you wish to load with the `channel_name` argument. "
-                    "To see what streams are available, call `Suite2pSegmentationExtractor.get_channel_names(folder_path=...)`."
+                    "To see what channels are available, call `Suite2pSegmentationExtractor.get_available_channels(folder_path=...)`.",
+                    UserWarning,
                 )
-            channel_name = channel_names["channel_streams"][0]
+            channel_name = channel_names[0]
 
-        if channel_name not in channel_names["channel_streams"]:
+        if channel_name not in channel_names:
             raise ValueError(
-                f"The selected channel '{channel_name}' is not a valid stream name. To see what channels are available, "
-                f"call `Suite2pSegmentationExtractor.get_channel_names(folder_path=...)`."
+                f"The selected channel '{channel_name}' is not a valid channel name. To see what channels are available, "
+                f"call `Suite2pSegmentationExtractor.get_available_channels(folder_path=...)`."
             )
 
         plane_names = self.get_available_planes(folder_path=folder_path)
-        if plane_name is not None and plane_name not in plane_names:
+        if plane_name is None:
+            if len(plane_names) > 1:
+                # For backward compatibility maybe it is better to warn first
+                warn(
+                    "More than one plane is detected! Please specify which plane you wish to load with the `plane_name` argument. "
+                    "To see what planes are available, call `Suite2pSegmentationExtractor.get_available_planes(folder_path=...)`.",
+                    UserWarning,
+                )
+            plane_name = plane_names[0]
+
+        if plane_name not in plane_names:
             raise ValueError(
-                f"The selected plane '{plane_name}' is not in the available plane_streams '{plane_names['plane_streams']}'!"
+                f"The selected plane '{plane_name}' is not a valid plane name. To see what planes are available, "
+                f"call `Suite2pSegmentationExtractor.get_available_planes(folder_path=...)`."
             )
         self.plane_name = plane_name
 
@@ -144,11 +159,11 @@ def __init__(
         neuropil_traces_file_name = "Fneu.npy" if channel_name == "chan1" else "Fneu_chan2.npy"
         self._roi_response_raw = self._load_npy(file_name=fluorescence_traces_file_name, mmap_mode="r", transpose=True)
         self._roi_response_neuropil = self._load_npy(file_name=neuropil_traces_file_name, mmap_mode="r", transpose=True)
-        self._roi_response_deconvolved = self._load_npy(file_name="spks.npy", mmap_mode="r", transpose=True)
+        self._roi_response_deconvolved = self._load_npy(file_name="spks.npy", mmap_mode="r", transpose=True) if channel_name == "chan1" else None
 
         self.iscell = self._load_npy("iscell.npy", mmap_mode="r")
 
-        channel_name = "OpticalChannel" if len(channel_names["channel_streams"]) == 1 else channel_name.capitalize()
+        channel_name = "OpticalChannel" if len(channel_names) == 1 else channel_name.capitalize()
         self._channel_names = [channel_name]
 
         self._image_correlation = self._correlation_image_read()
@@ -171,8 +186,7 @@ def _load_npy(self, file_name: str, mmap_mode=None, transpose: bool = False):
         -------
             The loaded .npy file.
         """
-        plane_name = self.plane_name
-        file_path = self.folder_path / plane_name / file_name
+        file_path = self.folder_path / self.plane_name / file_name
         if not file_path.exists():
             return