[tracing] Add memory usage of Android UI resources to memory-infra

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/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"
 #include "ui/android/window_android.h"
 #include "ui/gfx/android/java_bitmap.h"
@@ skipping 12 matching lines @@
 }
 
 ResourceManagerImpl::ResourceManagerImpl(gfx::NativeWindow native_window)
     : ui_resource_manager_(nullptr) {
   JNIEnv* env = base::android::AttachCurrentThread();
   java_obj_.Reset(
       env, Java_ResourceManager_create(env, native_window->GetJavaObject(),
                                        reinterpret_cast<intptr_t>(this))
                .obj());
   DCHECK(!java_obj_.is_null());
+  base::trace_event::MemoryDumpManager::GetInstance()->RegisterDumpProvider(
+      this, "android::ResourceManagerImpl",
+      base::ThreadTaskRunnerHandle::Get());
 }
 
 ResourceManagerImpl::~ResourceManagerImpl() {
+  base::trace_event::MemoryDumpManager::GetInstance()->UnregisterDumpProvider(
+      this);
   Java_ResourceManager_destroy(base::android::AttachCurrentThread(), java_obj_);
 }
 
 void ResourceManagerImpl::Init(cc::UIResourceManager* ui_resource_manager) {
   DCHECK(!ui_resource_manager_);
   DCHECK(ui_resource_manager);
   ui_resource_manager_ = ui_resource_manager;
 }
 
 base::android::ScopedJavaLocalRef<jobject>
@@ skipping 220 matching lines @@
                                    frame_ints[j+4],
                                    frame_ints[j+5]);
       frame_src_dst_rects.push_back(std::pair<gfx::Rect, gfx::Rect>(
           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();
+    }
+  }
+
+  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);
+
+  // 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;
+}
+
 void ResourceManagerImpl::OnCrushedSpriteResourceReloaded(
     JNIEnv* env,
     const JavaRef<jobject>& jobj,
     jint bitmap_res_id,
     const JavaRef<jobject>& bitmap) {
   std::unordered_map<int, std::unique_ptr<CrushedSpriteResource>>::iterator
       item = crushed_sprite_resources_.find(bitmap_res_id);
   if (item == crushed_sprite_resources_.end()) {
     // Cannot reload a resource that has not been previously loaded.
     return;
@@ skipping 31 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