def write_frame()

in tools/video_writer.py [0:0]

• 26 lines of code
• 10 McCabe index (conditional complexity)

    def write_frame(self, frame, resize=None):

        outfile = os.path.join(self.cache_dir, self.regexp % self.frame_num)

        ftype = type(frame)
        if ftype == matplotlib.figure.Figure:
            plt.savefig(outfile)
            im = None
        elif ftype == np.array or ftype == np.ndarray:
            im = Image.fromarray(frame)
        elif ftype == Image.Image:
            im = frame
        elif ftype == str:
            im = Image.open(frame).convert('RGB')
        else:
            raise ValueError('cant convert type %s' % str(ftype))

        if im is not None:
            if resize is not None:
                if type(resize) in (float,):
                    resize = [int(resize*s) for s in im.size]
                else:
                    resize = [int(resize[1]), int(resize[0])]
                resize[0] += resize[0] % 2
                resize[1] += resize[1] % 2
                im = im.resize(resize, Image.ANTIALIAS)
            im.save(outfile)

        self.frames.append(outfile)
        self.frame_num += 1