Add CVPixelBufferRef memory tracking

media/gpu/vt_video_decode_accelerator_mac.h

 // Copyright 2014 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 MEDIA_GPU_VT_VIDEO_DECODE_ACCELERATOR_MAC_H_
 #define MEDIA_GPU_VT_VIDEO_DECODE_ACCELERATOR_MAC_H_
 
 #include <stdint.h>
 
 #include <map>
 #include <memory>
 #include <queue>
 
 #include <VideoToolbox/VideoToolbox.h>
 
 #include "base/mac/scoped_cftyperef.h"
 #include "base/macros.h"
 #include "base/memory/linked_ptr.h"
 #include "base/memory/weak_ptr.h"
 #include "base/message_loop/message_loop.h"
 #include "base/threading/thread.h"
 #include "base/threading/thread_checker.h"
+#include "base/trace_event/memory_dump_provider.h"
 #include "media/filters/h264_parser.h"
 #include "media/gpu/gpu_video_decode_accelerator_helpers.h"
 #include "media/gpu/media_gpu_export.h"
 #include "media/video/h264_poc.h"
 #include "media/video/video_decode_accelerator.h"
 #include "ui/gfx/geometry/size.h"
 #include "ui/gl/gl_context_cgl.h"
 #include "ui/gl/gl_image_io_surface.h"
 
 namespace media {
 
 // Preload VideoToolbox libraries, needed for sandbox warmup.
 MEDIA_GPU_EXPORT bool InitializeVideoToolbox();
 
 // VideoToolbox.framework implementation of the VideoDecodeAccelerator
 // interface for Mac OS X (currently limited to 10.9+).
-class VTVideoDecodeAccelerator : public VideoDecodeAccelerator {
+class VTVideoDecodeAccelerator : public VideoDecodeAccelerator,
+                                 public base::trace_event::MemoryDumpProvider {
  public:
   explicit VTVideoDecodeAccelerator(
       const MakeGLContextCurrentCallback& make_context_current_cb,
       const BindGLImageCallback& bind_image_cb);
 
   ~VTVideoDecodeAccelerator() override;
 
   // VideoDecodeAccelerator implementation.
   bool Initialize(const Config& config, Client* client) override;
   void Decode(const BitstreamBuffer& bitstream) override;
   void AssignPictureBuffers(
       const std::vector<PictureBuffer>& pictures) override;
   void ReusePictureBuffer(int32_t picture_id) override;
   void Flush() override;
   void Reset() override;
   void Destroy() override;
   bool TryToSetupDecodeOnSeparateThread(
       const base::WeakPtr<Client>& decode_client,
       const scoped_refptr<base::SingleThreadTaskRunner>& decode_task_runner)
       override;
 
+  // MemoryDumpProvider implementation.
+  bool OnMemoryDump(const base::trace_event::MemoryDumpArgs& args,
+                    base::trace_event::ProcessMemoryDump* pmd) override;
+
   // Called by OutputThunk() when VideoToolbox finishes decoding a frame.
   void Output(void* source_frame_refcon,
               OSStatus status,
               CVImageBufferRef image_buffer);
 
   static VideoDecodeAccelerator::SupportedProfiles GetSupportedProfiles();
 
  private:
   // Logged to UMA, so never reuse values. Make sure to update
   // VTVDASessionFailureType in histograms.xml to match.
@@ skipping 190 matching lines @@
   // Last SPS and PPS the decoder was confgured with.
   std::vector<uint8_t> configured_sps_;
   std::vector<uint8_t> configured_spsext_;
   std::vector<uint8_t> configured_pps_;
   gfx::Size configured_size_;
 
   bool waiting_for_idr_ = true;
   bool missing_idr_logged_ = false;
   H264POC poc_;
 
+  // Id number for this instance for memory dumps.
+  int memory_dump_id_ = 0;
+
   //
   // Shared state (set up and torn down on GPU thread).
   //
   scoped_refptr<base::SingleThreadTaskRunner> gpu_task_runner_;
   base::WeakPtr<VTVideoDecodeAccelerator> weak_this_;
   base::Thread decoder_thread_;
 
   // Declared last to ensure that all weak pointers are invalidated before
   // other destructors run.
   base::WeakPtrFactory<VTVideoDecodeAccelerator> weak_this_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(VTVideoDecodeAccelerator);
 };
 
 }  // namespace media
 
 #endif  // MEDIA_GPU_VT_VIDEO_DECODE_ACCELERATOR_MAC_H_