Expose allocation tracing over service protocol

runtime/vm/profiler_service.cc

 // Copyright (c) 2015, 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/profiler_service.h"
 
 #include "vm/growable_array.h"
 #include "vm/native_symbol.h"
 #include "vm/object.h"
 #include "vm/os.h"
@@ skipping 1946 matching lines @@
   }
   current_index++;
   if (current_index >= parent_->NumChildren()) {
     return false;
   }
   current_ = parent_->At(current_index);
   return true;
 }
 
 
-class NoAllocationSampleFilter : public SampleFilter {
- public:
-  explicit NoAllocationSampleFilter(Isolate* isolate)
-      : SampleFilter(isolate) {
-  }
-
-  bool FilterSample(Sample* sample) {
-    return !sample->is_allocation_sample();
-  }
-};
-
-
-void ProfilerService::PrintJSON(JSONStream* stream,
-                                Profile::TagOrder tag_order) {
-  Isolate* isolate = Isolate::Current();
+void ProfilerService::PrintJSONImpl(Isolate* isolate,
+                                    JSONStream* stream,
+                                    Profile::TagOrder tag_order,
+                                    SampleFilter* filter) {
   // Disable profile interrupts while processing the buffer.
   Profiler::EndExecution(isolate);
 
   {
     MutexLocker profiler_data_lock(isolate->profiler_data_mutex());
     IsolateProfilerData* profiler_data = isolate->profiler_data();
     if (profiler_data == NULL) {
       stream->PrintError(kFeatureDisabled, NULL);
       return;
     }
   }
 
   {
     StackZone zone(isolate);
     HANDLESCOPE(isolate);
     Profile profile(isolate);
-    NoAllocationSampleFilter filter(isolate);
-    profile.Build(&filter, tag_order);
+    profile.Build(filter, tag_order);
     profile.PrintJSON(stream);
   }
 
   // Enable profile interrupts.
   Profiler::BeginExecution(isolate);
 }
 
 
+class NoAllocationSampleFilter : public SampleFilter {
+ public:
+  explicit NoAllocationSampleFilter(Isolate* isolate)
+      : SampleFilter(isolate) {
+  }
+
+  bool FilterSample(Sample* sample) {
+    return !sample->is_allocation_sample();
+  }
+};
+
+
+void ProfilerService::PrintJSON(JSONStream* stream,
+                                Profile::TagOrder tag_order) {
+  Isolate* isolate = Isolate::Current();
+  NoAllocationSampleFilter filter(isolate);
+  PrintJSONImpl(isolate, stream, tag_order, &filter);
+}
+
+
+class ClassAllocationSampleFilter : public SampleFilter {
+ public:
+  ClassAllocationSampleFilter(Isolate* isolate, const Class& cls)
+      : SampleFilter(isolate),
+        cls_(Class::Handle(cls.raw())) {
+    ASSERT(!cls_.IsNull());
+  }
+
+  bool FilterSample(Sample* sample) {
+    return sample->is_allocation_sample() &&
+           (sample->allocation_cid() == cls_.id());
+  }
+
+ private:
+  const Class& cls_;
+};
+
+
+void ProfilerService::PrintAllocationJSON(JSONStream* stream,
+                                          Profile::TagOrder tag_order,
+                                          const Class& cls) {
+  Isolate* isolate = Isolate::Current();
+  ClassAllocationSampleFilter filter(isolate, cls);
+  PrintJSONImpl(isolate, stream, tag_order, &filter);
+}
+
+
 void ProfilerService::ClearSamples() {
   Isolate* isolate = Isolate::Current();
 
   // Disable profile interrupts while processing the buffer.
   Profiler::EndExecution(isolate);
 
   MutexLocker profiler_data_lock(isolate->profiler_data_mutex());
   IsolateProfilerData* profiler_data = isolate->profiler_data();
   if (profiler_data == NULL) {
     return;
   }
   SampleBuffer* sample_buffer = profiler_data->sample_buffer();
   ASSERT(sample_buffer != NULL);
 
   ClearProfileVisitor clear_profile(isolate);
   sample_buffer->VisitSamples(&clear_profile);
 
   // Enable profile interrupts.
   Profiler::BeginExecution(isolate);
 }
 
 }  // namespace dart