Context Navigation

pclxl.py @ 357

Revision 357, 28.7 kB (checked in by jerome, 18 years ago)
Added missing docstrings, thanks to pylint.
Property svn:eol-style set to `native` Property svn:keywords set to `Auth Date Id Rev`

Rev	Line
[193]	1	#! /usr/bin/env python
	2	# -- coding: ISO-8859-15 --
[191]	3	#
	4	# pkpgcounter : a generic Page Description Language parser
	5	#
[302]	6	# (c) 2003, 2004, 2005, 2006 Jerome Alet <alet@librelogiciel.com>
[191]	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
[211]	19	# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
[191]	20	#
	21	# $Id$
	22	#
[193]	23
[357]	24	"""This modules implements a page counter for PCLXL (aka PCL6) documents."""
	25
[193]	26	import sys
[202]	27	import os
[193]	28	import mmap
	29	from struct import unpack
	30
[235]	31	import pdlparser
[252]	32	import pjl
[193]	33
[220]	34	class Parser(pdlparser.PDLParser) :
[193]	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",
[246]	50	12 : "B5",
[193]	51	14 : "JPostcard",
	52	15 : "JDoublePostcard",
	53	16 : "A5",
	54	17 : "A6",
	55	18 : "JB6",
[246]	56	19 : "JIS8K",
	57	20 : "JIS16K",
	58	21 : "JISExec",
	59	96 : "Default",
[193]	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",
[246]	78	4 : "Default",
[193]	79	}
	80
[220]	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))) :
[252]	87	self.logdebug("DEBUG: Input file is in the PCLXL (aka PCL6) format.")
[220]	88	return 1
	89	else :
	90	return 0
	91
[318]	92	def beginPage(self, nextpos) :
[193]	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"
[299]	101	duplexmode = None
[193]	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
[318]	106	pos = nextpos - 2
[193]	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))
[302]	114	pos -= 4
[193]	115	elif val == 0x25 :
[302]	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
[310]	132	# self.logdebug("Media size : %s" % mediasizelabel)
[193]	133	elif val == 0x28 :
	134	orientation = ord(minfile[pos - 2])
[252]	135	orientationlabel = self.orientations.get(orientation, str(orientation))
[302]	136	pos -= 4
[193]	137	elif val == 0x27 :
	138	savepos = pos
[302]	139	pos -= 1
[293]	140	startpos = size = None
[193]	141	while pos > 0 : # safety check : don't go back to far !
	142	val = ord(minfile[pos])
	143	pos -= 1
	144	if val == 0xc8 :
[293]	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
[316]	151	size = unpack(self.unpackShort, self.minfile[pos+3:startpos])[0]
[293]	152	elif length == 4 :
	153	startpos = pos + 7
[316]	154	size = unpack(self.unpackLong, self.minfile[pos+3:startpos])[0]
[293]	155	else :
[301]	156	raise pdlparser.PDLParserError, "Error on size at %s : %s" % (pos+2, length)
[193]	157	break
[293]	158	mediatypelabel = minfile[startpos:startpos+size]
[310]	159	# self.logdebug("Media type : %s" % mediatypelabel)
[299]	160	elif val == 0x34 :
	161	duplexmode = "Simplex"
[302]	162	pos -= 2
[299]	163	elif val in (0x35, 0x36) :
	164	duplexmode = "Duplex"
[302]	165	pos -= 2
[193]	166	# else : TODO : CUSTOM MEDIA SIZE AND UNIT !
	167	else :
[302]	168	pos -= 1 # ignored
[193]	169	self.pages[self.pagecount] = { "copies" : 1,
	170	"orientation" : orientationlabel,
	171	"mediatype" : mediatypelabel,
	172	"mediasize" : mediasizelabel,
	173	"mediasource" : mediasourcelabel,
[299]	174	"duplex" : duplexmode,
[193]	175	}
	176	return 0
	177
[318]	178	def endPage(self, nextpos) :
[193]	179	"""Indicates the end of a page."""
[318]	180	pos3 = nextpos - 3
[193]	181	minfile = self.minfile
[318]	182	if minfile[pos3:nextpos-1] == self.setNumberOfCopies :
[193]	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
[258]	187	try :
[317]	188	self.pages[self.pagecount]["copies"] = unpack(self.unpackShort, minfile[pos3-2:pos3])[0]
[258]	189	except KeyError :
	190	self.logdebug("It looks like this PCLXL file is corrupted.")
[193]	191	return 0
	192
[318]	193	def setColorSpace(self, nextpos) :
[205]	194	"""Changes the color space."""
[318]	195	if self.minfile[nextpos-4:nextpos-1] == self.RGBColorSpace : # TODO : doesn't seem to handle all cases !
[206]	196	self.iscolor = 1
[205]	197	return 0
	198
[318]	199	def array_Generic(self, nextpos, size) :
[317]	200	"""Handles all arrays."""
[318]	201	pos = nextpos
[317]	202	datatype = ord(self.minfile[pos])
[193]	203	pos += 1
[317]	204	length = self.tags[datatype]
[193]	205	if callable(length) :
[316]	206	length = length(pos)
[318]	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
[317]	211
[318]	212	def array_8(self, nextpos) :
[317]	213	"""Handles byte arrays."""
[318]	214	return self.array_Generic(nextpos, 1)
[193]	215
[318]	216	def array_16(self, nextpos) :
[193]	217	"""Handles byte arrays."""
[318]	218	return self.array_Generic(nextpos, 2)
[193]	219
[318]	220	def array_32(self, nextpos) :
[193]	221	"""Handles byte arrays."""
[318]	222	return self.array_Generic(nextpos, 4)
[193]	223
[318]	224	def embeddedDataSmall(self, nextpos) :
[193]	225	"""Handle small amounts of data."""
[318]	226	return 1 + ord(self.minfile[nextpos])
[193]	227
[318]	228	def embeddedData(self, nextpos) :
[193]	229	"""Handle normal amounts of data."""
[318]	230	return 4 + unpack(self.unpackLong, self.minfile[nextpos:nextpos+4])[0]
[193]	231
[318]	232	def littleEndian(self, nextpos) :
[193]	233	"""Toggles to little endianness."""
[318]	234	self.unpackType = { 1 : "B", 2 : "<H", 4 : "<I" }
	235	self.unpackShort = self.unpackType[2]
	236	self.unpackLong = self.unpackType[4]
[193]	237	return 0
	238
[318]	239	def bigEndian(self, nextpos) :
[193]	240	"""Toggles to big endianness."""
[318]	241	self.unpackType = { 1 : "B", 2 : ">H", 4 : ">I" }
	242	self.unpackShort = self.unpackType[2]
	243	self.unpackLong = self.unpackType[4]
[193]	244	return 0
	245
[318]	246	def reservedForFutureUse(self, nextpos) :
[245]	247	"""Outputs something when a reserved byte is encountered."""
[318]	248	self.logdebug("Byte at %x is out of the PCLXL Protocol Class 2.0 Specification" % nextpos)
[245]	249	return 0
	250
[318]	251	def x46_class3(self, nextpos) :
[310]	252	"""Undocumented tag 0x46 in class 3.0 streams."""
[318]	253	pos = nextpos - 3
[309]	254	minfile = self.minfile
[318]	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 :
[320]	264	pos -= offset
	265	length = self.tags[ord(self.minfile[pos])]
[318]	266	if callable(length) :
[320]	267	length = length(pos+1)
	268	if funcid == 0x92 : # we want to skip these blocks
[318]	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)
[309]	273	val = ord(minfile[pos])
	274	return 0
	275
[318]	276	def escape(self, nextpos) :
[245]	277	"""Handles the ESC code."""
[318]	278	pos = endpos = nextpos
[314]	279	minfile = self.minfile
	280	if minfile[pos : pos+8] == r"%-12345X" :
