[mojo] Add Mojo bindings support for IPC::ParamTraits

mojo/public/tools/bindings/generators/mojom_cpp_generator.py

 # Copyright 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 mojom.generate.generator as generator
 import mojom.generate.module as mojom
 import mojom.generate.pack as pack
 from mojom.generate.template_expander import UseJinja
@@ skipping 24 matching lines @@
 }
 
 _kind_to_cpp_literal_suffix = {
   mojom.UINT8:        "U",
   mojom.UINT16:       "U",
   mojom.UINT32:       "U",
   mojom.FLOAT:        "f",
   mojom.UINT64:       "ULL",
 }
 
+
 _current_variant = None
 
 
+_variant_metadata_attributes = {
+  "chromium": {
+    "typename": "ChromiumType",
+    "headers": "ChromiumHeaders",
+  },
+  "blink": {
+    "typename": "BlinkType",
+    "headers": "BlinkHeaders",
+  }
+}
+
+
+_variant_typemap = {}
+
+
 def ConstantValue(constant):
   return ExpressionToText(constant.value, kind=constant.kind)
 
 def DefaultValue(field):
   if field.default:
     if mojom.IsStructKind(field.kind):
       assert field.default == "default"
       return "%s::New()" % GetNameForKind(field.kind)
     return ExpressionToText(field.default, kind=field.kind)
   return ""
 
+def IsNativeKind(kind):
+  if hasattr(kind, "name") and kind.name in _variant_typemap:
+    return True
+
+def GetNativeTypeName(kind):
+  return _variant_typemap[kind.name]
+
 def NamespaceToArray(namespace):
   return namespace.split(".") if namespace else []
 
 def GetNameForKind(kind, internal = False):
   parts = []
   if kind.imported_from:
     parts.extend(NamespaceToArray(kind.imported_from["namespace"]))
   if _current_variant:
     parts.append(_current_variant)
   if internal:
     parts.append("internal")
   if kind.parent_kind:
     parts.append(kind.parent_kind.name)
   parts.append(kind.name)
   return "::".join(parts)
 
+def GetQualifiedGenericInterfaceName(interface_kind):
+  parts = NamespaceToArray(interface_kind.module.namespace)
+  parts.append(interface_kind.name)
+  return "::".join(parts)
+
 def GetCppType(kind):
   if mojom.IsArrayKind(kind):
     return "mojo::internal::Array_Data<%s>*" % GetCppType(kind.kind)
   if mojom.IsMapKind(kind):
     return "mojo::internal::Map_Data<%s, %s>*" % (
       GetCppType(kind.key_kind), GetCppType(kind.value_kind))
   if mojom.IsStructKind(kind):
     return "%s_Data*" % GetNameForKind(kind, internal=True)
   if mojom.IsUnionKind(kind):
     return "%s_Data" % GetNameForKind(kind, internal=True)
@@ skipping 43 matching lines @@
     return "mojo::ScopedDataPipeConsumerHandle"
   if mojom.IsDataPipeProducerKind(kind):
     return "mojo::ScopedDataPipeProducerHandle"
   if mojom.IsMessagePipeKind(kind):
     return "mojo::ScopedMessagePipeHandle"
   if mojom.IsSharedBufferKind(kind):
     return "mojo::ScopedSharedBufferHandle"
   return _kind_to_cpp_type[kind]
 
 def GetCppResultWrapperType(kind):
+  if IsNativeKind(kind):
+    return "const %s&" % GetNativeTypeName(kind)
   if mojom.IsEnumKind(kind):
     return GetNameForKind(kind)
   if mojom.IsStructKind(kind) or mojom.IsUnionKind(kind):
     return "%sPtr" % GetNameForKind(kind)
   if mojom.IsArrayKind(kind):
     return "mojo::Array<%s>" % GetCppArrayArgWrapperType(kind.kind)
   if mojom.IsMapKind(kind):
     return "mojo::Map<%s, %s>" % (GetCppArrayArgWrapperType(kind.key_kind),
                                   GetCppArrayArgWrapperType(kind.value_kind))
   if mojom.IsInterfaceKind(kind):
