zolfa-nd2reader/nd2reader/model/label.py

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!"))