Resolve "Image creation bugs with 0 size windows"

Closes #21 (closed)

skais/ais/ais_trajectory.py

@@ -7,6 +7,7 @@ from scipy.interpolate import interp1d
 
 from skais.utils.geography import great_circle, position_from_distance, get_coord
 from skais.ais.ais_points import AISPoints
+from skais.utils.geometry import bresenham
 
 
 @jit(nopython=True)
@@ -224,7 +225,8 @@ class AISTrajectory(AISPoints):
                 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, node_size=0):
+    def generate_array_from_positions(self, height=256, width=256, link=True, bounding_box='fit', features=None,
+                                      node_size=0):
         nb_channels = 1
         if features is not None:
             nb_channels = len(features)
@@ -248,5 +250,12 @@ class AISTrajectory(AISPoints):
                     data[x, y] = [1]
 
         if link:
-            raise ValueError("feature not implemented")
+            lon, lat = positions[0, 0], positions[0, 1]
+            for longitude, latitude in positions[1:]:
+                x_prv, y_prev = get_coord(lat, lon, height, width, min_lat, max_lat, min_lon, max_lon)
+                x_nxt, y_nxt = get_coord(latitude, longitude, height, width, min_lat, max_lat, min_lon, max_lon)
+
+                lon, lat = longitude, latitude
+                for x, y in bresenham(x_prv, y_prev, x_nxt, y_nxt):
+                    data[x, y] = [1]
         return data