Changeset 279 for pkpgcounter/trunk

Timestamp:

11/05/05 10:04:18 (19 years ago)

Author:

jerome

Message:

Now extracts black as well. For black the result is identical
to PrintBill?'s one when parsing the same 8-bit gray png images,
although my own code accepts other formats.

Files:

• pkpgcounter/trunk/pkpgpdls/inkcoverage.py (modified) (1 diff)

pkpgcounter/trunk/pkpgpdls/inkcoverage.py

@@ -34 +34 @@
 """        
 
-def percent_cmy(fname) :
+def getPercentCMY(img, nbpix) :
     result = []
-    img = Image.open(fname)
     (r, g, b) = [ p.histogram() for p in img.split() ]
-    nbpix = sum(r)
-    for histo in (r, g, b) :
-        result.append((100 * (reduce(lambda x,y: x + (y[1] * (255 - y[0])), enumerate(histo), 0) / 255.0)) / nbpix)
-    return tuple(result)        
+    for colorhisto in (r, g, b) :
+        result.append(100.0 * (reduce(lambda current, next: current + (next[1] * (255 - next[0])), enumerate(colorhisto), 0) / 255.0) / nbpix)
+    return tuple(result)
+    
+def getPercentBlack(img, nbpix) :
+    if img.mode != "L" :
+        img = img.convert("L")
+    return 100.0 * (reduce(lambda current, next: current + (next[1] * (255 - next[0])), enumerate(img.histogram()[:-1]), 0) / 255.0) / nbpix
+    
+def getPercents(fname) :
+    """Extracts the ink percentages from an image."""
+    image = Image.open(fname)
+    nbpixels = image.size[0] * image.size[1]
+    result = {}
+    try :
+        result["black"] = getPercentBlack(image, nbpixels)
+    except :     
+        sys.stderr.write("Problem when extracting BLACK !\n")
+    try :    
+        result["cmy"] = getPercentCMY(image, nbpixels)
+    except :     
+        sys.stderr.write("Problem when extracting CMY !\n")
+    return result
 
 if __name__ == "__main__" :
-    print percent_cmy(sys.argv[1])
+    print getPercents(sys.argv[1])