def __init__(self, filename, grey=False): if sys.version_info > (3,): filename = bytes(filename, "utf8") self.filename = filename self.grey = grey open(filename).close() self._myiter = None self._index = None self.systime_offset = 0 idx = index_file(self.filename) if os.path.exists(idx): index = json.load(open(idx)) assert index['version'] == 2 self.frame = index['frame'] self.systime_offset = index['systime_offset'] else: self.frame = [] m = lib.mkv_open(self.filename) frm = ffi.new('struct mkv_frame *') cluster_offsets = set() while lib.mkv_next(m, frm): if frm.key_frame: if frm.offset in cluster_offsets: print("FIXME: Multiple keyframes per cluster!") cluster_offsets.add(frm.key_frame) self.frame.append([frm.pts, frm.offset, frm.key_frame]) self.frame.sort() self.systime_offset = iter(self).estimate_systime_offset() # FIXME: This makes a second pass over the file parsing it lib.mkv_close(m) with open(idx, 'w') as fd: json.dump({'frame': self.frame, 'systime_offset': self.systime_offset, 'version': 2}, fd)
def __init__(self, filename, grey=False, reindex=False): # Be compatible with pathlib.Path filenames filename = str(filename).encode('utf-8') self.filename = filename self.grey = grey open(filename).close() self._myiter = None self.systime_offset = 0 need_index = True idx = index_file(self.filename) if os.path.exists(idx): try: index = json.load(open(idx)) except Exception: pass else: if index['version'] == INDEX_VERSION: self.frame = index['frame'] self.systime_offset = index['systime_offset'] self.mjpg_mode = index['mjpg_mode'] need_index = False if need_index or reindex: self.frame = [] m = lib.mkv_open(self.filename) frm = ffi.new('struct mkv_frame *') cluster_offsets = set() while lib.mkv_next(m, frm): if frm.key_frame: if frm.offset in cluster_offsets: print("FIXME: Multiple keyframes per cluster!") cluster_offsets.add(frm.key_frame) self.frame.append([frm.pts, frm.offset, frm.key_frame]) self.frame.sort() self.systime_offset = iter(self).estimate_systime_offset() # FIXME: This makes a second pass over the file parsing it self.mjpg_mode = (ffi.string(m.codec_id) == b'V_MS/VFW/FOURCC') lib.mkv_close(m) tmp = idx + '.tmp.%d' % os.getpid() with open(tmp, 'w') as fd: json.dump({'frame': self.frame, 'systime_offset': self.systime_offset, 'mjpg_mode': self.mjpg_mode, 'version': INDEX_VERSION}, fd) try: os.link(tmp, idx) except FileExistsError: pass os.unlink(tmp)
def __init__(self, filename, grey=False, reindex=False): if sys.version_info > (3, ): filename = bytes(filename, "utf8") self.filename = filename self.grey = grey open(filename).close() self._myiter = None self.systime_offset = 0 need_index = True idx = index_file(self.filename) if os.path.exists(idx): index = json.load(open(idx)) if index['version'] == INDEX_VERSION: self.frame = index['frame'] self.systime_offset = index['systime_offset'] self.mjpg_mode = index['mjpg_mode'] need_index = False if need_index or reindex: self.frame = [] m = lib.mkv_open(self.filename) frm = ffi.new('struct mkv_frame *') cluster_offsets = set() while lib.mkv_next(m, frm): if frm.key_frame: if frm.offset in cluster_offsets: print("FIXME: Multiple keyframes per cluster!") cluster_offsets.add(frm.key_frame) self.frame.append([frm.pts, frm.offset, frm.key_frame]) self.frame.sort() self.systime_offset = iter(self).estimate_systime_offset( ) # FIXME: This makes a second pass over the file parsing it self.mjpg_mode = (ffi.string(m.codec_id) == b'V_MS/VFW/FOURCC') lib.mkv_close(m) with open(idx, 'w') as fd: json.dump( { 'frame': self.frame, 'systime_offset': self.systime_offset, 'mjpg_mode': self.mjpg_mode, 'version': INDEX_VERSION }, fd)
def __init__(self, filename, grey=False): if sys.version_info > (3, ): filename = bytes(filename, "utf8") self.filename = filename self.grey = grey open(filename).close() self._myiter = None self._index = None self.systime_offset = 0 idx = index_file(self.filename) if os.path.exists(idx): index = json.load(open(idx)) assert index['version'] == 2 self.frame = index['frame'] self.systime_offset = index['systime_offset'] else: self.frame = [] m = lib.mkv_open(self.filename) frm = ffi.new('struct mkv_frame *') cluster_offsets = set() while lib.mkv_next(m, frm): if frm.key_frame: if frm.offset in cluster_offsets: print("FIXME: Multiple keyframes per cluster!") cluster_offsets.add(frm.key_frame) self.frame.append([frm.pts, frm.offset, frm.key_frame]) self.frame.sort() self.systime_offset = iter(self).estimate_systime_offset( ) # FIXME: This makes a second pass over the file parsing it lib.mkv_close(m) with open(idx, 'w') as fd: json.dump( { 'frame': self.frame, 'systime_offset': self.systime_offset, 'version': 2 }, fd)
def __del__(self): del self.decoder lib.mkv_close(self.m)
def __del__(self): with decode_open_lock: self.p = lib.decode_close(self.p) lib.mkv_close(self.m)
def __del__(self): try: del self.decoder except AttributeError: pass lib.mkv_close(self.m)