added plot labels

petercorke · petercorke · commit 5dfb1cfcf2c9 · 2021-09-13T12:05:18.000+10:00
diff --git a/roboticstoolbox/tools/trajectory.py b/roboticstoolbox/tools/trajectory.py
@@ -172,22 +172,22 @@ def plot(self, block=False, plotargs=None, textargs=None):
             # accel phase
             tf = self.t[-1]
             k = self.t <= self.tblend
-            ax.plot(self.t[k], self.s[k], color="red", **plotopts)
+            ax.plot(self.t[k], self.s[k], color="red", label="acceleration", **plotopts)
 
             # coast phase
             k = (self.t > self.tblend) & (self.t <= (tf - self.tblend))
             ax.plot(self.t[k], self.s[k], color="green", **plotopts)
             k = np.where(k)[0][0]
             ax.plot(
-                self.t[k - 1 : k + 1], self.s[k - 1 : k + 1], color="green", **plotopts
+                self.t[k - 1 : k + 1], self.s[k - 1 : k + 1], color="green", label="linear", **plotopts
             )
 
             # decel phase
             k = self.t > (tf - self.tblend)
             ax.plot(self.t[k], self.s[k], color="blue", **plotopts)
             k = np.where(k)[0][0]
             ax.plot(
-                self.t[k - 1 : k + 1], self.s[k - 1 : k + 1], color="blue", **plotopts
+                self.t[k - 1 : k + 1], self.s[k - 1 : k + 1], color="blue", label="deceleration", **plotopts
             )
 
             ax.grid(True)
