From fa156a13b4a8e8e209d9a76dfe7472afbec93b97 Mon Sep 17 00:00:00 2001
From: Noe Brucy <noe.brucy@cea.fr>
Date: Tue, 6 Jul 2021 14:53:57 +0200
Subject: [PATCH] [plotter] fix movies

---
 plotter.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

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"