dirImgProcess - multiprocessing

kcc/comic2ebook.py

@@ -32,6 +32,7 @@ from shutil import copytree
 from shutil import rmtree
 from shutil import make_archive
 from optparse import OptionParser
+from multiprocessing import Pool
 import image
 import cbxarchive
 import pdfjpgextract
@@ -255,7 +256,7 @@ def buildOPF(profile, dstdir, title, filelist, cover=None):
     f.write("</manifest>\n<spine toc=\"ncx\">\n")
     splitCountUsed = 1
     for entry in reflist:
-        if entry.endswith("-000001"):
+        if entry.endswith("-000001") or entry.endswith("-1"):
             # noinspection PyRedundantParentheses
             if ((options.righttoleft and facing == 'left') or (not options.righttoleft and facing == 'right')) and\
                     options.landscapemode:
@@ -265,7 +266,7 @@ def buildOPF(profile, dstdir, title, filelist, cover=None):
                 f.write("<itemref idref=\"page_" + entry + "\" properties=\"page-spread-" + facing1 + "\"/>\n")
             else:
                 f.write("<itemref idref=\"page_" + entry + "\"/>\n")
-        elif entry.endswith("-000002"):
+        elif entry.endswith("-000002") or entry.endswith("-2"):
             if options.landscapemode:
                 f.write("<itemref idref=\"page_" + entry + "\" properties=\"page-spread-" + facing2 + "\"/>\n")
             else:
@@ -323,7 +324,7 @@ def isInFilelist(filename, filelist):
     return seen
 
 
-def applyImgOptimization(img, isSplit=False, toRight=False):
+def applyImgOptimization(img, isSplit, toRight, options):
     img.cropWhiteSpace(10.0)
     if options.cutpagenumbers:
         img.cutPageNumber()
@@ -336,14 +337,32 @@ def applyImgOptimization(img, isSplit=False, toRight=False):
 
 def dirImgProcess(path):
     global options, splitCount
+    work = []
     pagenumber = 0
     pagenumbermodifier = 0
-    splitpages = []
+    pool = Pool()
-
     for (dirpath, dirnames, filenames) in os.walk(path):
         for afile in filenames:
             if getImageFileName(afile) is not None:
                 pagenumber += 1
+                work.append([afile, dirpath, pagenumber, options])
+    splitpages = pool.map(fileImgProcess, work)
+    pool.close()
+    pool.join()
+    splitpages = filter(None, splitpages)
+    splitpages.sort()
+    for page in splitpages:
+        if (page + pagenumbermodifier) % 2 == 0:
+            splitCount += 1
+            pagenumbermodifier += 1
+
+
+def fileImgProcess(work):
+    afile = work[0]
+    dirpath = work[1]
+    pagenumber = work[2]
+    options = work[3]
+    output = None
     if options.verbose:
         print "Optimizing " + afile + " for " + options.profile
     else:
@@ -359,26 +378,20 @@ def dirImgProcess(path):
         if options.righttoleft:
             toRight1 = False
             toRight2 = True
-                        splitpages.append(pagenumber)
         else:
             toRight1 = True
             toRight2 = False
-                        splitpages.append(pagenumber)
+        output = pagenumber
         img0 = image.ComicPage(split[0], options.profile)
-                    applyImgOptimization(img0, True, toRight1)
+        applyImgOptimization(img0, True, toRight1, options)
-                    img0.saveToDir(dirpath, options.forcepng, options.forcecolor, None)
+        img0.saveToDir(dirpath, options.forcepng, options.forcecolor)
         img1 = image.ComicPage(split[1], options.profile)
-                    applyImgOptimization(img1, True, toRight2)
+        applyImgOptimization(img1, True, toRight2, options)
-                    img1.saveToDir(dirpath, options.forcepng, options.forcecolor, None)
+        img1.saveToDir(dirpath, options.forcepng, options.forcecolor)
     else:
-                    applyImgOptimization(img)
+        applyImgOptimization(img, False, False, options)
         img.saveToDir(dirpath, options.forcepng, options.forcecolor, split)
-
+    return output
-    splitpages.sort()
-    for page in splitpages:
-        if (page + pagenumbermodifier) % 2 == 0:
-            splitCount += 1
-            pagenumbermodifier += 1
 
 
 def genEpubStruct(path):

kcc/image.py

@@ -119,7 +119,7 @@ class ComicPage:
             raise RuntimeError('Cannot read image file %s' % source)
         self.image = self.image.convert('RGB')
 
-    def saveToDir(self, targetdir, forcepng, color, sufix):
+    def saveToDir(self, targetdir, forcepng, color, sufix=None):
         filename = os.path.basename(self.origFileName)
         try:
             if not color: