mantarimay/Aegisub
1
0
Fork 0
forked from mia/Aegisub
Code Pull requests Activity

vapoursynth: Fix lwindex parsing for files with multiple video streams

Browse source
This commit is contained in:
arch1t3cht 2023-11-01 20:32:32 +01:00
parent 2dbee37ad8
commit 0d281af269
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
automation/vapoursynth/aegisub_vs.py
View file

@ -127,12 +127,14 @@ streaminfo_re = re.compile(r"Codec=(?P<Codec>[0-9]+),TimeBase=(?P<TimeBase>[0-9\
class LWIndexFrame: class LWIndexFrame:
pts: int pts: int
key: int key: int
index: int
def __init__(self, raw: list[str]): def __init__(self, raw: list[str]):
match1 = lwindex_re1.match(raw[0]) match1 = lwindex_re1.match(raw[0])
match2 = lwindex_re2.match(raw[1]) match2 = lwindex_re2.match(raw[1])
if not match1 or not match2: if not match1 or not match2:
raise ValueError("Invalid lwindex format") raise ValueError("Invalid lwindex format")
self.index = int(match1.group("Index"))
self.pts = int(match1.group("PTS")) self.pts = int(match1.group("PTS"))
self.key = int(match2.group("Key")) self.key = int(match2.group("Key"))
@ -150,11 +152,13 @@ def info_from_lwindex(indexfile: str) -> Dict[str, List[int]]:
with open(indexfile, encoding="latin1") as f: with open(indexfile, encoding="latin1") as f:
index = f.read().splitlines() index = f.read().splitlines()
indexstart, indexend = index.index("</StreamInfo>") + 1, index.index("</LibavReaderIndex>") # The picture list starts after the last </StreamInfo> tag
indexstart, indexend = (len(index) - index[::-1].index("</StreamInfo>")), index.index("</LibavReaderIndex>")
frames = [LWIndexFrame(index[i:i+2]) for i in range(indexstart, indexend, 2)] frames = [LWIndexFrame(index[i:i+2]) for i in range(indexstart, indexend, 2)]
frames = [f for f in frames if f.index == 0] # select the first stream
frames.sort(key=int) frames.sort(key=int)
streaminfo = streaminfo_re.match(index[indexstart - 2]) streaminfo = streaminfo_re.match(index[index.index("<StreamInfo=0,0>") + 1]) # info of first stream
if not streaminfo: if not streaminfo:
raise ValueError("Invalid lwindex format") raise ValueError("Invalid lwindex format")