raiot
/
PythonLinearNonlinearControl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #9 from Geonhee-LEE/master 

Bug fixed for saving .mp4 format
This commit is contained in:
Shunichi09 2020-07-04 16:47:32 +09:00 committed by GitHub
parent ebdb47b6ab bfe137414b
commit 969fee7e73
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
PythonLinearNonlinearControl/plotters/animator.py
View File

@ -4,6 +4,7 @@ from logging import getLogger
import numpy as np import numpy as np
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation from matplotlib.animation import FuncAnimation
import matplotlib.animation as animation
logger = getLogger(__name__) logger = getLogger(__name__)
@ -66,14 +67,18 @@ class Animator():
self._setup() self._setup()
_update_img = lambda i: self._update_img(i, history_x, history_g_x) _update_img = lambda i: self._update_img(i, history_x, history_g_x)
# Set up formatting for the movie files
Writer = animation.writers['ffmpeg']
writer = Writer(fps=15, metadata=dict(artist='Me'), bitrate=1800)
# call funcanimation # call funcanimation
animation = FuncAnimation( ani = FuncAnimation(
self.anim_fig, self.anim_fig,
_update_img, interval=self.interval, frames=len(history_x)-1) _update_img, interval=self.interval, frames=len(history_x)-1)
# save animation # save animation
path = os.path.join(self.result_dir, self.controller_type, path = os.path.join(self.result_dir, self.controller_type,
"animation-" + self.env_name + ".mp4") "animation-" + self.env_name + ".mp4")
logger.info("Saved Animation to {} ...".format(path)) logger.info("Saved Animation to {} ...".format(path))
animation.save(path, writer="ffmpeg") ani.save(path, writer=writer)