mirror of
https://github.com/roytam1/palemoon27.git
synced 2026-05-26 14:18:48 +00:00
trav90
0a7eb3d9c0
The logic was extracted from LibAV cmdutils.c. FFmpeg provides an API for that (av_frame_get_best_effort_timestamp()) unfortunately this isn't provided by LibAV. So copy the logic instead in order to keep compatibility with the two forks.
89 lines
2.6 KiB
C++
89 lines
2.6 KiB
C++
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* vim:set ts=2 sw=2 sts=2 et cindent: */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#ifndef __FFmpegH264Decoder_h__
|
|
#define __FFmpegH264Decoder_h__
|
|
|
|
#include "FFmpegDataDecoder.h"
|
|
|
|
namespace mozilla
|
|
{
|
|
|
|
template <int V>
|
|
class FFmpegH264Decoder : public FFmpegDataDecoder<V>
|
|
{
|
|
};
|
|
|
|
template <>
|
|
class FFmpegH264Decoder<LIBAV_VER> : public FFmpegDataDecoder<LIBAV_VER>
|
|
{
|
|
typedef mozilla::layers::Image Image;
|
|
typedef mozilla::layers::ImageContainer ImageContainer;
|
|
|
|
enum DecodeResult {
|
|
DECODE_FRAME,
|
|
DECODE_NO_FRAME,
|
|
DECODE_ERROR
|
|
};
|
|
|
|
public:
|
|
FFmpegH264Decoder(FlushableMediaTaskQueue* aTaskQueue,
|
|
MediaDataDecoderCallback* aCallback,
|
|
const VideoInfo& aConfig,
|
|
ImageContainer* aImageContainer);
|
|
virtual ~FFmpegH264Decoder();
|
|
|
|
virtual nsresult Init() override;
|
|
virtual nsresult Input(MediaRawData* aSample) override;
|
|
virtual nsresult Drain() override;
|
|
virtual nsresult Flush() override;
|
|
static AVCodecID GetCodecId(const nsACString& aMimeType);
|
|
|
|
private:
|
|
void DecodeFrame(MediaRawData* aSample);
|
|
DecodeResult DoDecodeFrame(MediaRawData* aSample);
|
|
void DoDrain();
|
|
void OutputDelayedFrames();
|
|
|
|
/**
|
|
* This method allocates a buffer for FFmpeg's decoder, wrapped in an Image.
|
|
* Currently it only supports Planar YUV420, which appears to be the only
|
|
* non-hardware accelerated image format that FFmpeg's H264 decoder is
|
|
* capable of outputting.
|
|
*/
|
|
int AllocateYUV420PVideoBuffer(AVCodecContext* aCodecContext,
|
|
AVFrame* aFrame);
|
|
|
|
static int AllocateBufferCb(AVCodecContext* aCodecContext, AVFrame* aFrame);
|
|
static void ReleaseBufferCb(AVCodecContext* aCodecContext, AVFrame* aFrame);
|
|
|
|
MediaDataDecoderCallback* mCallback;
|
|
nsRefPtr<ImageContainer> mImageContainer;
|
|
uint32_t mPictureWidth;
|
|
uint32_t mPictureHeight;
|
|
uint32_t mDisplayWidth;
|
|
uint32_t mDisplayHeight;
|
|
|
|
class PtsCorrectionContext {
|
|
public:
|
|
PtsCorrectionContext();
|
|
int64_t GuessCorrectPts(int64_t aPts, int64_t aDts);
|
|
void Reset();
|
|
|
|
private:
|
|
int64_t mNumFaultyPts; /// Number of incorrect PTS values so far
|
|
int64_t mNumFaultyDts; /// Number of incorrect DTS values so far
|
|
int64_t mLastPts; /// PTS of the last frame
|
|
int64_t mLastDts; /// DTS of the last frame
|
|
};
|
|
|
|
PtsCorrectionContext mPtsContext;
|
|
};
|
|
|
|
} // namespace mozilla
|
|
|
|
#endif // __FFmpegH264Decoder_h__
|