yttt/EncodeScripts
5
0
Fork 0
Code Releases Activity

yt_common: automation: allow processing only audio

Browse Source 
for when x265 segfaults and you need to do some appends
This commit is contained in:
louis f 2021-05-13 09:28:12 -04:00
parent 60e6ff8563
commit da10d6c379
Signed by: louis
GPG Key ID: 44D7E1DE4E23D6F2
1 changed files with 26 additions and 11 deletions

37
yt_common/yt_common/automation.py

@ -195,6 +195,7 @@ class SelfRunner():
parser.add_argument("-c", "--comparison", help="Output a comparison between workraw and final. \ parser.add_argument("-c", "--comparison", help="Output a comparison between workraw and final. \
Will search for the output file to include in comparison, if present.", Will search for the output file to include in comparison, if present.",
action="store_true") action="store_true")
parser.add_argument("-a", "--audio-only", help="Only process audio, no video.", action="store_true")
args = parser.parse_args() args = parser.parse_args()
self.workraw = args.workraw if workraw_filter else False self.workraw = args.workraw if workraw_filter else False
@ -225,7 +226,15 @@ class SelfRunner():
if start >= end: if start >= end:
raise ValueError("Start frame must be before end frame!") raise ValueError("Start frame must be before end frame!")
out_name = f"{self.config.title.lower()}_{self.config.desc}_{self.suffix}.mkv" out_name = f"{self.config.title.lower()}_{self.config.desc}_{self.suffix}"
if args.audio_only:
out_name += ".mka"
self._do_audio(start, end, audio_codec, out_name=out_name)
self.audio.do_cleanup()
return
out_name += ".mkv"
if args.comparison and workraw_filter: if args.comparison and workraw_filter:
log.status("Generating comparison...") log.status("Generating comparison...")
@ -247,18 +256,11 @@ class SelfRunner():
self.video_file = self.encoder.encode(self.clip, f"{self.config.desc}_{self.suffix}_{start}_{end}", self.video_file = self.encoder.encode(self.clip, f"{self.config.desc}_{self.suffix}_{start}_{end}",
start, end) start, end)
log.status("--- LOOKING FOR AUDIO ---") self._do_audio(start, end, audio_codec)
self.audio = AudioGetter(self.config, self.src)
log.status("--- TRIMMING AUDIO ---")
self.audio_file = self.audio.trim_audio((start, end))
if audio_codec:
log.status("--- TRANSCODING AUDIO ---")
self.audio_file = self.audio.encode_audio(self.audio_file, audio_codec)
try: try:
log.status("--- MUXING FILE ---") log.status("--- MUXING FILE ---")
if self._mux(out_name, start == 0 and end == self.clip.num_frames) != 0: if self._do_mux(out_name, start == 0 and end == self.clip.num_frames) != 0:
raise Exception("mkvmerge failed") raise Exception("mkvmerge failed")
except Exception: except Exception:
log.error("--- MUXING FAILED ---") log.error("--- MUXING FAILED ---")
@ -274,7 +276,20 @@ class SelfRunner():
log.success("--- ENCODE COMPLETE ---") log.success("--- ENCODE COMPLETE ---")
def _mux(self, name: str, chapters: bool = True) -> int: def _do_audio(self, start: int, end: int, codec: Optional[List[str]], out_name: Optional[str] = None) -> None:
log.status("--- LOOKING FOR AUDIO ---")
self.audio = AudioGetter(self.config, self.src)
log.status("--- TRIMMING AUDIO ---")
self.audio_file = self.audio.trim_audio((start, end))
if codec:
log.status("--- TRANSCODING AUDIO ---")
self.audio_file = self.audio.encode_audio(self.audio_file, codec)
if out_name:
shutil.copy(self.audio_file, out_name)
self.audio_file = out_name
def _do_mux(self, name: str, chapters: bool = True) -> int:
mkvtoolnix_args = [ mkvtoolnix_args = [
"mkvmerge", "mkvmerge",
"--output", name, "--output", name,