git.cworth.org Git - apitrace/blob - codegen.py

   1 ##########################################################################
   2 #
   3 # Copyright 2010 VMware, Inc.
   4 # All Rights Reserved.
   5 #
   6 # Permission is hereby granted, free of charge, to any person obtaining a copy
   7 # of this software and associated documentation files (the "Software"), to deal
   8 # in the Software without restriction, including without limitation the rights
   9 # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  10 # copies of the Software, and to permit persons to whom the Software is
  11 # furnished to do so, subject to the following conditions:
  12 #
  13 # The above copyright notice and this permission notice shall be included in
  14 # all copies or substantial portions of the Software.
  15 #
  16 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17 # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18 # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  19 # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20 # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  21 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  22 # THE SOFTWARE.
  23 #
  24 ##########################################################################/
  25
  26
  27 """C code generation helpers."""
  28
  29
  30 def _string_switch(var, prefix, suffixes, case, default):
  31     index = len(prefix)
  32     indent = '    '*(index + 1)
  33
  34     if len(suffixes) == 1:
  35         suffix = suffixes[0]
  36         chars = list(suffix) + ["\\0"]
  37         predicates = ['%s[%u] == \'%s\'' % (var, index + i, chars[i]) for i in range(len(chars))]
  38         print indent + 'if (%s) {' % (' && '.join(predicates),)
  39         value = prefix + suffix
  40         print indent + '    // %s' % value
  41         case(value)
  42         print indent + '}'
  43         if default is not None:
  44             print indent + 'else {'
  45             default()
  46             print indent + '}'
  47         return
  48
  49     print indent + 'switch (%s[%u]) {' % (var, index)
  50     if "" in suffixes:
  51         print indent + 'case \'\\0\':'
  52         print indent + '    // %s' % prefix
  53         case(prefix)
  54         print indent + '    break;'
  55
  56     chars = [suffix[0] for suffix in suffixes if suffix]
  57     chars = list(set(chars))
  58     chars.sort()
  59
  60     for char in chars:
  61         print indent + 'case \'%c\':' % (char)
  62         new_prefix = prefix + char
  63         new_suffixes = [suffix[1:] for suffix in suffixes if suffix.startswith(char)]
  64         _string_switch(var, new_prefix, new_suffixes, case, default)
  65         print indent + '    break;'
  66     if default is not None:
  67         print indent + 'default:'
  68         default()
  69         print indent + '    break;'
  70     print indent + '}'
  71
  72
  73 def _noop(*args, **kwargs):
  74     pass
  75
  76
  77 def string_switch(var, values, case=_noop, default=None):
  78     """Product a switch of strings, using a tree character switches."""
  79     values = list(values)
  80     values.sort()
  81     _string_switch(var, '', values, case, default)