| OLD | NEW |
| 1 # Copyright 2014 The Chromium Authors. All rights reserved. | 1 # Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 """Generates Python source files from a mojom.Module.""" | 5 """Generates Python source files from a mojom.Module.""" |
| 6 | 6 |
| 7 import re | 7 import re |
| 8 | 8 |
| 9 import mojom.generate.generator as generator | 9 import mojom.generate.generator as generator |
| 10 import mojom.generate.module as mojom | 10 import mojom.generate.module as mojom |
| (...skipping 16 matching lines...) Expand all Loading... |
| 27 uccc = UpperCamelCase(name) | 27 uccc = UpperCamelCase(name) |
| 28 return uccc[0].lower() + uccc[1:] | 28 return uccc[0].lower() + uccc[1:] |
| 29 | 29 |
| 30 def ConstantStyle(name): | 30 def ConstantStyle(name): |
| 31 components = NameToComponent(name) | 31 components = NameToComponent(name) |
| 32 if components[0] == 'k': | 32 if components[0] == 'k': |
| 33 components = components[1:] | 33 components = components[1:] |
| 34 return '_'.join([x.upper() for x in components]) | 34 return '_'.join([x.upper() for x in components]) |
| 35 | 35 |
| 36 def GetNameForElement(element): | 36 def GetNameForElement(element): |
| 37 if isinstance(element, mojom.EnumValue): |
| 38 return (UpperCamelCase(element.enum_name) + '.' + |
| 39 ConstantStyle(element.name)) |
| 37 if isinstance(element, (mojom.NamedValue, | 40 if isinstance(element, (mojom.NamedValue, |
| 38 mojom.Constant)): | 41 mojom.Constant)): |
| 39 return ConstantStyle(element.name) | 42 return ConstantStyle(element.name) |
| 40 raise Exception('Unexpected element: ' % element) | 43 raise Exception('Unexpected element: ' % element) |
| 41 | 44 |
| 42 def TranslateConstants(token): | 45 def ExpressionToText(token): |
| 43 if isinstance(token, (mojom.EnumValue, mojom.NamedValue)): | 46 if isinstance(token, (mojom.EnumValue, mojom.NamedValue)): |
| 44 # Both variable and enum constants are constructed like: | 47 # Both variable and enum constants are constructed like: |
| 45 # NamespaceUid.Struct[.Enum].CONSTANT_NAME | 48 # PythonModule[.Struct][.Enum].CONSTANT_NAME |
| 46 name = [] | 49 name = [] |
| 47 if token.imported_from: | 50 if token.imported_from: |
| 48 name.append(token.imported_from['python_module']) | 51 name.append(token.imported_from['python_module']) |
| 49 if token.parent_kind: | 52 if token.parent_kind: |
| 50 name.append(GetNameForElement(token.parent_kind)) | 53 name.append(GetNameForElement(token.parent_kind)) |
| 51 if isinstance(token, mojom.EnumValue): | 54 if isinstance(token, mojom.EnumValue): |
| 52 name.append(GetNameForElement(token)) | 55 name.append(GetNameForElement(token)) |
| 53 else: | 56 else: |
| 54 name.append(token.name) | 57 name.append(GetNameForElement(token)) |
| 55 return '.'.join(name) | 58 return '.'.join(name) |
| 56 | 59 |
| 57 if isinstance(token, mojom.BuiltinValue): | 60 if isinstance(token, mojom.BuiltinValue): |
| 58 if token.value == 'double.INFINITY' or token.value == 'float.INFINITY': | 61 if token.value == 'double.INFINITY' or token.value == 'float.INFINITY': |
| 59 return 'float(\'inf\')'; | 62 return 'float(\'inf\')'; |
| 60 if (token.value == 'double.NEGATIVE_INFINITY' or | 63 if (token.value == 'double.NEGATIVE_INFINITY' or |
| 61 token.value == 'float.NEGATIVE_INFINITY'): | 64 token.value == 'float.NEGATIVE_INFINITY'): |
| 62 return 'float(\'-inf\')' | 65 return 'float(\'-inf\')' |
| 63 if token.value == 'double.NAN' or token.value == 'float.NAN': | 66 if token.value == 'double.NAN' or token.value == 'float.NAN': |
| 64 return 'float(\'nan\')'; | 67 return 'float(\'nan\')'; |
| 65 | 68 |
| 66 return token | 69 return token |
| 67 | 70 |
| 68 | 71 |
| 69 def ExpressionToText(value): | |
| 70 return TranslateConstants(value) | |
| 71 | |
| 72 | |
| 73 class Generator(generator.Generator): | 72 class Generator(generator.Generator): |
| 74 | 73 |
| 75 python_filters = { | 74 python_filters = { |
| 76 'expression_to_text': ExpressionToText, | 75 'expression_to_text': ExpressionToText, |
| 77 'name': GetNameForElement, | 76 'name': GetNameForElement, |
| 78 } | 77 } |
| 79 | 78 |
| 80 @UseJinja('python_templates/module.py.tmpl', filters=python_filters) | 79 @UseJinja('python_templates/module.py.tmpl', filters=python_filters) |
| 81 def GeneratePythonModule(self): | 80 def GeneratePythonModule(self): |
| 82 return { | 81 return { |
| 83 'imports': self.GetImports(), | 82 'imports': self.GetImports(), |
| 83 'enums': self.module.enums, |
| 84 'module': self.module, | 84 'module': self.module, |
| 85 } | 85 } |
| 86 | 86 |
| 87 def GenerateFiles(self, args): | 87 def GenerateFiles(self, args): |
| 88 self.Write(self.GeneratePythonModule(), | 88 self.Write(self.GeneratePythonModule(), |
| 89 '%s.py' % self.module.name.replace('.mojom', '_mojom')) | 89 '%s.py' % self.module.name.replace('.mojom', '_mojom')) |
| 90 | 90 |
| 91 def GetImports(self): | 91 def GetImports(self): |
| 92 for each in self.module.imports: | 92 for each in self.module.imports: |
| 93 each['python_module'] = each['module_name'].replace('.mojom', '_mojom') | 93 each['python_module'] = each['module_name'].replace('.mojom', '_mojom') |
| 94 return self.module.imports | 94 return self.module.imports |
| 95 | 95 |
| 96 def GetJinjaParameters(self): | 96 def GetJinjaParameters(self): |
| 97 return { | 97 return { |
| 98 'lstrip_blocks': True, | 98 'lstrip_blocks': True, |
| 99 'trim_blocks': True, | 99 'trim_blocks': True, |
| 100 } | 100 } |
| OLD | NEW |