import six import struct import re class LabelMap(object): """ Contains pointers to metadata. This might only be valid for V3 files. """ def __init__(self, raw_binary_data): self._data = raw_binary_data self._image_data = {} def image_attributes(self): return self._get_location(six.b("ImageAttributesLV!")) def _get_location(self, label): try: label_location = self._data.index(label) + len(label) return self._parse_data_location(label_location) except ValueError: return None def _parse_data_location(self, label_location): location, length = struct.unpack("QQ", self._data[label_location: label_location + 16]) return location @property def image_text_info(self): return self._get_location(six.b("ImageTextInfoLV!")) @property def image_metadata(self): return self._get_location(six.b("ImageMetadataLV!")) @property def image_metadata_sequence(self): # there is always only one of these, even though it has a pipe followed by a zero, which is how they do indexes return self._get_location(six.b("ImageMetadataSeqLV|0!")) def get_image_data_location(self, index): if not self._image_data: regex = re.compile(six.b("""ImageDataSeq\|(\d+)!""")) for match in regex.finditer(self._data): if match: location = self._parse_data_location(match.end()) self._image_data[int(match.group(1))] = location return self._image_data[index] @property def image_calibration(self): return self._get_location(six.b("ImageCalibrationLV|0!")) @property def image_attributes(self): return self._get_location(six.b("ImageAttributesLV!")) @property def x_data(self): return self._get_location(six.b("CustomData|X!")) @property def y_data(self): return self._get_location(six.b("CustomData|Y!")) @property def z_data(self): return self._get_location(six.b("CustomData|Z!")) @property def roi_metadata(self): return self._get_location(six.b("CustomData|RoiMetadata_v1!")) @property def pfs_status(self): return self._get_location(six.b("CustomData|PFS_STATUS!")) @property def pfs_offset(self): return self._get_location(six.b("CustomData|PFS_OFFSET!")) @property def guid(self): return self._get_location(six.b("CustomData|GUIDStore!")) @property def description(self): return self._get_location(six.b("CustomData|CustomDescriptionV1_0!")) @property def camera_exposure_time(self): return self._get_location(six.b("CustomData|Camera_ExposureTime1!")) @property def camera_temp(self): return self._get_location(six.b("CustomData|CameraTemp1!")) @property def acquisition_times(self): return self._get_location(six.b("CustomData|AcqTimesCache!")) @property def acquisition_times_2(self): return self._get_location(six.b("CustomData|AcqTimes2Cache!")) @property def acquisition_frames(self): return self._get_location(six.b("CustomData|AcqFramesCache!")) @property def lut_data(self): return self._get_location(six.b("CustomDataVar|LUTDataV1_0!")) @property def grabber_settings(self): return self._get_location(six.b("CustomDataVar|GrabberCameraSettingsV1_0!")) @property def custom_data(self): return self._get_location(six.b("CustomDataVar|CustomDataV2_0!")) @property def app_info(self): return self._get_location(six.b("CustomDataVar|AppInfo_V1_0!"))