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

Revision 357, 28.7 kB (checked in by jerome, 16 years ago)

Added missing docstrings, thanks to pylint.

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