OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # | 2 # |
3 # Copyright 2014 The Chromium Authors. All rights reserved. | 3 # Copyright 2014 The Chromium Authors. All rights reserved. |
4 # Use of this source code is governed by a BSD-style license that can be | 4 # Use of this source code is governed by a BSD-style license that can be |
5 # found in the LICENSE file. | 5 # found in the LICENSE file. |
6 | 6 |
7 import collections | 7 import collections |
8 import re | 8 import re |
9 import optparse | 9 import optparse |
10 import os | 10 import os |
(...skipping 43 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
54 | 54 |
55 | 55 |
56 def _StripPrefix(self): | 56 def _StripPrefix(self): |
57 if not self.prefix_to_strip: | 57 if not self.prefix_to_strip: |
58 prefix_to_strip = re.sub('(?!^)([A-Z]+)', r'_\1', self.class_name).upper() | 58 prefix_to_strip = re.sub('(?!^)([A-Z]+)', r'_\1', self.class_name).upper() |
59 prefix_to_strip += '_' | 59 prefix_to_strip += '_' |
60 if not all([w.startswith(prefix_to_strip) for w in self.entries.keys()]): | 60 if not all([w.startswith(prefix_to_strip) for w in self.entries.keys()]): |
61 prefix_to_strip = '' | 61 prefix_to_strip = '' |
62 else: | 62 else: |
63 prefix_to_strip = self.prefix_to_strip | 63 prefix_to_strip = self.prefix_to_strip |
64 entries = ((k.replace(prefix_to_strip, '', 1), v) for (k, v) in | 64 |
65 self.entries.iteritems()) | 65 entries = collections.OrderedDict() |
66 self.entries = collections.OrderedDict(entries) | 66 for (k, v) in self.entries.iteritems(): |
| 67 stripped_key = k.replace(prefix_to_strip, '', 1) |
| 68 if isinstance(v, basestring): |
| 69 stripped_value = v.replace(prefix_to_strip, '', 1) |
| 70 else: |
| 71 stripped_value = v |
| 72 entries[stripped_key] = stripped_value |
| 73 |
| 74 self.entries = entries |
67 | 75 |
68 class HeaderParser(object): | 76 class HeaderParser(object): |
69 single_line_comment_re = re.compile(r'\s*//') | 77 single_line_comment_re = re.compile(r'\s*//') |
70 multi_line_comment_start_re = re.compile(r'\s*/\*') | 78 multi_line_comment_start_re = re.compile(r'\s*/\*') |
71 enum_start_re = re.compile(r'^\s*enum\s+(\w+)\s+{\s*$') | 79 enum_start_re = re.compile(r'^\s*enum\s+(\w+)\s+{\s*$') |
72 enum_line_re = re.compile(r'^\s*(\w+)(\s*\=\s*([^,\n]+))?,?') | 80 enum_line_re = re.compile(r'^\s*(\w+)(\s*\=\s*([^,\n]+))?,?') |
73 enum_end_re = re.compile(r'^\s*}\s*;\s*$') | 81 enum_end_re = re.compile(r'^\s*}\s*;\.*$') |
74 generator_directive_re = re.compile( | 82 generator_directive_re = re.compile( |
75 r'^\s*//\s+GENERATED_JAVA_(\w+)\s*:\s*([\.\w]+)$') | 83 r'^\s*//\s+GENERATED_JAVA_(\w+)\s*:\s*([\.\w]+)$') |
76 | 84 |
77 def __init__(self, lines): | 85 def __init__(self, lines): |
78 self._lines = lines | 86 self._lines = lines |
79 self._enum_definitions = [] | 87 self._enum_definitions = [] |
80 self._in_enum = False | 88 self._in_enum = False |
81 self._current_definition = None | 89 self._current_definition = None |
82 self._generator_directives = {} | 90 self._generator_directives = {} |
83 | 91 |
(...skipping 18 matching lines...) Expand all Loading... |
102 if enum_end: | 110 if enum_end: |
103 self._ApplyGeneratorDirectives() | 111 self._ApplyGeneratorDirectives() |
104 self._current_definition.Finalize() | 112 self._current_definition.Finalize() |
105 self._enum_definitions.append(self._current_definition) | 113 self._enum_definitions.append(self._current_definition) |
106 self._in_enum = False | 114 self._in_enum = False |
107 elif enum_entry: | 115 elif enum_entry: |
108 enum_key = enum_entry.groups()[0] | 116 enum_key = enum_entry.groups()[0] |
109 enum_value = enum_entry.groups()[2] | 117 enum_value = enum_entry.groups()[2] |
110 self._current_definition.AppendEntry(enum_key, enum_value) | 118 self._current_definition.AppendEntry(enum_key, enum_value) |
111 | 119 |
| 120 def _GetCurrentEnumClassNameOverride(self): |
| 121 return self._generator_directives.get('CLASS_NAME_OVERRIDE') |
| 122 |
112 def _GetCurrentEnumPackageName(self): | 123 def _GetCurrentEnumPackageName(self): |
113 return self._generator_directives.get('ENUM_PACKAGE') | 124 return self._generator_directives.get('ENUM_PACKAGE') |
114 | 125 |
115 def _GetCurrentEnumPrefixToStrip(self): | 126 def _GetCurrentEnumPrefixToStrip(self): |
116 return self._generator_directives.get('PREFIX_TO_STRIP', '') | 127 return self._generator_directives.get('PREFIX_TO_STRIP', '') |
117 | 128 |
118 def _ApplyGeneratorDirectives(self): | 129 def _ApplyGeneratorDirectives(self): |
119 current_definition = self._current_definition | 130 current_definition = self._current_definition |
120 current_definition.class_package = self._GetCurrentEnumPackageName() | 131 current_definition.class_package = self._GetCurrentEnumPackageName() |
121 current_definition.prefix_to_strip = self._GetCurrentEnumPrefixToStrip() | 132 current_definition.prefix_to_strip = self._GetCurrentEnumPrefixToStrip() |
122 self._generator_directives = {} | 133 self._generator_directives = {} |
123 | 134 |
124 def _ParseRegularLine(self, line): | 135 def _ParseRegularLine(self, line): |
125 enum_start = HeaderParser.enum_start_re.match(line) | 136 enum_start = HeaderParser.enum_start_re.match(line) |
126 generator_directive = HeaderParser.generator_directive_re.match(line) | 137 generator_directive = HeaderParser.generator_directive_re.match(line) |
127 if enum_start: | 138 if enum_start: |
128 if not self._GetCurrentEnumPackageName(): | 139 if not self._GetCurrentEnumPackageName(): |
129 return | 140 return |
130 self._current_definition = EnumDefinition() | 141 self._current_definition = EnumDefinition() |
131 self._current_definition.class_name = enum_start.groups()[0] | 142 self._current_definition.class_name = ( |
| 143 self._GetCurrentEnumClassNameOverride() or enum_start.groups()[0]) |
132 self._in_enum = True | 144 self._in_enum = True |
133 elif generator_directive: | 145 elif generator_directive: |
134 directive_name = generator_directive.groups()[0] | 146 directive_name = generator_directive.groups()[0] |
135 directive_value = generator_directive.groups()[1] | 147 directive_value = generator_directive.groups()[1] |
136 self._generator_directives[directive_name] = directive_value | 148 self._generator_directives[directive_name] = directive_value |
137 | 149 |
138 | 150 |
139 def GetScriptName(): | 151 def GetScriptName(): |
140 script_components = os.path.abspath(sys.argv[0]).split(os.path.sep) | 152 script_components = os.path.abspath(sys.argv[0]).split(os.path.sep) |
141 build_index = script_components.index('build') | 153 build_index = script_components.index('build') |
(...skipping 86 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
228 | 240 |
229 output_paths = DoGenerate(options, args) | 241 output_paths = DoGenerate(options, args) |
230 | 242 |
231 if options.assert_files_list: | 243 if options.assert_files_list: |
232 AssertFilesList(output_paths, options.assert_files_list) | 244 AssertFilesList(output_paths, options.assert_files_list) |
233 | 245 |
234 return " ".join(output_paths) | 246 return " ".join(output_paths) |
235 | 247 |
236 if __name__ == '__main__': | 248 if __name__ == '__main__': |
237 DoMain(sys.argv[1:]) | 249 DoMain(sys.argv[1:]) |
OLD | NEW |