From 6bcb6ef52f5564f3d5b5c429e22ed08397848416 Mon Sep 17 00:00:00 2001
From: Raphael Sturgis <raphael.sturgis@lis-lab.fr>
Date: Thu, 14 Apr 2022 10:10:25 +0200
Subject: [PATCH] refactor

---
 skais/ais/ais_trajectory.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/skais/ais/ais_trajectory.py b/skais/ais/ais_trajectory.py
index 90ba3e8..8a663cd 100644
--- a/skais/ais/ais_trajectory.py
+++ b/skais/ais/ais_trajectory.py
@@ -165,8 +165,8 @@ def generate_links(data, positions, height, width, lower_lat, upper_lat, lower_l
                                  upper_lon)
         lon, lat = longitude, latitude
         for x, y in bresenham(x_prv, y_prev, x_nxt, y_nxt):
-            for i, v in enumerate(current_value):
-                data[x, y, i] = v
+            for i in range(len(current_value)):
+                data[x, y, i] = current_value[i]
         current_value = nxt_value
 
 
@@ -374,6 +374,7 @@ class AISTrajectory(AISPoints):
 
         if features_vectors is not None:
             nb_channels = len(features_vectors.T)
+            bounds = np.array(bounds)
         else:
             nb_channels = 1
         data = np.zeros((height, width, nb_channels), dtype=np.float)
@@ -384,10 +385,9 @@ class AISTrajectory(AISPoints):
 
             if link:
                 generate_links(data, positions, height, width, lower_lat, upper_lat, lower_lon, upper_lon,
-                               np.ones(len(positions)))
+                               np.ones((len(positions), 1)))
 
         else:
-            print(features_vectors[0] / 10)
             generate_points_with_features(data, positions, features_vectors, np.array(bounds), node_size, height, width,
                                           lower_lat, upper_lat, lower_lon, upper_lon, )
             if link:
-- 
GitLab