Add argument `color_initial`

To function `plot_single_traj`.

Trajectories/Analysis/plot_traj.py

@@ -33,12 +33,25 @@ def get_extr_coord(traj, e_overestim, SHPC, orientation):
 
 
 def plot_single_traj(
-    traj, e_overestim, SHPC, orientation, ax, src_crs, annotate_flag, color
+    traj,
+    e_overestim,
+    SHPC,
+    orientation,
+    ax,
+    src_crs,
+    annotate_flag,
+    color,
+    color_initial,
 ):
     x, y = get_extr_coord(traj, e_overestim, SHPC, orientation)
     ax.plot(x, y, color, linewidth=0.5, transform=src_crs)
     ax.plot(
-        x[0], y[0], marker="s", markersize=2, color="green", transform=src_crs
+        x[0],
+        y[0],
+        marker="s",
+        markersize=2,
+        color=color_initial,
+        transform=src_crs,
     )
 
     if annotate_flag:
@@ -130,6 +143,7 @@ if __name__ == "__main__":
                     src_crs,
                     args.annotate,
                     color="red",
+                    color_initial="green",
                 )
                 break
         else:
@@ -144,6 +158,7 @@ if __name__ == "__main__":
             src_crs,
             args.annotate,
             color="red",
+            color_initial="green",
         )
     else:
         if args.min_duration == 1:
@@ -157,6 +172,7 @@ if __name__ == "__main__":
                     src_crs,
                     args.annotate,
                     color="red",
+                    color_initial="green",
                 )
         else:
             # args.min_duration > 1
@@ -177,6 +193,7 @@ if __name__ == "__main__":
                         src_crs,
                         args.annotate,
                         color="red",
+                        color_initial="green",
                     )
 
             print(