diff --git a/.gitignore b/.gitignore index 27cd58f..50970d1 100644 --- a/.gitignore +++ b/.gitignore @@ -145,6 +145,9 @@ cython_debug/ *.cbz *.pdf +# vscode +.vscode/ + *.bat .idea/ diff --git a/mangle/image.py b/mangle/image.py index 3928b98..d36dba6 100644 --- a/mangle/image.py +++ b/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: diff --git a/mangle/options.py b/mangle/options.py index 8829463..e798735 100644 --- a/mangle/options.py +++ b/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(): diff --git a/mangle/ui/options.ui b/mangle/ui/options.ui index 08cc082..b31b690 100644 --- a/mangle/ui/options.ui +++ b/mangle/ui/options.ui @@ -215,9 +215,23 @@ - + - Scale and crop images to fill screen width + Fit to screen with borders + + + + + + + Fill screen by cropping + + + + + + + Stretch images to fill screen