OpenVidTens2D/VideoWriter.py at main · ThHanke/OpenVidTens2D · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
from cv2 import VideoWriter
from multiprocessing import Process
import os


class VideoWriterProcess(Process):
    def __init__(self, filename, fourcc, queue):
        Process.__init__(self)
        self.filename = filename
        self.queue = queue
        self.daemon = True

        self.fourcc = fourcc
        self.videofilepart = 1
        self.videowriter = None

        self.start()

    def run(self):
        while True:
            msg = self.queue.get()
            if msg == 'TERMINATE':
                if self.videowriter is not None:
                    self.videowriter.release()
                    while self.videowriter.isOpened():
                        print('wait till videowriter is released')
                        pass
                print('i have been terminated')
                break

            else:
                if self.videowriter is None:
                    print('open video writer')
                    self.videowriter = VideoWriter(self.filename + '.' + str(self.videofilepart).rjust(2, '0') + '.avi',
                                                   self.fourcc, 30, (msg.shape[1], msg.shape[0]), isColor=False)
                    # self.videowriter.open(self.filename+'.'+str(self.videofilepart).rjust(2,'0')+'.avi',fourcc,30,(self.raw.shape[1],self.raw.shape[0]),isColor=False)
                    while not self.videowriter.isOpened():
                        #print 'wait till videowriter is ready'
                        pass
                # check filesize
                filesize = os.path.getsize(self.filename + '.' + str(self.videofilepart).rjust(2, '0') + '.avi')
                #print filesize
                # if filesize>=4294967296: #4GB
                if filesize >= 2000000000:  # 4GB
                    self.videowriter.release()
                    while self.videowriter.isOpened():
                        print('wait till videowriter is released')
                        pass
                    self.videofilepart += 1
                    self.videowriter.open(self.filename + '.' + str(self.videofilepart).rjust(2, '0') + '.avi',
                                          self.fourcc, 30, (msg.shape[1], msg.shape[0]), isColor=False)
                    while not self.videowriter.isOpened():
                        print('wait till videowriter is ready')
                self.videowriter.write(msg)