前文我们已经完成了一个集暂停、倍速、显示进度条功能为一体的视频播放器,今天我们再来增加一个新的功能——发送弹幕。
tkinter播放视频的原理,就是读取每一帧的图片,然后刷新画布。所以如果想实现弹幕功能,只要获取输入文本框中的文本,再写到图片上就好了。cv2.putText能够实现这个功能,但无法添加中文,所以我们需要换一种方法。
首先,我们需要from PIL import ImageDraw,ImageFont
然后用ImageDraw.Draw转换图片格式,转换后就可以调用.text()往图片上写中文了。
具体实现过程,只需要修改前文的一个函数:
def tkImage(n): global nowt,pointx,txt #倍速在这里实现 for i in range(n): ref,frame = vc1.read() pilImage = Image.fromarray(frame) draw = ImageDraw.Draw(pilImage) font = ImageFont.truetype("simhei.ttf", 40, encoding="utf-8") #参数1:字体文件路径,参数2:字体大小 #txt是读取的Entry内容,在button关联的函数中获取 if txt!="": draw.text((pointx, pointy), txt, (255, 255, 255), font=font) #pointx, pointy是文字添加的位置 pointx-=10 #每次向左移动10个单位 if(pointx==0): pointx=size[1] txt="" pilImage = cv2.cvtColor(np.array(pilImage), cv2.COLOR_BGR2RGB) pilImage = Image.fromarray(pilImage) pilImage = pilImage.resize((window_width, window_height),Image.ANTIALIAS) tkImage = ImageTk.PhotoImage(image=pilImage) nowt+=n #记录当前的帧数 return tkImage
现在我们把前文的显示四张图像改成一张,重新整理一下代码:
import time import tkinter as tk from tkinter import * from tkinter import ttk import cv2 from PIL import Image, ImageTk,ImageFilter,ImageDraw,ImageFont import multiprocessing import numpy as np import random import os filePath = 'D:\\movie\\' #电影存放路径 mlist=os.listdir(filePath) #文件夹下所有文件名称 window_width=960 #界面宽 window_height=760 #界面长 image_width=int(window_width*0.5) #图像宽 image_height=int(window_height*0.5) #图像长 imagepos_x=0 #画布位置x imagepos_y=0 #画布位置Y lock=0 #暂停标志 n=1 #初始倍速 nowt=0 #当前帧数 nows=0 #当前播放时长(秒) pointx=0 #弹幕起始点横坐标 pointy=0 #弹幕起始点纵坐标 txt="" #弹幕内容 #获取当前图像 def tkImage(n): #倍速在这里实现 global nowt,pointx,pointy,txt for i in range(n): ref,frame = vc1.read() pilImage = Image.fromarray(frame) draw = ImageDraw.Draw(pilImage) font = ImageFont.truetype("simhei.ttf", 40, encoding="utf-8")#参数1:字体文件路径,参数2:字体大小 if txt!="": draw.text((pointx, pointy), txt, (255, 255, 255), font=font) pointx-=10 if(pointx==-50): pointx=size[1] txt="" pilImage = cv2.cvtColor(np.array(pilImage), cv2.COLOR_BGR2RGB) pilImage = Image.fromarray(pilImage) pilImage = pilImage.resize((window_width, 720),Image.ANTIALIAS) tkImage1 = ImageTk.PhotoImage(image=pilImage) nowt+=n return tkImage1 #图像的显示与更新 def video(): global nows,nowt def video_loop(): global nows,nowt try: while True: if lock % 2 == 0: #是否暂停 picture1=tkImage(n) if nowt >= fps: nowt=0 #每过一秒则清零重计 nows+=1 #每过一秒,当前播放时间也加1 mi=str(int(nows/60)) #分钟 se=str(int(nows%60)) #秒 if int(mi)<10: mi="0"+mi if int(se)<10: se="0"+se showt=mi+":"+se #当前时间 show=showt+"/"+totalt #最终显示格式(当前时间/总时长) tlabel.config(text=show) canvas.coords(fill_line, (0, 0, int(window_width/tt*nows), 15)) #填充进度条 canvas1.create_image(0,0,anchor='nw',image=picture1) win.update_idletasks() win.update() else: win.update_idletasks() win.update() except: return #每次开始播放前初始化 nows,nowt=0,0 canvas.coords(fill_line, (0, 0, 0, 15)) video_loop() vc1.release() cv2.destroyAllWindows() #右键事件:倍速 def right(self): global n n+=1 if n>4: n=1 #左键事件:暂停 def left(self): global lock lock+=1 #按钮事件:获取弹幕 def get_txt(): global txt,pointx,pointy txt=E1.get() pointx=size[1] #初始x坐标 pointy=random.randint(50,150) #初始y坐标随机 E1.delete(0, END) #播放选中文件 def start(): global vc1,size,frames_num,fps,totalt,tt,pointx val = theLB.get() #获得下拉框当前内容 vc1 = cv2.VideoCapture(filePath+val) #读取视频 size = (int(vc1.get(cv2.CAP_PROP_FRAME_HEIGHT)), int(vc1.get(cv2.CAP_PROP_FRAME_WIDTH))) #视频图像的长和宽 frames_num=vc1.get(cv2.CAP_PROP_FRAME_COUNT) #总帧数 fps = vc1.get(cv2.CAP_PROP_FPS) #帧率 totalt=str(int(frames_num/fps/60))+":"+str(int(frames_num/fps)%60) #视频时长 tt=int(frames_num/fps) #进度条叠满所需次数 pointx=size[1] #弹幕起始坐标x video() #跳转到指定位置 def kj(): global nows,vc1 time=int(int(s1.get())*frames_num/fps/100) #要跳转到的时长 vc1.set(cv2.CAP_PROP_POS_FRAMES,int(time*fps)) #读取到指定帧数 ref,frame = vc1.read() nows=time '''布局''' win = tk.Tk() win.geometry(str(window_width+120)+'x'+str(window_height+20)) #显示视频的画布 canvas1 =Canvas(win,bg='white',width=window_width,height=720) canvas1.place(x=imagepos_x,y=imagepos_y) canvas1.bind('<Button-1>', left) canvas1.bind('<Button-3>', right) #显示进度条的画布 canvas = Canvas(win, width=image_width*2, height=15, bg="white") canvas.place(x=0, y=722) fill_line=canvas.create_rectangle(0,0,0,15,fill = 'LightGreen') #坐标是相对于画布的 #弹幕输入框 E1 = Entry(win, bd =5,width=100) E1.place(x=0, y=745) #发送弹幕按钮 B1 = Button(win, text="发送", command=get_txt,font=('黑体', 10),fg='blue',width=10,height=2) B1.place(x=750, y=742) #显示时间的Label tlabel = Label(win,font=('黑体', 13),text='') tlabel.place(x=850, y=750) #选择影片下拉框 theLB = ttk.Combobox(win,width=12,height=10) theLB["values"] = mlist theLB.current(0) #默认选第一个 theLB.place(x=965, y=0) #播放影片按钮 B2 = Button(win, text="播放", command=start,font=('黑体', 10),fg='red',width=10,height=2) B2.place(x=980, y=50) #跳转按钮 B3 = Button(win, text="跳转", command=kj,font=('黑体', 10),fg='red',width=10,height=2) B3.place(x=980, y=150) #选择跳转位置的滚动条 s1 = Scale(win,from_=0,to=99,orient=HORIZONTAL) #orient=HORIZONTAL设置水平方向显示 s1.place(x=970, y=100) win.mainloop()
上述代码不仅实现了我们所说的功能,我还用下拉框Combobox自动加载文件夹下的所有文件名,以供选择文件播放;并且能够拖动滚动条跳转到指定位置,利用了cv2.VideoCapture的.set()方法。
怎么样,更像一个视频播放器了吧?如果你还想往下做,可以加入“添加到播放列表”的功能,选择其他路径下的文件添加到全局变量mlist里,然后更新一下Combobox的values即可。