source: proto/charsetcompiler/charset_compiler.py @ 915

Last change on this file since 915 was 915, checked in by ksherdy, 8 years ago

Modularize charset compiler and character set definitions.

File size: 21.9 KB
Line 
1# -*- coding: utf-8 -*-
2#
3#  Character Class Compiler
4#
5#  Version 0.7 - August 14, 2009
6#     Adding generation of Python code;  parabix 2 definitions
7#  Copyright (c) 2007, Robert D. Cameron
8#  All rights reserved.
9#
10#  TO DO Notes
11#
12#  1.  Perhaps the character set definition should be extended to
13#      allow other set operations.   For example, the 'Restricted' set
14#      of XML might be defined as ['\x00-\x1f'] - ['\x09', '\x0a', '\x0d']
15#      This would require fewer operations to compute.
16#
17#  2.  The range logic generator should be modified to group
18#      bit variables the same way as the individual character logic
19#      generator (i.e., combining bits 0 and 1, 2 and 3, 4 and 5 and
20#      6 and 7 first.
21#
22#  3.  Extend for 16-bit and full Unicode character values.
23#
24#
25#--------------------------------------------------------------------------
26#
27#  Data types
28#  1. Character Set Definitions
29#  2. Boolean Expressions
30#  3. Code Generator Objects
31#
32import sys
33import re, binascii, string
34
35import charset_def
36CharDef = charset_def.CharDef
37CharSetDef = charset_def.CharSetDef
38
39import charsets
40DefinitionSet = charsets.DefinitionSet
41
42import charset_input_parser
43
44#
45# Symbolic Representations of Boolean Expressions
46#
47
48class BoolExpr:
49   """The BoolExpr class and its subclasses provide a symbolic
50      representation of Boolean expressions.
51   """
52   pass
53
54class Var(BoolExpr):
55    def __init__(self, varname):
56        self.varname = varname
57    def show(self): return 'Var("' + self.varname + '")'
58
59class TrueLiteral(BoolExpr):
60    def __init__(self):
61        self.value = True
62    def show(self): return 'T'
63
64class FalseLiteral(BoolExpr):
65    def __init__(self):
66        self.value = False
67    def show(self): return 'F'
68
69class Not(BoolExpr):
70    def __init__(self, expr):
71        self.operand = expr
72    def show(self): return 'Not(%s)' % (self.operand.show())
73
74class And(BoolExpr):
75    def __init__(self, expr1, expr2):
76        self.operand1 = expr1
77        self.operand2 = expr2
78    def show(self): return 'And(%s, %s)' % (self.operand1.show(), self.operand2.show())
79
80class Or(BoolExpr):
81    def __init__(self, expr1, expr2):
82        self.operand1 = expr1
83        self.operand2 = expr2
84    def show(self): return 'Or(%s, %s)' % (self.operand1.show(), self.operand2.show())
85
86class Xor(BoolExpr):
87    def __init__(self, expr1, expr2):
88        self.operand1 = expr1
89        self.operand2 = expr2
90    def show(self): return 'Xor(%s, %s)' % (self.operand1.show(), self.operand2.show())
91
92class Sel(BoolExpr):
93    def __init__(self, expr1, expr2, expr3):
94        self.sel = expr1
95        self.true_branch = expr2
96        self.false_branch = expr3
97    def show(self): return 'Sel(%s, %s, %s)' % (self.sel.show(), self.true_branch.show(), self.false_branch.show())
98
99
100#
101# Optimizing Constructors for Boolean Expressions
102# - Maintaining Assembler Instruction Form:
103#   - All boolean algebraic rules involving true/false applied.
104#   - Negations restricted:
105#       - no negations within or (DeMorgan's to nand)
106#       - at most one negation within and
107#
108
109def make_not(expr):
110    if isinstance(expr, TrueLiteral):
111        return FalseLiteral()
112    elif isinstance(expr, FalseLiteral):
113        return TrueLiteral()
114    elif isinstance(expr, Not):
115        return expr.operand
116    else: return Not(expr)
117
118def make_and(expr1, expr2):
119    if isinstance(expr1, TrueLiteral):
120        return expr2
121    elif isinstance(expr2, TrueLiteral):
122        return expr1
123    elif isinstance(expr1, FalseLiteral):
124        return FalseLiteral()
125    elif isinstance(expr2, FalseLiteral):
126        return FalseLiteral()
127    elif equal_exprs(expr1, expr2): return expr1
128    elif isinstance(expr1, Not):
129        if isinstance(expr2, Not):
130            return make_not(make_or(expr1.operand, expr2.operand))
131        elif equal_exprs(expr1.operand, expr2): return FalseLiteral()
132        else: return And(expr1, expr2)
133    elif isinstance(expr2, Not):
134        if equal_exprs(expr1, expr2.operand): return FalseLiteral()
135        else: return And(expr1, expr2)
136    else: return And(expr1, expr2)
137
138def make_or(expr1, expr2):
139    if isinstance(expr1, FalseLiteral):
140        return expr2
141    elif isinstance(expr2, FalseLiteral):
142        return expr1
143    elif isinstance(expr1, TrueLiteral):
144        return TrueLiteral()
145    elif isinstance(expr2, TrueLiteral):
146        return TrueLiteral()
147    elif isinstance(expr1, Not):
148        return make_not(make_and(expr1.operand, make_not(expr2)))
149    elif isinstance(expr2, Not):
150        return make_not(make_and(make_not(expr1), expr2.operand))
151    elif equal_exprs(expr1, expr2): return expr1
152    elif isinstance(expr1, And) and isinstance(expr2, And):
153        # These optimizations factor out common components that
154        # can occur when sets are formed by union (e.g., union of
155        # [a-z] and [A-Z].
156        if equal_exprs(expr1.operand1, expr2.operand1):
157            return make_and(expr1.operand1, make_or(expr1.operand2, expr2.operand2))
158        elif equal_exprs(expr1.operand2, expr2.operand2):
159            return make_and(expr1.operand2, make_or(expr1.operand1, expr2.operand1))
160        elif equal_exprs(expr1.operand1, expr2.operand2):
161            return make_and(expr1.operand1, make_or(expr1.operand2, expr2.operand1))
162        elif equal_exprs(expr1.operand2, expr2.operand1):
163            return make_and(expr1.operand2, make_or(expr1.operand1, expr2.operand2))
164        else: return Or(expr1, expr2)
165    else: return Or(expr1, expr2)
166
167def make_sel(if_expr, T_expr, F_expr):
168    if isinstance(if_expr, TrueLiteral):
169        return T_expr
170    elif isinstance(if_expr, FalseLiteral):
171        return F_expr
172    elif isinstance(T_expr, TrueLiteral):
173        # if x then T else y = x or y
174        return make_or(if_expr, F_expr) 
175    elif isinstance(T_expr, FalseLiteral):
176        # if x then F else y = not(x) and y
177        return make_and(make_not(if_expr), F_expr) 
178    elif isinstance(F_expr, FalseLiteral):
179        # if x then y else F = x and y
180        return make_and(if_expr, T_expr) 
181    elif isinstance(F_expr, TrueLiteral):
182        # if x then y else T = not(x) or y
183        return make_or(make_not(if_expr), T_expr)
184    elif equal_exprs(T_expr, F_expr):
185        return T_expr
186    else: return Sel(if_expr, T_expr, F_expr)
187
188
189def make_xor(expr1, expr2):
190    if isinstance(expr1, FalseLiteral):
191        return expr2
192    elif isinstance(expr1, FalseLiteral):
193        return expr1
194    elif isinstance(expr1, TrueLiteral):
195        return make_not(expr2)
196    elif isinstance(expr2, TrueLiteral):
197        return make_not(expr1)
198    elif isinstance(expr1, Not) and isinstance(expr2, Not):
199        return make_xor(expr1.operand, expr2.operand)
200    else: return Xor(expr1, expr2)
201
202def equal_exprs(e1, e2):
203    if isinstance(e1, FalseLiteral): return isinstance(e2, FalseLiteral)
204    elif isinstance(e1, TrueLiteral): return isinstance(e2, TrueLiteral)
205    elif isinstance(e1, Var) and isinstance(e2, Var):
206        return e1.varname == e2.varname
207    elif isinstance(e1, Not) and isinstance(e2, Not):
208        return equal_exprs(e1.operand, e2.operand)
209    elif isinstance(e1, And) and isinstance(e2, And):
210        if equal_exprs(e1.operand1, e2.operand1):
211            return equal_exprs(e1.operand2, e2.operand2)
212        elif equal_exprs(e1.operand1, e2.operand2):
213            return equal_exprs(e1.operand2, e2.operand1)
214        else: return False
215    elif isinstance(e1, Or) and isinstance(e2, Or):
216        if equal_exprs(e1.operand1, e2.operand1):
217            return equal_exprs(e1.operand2, e2.operand2)
218        elif equal_exprs(e1.operand1, e2.operand2):
219            return equal_exprs(e1.operand2, e2.operand1)
220        else: return False
221    elif isinstance(e1, Xor) and isinstance(e2, Xor):
222        if equal_exprs(e1.operand1, e2.operand1):
223            return equal_exprs(e1.operand2, e2.operand2)
224        elif equal_exprs(e1.operand1, e2.operand2):
225            return equal_exprs(e1.operand2, e2.operand1)
226        else: return False
227    elif isinstance(e1, Sel) and isinstance(e2, Sel):
228        if equal_exprs(e1.sel, e2.sel):
229             if equal_exprs(e1.true_branch, e2.true_branch):
230                 return equal_exprs(e1.false_branch, e2.false_branch)
231             else: return False
232        else: return False
233
234
235#
236#
237#  Boolean Expressions for Character Class Definitions
238#
239
240def bit_var(n):
241    return 'bit[%i]' % n
242def make_bitv(n):
243    return Var(bit_var(n))
244
245# Deprecated
246def make_2bit_test(var1, var2, bit_pattern):
247  if bit_pattern == 0: 
248    return make_not(make_or(var1, var2))
249  elif bit_pattern == 1:
250    return make_and(make_not(var1), var2)
251  elif bit_pattern == 2:
252    return make_and(var1, make_not(var2))
253  else: return make_and(var1, var2)
254
255# Deprecated
256def make_8bit_test(bit_pattern):
257  return make_and(make_and(make_2bit_test(make_bitv(0), make_bitv(1), (bit_pattern >> 6) & 3),
258                           make_2bit_test(make_bitv(2), make_bitv(3), (bit_pattern >> 4) & 3)),
259                  make_and(make_2bit_test(make_bitv(4), make_bitv(5), (bit_pattern >> 2) & 3),
260                           make_2bit_test(make_bitv(6), make_bitv(7), (bit_pattern) & 3)))
261
262
263def make_bit_test(pattern, bit_count):
264    bit_terms = []
265    test_bit = 2**(bit_count - 1)
266    for i in range(0, bit_count):
267        if (pattern & test_bit) == 0:
268            bit_terms.append(make_not(make_bitv(i)))
269        else: bit_terms.append(make_bitv(i))
270        test_bit >>= 1
271    while len(bit_terms) > 1:
272        new_terms = []
273        for i in range(0, len(bit_terms)/ 2):
274            new_terms.append(make_and(bit_terms[2*i], bit_terms[2*i+1]))
275        if len(bit_terms) % 2 == 1:
276            new_terms.append(bit_terms[-1])
277        bit_terms = new_terms
278    return bit_terms[0]
279
280
281def char_test_expr(ch):
282    return make_bit_test(ord(ch), 8)
283    #return make_8bit_test(ord(ch))
284
285def GE_Range(N, n):
286    if N == 0: return TrueLiteral()
287    elif N % 2 == 0 and (n >> (N - 2)) == 0:
288        return make_or(make_or(make_bitv(8-N), make_bitv(9-N)),
289                        GE_Range(N - 2, n))
290    elif N % 2 == 0 and (n >> (N - 2)) == 3:   # >= 11xxxx
291        return make_and(make_and(make_bitv(8-N), make_bitv(9-N)),
292                        GE_Range(N - 2, n - (3 << (N-2))))
293    elif N >= 1:
294        hi_bit = n & (1 << (N-1))
295        lo_bits = n - hi_bit
296        lo_range = GE_Range(N-1, lo_bits)
297        if hi_bit == 0:
298            # If the hi_bit of n is not set, then whenever the corresponding bit
299            # is set in the target, the target will certainly be >=.  Otherwise,
300            # the value of GE_range(N-1, lo_bits) is required.
301            return make_or(make_bitv(8-N), lo_range)
302        else: 
303            # If the hi_bit of n is set, then the corresponding bit must be set
304            # in the target for >= and GE_range(N-1, lo_bits) must also be true.
305            return make_and(make_bitv(8-N), lo_range)
306
307def LE_Range(N, n):
308    # If an N-bit pattern is all ones, then it is always
309    # true that any n-bit value is LE this pattern.
310    # Handling this as a special case avoids an overflow
311    # issue with n+1 requiring more than N bits.
312    if n+1 == 2 ** N:
313        return TrueLiteral()
314    else:
315        return make_not(GE_Range(N, n+1))
316
317BadRange = Exception()
318
319
320def Make_Range(n1, n2):  # require n2 >= n1
321    diff_bits = n1 ^ n2
322    diff_count = 0
323    while diff_bits > 0:
324        diff_count += 1
325        diff_bits >>= 1
326    if n2 < n1 or diff_count > 8: raise BadRange()
327    common = make_bit_test(n1 >> diff_count, 8 - diff_count)
328    if diff_count == 0: return common
329    mask = 2**(diff_count-1) - 1
330    lo_test = GE_Range(diff_count-1, n1 & mask)
331    hi_test = LE_Range(diff_count-1, n2 & mask)
332    return make_and(common, make_sel(make_bitv(8-diff_count), hi_test, lo_test))
333
334# Deprecated   
335def Inclusive_Range(N, n1, n2):  # require n2 >= n1
336    if N == 0: return TrueLiteral()
337    elif n1 >= 2**(N-1):
338        return make_and(make_bitv(8-N), Inclusive_Range(N-1, n1 - 2**(N-1), n2 - 2**(N-1)))
339    elif n2 < 2**(N-1):
340        return make_and(make_not(make_bitv(8-N)), Inclusive_Range(N-1, n1, n2))
341    else:
342        n2_lo = n2 - 2**(N-1)
343        lo_test = GE_Range(N-1, n1)
344        hi_test = LE_Range(N-1, n2_lo)
345        # special optimization?
346        # if n2_lo + 1 == n1: return make_xor(make_bit_test(8-N), lo_test)
347        return make_sel(make_bitv(8-N), lo_test, hi_test)
348
349BadCharSetItem = Exception()
350
351def char_or_range_expr(charset_item):
352    if len(charset_item) == 1:
353        return char_test_expr(charset_item[0])
354    elif len(charset_item) == 3:
355        if charset_item[1] == '-' and ord(charset_item[0]) <= ord(charset_item[2]):
356             return Make_Range(ord(charset_item[0]), ord(charset_item[2]))
357             #  return Inclusive_Range(8, ord(charset_item[0]), ord(charset_item[2]))
358    print charset_item
359    raise BadCharSetItem
360
361def charset_expr(chardef):
362    e1 = char_or_range_expr(chardef.items[0])
363    for i in range(1, len(chardef.items)):
364        e1 = make_or(e1, char_or_range_expr(chardef.items[i]))
365    if chardef.complemented: return make_not(e1)
366    else: return e1
367
368#
369#
370#  Code Generation
371#
372class CodeGenObject:
373    def __init__(self, predeclared, typedecl='BitBlock '):
374        self.gensym_template = 'temp%i'
375        self.gensym_counter = 0
376        self.generated_code = []
377        self.common_expression_map = {}
378        for sym in predeclared: self.common_expression_map[sym] = sym
379        self.typedecl = typedecl
380    def add_assignment(self, varname, expr):
381        self.common_expression_map[expr] = varname
382        self.generated_code.append('%s%s = %s;\n' % (self.typedecl, varname, expr))
383    def expr_string_to_variable(self, expr_string):
384        if self.common_expression_map.has_key(expr_string): 
385            return self.common_expression_map[expr_string]
386        else:
387            self.gensym_counter += 1
388            sym = self.gensym_template % self.gensym_counter
389            self.add_assignment(sym, expr_string)
390            return sym
391    def showcode(self):
392        s = ''
393        for stmt in self.generated_code: s += stmt
394        return s
395
396def expr2simd(genobj, expr):
397    """Translate a Boolean expression into three-address Altivec code
398       using code generator object genobj.
399    """
400    if isinstance(expr, TrueLiteral): return 'simd_const_1(1)'
401    elif isinstance(expr, FalseLiteral): return 'simd_const_1(0)'
402    elif isinstance(expr, Var): return expr.varname
403    elif isinstance(expr, Not):
404       e = genobj.expr_string_to_variable(expr2simd(genobj, expr.operand))
405       return 'simd_andc(simd_const_1(1), %s)' % (e)
406    elif isinstance(expr, Or):
407       e1 = genobj.expr_string_to_variable(expr2simd(genobj, expr.operand1))
408       e2 = genobj.expr_string_to_variable(expr2simd(genobj, expr.operand2))
409       return 'simd_or(%s, %s)' % (e1, e2)
410    elif isinstance(expr, Xor):
411       e1 = genobj.expr_string_to_variable(expr2simd(genobj, expr.operand1))
412       e2 = genobj.expr_string_to_variable(expr2simd(genobj, expr.operand2))
413       return 'simd_xor(%s, %s)' % (e1, e2)
414    elif isinstance(expr, And):
415       if isinstance(expr.operand1, Not):
416           e1 = genobj.expr_string_to_variable(expr2simd(genobj, expr.operand1.operand))
417           e2 = genobj.expr_string_to_variable(expr2simd(genobj, expr.operand2))
418           return 'simd_andc(%s, %s)' % (e2, e1)
419       elif isinstance(expr.operand2, Not):
420           e1 = genobj.expr_string_to_variable(expr2simd(genobj, expr.operand1))
421           e2 = genobj.expr_string_to_variable(expr2simd(genobj, expr.operand2.operand))
422           return 'simd_andc(%s, %s)' % (e1, e2)
423       else:
424           e1 = genobj.expr_string_to_variable(expr2simd(genobj, expr.operand1))
425           e2 = genobj.expr_string_to_variable(expr2simd(genobj, expr.operand2))
426           return 'simd_and(%s, %s)' % (e1, e2)
427    elif isinstance(expr, Sel):
428       sel = genobj.expr_string_to_variable(expr2simd(genobj, expr.sel))
429       e1 = genobj.expr_string_to_variable(expr2simd(genobj, expr.true_branch))
430       e2 = genobj.expr_string_to_variable(expr2simd(genobj, expr.false_branch))
431       return 'simd_if(%s, %s, %s)' %(sel, e1, e2)
432
433def chardef2simd(genobj, chardef):
434    genobj.add_assignment(chardef.name, expr2simd(genobj, charset_expr(chardef)))
435
436def chardeflist2simd(chardeflist):
437    cgo = CodeGenObject([bit_var(i) for i in range(0,8)])
438    for d in chardeflist:
439        chardef2simd(cgo, d)
440    return cgo.showcode()
441
442def expr2py(genobj, expr):
443    """Translate a Boolean expression into three-address python code
444       using code generator object genobj.
445    """
446    if isinstance(expr, TrueLiteral): return '-1'
447    elif isinstance(expr, FalseLiteral): return '0'
448    elif isinstance(expr, Var): return expr.varname
449    elif isinstance(expr, Not):
450       e = genobj.expr_string_to_variable(expr2py(genobj, expr.operand))
451       return '(~%s)' % (e)
452    elif isinstance(expr, Or):
453       e1 = genobj.expr_string_to_variable(expr2py(genobj, expr.operand1))
454       e2 = genobj.expr_string_to_variable(expr2py(genobj, expr.operand2))
455       return '(%s | %s)' % (e1, e2)
456    elif isinstance(expr, Xor):
457       e1 = genobj.expr_string_to_variable(expr2py(genobj, expr.operand1))
458       e2 = genobj.expr_string_to_variable(expr2py(genobj, expr.operand2))
459       return '(%s ^ %s)' % (e1, e2)
460    elif isinstance(expr, And):
461       if isinstance(expr.operand1, Not):
462           e1 = genobj.expr_string_to_variable(expr2py(genobj, expr.operand1.operand))
463           e2 = genobj.expr_string_to_variable(expr2py(genobj, expr.operand2))
464           return '(%s &~ %s)' % (e2, e1)
465       elif isinstance(expr.operand2, Not):
466           e1 = genobj.expr_string_to_variable(expr2py(genobj, expr.operand1))
467           e2 = genobj.expr_string_to_variable(expr2py(genobj, expr.operand2.operand))
468           return '(%s &~ %s)' % (e1, e2)
469       else:
470           e1 = genobj.expr_string_to_variable(expr2py(genobj, expr.operand1))
471           e2 = genobj.expr_string_to_variable(expr2py(genobj, expr.operand2))
472           return '(%s & %s)' % (e1, e2)
473    elif isinstance(expr, Sel):
474       sel = genobj.expr_string_to_variable(expr2py(genobj, expr.sel))
475       e1 = genobj.expr_string_to_variable(expr2py(genobj, expr.true_branch))
476       e2 = genobj.expr_string_to_variable(expr2py(genobj, expr.false_branch))
477       return '((%s & %s)|(~(%s) & %s))' %(sel, e1, sel, e2)
478
479def chardef2py(genobj, chardef):
480    genobj.add_assignment(chardef.name, expr2py(genobj, charset_expr(chardef)))
481
482
483def py_chardefmap(chardeflist):
484    defs = ["'%s' : %s" % (d.name,d.name) for d in chardeflist]
485    return  '{%s}' % string.join(defs, ',\n\t')
486
487def chardeflist2py(chardeflist):
488    cgo = CodeGenObject([bit_var(i) for i in range(0,8)],'')
489    for d in chardeflist:
490        chardef2py(cgo, d) 
491    return cgo.showcode()# + "  return "+ py_chardefmap(chardeflist) + "\n"
492
493
494def code_gen_for_transcode(transcode_tbl_h, transcode_tbl_l):
495  cgo = CodeGenObject([bit_var(i) for i in range(0,8)])
496  xor_tbl_l = [transcode_tbl_l[code] ^ code for code in range(256)]
497  code_gen_for_transcode_8bit(cgo, "x16h", transcode_tbl_h)
498  code_gen_for_transcode_8bit(cgo, "x16l", xor_tbl_l)
499  return cgo.showcode()
500
501def code_gen_for_transcode_8bit(cgo, pfx, tbl):
502  for bit in range(8):
503    bit_mask = 256 >> bit
504    in_run = False
505    first_expr_found = False
506    bit_expr = FalseLiteral()
507    for code in range(256):
508      if tbl[code] & bit_mask != 0:
509        if not in_run:
510          in_run = True
511          run_start = code
512      else:
513        if in_run:
514          if run_start == code-1:
515            e1 = char_test_expr(chr(code - 1))
516          else:
517            e1 = Make_Range(run_start, code - 1)
518          if first_expr_found:
519            bit_expr = make_or(bit_expr, e1)
520          else:
521            first_expr_found = True
522            bit_expr = e1
523    if first_expr_found:
524      cgo.add_assignment("%s[%i]" % (pfx, bit), expr2simd(cgo, bit_expr))
525
526# Work with tables from
527# /home/cameron/glibc-2.3.5/localedata/charmaps/
528
529charmap_line_RE = re.compile("<U([0-9A-F][0-9A-F])([0-9A-Z][0-9A-Z])>\s+/x([0-9a-f][0-9a-f])\s")
530
531def read_char_map(file):
532  f = open(file)
533  lines = f.readlines()
534  matches = [charmap_line_RE.match(l) for l in lines]
535  u16hi = [ord(binascii.unhexlify(m.group(1))) for m in matches if m]
536  u16lo = [ord(binascii.unhexlify(m.group(2))) for m in matches if m]
537  codes = [ord(binascii.unhexlify(m.group(3))) for m in matches if m]
538  codes_OK = [c for c in range(256) if codes[c] == c]
539  if len(codes_OK) != 256:
540    print("Code map failure reading %s" % file)
541  return (u16hi, u16lo)
542
543import codecs
544def ascii2ebcdic_chardeflist(defs):
545        encoder = codecs.getencoder('cp037')
546        return [xlate_chardef(d, encoder) for d in defs]
547
548def xlate_char_or_range(charset_item, encoder):
549    if len(charset_item) == 1:
550        return encoder(charset_item[0])
551    elif len(charset_item) == 3:
552        if charset_item[1] == '-' and ord(charset_item[0]) <= ord(charset_item[2]):
553             return Make_Range(ord(charset_item[0]), ord(charset_item[2]))
554             #  return Inclusive_Range(8, ord(charset_item[0]), ord(charset_item[2]))
555    print charset_item
556    raise BadCharSetItem
557       
558def xlate_chardef(chardef, encoder):
559  if isinstance(chardef, CharDef):
560    return CharDef(chardef.name, encoder(chardef.items[0])[0], chardef.complemented)
561  else:
562    cdefs = []
563    for item in chardef.items:
564        if len(item) == 1: cdefs.append(encoder(item)[0])
565        elif len(item) == 3:
566          for v in range(ord(item[0]), ord(item[-1])+1):
567            cdefs.append(encoder(chr(v))[0])
568        else: raise BadCharSetItem
569    return CharSetDef(chardef.name, cdefs, chardef.complemented)
570
571def main():
572    if len(sys.argv) == 2:
573        # if the specified argument is not in the DefinitionSet, then assume that it's a filename
574        if sys.argv[1] not in DefinitionSet:
575            #define the characters in the list
576            defs = charset_input_parser.input_chardef(sys.argv[1])
577                                               
578            #print chardeflist2simd(defs)
579            print chardeflist2py(defs)
580        else:
581            #print chardeflist2simd(DefinitionSet[sys.argv[1]])
582            print chardeflist2py(DefinitionSet[sys.argv[1]])
583    elif len(sys.argv) == 3 and sys.argv[2] == 'EBCDIC':
584        defs = ascii2ebcdic_chardeflist(DefinitionSet[sys.argv[1]])
585        print chardeflist2simd(defs)
586    else:
587        print "Usage python charset_compiler.py <name> [EBCDIC]"
588        for k in DefinitionSet.keys(): print k
589
590if __name__ == "__main__": main()
Note: See TracBrowser for help on using the repository browser.