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 |