mirror of
https://gitea.pjck.top/Cookies/CookiesChartConverter.git
synced 2025-12-14 12:56:54 +08:00
58 lines
1.9 KiB
Python
58 lines
1.9 KiB
Python
import os
|
||
import subprocess
|
||
import sys
|
||
|
||
def get_video_resolution(video_path):
|
||
"""获取视频的宽高信息"""
|
||
cmd = [
|
||
"ffprobe", "-v", "error", "-select_streams", "v:0", "-show_entries",
|
||
"stream=width,height", "-of", "csv=p=0", video_path
|
||
]
|
||
result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
|
||
width, height = map(int, result.stdout.strip().split(","))
|
||
return width, height
|
||
|
||
def calculate_padding(width, height):
|
||
"""计算填充黑边，使其变成 1:1（正方形）"""
|
||
max_side = max(width, height) # 以最大边长作为正方形边长
|
||
pad_x = (max_side - width) // 2
|
||
pad_y = (max_side - height) // 2
|
||
return max_side, pad_x, pad_y
|
||
|
||
def process_video(input_file, output_file):
|
||
"""填充黑色边框，调整为 1:1，并转换为 H.264"""
|
||
width, height = get_video_resolution(input_file)
|
||
target_size, pad_x, pad_y = calculate_padding(width, height)
|
||
|
||
if pad_x == 0 and pad_y == 0:
|
||
print("视频已经是 1:1，无需填充。")
|
||
ffmpeg_cmd = [
|
||
"ffmpeg", "-i", input_file,
|
||
"-c:v", "h264_videotoolbox",
|
||
"-b:v", "50M", "-maxrate", "70M", "-bufsize", "100M",
|
||
"-c:a", "aac", "-b:a", "320k",
|
||
output_file
|
||
]
|
||
else:
|
||
print(f"填充黑边，使视频变为 {target_size}x{target_size}")
|
||
ffmpeg_cmd = [
|
||
"ffmpeg", "-i", input_file,
|
||
"-vf", f"pad={target_size}:{target_size}:{pad_x}:{pad_y}:black",
|
||
"-c:v", "h264_videotoolbox",
|
||
"-b:v", "50M", "-maxrate", "70M", "-bufsize", "100M",
|
||
"-c:a", "aac", "-b:a", "320k",
|
||
output_file
|
||
]
|
||
|
||
subprocess.run(ffmpeg_cmd)
|
||
|
||
if __name__ == "__main__":
|
||
if len(sys.argv) < 3:
|
||
print("用法: python pv_convert.py 输入文件 输出文件")
|
||
sys.exit(1)
|
||
|
||
input_file = sys.argv[1]
|
||
output_file = sys.argv[2]
|
||
|
||
process_video(input_file, output_file)
|