blob: 4f6a0d7f59dab2aa924c3d947975ea17858332c5 [file] [log] [blame]
// Copyright (c) 2010 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.
// Interface and some concrete classes for applying various transforms
// to AVPackets. FFmpegBitstreamConverter, in particular, can be used
// to apply FFmpeg bitstream filters to the incoming AVPacket to transcode
// the packet format.
#ifndef MEDIA_FILTERS_BITSTREAM_CONVERTER_H_
#define MEDIA_FILTERS_BITSTREAM_CONVERTER_H_
#include <string>
#include "base/basictypes.h"
#include "base/gtest_prod_util.h"
// FFmpeg types.
struct AVBitStreamFilterContext;
struct AVCodecContext;
struct AVPacket;
namespace media {
class BitstreamConverter {
public:
BitstreamConverter() {}
virtual ~BitstreamConverter() {}
// Attemps to convert the AVPacket from one format to another, based on the
// specific type of BitstreamConverter that was instantiated.
virtual bool Initialize() = 0;
virtual bool ConvertPacket(AVPacket* packet) = 0;
private:
DISALLOW_COPY_AND_ASSIGN(BitstreamConverter);
};
class IdentityBitstreamConverter : public BitstreamConverter {
public:
IdentityBitstreamConverter() {}
virtual ~IdentityBitstreamConverter() {}
virtual bool Initialize() { return true; }
virtual bool ConvertPacket(AVPacket* packet) { return true; }
private:
DISALLOW_COPY_AND_ASSIGN(IdentityBitstreamConverter);
};
class FFmpegBitstreamConverter : public BitstreamConverter {
public:
// Creates FFmpegBitstreamConverter based on the FFmpeg bistream filter
// corresponding to |filter_name|.
//
// The |stream_context| will be used during conversion and should be the
// AVCodecContext for the stream sourcing these packets. A reference to
// |stream_context| is retained, so it must outlive this class.
FFmpegBitstreamConverter(const std::string& filter_name,
AVCodecContext* stream_context);
virtual ~FFmpegBitstreamConverter();
virtual bool Initialize();
virtual bool ConvertPacket(AVPacket* packet);
private:
FRIEND_TEST_ALL_PREFIXES(BitstreamConverterTest, ConvertPacket_FailedFilter);
FRIEND_TEST_ALL_PREFIXES(BitstreamConverterTest, ConvertPacket_Success);
FRIEND_TEST_ALL_PREFIXES(BitstreamConverterTest,
ConvertPacket_SuccessInPlace);
std::string filter_name_;
AVBitStreamFilterContext* stream_filter_;
AVCodecContext* stream_context_;
DISALLOW_COPY_AND_ASSIGN(FFmpegBitstreamConverter);
};
} // namespace media
#endif // MEDIA_FILTERS_BITSTREAM_CONVERTER_H_