[245]	281	endpos = pos + 9
[252]	282	endmark = chr(0x0c) + chr(0x00) + chr(0x1b)
[246]	283	asciilimit = chr(0x80)
[269]	284	quotes = 0
[314]	285	while (minfile[endpos] not in endmark) and \
	286	((minfile[endpos] < asciilimit) or (quotes % 2)) :
	287	if minfile[endpos] == '"' :
[269]	288	quotes += 1
[245]	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, [])
[314]	294	stuff.append(minfile[pos : endpos])
	295	self.logdebug("Escaped datas : [%s]" % repr(minfile[pos : endpos]))
[245]	296	return endpos - pos
	297
[318]	298	def skipKyoceraPrescribe(self, nextpos) :
[312]	299	"""Skips Kyocera Prescribe commands."""
[318]	300	pos = nextpos - 1
[312]	301	minfile = self.minfile
	302	if minfile[pos:pos+3] == "!R!" :
[318]	303	while (pos - nextpos) < 1024 : # This is a realistic upper bound, to avoid infinite loops
[312]	304	if (minfile[pos] == ";") and (minfile[pos-4:pos] == "EXIT") :
[314]	305	pos += 1
	306	prescribe = self.prescribeStuff.setdefault(self.pagecount, [])
[318]	307	prescribe.append(minfile[nextpos-1:pos])
	308	self.logdebug("Prescribe commands : [%s]" % repr(minfile[nextpos-1:pos]))
	309	return (pos - nextpos)
[312]	310	pos += 1
	311	else :
	312	return 0
	313
[193]	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
[246]	324
	325	Protocol Class 2.1 Supplement
	326	xl_ref21.pdf
	327
	328	Protocol Class 3.0 Supplement
	329	xl_refsup30r089.pdf
