mojo: break struct headers into headers and source

mojo/public/bindings/generators/mojom_cpp_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.
 
+"""Generates C++ source files from a mojom.Module."""
+
 import datetime
 import mojom
 import mojom_pack
 import os
 import sys
 
 from string import Template
 
 # mojom_cpp_generator provides a way to generate c++ code from a mojom.Module.
 # cpp = mojom_cpp_generator.CPPGenerator(module)
 # cpp.GenerateFiles("/tmp/g")
 def ReadTemplate(filename):
+  """Reads a string.Template from |filename|."""
   dir = os.path.dirname(__file__)
   with open(dir + '/' + filename, 'r') as file:
     return Template(file.read())
 
 
+def AddForward(forwards, kind):
+  """Adds a forward class declaration line to the |forwards| set."""
+  if isinstance(kind, mojom.Struct):
+    forwards.add("class %s;" % kind.name.capitalize())
+  if isinstance(kind, mojom.Array):
+    AddForward(forwards, kind.kind)
+
+
+def GetForwards(forwards):
+  """Returns a string suitable for substituting into a $FORWARDS section."""
+  if len(forwards) > 0:
+    return '\n'.join(sorted(forwards)) + '\n'
+  return ""
+
+
 class CPPGenerator(object):
-  struct_template = ReadTemplate("cpp_struct.template")
-  interface_template = ReadTemplate("cpp_interface.template")
+
+  struct_header_template = ReadTemplate("module_struct.h")
+  struct_source_template = ReadTemplate("module_struct.cc")
+  interface_header_template = ReadTemplate("module_interface.h")
   field_template = Template("  $itype ${field}_;")
   bool_field_template = Template("  uint8_t ${field}_ : 1;")
   setter_template = \
       Template("  void set_$field($stype $field) { ${field}_ = $field; }")
   getter_template = \
       Template("  $gtype $field() const { return ${field}_; }")
   ptr_getter_template = \
       Template("  $gtype $field() const { return ${field}_.ptr; }")
   pad_template = Template("  uint8_t _pad${count}_[$pad];")
