yttt/EncodeScripts
5
0
Fork 0
Code Releases Activity

yt_common: source: cache opened files

Browse Source
This commit is contained in:
louis f 2021-06-05 20:11:54 -04:00
parent 62c8355fb1
commit f1bc1f8177
Signed by: louis
GPG Key ID: 44D7E1DE4E23D6F2
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
yt_common/yt_common/source.py
View File

@ -10,7 +10,7 @@ import glob
import os import os
from abc import ABC, abstractmethod from abc import ABC, abstractmethod
from typing import Any, List, NamedTuple, Optional, Tuple, Union from typing import Any, Dict, List, NamedTuple, Optional, Tuple, Union
from .audio import AudioStream, CodecPassthrough from .audio import AudioStream, CodecPassthrough
from .config import Config from .config import Config
@ -63,10 +63,18 @@ def glob_filename(pattern: str) -> str:
class FileSource(ABC): class FileSource(ABC):
nodes: Dict[str, vs.VideoNode]
def __init__(self) -> None:
self.nodes = {}
def _open(self, path: str) -> vs.VideoNode: def _open(self, path: str) -> vs.VideoNode:
return depth(core.lsmas.LWLibavSource(path), 16) if path.lower().endswith(".m2ts") \ if path in self.nodes:
return self.nodes[path]
self.nodes[path] = depth(core.lsmas.LWLibavSource(path), 16) if path.lower().endswith(".m2ts") \
else depth(core.d2v.Source(path), 16) if path.lower().endswith(".d2v") \ else depth(core.d2v.Source(path), 16) if path.lower().endswith(".d2v") \
else depth(core.ffms2.Source(path), 16) else depth(core.ffms2.Source(path), 16)
return self.nodes[path]
def audio_ref(self) -> Optional[vs.VideoNode]: def audio_ref(self) -> Optional[vs.VideoNode]:
return None return None
@ -97,6 +105,7 @@ class SimpleSource(FileSource):
self.sclip = None self.sclip = None
self.aref = aref self.aref = aref
self.asrc = asrc if isinstance(asrc, list) else [asrc] if asrc is not None else None self.asrc = asrc if isinstance(asrc, list) else [asrc] if asrc is not None else None
super().__init__()
def audio_ref(self) -> Optional[vs.VideoNode]: def audio_ref(self) -> Optional[vs.VideoNode]:
return self.aref return self.aref
@ -116,6 +125,7 @@ class DehardsubFileFinder(FileSource):
def __init__(self, config: Config) -> None: def __init__(self, config: Config) -> None:
self.config = config self.config = config
super().__init__()
@abstractmethod @abstractmethod
def get_waka_filenames(self) -> List[str]: def get_waka_filenames(self) -> List[str]: