Added convert_mode param when saving

Author: radarhere

Tests/images/pil123rgba_red.jpg

@@ -566,6 +566,16 @@ def test_save_I(self):
         reloaded = Image.open(out)
         self.assert_image_equal(reloaded.convert('L'), im.convert('L'))
 
+    def test_save_wrong_modes(self):
+        out = BytesIO()
+        for mode in ['CMYK']:
+            img = Image.new(mode, (20, 20))
+            self.assertRaises(ValueError, img.save, out, "GIF")
+
+        for mode in ['CMYK', 'LA']:
+            img = Image.new(mode, (20, 20))
+            img.save(out, "GIF", convert_mode=True)
+
     def test_getdata(self):
         # test getheader/getdata against legacy values
         # Create a 'P' image with holes in the palette

Tests/test_file_gif.py

@@ -507,10 +507,22 @@ def test_save_correct_modes(self):
     def test_save_wrong_modes(self):
         # ref https://github.com/python-pillow/Pillow/issues/2005
         out = BytesIO()
-        for mode in ['LA', 'La', 'RGBA', 'RGBa', 'P']:
+        for mode in ['LA', 'La', 'RGBA', 'RGBa', 'P', 'I']:
             img = Image.new(mode, (20, 20))
             self.assertRaises(IOError, img.save, out, "JPEG")
 
+        for mode in ['LA', 'RGBA', 'P', 'I']:
+            img = Image.new(mode, (20, 20))
+            img.save(out, "JPEG", convert_mode=True)
+
+        img = Image.open('Tests/images/pil123rgba.png')
+        temp_file = self.tempfile("temp.jpg")
+        img.save(temp_file, convert_mode=True, fill_color='red')
+
+        reloaded = Image.open(temp_file)
+        target = Image.open('Tests/images/pil123rgba_red.jpg')
+        self.assert_image_similar(reloaded, target, 4)
+
     def test_save_tiff_with_dpi(self):
         # Arrange
         outfile = self.tempfile("temp.tif")

Tests/test_file_jpeg.py

@@ -227,6 +227,13 @@ def test_load_transparent_rgb(self):
         # image has 876 transparent pixels
         self.assertEqual(im.getchannel('A').getcolors()[0][0], 876)
 
+    def test_save_CMYK(self):
+        out = BytesIO()
+        img = Image.new('CMYK', (20, 20))
+        self.assertRaises(IOError, img.save, out, "PNG")
+
+        img.save(out, "PNG", convert_mode=True)
+
     def test_save_p_transparent_palette(self):
         in_file = "Tests/images/pil123p.png"
         im = Image.open(in_file)

Tests/test_file_png.py

@@ -2,6 +2,8 @@
 
 from PIL import Image, WebPImagePlugin
 
+from io import BytesIO
+
 try:
     from PIL import _webp
     HAVE_WEBP = True
@@ -53,6 +55,13 @@ def test_read_rgb(self):
         self.assert_image_similar_tofile(
             image, 'Tests/images/hopper_webp_bits.ppm', 1.0)
 
