55 lines
2.7 KiB
Python
55 lines
2.7 KiB
Python
from mkv.chapter import get_intro_outro
|
|
from audio.audio_utils import extract_audio_segment_tmp, get_audio_streams
|
|
from audio.audio_cross_correlation import get_audio_offset
|
|
from pathlib import Path
|
|
import numpy as np
|
|
from typing import Tuple
|
|
|
|
def check_intro_outro_sync(file: Path, length: float = 90.0, uncertainty: float = 0.1, max_offset: float = 0.1) -> Tuple[bool, float|None, int|None]:
|
|
intro_outro = get_intro_outro(file, length)
|
|
if intro_outro[0][0]<intro_outro[1][0]:
|
|
intro = intro_outro[0]
|
|
outro = intro_outro[1]
|
|
else:
|
|
intro = intro_outro[1]
|
|
outro = intro_outro[0]
|
|
|
|
### Cut away any possible part that is not of the intro/outro
|
|
intro = (intro[0]+intro[1]*(uncertainty/2), intro[1]-intro[1]*uncertainty)
|
|
outro = (outro[0]+outro[1]*(uncertainty/2), outro[1]-outro[1]*uncertainty)
|
|
|
|
### Get the intro and outro Segments
|
|
audio_streams = get_audio_streams(file)
|
|
intros : list[Path] = []
|
|
outros : list[Path] = []
|
|
# for i in audio_streams:
|
|
# intros.append(extract_audio_segment_tmp(file, intro, None, i))
|
|
# outros.append(extract_audio_segment_tmp(file, outro, None, i))
|
|
intros.append(extract_audio_segment_tmp(file, intro, None, audio_streams[0]))
|
|
outros.append(extract_audio_segment_tmp(file, outro, None, audio_streams[0]))
|
|
intros.append(extract_audio_segment_tmp(file, intro, None, audio_streams[-1]))
|
|
outros.append(extract_audio_segment_tmp(file, outro, None, audio_streams[-1]))
|
|
|
|
### correlate
|
|
intro_offsets = []
|
|
outro_offsets = []
|
|
for intro_a, outro_a in zip(intros, outros):
|
|
idx = intros.index(intro_a)
|
|
odx = outros.index(outro_a)
|
|
for intro_b, outro_b in zip(intros[idx+1:], outros[odx+1:]):
|
|
intro_offset = get_audio_offset(intro_a, intro_b, intro[1])
|
|
intro_offsets.append(intro_offset)
|
|
# print(f"intro_offset between Audioline {intros.index(intro_a)} and {intros.index(intro_b)} is {intro_offset}")
|
|
outro_offset = get_audio_offset(outro_a, outro_b, outro[1])
|
|
outro_offsets.append(outro_offset)
|
|
# print(f"outro_offset between Audioline {outros.index(outro_a)} and {outros.index(outro_b)} is {outro_offset}")
|
|
# print(f"Offset of offsets: {(outro_offset - intro_offset)}")
|
|
mean_offset = float(np.mean([intro_offsets[0], outro_offsets[0]]))
|
|
|
|
# print(f"Mean Intro Offset: {round(np.mean(intro_offsets), 1)}")
|
|
# print(f"Mean Outro Offset: {round(np.mean(outro_offsets), 1)}")
|
|
# print(f"Offset of mean offsets: {round(np.mean(outro_offsets) - np.mean(intro_offsets)), 3}")
|
|
|
|
if np.abs(np.mean(outro_offsets) - np.mean(intro_offsets)) < max_offset:
|
|
return True, mean_offset, audio_streams[-1]
|
|
return False, None, None |