Retained size for instance in Observatory.

runtime/vm/object_graph.cc

 // Copyright (c) 2014, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 #include "vm/object_graph.h"
 
 #include "vm/dart.h"
 #include "vm/growable_array.h"
 #include "vm/isolate.h"
 #include "vm/object.h"
@@ skipping 128 matching lines @@
                                      ObjectGraph::Visitor* visitor) {
   NoGCScope no_gc_scope_;
   Stack stack(isolate_);
   RawObject* root_raw = root.raw();
   stack.VisitPointer(&root_raw);
   stack.TraverseGraph(visitor);
   // TODO(koda): Optimize if we only visited a small subgraph.
   Unmarker::UnmarkAll(isolate_);
 }
 
+
+class SizeVisitor : public ObjectGraph::Visitor {
+ public:
+  SizeVisitor() : size_(0) { }
+  intptr_t size() const { return size_; }
+  virtual bool ShouldSkip(RawObject* obj) const { return false; }
+  virtual Direction VisitObject(ObjectGraph::StackIterator* it) {
+    RawObject* obj = it->Get();
+    if (ShouldSkip(obj)) {
+      return kBacktrack;
+    }
+    size_ += obj->Size();
+    return kProceed;
+  }
+ private:
+  intptr_t size_;
+};
+
+
+class SizeExcludingObjectVisitor : public SizeVisitor {
+ public:
+  explicit SizeExcludingObjectVisitor(const Object& skip) : skip_(skip) { }
+  virtual bool ShouldSkip(RawObject* obj) const { return obj == skip_.raw(); }
+ private:
+  const Object& skip_;
+};
+
+
+class SizeExcludingClassVisitor : public SizeVisitor {
+ public:
+  explicit SizeExcludingClassVisitor(intptr_t skip) : skip_(skip) { }
+  virtual bool ShouldSkip(RawObject* obj) const {
+    return obj->GetClassId() == skip_;
+  }
+ private:
+  const intptr_t skip_;
+};
+
+
+intptr_t ObjectGraph::SizeRetainedByInstance(const Object& obj) {
+  SizeVisitor total;
+  IterateObjects(&total);
+  intptr_t size_total = total.size();
+  SizeExcludingObjectVisitor excluding_obj(obj);
+  IterateObjects(&excluding_obj);
+  intptr_t size_excluding_obj = excluding_obj.size();
+  return size_total - size_excluding_obj;
+}
+
+
+intptr_t ObjectGraph::SizeRetainedByClass(intptr_t class_id) {
+  SizeVisitor total;
+  IterateObjects(&total);
+  intptr_t size_total = total.size();
+  SizeExcludingClassVisitor excluding_class(class_id);
+  IterateObjects(&excluding_class);
+  intptr_t size_excluding_class = excluding_class.size();
+  return size_total - size_excluding_class;
+}
+
 }  // namespace dart