diff --git a/kcc/comic2panel.py b/kcc/comic2panel.py index e5962b4..7f0097e 100644 --- a/kcc/comic2panel.py +++ b/kcc/comic2panel.py @@ -53,7 +53,8 @@ def getImageFileName(imgfile): return filename -def getImageFill(histogram): +def getImageHistogram(image): + histogram = image.histogram() RBGW = [] for i in range(256): RBGW.append(histogram[i] + histogram[256 + i] + histogram[512 + i]) @@ -69,6 +70,18 @@ def getImageFill(histogram): return 'KCCFB' +def getImageFill(image): + imageSize = image.size + imageT = image.crop((0, 0, imageSize[0], 1)) + imageB = image.crop((0, imageSize[1]-1, imageSize[0], imageSize[1])) + imageT = getImageHistogram(imageT) + imageB = getImageHistogram(imageB) + if imageT == imageB: + return imageT + else: + return 'KCCFW' + + def sanitizePanelSize(panel, options): newPanels = [] if panel[2] > 1.5 * options.height: @@ -171,7 +184,7 @@ def splitImage(work): newPage.paste(panelImg, (0, targetHeight)) targetHeight += panels[panel][2] newPage.save(os.path.join(path, fileExpanded[0] + '-' + - str(pageNumber) + '-' + getImageFill(newPage.histogram()) + '.png'), 'PNG') + str(pageNumber) + '-' + getImageFill(newPage) + '.png'), 'PNG') pageNumber += 1 os.remove(os.path.join(path, name))