jamie/wavinfo
1
0
Fork 0
mirror of https://github.com/iluvcapra/wavinfo.git synced 2025-12-31 08:50:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improved exceptions in certain EOF cases

Browse Source 
Pursuant to #4
This commit is contained in:
Jamie Hardt
2020-01-02 10:50:36 -08:00
parent 15db4c9ffa
commit 60e329fdb4
2 changed files with 27 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
wavinfo/__init__.py
View File

@@ -5,7 +5,8 @@ Go to the documentation for wavinfo.WavInfoReader for more information.
"""
from .wave_reader import WavInfoReader
from .riff_parser import WavInfoEOFError
__version__ = '1.3'
__version__ = '1.3.1'
__author__ = 'Jamie Hardt <jamiehardt@gmail.com>'
__license__ = "MIT"

38
wavinfo/riff_parser.py
View File

@@ -1,12 +1,17 @@
import struct
import pdb
from collections import namedtuple
from .rf64_parser import parse_rf64
class ListChunkDescriptor(namedtuple('ListChunkDescriptor' , 'signature children')):
def find(chunk_path):
class WavInfoEOFError(EOFError):
def __init__(self, identifier, chunk_start):
self.identifier = identifier
self.chunk_start = chunk_start
class ListChunkDescriptor(namedtuple('ListChunkDescriptor', 'signature children')):
def find(self, chunk_path):
if len(chunk_path) > 1:
for chunk in self.children:
if type(chunk) is ListChunkDescriptor and \
@@ -19,20 +24,20 @@ class ListChunkDescriptor(namedtuple('ListChunkDescriptor' , 'signature children
return chunk
class ChunkDescriptor(namedtuple('ChunkDescriptor', 'ident start length rf64_context') ):
class ChunkDescriptor(namedtuple('ChunkDescriptor', 'ident start length rf64_context')):
def read_data(self, from_stream):
from_stream.seek(self.start)
return from_stream.read(self.length)
def parse_list_chunk(stream, length, rf64_context =None):
def parse_list_chunk(stream, length, rf64_context=None):
start = stream.tell()
signature = stream.read(4)
#print("Parsing list chunk with siganture: ", signature)
children = []
while (stream.tell() - start) < length:
child_chunk = parse_chunk(stream, rf64_context= rf64_context)
child_chunk = parse_chunk(stream, rf64_context=rf64_context)
if child_chunk:
children.append(child_chunk)
else:
@@ -40,13 +45,16 @@ def parse_list_chunk(stream, length, rf64_context =None):
return ListChunkDescriptor(signature=signature, children=children)
def parse_chunk(stream, rf64_context=None):
ident = stream.read(4)
if len(ident) != 4:
return
size_bytes = stream.read(4)
sizeb = stream.read(4)
size = struct.unpack('<I',sizeb)[0]
if len(ident) != 4 or len(size_bytes) < 4:
raise WavInfoEOFError(identifier=ident, chunk_start=stream.tell() - 4)
size_bytes = stream.read(4)
size = struct.unpack('<I', size_bytes)[0]
if size == 0xFFFFFFFF:
if rf64_context is None and ident == b'RF64':
@@ -58,13 +66,9 @@ def parse_chunk(stream, rf64_context=None):
if displacement % 2 is not 0:
displacement = displacement + 1
if ident in [b'RIFF',b'LIST', b'RF64']:
#print("Parsing list chunk with ident: ", ident)
if ident in [b'RIFF', b'LIST', b'RF64']:
return parse_list_chunk(stream=stream, length=size, rf64_context=rf64_context)
else:
start = stream.tell()
stream.seek(displacement,1)
#print("Parsing chunk with start=%i, ident=%s" % (start, ident))
stream.seek(displacement, 1)
return ChunkDescriptor(ident=ident, start=start, length=size, rf64_context=rf64_context)