implementation of generate_array_from_positions

skais/ais/ais_trajectory.py

@@ -223,3 +223,21 @@ class AISTrajectory(AISPoints):
                 current_label = row[label_column]
                 result.append((row['ts_sec'], current_label))
         return result
+
+    def generate_array_from_positions(self, height=256, width=256, link=True, bounding_box='fit', features=None):
+        nb_channels = 1
+        if features is not None:
+            nb_channels = len(features)
+        data = np.zeros((height, width, nb_channels), dtype=np.uint8)
+        if link:
+            raise ValueError("feature not implemented")
+        if bounding_box != 'fit':
+            raise ValueError("feature not implemented")
+        positions = self.df[['longitude', 'latitude']].to_numpy()
+        range_longitude = (min(positions[0, :]), max(positions[0, :]))
+        range_latitude = (min(positions[1, :]), max(positions[1, :]))
+        for longitude, latitude in positions:
+            x_coord = width * (longitude - range_longitude[0]) / (range_longitude[1] - range_longitude[0])
+            y_coord = height * (longitude - range_latitude[0]) / (range_latitude[1] - range_latitude[0])
+            data[x_coord, y_coord, :] = 1
+        return data