/usr/share/pyshared/dicom/fileutil.py is in python-dicom 0.9.6-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 | # fileutil.py
"""Functions for reading to certain bytes, e.g. delimiters"""
# Copyright (c) 2009 Darcy Mason
# This file is part of pydicom, released under a modified MIT license.
# See the file license.txt included with this distribution, also
# available at http://pydicom.googlecode.com
from struct import pack, unpack
from dicom.tag import TupleTag, Tag
from dicom.datadict import dictionaryDescription
import logging
logger = logging.getLogger('pydicom')
def absorb_delimiter_item(fp, is_little_endian, delimiter):
"""Read (and ignore) undefined length sequence or item terminators."""
if is_little_endian:
struct_format = "<HHL"
else:
struct_format = ">HHL"
group, elem, length = unpack(struct_format, fp.read(8))
tag = TupleTag((group, elem))
if tag != delimiter:
msg = "Did not find expected delimiter '%s'" % dictionaryDescription(delimiter)
msg += ", instead found %s at file position 0x%x" %(str(tag), fp.tell()-8)
logger.warn(msg)
fp.seek(fp.tell()-8)
return
logger.debug("%04x: Found Delimiter '%s'", fp.tell()-8, dictionaryDescription(delimiter))
if length == 0:
logger.debug("%04x: Read 0 bytes after delimiter", fp.tell()-4)
else:
logger.debug("%04x: Expected 0x00000000 after delimiter, found 0x%x", fp.tell()-4, length)
def find_bytes(fp, bytes_to_find, read_size=128, rewind=True):
"""Read in the file until a specific byte sequence found
bytes_to_find -- a string containing the bytes to find. Must be in correct
endian order already
read_size -- number of bytes to read at a time
"""
data_start = fp.tell()
search_rewind = len(bytes_to_find)-1
found = False
EOF = False
while not found:
chunk_start = fp.tell()
bytes = fp.read(read_size)
if len(bytes) < read_size:
# try again - if still don't get required amount, this is last block
new_bytes = fp.read(read_size - len(bytes))
bytes += new_bytes
if len(bytes) < read_size:
EOF = True # but will still check whatever we did get
index = bytes.find(bytes_to_find)
if index != -1:
found = True
elif EOF:
if rewind:
fp.seek(data_start)
return None
else:
fp.seek(fp.tell()-search_rewind) # rewind a bit in case delimiter crossed read_size boundary
# if get here then have found the byte string
found_at = chunk_start + index
if rewind:
fp.seek(data_start)
return found_at
def read_undefined_length_value(fp, is_little_endian, delimiter_tag, defer_size=None,
read_size=128):
"""Read until the delimiter tag found and return the value, ignore the delimiter
fp -- a file-like object with read(), seek() functions
is_little_endian -- True if file transfer syntax is little endian, else False
read_size -- number of bytes to read at one time (default 128)
On completion, the file will be set to the first byte after the delimiter and its
following four zero bytes.
If end-of-file is hit before the delimiter was found, raises EOFError
"""
data_start = fp.tell()
search_rewind = 3
if is_little_endian:
bytes_format = "<HH"
else:
bytes_format = ">HH"
bytes_to_find = pack(bytes_format, delimiter_tag.group, delimiter_tag.elem)
found = False
EOF = False
value_chunks = []
byte_count = 0 # for defer_size checks
while not found:
chunk_start = fp.tell()
bytes = fp.read(read_size)
if len(bytes) < read_size:
# try again - if still don't get required amount, this is last block
new_bytes = fp.read(read_size - len(bytes))
bytes += new_bytes
if len(bytes) < read_size:
EOF = True # but will still check whatever we did get
index = bytes.find(bytes_to_find)
if index != -1:
found = True
new_bytes = bytes[:index]
byte_count += len(new_bytes)
if defer_size is None or byte_count < defer_size:
value_chunks.append(bytes[:index])
fp.seek(chunk_start + index + 4) # rewind to end of delimiter
length = fp.read(4)
if length != "\0\0\0\0":
msg = "Expected 4 zero bytes after undefined length delimiter at pos %x"
logger.error(msg % (fp.tell()-4,))
elif EOF:
fp.seek(data_start)
raise EOFError, "End of file reached before delimiter %r found" % delimiter_tag
else:
fp.seek(fp.tell()-search_rewind) # rewind a bit in case delimiter crossed read_size boundary
# accumulate the bytes read (not including the rewind)
new_bytes = bytes[:-search_rewind]
byte_count += len(new_bytes)
if defer_size is None or byte_count < defer_size:
value_chunks.append(new_bytes)
# if get here then have found the byte string
if defer_size is not None and defer_size >= defer_size:
return None
else:
return "".join(value_chunks)
def find_delimiter(fp, delimiter, is_little_endian, read_size=128, rewind=True):
"""Return file position where 4-byte delimiter is located.
Return None if reach end of file without finding the delimiter.
On return, file position will be where it was before this function,
unless rewind argument is False.
"""
struct_format = "<H"
if not is_little_endian:
struct_format = ">H"
delimiter = Tag(delimiter)
bytes_to_find = pack(struct_format, delimiter.group) + pack(struct_format, delimiter.elem)
return find_bytes(fp, bytes_to_find, rewind=rewind)
def length_of_undefined_length(fp, delimiter, is_little_endian, read_size=128, rewind=True):
"""Search through the file to find the delimiter, return the length of the data
element.
Return the file to the start of the data, ready to read it.
Note the data element that the delimiter starts is not read here, the calling
routine must handle that.
delimiter must be 4 bytes long
rewind == if True, file will be returned to position before seeking the bytes
"""
chunk = 0
data_start = fp.tell()
delimiter_pos = find_delimiter(fp, delimiter, is_little_endian, rewind=rewind)
length = delimiter_pos - data_start
return length
def read_delimiter_item(fp, delimiter):
"""Read and ignore an expected delimiter.
If the delimiter is not found or correctly formed, a warning is logged.
"""
found = fp.read(4)
if found != delimiter:
logger.warn("Expected delimitor %s, got %s at file position 0x%x", Tag(delimiter), Tag(found), fp.tell()-4)
length = fp.read_UL()
if length != 0:
logger.warn("Expected delimiter item to have length 0, got %d at file position 0x%x", length, fp.tell()-4)
|