Ability for demuxer clients to get to FFmpeg's AVStream object exposed as an ...

media/base/mock_media_filters.h

 // Copyright (c) 2009 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_BASE_MOCK_MEDIA_FILTERS_H_
 #define MEDIA_BASE_MOCK_MEDIA_FILTERS_H_
 
 #include <string>
 
 #include "base/waitable_event.h"
@@ skipping 177 matching lines @@
 
   // Set to true inside the destructor.  Used in FFmpegGlue unit tests for
   // testing proper reference counting.
   bool* deleted_;
 
   DISALLOW_COPY_AND_ASSIGN(MockDataSource);
 };
 
 //------------------------------------------------------------------------------
 
+class MockDemuxerStream : public DemuxerStream {
+ public:
+  MockDemuxerStream(const MockFilterConfig* config, bool is_audio) {
+    if (is_audio) {
+      media_format_.SetAsString(MediaFormat::kMimeType,
+                                config->compressed_audio_mime_type);
+    } else {
+      media_format_.SetAsString(MediaFormat::kMimeType,
+                                config->compressed_video_mime_type);
+      media_format_.SetAsInteger(MediaFormat::kWidth, config->video_width);
+      media_format_.SetAsInteger(MediaFormat::kHeight, config->video_height);
+    }
+  }
+
+  // Implementation of DemuxerStream.
+  virtual const MediaFormat* GetMediaFormat() {
+    return &media_format_;
+  }
+
+  virtual void Read(Assignable<Buffer>* buffer) {
+    NOTREACHED();  // TODO(ralphl): fix me!!
+  }
+
+ private:
+  virtual ~MockDemuxerStream() {}
+
+  MediaFormat media_format_;
+
+  DISALLOW_COPY_AND_ASSIGN(MockDemuxerStream);
+};
+
+//------------------------------------------------------------------------------
+
 class MockDemuxer : public Demuxer {
  public:
   static FilterFactory* CreateFactory(const MockFilterConfig* config) {
      return new FilterFactoryImpl1<MockDemuxer,
                                    const MockFilterConfig*>(config);
   }
 
   explicit MockDemuxer(const MockFilterConfig* config)
       : config_(config),
-        mock_audio_stream_(config, true),
-        mock_video_stream_(config, false) {
+        mock_audio_stream_(new MockDemuxerStream(config, true)),
+        mock_video_stream_(new MockDemuxerStream(config, false)) {
   }
 
   // Implementation of MediaFilter.
   virtual void Stop() {}
 
   // Implementation of Demuxer.
   virtual bool Initialize(DataSource* data_source) {
     host_->InitializationComplete();
     return true;
   }
 
   virtual size_t GetNumberOfStreams() {
     size_t num_streams = 0;
     if (config_->has_audio) {
       ++num_streams;
     }
     if (config_->has_video) {
       ++num_streams;
     }
     return num_streams;
   }
 
-  virtual DemuxerStream* GetStream(int stream_id) {
+  virtual scoped_refptr<DemuxerStream> GetStream(int stream_id) {
     switch (stream_id) {
       case 0:
         if (config_->has_audio) {
-          return &mock_audio_stream_;
+          return mock_audio_stream_;
         } else if (config_->has_video) {
-          return &mock_video_stream_;
+          return mock_video_stream_;
         }
         break;
       case 1:
         if (config_->has_audio && config_->has_video) {
-          return &mock_video_stream_;
+          return mock_video_stream_;
         }
         break;
     }
     ADD_FAILURE();
     return NULL;
   }
 
  private:
   virtual ~MockDemuxer() {}
 
-  // Internal class implements DemuxerStream interface.
-  class MockDemuxerStream : public DemuxerStream {
-   public:
-    MockDemuxerStream(const MockFilterConfig* config, bool is_audio) {
-      if (is_audio) {
-        media_format_.SetAsString(MediaFormat::kMimeType,
-                                  config->compressed_audio_mime_type);
-      } else {
-        media_format_.SetAsString(MediaFormat::kMimeType,
-                                  config->compressed_video_mime_type);
-        media_format_.SetAsInteger(MediaFormat::kWidth, config->video_width);
-        media_format_.SetAsInteger(MediaFormat::kHeight, config->video_height);
-      }
-    }
-
-    virtual ~MockDemuxerStream() {}
-
-    // Implementation of DemuxerStream.
-    virtual const MediaFormat* GetMediaFormat() {
-      return &media_format_;
-    }
-
-    virtual void Read(Assignable<Buffer>* buffer) {
-      NOTREACHED();  // TODO(ralphl): fix me!!
-    }
-
-   private:
-    MediaFormat media_format_;
-
-    DISALLOW_COPY_AND_ASSIGN(MockDemuxerStream);
-  };
-
   const MockFilterConfig* config_;
-  MockDemuxerStream mock_audio_stream_;
-  MockDemuxerStream mock_video_stream_;
+  scoped_refptr<DemuxerStream> mock_audio_stream_;
+  scoped_refptr<DemuxerStream> mock_video_stream_;
 
   DISALLOW_COPY_AND_ASSIGN(MockDemuxer);
 };
 
 //------------------------------------------------------------------------------
 
 class MockAudioDecoder : public AudioDecoder {
  public:
   static FilterFactory* CreateFactory(const MockFilterConfig* config) {
     return new FilterFactoryImpl1<MockAudioDecoder,
@@ skipping 289 matching lines @@
   base::WaitableEvent event_;
   bool callback_success_status_;
   bool waiting_for_callback_;
 
   DISALLOW_COPY_AND_ASSIGN(InitializationHelper);
 };
 
 }  // namespace media
 
 #endif  // MEDIA_BASE_MOCK_MEDIA_FILTERS_H_