public/kcc
1
0
Fork 0
mirror of https://github.com/ciromattia/kcc synced 2025-12-16 03:06:33 +00:00
Code Issues Releases Wiki Activity

Margin color detection tweaks

Browse Source
This commit is contained in:
Paweł Jastrzębski
2014-01-26 10:25:26 +01:00
parent 60b7a90589
commit 63d752280a
1 changed files with 15 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
kcc/image.py
View File

@@ -107,6 +107,7 @@ class ComicPage:
self.filename = os.path.basename(self.origFileName) self.filename = os.path.basename(self.origFileName)
self.image = Image.open(source) self.image = Image.open(source)
self.image = self.image.convert('RGB') self.image = self.image.convert('RGB')
self.color = self.isImageColor()
self.rotated = None self.rotated = None
self.border = None self.border = None
self.noHPV = None self.noHPV = None
@@ -156,7 +157,7 @@ class ComicPage:
def optimizeImage(self, gamma): def optimizeImage(self, gamma):
if gamma < 0.1: if gamma < 0.1:
gamma = self.gamma gamma = self.gamma
if self.gamma != 1.0 and self.isImageColor(self.image): if self.gamma != 1.0 and self.color:
gamma = 1.0 gamma = 1.0
if gamma == 1.0: if gamma == 1.0:
self.image = ImageOps.autocontrast(self.image) self.image = ImageOps.autocontrast(self.image)
@@ -414,30 +415,37 @@ class ComicPage:
return False return False
def getImageFill(self, webtoon): def getImageFill(self, webtoon):
if not webtoon and self.color:
self.fill = 'black'
return
fill = 0 fill = 0
if not webtoon and not self.rotated: if not webtoon and not self.rotated:
# Search for horizontal solid lines # Search for horizontal solid lines
startY = 0 startY = 0
while startY < self.image.size[1]: while startY < self.image.size[1]:
checkSolid = self.getImageHistogram(self.image.crop((0, startY, self.image.size[0], startY+1))) if startY + 5 > self.image.size[1]:
startY = self.image.size[1] - 5
checkSolid = self.getImageHistogram(self.image.crop((0, startY, self.image.size[0], startY+5)))
if checkSolid: if checkSolid:
fill += checkSolid fill += checkSolid
startY += 1 startY += 5
else: else:
# Search for vertical solid lines # Search for vertical solid lines
startX = 0 startX = 0
while startX < self.image.size[0]: while startX < self.image.size[0]:
checkSolid = self.getImageHistogram(self.image.crop((startX, 0, startX+1, self.image.size[1]))) if startX + 5 > self.image.size[0]:
startX = self.image.size[0] - 5
checkSolid = self.getImageHistogram(self.image.crop((startX, 0, startX+5, self.image.size[1])))
if checkSolid: if checkSolid:
fill += checkSolid fill += checkSolid
startX += 1 startX += 5
if fill > 0: if fill > 0:
self.fill = 'black' self.fill = 'black'
else: else:
self.fill = 'white' self.fill = 'white'
def isImageColor(self, image): def isImageColor(self):
v = ImageStat.Stat(image).var v = ImageStat.Stat(self.image).var
isMonochromatic = reduce(lambda x, y: x and y < 0.005, v, True) isMonochromatic = reduce(lambda x, y: x and y < 0.005, v, True)
if isMonochromatic: if isMonochromatic:
# Monochromatic # Monochromatic