+    @unittest.skipIf(not HAVE_WEBP, "WebP support not installed")
+    def test_save_convert_mode(self):
+        out = BytesIO()
+        for mode in ['CMYK', 'I', 'L', 'LA', 'P']:
+            img = Image.new(mode, (20, 20))
+            img.save(out, "WEBP", convert_mode=True)
+
     @unittest.skipIf(not HAVE_WEBP, "WebP support not installed")
     def test_write_rgb(self):
         """

Tests/test_file_webp.py

@@ -1,6 +1,6 @@
 from helper import unittest, PillowTestCase, hopper
 
-from PIL import Image
+from PIL import Image, TiffImagePlugin
 from PIL._util import py3
 import os
 
@@ -323,6 +323,14 @@ def test_registered_extensions(self):
         for ext in ['.cur', '.icns', '.tif', '.tiff']:
             self.assertIn(ext, extensions)
 
+    def test_no_convert_mode(self):
+        self.assertTrue(not hasattr(TiffImagePlugin, '_convert_mode'))
+
+        temp_file = self.tempfile("temp.tiff")
+
+        im = hopper()
+        im.save(temp_file, convert_mode=True)
+
     def test_effect_mandelbrot(self):
         # Arrange
         size = (512, 512)

Tests/test_image.py

@@ -809,6 +809,13 @@ def write(self, data):
     return fp.data
 
 
+def _convert_mode(im):
+    return {
+        'LA':'P',
+        'CMYK':'RGB'
+    }.get(im.mode)
+
+
 # --------------------------------------------------------------------
 # Registry
 

src/PIL/GifImagePlugin.py

@@ -1934,10 +1934,6 @@ def save(self, fp, format=None, **params):
         # may mutate self!
         self.load()
 
-        save_all = params.pop('save_all', False)
-        self.encoderinfo = params
-        self.encoderconfig = ()
-
         preinit()
 
         ext = os.path.splitext(filename)[1].lower()
@@ -1952,11 +1948,20 @@ def save(self, fp, format=None, **params):
 
         if format.upper() not in SAVE:
             init()
-        if save_all:
+        if params.pop('save_all', False):
             save_handler = SAVE_ALL[format.upper()]
         else:
             save_handler = SAVE[format.upper()]
 
+        if params.get('convert_mode'):
+            plugin = sys.modules[save_handler.__module__]
+            converted_im = self._convert_mode(plugin, params)
+            if converted_im:
+                return converted_im.save(fp, format, **params)
+
+        self.encoderinfo = params
+        self.encoderconfig = ()
+
         if open_fp:
             if params.get('append', False):
                 fp = builtins.open(filename, "r+b")
@@ -1972,6 +1977,37 @@ def save(self, fp, format=None, **params):
             if open_fp:
                 fp.close()
 
+    def _convert_mode(self, plugin, params):
+        if not hasattr(plugin, '_convert_mode'):
+            return
+        new_mode = plugin._convert_mode(self)
+        if self.mode == 'LA' and new_mode == 'P':
+            alpha = self.getchannel('A')
+            # Convert the image into P mode but only use 255 colors
+            # in the palette out of 256.
+            im = self.convert('L') \
+                .convert('P', palette=ADAPTIVE, colors=255)
+            # Set all pixel values below 128 to 255, and the rest to 0.
+            mask = eval(alpha, lambda px: 255 if px < 128 else 0)
+            # Paste the color of index 255 and use alpha as a mask.
+            im.paste(255, mask)
+            # The transparency index is 255.
+            im.info['transparency'] = 255
+            return im
+
+        elif self.mode == 'I':
+            im = self.point([i//256 for i in range(65536)], 'L')
+            return im.convert(new_mode) if new_mode != 'L' else im
+
+        elif self.mode in ('RGBA', 'LA') and new_mode in ('RGB', 'L'):
+            fill_color = params.get('fill_color', 'white')
+            background = new(new_mode, self.size, fill_color)
+            background.paste(self, self.getchannel('A'))
+            return background
+
+        elif new_mode:
+            return self.convert(new_mode)
+
     def seek(self, frame):
         """
         Seeks to the given frame in this sequence file. If you seek

src/PIL/Image.py

@@ -792,6 +792,17 @@ def jpeg_factory(fp=None, filename=None):
     return im
 
 
+def _convert_mode(im):
+    mode = im.mode
+    if mode == 'P':
+        return 'RGBA' if 'A' in im.im.getpalettemode() else 'RGB'
+    return {
+        'RGBA':'RGB',
+        'LA':'L',
+        'I':'L'
+    }.get(mode)
+
+
 # ---------------------------------------------------------------------
 # Registry stuff
 

src/PIL/JpegImagePlugin.py

@@ -860,6 +860,12 @@ def append(fp, cid, *data):
     return fp.data
 
 
+def _convert_mode(im):
+    return {
+        'CMYK':'RGB'
+    }.get(im.mode)
+
+
 # --------------------------------------------------------------------
 # Registry