@@ skipping 18 matching lines @@
     return "mojo::ScopedSharedBufferHandle"
   # TODO(rudominer) After improvements to compiler front end have landed,
   # revisit strategy used below for emitting a useful error message when an
   # undefined identifier is referenced.
   val =  _kind_to_cpp_type.get(kind)
   if (val is not None):
     return val
   raise Exception("Unrecognized kind %s" % kind.spec)
 
 def GetCppWrapperType(kind):
+  if IsNativeKind(kind):
+    return GetNativeTypeName(kind)
   if mojom.IsEnumKind(kind):
     return GetNameForKind(kind)
   if mojom.IsStructKind(kind) or mojom.IsUnionKind(kind):
     return "%sPtr" % GetNameForKind(kind)
   if mojom.IsArrayKind(kind):
     return "mojo::Array<%s>" % GetCppArrayArgWrapperType(kind.kind)
   if mojom.IsMapKind(kind):
     return "mojo::Map<%s, %s>" % (GetCppArrayArgWrapperType(kind.key_kind),
                                   GetCppArrayArgWrapperType(kind.value_kind))
   if mojom.IsInterfaceKind(kind):
@@ skipping 12 matching lines @@
     return "mojo::ScopedDataPipeConsumerHandle"
   if mojom.IsDataPipeProducerKind(kind):
     return "mojo::ScopedDataPipeProducerHandle"
   if mojom.IsMessagePipeKind(kind):
     return "mojo::ScopedMessagePipeHandle"
   if mojom.IsSharedBufferKind(kind):
     return "mojo::ScopedSharedBufferHandle"
   return _kind_to_cpp_type.get(kind)
 
 def GetCppConstWrapperType(kind):
+  if IsNativeKind(kind):
+    return "const %s&" % GetNativeTypeName(kind)
   if mojom.IsStructKind(kind) or mojom.IsUnionKind(kind):
     return "%sPtr" % GetNameForKind(kind)
   if mojom.IsArrayKind(kind):
     return "mojo::Array<%s>" % GetCppArrayArgWrapperType(kind.kind)
   if mojom.IsMapKind(kind):
     return "mojo::Map<%s, %s>" % (GetCppArrayArgWrapperType(kind.key_kind),
                                   GetCppArrayArgWrapperType(kind.value_kind))
   if mojom.IsInterfaceKind(kind):
     return "%sPtr" % GetNameForKind(kind)
   if mojom.IsInterfaceRequestKind(kind):
@@ skipping 14 matching lines @@
     return "mojo::ScopedDataPipeProducerHandle"
   if mojom.IsMessagePipeKind(kind):
     return "mojo::ScopedMessagePipeHandle"
   if mojom.IsSharedBufferKind(kind):
     return "mojo::ScopedSharedBufferHandle"
   if not kind in _kind_to_cpp_type:
     print "missing:", kind.spec
   return _kind_to_cpp_type[kind]
 
 def GetCppFieldType(kind):
+  if IsNativeKind(kind):
+    return "mojo::internal::NativePointer<%s>" % GetNativeTypeName(kind)
   if mojom.IsStructKind(kind):
     return ("mojo::internal::StructPointer<%s_Data>" %
         GetNameForKind(kind, internal=True))
   if mojom.IsUnionKind(kind):
     return "%s_Data" % GetNameForKind(kind, internal=True)
   if mojom.IsArrayKind(kind):
     return "mojo::internal::ArrayPointer<%s>" % GetCppType(kind.kind)
   if mojom.IsMapKind(kind):
     return ("mojo::internal::StructPointer<mojo::internal::Map_Data<%s, %s>>" %
             (GetCppType(kind.key_kind), GetCppType(kind.value_kind)))
@@ skipping 138 matching lines @@
     "cpp_result_type": GetCppResultWrapperType,
     "cpp_type": GetCppType,
     "cpp_union_getter_return_type": GetUnionGetterReturnType,
     "cpp_wrapper_type": GetCppWrapperType,
     "default_value": DefaultValue,
     "expression_to_text": ExpressionToText,
     "get_array_validate_params_ctor_args": GetArrayValidateParamsCtorArgs,
     "get_map_validate_params_ctor_args": GetMapValidateParamsCtorArgs,
     "get_name_for_kind": GetNameForKind,
     "get_pad": pack.GetPad,
