Coverage for /builds/ase/ase/ase/io/animation.py: 100.00%

1# fmt: off 

2 

3from ase.visualize.plot import animate 

4 

5 

6def write_animation(filename, images, writer=None, 

7 interval=200, 

8 save_count=None, # ignored for newer matplotlib (2023) 

9 save_parameters=None, ax=None, **kwargs): 

10 import matplotlib.pyplot as plt 

11 

12 if save_parameters is None: 

13 save_parameters = {} 

14 

15 if ax is None: 

16 ax = plt.gca() 

17 

18 animation = animate(images, ax=ax, 

19 interval=interval, 

20 **kwargs) 

21 animation.save(filename, writer=writer, 

22 **save_parameters) 

23 

24 

25# Shortcuts for ase.io.formats (guessing file type from extension): 

26write_gif = write_animation 

27write_mp4 = write_animation