root / pkpgcounter / trunk / pkpgpdls / pclxl.py @ 375

Revision 375, 28.9 kB (checked in by jerome, 16 years ago)

The computation of ink coverage now works for both the PCL3/4/5
and PCLXL file formats, through ghostpcl.

  • 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, 2006 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
24"""This modules implements a page counter for PCLXL (aka PCL6) documents."""
25
26import sys
27import os
28import mmap
29from struct import unpack
30
31import pdlparser
32import pjl
33
34class Parser(pdlparser.PDLParser) :
35    """A parser for PCLXL (aka PCL6) documents."""
36    totiffcommand = 'pcl6 -sDEVICE=pdfwrite -dPARANOIDSAFER -dNOPAUSE -dBATCH -dQUIET -sOutputFile=- - | gs -sDEVICE=tiff24nc -dPARANOIDSAFER -dNOPAUSE -dBATCH -dQUIET -r%(dpi)i -sOutputFile="%(fname)s" -'
37    mediasizes = { 
38                    0 : "Letter",
39                    1 : "Legal",
40                    2 : "A4",
41                    3 : "Executive",
42                    4 : "Ledger",
43                    5 : "A3",
44                    6 : "COM10Envelope",
45                    7 : "MonarchEnvelope",
46                    8 : "C5Envelope",
47                    9 : "DLEnvelope",
48                    10 : "JB4",
49                    11 : "JB5",
50                    12 : "B5Envelope",
51                    12 : "B5",
52                    14 : "JPostcard",
53                    15 : "JDoublePostcard",
54                    16 : "A5",
55                    17 : "A6",
56                    18 : "JB6",
57                    19 : "JIS8K",
58                    20 : "JIS16K",
59                    21 : "JISExec",
60                    96 : "Default",
61                 }   
62                 
63    mediasources = {             
64                     0 : "Default",
65                     1 : "Auto",
66                     2 : "Manual",
67                     3 : "MultiPurpose",
68                     4 : "UpperCassette",
69                     5 : "LowerCassette",
70                     6 : "EnvelopeTray",
71                     7 : "ThirdCassette",
72                   }
73                   
74    orientations = {               
75                     0 : "Portrait",
76                     1 : "Landscape",
77                     2 : "ReversePortrait",
78                     3 : "ReverseLandscape",
79                     4 : "Default",
80                   }
81           
82    def isValid(self) :   
83        """Returns 1 if data is PCLXL aka PCL6, else 0."""
84        if ((self.firstblock[:128].find("\033%-12345X") != -1) and \
85             (self.firstblock.find(" HP-PCL XL;") != -1) and \
86             ((self.firstblock.find("LANGUAGE=PCLXL") != -1) or \
87              (self.firstblock.find("LANGUAGE = PCLXL") != -1))) :
88            self.logdebug("DEBUG: Input file is in the PCLXL (aka PCL6) format.")
89            return 1
90        else :   
91            return 0
92           
93    def beginPage(self, nextpos) :
94        """Indicates the beginning of a new page, and extracts media information."""
95        self.pagecount += 1
96       
97        # Default values
98        mediatypelabel = "Plain"
99        mediasourcelabel = "Main"
100        mediasizelabel = "Default"
101        orientationlabel = "Portrait"
102        duplexmode = None
103       
104        # Now go upstream to decode media type, size, source, and orientation
105        # this saves time because we don't need a complete parser !
106        minfile = self.minfile
107        pos = nextpos - 2
108        while pos > 0 : # safety check : don't go back to far !
109            val = ord(minfile[pos])
110            if val in (0x44, 0x48, 0x41) : # if previous endPage or openDataSource or beginSession (first page)
111                break
112            if val == 0x26 :   
113                mediasource = ord(minfile[pos - 2])
114                mediasourcelabel = self.mediasources.get(mediasource, str(mediasource))
115                pos -= 4
116            elif val == 0x25 :
117                while (pos > 0) and (ord(minfile[pos]) != 0xc0) :
118                    # we search the preceding ubyte tag
119                    pos -= 1
120                if pos > 0 :
121                    if ord(minfile[pos-1]) == 0xc8 :
122                        # if we found an ubyte_array then the media
123                        # size is completely spelled
124                        arraylength = ord(minfile[pos+1])
125                        mediasizelabel = minfile[pos+2:pos+2+arraylength].title()
126                        pos -= 1
127                    else :   
128                        # if we just found an ubyte, then the media
129                        # size is known by its index
130                        mediasize = ord(minfile[pos+1])
131                        mediasizelabel = self.mediasizes.get(mediasize, str(mediasize))
132                    pos -= 1 
133                    # self.logdebug("Media size : %s" % mediasizelabel)
134            elif val == 0x28 :   
135                orientation = ord(minfile[pos - 2])
136                orientationlabel = self.orientations.get(orientation, str(orientation))
137                pos -= 4
138            elif val == 0x27 :   
139                savepos = pos
140                pos -= 1
141                startpos = size = None 
142                while pos > 0 : # safety check : don't go back to far !
143                    val = ord(minfile[pos])
144                    pos -= 1   
145                    if val == 0xc8 :
146                        length = self.tags[ord(minfile[pos+2])] # will probably always be a byte or uint16
147                        if length == 1 :   
148                            startpos = pos + 4
149                            size = unpack("B", self.minfile[pos+3:startpos])[0]
150                        elif length == 2 :   
151                            startpos = pos + 5
152                            size = unpack(self.unpackShort, self.minfile[pos+3:startpos])[0]
153                        elif length == 4 :   
154                            startpos = pos + 7
155                            size = unpack(self.unpackLong, self.minfile[pos+3:startpos])[0]
156                        else :   
157                            raise pdlparser.PDLParserError, "Error on size at %s : %s" % (pos+2, length)
158                        break
159                mediatypelabel = minfile[startpos:startpos+size]
160                # self.logdebug("Media type : %s" % mediatypelabel)
161            elif val == 0x34 :   
162                duplexmode = "Simplex"
163                pos -= 2
164            elif val in (0x35, 0x36) :   
165                duplexmode = "Duplex"
166                pos -= 2
167            # else : TODO : CUSTOM MEDIA SIZE AND UNIT !
168            else :   
169                pos -= 1  # ignored
170        self.pages[self.pagecount] = { "copies" : 1, 
171                                       "orientation" : orientationlabel, 
172                                       "mediatype" : mediatypelabel, 
173                                       "mediasize" : mediasizelabel,
174                                       "mediasource" : mediasourcelabel,
175                                       "duplex" : duplexmode,
176                                     } 
177        return 0
178       
179    def endPage(self, nextpos) :   
180        """Indicates the end of a page."""
181        pos3 = nextpos - 3
182        minfile = self.minfile
183        if minfile[pos3:nextpos-1] == self.setNumberOfCopies :
184            # The EndPage operator may be preceded by a PageCopies attribute
185            # So set number of copies for current page.
186            # From what I read in PCLXL documentation, the number
187            # of copies is an unsigned 16 bits integer
188            try :
189                self.pages[self.pagecount]["copies"] = unpack(self.unpackShort, minfile[pos3-2:pos3])[0]
190            except KeyError :   
191                self.logdebug("It looks like this PCLXL file is corrupted.")
192        return 0
193       
194    def setColorSpace(self, nextpos) :   
195        """Changes the color space."""
196        if self.minfile[nextpos-4:nextpos-1] == self.RGBColorSpace : # TODO : doesn't seem to handle all cases !
197            self.iscolor = 1
198        return 0
199           
200    def array_Generic(self, nextpos, size) :
201        """Handles all arrays."""
202        pos = nextpos
203        datatype = ord(self.minfile[pos])
204        pos += 1
205        length = self.tags[datatype]
206        if callable(length) :
207            length = length(pos)
208        try :
209            return 1 + length + size * unpack(self.unpackType[length], self.minfile[pos:pos+length])[0]
210        except KeyError :
211            raise pdlparser.PDLParserError, "Error on array size at %x" % nextpos
212           
213    def array_8(self, nextpos) :   
214        """Handles byte arrays."""
215        return self.array_Generic(nextpos, 1)
216       
217    def array_16(self, nextpos) :
218        """Handles byte arrays."""
219        return self.array_Generic(nextpos, 2)
220       
221    def array_32(self, nextpos) :
222        """Handles byte arrays."""
223        return self.array_Generic(nextpos, 4)
224       
225    def embeddedDataSmall(self, nextpos) :
226        """Handle small amounts of data."""
227        return 1 + ord(self.minfile[nextpos])
228       
229    def embeddedData(self, nextpos) :
230        """Handle normal amounts of data."""
231        return 4 + unpack(self.unpackLong, self.minfile[nextpos:nextpos+4])[0]
232       
233    def littleEndian(self, nextpos) :
234        """Toggles to little endianness."""
235        self.unpackType = { 1 : "B", 2 : "<H", 4 : "<I" }
236        self.unpackShort = self.unpackType[2]
237        self.unpackLong = self.unpackType[4]
238        return 0
239       
240    def bigEndian(self, nextpos) :
241        """Toggles to big endianness."""
242        self.unpackType = { 1 : "B", 2 : ">H", 4 : ">I" }
243        self.unpackShort = self.unpackType[2]
244        self.unpackLong = self.unpackType[4]
245        return 0
246   
247    def reservedForFutureUse(self, nextpos) :
248        """Outputs something when a reserved byte is encountered."""
249        self.logdebug("Byte at %x is out of the PCLXL Protocol Class 2.0 Specification" % nextpos)
250        return 0   
251       
252    def x46_class3(self, nextpos) :
253        """Undocumented tag 0x46 in class 3.0 streams."""
254        pos = nextpos - 3
255        minfile = self.minfile
256        val = ord(minfile[pos])
257        while val == 0xf8 :
258            funcid = ord(minfile[pos+1])
259            try :
260                offset = self.x46_functions[funcid]
261            except KeyError :   
262                self.logdebug("Unexpected subfunction 0x%02x for undocumented tag 0x46 at %x" % (funcid, nextpos))
263                break
264            else :   
265                pos -= offset
266                length = self.tags[ord(self.minfile[pos])]
267                if callable(length) :
268                    length = length(pos+1)
269                if funcid == 0x92 : # we want to skip these blocks
270                    try :
271                        return unpack(self.unpackType[length], self.minfile[pos+1:pos+length+1])[0]
272                    except KeyError :
273                        raise pdlparser.PDLParserError, "Error on size '%s' at %x" % (length, pos+1)
274            val = ord(minfile[pos])
275        return 0   
276       
277    def escape(self, nextpos) :   
278        """Handles the ESC code."""
279        pos = endpos = nextpos
280        minfile = self.minfile
281        if minfile[pos : pos+8] == r"%-12345X" :
282            endpos = pos + 9
283            endmark = chr(0x0c) + chr(0x00) + chr(0x1b)
284            asciilimit = chr(0x80)
285            quotes = 0
286            while (minfile[endpos] not in endmark) and \
287                   ((minfile[endpos] < asciilimit) or (quotes % 2)) :
288                if minfile[endpos] == '"' :
289                    quotes += 1
290                endpos += 1
291               
292            # Store this in a per page mapping.   
293            # NB : First time will be at page 0 (i.e. **before** page 1) !
294            stuff = self.escapedStuff.setdefault(self.pagecount, [])
295            stuff.append(minfile[pos : endpos])
296            self.logdebug("Escaped datas : [%s]" % repr(minfile[pos : endpos]))
297        return endpos - pos
298       
299    def skipKyoceraPrescribe(self, nextpos) :
300        """Skips Kyocera Prescribe commands."""
301        pos = nextpos - 1
302        minfile = self.minfile
303        if minfile[pos:pos+3] == "!R!" :
304            while (pos - nextpos) < 1024 :   # This is a realistic upper bound, to avoid infinite loops
305                if (minfile[pos] == ";") and (minfile[pos-4:pos] == "EXIT") :
306                    pos += 1
307                    prescribe = self.prescribeStuff.setdefault(self.pagecount, [])
308                    prescribe.append(minfile[nextpos-1:pos])
309                    self.logdebug("Prescribe commands : [%s]" % repr(minfile[nextpos-1:pos]))
310                    return (pos - nextpos)
311                pos += 1   
312        else :
313            return 0
314           
315    def getJobSize(self) :
316        """Counts pages in a PCLXL (PCL6) document.
317       
318           Algorithm by Jerome Alet.
319           
320           The documentation used for this was :
321         
322           HP PCL XL Feature Reference
323           Protocol Class 2.0
324           http://www.hpdevelopersolutions.com/downloads/64/358/xl_ref20r22.pdf
325           
326           Protocol Class 2.1 Supplement
327           xl_ref21.pdf
328           
329           Protocol Class 3.0 Supplement
330           xl_refsup30r089.pdf
331        """
332        self.iscolor = None
333        found = 0
334        while not found :
335            line = self.infile.readline()
336            if not line :
337                break
338            pos = line.find(" HP-PCL XL;")   
339            if pos != -1 :
340                found = 1
341                endian = ord(line[pos - 1])
342                if endian == 0x29 :
343                    self.littleEndian(0)
344                elif endian == 0x28 :   
345                    self.bigEndian(0)
346                # elif endian == 0x27 : # TODO : This is the ESC code : parse it for PJL statements !
347                #
348                else :   
349                    raise pdlparser.PDLParserError, "Unknown endianness marker 0x%02x at start !" % endian
350        if not found :
351            raise pdlparser.PDLParserError, "This file doesn't seem to be PCLXL (aka PCL6)"
352           
353        # Initialize Media Sources
354        for i in range(8, 256) :
355            self.mediasources[i] = "ExternalTray%03i" % (i - 7)
356           
357        # Initialize table of tags
358        self.tags = [ 0 ] * 256   
359       
360        self.tags[0x1b] = self.escape # The escape code
361       
362        self.tags[0x21] = self.skipKyoceraPrescribe # 0x21 is not normally used
363       
364        # GhostScript's sources tell us that HP printers
365        # only accept little endianness, but we can handle both.
366        self.tags[0x28] = self.bigEndian    # BigEndian
367        self.tags[0x29] = self.littleEndian # LittleEndian
368       
369        self.tags[0x43] = self.beginPage    # BeginPage
370        self.tags[0x44] = self.endPage      # EndPage
371        self.tags[0x45] = self.reservedForFutureUse # reserved
372       
373        self.tags[0x46] = self.x46_class3 
374       
375        self.tags[0x4a] = self.reservedForFutureUse # reserved
376        self.tags[0x4b] = self.reservedForFutureUse # reserved
377        self.tags[0x4c] = self.reservedForFutureUse # reserved
378        self.tags[0x4d] = self.reservedForFutureUse # reserved
379        self.tags[0x4e] = self.reservedForFutureUse # reserved
380       
381        self.tags[0x56] = self.reservedForFutureUse # TODO : documentation not clear about reserved status
382       
383        self.tags[0x57] = self.reservedForFutureUse # reserved
384       
385        self.tags[0x59] = self.reservedForFutureUse # reserved
386        self.tags[0x5a] = self.reservedForFutureUse # reserved
387       
388        self.tags[0x6a] = self.setColorSpace    # to detect color/b&w mode
389       
390        self.tags[0x87] = self.reservedForFutureUse # reserved
391        self.tags[0x88] = self.reservedForFutureUse # reserved
392        self.tags[0x89] = self.reservedForFutureUse # reserved
393        self.tags[0x8a] = self.reservedForFutureUse # reserved
394       
395        self.tags[0x8b] = self.reservedForFutureUse # reserved
396       
397        self.tags[0x8c] = self.reservedForFutureUse # reserved
398        self.tags[0x8d] = self.reservedForFutureUse # reserved
399        self.tags[0x8e] = self.reservedForFutureUse # reserved
400        self.tags[0x8f] = self.reservedForFutureUse # reserved
401        self.tags[0x90] = self.reservedForFutureUse # reserved
402       
403        self.tags[0x9a] = self.reservedForFutureUse # reserved
404        self.tags[0x9c] = self.reservedForFutureUse # reserved
405       
406        self.tags[0xa4] = self.reservedForFutureUse # reserved
407        self.tags[0xa5] = self.reservedForFutureUse # reserved
408        self.tags[0xa6] = self.reservedForFutureUse # reserved
409        self.tags[0xa7] = self.reservedForFutureUse # reserved
410       
411        self.tags[0xaa] = self.reservedForFutureUse # reserved
412        self.tags[0xab] = self.reservedForFutureUse # reserved
413        self.tags[0xac] = self.reservedForFutureUse # reserved
414        self.tags[0xad] = self.reservedForFutureUse # reserved
415        self.tags[0xae] = self.reservedForFutureUse # reserved
416        self.tags[0xaf] = self.reservedForFutureUse # reserved
417       
418        self.tags[0xb7] = self.reservedForFutureUse # reserved
419       
420        self.tags[0xba] = self.reservedForFutureUse # reserved
421        self.tags[0xbb] = self.reservedForFutureUse # reserved
422        self.tags[0xbc] = self.reservedForFutureUse # reserved
423        self.tags[0xbd] = self.reservedForFutureUse # reserved
424        self.tags[0xbe] = self.reservedForFutureUse # reserved
425       
426        # self.tags[0xbf] = self.passThrough # PassThrough mode should already be taken care of automatically
427       
428        self.tags[0xc0] = 1 # ubyte
429        self.tags[0xc1] = 2 # uint16
430        self.tags[0xc2] = 4 # uint32
431        self.tags[0xc3] = 2 # sint16
432        self.tags[0xc4] = 4 # sint32
433        self.tags[0xc5] = 4 # real32
434       
435        self.tags[0xc6] = self.reservedForFutureUse # reserved
436        self.tags[0xc7] = self.reservedForFutureUse # reserved
437       
438        self.tags[0xc8] = self.array_8  # ubyte_array
439        self.tags[0xc9] = self.array_16 # uint16_array
440        self.tags[0xca] = self.array_32 # uint32_array
441        self.tags[0xcb] = self.array_16 # sint16_array
442        self.tags[0xcc] = self.array_32 # sint32_array
443        self.tags[0xcd] = self.array_32 # real32_array
444       
445        self.tags[0xce] = self.reservedForFutureUse # reserved
446        self.tags[0xcf] = self.reservedForFutureUse # reserved
447       
448        self.tags[0xd0] = 2 # ubyte_xy
449        self.tags[0xd1] = 4 # uint16_xy
450        self.tags[0xd2] = 8 # uint32_xy
451        self.tags[0xd3] = 4 # sint16_xy
452        self.tags[0xd4] = 8 # sint32_xy
453        self.tags[0xd5] = 8 # real32_xy
454        self.tags[0xd6] = self.reservedForFutureUse # reserved
455        self.tags[0xd7] = self.reservedForFutureUse # reserved
456        self.tags[0xd8] = self.reservedForFutureUse # reserved
457        self.tags[0xd9] = self.reservedForFutureUse # reserved
458        self.tags[0xda] = self.reservedForFutureUse # reserved
459        self.tags[0xdb] = self.reservedForFutureUse # reserved
460        self.tags[0xdc] = self.reservedForFutureUse # reserved
461        self.tags[0xdd] = self.reservedForFutureUse # reserved
462        self.tags[0xde] = self.reservedForFutureUse # reserved
463        self.tags[0xdf] = self.reservedForFutureUse # reserved
464       
465        self.tags[0xe0] = 4  # ubyte_box
466        self.tags[0xe1] = 8  # uint16_box
467        self.tags[0xe2] = 16 # uint32_box
468        self.tags[0xe3] = 8  # sint16_box
469        self.tags[0xe4] = 16 # sint32_box
470        self.tags[0xe5] = 16 # real32_box
471        self.tags[0xe6] = self.reservedForFutureUse # reserved
472        self.tags[0xe7] = self.reservedForFutureUse # reserved
473        self.tags[0xe8] = self.reservedForFutureUse # reserved
474        self.tags[0xe9] = self.reservedForFutureUse # reserved
475        self.tags[0xea] = self.reservedForFutureUse # reserved
476        self.tags[0xeb] = self.reservedForFutureUse # reserved
477        self.tags[0xec] = self.reservedForFutureUse # reserved
478        self.tags[0xed] = self.reservedForFutureUse # reserved
479        self.tags[0xee] = self.reservedForFutureUse # reserved
480        self.tags[0xef] = self.reservedForFutureUse # reserved
481       
482        self.tags[0xf0] = self.reservedForFutureUse # reserved
483        self.tags[0xf1] = self.reservedForFutureUse # reserved
484        self.tags[0xf2] = self.reservedForFutureUse # reserved
485        self.tags[0xf3] = self.reservedForFutureUse # reserved
486        self.tags[0xf4] = self.reservedForFutureUse # reserved
487        self.tags[0xf5] = self.reservedForFutureUse # reserved
488        self.tags[0xf6] = self.reservedForFutureUse # reserved
489        self.tags[0xf7] = self.reservedForFutureUse # reserved
490       
491        self.tags[0xf8] = 1 # attr_ubyte
492        self.tags[0xf9] = 2 # attr_uint16
493       
494        self.tags[0xfa] = self.embeddedData      # dataLength
495        self.tags[0xfb] = self.embeddedDataSmall # dataLengthByte
496       
497        self.tags[0xfc] = self.reservedForFutureUse # reserved
498        self.tags[0xfd] = self.reservedForFutureUse # reserved
499        self.tags[0xfe] = self.reservedForFutureUse # reserved
500        self.tags[0xff] = self.reservedForFutureUse # reserved
501           
502        # color spaces   
503        self.BWColorSpace = "".join([chr(0x00), chr(0xf8), chr(0x03)])
504        self.GrayColorSpace = "".join([chr(0x01), chr(0xf8), chr(0x03)])
505        self.RGBColorSpace = "".join([chr(0x02), chr(0xf8), chr(0x03)])
506       
507        # set number of copies
508        self.setNumberOfCopies = "".join([chr(0xf8), chr(0x31)]) 
509       
510        # subcodes for undocumented tag 0x46 and the negative
511        # offset to grab the value from.
512        self.x46_functions = { 0x91 : 5,
513                               0x92 : 5,
514                               0x93 : 3,
515                               0x94 : 3,
516                               0x95 : 5,
517                               0x96 : 2,
518                               0x97 : 2,
519                               0x98 : 2,
520                             }
521                             
522        infileno = self.infile.fileno()
523        self.pages = { 0 : { "copies" : 1, 
524                             "orientation" : "Default", 
525                             "mediatype" : "Plain", 
526                             "mediasize" : "Default", 
527                             "mediasource" : "Default", 
528                             "duplex" : None,
529                           } 
530                     }     
531        self.minfile = minfile = mmap.mmap(infileno, os.fstat(infileno)[6], prot=mmap.PROT_READ, flags=mmap.MAP_SHARED)
532        tags = self.tags
533        self.pagecount = 0
534        self.escapedStuff = {}   # For escaped datas, mostly PJL commands
535        self.prescribeStuff = {} # For Kyocera Prescribe commands
536        pos = oldpos = 0
537        try :
538            try :
539                while 1 :
540                    try :
541                        tag = ord(minfile[pos])
542                    except OverflowError :   
543                        pos = oldpos + 1
544                    pos += 1
545                    length = tags[tag]
546                    if length :
547                        if callable(length) :   
548                            length = length(pos)
549                        oldpos = pos   
550                        pos += length   
551            except IndexError : # EOF ?           
552                pass
553        finally :
554            self.minfile.close()
555           
556        # now handle number of copies for each page (may differ).
557        if self.iscolor :
558            colormode = "Color"
559        else :   
560            colormode = "BW"
561           
562        defaultduplexmode = "Simplex"
563        defaultpapersize = ""
564        defaultpjlcopies = 1   
565        oldpjlcopies = -1
566        oldduplexmode = ""
567        oldpapersize = ""
568        for pnum in range(1, self.pagecount + 1) :
569            # if no number of copies defined, take 1, as explained
570            # in PCLXL documentation.
571            # NB : is number of copies is 0, the page won't be output
572            # but the formula below is still correct : we want
573            # to decrease the total number of pages in this case.
574            page = self.pages.get(pnum, self.pages.get(1, { "copies" : 1, "mediasize" : "Default", "duplex" : None }))
575            pjlstuff = self.escapedStuff.get(pnum, self.escapedStuff.get(0, []))
576            if pjlstuff :
577                pjlparser = pjl.PJLParser("".join(pjlstuff))
578                nbdefaultcopies = int(pjlparser.default_variables.get("COPIES", -1))
579                nbcopies = int(pjlparser.environment_variables.get("COPIES", -1))
580                nbdefaultqty = int(pjlparser.default_variables.get("QTY", -1))
581                nbqty = int(pjlparser.environment_variables.get("QTY", -1))
582                if nbdefaultcopies > -1 :
583                    defaultpjlcopies = nbdefaultcopies
584                if nbdefaultqty > -1 :
585                    defaultpjlcopies = nbdefaultqty
586                if nbcopies > -1 :
587                    pjlcopies = nbcopies
588                elif nbqty > -1 :
589                    pjlcopies = nbqty
590                else :
591                    if oldpjlcopies == -1 :   
592                        pjlcopies = defaultpjlcopies
593                    else :   
594                        pjlcopies = oldpjlcopies   
595                if page["duplex"] :       
596                    duplexmode = page["duplex"]
597                else :   
598                    defaultdm = pjlparser.default_variables.get("DUPLEX", "")
599                    if defaultdm :
600                        if defaultdm.upper() == "ON" :
601                            defaultduplexmode = "Duplex"
602                        else :   
603                            defaultduplexmode = "Simplex"
604                    envdm = pjlparser.environment_variables.get("DUPLEX", "")
605                    if envdm :
606                        if envdm.upper() == "ON" :
607                            duplexmode = "Duplex"
608                        else :   
609                            duplexmode = "Simplex"
610                    else :       
611                        if not oldduplexmode :
612                            duplexmode = defaultduplexmode
613                        else :   
614                            duplexmode = oldduplexmode
615                defaultps = pjlparser.default_variables.get("PAPER", "")
616                if defaultps :
617                    defaultpapersize = defaultps
618                envps = pjlparser.environment_variables.get("PAPER", "")
619                if envps :
620                    papersize = envps
621                else :   
622                    if not oldpapersize :
623                        papersize = defaultpapersize
624                    else :   
625                        papersize = oldpapersize
626            else :       
627                if oldpjlcopies == -1 :
628                    pjlcopies = defaultpjlcopies
629                else :   
630                    pjlcopies = oldpjlcopies
631                if not oldduplexmode :
632                    duplexmode = defaultduplexmode
633                else :   
634                    duplexmode = oldduplexmode
635                if not oldpapersize :   
636                    papersize = defaultpapersize
637                else :   
638                    papersize = oldpapersize
639                duplexmode = oldduplexmode
640                papersize = oldpapersize or page["mediasize"]
641            if page["mediasize"] != "Default" :
642                papersize = page["mediasize"]
643            if not duplexmode :   
644                duplexmode = oldduplexmode or defaultduplexmode
645            oldpjlcopies = pjlcopies   
646            oldduplexmode = duplexmode
647            oldpapersize = papersize
648            copies = pjlcopies * page["copies"]
649            self.pagecount += (copies - 1)
650            self.logdebug("%s*%s*%s*%s*%s*%s*%s" % (copies, 
651                                                 page["mediatype"], 
652                                                 papersize, 
653                                                 page["orientation"], 
654                                                 page["mediasource"], 
655                                                 duplexmode, 
656                                                 colormode))
657        return self.pagecount
658       
659def test() :       
660    """Test function."""
661    if (len(sys.argv) < 2) or ((not sys.stdin.isatty()) and ("-" not in sys.argv[1:])) :
662        sys.argv.append("-")
663    totalsize = 0   
664    for arg in sys.argv[1:] :
665        if arg == "-" :
666            infile = sys.stdin
667            mustclose = 0
668        else :   
669            infile = open(arg, "rb")
670            mustclose = 1
671        try :
672            parser = Parser(infile, debug=1)
673            totalsize += parser.getJobSize()
674        except pdlparser.PDLParserError, msg :   
675            sys.stderr.write("ERROR: %s\n" % msg)
676            sys.stderr.flush()
677        if mustclose :   
678            infile.close()
679    print "%s" % totalsize
680   
681if __name__ == "__main__" :   
682    test()
Note: See TracBrowser for help on using the browser.