+  HEADER_SIZE = 8
 
   kind_to_type = {
     mojom.BOOL:   "bool",
     mojom.INT8:   "int8_t",
     mojom.UINT8:  "uint8_t",
     mojom.INT16:  "int16_t",
     mojom.UINT16: "uint16_t",
     mojom.INT32:  "int32_t",
     mojom.UINT32: "uint32_t",
     mojom.FLOAT:  "float",
@@ skipping 52 matching lines @@
     elif kind.spec == 's':
       itype = "mojo::internal::StringPointer"
     else:
       itype = cls.kind_to_type[kind]
     return cls.field_template.substitute(field=pf.field.name, itype=itype)
 
   def GetHeaderGuard(self, component):
     return "MOJO_GENERATED_BINDINGS_%s_%s_H_" % \
         (self.module.name.upper(), component.name.upper())
 
-  def OpenComponentFile(self, directory, component):
-    filename = "%s_%s.h" % \
-        (self.module.name.lower(), component.name.lower())
-    return open(directory + '/' + filename, "w+")
+  # Pass |output_dir| to emit files to disk. Omit |output_dir| to echo all files
+  # to stdout.
+  def __init__(self, module, header_dir, output_dir=None):
+    self.module = module
+    self.header_dir = header_dir
+    self.output_dir = output_dir
 
-  def __init__(self, module):
-    self.module = module
-    self.file = file
+  def WriteTemplateToFile(self, template, name, ext, **substitutions):
+    if self.output_dir is None:
+      file = sys.stdout
+    else:
+      filename = "%s_%s.%s" % \
+          (self.module.name.lower(), name.lower(), ext)
+      file = open(os.path.join(self.output_dir, filename), "w+")
+    try:
+      file.write(template.substitute(substitutions))
+    finally:
+      if self.output_dir is not None:
+        file.close()
 
-  def GenerateStruct(self, struct, file):
+  def GenerateStructHeader(self, struct):
     fields = []
     setters = []
     getters = []
+    forwards = set()
 
     ps = mojom_pack.PackedStruct(struct)
     pad_count = 0
     num_fields = len(ps.packed_fields)
     for i in xrange(num_fields):
       pf = ps.packed_fields[i]
       fields.append(self.GetFieldLine(pf))
+      AddForward(forwards, pf.field.kind)
       if i < (num_fields - 2):
         next_pf = ps.packed_fields[i+1]
         pad = next_pf.offset - (pf.offset + pf.size)
         if pad > 0:
           fields.append(self.pad_template.substitute(count=pad_count, pad=pad))
           pad_count += 1
       setters.append(self.GetSetterLine(pf))
       getters.append(self.GetGetterLine(pf))
 
     if num_fields > 0:
       last_field = ps.packed_fields[num_fields - 1]
       offset = last_field.offset + last_field.size
       pad = mojom_pack.GetPad(offset, 8)
       if pad > 0:
         fields.append(self.pad_template.substitute(count=pad_count, pad=pad))
         pad_count += 1
       size = offset + pad
     else:
       size = 0
 
-    file.write(self.struct_template.substitute(
-      year = datetime.date.today().year,
-      header_guard = self.GetHeaderGuard(struct),
-      namespace = self.module.namespace,
-      classname = struct.name.capitalize(),
-      num_fields = len(ps.packed_fields),
-      size = size + 8,
-      setters = '\n'.join(setters),
-      getters = '\n'.join(getters),
-      fields = '\n'.join(fields)))
+    self.WriteTemplateToFile(self.struct_header_template, struct.name, 'h',
+        YEAR = datetime.date.today().year,
+        HEADER_GUARD = self.GetHeaderGuard(struct),
+        NAMESPACE = self.module.namespace,
+        CLASS = struct.name.capitalize(),
+        SIZE = size + self.HEADER_SIZE,
+        FORWARDS = GetForwards(forwards),
+        SETTERS = '\n'.join(setters),
+        GETTERS = '\n'.join(getters),
+        FIELDS = '\n'.join(fields))
 
-  def GenerateInterface(self, interface, file):
+  def GenerateStructSource(self, struct):
+    header = "%s_%s.h" % (self.module.name.lower(), struct.name.lower())
+    header = os.path.join(self.header_dir, header)
+    self.WriteTemplateToFile(self.struct_source_template, struct.name, 'cc',
+        YEAR = datetime.date.today().year,
+        NAMESPACE = self.module.namespace,
+        CLASS = struct.name.capitalize(),
+        NUM_FIELDS = len(struct.fields),
+        HEADER = header)
+
+  def GenerateInterfaceHeader(self, interface):
     cpp_methods = []
+    forwards = set()
     for method in interface.methods:
       cpp_method = "  virtual void %s(" % method.name
       first_param = True
       for param in method.parameters:
         if first_param == True:
           first_param = False
         else:
           cpp_method += ", "
+        AddForward(forwards, param.kind)
         cpp_method += "%s %s" % (self.GetConstType(param.kind), param.name)
       cpp_method += ") = 0;"
       cpp_methods.append(cpp_method)
 
-    file.write(self.interface_template.substitute(
-      year = datetime.date.today().year,
-      header_guard = self.GetHeaderGuard(interface),
-      namespace = self.module.namespace,
-      classname = interface.name.capitalize(),
-      methods = '\n'.join(cpp_methods)))
+    self.WriteTemplateToFile(
+        self.interface_header_template,
+        interface.name,
+        'h',
+        YEAR = datetime.date.today().year,
+        HEADER_GUARD = self.GetHeaderGuard(interface),
+        NAMESPACE = self.module.namespace,
+        CLASS = interface.name.capitalize(),
+        FORWARDS = GetForwards(forwards),
+        METHODS = '\n'.join(cpp_methods))
 
-  # Pass |directory| to emit files to disk. Omit |directory| to echo all files
-  # to stdout.
-  def GenerateFiles(self, directory=None):
+  def GenerateFiles(self):
     for struct in self.module.structs:
-      if directory is None:
-        self.GenerateStruct(struct, sys.stdout)
-      else:
-        with self.OpenComponentFile(directory, struct) as file:
-          self.GenerateStruct(struct, file)
+      self.GenerateStructHeader(struct)
+      self.GenerateStructSource(struct)
     for interface in self.module.interfaces:
-      if directory is None:
-        self.GenerateInterface(interface, sys.stdout)
-      else:
-        with self.OpenComponentFile(directory, interface) as file:
-          self.GenerateInterface(interface, file)
+      self.GenerateInterfaceHeader(interface)