Initial code generation for features.

tools/json_schema_compiler/features_h_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 HGenerator(object):
+  def Generate(self, features, source_file, namespace):
+    return _Generator(features, source_file, namespace).Generate()
+
+
+class _Generator(object):
+  """A .cc generator for features.
+  """
+  def __init__(self, features, source_file, namespace):
+    self._feature_defs = features
+    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()
+    )
+    ifndef_name = cpp_util.GenerateIfndefName(self._source_file_filename,
+                                              self._class_name)
+    (c.Append('#ifndef %s' % ifndef_name)
+      .Append('#define %s' % ifndef_name)
+      .Append()
+    )
+
+    (c.Append('#include <map>')
+      .Append('#include <string>')
+      .Append()
+      .Concat(cpp_util.OpenNamespace(self._namespace))
+      .Append()
+    )
+
+    (c.Append('class %s {' % self._class_name)
+      .Append(' public:')
+      .Sblock()
+      .Concat(self._GeneratePublicBody())
+      .Eblock()
+      .Append(' private:')
+      .Sblock()
+      .Concat(self._GeneratePrivateBody())
+      .Eblock('};')
+      .Append()
+      .Cblock(cpp_util.CloseNamespace(self._namespace))
+    )
+    (c.Append('#endif  // %s' % ifndef_name)
+      .Append()
+    )
+    return c
+
+  def _GeneratePublicBody(self):
+    c = Code()
+
+    (c.Append('%s();' % self._class_name)
+      .Append()
+      .Append('enum ID {')
+      .Concat(self._GenerateEnumConstants())
+      .Eblock('};')
+      .Append()
+      .Append('const char* ToString(ID id) const;')
+      .Append('ID FromString(const std::string& id) const;')
+      .Append()
+    )
+    return c
+
+  def _GeneratePrivateBody(self):
+    return Code().Append('std::map<std::string, '
+                         '%s::ID> features_;' % self._class_name)
+
+  def _GenerateEnumConstants(self):
+    c = Code()
+
+    (c.Sblock()
+      .Append('kUnknown,')
+    )
+    for feature in self._feature_defs:
+      c.Append('%s,' % cpp_util.ConstantName(feature.name))
+    c.Append('kEnumBoundary')
+
+    return c