Implement about:tracing UI for the sampling profiler (Chromium part)

content/browser/tracing/trace_controller_impl.h

 // Copyright (c) 2012 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.
 
 #ifndef CONTENT_BROWSER_TRACING_TRACE_CONTROLLER_IMPL_H_
 #define CONTENT_BROWSER_TRACING_TRACE_CONTROLLER_IMPL_H_
 
 #include <set>
 #include <string>
 #include <vector>
@@ skipping 12 matching lines @@
   static TraceControllerImpl* GetInstance();
 
   // Called on the main thread of the browser process to initialize
   // startup tracing.
   void InitStartupTracing(const CommandLine& command_line);
 
   // TraceController implementation:
   virtual bool BeginTracing(TraceSubscriber* subscriber,
                             const std::string& category_patterns,
                             base::debug::TraceLog::Options options) OVERRIDE;
-  virtual  bool EndTracingAsync(TraceSubscriber* subscriber) OVERRIDE;
+  virtual bool EndTracingAsync(TraceSubscriber* subscriber) OVERRIDE;
+  virtual bool BeginContinuousSamplingTracing(
+      TraceSubscriber* subscriber) OVERRIDE;
+  virtual bool EndContinuousSamplingTracing(
+      TraceSubscriber* subscriber) OVERRIDE;
+  virtual bool ShowContinuousSamplingTracingAsync(
+      TraceSubscriber* subscriber) OVERRIDE;
   virtual bool GetTraceBufferPercentFullAsync(
       TraceSubscriber* subscriber) OVERRIDE;
   virtual bool SetWatchEvent(TraceSubscriber* subscriber,
                              const std::string& category_name,
                              const std::string& event_name) OVERRIDE;
   virtual bool CancelWatchEvent(TraceSubscriber* subscriber) OVERRIDE;
   virtual void CancelSubscriber(TraceSubscriber* subscriber) OVERRIDE;
   virtual bool GetKnownCategoryGroupsAsync(TraceSubscriber* subscriber)
       OVERRIDE;
 
  private:
   typedef std::set<scoped_refptr<TraceMessageFilter> > FilterMap;
 
   friend struct base::DefaultLazyInstanceTraits<TraceControllerImpl>;
   friend class TraceMessageFilter;
 
   TraceControllerImpl();
   virtual ~TraceControllerImpl();
 
-  bool is_tracing_enabled() const {
-    return can_end_tracing();
+  bool can_begin_tracing(TraceSubscriber* subscriber) const {
+    return !is_tracing_ &&
+        (subscriber_ == NULL || subscriber == subscriber_);
   }
 
-  bool can_end_tracing() const {
+  bool can_end_tracing(TraceSubscriber* subscriber) const {
+    return is_tracing_ && pending_end_ack_count_ == 0 &&
+      subscriber == subscriber_;
+  }
+
+  bool is_tracing_enabled() const {
     return is_tracing_ && pending_end_ack_count_ == 0;
   }
 
+  bool can_begin_continuous_sampling_tracing(
+      TraceSubscriber* subscriber) const {
+    return !is_continuous_sampling_tracing_ &&
+        (continuous_sampling_subscriber_ == NULL ||
+         subscriber == continuous_sampling_subscriber_);
+  }
+
+  bool can_end_continuous_sampling_tracing(TraceSubscriber* subscriber) const {
+    return is_continuous_sampling_tracing_ &&
+        pending_show_continuous_sampling_ack_count_ == 0 &&
+        subscriber == continuous_sampling_subscriber_;
+  }
+
+  bool is_continuous_sampling_tracing_enabled() const {
+    return is_continuous_sampling_tracing_ &&
+        pending_show_continuous_sampling_ack_count_ == 0;
+  }
+
   // Can get Buffer Percent Full
   bool can_get_buffer_percent_full() const {
     return is_tracing_ &&
         pending_end_ack_count_ == 0 &&
         pending_bpf_ack_count_ == 0;
   }
 
-  bool can_begin_tracing(TraceSubscriber* subscriber) const {
-    return !is_tracing_ &&
-        (subscriber_ == NULL || subscriber == subscriber_);
-  }
-
   // Methods for use by TraceMessageFilter.
 
   void AddFilter(TraceMessageFilter* filter);
   void RemoveFilter(TraceMessageFilter* filter);
   void OnTracingBegan(TraceSubscriber* subscriber);
   void OnEndTracingAck(const std::vector<std::string>& known_category_groups);
+  void OnShowContinuousSamplingTracingAck();
   void OnTraceDataCollected(
       const scoped_refptr<base::RefCountedString>& events_str_ptr);
+  void OnContinuousSamplingTraceDataCollected(
+      const scoped_refptr<base::RefCountedString>& events_str_ptr);
   void OnTraceNotification(int notification);
   void OnTraceBufferPercentFullReply(float percent_full);
 
   FilterMap filters_;
   TraceSubscriber* subscriber_;
+  TraceSubscriber* continuous_sampling_subscriber_;
   // Pending acks for EndTracingAsync:
   int pending_end_ack_count_;
+  // Pending acks for ShowContinuousSamplingTracingAsync:
+  int pending_show_continuous_sampling_ack_count_;
   // Pending acks for GetTraceBufferPercentFullAsync:
   int pending_bpf_ack_count_;
   float maximum_bpf_;
   bool is_tracing_;
+  bool is_continuous_sampling_tracing_;
   bool is_get_category_groups_;
   std::set<std::string> known_category_groups_;
   std::string watch_category_;
   std::string watch_name_;
   base::debug::TraceLog::Options trace_options_;
   base::debug::CategoryFilter category_filter_;
 
   DISALLOW_COPY_AND_ASSIGN(TraceControllerImpl);
 };
 
 }  // namespace content
 
 #endif  // CONTENT_BROWSER_TRACING_TRACE_CONTROLLER_IMPL_H_