root / pkpgcounter / trunk / pkpgpdls / inkcoverage.py @ 3474

Revision 3474, 4.6 kB (checked in by jerome, 13 years ago)

Changed copyright years.

  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Revision Id
Line 
1# -*- coding: utf-8 -*-
2#
3# pkpgcounter : a generic Page Description Language parser
4#
5# (c) 2003-2009 Jerome Alet <alet@librelogiciel.com>
6# This program is free software: you can redistribute it and/or modify
7# it under the terms of the GNU General Public License as published by
8# the Free Software Foundation, either version 3 of the License, or
9# (at your option) any later version.
10#
11# This program is distributed in the hope that it will be useful,
12# but WITHOUT ANY WARRANTY; without even the implied warranty of
13# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14# GNU General Public License for more details.
15#
16# You should have received a copy of the GNU General Public License
17# along with this program.  If not, see <http://www.gnu.org/licenses/>.
18#
19# $Id$
20#
21
22"""This modules implements the computation of ink coverage in different colorspaces."""
23
24import sys
25
26import pdlparser
27
28try :
29    from PIL import Image
30except ImportError :
31    sys.stderr.write("ERROR: You MUST install the Python Imaging Library (python-imaging) for pkpgcounter to work.\n")
32    raise pdlparser.PDLParserError, "The Python Imaging Library is missing."
33
34def getPercent(img, nbpix) :
35    """Extracts the percents per color component from a picture.
36
37       Faster without Psyco on my own machine.
38    """
39    result = {}
40    bands = img.split()
41    for (i, bandname) in enumerate(img.getbands()) :
42        result[bandname] = 100.0 * (reduce(lambda current, next: current + (next[1] * next[0]), enumerate(bands[i].histogram()), 0) / 255.0) / nbpix
43    return result
44
45def getPercentCMYK(img, nbpix) :
46    """Extracts the percents of Cyan, Magenta, Yellow, and Black from a picture.
47
48       PIL doesn't produce useable CMYK for our algorithm, so we use the algorithm from PrintBill.
49       Psyco speeds this function up by around 2.5 times on my computer.
50    """
51    if img.mode != "RGB" :
52        img = img.convert("RGB")
53    cyan = magenta = yellow = black = 0
54    for (r, g, b) in img.getdata() :
55        pixblack = 255 - max(r, g, b)
56        black += pixblack
57        cyan += 255 - r - pixblack
58        magenta += 255 - g - pixblack
59        yellow += 255 - b - pixblack
60
61    frac = 100.0 / nbpix
62    return { "C" : frac * (cyan / 255.0),
63             "M" : frac * (magenta / 255.0),
64             "Y" : frac * (yellow / 255.0),
65             "K" : frac * (black / 255.0),
66           }
67
68def getPercentGC(img, nbpix) :
69    """Determines if a page is in grayscale or colour mode."""
70    if img.mode != "RGB" :
71        img = img.convert("RGB")
72    gray = 0
73    for (r, g, b) in img.getdata() :
74        if not (r == g == b) :
75            # optimize : if a single pixel is not gray the whole page is colored.
76            return { "G" : 0.0, "C" : 100.0 }
77    return { "G" : 100.0, "C" : 0.0 }
78
79def getPercentBW(img, nbpix) :
80    """Extracts the percents of Black from a picture, once converted to gray levels."""
81    if img.mode != "L" :
82        img = img.convert("L")
83    return { "B" : 100.0 - getPercent(img, nbpix)["L"] }
84
85def getPercentRGB(img, nbpix) :
86    """Extracts the percents of Red, Green, Blue from a picture, once converted to RGB."""
87    if img.mode != "RGB" :
88        img = img.convert("RGB")
89    return getPercent(img, nbpix)
90
91def getPercentCMY(img, nbpix) :
92    """Extracts the percents of Cyan, Magenta, and Yellow from a picture once converted to RGB."""
93    result = getPercentRGB(img, nbpix)
94    return { "C" : 100.0 - result["R"],
95             "M" : 100.0 - result["G"],
96             "Y" : 100.0 - result["B"],
97           }
98
99def getInkCoverage(fname, colorspace) :
100    """Returns a list of dictionnaries containing for each page,
101       for each color component, the percent of ink coverage on
102       that particular page.
103    """
104    result = []
105    colorspace = colorspace.upper()
106    computation = globals()["getPercent%s" % colorspace]
107    if colorspace in ("CMYK", "GC") : # faster with psyco on my machine
108        try :
109            import psyco
110        except ImportError :
111            pass
112        else :
113            psyco.bind(getPercentCMYK)
114
115    index = 0
116    try :
117        image = Image.open(fname)
118    except (IOError, OverflowError), msg :
119        raise pdlparser.PDLParserError, "%s (%s)" % (msg, fname)
120    else :
121        try :
122            while True :
123                nbpixels = image.size[0] * image.size[1]
124                result.append(computation(image, nbpixels))
125                index += 1
126                image.seek(index)
127        except EOFError :
128            pass
129        return (colorspace, result)
130
131if __name__ == "__main__" :
132    # NB : length of result gives number of pages !
133    sys.stdout.write("%s\n" % getInkCoverage(sys.argv[1], "CMYK"))
Note: See TracBrowser for help on using the browser.