blob: 8f9518fd683f5fa41dcbf69a570cb04e169acb9a [file] [log] [blame]
// Copyright 2018 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.
#include <memory>
#include <vector>
#include "base/callback.h"
#include "base/macros.h"
#include "base/memory/ref_counted.h"
#include "base/memory/weak_ptr.h"
#include "media/base/media_export.h"
#include "media/base/video_decoder.h"
#include "media/base/video_decoder_config.h"
namespace media {
class VideoFrame;
// Class to decode a single video frame for video thumbnail generation.
class MEDIA_EXPORT VideoThumbnailDecoder {
using VideoFrameCallback =
// Creates the thumbnail decoder with |decoder| that performs the actual work.
// |encoded_data| is the encoded frame extracted with ffmpeg. A video frame
// will be returned through |video_frame_callback|.
VideoThumbnailDecoder(std::unique_ptr<VideoDecoder> decoder,
const VideoDecoderConfig& config,
std::vector<uint8_t> encoded_data);
// Starts to decode the video frame.
void Start(VideoFrameCallback video_frame_callback);
void OnVideoDecoderInitialized(bool success);
void OnVideoBufferDecoded(DecodeStatus status);
void OnEosBufferDecoded(DecodeStatus status);
// Called when the output frame is generated.
void OnVideoFrameDecoded(const scoped_refptr<VideoFrame>& frame);
void NotifyComplete(scoped_refptr<VideoFrame> frame);
// The decoder that does the actual decoding.
std::unique_ptr<VideoDecoder> decoder_;
const VideoDecoderConfig config_;
// The demuxed, encoded video frame waiting to be decoded.
std::vector<uint8_t> encoded_data_;
VideoFrameCallback video_frame_callback_;
base::WeakPtrFactory<VideoThumbnailDecoder> weak_factory_;
} // namespace media