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

Revision 266, 25.2 kB (checked in by jerome, 19 years ago)

Improved PCLXL parser.

  • Property svn:eol-style set to native
  • Property svn:keywords set to Auth Date Id Rev
Line 
1#! /usr/bin/env python
2# -*- coding: ISO-8859-15 -*-
3#
4# pkpgcounter : a generic Page Description Language parser
5#
6# (c) 2003, 2004, 2005 Jerome Alet <alet@librelogiciel.com>
7# This program is free software; you can redistribute it and/or modify
8# it under the terms of the GNU General Public License as published by
9# the Free Software Foundation; either version 2 of the License, or
10# (at your option) any later version.
11#
12# This program is distributed in the hope that it will be useful,
13# but WITHOUT ANY WARRANTY; without even the implied warranty of
14# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15# GNU General Public License for more details.
16#
17# You should have received a copy of the GNU General Public License
18# along with this program; if not, write to the Free Software
19# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
20#
21# $Id$
22#
23
24import sys
25import 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       
100        # Now go upstream to decode media type, size, source, and orientation
101        # this saves time because we don't need a complete parser !
102        minfile = self.minfile
103        pos = self.pos - 2
104        while pos > 0 : # safety check : don't go back to far !
105            val = ord(minfile[pos])
106            if val in (0x44, 0x48, 0x41) : # if previous endPage or openDataSource or beginSession (first page)
107                break
108            if val == 0x26 :   
109                mediasource = ord(minfile[pos - 2])
110                mediasourcelabel = self.mediasources.get(mediasource, str(mediasource))
111                pos = pos - 4
112            elif val == 0x25 :
113                mediasize = ord(minfile[pos - 2])
114                mediasizelabel = self.mediasizes.get(mediasize, str(mediasize))
115                pos = pos - 4
116            elif val == 0x28 :   
117                orientation = ord(minfile[pos - 2])
118                orientationlabel = self.orientations.get(orientation, str(orientation))
119                pos = pos - 4
120            elif val == 0x27 :   
121                savepos = pos
122                pos = pos - 1
123                while pos > 0 : # safety check : don't go back to far !
124                    val = ord(minfile[pos])
125                    pos -= 1   
126                    if val == 0xc8 :
127                        break
128                mediatypelabel = minfile[pos:savepos] # TODO : INCORRECT, WE HAVE TO STRIP OUT THE UBYTE ARRAY'S LENGTH !!!
129            # else : TODO : CUSTOM MEDIA SIZE AND UNIT !
130            else :   
131                pos = pos - 2   # ignored
132        self.pages[self.pagecount] = { "copies" : 1, 
133                                       "orientation" : orientationlabel, 
134                                       "mediatype" : mediatypelabel, 
135                                       "mediasize" : mediasizelabel,
136                                       "mediasource" : mediasourcelabel,
137                                     } 
138        return 0
139       
140    def endPage(self) :   
141        """Indicates the end of a page."""
142        pos = self.pos
143        pos3 = pos - 3
144        minfile = self.minfile
145        if minfile[pos3:pos-1] == self.setNumberOfCopies :
146            # The EndPage operator may be preceded by a PageCopies attribute
147            # So set number of copies for current page.
148            # From what I read in PCLXL documentation, the number
149            # of copies is an unsigned 16 bits integer
150            try :
151                self.pages[self.pagecount]["copies"] = unpack(self.endianness + "H", minfile[pos-5:pos3])[0]
152            except KeyError :   
153                self.logdebug("It looks like this PCLXL file is corrupted.")
154        return 0
155       
156    def setColorSpace(self) :   
157        """Changes the color space."""
158        if self.minfile[self.pos-4:self.pos-1] == self.RGBColorSpace :
159            self.iscolor = 1
160        return 0
161           
162    def array_8(self) :   
163        """Handles byte arrays."""
164        pos = self.pos
165        datatype = self.minfile[pos]
166        pos += 1
167        length = self.tags[ord(datatype)]
168        if callable(length) :
169            self.pos = pos
170            length = length()
171            pos = self.pos
172        posl = pos + length
173        self.pos = posl
174        if length == 1 :   
175            return unpack("B", self.minfile[pos:posl])[0]
176        elif length == 2 :   
177            return unpack(self.endianness + "H", self.minfile[pos:posl])[0]
178        elif length == 4 :   
179            return unpack(self.endianness + "I", self.minfile[pos:posl])[0]
180        else :   
181            raise pdlparser.PDLParserError, "Error on array size at %s" % self.pos
182       
183    def array_16(self) :   
184        """Handles byte arrays."""
185        pos = self.pos
186        datatype = self.minfile[pos]
187        pos += 1
188        length = self.tags[ord(datatype)]
189        if callable(length) :
190            self.pos = pos
191            length = length()
192            pos = self.pos
193        posl = pos + length
194        self.pos = posl
195        if length == 1 :   
196            return 2 * unpack("B", self.minfile[pos:posl])[0]
197        elif length == 2 :   
198            return 2 * unpack(self.endianness + "H", self.minfile[pos:posl])[0]
199        elif length == 4 :   
200            return 2 * unpack(self.endianness + "I", self.minfile[pos:posl])[0]
201        else :   
202            raise pdlparser.PDLParserError, "Error on array size at %s" % self.pos
203       
204    def array_32(self) :   
205        """Handles byte arrays."""
206        pos = self.pos
207        datatype = self.minfile[pos]
208        pos += 1
209        length = self.tags[ord(datatype)]
210        if callable(length) :
211            self.pos = pos
212            length = length()
213            pos = self.pos
214        posl = pos + length
215        self.pos = posl
216        if length == 1 :   
217            return 4 * unpack("B", self.minfile[pos:posl])[0]
218        elif length == 2 :   
219            return 4 * unpack(self.endianness + "H", self.minfile[pos:posl])[0]
220        elif length == 4 :   
221            return 4 * unpack(self.endianness + "I", self.minfile[pos:posl])[0]
222        else :   
223            raise pdlparser.PDLParserError, "Error on array size at %s" % self.pos
224       
225    def embeddedDataSmall(self) :
226        """Handle small amounts of data."""
227        pos = self.pos
228        length = ord(self.minfile[pos])
229        self.pos = pos + 1
230        return length
231       
232    def embeddedData(self) :
233        """Handle normal amounts of data."""
234        pos = self.pos
235        pos4 = pos + 4
236        self.pos = pos4
237        return unpack(self.endianness + "I", self.minfile[pos:pos4])[0]
238       
239    def littleEndian(self) :       
240        """Toggles to little endianness."""
241        self.endianness = "<" # little endian
242        return 0
243       
244    def bigEndian(self) :   
245        """Toggles to big endianness."""
246        self.endianness = ">" # big endian
247        return 0
248   
249    def reservedForFutureUse(self) :
250        """Outputs something when a reserved byte is encountered."""
251        self.logdebug("Byte at %s is out of the PCLXL Protocol Class 2.0 Specification" % self.pos)
252        return 0   
253       
254    def escape(self) :   
255        """Handles the ESC code."""
256        pos = endpos = self.pos
257        if self.minfile[pos : pos+8] == r"%-12345X" :
258            endpos = pos + 9
259            endmark = chr(0x0c) + chr(0x00) + chr(0x1b)
260            asciilimit = chr(0x80)
261            while (self.minfile[endpos] not in endmark) and (self.minfile[endpos] < asciilimit) :
262                endpos += 1
263               
264            # Store this in a per page mapping.   
265            # NB : First time will be at page 0 (i.e. **before** page 1) !
266            stuff = self.escapedStuff.setdefault(self.pagecount, [])
267            stuff.append(self.minfile[pos : endpos])
268            self.logdebug("Escaped datas : [%s]" % repr(self.minfile[pos : endpos]))
269        return endpos - pos
270       
271    def getJobSize(self) :
272        """Counts pages in a PCLXL (PCL6) document.
273       
274           Algorithm by Jerome Alet.
275           
276           The documentation used for this was :
277         
278           HP PCL XL Feature Reference
279           Protocol Class 2.0
280           http://www.hpdevelopersolutions.com/downloads/64/358/xl_ref20r22.pdf
281           
282           Protocol Class 2.1 Supplement
283           xl_ref21.pdf
284           
285           Protocol Class 3.0 Supplement
286           xl_refsup30r089.pdf
287        """
288        self.iscolor = None
289        self.endianness = None
290        found = 0
291        while not found :
292            line = self.infile.readline()
293            if not line :
294                break
295            if line[1:12] == " HP-PCL XL;" :
296                found = 1
297                endian = ord(line[0])
298                if endian == 0x29 :
299                    self.littleEndian()
300                elif endian == 0x28 :   
301                    self.bigEndian()
302                # elif endian == 0x27 : # TODO : This is the ESC code : parse it for PJL statements !
303                #
304                else :   
305                    raise pdlparser.PDLParserError, "Unknown endianness marker 0x%02x at start !" % endian
306        if not found :
307            raise pdlparser.PDLParserError, "This file doesn't seem to be PCLXL (aka PCL6)"
308           
309        # Initialize Media Sources
310        for i in range(8, 256) :
311            self.mediasources[i] = "ExternalTray%03i" % (i - 7)
312           
313        # Initialize table of tags
314        self.tags = [ 0 ] * 256   
315       
316        self.tags[0x1b] = self.escape # The escape code
317       
318        # GhostScript's sources tell us that HP printers
319        # only accept little endianness, but we can handle both.
320        self.tags[0x28] = self.bigEndian    # BigEndian
321        self.tags[0x29] = self.littleEndian # LittleEndian
322       
323        self.tags[0x43] = self.beginPage    # BeginPage
324        self.tags[0x44] = self.endPage      # EndPage
325        self.tags[0x45] = self.reservedForFutureUse # reserved
326        self.tags[0x46] = self.reservedForFutureUse # reserved
327       
328        self.tags[0x4a] = self.reservedForFutureUse # reserved
329        self.tags[0x4b] = self.reservedForFutureUse # reserved
330        self.tags[0x4c] = self.reservedForFutureUse # reserved
331        self.tags[0x4d] = self.reservedForFutureUse # reserved
332        self.tags[0x4e] = self.reservedForFutureUse # reserved
333       
334        self.tags[0x56] = self.reservedForFutureUse # TODO : documentation not clear about reserved status
335       
336        self.tags[0x57] = self.reservedForFutureUse # reserved
337        self.tags[0x58] = self.reservedForFutureUse # reserved
338        self.tags[0x59] = self.reservedForFutureUse # reserved
339        self.tags[0x5a] = self.reservedForFutureUse # reserved
340       
341        self.tags[0x6a] = self.setColorSpace    # to detect color/b&w mode
342       
343        self.tags[0x83] = self.reservedForFutureUse # reserved
344       
345        self.tags[0x87] = self.reservedForFutureUse # reserved
346        self.tags[0x88] = self.reservedForFutureUse # reserved
347        self.tags[0x89] = self.reservedForFutureUse # reserved
348        self.tags[0x8a] = self.reservedForFutureUse # reserved
349        self.tags[0x8b] = self.reservedForFutureUse # reserved
350        self.tags[0x8c] = self.reservedForFutureUse # reserved
351        self.tags[0x8d] = self.reservedForFutureUse # reserved
352        self.tags[0x8e] = self.reservedForFutureUse # reserved
353        self.tags[0x8f] = self.reservedForFutureUse # reserved
354        self.tags[0x90] = self.reservedForFutureUse # reserved
355       
356        self.tags[0x92] = self.reservedForFutureUse # reserved
357       
358        self.tags[0x94] = self.reservedForFutureUse # reserved
359       
360        self.tags[0x9a] = self.reservedForFutureUse # reserved
361        self.tags[0x9c] = self.reservedForFutureUse # reserved
362       
363        self.tags[0xa4] = self.reservedForFutureUse # reserved
364        self.tags[0xa5] = self.reservedForFutureUse # reserved
365        self.tags[0xa6] = self.reservedForFutureUse # reserved
366        self.tags[0xa7] = self.reservedForFutureUse # reserved
367       
368        self.tags[0xaa] = self.reservedForFutureUse # reserved
369        self.tags[0xab] = self.reservedForFutureUse # reserved
370        self.tags[0xac] = self.reservedForFutureUse # reserved
371        self.tags[0xad] = self.reservedForFutureUse # reserved
372        self.tags[0xae] = self.reservedForFutureUse # reserved
373        self.tags[0xaf] = self.reservedForFutureUse # reserved
374       
375        self.tags[0xb7] = self.reservedForFutureUse # reserved
376       
377        self.tags[0xba] = self.reservedForFutureUse # reserved
378        self.tags[0xbb] = self.reservedForFutureUse # reserved
379        self.tags[0xbc] = self.reservedForFutureUse # reserved
380        self.tags[0xbd] = self.reservedForFutureUse # reserved
381        self.tags[0xbe] = self.reservedForFutureUse # reserved
382       
383        self.tags[0xc0] = 1 # ubyte
384        self.tags[0xc1] = 2 # uint16
385        self.tags[0xc2] = 4 # uint32
386        self.tags[0xc3] = 2 # sint16
387        self.tags[0xc4] = 4 # sint32
388        self.tags[0xc5] = 4 # real32
389       
390        self.tags[0xc6] = self.reservedForFutureUse # reserved
391        self.tags[0xc7] = self.reservedForFutureUse # reserved
392       
393        self.tags[0xc8] = self.array_8  # ubyte_array
394        self.tags[0xc9] = self.array_16 # uint16_array
395        self.tags[0xca] = self.array_32 # uint32_array
396        self.tags[0xcb] = self.array_16 # sint16_array
397        self.tags[0xcc] = self.array_32 # sint32_array
398        self.tags[0xcd] = self.array_32 # real32_array
399       
400        self.tags[0xce] = self.reservedForFutureUse # reserved
401        self.tags[0xcf] = self.reservedForFutureUse # reserved
402       
403        self.tags[0xd0] = 2 # ubyte_xy
404        self.tags[0xd1] = 4 # uint16_xy
405        self.tags[0xd2] = 8 # uint32_xy
406        self.tags[0xd3] = 4 # sint16_xy
407        self.tags[0xd4] = 8 # sint32_xy
408        self.tags[0xd5] = 8 # real32_xy
409        self.tags[0xd6] = self.reservedForFutureUse # reserved
410        self.tags[0xd7] = self.reservedForFutureUse # reserved
411        self.tags[0xd8] = self.reservedForFutureUse # reserved
412        self.tags[0xd9] = self.reservedForFutureUse # reserved
413        self.tags[0xda] = self.reservedForFutureUse # reserved
414        self.tags[0xdb] = self.reservedForFutureUse # reserved
415        self.tags[0xdc] = self.reservedForFutureUse # reserved
416        self.tags[0xdd] = self.reservedForFutureUse # reserved
417        self.tags[0xde] = self.reservedForFutureUse # reserved
418        self.tags[0xdf] = self.reservedForFutureUse # reserved
419       
420        self.tags[0xe0] = 4  # ubyte_box
421        self.tags[0xe1] = 8  # uint16_box
422        self.tags[0xe2] = 16 # uint32_box
423        self.tags[0xe3] = 8  # sint16_box
424        self.tags[0xe4] = 16 # sint32_box
425        self.tags[0xe5] = 16 # real32_box
426        self.tags[0xe6] = self.reservedForFutureUse # reserved
427        self.tags[0xe7] = self.reservedForFutureUse # reserved
428        self.tags[0xe8] = self.reservedForFutureUse # reserved
429        self.tags[0xe9] = self.reservedForFutureUse # reserved
430        self.tags[0xea] = self.reservedForFutureUse # reserved
431        self.tags[0xeb] = self.reservedForFutureUse # reserved
432        self.tags[0xec] = self.reservedForFutureUse # reserved
433        self.tags[0xed] = self.reservedForFutureUse # reserved
434        self.tags[0xee] = self.reservedForFutureUse # reserved
435        self.tags[0xef] = self.reservedForFutureUse # reserved
436       
437        self.tags[0xf0] = self.reservedForFutureUse # reserved
438        self.tags[0xf1] = self.reservedForFutureUse # reserved
439        self.tags[0xf2] = self.reservedForFutureUse # reserved
440        self.tags[0xf3] = self.reservedForFutureUse # reserved
441        self.tags[0xf4] = self.reservedForFutureUse # reserved
442        self.tags[0xf5] = self.reservedForFutureUse # reserved
443        self.tags[0xf6] = self.reservedForFutureUse # reserved
444        self.tags[0xf7] = self.reservedForFutureUse # reserved
445       
446        self.tags[0xf8] = 1 # attr_ubyte
447        self.tags[0xf9] = 2 # attr_uint16
448       
449        self.tags[0xfa] = self.embeddedData      # dataLength
450        self.tags[0xfb] = self.embeddedDataSmall # dataLengthByte
451       
452        self.tags[0xfc] = self.reservedForFutureUse # reserved
453        self.tags[0xfd] = self.reservedForFutureUse # reserved
454        self.tags[0xfe] = self.reservedForFutureUse # reserved
455        self.tags[0xff] = self.reservedForFutureUse # reserved
456           
457        # color spaces   
458        self.BWColorSpace = "".join([chr(0x00), chr(0xf8), chr(0x03)])
459        self.GrayColorSpace = "".join([chr(0x01), chr(0xf8), chr(0x03)])
460        self.RGBColorSpace = "".join([chr(0x02), chr(0xf8), chr(0x03)])
461       
462        # set number of copies
463        self.setNumberOfCopies = "".join([chr(0xf8), chr(0x31)]) 
464       
465        infileno = self.infile.fileno()
466        self.pages = { 0 : { "copies" : 1, 
467                             "orientation" : "Default", 
468                             "mediatype" : "Plain", 
469                             "mediasize" : "Default", 
470                             "mediasource" : "Default", 
471                           } 
472                     }     
473        self.minfile = minfile = mmap.mmap(infileno, os.fstat(infileno)[6], prot=mmap.PROT_READ, flags=mmap.MAP_SHARED)
474        tags = self.tags
475        self.pagecount = 0
476        self.escapedStuff = {}
477        self.pos = pos = 0
478        try :
479            while 1 :
480                char = minfile[pos]
481                pos += 1
482                length = tags[ord(char)]
483                if not length :
484                    continue
485                if callable(length) :   
486                    self.pos = pos
487                    length = length()
488                    pos = self.pos
489                pos += length   
490        except IndexError : # EOF ?
491            self.minfile.close() # reached EOF
492           
493        # now handle number of copies for each page (may differ).
494        if self.iscolor :
495            colormode = "Color"
496        else :   
497            colormode = "BW"
498           
499        defaultduplexmode = "Simplex"
500        defaultpapersize = ""
501        defaultpjlcopies = 1   
502        oldpjlcopies = -1
503        oldduplexmode = ""
504        oldpapersize = ""
505        for pnum in range(1, self.pagecount + 1) :
506            # if no number of copies defined, take 1, as explained
507            # in PCLXL documentation.
508            # NB : is number of copies is 0, the page won't be output
509            # but the formula below is still correct : we want
510            # to decrease the total number of pages in this case.
511            page = self.pages.get(pnum, self.pages.get(1, { "copies" : 1, "mediasize" : "Default" }))
512            pjlstuff = self.escapedStuff.get(pnum, self.escapedStuff.get(0, []))
513            if pjlstuff :
514                pjlparser = pjl.PJLParser("".join(pjlstuff))
515                nbdefaultcopies = int(pjlparser.default_variables.get("COPIES", -1))
516                nbcopies = int(pjlparser.environment_variables.get("COPIES", -1))
517                nbdefaultqty = int(pjlparser.default_variables.get("QTY", -1))
518                nbqty = int(pjlparser.environment_variables.get("QTY", -1))
519                if nbdefaultcopies > -1 :
520                    defaultpjlcopies = nbdefaultcopies
521                if nbdefaultqty > -1 :
522                    defaultpjlcopies = nbdefaultqty
523                if nbcopies > -1 :
524                    pjlcopies = nbcopies
525                elif nbqty > -1 :
526                    pjlcopies = nbqty
527                else :
528                    if oldpjlcopies == -1 :   
529                        pjlcopies = defaultpjlcopies
530                    else :   
531                        pjlcopies = oldpjlcopies   
532                defaultdm = pjlparser.default_variables.get("DUPLEX", "")
533                if defaultdm :
534                    if defaultdm.upper() == "ON" :
535                        defaultduplexmode = "Duplex"
536                    else :   
537                        defaultduplexmode = "Simplex"
538                envdm = pjlparser.environment_variables.get("DUPLEX", "")
539                if envdm :
540                    if envdm.upper() == "ON" :
541                        duplexmode = "Duplex"
542                    else :   
543                        duplexmode = "Simplex"
544                else :       
545                    if not oldduplexmode :
546                        duplexmode = defaultduplexmode
547                    else :   
548                        duplexmode = oldduplexmode
549                defaultps = pjlparser.default_variables.get("PAPER", "")
550                if defaultps :
551                    defaultpapersize = defaultps
552                envps = pjlparser.environment_variables.get("PAPER", "")
553                if envps :
554                    papersize = envps
555                else :   
556                    if not oldpapersize :
557                        papersize = defaultpapersize
558                    else :   
559                        papersize = oldpapersize
560            else :       
561                if oldpjlcopies == -1 :
562                    pjlcopies = defaultpjlcopies
563                else :   
564                    pjlcopies = oldpjlcopies
565                if not oldduplexmode :
566                    duplexmode = defaultduplexmode
567                else :   
568                    duplexmode = oldduplexmode
569                if not oldpapersize :   
570                    papersize = defaultpapersize
571                else :   
572                    papersize = oldpapersize
573                duplexmode = oldduplexmode
574                papersize = oldpapersize or page["mediasize"]
575            if page["mediasize"] != "Default" :
576                papersize = page["mediasize"]
577            if not duplexmode :   
578                duplexmode = oldduplexmode or defaultduplexmode
579            oldpjlcopies = pjlcopies   
580            oldduplexmode = duplexmode
581            oldpapersize = papersize
582            copies = pjlcopies * page["copies"]
583            self.pagecount += (copies - 1)
584            self.logdebug("%s*%s*%s*%s*%s*%s*%s" % (copies, 
585                                                 page["mediatype"], 
586                                                 papersize, 
587                                                 page["orientation"], 
588                                                 page["mediasource"], 
589                                                 duplexmode, 
590                                                 colormode))
591        return self.pagecount
592       
593def test() :       
594    """Test function."""
595    if (len(sys.argv) < 2) or ((not sys.stdin.isatty()) and ("-" not in sys.argv[1:])) :
596        sys.argv.append("-")
597    totalsize = 0   
598    for arg in sys.argv[1:] :
599        if arg == "-" :
600            infile = sys.stdin
601            mustclose = 0
602        else :   
603            infile = open(arg, "rb")
604            mustclose = 1
605        try :
606            parser = Parser(infile, debug=1)
607            totalsize += parser.getJobSize()
608        except pdlparser.PDLParserError, msg :   
609            sys.stderr.write("ERROR: %s\n" % msg)
610            sys.stderr.flush()
611        if mustclose :   
612            infile.close()
613    print "%s" % totalsize
614   
615if __name__ == "__main__" :   
616    test()
Note: See TracBrowser for help on using the browser.