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

Revision 309, 29.6 kB (checked in by jerome, 18 years ago)

Improved PCLXL parser with regard to undocumented PCLXL 3.0 tags

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