前文我们已经完成了一个集暂停、倍速、显示进度条功能为一体的视频播放器,今天我们再来增加一个新的功能——发送弹幕。
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即可。
来看看我们的成果吧:
费了半天劲,反正还是只能看默剧。。