mawei/StarPilot
1
0
Fork 0
mirror of https://github.com/firestar5683/StarPilot.git synced 2026-07-21 09:12:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

framereader single threaded env var

Browse Source 
old-commit-hash: 2d0686faa1876d65642f6c982a4352630a842676
This commit is contained in:
Greg Hogan
2020-05-28 13:39:28 -07:00
parent b8a63ef047
commit 5db5ecd0b9
1 changed files with 13 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/lib/framereader.py
+13 -7
View File
@@ -210,9 +210,12 @@ def rgb24toyuv420(rgb):
return yuv420.clip(0,255).astype('uint8')
def decompress_video_data(rawdat, vid_fmt, w, h, pix_fmt, multithreaded=False):
def decompress_video_data(rawdat, vid_fmt, w, h, pix_fmt, multithreaded=None):
# using a tempfile is much faster than proc.communicate for some reason
if multithreaded is None:
multithreaded = not os.getenv('SINGLETHREADED')
with tempfile.TemporaryFile() as tmpf:
tmpf.write(rawdat)
tmpf.seek(0)
@@ -263,7 +266,7 @@ class BaseFrameReader:
raise NotImplementedError
def FrameReader(fn, cache_prefix=None, readahead=False, readbehind=False, multithreaded=True, index_data=None):
def FrameReader(fn, cache_prefix=None, readahead=False, readbehind=False, multithreaded=None, index_data=None):
frame_type = fingerprint_video(fn)
if frame_type == FrameType.raw:
return RawFrameReader(fn)
@@ -322,12 +325,15 @@ class RawFrameReader(BaseFrameReader):
class VideoStreamDecompressor:
def __init__(self, vid_fmt, w, h, pix_fmt, multithreaded=False):
def __init__(self, vid_fmt, w, h, pix_fmt, multithreaded=None):
self.vid_fmt = vid_fmt
self.w = w
self.h = h
self.pix_fmt = pix_fmt
if multithreaded is None:
multithreaded = not os.getenv('SINGLETHREADED')
if pix_fmt == "yuv420p":
self.out_size = w*h*3//2 # yuv420p
elif pix_fmt in ("rgb24", "yuv444p"):
@@ -473,7 +479,7 @@ class StreamGOPReader(GOPReader):
class GOPFrameReader(BaseFrameReader):
#FrameReader with caching and readahead for formats that are group-of-picture based
def __init__(self, readahead=False, readbehind=False, multithreaded=True):
def __init__(self, readahead=False, readbehind=False, multithreaded=None):
self.open_ = True
self.multithreaded = multithreaded
@@ -567,12 +573,12 @@ class GOPFrameReader(BaseFrameReader):
class StreamFrameReader(StreamGOPReader, GOPFrameReader):
def __init__(self, fn, frame_type, index_data, readahead=False, readbehind=False, multithreaded=False):
def __init__(self, fn, frame_type, index_data, readahead=False, readbehind=False, multithreaded=None):
StreamGOPReader.__init__(self, fn, frame_type, index_data)
GOPFrameReader.__init__(self, readahead, readbehind, multithreaded)
def GOPFrameIterator(gop_reader, pix_fmt, multithreaded=True):
def GOPFrameIterator(gop_reader, pix_fmt, multithreaded=None):
# this is really ugly. ill think about how to refactor it when i can think good
IN_FLIGHT_GOPS = 6 # should be enough that the stream decompressor starts returning data
@@ -615,7 +621,7 @@ def GOPFrameIterator(gop_reader, pix_fmt, multithreaded=True):
def FrameIterator(fn, pix_fmt, **kwargs):
fr = FrameReader(fn, **kwargs)
if isinstance(fr, GOPReader):
for v in GOPFrameIterator(fr, pix_fmt, kwargs.get("multithreaded", True)):
for v in GOPFrameIterator(fr, pix_fmt, kwargs.get("multithreaded", None)):
yield v
else:
for i in range(fr.frame_count):