Esempio n. 1
0
 def readFile(self, filename):
     self.seq = YUV_sequence()
     [width, height, fps, depth] = self.seq.get_seq_params(filename)
     print([width, height, fps, depth])
     self.NumFrame = self.seq.open_file(filename, width, height, fps, depth)
     self.parent.geometry("{0}x{1}".format(width, height))
     self.showImage(0)
Esempio n. 2
0
class MainDiag(Frame):
    def __init__(self, parent):
        Frame.__init__(self, parent, width=832, height=480)
        self.parent = parent
        self.initUI()
        self.data = 0
        self.chn = 'A'
        self.play = False
        self.onOpen()

    def initUI(self):
        self.parent.title("YUV Player")
        self.pack(fill=BOTH, expand=1)

        menubar = Menu(self.parent)
        self.parent.config(menu=menubar)

        filemenu = Menu(menubar)
        filemenu.add_command(label="Open", command=self.onOpen)
        menubar.add_cascade(label="File", menu=filemenu)

        self.bind("<Configure>", self.on_resize)
        self.parent.bind("<Left>", self.prev1)
        self.parent.bind("<Right>", self.next1)
        self.parent.bind("<Shift-Left>", self.prev_sec)
        self.parent.bind("<Shift-Right>", self.next_sec)
        self.parent.bind("n", self.next_chn)
        self.parent.bind("<space>", self.play_pause)

    def onOpen(self):
        ftypes = [('YUV files', '*.yuv'), ('All files', '*')]
        fl = filedialog.askopenfilename(filetypes=ftypes)
        if fl != '':
            self.readFile(fl)

    def readFile(self, filename):
        self.seq = YUV_sequence()
        [width, height, fps, depth] = self.seq.get_seq_params(filename)
        print([width, height, fps, depth])
        self.NumFrame = self.seq.open_file(filename, width, height, fps, depth)
        self.parent.geometry("{0}x{1}".format(width, height))
        self.showImage(0)

    def prev_sec(self, event):
        self.showImage((self.nf - self.seq.fps) % self.NumFrame)

    def next_sec(self, event):
        self.showImage((self.nf + self.seq.fps) % self.NumFrame)

    def prev1(self, event):
        self.showImage((self.nf - 1) % self.NumFrame)

    def next1(self, event=0):
        self.showImage((self.nf + 1) % self.NumFrame)

    def next_chn(self, event):
        chn = {'A': 'Y', 'Y': 'U', 'U': 'V', 'V': 'A'}
        self.chn = chn[self.chn]
        self.showImage(self.nf)

    def play_pause(self, event):
        self.play = not self.play
        if self.play:
            self.parent.title("YUV Player <PLAY>")
        else:
            self.parent.title("YUV Player <PAUSE>")
        self.showImage()

    def showImage(self, FrameNum=-1):
        import time
        se = time.time()
        if FrameNum >= 0:
            self.nf = FrameNum
            self.data = self.seq.getFrame(FrameNum)
        elif not self.data:
            return
        load = Image.fromarray(
            self.data.GetImg(self.chn, self.winfo_width(),
                             self.winfo_height()))
        render = ImageTk.PhotoImage(load)
        # labels can be text or images
        img = Label(self, image=render)
        img.image = render
        img.place(x=0, y=0)
        if self.play:
            cur = time.time()
            rem = round((1. / self.seq.fps - (cur - se)) * 1000)
            if rem < 0:
                print("delayed ", rem)
                self.next1()
            else:
                self.after(rem, self.next1)

    def on_resize(self, event):
        self.showImage(-1)