diff options
author | Elliott Hughes <enh@google.com> | 2012-04-03 16:04:52 -0700 |
---|---|---|
committer | Elliott Hughes <enh@google.com> | 2012-04-04 11:18:48 -0700 |
commit | 0e57ccbbc2de9eeaeecd699575aab22a3f555619 (patch) | |
tree | 3137981abc7f07f3cdcc87bfbf5fbd73fa0c966a /tools | |
parent | 0879504b6ac72f6d93dcccf9e1b7e1769a73b230 (diff) | |
download | android_art-0e57ccbbc2de9eeaeecd699575aab22a3f555619.tar.gz android_art-0e57ccbbc2de9eeaeecd699575aab22a3f555619.tar.bz2 android_art-0e57ccbbc2de9eeaeecd699575aab22a3f555619.zip |
Start automatically generating operator<<s for enums.
Change-Id: I0f53db089b9a1ba38ce82b75ab22448877be67e0
Diffstat (limited to 'tools')
-rwxr-xr-x | tools/generate-operator-out.py | 150 |
1 files changed, 150 insertions, 0 deletions
diff --git a/tools/generate-operator-out.py b/tools/generate-operator-out.py new file mode 100755 index 0000000000..b1d102eab6 --- /dev/null +++ b/tools/generate-operator-out.py @@ -0,0 +1,150 @@ +#!/usr/bin/python2.4 +# +# Copyright 2012 Google Inc. All Rights Reserved. +# +# Redistribution and use in source and binary forms, with or without +# modification, are permitted provided that the following conditions are +# met: +# +# * Redistributions of source code must retain the above copyright +# notice, this list of conditions and the following disclaimer. +# * Redistributions in binary form must reproduce the above +# copyright notice, this list of conditions and the following disclaimer +# in the documentation and/or other materials provided with the +# distribution. +# * Neither the name of Google Inc. nor the names of its +# contributors may be used to endorse or promote products derived from +# this software without specific prior written permission. +# +# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS +# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT +# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR +# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT +# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, +# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT +# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, +# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY +# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT +# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE +# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + +"""Generates default implementations of operator<< for enum types.""" + +import codecs +import os +import re +import string +import sys + + +_ENUM_START_RE = re.compile(r'\benum\b\s+(\S+)\s+\{') +_ENUM_VALUE_RE = re.compile(r'([A-Za-z0-9_]+)(.*)') +_ENUM_END_RE = re.compile(r'^\s*\};$') +_ENUMS = {} + +def Confused(filename, line_number, line): + sys.stderr.write('%s:%d: confused by:\n%s\n' % (filename, line_number, line)) + sys.exit(1) + + +def ProcessFile(filename): + lines = codecs.open(filename, 'r', 'utf8', 'replace').read().split('\n') + in_enum = False + line_number = 0 + for raw_line in lines: + line_number += 1 + + # TODO: take namespaces and enclosing classes/structs into account. + + # Is this the start of a new enum? + if not in_enum: + m = _ENUM_START_RE.search(raw_line) + if m: + # Yes, so add an empty entry to _ENUMS for this enum. + enum_name = m.group(1) + _ENUMS[enum_name] = [] + in_enum = True + continue + + # Is this the end of the current enum? + m = _ENUM_END_RE.search(raw_line) + if m: + if not in_enum: + Confused(filename, line_number, raw_line) + in_enum = False + continue + + # Is this another enum value? + m = _ENUM_VALUE_RE.search(raw_line.strip()) + if not m: + Confused(filename, line_number, raw_line) + + enum_value = m.group(1) + + # By default, we turn "kSomeValue" into "SomeValue". + enum_text = enum_value + if enum_text.startswith('k'): + enum_text = enum_text[1:] + + # Lose literal values because we don't care; turn "= 123, // blah" into ", // blah". + rest = m.group(2).strip() + m_literal = re.compile(r'= (0x[0-9a-f]+|-?[0-9]+)').search(rest) + if m_literal: + rest = rest[(len(m_literal.group(0))):] + + # With "kSomeValue = kOtherValue," we take the original and skip later synonyms. + # TODO: check that the rhs is actually an existing value. + if rest.startswith('= k'): + continue + + # Remove any trailing comma and whitespace + if rest.startswith(','): + rest = rest[1:] + rest = rest.strip() + + # Anything left should be a comment. + if len(rest) and not rest.startswith('// '): + print rest + Confused(filename, line_number, raw_line) + + m_comment = re.compile(r'<<(.*?)>>').search(rest) + if m_comment: + enum_text = m_comment.group(1) + + _ENUMS[enum_name].append((enum_value, enum_text)) + +def main(): + header_files = [] + for header_file in sys.argv[1:]: + header_files.append(header_file) + ProcessFile(header_file) + + print '#include <iostream>' + print + + for header_file in header_files: + print '#include "%s"' % header_file + + print + print 'namespace art {' + print + + for enum_name in _ENUMS: + print '// This was automatically generated by %s --- do not edit!' % sys.argv[0] + print 'std::ostream& operator<<(std::ostream& os, const %s& rhs) {' % enum_name + print ' switch (rhs) {' + for (enum_value, enum_text) in _ENUMS[enum_name]: + print ' case %s: os << "%s"; break;' % (enum_value, enum_text) + print ' default: os << "%s[" << static_cast<int>(rhs) << "]"; break;' % enum_name + print ' }' + print ' return os;' + print '}' + print + + print '} // namespace art' + + sys.exit(0) + + +if __name__ == '__main__': + main() |