small bugs and nb

Author: jbuces

dl1_data_handler/image_mapper.py

@@ -704,7 +704,7 @@ def __init__(
             self.num_patches = len(self.trigger_patches)
             self.patch_size = len(self.index_map[0])
 
-        elif geometry.name == "LSTCam":
+        else:
             neighbor_matrix = geometry.neighbor_matrix
             num_pixels = neighbor_matrix.shape[0]
             neighbor_lists = []
@@ -716,6 +716,7 @@ def __init__(
             self.cam_neighbor_array = np.full((num_pixels, 7), -1, dtype=int)
             for i, neighbors in enumerate(neighbor_lists):
                 self.cam_neighbor_array[i, :len(neighbors)] = neighbors
+            print("Computed neighbor array is", self.cam_neighbor_array)
 
     def get_reordered_patch(self, raw_vector, patch_index):
         # Retrieve the patch needed

dl1_data_handler/reader.py

@@ -1222,14 +1222,15 @@ def __init__(
 
         # Set the input shape based on the selected mode
         if self.mode == "mono":
-            if self.image_mappers[self.cam_name].cam_neighbor_array is not None and self.image_mappers[self.cam_name].index_matrix is None:
+            if self.image_mappers[self.cam_name].cam_neighbor_array is None:
                 self.input_shape = (
                     self.image_mappers[self.cam_name].image_shape,
                     self.image_mappers[self.cam_name].image_shape,
                     len(self.channels),
                 )
             else:
                 self.input_shape = (self.image_mappers[self.cam_name].geometry.n_pixels, len(self.channels))
+                
         elif self.mode == "stereo":
             self.input_shape = {}
             for tel_type in self.selected_telescopes:
@@ -1512,11 +1513,14 @@ def __init__(
 
         # Set the input shape based on the selected mode
         if self.mode == "mono":
-            self.input_shape = (
-                self.image_mappers[self.cam_name].image_shape,
-                self.image_mappers[self.cam_name].image_shape,
-                self.sequence_length,
-            )
+            if self.image_mappers[self.cam_name].cam_neighbor_array is None:
+                self.input_shape = (
+                    self.image_mappers[self.cam_name].image_shape,
+                    self.image_mappers[self.cam_name].image_shape,
+                    self.sequence_length,
+                )
+            else:
+                self.input_shape = (self.image_mappers[self.cam_name].geometry.n_pixels, self.sequence_length)
         elif self.mode == "stereo":
             self.input_shape = {}
             for tel_type in self.selected_telescopes: