Implemented orbital speed change

poliastro · Mar 31, 2021 · 4705ace · 4705ace
1 parent c684748
commit 4705ace
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/src/poliastro/plotting/_base.py b/src/poliastro/plotting/_base.py
@@ -190,11 +190,19 @@ def _anim(self, orbit, *, label=None, color=None, trail=False):
  colors = self._get_colors(color, trail)
 
  self.set_attractor(orbit.attractor)
+ from poliastro.twobody import propagation
 
  orbit = orbit.change_plane(self.plane)
 
  label = generate_label(orbit.epoch, label)
- coordinates = orbit.sample(self._num_points)
+ time_laps = [i for i in range(int(orbit.period.value))]
+ tofs = time_laps * u.day * 2
+ k = propagation.danby(orbit.attractor.k, orbit.r, orbit.v, tofs)[0]
+ x = k.to(u.km).value[:, 0]
+ y = k.to(u.km).value[:, 1]
+ z = k.to(u.km).value[:, 2]
+ cartesian_repr = CartesianRepresentation(x * (u.km), y * (u.km), z * (u.km))
+ coordinates = cartesian_repr
 
  return self.__anim_trajectory(
  coordinates, orbit.r, label=label, colors=colors, dashed=True

diff --git a/src/poliastro/plotting/static.py b/src/poliastro/plotting/static.py
@@ -180,7 +180,7 @@ def animate(i):
  self.fig,
  animate,
  frames=np.arange(0, len(x), 1),
- interval=40,
+ interval=30,
  blit=True,
  repeat=True,
  )