Added UMA stats for extensions events on addListener.

tools/json_schema_compiler/cpp_bundle_generator.py

 # Copyright (c) 2012 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 code
 import cpp_util
 from model import Platforms
 from schema_util import CapitalizeFirstLetter
+from schema_util import JsEventNameToHistogramValue
 from schema_util import JsFunctionNameToClassName
 
 import json
 import os
 import re
 
 # TODO(miket/asargent) - parameterize this.
 SOURCE_BASE_PATH = 'chrome/common/extensions/api'
 
 def _RemoveDescriptions(node):
@@ skipping 70 matching lines @@
       c.Append("#if %s" % function_ifdefs, indent_level=0)
 
     function_name = JsFunctionNameToClassName(namespace_name, function.name)
     c.Append("registry->RegisterFunction<%sFunction>();" % (
         function_name))
 
     if function_ifdefs is not None:
       c.Append("#endif  // %s" % function_ifdefs, indent_level=0)
     return c
 
+  def _GenerateEventHistogramMapAddAll(self):
+    c = code.Code()
+    c.Append('// static')
+    c.Append('void GeneratedEventHistogramMap::AddAll(std::map<std::string,')
+    c.Sblock('    extensions::events::HistogramValue>& map) {')
+    for namespace in self._model.namespaces.values():
+      for event in namespace.events.values():
+        if event.nocompile:
+          continue
+        event_name = "%s.%s" % (namespace.name, event.name)
+        histogram_value = event.histogram_name
+        if not histogram_value:
+          histogram_value = JsEventNameToHistogramValue(namespace.name,
+                                                      event.name)
+        c.Append('map.insert(std::make_pair("%s", extensions::events::%s));' %
+            (event_name, histogram_value))
+
+    c.Eblock("}")
+    return c
+
   def _GenerateFunctionRegistryRegisterAll(self):
     c = code.Code()
     c.Append('// static')
     c.Sblock('void GeneratedFunctionRegistry::RegisterAll('
                  'ExtensionFunctionRegistry* registry) {')
     for namespace in self._model.namespaces.values():
       namespace_ifdefs = self._GetPlatformIfdefs(namespace)
       if namespace_ifdefs is not None:
         c.Append("#if %s" % namespace_ifdefs, indent_level=0)
 
@@ skipping 19 matching lines @@
     return c
 
 class _APIHGenerator(object):
   """Generates the header for API registration / declaration"""
   def __init__(self, cpp_bundle):
     self._bundle = cpp_bundle
 
   def Generate(self, namespace):
     c = code.Code()
 
+    c.Append('#include <map>')
     c.Append('#include <string>')
     c.Append()
     c.Append('#include "base/basictypes.h"')
+
+    c.Append('#include '
+        '"chrome/browser/extensions/extension_event_histogram_value.h"')
+    c.Append('#include '
+        '"chrome/browser/extensions/extension_function_registry.h"')
     c.Append()
     c.Append("class ExtensionFunctionRegistry;")
     c.Append()
     c.Concat(cpp_util.OpenNamespace(self._bundle._cpp_namespace))
     c.Append()
     c.Append('class GeneratedFunctionRegistry {')
     c.Sblock(' public:')
     c.Append('static void RegisterAll('
                  'ExtensionFunctionRegistry* registry);')
     c.Eblock('};');
     c.Append()
+    c.Append('class GeneratedEventHistogramMap {')
+    c.Sblock(' public:')
+    c.Append('static void AddAll(std::map<std::string,')
+    c.Append('    extensions::events::HistogramValue>& map);')
+    c.Eblock('};');
+    c.Append()
     c.Concat(cpp_util.CloseNamespace(self._bundle._cpp_namespace))
     return self._bundle._GenerateHeader('generated_api', c)
 
 class _APICCGenerator(object):
   """Generates a code.Code object for the generated API .cc file"""
 
   def __init__(self, cpp_bundle):
     self._bundle = cpp_bundle
 
   def Generate(self, namespace):
@@ skipping 26 matching lines @@
       if ifdefs is not None:
         c.Append("#endif  // %s" % ifdefs, indent_level=0)
     c.Append()
     c.Append('#include '
                  '"chrome/browser/extensions/extension_function_registry.h"')
     c.Append()
     c.Concat(cpp_util.OpenNamespace(self._bundle._cpp_namespace))
     c.Append()
     c.Concat(self._bundle._GenerateFunctionRegistryRegisterAll())
     c.Append()
+    c.Concat(self._bundle._GenerateEventHistogramMapAddAll())
+    c.Append()
     c.Concat(cpp_util.CloseNamespace(self._bundle._cpp_namespace))
     c.Append()
     return c
 
 class _SchemasHGenerator(object):
   """Generates a code.Code object for the generated schemas .h file"""
   def __init__(self, cpp_bundle):
     self._bundle = cpp_bundle
 
   def Generate(self, namespace):
@@ skipping 40 matching lines @@
                                 separators=(',', ':'))
       # Escape all double-quotes and backslashes. For this to output a valid
       # JSON C string, we need to escape \ and ".
       json_content = json_content.replace('\\', '\\\\').replace('"', '\\"')
       c.Append('(*schemas)["%s"] = "%s";' % (namespace.name, json_content))
     c.Eblock('}')
     c.Append()
     c.Concat(cpp_util.CloseNamespace(self._bundle._cpp_namespace))
     c.Append()
     return c