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

Bug fixed for saving .mp4 format

This commit is contained in:
Geonhee 2020-07-02 11:50:54 +09:00 committed by GitHub
parent ebdb47b6ab
commit bfe137414b
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

9
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,8 +67,12 @@ 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)
@ -76,4 +81,4 @@ class Animator():
"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)