| OLD | NEW |
| (Empty) |
| 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) | |
| OLD | NEW |