71 lines
2.5 KiB
C++
71 lines
2.5 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/. */
|
|
#if !defined(DAV1DDecoder_h_)
|
|
# define DAV1DDecoder_h_
|
|
|
|
# include "PerformanceRecorder.h"
|
|
# include "PlatformDecoderModule.h"
|
|
# include "dav1d/dav1d.h"
|
|
# include "mozilla/Result.h"
|
|
# include "nsRefPtrHashtable.h"
|
|
|
|
namespace mozilla {
|
|
|
|
DDLoggedTypeDeclNameAndBase(DAV1DDecoder, MediaDataDecoder);
|
|
|
|
typedef nsRefPtrHashtable<nsPtrHashKey<const uint8_t>, MediaRawData>
|
|
MediaRawDataHashtable;
|
|
|
|
class DAV1DDecoder final : public MediaDataDecoder,
|
|
public DecoderDoctorLifeLogger<DAV1DDecoder> {
|
|
public:
|
|
NS_INLINE_DECL_THREADSAFE_REFCOUNTING(DAV1DDecoder, final);
|
|
|
|
explicit DAV1DDecoder(const CreateDecoderParams& aParams);
|
|
|
|
RefPtr<InitPromise> Init() override;
|
|
RefPtr<DecodePromise> Decode(MediaRawData* aSample) override;
|
|
RefPtr<DecodePromise> Drain() override;
|
|
RefPtr<FlushPromise> Flush() override;
|
|
RefPtr<ShutdownPromise> Shutdown() override;
|
|
nsCString GetDescriptionName() const override {
|
|
return "av1 libdav1d video decoder"_ns;
|
|
}
|
|
nsCString GetCodecName() const override { return "av1"_ns; }
|
|
|
|
void ReleaseDataBuffer(const uint8_t* buf);
|
|
|
|
static Maybe<gfx::YUVColorSpace> GetColorSpace(const Dav1dPicture& aPicture,
|
|
LazyLogModule& aLogger);
|
|
|
|
static Maybe<gfx::ColorSpace2> GetColorPrimaries(const Dav1dPicture& aPicture,
|
|
LazyLogModule& aLogger);
|
|
|
|
private:
|
|
virtual ~DAV1DDecoder();
|
|
RefPtr<DecodePromise> InvokeDecode(MediaRawData* aSample);
|
|
Result<already_AddRefed<VideoData>, MediaResult> GetPicture();
|
|
Result<already_AddRefed<VideoData>, MediaResult> ConstructImage(
|
|
const Dav1dPicture& aPicture);
|
|
|
|
Dav1dContext* mContext = nullptr;
|
|
|
|
const VideoInfo mInfo;
|
|
const RefPtr<TaskQueue> mTaskQueue;
|
|
const RefPtr<layers::ImageContainer> mImageContainer;
|
|
const RefPtr<layers::KnowsCompositor> mImageAllocator;
|
|
const Maybe<TrackingId> mTrackingId;
|
|
const bool mLowLatency;
|
|
PerformanceRecorderMulti<DecodeStage> mPerformanceRecorder;
|
|
|
|
// Keep the buffers alive until dav1d
|
|
// does not need them any more.
|
|
MediaRawDataHashtable mDecodingBuffers;
|
|
};
|
|
|
|
} // namespace mozilla
|
|
|
|
#endif // DAV1DDecoder_h_
|