-
Notifications
You must be signed in to change notification settings - Fork 9
/
Copy pathffmpeg_vf.py
182 lines (165 loc) · 11.8 KB
/
ffmpeg_vf.py
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
#!/usr/bin/python
# -*- coding: UTF-8 -*-
#/**************************************************************************
#* ffmpeg_vf.py: video filter cases on windows: python 3.x, ffmpeg-4.3.1
#***************************************************************************
#* Copyright (c) 2020-2020 Hybase@qq.com
#*
#* This program is free software: you can redistribute it and/or modify
#* it under the terms of the GNU General Public License as published by
#* the Free Software Foundation, either version 3 of the License, or
#* (at your option) any later version.
#*
#* This program is distributed in the hope that it will be useful,
#* but WITHOUT ANY WARRANTY; without even the implied warranty of
#* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
#* GNU General Public License for more details.
#*
#****************************************************************************/
import os
import re
import sys
import hysmm_def
import ffmpeg_run
################################################################################
# 通用参数字符串
dst_dir = hysmm_def.dst_dir
log_param = hysmm_def.log_param
tt_src_dir = 'e:/material/ffmpeg_hys_python'
input_mkv_file = ' -i ' + tt_src_dir + '/tt.mkv '
input_mp4_file = ' -i ' + tt_src_dir + '/tt.mp4 '
input_ts_file = ' -i ' + tt_src_dir + '/tt.ts '
input_mp3_file = ' -i ' + tt_src_dir + '/es_mp3.mp3 '
input_png_file = ' -f image2 -i ' + tt_src_dir + '/image/image-%03.png '
hls_cancat_file= tt_src_dir + '/hls_part/hls_concat.txt '
vf_out = dst_dir + '/vf_effect'
#log, vf_str, 'd:/otest/case019_vf_crop.mp4', flag_vf_crop, 'case_vf_crop'
def ffmpeg_vf_exec(log, vf_str, output_file_name, flag, case_name):
ffmpeg_run.ffmpeg_vf_run(case_name, flag, log, input_mkv_file, vf_str, output_file_name)
return
################################################################################
# case flags: (default: the first be set, please set others by yourself)
overlay_flag = False
play_flag = False # true - ffplay, false - ffmpeg
sfx = '.ts' # .gif or .ts
start_num = 3
end_num = 39
class vf_case_c:
def __init__(self):
# crop=w=%d:h=%d:x=%d:y=%d
# 注释：宽度:高度:x:y(x,y如果不写则从中心裁剪)
# 注意事项：width, height, x, y 不要超出原始视频边界
self.case002 = [vf_out + '/d002_crop' + sfx, " crop=w=960:h=540:x=80:y=40 "]
self.case003 = [vf_out + '/d003_rotate_1_dynamic' + sfx, " rotate='PI*2/10*t' "] # 视频动态旋转 (T为旋转360度的时间常量, 单位:秒)
self.case004 = [vf_out + '/d004_rotate_2_45degress' + sfx, " rotate='PI/4' "] # 视频旋转45度 (保留原始尺寸)
self.case005 = [vf_out + '/d005_rotate_3_45degress' + sfx, " rotate='PI/4:ow=floor(hypot(iw,ih)/4)*4:oh=ow:c=none' "] # 视频旋转45度，以对角线长度为宽高
self.case006 = [vf_out + '/d006_transpose_1_cw90' + sfx, " transpose=1:portrait "] # 视频顺时针旋转90 (transpose=1)
self.case007 = [vf_out + '/d007_transpose_2_ccw90' + sfx, " transpose=2 "] # 视频逆时针旋转90 (transpose=2)
self.case008 = [vf_out + '/d008_transpose_3' + sfx, " transpose=3 "] # 视频逆时针旋转90度后垂直镜像 (transpose=3)
self.case009 = [vf_out + '/d009_hflip_vflip' + sfx, " hflip,vflip "] # 视频顺时针旋转180度 (hflip,vflip)
self.case010 = [vf_out + '/d010_hflip' + sfx, " hflip "] # 视频水平镜像 (hflip)
self.case011 = [vf_out + '/d011_vflip' + sfx, " vflip "] # 视频竖直镜像 (vflip)
# 视频文字跑马灯效果 (drawtext)
vf_str1= '''drawtext=fontfile=c:/Windows/Fonts/Medium.ttf:text='welcome-hr-left2right':x=90+t*40:y=60:fontsize=36:fontcolor=#ffffff@0.7:shadowy=-1'''
vf_str2= '''drawtext=fontfile=c:/Windows/Fonts/Medium.ttf:text='welcome-hr-right2left':x=1200-t*40:y=360:fontsize=36:fontcolor=#ffffff@0.7:shadowy=-1'''
vf_str3= '''drawtext=fontfile=c:/Windows/Fonts/Medium.ttf:text='welcome-ve-up2down':x=100:y=10+t*20:fontsize=36:fontcolor=#ffffff@0.7:shadowy=-1'''
vf_str4= '''drawtext=fontfile=c:/Windows/Fonts/Medium.ttf:text='welcome-ve-down2up':x=400:y=700-t*20:fontsize=36:fontcolor=#ffffff@0.7:shadowy=-1'''
vf_marquee_str= vf_str1 + ',' + vf_str2 + ',' + vf_str3 + ',' + vf_str4
self.case012 = [vf_out + '/d012_drawtext' + sfx, vf_marquee_str]
self.case013 = [vf_out + '/d013_drawgrid' + sfx, " drawgrid=width=80:height=80:thickness=2:color=yellow@0.9 "] # 网格
self.case014 = [vf_out + '/d014_drawbox' + sfx, " drawbox=x=10:y=10:w=860:h=480:color=pink@0.5:t=fill "] # box框
self.case015 = [vf_out + '/d015_boxblur' + sfx, " boxblur=2:1:cr=0:ar=0 "]
self.case016 = [vf_out + '/d016_edgedetect' + sfx, " edgedetect=low=0.1:high=0.4 "]
self.case017 = [vf_out + '/d017_eq' + sfx, " eq=contrast=1.5:brightness=0.5 "]
self.case018 = [vf_out + '/d018_fadein' + sfx, " fade=in:0:250 "]
self.case019 = [vf_out + '/d019_fadeout' + sfx, " fade=out:0:250 "]
self.case020 = [vf_out + '/d020_geq' + sfx, " format=gray,geq=lum_expr='(p(X,Y)+(256-p(X-4,Y-4)))/2' "]
self.case021 = [vf_out + '/d021_histogram' + sfx, " histogram "]
self.case022 = [vf_out + '/d022_histeq' + sfx, " histeq=strength=0.5:intensity=0.4:antibanding=weak "]
self.case023 = [vf_out + '/d023_hqdn3d' + sfx, " hqdn3d "]
self.case024 = [vf_out + '/d024_hue' + sfx, " hue=h=90:s=1:b=5 "]
self.case025 = [vf_out + '/d025_il' + sfx, " il=l=d:c=d "]
self.case026 = [vf_out + '/d026_lutyuv' + sfx, " lutyuv='y=maxval+minval-val:u=maxval+minval-val:v=maxval+minval-val' "]
self.case027 = [vf_out + '/d027_negate' + sfx, " negate "]
self.case028 = [vf_out + '/d028_oscilloscope' + sfx, " oscilloscope=x=1:y=0.5:s=1:t=1 "]
self.case029 = [vf_out + '/d029_pad' + sfx, " pad='iw*1.1:ih*1.1:(ow-iw)/2:(oh-ih)/2:color=violet' "]
self.case030 = [vf_out + '/d030_showpalette' + sfx, " showpalette "]
self.case031 = [vf_out + '/d031_shuffleplanes' + sfx, " shuffleplanes=0:2:1:3 "]
self.case032 = [vf_out + '/d032_sobel' + sfx, " sobel "]
self.case033 = [vf_out + '/d033_stereo3d' + sfx, " stereo3d=abl:sbsr "]
self.case034 = [vf_out + '/d034_swapuv' + sfx, " swapuv "]
self.case035 = [vf_out + '/d035_tile' + sfx, " scale=224:180,tile=4x3:nb_frames=12:padding=2:margin=2 "]# 42 tile 填充参数需注意，确保不造成奇数宽或高
self.case036 = [vf_out + '/d036_unsharp' + sfx, " unsharp "]
self.case037 = [vf_out + '/d037_vignette' + sfx, " vignette='PI/4+random(1)*PI/50' "] # 闪烁渐晕
self.case038 = [vf_out + '/d038_delogo' + sfx, " delogo=w=200:h=80:x=1000:y=50 "]
#################################################################################
# main process
if __name__ == "__main__":
# init logger
log = hysmm_def.Logger('sys_all.log',level='debug')
hysmm_def.Logger('sys_err.log', level='error')
log.logger.info("========= program start ============")
# check source and output dir, create output dir if necessary
hysmm_def.checkDir(log, tt_src_dir)
hysmm_def.createDirIfNotFound(log, dst_dir)
hysmm_def.createDirIfNotFound(log, vf_out)
# init vf case
vf_case = vf_case_c()
vf_case_dict = vf_case.__dict__
#####################################################
# overlay image logos(png/bmp/webp/gif/jpeg) , draw text
# ffmpeg -fflags +genpts -i main_video.mkv -i logo1.png -i logo2.jpg -max_muxing_queue_size 256
# -filter_complex "[0:v][1:v]overlay=40:40[bkg1];[bkg1][2:v]overlay=200:40,"\
# "drawtext=fontfile=/fonts/mingliu.ttc:text='hybase@qq.com 视频水印1':x=50:y=450:fontsize=32:fontcolor=0xFFEE00@0.7:shadowy=-1,"\
# "drawtext=fontfile=/fonts/mingliu.ttc:text='hybase@qq.com 视频水印2':x=350:y=450:fontsize=32:fontcolor=0xFFEE00@0.7:shadowy=-1"\
# -map 0:0 -c:v:0 libx264 -s 1280x720 -b:v:0 1500000 -minrate 1500000 -maxrate 3000000 -bufsize 1500000 -g 25 -bf 1 -pix_fmt yuv420p -r:v:0 25.000000 \
# -map 0:1 -c:a:0 aac -ab:a:0 64k -ar:a:0 48000 -ac:a:0 2 -vol:a:0 256 \
# -map 0:1 -c:a:1 ac3 -ab:a:1 128k -ar:a:1 48000 -ac:a:1 2 -vol:a:1 256 \
# -muxrate 0 -f mpegts d:/output.ts
tsmux_str = hysmm_def.tsmux_vbr_str
png_logo = ' -i e:/material/image/logo/jxtv_96x96.png '
jpg_logo = ' -i e:/material/image/logo/jxtv_96x96.jpg '
overlay_str = '[0:v][1:v]overlay=60:40[bkg1];[bkg1][2:v]overlay=200:40'
drawtext1 = '''drawtext=fontfile=c:/Windows/Fonts/msyh.ttf:text='hybase@qq.com视频水印1':x=50:y=550:fontsize=36:fontcolor=0xFFFF00@0.7:shadowy=-1,'''
drawtext2 = '''drawtext=fontfile=c:/Windows/Fonts/msyh.ttf:text='hybase@qq.com视频水印2':x=50:y=650:fontsize=36:fontcolor=0xFFFF00@0.7:shadowy=-1'''
video_param = ' -map 0:0 -c:v:0 libx264 -s 1280x720 -b:v:0 1500000 -minrate 1500000 -maxrate 3000000 -bufsize 1500000 -g 25 -bf 1 -pix_fmt yuv420p -r:v:0 25.000000 '
aac_str = ' -map 0:1 -c:a:0 aac -ab:a:0 64k -ar:a:0 48000 -ac:a:0 2 -vol:a:0 256 '
ac3_str = ' -map 0:1 -c:a:1 ac3 -ab:a:1 128k -ar:a:1 48000 -ac:a:1 2 -vol:a:1 256 '
common_str = 'ffmpeg -y -fflags +genpts ' + log_param + input_ts_file + png_logo + jpg_logo + ' -max_muxing_queue_size 256 -filter_complex ' + overlay_str
cmd1 = common_str + video_param + aac_str + ac3_str + tsmux_str + vf_out + '/case000_overlay_logos.ts'
cmd2 = common_str + ',' + drawtext1 + drawtext2 + video_param + aac_str + ac3_str + tsmux_str + vf_out + '/case001_overlay_logo_drawtext.ts '
ffmpeg_run.ffmpeg_run('case_vf_overlay_logos' , overlay_flag, log, cmd1)
ffmpeg_run.ffmpeg_run('case_vf_overlay_drawtext', overlay_flag, log, cmd2)
# ***********************************
# gif or video format string
video_param_str = ' '
gif_str = ' -an -s 1280x720 -r 15 -pix_fmt rgb8 '
mpegts_str = ' -an -s 1280x720 -r 25 -pix_fmt yuv420p -vcodec libx264 -b 1500k -minrate 1500k -maxrate 2000k -bufsize 1500k -g 25 -bf 2 ' + hysmm_def.tsmux_vbr_str
if (sfx == '.gif'):
video_param_str = gif_str
else:
video_param_str = mpegts_str
log.logger.info(video_param_str)
for i in range(start_num, end_num):
idx = '%03d' %i
key = "case" + idx
input_file = input_mp4_file
output_file = vf_case_dict[key][0]
if (play_flag):
vf_str = ' -vf ' + vf_case_dict[key][1]
log.logger.info("key:%s, vf: %s, dst: %s", key, vf_str, output_file)
cmd = 'ffplay ' + log_param + input_file + vf_str
ffmpeg_run.ffplay_run(key, True, log, cmd)
else:
vf_str = ' -filter_complex ' + vf_case_dict[key][1]
log.logger.info("key:%s, vf: %s, dst: %s", key, vf_str, output_file)
cmd = 'ffmpeg -y ' + log_param + input_file + ' -t 10 ' + vf_str + video_param_str + output_file
ffmpeg_run.ffmpeg_run(key, True, log, cmd)
# frei0r cases please reference to "ffmpeg_frei0r.py"
## end of all cases
log.logger.info("=== all cases run ===")
log.logger.info("============ program exit ============")
sys.exit(0)
################################################################################
# end
#