# -*- coding: utf-8 -*-
import sys
if '/opt/ros/kinetic/lib/python2.7/dist-packages' in sys.path:
    sys.path.remove('/opt/ros/kinetic/lib/python2.7/dist-packages') 
import cv2

class CVCreateVideo():

    def __init__(self):
        pass

    def start(self, path, fps):
        self.path = path
        self.fps = fps
        self.size = None

    def add(self, image):
        self.auto_size(image)
        self.videowriter.write(image)

    def end(self):
        self.videowriter.release()

    def auto_size(self, image):
        if self.size == None:
            self.size = (image.shape[1], image.shape[0])
            self.videowriter = cv2.VideoWriter(self.path, cv2.VideoWriter_fourcc('M','J','P','G'), self.fps, self.size)
        else:
            assert self.size == (image.shape[1], image.shape[0])

def main():
    cv_create_video = CVCreateVideo()
    cv_create_video.start(path='out.avi', fps=16)
    for i in range(1,10):
        image = cv2.imread('camera.png')
        cv_create_video.add(image)
    cv_create_video.end()

if __name__ == '__main__':
    main()