如何解决MoviePy制作pyqt5进度条
使用moviepy模块,我希望有一个回调函数来显示%音频导出。
我已经完成了模块中的更改:
a)调用audio.write_audiofile("../disket Box/mp3 files/"+title+".mp3",progress_function=self.update_progress_bar_local,verbose=True,logger=None)
(添加progress_function参数)
b)文件python38 \ Lib \ site-packages \ moviepy \ audio \ AudioClip.py行
def write_audiofile(self,filename,fps=None,nbytes=2,buffersize=2000,codec=None,bitrate=None,ffmpeg_params=None,write_logfile=False,logger='bar',progress_function=None):
c)与b)第206行return ffmpeg_audiowrite(self,fps,nbytes,buffersize,codec=codec,bitrate=bitrate,write_logfile=write_logfile,verbose=verbose,ffmpeg_params=ffmpeg_params,logger=logger,progress_function=progress_function)
d)文件python38 \ lib \ site-packages \ moviepy \ audio \ io \ ffmpeg_audiowriter.py第144行-结束
def ffmpeg_audiowrite(clip,codec='libvorbis',progress_function=None):
"""
A function that wraps the FFMPEG_AudioWriter to write an AudioClip
to a file.
NOTE: verbose is deprecated.
"""
if write_logfile:
logfile = open(filename + ".log",'w+')
else:
logfile = None
logger = proglog.default_bar_logger(logger)
logger(message="MoviePy - Writing audio in %s" % filename)
writer = FFMPEG_AudioWriter(filename,clip.nchannels,logfile=logfile,ffmpeg_params=ffmpeg_params)
all_chunks = clip.iter_chunks(chunksize=buffersize,quantize=True,nbytes=nbytes,fps=fps,logger=logger)
counter = 0
total_chunks = "?????"
for chunk in all_chunks:
counter = counter + 1
writer.write_frames(chunk)
progress_function(str(counter),str(total_chunks))
writer.close()
if write_logfile:
logfile.close()
logger(message="MoviePy - Done.")
好的,进度函数可以工作,但是我不知道如何计算total_chunks。我尝试了len(all_chunks),但失败了,因为all_chunks不是列表(它是生成器)。
关于如何计算total_chunks的任何建议都是有用的。
预先感谢, 克里斯·帕帕斯(Chris Pappas)
解决方法
我修改了\ Python38 \ Lib \ site-packages \ moviepy \ audio \ io \ AudioFileClip.py
我添加了一个新方法:
def len(self):
return self.duration
现在,Python38 \ Lib \ site-packages \ moviepy \ audio \ io \ ffmpeg_audiowriter.py如下:
import os
import subprocess as sp
import proglog
from moviepy.compat import DEVNULL
from moviepy.config import get_setting
from moviepy.decorators import requires_duration
import math
class FFMPEG_AudioWriter:
"""
A class to write an AudioClip into an audio file.
Parameters
------------
filename
Name of any video or audio file,like ``video.mp4`` or ``sound.wav`` etc.
size
Size (width,height) in pixels of the output video.
fps_input
Frames per second of the input audio (given by the AUdioClip being
written down).
codec
Name of the ffmpeg codec to use for the output.
bitrate:
A string indicating the bitrate of the final video. Only
relevant for codecs which accept a bitrate.
"""
def __init__(self,filename,fps_input,nbytes=2,nchannels=2,codec='libfdk_aac',bitrate=None,input_video=None,logfile=None,ffmpeg_params=None):
self.filename = filename
self.codec = codec
if logfile is None:
logfile = sp.PIPE
cmd = ([get_setting("FFMPEG_BINARY"),'-y',"-loglevel","error" if logfile == sp.PIPE else "info","-f",'s%dle' % (8*nbytes),"-acodec",'pcm_s%dle' % (8*nbytes),'-ar',"%d" % fps_input,'-ac',"%d" % nchannels,'-i','-']
+ (['-vn'] if input_video is None else ["-i",input_video,'-vcodec','copy'])
+ ['-acodec',codec]
+ ['-ar',"%d" % fps_input]
+ ['-strict','-2'] # needed to support codec 'aac'
+ (['-ab',bitrate] if (bitrate is not None) else [])
+ (ffmpeg_params if ffmpeg_params else [])
+ [filename])
popen_params = {"stdout": DEVNULL,"stderr": logfile,"stdin": sp.PIPE}
if os.name == "nt":
popen_params["creationflags"] = 0x08000000
self.proc = sp.Popen(cmd,**popen_params)
def write_frames(self,frames_array):
try:
try:
self.proc.stdin.write(frames_array.tobytes())
except NameError:
self.proc.stdin.write(frames_array.tostring())
except IOError as err:
ffmpeg_error = self.proc.stderr.read()
error = (str(err) + ("\n\nMoviePy error: FFMPEG encountered "
"the following error while writing file %s:" % self.filename
+ "\n\n" + str(ffmpeg_error)))
if b"Unknown encoder" in ffmpeg_error:
error = (error +
("\n\nThe audio export failed because FFMPEG didn't "
"find the specified codec for audio encoding (%s). "
"Please install this codec or change the codec when "
"calling to_videofile or to_audiofile. For instance "
"for mp3:\n"
" >>> to_videofile('myvid.mp4',audio_codec='libmp3lame')"
) % (self.codec))
elif b"incorrect codec parameters ?" in ffmpeg_error:
error = (error +
("\n\nThe audio export failed,possibly because the "
"codec specified for the video (%s) is not compatible"
" with the given extension (%s). Please specify a "
"valid 'codec' argument in to_videofile. This would "
"be 'libmp3lame' for mp3,'libvorbis' for ogg...")
% (self.codec,self.ext))
elif b"encoder setup failed" in ffmpeg_error:
error = (error +
("\n\nThe audio export failed,possily because the "
"bitrate you specified was two high or too low for "
"the video codec."))
else:
error = (error +
("\n\nIn case it helps,make sure you are using a "
"recent version of FFMPEG (the versions in the "
"Ubuntu/Debian repos are deprecated)."))
raise IOError(error)
def close(self):
if hasattr(self,'proc') and self.proc:
self.proc.stdin.close()
self.proc.stdin = None
if self.proc.stderr is not None:
self.proc.stderr.close()
self.proc.stdee = None
# If this causes deadlocks,consider terminating instead.
self.proc.wait()
self.proc = None
def __del__(self):
# If the garbage collector comes,make sure the subprocess is terminated.
self.close()
# Support the Context Manager protocol,to ensure that resources are cleaned up.
def __enter__(self):
return self
def __exit__(self,exc_type,exc_value,traceback):
self.close()
@requires_duration
def ffmpeg_audiowrite(clip,fps,nbytes,buffersize,codec='libvorbis',write_logfile=False,verbose=True,ffmpeg_params=None,logger='bar',progress_function=None):
"""
A function that wraps the FFMPEG_AudioWriter to write an AudioClip
to a file.
NOTE: verbose is deprecated.
"""
if write_logfile:
logfile = open(filename + ".log",'w+')
else:
logfile = None
logger = proglog.default_bar_logger(logger)
logger(message="MoviePy - Writing audio in %s" % filename)
writer = FFMPEG_AudioWriter(filename,clip.nchannels,codec=codec,bitrate=bitrate,logfile=logfile,ffmpeg_params=ffmpeg_params)
all_chunks = clip.iter_chunks(chunksize=buffersize,quantize=True,nbytes=nbytes,fps=fps,logger=logger)
counter = 0
total_chunks = math.ceil(fps*clip.len()/buffersize)
print(clip.len())
for chunk in all_chunks:
counter = counter + 1
writer.write_frames(chunk)
progress_function(str(counter),str(total_chunks))
writer.close()
if write_logfile:
logfile.close()
logger(message="MoviePy - Done.")
total_chunks与最终计数器相同!
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。