[193]	330	"""
[206]	331	self.iscolor = None
[193]	332	found = 0
	333	while not found :
	334	line = self.infile.readline()
	335	if not line :
	336	break
[293]	337	pos = line.find(" HP-PCL XL;")
	338	if pos != -1 :
[193]	339	found = 1
[293]	340	endian = ord(line[pos - 1])
[193]	341	if endian == 0x29 :
[316]	342	self.littleEndian(0)
[193]	343	elif endian == 0x28 :
[316]	344	self.bigEndian(0)
[193]	345	# elif endian == 0x27 : # TODO : This is the ESC code : parse it for PJL statements !
	346	#
	347	else :
[200]	348	raise pdlparser.PDLParserError, "Unknown endianness marker 0x%02x at start !" % endian
[193]	349	if not found :
[200]	350	raise pdlparser.PDLParserError, "This file doesn't seem to be PCLXL (aka PCL6)"
[193]	351
[246]	352	# Initialize Media Sources
	353	for i in range(8, 256) :
	354	self.mediasources[i] = "ExternalTray%03i" % (i - 7)
	355
[193]	356	# Initialize table of tags
	357	self.tags = [ 0 ] * 256
	358
[245]	359	self.tags[0x1b] = self.escape # The escape code
	360
[352]	361	self.tags[0x21] = self.skipKyoceraPrescribe # 0x21 is not normally used
[312]	362
[193]	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
[245]	370	self.tags[0x45] = self.reservedForFutureUse # reserved
[313]	371
[309]	372	self.tags[0x46] = self.x46_class3
[193]	373
[245]	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
[313]	383
[245]	384	self.tags[0x59] = self.reservedForFutureUse # reserved
	385	self.tags[0x5a] = self.reservedForFutureUse # reserved
	386
[206]	387	self.tags[0x6a] = self.setColorSpace # to detect color/b&w mode
[205]	388
[245]	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
[313]	393
[245]	394	self.tags[0x8b] = self.reservedForFutureUse # reserved
[313]	395
[245]	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
[316]	425	# self.tags[0xbf] = self.passThrough # PassThrough mode should already be taken care of automatically
[298]	426
[193]	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
[245]	434	self.tags[0xc6] = self.reservedForFutureUse # reserved
	435	self.tags[0xc7] = self.reservedForFutureUse # reserved
	436
[193]	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
[245]	444	self.tags[0xce] = self.reservedForFutureUse # reserved
	445	self.tags[0xcf] = self.reservedForFutureUse # reserved
	446
[193]	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
[245]	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
[193]	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
[245]	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
[193]	480
[245]	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
[193]	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
[245]	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
[193]	500
[206]	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
[318]	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
[193]	521	infileno = self.infile.fileno()
[258]	522	self.pages = { 0 : { "copies" : 1,
	523	"orientation" : "Default",
	524	"mediatype" : "Plain",
	525	"mediasize" : "Default",
	526	"mediasource" : "Default",
[299]	527	"duplex" : None,
[258]	528	}
	529	}
[193]	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
[314]	533	self.escapedStuff = {} # For escaped datas, mostly PJL commands
	534	self.prescribeStuff = {} # For Kyocera Prescribe commands
[316]	535	pos = oldpos = 0
[193]	536	try :
[307]	537	try :
	538	while 1 :
	539	try :
[316]	540	tag = ord(minfile[pos])
[307]	541	except OverflowError :
	542	pos = oldpos + 1
	543	pos += 1
[316]	544	length = tags[tag]
[307]	545	if length :
	546	if callable(length) :
[316]	547	length = length(pos)
[307]	548	oldpos = pos
	549	pos += length
	550	except IndexError : # EOF ?
	551	pass
	552	finally :
	553	self.minfile.close()
[193]	554
	555	# now handle number of copies for each page (may differ).
[206]	556	if self.iscolor :
	557	colormode = "Color"
	558	else :
[253]	559	colormode = "BW"
[252]	560
[253]	561	defaultduplexmode = "Simplex"
	562	defaultpapersize = ""
[252]	563	defaultpjlcopies = 1
	564	oldpjlcopies = -1
[253]	565	oldduplexmode = ""
	566	oldpapersize = ""
[193]	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.
[299]	573	page = self.pages.get(pnum, self.pages.get(1, { "copies" : 1, "mediasize" : "Default", "duplex" : None }))
[266]	574	pjlstuff = self.escapedStuff.get(pnum, self.escapedStuff.get(0, []))
[252]	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 :
[253]	586	pjlcopies = nbcopies
[252]	587	elif nbqty > -1 :
[253]	588	pjlcopies = nbqty
[252]	589	else :
	590	if oldpjlcopies == -1 :
[253]	591	pjlcopies = defaultpjlcopies
	592	else :
	593	pjlcopies = oldpjlcopies
[299]	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
[253]	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
[252]	625	else :
	626	if oldpjlcopies == -1 :
	627	pjlcopies = defaultpjlcopies
	628	else :
	629	pjlcopies = oldpjlcopies
[253]	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
[252]	647	copies = pjlcopies * page["copies"]
[193]	648	self.pagecount += (copies - 1)
[253]	649	self.logdebug("%s%s%s%s%s%s%s" % (copies,
[252]	650	page["mediatype"],
[253]	651	papersize,
[252]	652	page["orientation"],
	653	page["mediasource"],
[253]	654	duplexmode,
[252]	655	colormode))
[193]	656	return self.pagecount
	657
	658	def test() :
	659	"""Test function."""
[196]	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 :
[220]	671	parser = Parser(infile, debug=1)
[196]	672	totalsize += parser.getJobSize()
[200]	673	except pdlparser.PDLParserError, msg :
[196]	674	sys.stderr.write("ERROR: %s\n" % msg)
	675	sys.stderr.flush()
	676	if mustclose :
	677	infile.close()
	678	print "%s" % totalsize
[193]	679
	680	if __name__ == "__main__" :
	681	test()

Note: See TracBrowser for help on using the browser.

Context Navigation

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

Download in other formats: