Merge pull request #42 from axu2/options

Add new fit, fill, stretch options

.gitignore

@@ -145,6 +145,9 @@ cython_debug/
 *.cbz
 *.pdf
 
+# vscode
+.vscode/
+
 *.bat
 .idea/
 

mangle/image.py

@@ -24,12 +24,14 @@ class ImageFlags:
     Resize = 1 << 1
     Frame = 1 << 2
     Quantize = 1 << 3
-    ScaleCrop = 1 << 4
+    Fit = 1 << 4
     SplitRightLeft = 1 << 5    # split right then left
     SplitRight = 1 << 6        # split only the right page
     SplitLeft = 1 << 7         # split only the left page
     SplitLeftRight = 1 << 8    # split left then right page
     AutoCrop = 1 << 9
+    Fill = 1 << 10
+    Stretch = 1 << 11
 
 
 class KindleData:
@@ -134,7 +136,27 @@ def quantizeImage(image, palette):
 
 
 @protect_bad_image
-def scaleCropImage(image, size):
+def fitImage(image, size, method=Image.ANTIALIAS):
+    # copied from ImageOps.contain() from the Python3 version of Pillow
+    # with division related modifications for Python2
+
+    im_ratio = 1.0 * image.width / image.height
+    dest_ratio = 1.0 * size[0] / size[1]
+
+    if im_ratio != dest_ratio:
+        if im_ratio > dest_ratio:
+            new_height = int(1.0 * image.height / image.width * size[0])
+            if new_height != size[1]:
+                size = (size[0], new_height)
+        else:
+            new_width = int(1.0 * image.width / image.height * size[1])
+            if new_width != size[0]:
+                size = (new_width, size[1])
+    return image.resize(size, resample=method)
+
+
+@protect_bad_image
+def fillImage(image, size):
     widthDev, heightDev = size
     widthImg, heightImg = image.size
     
@@ -148,6 +170,11 @@ def scaleCropImage(image, size):
     return ImageOps.fit(image, size, Image.ANTIALIAS)
 
 
+@protect_bad_image
+def stretchImage(image, size):
+    return image.resize(size, Image.ANTIALIAS)
+
+
 @protect_bad_image
 def resizeImage(image, size):
     widthDev, heightDev = size
@@ -288,8 +315,12 @@ def convertImage(source, target, device, flags):
         image = orientImage(image, size)
     if flags & ImageFlags.Resize:
         image = resizeImage(image, size)
-    if flags & ImageFlags.ScaleCrop:
-        image = scaleCropImage(image, size)
+    if flags & ImageFlags.Fit:
+        image = fitImage(image, size)
+    if flags & ImageFlags.Fill:
+        image = fillImage(image, size)
+    if flags & ImageFlags.Stretch:
+        image = stretchImage(image, size)
     if flags & ImageFlags.Frame:
         image = frameImage(image, tuple(palette[:3]), tuple(palette[-3:]), size)
     if flags & ImageFlags.Quantize:

mangle/options.py

@@ -43,7 +43,9 @@ class DialogOptions(QtGui.QDialog):
         self.checkboxOverwrite.setChecked(self.book.overwrite)
         self.checkboxOrient.setChecked(self.book.imageFlags & ImageFlags.Orient)
         self.checkboxResize.setChecked(self.book.imageFlags & ImageFlags.Resize)
-        self.checkboxScaleCrop.setChecked(self.book.imageFlags & ImageFlags.ScaleCrop)
+        self.checkboxFit.setChecked(self.book.imageFlags & ImageFlags.Fit)
+        self.checkboxFill.setChecked(self.book.imageFlags & ImageFlags.Fill)
+        self.checkboxStretch.setChecked(self.book.imageFlags & ImageFlags.Stretch)
         self.checkboxQuantize.setChecked(self.book.imageFlags & ImageFlags.Quantize)
         self.checkboxFrame.setChecked(self.book.imageFlags & ImageFlags.Frame)
 
@@ -62,8 +64,12 @@ class DialogOptions(QtGui.QDialog):
             imageFlags |= ImageFlags.Orient
         if self.checkboxResize.isChecked():
             imageFlags |= ImageFlags.Resize
-        if self.checkboxScaleCrop.isChecked():
-            imageFlags |= ImageFlags.ScaleCrop
+        if self.checkboxFit.isChecked():
+            imageFlags |= ImageFlags.Fit
+        if self.checkboxFill.isChecked():
+            imageFlags |= ImageFlags.Fill
+        if self.checkboxStretch.isChecked():
+            imageFlags |= ImageFlags.Stretch
         if self.checkboxQuantize.isChecked():
             imageFlags |= ImageFlags.Quantize
         if self.checkboxFrame.isChecked():

mangle/ui/options.ui

@@ -215,9 +215,23 @@
        </widget>
       </item>
       <item>
-       <widget class="QRadioButton" name="checkboxScaleCrop">
+       <widget class="QRadioButton" name="checkboxFit">
         <property name="text">
-         <string>Scale and crop images to fill screen width</string>
+         <string>Fit to screen with borders</string>
+        </property>
+       </widget>
+      </item>
+      <item>
+       <widget class="QRadioButton" name="checkboxFill">
+        <property name="text">
+         <string>Fill screen by cropping</string>
+        </property>
+       </widget>
+      </item>      
+      <item>
+       <widget class="QRadioButton" name="checkboxStretch">
+        <property name="text">
+         <string>Stretch images to fill screen</string>
         </property>
        </widget>
       </item>