Sampling heap profiler data structure changes

src/profiler/sampling-heap-profiler.h

 // Copyright 2015 the V8 project authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef V8_PROFILER_SAMPLING_HEAP_PROFILER_H_
 #define V8_PROFILER_SAMPLING_HEAP_PROFILER_H_
 
 #include <deque>
 #include <map>
 #include <set>
@@ skipping 30 matching lines @@
 class SamplingHeapProfiler {
  public:
   SamplingHeapProfiler(Heap* heap, StringsStorage* names, uint64_t rate,
                        int stack_depth);
   ~SamplingHeapProfiler();
 
   v8::AllocationProfile* GetAllocationProfile();
 
   StringsStorage* names() const { return names_; }
 
-  class FunctionInfo {
+  class AllocationNode;
+
+  struct Sample {
+    size_t size;
+    AllocationNode* owner;
+    Global<Value> global;
+    SamplingHeapProfiler* profiler;

[ulan, 2016/02/17 11:14:50]

Disallow copy and assign.

[mattloring, 2016/02/17 17:50:21]

Done.

+  };
+
+  class AllocationNode {
    public:
-    FunctionInfo(SharedFunctionInfo* shared, StringsStorage* names);
-    explicit FunctionInfo(const char* name)
-        : name_(name),
-          script_name_(""),
-          script_id_(v8::UnboundScript::kNoScriptId),
-          start_position_(0) {}
-
-    const char* get_name() const { return name_; }
-    const char* get_script_name() const { return script_name_; }
-    int get_script_id() const { return script_id_; }
-    int get_start_position() const { return start_position_; }
+    explicit AllocationNode(const char* const name, int script_id,

[ulan, 2016/02/17 11:14:50]

No need for explicit here.

[mattloring, 2016/02/17 17:50:21]

Done.

+                            const int start_position)
+        : script_id_(script_id),
+          script_position_(start_position),
+          name_(name) {}
+    ~AllocationNode() {
+      for (auto child : children_) {
+        delete child;
+      }
+    }
 
    private:
-    const char* const name_;
-    const char* script_name_;
+    std::map<size_t, unsigned int> allocations_;
+    std::vector<AllocationNode*> children_;
     int script_id_;
-    const int start_position_;
-  };
+    int script_position_;
+    const char* name_;
 
-  class SampledAllocation {
-   public:
-    SampledAllocation(SamplingHeapProfiler* sampling_heap_profiler,
-                      Isolate* isolate, Local<Value> local, size_t size,
-                      int max_frames);
-    ~SampledAllocation() {
-      for (auto info : stack_) {
-        delete info;
-      }
-      global_.Reset();  // drop the reference.
-    }
-    size_t get_size() const { return size_; }
-    const std::vector<FunctionInfo*>& get_stack() const { return stack_; }
+    friend class SamplingHeapProfiler;
 
-   private:
-    static void OnWeakCallback(const WeakCallbackInfo<SampledAllocation>& data);
-
-    SamplingHeapProfiler* const sampling_heap_profiler_;
-    Global<Value> global_;
-    std::vector<FunctionInfo*> stack_;
-    const size_t size_;
-
-    DISALLOW_COPY_AND_ASSIGN(SampledAllocation);
+    DISALLOW_COPY_AND_ASSIGN(AllocationNode);
   };
 
  private:
   Heap* heap() const { return heap_; }
 
   void SampleObject(Address soon_object, size_t size);
 
+  static void OnWeakCallback(const WeakCallbackInfo<Sample>& data);
+
   // Methods that construct v8::AllocationProfile.
-  v8::AllocationProfile::Node* AddStack(
-      AllocationProfile* profile, const std::map<int, Script*>& scripts,
-      const std::vector<FunctionInfo*>& stack);
-  v8::AllocationProfile::Node* FindOrAddChildNode(
-      AllocationProfile* profile, const std::map<int, Script*>& scripts,
-      v8::AllocationProfile::Node* parent, FunctionInfo* function_info);
-  v8::AllocationProfile::Node* AllocateNode(
-      AllocationProfile* profile, const std::map<int, Script*>& scripts,
-      FunctionInfo* function_info);
+  v8::AllocationProfile::Node* GenerateProfile(

[ulan, 2016/02/17 11:14:50]

Could you please add a comment describing pre/post conditions of this function?
i.e. what is node, what is expected to be in scripts, what is returned.

[mattloring, 2016/02/17 17:50:21]

Done.

+      AllocationProfile* profile, SamplingHeapProfiler::AllocationNode* node,
+      const std::map<int, Script*>& scripts);
+  AllocationNode* AddStack();
+  AllocationNode* FindOrAddChildNode(AllocationNode* parent, const char* name,
+                                     int script_id, int start_position);
 
   Isolate* const isolate_;
   Heap* const heap_;
   base::SmartPointer<SamplingAllocationObserver> new_space_observer_;
   base::SmartPointer<SamplingAllocationObserver> other_spaces_observer_;
   StringsStorage* const names_;
-  std::set<SampledAllocation*> samples_;
+  AllocationNode* profile_root_;
+  std::set<Sample*> samples_;
   const int stack_depth_;
 
   friend class SamplingAllocationObserver;
 };
 
 class SamplingAllocationObserver : public AllocationObserver {
  public:
   SamplingAllocationObserver(Heap* heap, intptr_t step_size, uint64_t rate,
                              SamplingHeapProfiler* profiler,
                              base::RandomNumberGenerator* random)
@@ skipping 19 matching lines @@
   SamplingHeapProfiler* const profiler_;
   Heap* const heap_;
   base::RandomNumberGenerator* const random_;
   uint64_t const rate_;
 };
 
 }  // namespace internal
 }  // namespace v8
 
 #endif  // V8_PROFILER_SAMPLING_HEAP_PROFILER_H_