首頁 >後端開發 >Python教學 >Python怎麼實現錄影功能

Python怎麼實現錄影功能

WBOY
WBOY轉載
2023-05-11 11:04:153424瀏覽

    一、介面開發

        #设置主界面
    def set_init_window(self):
        # 去掉tkinter默认的标题
        self.tk.title('')
        # 隐藏默认图标
        self.tk.iconbitmap(self.icon_path())
        #获取屏幕的宽度
        screeWidth = self.tk.winfo_screenwidth()
        #获取屏幕高度
        screeHeight = self.tk.winfo_screenheight()
        
        width = int((screeWidth - 500) / 2)
        height = int((screeHeight - 300) / 2)
        # 设置主界面的大小和默认位置
        self.tk.geometry(f'500x100+{width}+{height}')
        
        #添加开始录制按钮,点击之后开启两个线程:一个录屏、一个监听键盘
        btn1 = tkinter.Button(self.tk, width=5, height=1, text='开始',
        command=lambda:[threading.Thread(target=self.video_record).start(),threading.Thread(target=self.start_listen).start()])
        btn1.pack()
        # 设置按钮位置
        btn1.place(x=110, y=50, anchor='n')
        
        #开启新线程设置录屏范围
        btn2 = tkinter.Button(self.tk, width=15, height=1, text='设置录制区域', command=lambda:threading.Thread(target=self.set_range).start())
        btn2.pack()
        btn2.place(x=230, y=50, anchor='n')
    
    
    #生成透明的icon图标
    def icon_path(self):
        ICON = (b'\x00\x00\x01\x00\x01\x00\x10\x10\x00\x00\x01\x00\x08\x00h\x05\x00\x00'
                b'\x16\x00\x00\x00(\x00\x00\x00\x10\x00\x00\x00 \x00\x00\x00\x01\x00'
                b'\x08\x00\x00\x00\x00\x00@\x05\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'
                b'\x00\x01\x00\x00\x00\x01') + b'\x00' * 1282 + b'\xff' * 64
        _, ICON_PATH = tempfile.mkstemp()
        with open(ICON_PATH, 'wb') as icon_file:
            icon_file.write(ICON)
        return ICON_PATH

    二、錄影機參數設定

    1.設定錄影範圍
    #设置录屏范围
    def set_range(self):
        #self.init_window.withdraw() #隐藏窗口
        self.tk.state('icon')#窗口最小化
        screeWidth = self.tk.winfo_screenwidth()
        screeHeight = self.tk.winfo_screenheight()
    
        self.newFrame = tkinter.Toplevel(self.tk,width=screeWidth,height=screeHeight,bg='white')#开启新窗口
        self.newFrame.attributes('-transparentcolor', 'white')  # 使白色为透明色
        self.newFrame.overrideredirect(True)  # 隐藏导航栏
    
        self.canvas = tkinter.Canvas(self.newFrame,bg='white',bd=0,width=screeWidth,height=screeHeight)
        self.canvas.bind(&#39;<Button-1>&#39;, self.onLeftButtonDown)#按下左键
        self.canvas.bind(&#39;<B1-Motion>&#39;, self.onLeftButtonMove)#移动鼠标
        self.canvas.bind(&#39;<ButtonRelease-1>&#39;, self.onLeftButtonUp)#抬起左键
        self.canvas.pack(fill=tkinter.BOTH, expand=tkinter.YES)
    
        time.sleep(0.3)
        im = ImageGrab.grab()
        # 暂存全屏截图
        im.save(&#39;temp.png&#39;)
        im.close()
        self.image = tkinter.PhotoImage(file=&#39;temp.png&#39;)
        os.remove(&#39;temp.png&#39;)
        self.canvas.create_image(screeWidth//2, screeHeight//2, image=self.image)
    2.滑鼠事件監聽
    #按下鼠标
    def onLeftButtonDown(self,event):
        try:
            self.canvas.delete(self.lastDraw)
            self.canvas.delete(self.dot1)
            self.canvas.delete(self.dot2)
            self.btn3.destroy()
        except Exception as e:
            pass
    
        self.X = event.x
        self.Y = event.y
        self.X2 = 0
        self.Y2 = 0
    
    #移动鼠标
    def onLeftButtonMove(self,event):
        try:  # 删除刚画完的图形,不然所有画的框都会出现
            self.canvas.delete(self.lastDraw)
            self.canvas.delete(self.dot1)
            self.canvas.delete(self.dot2)
            self.btn3.destroy()
        except Exception as e:
            pass
    
        self.X2 = event.x
        self.Y2 = event.y
        self.lastDraw = self.canvas.create_rectangle(self.X, self.Y, event.x, event.y,width=2, outline=&#39;pink&#39;)
    
    #松开鼠标
    def onLeftButtonUp(self,event):
        print("起点", self.X, self.Y)
        print("终点", self.X2, self.Y2)
    
        if self.X2==0 and self.X2==0:
            return
    
        self.width, self.high = self.X2-self.X,self.Y2-self.Y
        self.region = (self.X, self.Y, self.X2, self.Y2)
    
        self.dot1=self.canvas.create_text(self.X, self.Y - 10, text=f&#39;({self.X},{self.Y})&#39;, font=("Purisa", 12), fill="pink")
        self.dot2=self.canvas.create_text(self.X2, self.Y2 + 10, text=f&#39;({self.X2},{self.Y2})&#39;, font=("Purisa", 12), fill="pink")
    
        # self.newFrame.destroy()#销毁窗口
        # self.init_window.deiconify()#显示窗口
        # self.tk.state(&#39;normal&#39;)  # 取消窗口最小化
    
        self.btn3 = tkinter.Button(self.canvas, width=15, height=1, text=&#39;确定录制区域&#39;,bg=&#39;pink&#39;,fg=&#39;#64854c&#39;,command=lambda:[self.newFrame.destroy(),self.tk.state(&#39;normal&#39;)])
        self.btn3.pack()
        self.btn3.place(x=self.X2-20, y=self.Y2+20, anchor=&#39;n&#39;)
    3.鍵盤事件監聽
    # 开始监听
    def start_listen(self):
        with keyboard.Listener(on_press=self.on_press) as listener:
            listener.join()
        # 监听按键
    def on_press(self,key):
        if key == keyboard.Key.esc:
            self.flag = True  # 改变
            return False  # 返回False,键盘监听结束!

    三、錄影動作

    def video_record(self):
         fourcc = cv2.VideoWriter_fourcc(&#39;X&#39;, &#39;V&#39;, &#39;I&#39;, &#39;D&#39;)
         out = cv2.VideoWriter(&#39;output.mp4&#39;, fourcc, 14, (self.width, self.high))  # 参数分别为 输出文件名,解码方式,帧数,录像范围
         self.count = 1
         while (True):
             img = ImageGrab.grab(self.region) #指定截取坐标(左边X,上边Y,右边X,下边Y)
             img_np = numpy.array(img)
             frame = cv2.cvtColor(img_np, cv2.COLOR_BGR2RGB)  # ImageGrab获取的颜色为BGR排序,需转换为RGB
             out.write(frame)
             self.count += 1
             label = tkinter.Label(self.tk, text=f"{int(self.count / 14)}秒")
             label.pack()
             label.place(x=320, y=55, anchor=&#39;n&#39;)
             # 点击ESC退出
             if self.flag == True:
                 tkinter.messagebox.showinfo(&#39;提示&#39;, &#39;录屏结束&#39;)
                 self.flag = False  # 改变录屏状态
                 break
         out.release()
         cv2.destroyAllWindows()

    以上是Python怎麼實現錄影功能的詳細內容。更多資訊請關注PHP中文網其他相關文章!

    陳述:
    本文轉載於:yisu.com。如有侵權,請聯絡admin@php.cn刪除