[tracing] Implement composable memory usage estimators.

ui/android/resources/resource_manager_impl.cc

 // Copyright 2014 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.
 
 #include "ui/android/resources/resource_manager_impl.h"
 
 #include <inttypes.h>
 #include <stddef.h>
 
 #include <utility>
 #include <vector>
 
 #include "base/android/jni_array.h"
 #include "base/android/jni_string.h"
 #include "base/memory/ptr_util.h"
 #include "base/strings/stringprintf.h"
 #include "base/threading/thread_task_runner_handle.h"
+#include "base/trace_event/estimate_memory_usage.h"
 #include "base/trace_event/memory_dump_manager.h"
 #include "base/trace_event/process_memory_dump.h"
 #include "base/trace_event/trace_event.h"
 #include "cc/resources/scoped_ui_resource.h"
 #include "cc/resources/ui_resource_manager.h"
 #include "jni/ResourceManager_jni.h"
 #include "third_party/skia/include/core/SkBitmap.h"
 #include "third_party/skia/include/core/SkCanvas.h"
 #include "third_party/skia/include/core/SkColorFilter.h"
 #include "ui/android/resources/ui_resource_provider.h"
@@ skipping 266 matching lines @@
           sprite_rect_source, sprite_rect_destination));
     }
     src_dst_rects.push_back(frame_src_dst_rects);
   }
   return src_dst_rects;
 }
 
 bool ResourceManagerImpl::OnMemoryDump(
     const base::trace_event::MemoryDumpArgs& args,
     base::trace_event::ProcessMemoryDump* pmd) {
-  size_t size = 0;
-  for (const auto& resource_map : resources_) {
-    for (const auto& id_and_resource : resource_map) {
-      if (id_and_resource.second && id_and_resource.second->ui_resource)
-        size += id_and_resource.second->ui_resource->GetAllocatedSizeInBytes();
-    }
-  }
-  for (const auto& id_and_resource : crushed_sprite_resources_) {
-    if (id_and_resource.second)
-      size += id_and_resource.second->GetAllocatedSizeInBytes();
-  }
-  for (const auto& color_and_resources : tinted_resources_) {
-    for (const auto& id_and_resource : *color_and_resources.second) {
-      if (id_and_resource.second && id_and_resource.second->ui_resource)
-        size += id_and_resource.second->ui_resource->GetAllocatedSizeInBytes();
-    }
-  }
+  using base::trace_event::EstimateMemoryUsage;
+  size_t memory_usage = EstimateMemoryUsage(resources_) +
+                        EstimateMemoryUsage(crushed_sprite_resources_) +
+                        EstimateMemoryUsage(tinted_resources_);
 
   base::trace_event::MemoryAllocatorDump* dump = pmd->CreateAllocatorDump(
       base::StringPrintf("ui/resource_manager_0x%" PRIXPTR,
                          reinterpret_cast<uintptr_t>(this)));
   dump->AddScalar(base::trace_event::MemoryAllocatorDump::kNameSize,
-                  base::trace_event::MemoryAllocatorDump::kUnitsBytes, size);
+                  base::trace_event::MemoryAllocatorDump::kUnitsBytes,
+                  memory_usage);
 
-  // Bitmaps are allocated from malloc.
   const char* system_allocator_name =
       base::trace_event::MemoryDumpManager::GetInstance()
           ->system_allocator_pool_name();
   if (system_allocator_name) {
     pmd->AddSuballocation(dump->guid(), system_allocator_name);
   }
 
   return true;
 }
 
@@ skipping 41 matching lines @@
   TRACE_EVENT2("ui",
                "ResourceManagerImpl::RequestCrushedSpriteResourceFromJava",
                "bitmap_res_id", bitmap_res_id,
                "metadata_res_id", metadata_res_id);
   Java_ResourceManager_crushedSpriteResourceRequested(
       base::android::AttachCurrentThread(), java_obj_, bitmap_res_id,
       metadata_res_id, reloading);
 }
 
 }  // namespace ui