src/tools/json_schema_compiler/ppapi_generator.py

   1 # Copyright 2013 The Chromium Authors. All rights reserved.
   2 # Use of this source code is governed by a BSD-style license that can be
   3 # found in the LICENSE file.
   4
   5 import collections
   6 import datetime
   7 import os.path
   8 import sys
   9
  10 import code
  11 import cpp_util
  12 import model
  13
  14 try:
  15   import jinja2
  16 except ImportError:
  17   sys.path.append(os.path.join(os.path.dirname(__file__), '..', '..',
  18                                'third_party'))
  19   import jinja2
  20
  21
  22 class _PpapiGeneratorBase(object):
  23   """A base class for ppapi generators.
  24
  25   Implementations should set TEMPLATE_NAME to a string containing the name of
  26   the template file without its extension. The template will be rendered with
  27   the following symbols available:
  28     name: A string containing the name of the namespace.
  29     enums: A list of enums within the namespace.
  30     types: A list of types within the namespace, sorted such that no element
  31         depends on an earlier element.
  32     events: A dict of events within the namespace.
  33     functions: A dict of functions within the namespace.
  34     year: An int containing the current year.
  35     source_file: The name of the input file.
  36   """
  37
  38   def __init__(self, namespace):
  39     self._namespace = namespace
  40     self._required_types = {}
  41     self._array_types = set()
  42     self._optional_types = set()
  43     self._optional_array_types = set()
  44     self._dependencies = collections.OrderedDict()
  45     self._types = []
  46     self._enums = []
  47
  48     self.jinja_environment = jinja2.Environment(
  49         loader=jinja2.FileSystemLoader(os.path.join(os.path.dirname(__file__),
  50                                                     'templates', 'ppapi')))
  51     self._SetupFilters()
  52     self._ResolveTypeDependencies()
  53
  54   def _SetupFilters(self):
  55     self.jinja_environment.filters.update({
  56       'ppapi_type': self.ToPpapiType,
  57       'classname': cpp_util.Classname,
  58       'enum_value': self.EnumValueName,
  59       'return_type': self.GetFunctionReturnType,
  60       'format_param_type': self.FormatParamType,
  61       'needs_optional': self.NeedsOptional,
  62       'needs_array': self.NeedsArray,
  63       'needs_optional_array': self.NeedsOptionalArray,
  64       'has_array_outs': self.HasArrayOuts,
  65     })
  66
  67   def Render(self, template_name, values):
  68     generated_code = code.Code()
  69     template = self.jinja_environment.get_template(
  70         '%s.template' % template_name)
  71     generated_code.Append(template.render(values))
  72     return generated_code
  73
  74   def Generate(self):
  75     """Generates a Code object for a single namespace."""
  76     return self.Render(self.TEMPLATE_NAME, {
  77       'name': self._namespace.name,
  78       'enums': self._enums,
  79       'types': self._types,
  80       'events': self._namespace.events,
  81       'functions': self._namespace.functions,
  82       # TODO(sammc): Don't change years when regenerating existing output files.
  83       'year': datetime.date.today().year,
  84       'source_file': self._namespace.source_file,
  85     })
  86
  87   def _ResolveTypeDependencies(self):
  88     """Calculates the transitive closure of the types in _required_types.
  89
  90     Returns a tuple containing the list of struct types and the list of enum
  91     types. The list of struct types is ordered such that no type depends on a
  92     type later in the list.
  93
  94     """
  95     if self._namespace.functions:
  96       for function in self._namespace.functions.itervalues():
  97         self._FindFunctionDependencies(function)
  98
  99     if self._namespace.events:
 100       for event in self._namespace.events.itervalues():
 101         self._FindFunctionDependencies(event)
 102     resolved_types = set()
 103     while resolved_types < set(self._required_types):
 104       for typename in sorted(set(self._required_types) - resolved_types):
 105         type_ = self._required_types[typename]
 106         self._dependencies.setdefault(typename, set())
 107         for member in type_.properties.itervalues():
 108           self._RegisterDependency(member, self._NameComponents(type_))
 109         resolved_types.add(typename)
 110     while self._dependencies:
 111       for name, deps in self._dependencies.items():
 112         if not deps:
 113           if (self._required_types[name].property_type ==
 114               model.PropertyType.ENUM):
 115             self._enums.append(self._required_types[name])
 116           else:
 117             self._types.append(self._required_types[name])
 118           for deps in self._dependencies.itervalues():
 119             deps.discard(name)
 120           del self._dependencies[name]
 121           break
 122       else:
 123         raise ValueError('Circular dependency %s' % self._dependencies)
 124
 125   def _FindFunctionDependencies(self, function):
 126     for param in function.params:
 127       self._RegisterDependency(param, None)
 128     if function.callback:
 129       for param in function.callback.params:
 130         self._RegisterDependency(param, None)
 131     if function.returns:
 132       self._RegisterTypeDependency(function.returns, None, False, False)
 133
 134   def _RegisterDependency(self, member, depender):
 135     self._RegisterTypeDependency(member.type_, depender, member.optional, False)
 136
 137   def _RegisterTypeDependency(self, type_, depender, optional, array):
 138     if type_.property_type == model.PropertyType.ARRAY:
 139       self._RegisterTypeDependency(type_.item_type, depender, optional, True)
 140     elif type_.property_type == model.PropertyType.REF:
 141       self._RegisterTypeDependency(self._namespace.types[type_.ref_type],
 142                                    depender, optional, array)
 143     elif type_.property_type in (model.PropertyType.OBJECT,
 144                                  model.PropertyType.ENUM):
 145       name_components = self._NameComponents(type_)
 146       self._required_types[name_components] = type_
 147       if depender:
 148         self._dependencies.setdefault(depender, set()).add(
 149             name_components)
 150       if array:
 151         self._array_types.add(name_components)
 152         if optional:
 153           self._optional_array_types.add(name_components)
 154       elif optional:
 155         self._optional_types.add(name_components)
 156
 157   @staticmethod
 158   def _NameComponents(entity):
 159     """Returns a tuple of the fully-qualified name of an entity."""
 160     names = []
 161     while entity:
 162       if (not isinstance(entity, model.Type) or
 163           entity.property_type != model.PropertyType.ARRAY):
 164         names.append(entity.name)
 165       entity = entity.parent
 166     return tuple(reversed(names[:-1]))
 167
 168   def ToPpapiType(self, type_, array=False, optional=False):
 169     """Returns a string containing the name of the Pepper C type for |type_|.
 170
 171     If array is True, returns the name of an array of |type_|. If optional is
 172     True, returns the name of an optional |type_|. If both array and optional
 173     are True, returns the name of an optional array of |type_|.
 174     """
 175     if isinstance(type_, model.Function) or type_.property_type in (
 176         model.PropertyType.OBJECT, model.PropertyType.ENUM):
 177       return self._FormatPpapiTypeName(
 178           array, optional, '_'.join(
 179               cpp_util.Classname(s) for s in self._NameComponents(type_)),
 180               namespace=cpp_util.Classname(self._namespace.name))
 181     elif type_.property_type == model.PropertyType.REF:
 182       return self.ToPpapiType(self._namespace.types[type_.ref_type],
 183                               optional=optional, array=array)
 184     elif type_.property_type == model.PropertyType.ARRAY:
 185       return self.ToPpapiType(type_.item_type, array=True,
 186                               optional=optional)
 187     elif type_.property_type == model.PropertyType.STRING and not array:
 188       return 'PP_Var'
 189     elif array or optional:
 190       if type_.property_type in self._PPAPI_COMPOUND_PRIMITIVE_TYPE_MAP:
 191         return self._FormatPpapiTypeName(
 192             array, optional,
 193             self._PPAPI_COMPOUND_PRIMITIVE_TYPE_MAP[type_.property_type], '')
 194     return self._PPAPI_PRIMITIVE_TYPE_MAP.get(type_.property_type, 'PP_Var')
 195
 196   _PPAPI_PRIMITIVE_TYPE_MAP = {
 197     model.PropertyType.BOOLEAN: 'PP_Bool',
 198     model.PropertyType.DOUBLE: 'double_t',
 199     model.PropertyType.INT64: 'int64_t',
 200     model.PropertyType.INTEGER: 'int32_t',
 201   }
 202   _PPAPI_COMPOUND_PRIMITIVE_TYPE_MAP = {
 203     model.PropertyType.BOOLEAN: 'Bool',
 204     model.PropertyType.DOUBLE: 'Double',
 205     model.PropertyType.INT64: 'Int64',
 206     model.PropertyType.INTEGER: 'Int32',
 207     model.PropertyType.STRING: 'String',
 208   }
 209
 210   @staticmethod
 211   def _FormatPpapiTypeName(array, optional, name, namespace=''):
 212     if namespace:
 213       namespace = '%s_' % namespace
 214     if array:
 215       if optional:
 216         return 'PP_%sOptional_%s_Array' % (namespace, name)
 217       return 'PP_%s%s_Array' % (namespace, name)
 218     if optional:
 219       return 'PP_%sOptional_%s' % (namespace, name)
 220     return 'PP_%s%s' % (namespace, name)
 221
 222   def NeedsOptional(self, type_):
 223     """Returns True if an optional |type_| is required."""
 224     return self._NameComponents(type_) in self._optional_types
 225
 226   def NeedsArray(self, type_):
 227     """Returns True if an array of |type_| is required."""
 228     return self._NameComponents(type_) in self._array_types
 229
 230   def NeedsOptionalArray(self, type_):
 231     """Returns True if an optional array of |type_| is required."""
 232     return self._NameComponents(type_) in self._optional_array_types
 233
 234   def FormatParamType(self, param):
 235     """Formats the type of a parameter or property."""
 236     return self.ToPpapiType(param.type_, optional=param.optional)
 237
 238   @staticmethod
 239   def GetFunctionReturnType(function):
 240     return 'int32_t' if function.callback or function.returns else 'void'
 241
 242   def EnumValueName(self, enum_value, enum_type):
 243     """Returns a string containing the name for an enum value."""
 244     return '%s_%s' % (self.ToPpapiType(enum_type).upper(),
 245                       enum_value.name.upper())
 246
 247   def _ResolveType(self, type_):
 248     if type_.property_type == model.PropertyType.REF:
 249       return self._ResolveType(self._namespace.types[type_.ref_type])
 250     if type_.property_type == model.PropertyType.ARRAY:
 251       return self._ResolveType(type_.item_type)
 252     return type_
 253
 254   def _IsOrContainsArray(self, type_):
 255     if type_.property_type == model.PropertyType.ARRAY:
 256       return True
 257     type_ = self._ResolveType(type_)
 258     if type_.property_type == model.PropertyType.OBJECT:
 259       return any(self._IsOrContainsArray(param.type_)
 260                  for param in type_.properties.itervalues())
 261     return False
 262
 263   def HasArrayOuts(self, function):
 264     """Returns True if the function produces any arrays as outputs.
 265
 266     This includes arrays that are properties of other objects.
 267     """
 268     if function.callback:
 269       for param in function.callback.params:
 270         if self._IsOrContainsArray(param.type_):
 271           return True
 272     return function.returns and self._IsOrContainsArray(function.returns)
 273
 274
 275 class _IdlGenerator(_PpapiGeneratorBase):
 276   TEMPLATE_NAME = 'idl'
 277
 278
 279 class _GeneratorWrapper(object):
 280   def __init__(self, generator_factory):
 281     self._generator_factory = generator_factory
 282
 283   def Generate(self, namespace):
 284     return self._generator_factory(namespace).Generate()
 285
 286
 287 class PpapiGenerator(object):
 288   def __init__(self):
 289     self.idl_generator = _GeneratorWrapper(_IdlGenerator)