root / pkpgcounter / trunk / pkpgpdls / postscript.py @ 270

Revision 263, 6.9 kB (checked in by jerome, 19 years ago)

Fixed special case in PostScript? parser.

  • Property svn:eol-style set to native
  • Property svn:keywords set to Auth Date Id Rev
Line 
1#! /usr/bin/env python
2# -*- coding: ISO-8859-15 -*-
3#
4# pkpgcounter : a generic Page Description Language parser
5#
6# (c) 2003, 2004, 2005 Jerome Alet <alet@librelogiciel.com>
7# This program is free software; you can redistribute it and/or modify
8# it under the terms of the GNU General Public License as published by
9# the Free Software Foundation; either version 2 of the License, or
10# (at your option) any later version.
11#
12# This program is distributed in the hope that it will be useful,
13# but WITHOUT ANY WARRANTY; without even the implied warranty of
14# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15# GNU General Public License for more details.
16#
17# You should have received a copy of the GNU General Public License
18# along with this program; if not, write to the Free Software
19# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
20#
21# $Id$
22#
23
24import sys
25import popen2
26
27import pdlparser
28
29class Parser(pdlparser.PDLParser) :
30    """A parser for PostScript documents."""
31    def isValid(self) :   
32        """Returns 1 if data is PostScript, else 0."""
33        if self.firstblock.startswith("%!") or \
34           self.firstblock.startswith("\004%!") or \
35           self.firstblock.startswith("\033%-12345X%!PS") or \
36           ((self.firstblock[:128].find("\033%-12345X") != -1) and \
37             ((self.firstblock.find("LANGUAGE=POSTSCRIPT") != -1) or \
38              (self.firstblock.find("LANGUAGE = POSTSCRIPT") != -1) or \
39              (self.firstblock.find("LANGUAGE = Postscript") != -1))) or \
40              (self.firstblock.find("%!PS-Adobe") != -1) :
41            self.logdebug("DEBUG: Input file is in the PostScript format.")
42            return 1
43        else :   
44            return 0
45       
46    def throughGhostScript(self) :
47        """Get the count through GhostScript, useful for non-DSC compliant PS files."""
48        self.logdebug("Internal parser sucks, using GhostScript instead...")
49        self.infile.seek(0)
50        command = 'gs -sDEVICE=bbox -dNOPAUSE -dBATCH -dQUIET - 2>&1 | grep -c "%%HiResBoundingBox:" 2>/dev/null'
51        child = popen2.Popen4(command)
52        try :
53            data = self.infile.read(pdlparser.MEGABYTE)   
54            while data :
55                child.tochild.write(data)
56                data = self.infile.read(pdlparser.MEGABYTE)
57            child.tochild.flush()
58            child.tochild.close()   
59        except (IOError, OSError), msg :   
60            raise pdlparser.PDLParserError, "Problem during analysis of Binary PostScript document : %s" % msg
61           
62        pagecount = 0
63        try :
64            pagecount = int(child.fromchild.readline().strip())
65        except (IOError, OSError, AttributeError, ValueError), msg :
66            raise pdlparser.PDLParserError, "Problem during analysis of Binary PostScript document : %s" % msg
67        child.fromchild.close()
68       
69        try :
70            child.wait()
71        except OSError, msg :   
72            raise pdlparser.PDLParserError, "Problem during analysis of Binary PostScript document : %s" % msg
73        return pagecount * self.copies
74       
75    def natively(self) :
76        """Count pages in a DSC compliant PostScript document."""
77        self.infile.seek(0)
78        pagecount = 0
79        pages = {}
80        pages[0] = { "copies" : 1 }
81        oldpagenum = None
82        previousline = ""
83        for line in self.infile.xreadlines() : 
84            if line.startswith(r"%%Page: ") or line.startswith(r"(%%[Page: ") :
85                proceed = 1
86                try :
87                    newpagenum = int(line.split(']')[0].split()[1])
88                except :   
89                    pass
90                else :   
91                    if newpagenum == oldpagenum :
92                        proceed = 0
93                    else :
94                        oldpagenum = newpagenum
95                if proceed :       
96                    pagecount += 1
97                    pages[pagecount] = { "copies" : pages[pagecount-1]["copies"] }
98            elif line.startswith(r"%%Requirements: numcopies(") :   
99                try :
100                    number = int(line.strip().split('(')[1].split(')')[0])
101                except :     
102                    pass
103                else :   
104                    if number > pages[pagecount]["copies"] :
105                        pages[pagecount]["copies"] = number
106            elif line.startswith(r"%%BeginNonPPDFeature: NumCopies ") :
107                # handle # of copies set by some Windows printer driver
108                try :
109                    number = int(line.strip().split()[2])
110                except :     
111                    pass
112                else :   
113                    if number > pages[pagecount]["copies"] :
114                        pages[pagecount]["copies"] = number
115            elif line.startswith("1 dict dup /NumCopies ") :
116                # handle # of copies set by mozilla/kprinter
117                try :
118                    number = int(line.strip().split()[4])
119                except :     
120                    pass
121                else :   
122                    if number > pages[pagecount]["copies"] :
123                        pages[pagecount]["copies"] = number
124            elif line.startswith("/languagelevel where{pop languagelevel}{1}ifelse 2 ge{1 dict dup/NumCopies") :
125                try :
126                    number = int(previousline.strip()[2:])
127                except :
128                    pass
129                else :
130                    if number > pages[pagecount]["copies"] :
131                        pages[pagecount]["copies"] = number
132            previousline = line
133           
134        # extract max number of copies to please the ghostscript parser, just   
135        # in case we will use it later
136        self.copies = max([ v["copies"] for (k, v) in pages.items() ])
137       
138        # now apply the number of copies to each page
139        for pnum in range(1, pagecount + 1) :
140            page = pages.get(pnum, pages.get(1, { "copies" : 1 }))
141            copies = page["copies"]
142            pagecount += (copies - 1)
143            self.logdebug("%s * page #%s" % (copies, pnum))
144        return pagecount
145       
146    def getJobSize(self) :   
147        """Count pages in PostScript document."""
148        self.copies = 1
149        return self.natively() or self.throughGhostScript()
150       
151def test() :       
152    """Test function."""
153    if (len(sys.argv) < 2) or ((not sys.stdin.isatty()) and ("-" not in sys.argv[1:])) :
154        sys.argv.append("-")
155    totalsize = 0   
156    for arg in sys.argv[1:] :
157        if arg == "-" :
158            infile = sys.stdin
159            mustclose = 0
160        else :   
161            infile = open(arg, "rb")
162            mustclose = 1
163        try :
164            parser = Parser(infile, debug=1)
165            totalsize += parser.getJobSize()
166        except pdlparser.PDLParserError, msg :   
167            sys.stderr.write("ERROR: %s\n" % msg)
168            sys.stderr.flush()
169        if mustclose :   
170            infile.close()
171    print "%s" % totalsize
172   
173if __name__ == "__main__" :   
174    test()
Note: See TracBrowser for help on using the browser.