1 files changed, 69 insertions, 0 deletions
diff --git a/ddms/libs/ddmuilib/src/com/android/ddmuilib/heap/NativeHeapProviderByLibrary.java b/ddms/libs/ddmuilib/src/com/android/ddmuilib/heap/NativeHeapProviderByLibrary.java
new file mode 100644
index 0000000..858b104
--- /dev/null
+++ b/ddms/libs/ddmuilib/src/com/android/ddmuilib/heap/NativeHeapProviderByLibrary.java
@@ -0,0 +1,69 @@
+/*
+ * Copyright (C) 2011 The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.android.ddmuilib.heap;
+
+import org.eclipse.jface.viewers.ILazyTreeContentProvider;
+import org.eclipse.jface.viewers.TreeViewer;
+import org.eclipse.jface.viewers.Viewer;
+
+/**
+ * Content Provider for the native heap tree viewer in {@link NativeHeapPanel}.
+ * It expects input of type {@link NativeHeapSnapshot}, and provides heap allocations
+ * grouped by library to the UI.
+ */
+public class NativeHeapProviderByLibrary implements ILazyTreeContentProvider {
+    private TreeViewer mViewer;
+
+    public NativeHeapProviderByLibrary(TreeViewer viewer) {
+        mViewer = viewer;
+    }
+
+    public void dispose() {
+    }
+
+    public void inputChanged(Viewer viewer, Object oldInput, Object newInput) {
+    }
+
+    public Object getParent(Object element) {
+        return null;
+    }
+
+    public void updateChildCount(Object element, int currentChildCount) {
+        int childCount = 0;
+
+        if (element instanceof NativeHeapSnapshot) {
+            childCount = ((NativeHeapSnapshot) element).getAllocationsByLibrary().size();
+        }
+
+        mViewer.setChildCount(element, childCount);
+    }
+
+    public void updateElement(Object parent, int index) {
+        Object item = null;
+        int childCount = 0;
+
+        if (parent instanceof NativeHeapSnapshot) { // root element
+            item = ((NativeHeapSnapshot) parent).getAllocationsByLibrary().get(index);
+            childCount = ((NativeLibraryAllocationInfo) item).getAllocations().size();
+        } else if (parent instanceof NativeLibraryAllocationInfo) {
+            item = ((NativeLibraryAllocationInfo) parent).getAllocations().get(index);
+        }
+
+        mViewer.replace(parent, index, item);
+        mViewer.setChildCount(item, childCount);
+    }
+}