Initial code generation for features.

tools/json_schema_compiler/features_cc_generator.py

+# Copyright (c) 2013 The Chromium Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+import os.path
+
+from code import Code
+import cpp_util
+
+
+class CCGenerator(object):
+  def Generate(self, feature_defs, source_file, namespace):
+    return _Generator(feature_defs, source_file, namespace).Generate()
+
+
+class _Generator(object):
+  """A .cc generator for features.
+  """
+  def __init__(self, feature_defs, source_file, namespace):
+    self._feature_defs = feature_defs
+    self._source_file = source_file
+    self._source_file_filename, _ = os.path.splitext(source_file)
+    self._class_name = cpp_util.ClassName(self._source_file_filename)
+    self._namespace = namespace
+
+  def Generate(self):
+    """Generates a Code object for features.
+    """
+    c = Code()
+    (c.Append(cpp_util.CHROMIUM_LICENSE)
+      .Append()
+      .Append(cpp_util.GENERATED_FEATURE_MESSAGE % self._source_file)
+      .Append()
+      .Append('#include <string>')
+      .Append()
+      .Append('#include "%s.h"' % self._source_file_filename)
+      .Append()
+      .Append('#include "base/logging.h"')
+      .Append()
+      .Concat(cpp_util.OpenNamespace(self._namespace))
+      .Append()
+    )
+
+    # Generate the constructor.
+    (c.Append('%s::%s() {' % (self._class_name, self._class_name))
+      .Sblock()
+    )
+    for feature in self._feature_defs:
+      c.Append('features_["%s"] = %s;'
+                       % (feature.name, cpp_util.ConstantName(feature.name)))
+    (c.Eblock()
+      .Append('}')
+      .Append()
+    )
+
+    # Generate the ToString function.
+    (c.Append('const char* %s::ToString('
+                  '%s::ID id) const {' % (self._class_name, self._class_name))
+      .Sblock()
+      .Append('switch (id) {')
+      .Sblock()
+    )
+    for feature in self._feature_defs:
+      c.Append('case %s: return "%s";' %
+          (cpp_util.ConstantName(feature.name), feature.name))
+    (c.Append('case kUnknown: break;')
+      .Append('case kEnumBoundary: break;')
+      .Eblock()
+      .Append('}')
+      .Append('NOTREACHED();')
+      .Append('return "";')
+    )
+    (c.Eblock()
+      .Append('}')
+      .Append()
+    )
+
+    # Generate the FromString function.
+
+    (c.Append('%s::ID %s::FromString('
+                  'const std::string& id) const {'
+                  % (self._class_name, self._class_name))
+      .Sblock()
+      .Append('std::map<std::string, %s::ID>::const_iterator it'
+                  ' = features_.find(id);' % self._class_name)
+      .Append('if (it == features_.end())')
+      .Append('  return kUnknown;')
+      .Append('return it->second;')
+      .Eblock()
+      .Append('}')
+      .Append()
+      .Cblock(cpp_util.CloseNamespace(self._namespace))
+    )
+
+    return c