Apply ICC profiles.

This is done on tiff and openslide tiff-based sources.

CHANGELOG.md

@@ -1,6 +1,9 @@
 # Change Log
 
-## 1.19.4
+## 1.20.0
+
+### Features
+- ICC color profile support ([#1037](../../pull/1037))
 
 ### Improvements
 - Speed up generating tiles for some multi source files ([#1035](../../pull/1035))

docs/tilesource_options.rst

@@ -64,6 +64,8 @@ A band definition is an object which can contain the following keys:
 
 - ``axis``: if specified, keep on the specified axis (channel) of the intermediate numpy array.  This is typically between 0 and 3 for the red, green, blue, and alpha channels.  Only the first such value is used, and this can be specified as a base key if ``bands`` is specified.
 
+- ``icc``: by default, sources that expose ICC color profiles will apply those profiles to the image data, converting the results to the sRGB profile.  To use the raw image data without ICC profile adjustments, specify an ``icc`` value of ``false``.  If the entire style is ``{"icc": false}``, the results will be the same as the default bands with only the adjustment being skipped.  Note that not all tile sources expose ICC color profile information, even if the base file format contains it.
+
 - ``function``: if specified, call a function to modify the resulting image.  This can be specified as a base key and as a band key.  Style functions can be called at multiple stages in the styling pipeline:
 
   - ``pre`` stage: this passes the original tile image to the function before any band data is applied.

girder/girder_large_image/models/image_item.py

@@ -18,6 +18,7 @@
 import json
 import pickle
 
+import PIL.ImageCms
 import pymongo
 from girder_jobs.constants import JobStatus
 from girder_jobs.models.job import Job
@@ -246,6 +247,11 @@ def getMetadata(self, item, **kwargs):
     def getInternalMetadata(self, item, **kwargs):
         tileSource = self._loadTileSource(item, **kwargs)
         result = tileSource.getInternalMetadata() or {}
+        if tileSource.getICCProfiles():
+            print(dir(tileSource.getICCProfiles()[0]))
+            result['iccprofiles'] = [
+                PIL.ImageCms.getProfileInfo(prof).strip() or 'present' if prof else None
+                for prof in tileSource.getICCProfiles()]
         result['tilesource'] = tileSource.name
         return result
 

large_image/tilesource/base.py

@@ -1,3 +1,4 @@
+import io
 import json
 import math
 import os
@@ -11,6 +12,7 @@
 import numpy
 import PIL
 import PIL.Image
+import PIL.ImageCms
 import PIL.ImageColor
 import PIL.ImageDraw
 
@@ -1191,6 +1193,73 @@ def _applyStyleFunction(self, image, sc, stage, function=None):
                 self.logger.exception('Failed to execute style function %s' % function['name'])
             return image
 
+    def getICCProfiles(self, idx=None):
+        """
+        Get a list of all ICC profiles that are available for the source, or
+        get a specific profile.
+
+        :param idx: a 0-based index into the profiles to get one profile, or
+            None to get a list of all profiles.
+        :returns: either one or a list of PIL.ImageCms.CmsProfile objects, or
+            None if no profiles are available.  If a list, entries in the list
+            may be None.
+        """
+        if not hasattr(self, '_iccprofiles'):
+            return None
+        results = []
+        for pidx, prof in enumerate(self._iccprofiles):
+            if idx is not None and pidx != idx:
+                continue
+            if hasattr(self, '_iccprofilesObjects') and self._iccprofilesObjects[pidx] is not None:
+                prof = self._iccprofilesObjects[pidx]['profile']
+            elif not isinstance(prof, PIL.ImageCms.ImageCmsProfile):
+                prof = PIL.ImageCms.getOpenProfile(io.BytesIO(prof))
+            if idx == pidx:
+                return prof
+            results.append(prof)
+        return results
+
+    def _applyICCProfile(self, sc, frame):
+        """
+        Apply an ICC profile to an image.
+
+        :param sc: the style context.
+        :param frame: the frame to use for auto ranging.
+        :returns: an image with the icc profile, if any, applied.
+        """
+        profileIdx = frame if frame and len(self._iccprofiles) >= frame + 1 else 0
+        sc.iccimage = sc.image
+        sc.iccapplied = False
+        if not self._iccprofiles[profileIdx]:
+            return sc.image
+        if not hasattr(self, '_iccprofilesObjects'):
+            self._iccprofilesObjects = [None] * len(self._iccprofiles)
+        image = _imageToPIL(sc.image)
+        mode = image.mode
+        if not hasattr(self, '_iccsrgbprofile'):
+            self._iccsrgbprofile = PIL.ImageCms.createProfile('sRGB')
+        try:
+            if self._iccprofilesObjects[profileIdx] is None:
+                self._iccprofilesObjects[profileIdx] = {
+                    'profile': self.getICCProfiles(profileIdx)
+                }
+                if mode not in self._iccprofilesObjects[profileIdx]:
+                    self._iccprofilesObjects[profileIdx][mode] = \
+                        PIL.ImageCms.buildTransformFromOpenProfiles(
+                            self._iccprofilesObjects[profileIdx]['profile'],
+                            self._iccsrgbprofile, mode, mode)
+            transform = self._iccprofilesObjects[profileIdx][mode]
+
+            PIL.ImageCms.applyTransform(image, transform, inPlace=True)
+            sc.iccimage = _imageToNumpy(image)[0]
+            sc.iccapplied = True
+        except Exception as exc:
+            raise
+            if not hasattr(self, '_iccerror'):
+                self._iccerror = exc
+                self.logger.exception('Failed to apply ICC profile')
+        return sc.iccimage
+
     def _applyStyle(self, image, style, x, y, z, frame=None):  # noqa
         """
         Apply a style to a numpy image.
@@ -1205,11 +1274,19 @@ def _applyStyle(self, image, style, x, y, z, frame=None):  # noqa
         """
         sc = types.SimpleNamespace(
             image=image, originalStyle=style, x=x, y=y, z=z, frame=frame,
-            mainImage=image, mainFrame=frame)
-        sc.dtype = style.get('dtype')
-        sc.axis = style.get('axis')
-        sc.style = style if 'bands' in style else {'bands': [style]}
-        sc.output = numpy.zeros((image.shape[0], image.shape[1], 4), float)
+            mainImage=image, mainFrame=frame, dtype=None, axis=None)
+        if style is None:
+            sc.style = {'bands': []}
+        else:
+            sc.style = style if 'bands' in style else {'bands': [style]}
+            sc.dtype = style.get('dtype')
+            sc.axis = style.get('axis')
+        if hasattr(self, '_iccprofiles') and sc.style.get('icc', True):
+            image = self._applyICCProfile(sc, frame)
+        if style is None or (len(style) == 1 and 'icc' in style):
+            sc.output = image
+        else:
+            sc.output = numpy.zeros((image.shape[0], image.shape[1], 4), float)
         image = self._applyStyleFunction(image, sc, 'pre')
         for eidx, entry in enumerate(sc.style['bands']):
             sc.styleIndex = eidx
@@ -1343,10 +1420,10 @@ def _outputTileNumpyStyle(self, tile, applyStyle, x, y, z, frame=None):
         :returns: a numpy array and a target PIL image mode.
         """
         tile, mode = _imageToNumpy(tile)
-        if applyStyle and getattr(self, 'style', None):
+        if applyStyle and (getattr(self, 'style', None) or hasattr(self, '_iccprofiles')):
             with self._styleLock:
                 if not getattr(self, '_skipStyle', False):
-                    tile = self._applyStyle(tile, self.style, x, y, z, frame)
+                    tile = self._applyStyle(tile, getattr(self, 'style', None), x, y, z, frame)
         if tile.shape[0] != self.tileHeight or tile.shape[1] != self.tileWidth:
             extend = numpy.zeros(
                 (self.tileHeight, self.tileWidth, tile.shape[2]),
@@ -1386,8 +1463,10 @@ def _outputTile(self, tile, tileEncoding, x, y, z, pilImageAllowed=False,
                 not isEdge and (not applyStyle or not getattr(self, 'style', None))):
             return tile
         mode = None
-        if (numpyAllowed == 'always' or tileEncoding == TILE_FORMAT_NUMPY or
-                (applyStyle and getattr(self, 'style', None)) or isEdge):
+        if (numpyAllowed == 'always' or tileEncoding == TILE_FORMAT_NUMPY or (applyStyle and (
+                (getattr(self, 'style', None) or hasattr(self, '_iccprofiles')) and
+                getattr(self, 'style', None) != {'icc': False})) or
+                isEdge):
             tile, mode = self._outputTileNumpyStyle(
                 tile, applyStyle, x, y, z, self._getFrame(**kwargs))
         if isEdge:

sources/openslide/large_image_source_openslide/__init__.py

@@ -95,6 +95,9 @@ def __init__(self, path, **kwargs):  # noqa
         if libtiff_ctypes:
             try:
                 self._tiffinfo = tifftools.read_tiff(self._largeImagePath)
+                if tifftools.Tag.ICCProfile.value in self._tiffinfo['ifds'][0]['tags']:
+                    self._iccprofiles = [self._tiffinfo['ifds'][0]['tags'][
+                        tifftools.Tag.ICCProfile.value]['data']]
             except Exception:
                 pass
 

sources/tiff/large_image_source_tiff/__init__.py

@@ -308,6 +308,13 @@ def _initWithTiffTools(self):  # noqa
                             frames[-1][key] = frameMetadata[key]
                 except Exception:
                     pass
+                if tifftools.Tag.ICCProfile.value in ifd['tags']:
+                    if not hasattr(self, '_iccprofiles'):
+                        self._iccprofiles = []
+                    while len(self._iccprofiles) < len(frames) - 1:
+                        self._iccprofiles.append(None)
+                    self._iccprofiles.append(ifd['tags'][
+                        tifftools.Tag.ICCProfile.value]['data'])
             # otherwise, add to the first frame missing that level
             elif level < self.levels - 1 and any(
                     frame for frame in frames if frame['dirs'][level] is None):

test/test_converter.py

@@ -146,7 +146,9 @@ def testConvertJp2kCompression(tmpdir):
     source = large_image_source_tiff.open(outputPath)
     image, _ = source.getRegion(
         output={'maxWidth': 200, 'maxHeight': 200}, format=constants.TILE_FORMAT_NUMPY)
-    assert (image[12][167] == [215, 135, 172]).all()
+    # Without or with icc adjustment
+    assert ((image[12][167] == [215, 135, 172]).all() or
+            (image[12][167] == [216, 134, 172]).all())
 
     outputPath2 = os.path.join(tmpdir, 'out2.tiff')
     large_image_converter.convert(imagePath, outputPath2, compression='jp2k', psnr=50)

test/test_source_openslide.py

@@ -318,7 +318,7 @@ def testGetPixel():
     imagePath = datastore.fetch(
         'sample_jp2k_33003_TCGA-CV-7242-11A-01-TS1.1838afb1-9eee-'
         '4a70-9ae3-50e3ab45e242.svs')
-    source = large_image_source_openslide.open(imagePath)
+    source = large_image_source_openslide.open(imagePath, style={'icc': False})
 
     pixel = source.getPixel(region={'left': 12125, 'top': 10640})
     assert pixel == {'r': 156, 'g': 98, 'b': 138, 'a': 255}