+    "get_qualified_generic_interface_name": GetQualifiedGenericInterfaceName,
     "has_callbacks": mojom.HasCallbacks,
     "should_inline": ShouldInlineStruct,
     "should_inline_union": ShouldInlineUnion,
     "is_array_kind": mojom.IsArrayKind,
     "is_cloneable_kind": mojom.IsCloneableKind,
     "is_enum_kind": mojom.IsEnumKind,
     "is_integral_kind": mojom.IsIntegralKind,
     "is_move_only_kind": mojom.IsMoveOnlyKind,
+    "is_native_kind": IsNativeKind,
     "is_any_handle_kind": mojom.IsAnyHandleKind,
     "is_interface_kind": mojom.IsInterfaceKind,
     "is_interface_request_kind": mojom.IsInterfaceRequestKind,
     "is_associated_interface_kind": mojom.IsAssociatedInterfaceKind,
     "is_associated_interface_request_kind":
         mojom.IsAssociatedInterfaceRequestKind,
     "is_associated_kind": mojom.IsAssociatedKind,
     "is_map_kind": mojom.IsMapKind,
     "is_nullable_kind": mojom.IsNullableKind,
     "is_object_kind": mojom.IsObjectKind,
     "is_string_kind": mojom.IsStringKind,
     "is_struct_kind": mojom.IsStructKind,
     "is_union_kind": mojom.IsUnionKind,
     "passes_associated_kinds": mojom.PassesAssociatedKinds,
     "struct_size": lambda ps: ps.GetTotalSize() + _HEADER_SIZE,
     "stylize_method": generator.StudlyCapsToCamel,
     "to_all_caps": generator.CamelCaseToAllCaps,
     "under_to_camel": generator.UnderToCamel,
   }
 
   chromium_cpp_filters = dict(cpp_filters)
   blink_cpp_filters = dict(cpp_filters)
 
+  def GetExtraHeaders(self):
+    if not _current_variant:
+      return []
+    extra_headers = []
+    headers_attribute = \
+        _variant_metadata_attributes[_current_variant]["headers"]
+    for struct in self.GetStructs():
+      if struct.attributes:
+        headers = struct.attributes.get(headers_attribute, [])
+        extra_headers.extend(headers)
+    return extra_headers
+
   def GetJinjaExports(self):
+    global _variant_typemap
+    _variant_typemap = {}
+    if _current_variant:
+      typename_attribute = \
+          _variant_metadata_attributes[_current_variant]["typename"]
+      for struct in self.GetStructs():
+        if struct.attributes:
+          typename = struct.attributes.get(typename_attribute)
+          if typename:
+            _variant_typemap[struct.name] = typename
+
     return {
       "module": self.module,
       "namespace": self.module.namespace,
       "namespaces_as_array": NamespaceToArray(self.module.namespace),
       "imports": self.module.imports,
       "kinds": self.module.kinds,
       "enums": self.module.enums,
       "structs": self.GetStructs(),
       "unions": self.GetUnions(),
       "interfaces": self.GetInterfaces(),
       "variant": _current_variant,
+      "extra_headers": self.GetExtraHeaders(),
     }
 
   @UseJinja("cpp_templates/module.h.tmpl", filters=cpp_filters)
   def GenerateModuleHeader(self):
     return self.GetJinjaExports()
 
   @UseJinja("cpp_templates/module-internal.h.tmpl", filters=cpp_filters)
   def GenerateModuleInternalHeader(self):
     return self.GetJinjaExports()
 
@@ skipping 43 matching lines @@
     self.Write(self.GenerateChromiumModuleSource(),
         self.MatchMojomFilePath("%s-chromium.cc" % self.module.name))
 
     _current_variant = "blink"
     self.Write(self.GenerateBlinkModuleHeader(),
         self.MatchMojomFilePath("%s-blink.h" % self.module.name))
     self.Write(self.GenerateBlinkModuleInternalHeader(),
         self.MatchMojomFilePath("%s-blink-internal.h" % self.module.name))
     self.Write(self.GenerateBlinkModuleSource(),
         self.MatchMojomFilePath("%s-blink.cc" % self.module.name))