Browse Source

#11 made parser section more readable

zolfa-add_slices_loading
jim 10 years ago
parent
commit
5930c63e51
1 changed files with 31 additions and 30 deletions
  1. +31
    -30
      nd2reader/reader.py

+ 31
- 30
nd2reader/reader.py View File

@ -228,48 +228,49 @@ class Nd2FileReader(object):
def read_lv_encoding(self, data, count):
data = StringIO(data)
res = {}
metadata = {}
total_count = 0
for c in range(count):
lastpos = data.tell()
for _ in xrange(count):
cursor_position = data.tell()
total_count += 1
hdr = data.read(2)
if not hdr:
header = data.read(2)
if not header:
break
typ = ord(hdr[0])
bname = data.read(2*ord(hdr[1]))
name = bname.decode("utf16")[:-1].encode("utf8")
if typ == 1:
data_type, name_length = map(ord, header)
name = data.read(name_length * 2).decode("utf16")[:-1].encode("utf8")
if data_type == 1:
value, = struct.unpack("B", data.read(1))
elif typ in [2, 3]:
elif data_type in [2, 3]:
value, = struct.unpack("I", data.read(4))
elif typ == 5:
elif data_type == 5:
value, = struct.unpack("Q", data.read(8))
elif typ == 6:
elif data_type == 6:
value, = struct.unpack("d", data.read(8))
elif typ == 8:
elif data_type == 8:
value = data.read(2)
while value[-2:] != "\x00\x00":
value += data.read(2)
value = value.decode("utf16")[:-1].encode("utf8")
elif typ == 9:
elif data_type == 9:
cnt, = struct.unpack("Q", data.read(8))
value = array.array("B", data.read(cnt))
elif typ == 11:
newcount, length = struct.unpack("<IQ", data.read(12))
length -= data.tell()-lastpos
nextdata = data.read(length)
value = self.read_lv_encoding(nextdata, newcount)
elif data_type == 11:
new_count, length = struct.unpack("<IQ", data.read(12))
length -= data.tell() - cursor_position
next_data_length = data.read(length)
value = self.read_lv_encoding(next_data_length, new_count)
# Skip some offsets
data.read(newcount * 8)
data.read(new_count * 8)
if name not in metadata:
metadata[name] = value
else:
assert 0, "%s hdr %x:%x unknown" % (name, ord(hdr[0]), ord(hdr[1]))
if not name in res:
res[name] = value
else:
if not isinstance(res[name], list):
res[name] = [res[name]]
res[name].append(value)
x = data.read()
assert not x, "skip %d %s" % (len(x), repr(x[:30]))
return res
if not isinstance(metadata[name], list):
# We have encountered this key exactly once before. Since we're seeing it again, we know we
# need to convert it to a list before proceeding.
metadata[name] = [metadata[name]]
# We've encountered this key before so we're guaranteed to be dealing with a list. Thus we append
# the value to the already-existing list.
metadata[name].append(value)
# x = data.read()
# assert not x, "skip %d %s" % (len(x), repr(x[:30]))
return metadata

Loading…
Cancel
Save