DAV1DDecoder.h - mozsearch

Enable keyboard shortcuts

/* -*- 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_