diff --git a/plotter.py b/plotter.py index 9f21eff..b325624 100644 --- a/plotter.py +++ b/plotter.py @@ -26,7 +26,7 @@ if os.environ.get("DISPLAY", "") == "": import datetime import matplotlib.pyplot as plt -from moviepy.video.io import ImageSequenceClip +from moviepy.video.io.ImageSequenceClip import ImageSequenceClip import pspec_read from baseprocessor import Rule, BaseProcessor @@ -301,6 +301,8 @@ class Plotter(Aggregator, BaseProcessor): nums = self.nums datafiles = [] + if movie: + filenames = [] if rule.kind == "snapshot" or rule.kind == "cells": run_num = [(run, num) for run in runs for num in nums[run]] @@ -362,6 +364,8 @@ class Plotter(Aggregator, BaseProcessor): # Save in astrophysix format df = rule.datafile(name, arg) df[filetype] = plot_filename + if movie: + filenames.append(plot_filename) if plot_info is not None: df.plot_info = plot_info if num is not None: @@ -378,7 +382,6 @@ class Plotter(Aggregator, BaseProcessor): datafiles.append(df) if movie: - filenames = [df[filetype] for df in datafiles] clip = ImageSequenceClip(filenames, fps=movie_fps) movie_filename = self._find_filename(name_full) movie.filename = os.path.splitext(movie_filename)[0] + ".mp4"