source: proto/charsetcompiler/charset_compiler.py @ 1200

Last change on this file since 1200 was 1200, checked in by cameron, 8 years ago

Tom's fix for make_xor; doc for equal_exprs

File size: 22.2 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(expr2, 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
202# Return True if e1 and e2 can be proven equivalent according
203# to some rules, False otherwise.   Note that False may
204# be returned in some cases when the exprs are equivalent.
205def equal_exprs(e1, e2):
206    if isinstance(e1, FalseLiteral): return isinstance(e2, FalseLiteral)
207    elif isinstance(e1, TrueLiteral): return isinstance(e2, TrueLiteral)
208    elif isinstance(e1, Var) and isinstance(e2, Var):
209        return e1.varname == e2.varname
210    elif isinstance(e1, Not) and isinstance(e2, Not):
211        return equal_exprs(e1.operand, e2.operand)
212    elif isinstance(e1, And) and isinstance(e2, And):
213        if equal_exprs(e1.operand1, e2.operand1):
214            return equal_exprs(e1.operand2, e2.operand2)
215        elif equal_exprs(e1.operand1, e2.operand2):
216            return equal_exprs(e1.operand2, e2.operand1)
217        else: return False
218    elif isinstance(e1, Or) and isinstance(e2, Or):
219        if equal_exprs(e1.operand1, e2.operand1):
220            return equal_exprs(e1.operand2, e2.operand2)
221        elif equal_exprs(e1.operand1, e2.operand2):
222            return equal_exprs(e1.operand2, e2.operand1)
223        else: return False
224    elif isinstance(e1, Xor) and isinstance(e2, Xor):
225        if equal_exprs(e1.operand1, e2.operand1):
226            return equal_exprs(e1.operand2, e2.operand2)
227        elif equal_exprs(e1.operand1, e2.operand2):
228            return equal_exprs(e1.operand2, e2.operand1)
229        else: return False
230    elif isinstance(e1, Sel) and isinstance(e2, Sel):
231        if equal_exprs(e1.sel, e2.sel):
232             if equal_exprs(e1.true_branch, e2.true_branch):
233                 return equal_exprs(e1.false_branch, e2.false_branch)
234             else: return False
235        else: return False
236
237
238#
239#
240#  Boolean Expressions for Character Class Definitions
241#
242
243def bit_var(n):
244#    return 'bit[%i]' % n
245    return 'basis_bits.bit_%i' % n
246def make_bitv(n):
247    return Var(bit_var(n))
248
249# Deprecated
250def make_2bit_test(var1, var2, bit_pattern):
251  if bit_pattern == 0: 
252    return make_not(make_or(var1, var2))
253  elif bit_pattern == 1:
254    return make_and(make_not(var1), var2)
255  elif bit_pattern == 2:
256    return make_and(var1, make_not(var2))
257  else: return make_and(var1, var2)
258
259# Deprecated
260def make_8bit_test(bit_pattern):
261  return make_and(make_and(make_2bit_test(make_bitv(0), make_bitv(1), (bit_pattern >> 6) & 3),
262                           make_2bit_test(make_bitv(2), make_bitv(3), (bit_pattern >> 4) & 3)),
263                  make_and(make_2bit_test(make_bitv(4), make_bitv(5), (bit_pattern >> 2) & 3),
264                           make_2bit_test(make_bitv(6), make_bitv(7), (bit_pattern) & 3)))
265
266
267def make_bit_test(pattern, bit_count):
268    bit_terms = []
269    test_bit = 2**(bit_count - 1)
270    for i in range(0, bit_count):
271        if (pattern & test_bit) == 0:
272            bit_terms.append(make_not(make_bitv(i)))
273        else: bit_terms.append(make_bitv(i))
274        test_bit >>= 1
275    while len(bit_terms) > 1:
276        new_terms = []
277        for i in range(0, len(bit_terms)/ 2):
278            new_terms.append(make_and(bit_terms[2*i], bit_terms[2*i+1]))
279        if len(bit_terms) % 2 == 1:
280            new_terms.append(bit_terms[-1])
281        bit_terms = new_terms
282    return bit_terms[0]
283
284
285def char_test_expr(ch):
286    return make_bit_test(ord(ch), 8)
287    #return make_8bit_test(ord(ch))
288
289def GE_Range(N, n):
290    if N == 0: return TrueLiteral()
291    elif N % 2 == 0 and (n >> (N - 2)) == 0:
292        return make_or(make_or(make_bitv(8-N), make_bitv(9-N)),
293                        GE_Range(N - 2, n))
294    elif N % 2 == 0 and (n >> (N - 2)) == 3:   # >= 11xxxx
295        return make_and(make_and(make_bitv(8-N), make_bitv(9-N)),
296                        GE_Range(N - 2, n - (3 << (N-2))))
297    elif N >= 1:
298        hi_bit = n & (1 << (N-1))
299        lo_bits = n - hi_bit
300        lo_range = GE_Range(N-1, lo_bits)
301        if hi_bit == 0:
302            # If the hi_bit of n is not set, then whenever the corresponding bit
303            # is set in the target, the target will certainly be >=.  Otherwise,
304            # the value of GE_range(N-1, lo_bits) is required.
305            return make_or(make_bitv(8-N), lo_range)
306        else: 
307            # If the hi_bit of n is set, then the corresponding bit must be set
308            # in the target for >= and GE_range(N-1, lo_bits) must also be true.
309            return make_and(make_bitv(8-N), lo_range)
310
311def LE_Range(N, n):
312    # If an N-bit pattern is all ones, then it is always
313    # true that any n-bit value is LE this pattern.
314    # Handling this as a special case avoids an overflow
315    # issue with n+1 requiring more than N bits.
316    if n+1 == 2 ** N:
317        return TrueLiteral()
318    else:
319        return make_not(GE_Range(N, n+1))
320
321BadRange = Exception()
322
323
324def Make_Range(n1, n2):  # require n2 >= n1
325    diff_bits = n1 ^ n2
326    diff_count = 0
327    while diff_bits > 0:
328        diff_count += 1
329        diff_bits >>= 1
330    if n2 < n1 or diff_count > 8: raise BadRange()
331    common = make_bit_test(n1 >> diff_count, 8 - diff_count)
332    if diff_count == 0: return common
333    mask = 2**(diff_count-1) - 1
334    lo_test = GE_Range(diff_count-1, n1 & mask)
335    hi_test = LE_Range(diff_count-1, n2 & mask)
336    return make_and(common, make_sel(make_bitv(8-diff_count), hi_test, lo_test))
337
338# Deprecated   
339def Inclusive_Range(N, n1, n2):  # require n2 >= n1
340    if N == 0: return TrueLiteral()
341    elif n1 >= 2**(N-1):
342        return make_and(make_bitv(8-N), Inclusive_Range(N-1, n1 - 2**(N-1), n2 - 2**(N-1)))
343    elif n2 < 2**(N-1):
344        return make_and(make_not(make_bitv(8-N)), Inclusive_Range(N-1, n1, n2))
345    else:
346        n2_lo = n2 - 2**(N-1)
347        lo_test = GE_Range(N-1, n1)
348        hi_test = LE_Range(N-1, n2_lo)
349        # special optimization?
350        # if n2_lo + 1 == n1: return make_xor(make_bit_test(8-N), lo_test)
351        return make_sel(make_bitv(8-N), lo_test, hi_test)
352
353BadCharSetItem = Exception()
354
355def char_or_range_expr(charset_item):
356    if len(charset_item) == 1:
357        return char_test_expr(charset_item[0])
358    elif len(charset_item) == 3:
359        if charset_item[1] == '-' and ord(charset_item[0]) <= ord(charset_item[2]):
360             return Make_Range(ord(charset_item[0]), ord(charset_item[2]))
361             #  return Inclusive_Range(8, ord(charset_item[0]), ord(charset_item[2]))
362    print charset_item
363    raise BadCharSetItem
364
365def charset_expr(chardef):
366    e1 = char_or_range_expr(chardef.items[0])
367    for i in range(1, len(chardef.items)):
368        e1 = make_or(e1, char_or_range_expr(chardef.items[i]))
369    if chardef.complemented: return make_not(e1)
370    else: return e1
371
372#
373#
374#  Code Generation
375#
376class CodeGenObject:
377    def __init__(self, predeclared, typedecl='BitBlock '):
378        self.gensym_template = 'temp%i'
379        self.gensym_counter = 0
380        self.generated_code = []
381        self.common_expression_map = {}
382        for sym in predeclared: self.common_expression_map[sym] = sym
383        self.typedecl = typedecl
384    def add_assignment(self, varname, expr):
385        self.common_expression_map[expr] = varname
386        #self.generated_code.append('%s%s = %s;\n' % (self.typedecl, varname, expr))
387        self.generated_code.append('\t%s%s = %s\n' % (self.typedecl, varname, expr))
388    def expr_string_to_variable(self, expr_string):
389        if self.common_expression_map.has_key(expr_string): 
390            return self.common_expression_map[expr_string]
391        else:
392            self.gensym_counter += 1
393            sym = self.gensym_template % self.gensym_counter
394            self.add_assignment(sym, expr_string)
395            return sym
396    def showcode(self):
397        s = ''
398        for stmt in self.generated_code: s += stmt
399        return s
400
401def expr2simd(genobj, expr):
402    """Translate a Boolean expression into three-address Altivec code
403       using code generator object genobj.
404    """
405    if isinstance(expr, TrueLiteral): return 'simd_const_1(1)'
406    elif isinstance(expr, FalseLiteral): return 'simd_const_1(0)'
407    elif isinstance(expr, Var): return expr.varname
408    elif isinstance(expr, Not):
409       e = genobj.expr_string_to_variable(expr2simd(genobj, expr.operand))
410       return 'simd_andc(simd_const_1(1), %s)' % (e)
411    elif isinstance(expr, Or):
412       e1 = genobj.expr_string_to_variable(expr2simd(genobj, expr.operand1))
413       e2 = genobj.expr_string_to_variable(expr2simd(genobj, expr.operand2))
414       return 'simd_or(%s, %s)' % (e1, e2)
415    elif isinstance(expr, Xor):
416       e1 = genobj.expr_string_to_variable(expr2simd(genobj, expr.operand1))
417       e2 = genobj.expr_string_to_variable(expr2simd(genobj, expr.operand2))
418       return 'simd_xor(%s, %s)' % (e1, e2)
419    elif isinstance(expr, And):
420       if isinstance(expr.operand1, Not):
421           e1 = genobj.expr_string_to_variable(expr2simd(genobj, expr.operand1.operand))
422           e2 = genobj.expr_string_to_variable(expr2simd(genobj, expr.operand2))
423           return 'simd_andc(%s, %s)' % (e2, e1)
424       elif isinstance(expr.operand2, Not):
425           e1 = genobj.expr_string_to_variable(expr2simd(genobj, expr.operand1))
426           e2 = genobj.expr_string_to_variable(expr2simd(genobj, expr.operand2.operand))
427           return 'simd_andc(%s, %s)' % (e1, e2)
428       else:
429           e1 = genobj.expr_string_to_variable(expr2simd(genobj, expr.operand1))
430           e2 = genobj.expr_string_to_variable(expr2simd(genobj, expr.operand2))
431           return 'simd_and(%s, %s)' % (e1, e2)
432    elif isinstance(expr, Sel):
433       sel = genobj.expr_string_to_variable(expr2simd(genobj, expr.sel))
434       e1 = genobj.expr_string_to_variable(expr2simd(genobj, expr.true_branch))
435       e2 = genobj.expr_string_to_variable(expr2simd(genobj, expr.false_branch))
436       return 'simd_if(%s, %s, %s)' %(sel, e1, e2)
437
438def chardef2simd(genobj, chardef):
439    genobj.add_assignment(chardef.name, expr2simd(genobj, charset_expr(chardef)))
440
441def chardeflist2simd(chardeflist):
442    cgo = CodeGenObject([bit_var(i) for i in range(0,8)])
443    for d in chardeflist:
444        chardef2simd(cgo, d)
445    return cgo.showcode()
446
447def expr2py(genobj, expr):
448    """Translate a Boolean expression into three-address python code
449       using code generator object genobj.
450    """
451    if isinstance(expr, TrueLiteral): return '-1'
452    elif isinstance(expr, FalseLiteral): return '0'
453    elif isinstance(expr, Var): return expr.varname
454    elif isinstance(expr, Not):
455       e = genobj.expr_string_to_variable(expr2py(genobj, expr.operand))
456       return '(~%s)' % (e)
457    elif isinstance(expr, Or):
458       e1 = genobj.expr_string_to_variable(expr2py(genobj, expr.operand1))
459       e2 = genobj.expr_string_to_variable(expr2py(genobj, expr.operand2))
460       return '(%s | %s)' % (e1, e2)
461    elif isinstance(expr, Xor):
462       e1 = genobj.expr_string_to_variable(expr2py(genobj, expr.operand1))
463       e2 = genobj.expr_string_to_variable(expr2py(genobj, expr.operand2))
464       return '(%s ^ %s)' % (e1, e2)
465    elif isinstance(expr, And):
466       if isinstance(expr.operand1, Not):
467           e1 = genobj.expr_string_to_variable(expr2py(genobj, expr.operand1.operand))
468           e2 = genobj.expr_string_to_variable(expr2py(genobj, expr.operand2))
469           return '(%s &~ %s)' % (e2, e1)
470       elif isinstance(expr.operand2, Not):
471           e1 = genobj.expr_string_to_variable(expr2py(genobj, expr.operand1))
472           e2 = genobj.expr_string_to_variable(expr2py(genobj, expr.operand2.operand))
473           return '(%s &~ %s)' % (e1, e2)
474       else:
475           e1 = genobj.expr_string_to_variable(expr2py(genobj, expr.operand1))
476           e2 = genobj.expr_string_to_variable(expr2py(genobj, expr.operand2))
477           return '(%s & %s)' % (e1, e2)
478    elif isinstance(expr, Sel):
479       sel = genobj.expr_string_to_variable(expr2py(genobj, expr.sel))
480       e1 = genobj.expr_string_to_variable(expr2py(genobj, expr.true_branch))
481       e2 = genobj.expr_string_to_variable(expr2py(genobj, expr.false_branch))
482       return '((%s & %s)|(~(%s) & %s))' %(sel, e1, sel, e2)
483
484def chardef2py(genobj, chardef):
485    genobj.add_assignment(chardef.name, expr2py(genobj, charset_expr(chardef)))
486
487
488def py_chardefmap(chardeflist):
489    defs = ["'%s' : %s" % (d.name,d.name) for d in chardeflist]
490    return  '{%s}' % string.join(defs, ',\n\t')
491
492def chardeflist2py(chardeflist):
493    cgo = CodeGenObject([bit_var(i) for i in range(0,8)],'')
494    for d in chardeflist:
495        chardef2py(cgo, d) 
496    return cgo.showcode()# + "  return "+ py_chardefmap(chardeflist) + "\n"
497
498
499def code_gen_for_transcode(transcode_tbl_h, transcode_tbl_l):
500  cgo = CodeGenObject([bit_var(i) for i in range(0,8)])
501  xor_tbl_l = [transcode_tbl_l[code] ^ code for code in range(256)]
502  code_gen_for_transcode_8bit(cgo, "x16h", transcode_tbl_h)
503  code_gen_for_transcode_8bit(cgo, "x16l", xor_tbl_l)
504  return cgo.showcode()
505
506def code_gen_for_transcode_8bit(cgo, pfx, tbl):
507  for bit in range(8):
508    bit_mask = 256 >> bit
509    in_run = False
510    first_expr_found = False
511    bit_expr = FalseLiteral()
512    for code in range(256):
513      if tbl[code] & bit_mask != 0:
514        if not in_run:
515          in_run = True
516          run_start = code
517      else:
518        if in_run:
519          if run_start == code-1:
520            e1 = char_test_expr(chr(code - 1))
521          else:
522            e1 = Make_Range(run_start, code - 1)
523          if first_expr_found:
524            bit_expr = make_or(bit_expr, e1)
525          else:
526            first_expr_found = True
527            bit_expr = e1
528    if first_expr_found:
529      cgo.add_assignment("%s[%i]" % (pfx, bit), expr2simd(cgo, bit_expr))
530
531# Work with tables from
532# /home/cameron/glibc-2.3.5/localedata/charmaps/
533
534charmap_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")
535
536def read_char_map(file):
537  f = open(file)
538  lines = f.readlines()
539  matches = [charmap_line_RE.match(l) for l in lines]
540  u16hi = [ord(binascii.unhexlify(m.group(1))) for m in matches if m]
541  u16lo = [ord(binascii.unhexlify(m.group(2))) for m in matches if m]
542  codes = [ord(binascii.unhexlify(m.group(3))) for m in matches if m]
543  codes_OK = [c for c in range(256) if codes[c] == c]
544  if len(codes_OK) != 256:
545    print("Code map failure reading %s" % file)
546  return (u16hi, u16lo)
547
548import codecs
549def ascii2ebcdic_chardeflist(defs):
550        encoder = codecs.getencoder('cp037')
551        return [xlate_chardef(d, encoder) for d in defs]
552
553def xlate_char_or_range(charset_item, encoder):
554    if len(charset_item) == 1:
555        return encoder(charset_item[0])
556    elif len(charset_item) == 3:
557        if charset_item[1] == '-' and ord(charset_item[0]) <= ord(charset_item[2]):
558             return Make_Range(ord(charset_item[0]), ord(charset_item[2]))
559             #  return Inclusive_Range(8, ord(charset_item[0]), ord(charset_item[2]))
560    print charset_item
561    raise BadCharSetItem
562       
563def xlate_chardef(chardef, encoder):
564  if isinstance(chardef, CharDef):
565    return CharDef(chardef.name, encoder(chardef.items[0])[0], chardef.complemented)
566  else:
567    cdefs = []
568    for item in chardef.items:
569        if len(item) == 1: cdefs.append(encoder(item)[0])
570        elif len(item) == 3:
571          for v in range(ord(item[0]), ord(item[-1])+1):
572            cdefs.append(encoder(chr(v))[0])
573        else: raise BadCharSetItem
574    return CharSetDef(chardef.name, cdefs, chardef.complemented)
575
576def main():
577    if len(sys.argv) == 2:
578        # if the specified argument is not in the DefinitionSet, then assume that it's a filename
579        if sys.argv[1] not in DefinitionSet:
580            #define the characters in the list
581            defs = charset_input_parser.input_chardef(sys.argv[1])
582                                               
583            #print chardeflist2simd(defs)
584            print chardeflist2py(defs)
585        else:
586            #print chardeflist2simd(DefinitionSet[sys.argv[1]])
587            print chardeflist2py(DefinitionSet[sys.argv[1]])
588    elif len(sys.argv) == 3 and sys.argv[2] == 'EBCDIC':
589        defs = ascii2ebcdic_chardeflist(DefinitionSet[sys.argv[1]])
590        print chardeflist2simd(defs)
591    else:
592        print "Usage python charset_compiler.py <name> [EBCDIC]"
593        for k in DefinitionSet.keys(): print k
594
595if __name__ == "__main__": main()
Note: See TracBrowser for help on using the repository browser.