d3d11_av1_accelerator.h 2.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879
  1. // Copyright 2020 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #ifndef MEDIA_GPU_WINDOWS_D3D11_AV1_ACCELERATOR_H_
  5. #define MEDIA_GPU_WINDOWS_D3D11_AV1_ACCELERATOR_H_
  6. #include <d3d11_1.h>
  7. #include <d3d9.h>
  8. #include <dxva.h>
  9. #include <windows.h>
  10. #include <wrl/client.h>
  11. #include "base/callback_helpers.h"
  12. #include "base/memory/raw_ptr.h"
  13. #include "media/base/media_log.h"
  14. #include "media/gpu/av1_decoder.h"
  15. #include "media/gpu/windows/d3d11_com_defs.h"
  16. #include "media/gpu/windows/d3d11_status.h"
  17. #include "media/gpu/windows/d3d11_video_context_wrapper.h"
  18. #include "media/gpu/windows/d3d11_video_decoder_client.h"
  19. typedef struct _DXVA_PicParams_AV1 DXVA_PicParams_AV1;
  20. typedef struct _DXVA_Tile_AV1 DXVA_Tile_AV1;
  21. namespace media {
  22. class D3D11AV1Accelerator : public AV1Decoder::AV1Accelerator {
  23. public:
  24. D3D11AV1Accelerator(D3D11VideoDecoderClient* client,
  25. MediaLog* media_log,
  26. ComD3D11VideoDevice video_device,
  27. std::unique_ptr<VideoContextWrapper> video_context);
  28. D3D11AV1Accelerator(const D3D11AV1Accelerator&) = delete;
  29. D3D11AV1Accelerator& operator=(const D3D11AV1Accelerator&) = delete;
  30. ~D3D11AV1Accelerator() override;
  31. scoped_refptr<AV1Picture> CreateAV1Picture(bool apply_grain) override;
  32. Status SubmitDecode(const AV1Picture& pic,
  33. const libgav1::ObuSequenceHeader& seq_header,
  34. const AV1ReferenceFrameVector& ref_frames,
  35. const libgav1::Vector<libgav1::TileBuffer>& tile_buffers,
  36. base::span<const uint8_t> data) override;
  37. bool OutputPicture(const AV1Picture& pic) override;
  38. private:
  39. class ScopedDecoderBuffer;
  40. ScopedDecoderBuffer GetBuffer(D3D11_VIDEO_DECODER_BUFFER_TYPE type);
  41. bool SubmitDecoderBuffer(
  42. const DXVA_PicParams_AV1& pic_params,
  43. const libgav1::Vector<libgav1::TileBuffer>& tile_buffers);
  44. void RecordFailure(const std::string& fail_type, D3D11Status error);
  45. void RecordFailure(const std::string& fail_type,
  46. const std::string& message,
  47. D3D11Status::Codes reason);
  48. void SetVideoDecoder(ComD3D11VideoDecoder video_decoder);
  49. void FillPicParams(size_t picture_index,
  50. bool apply_grain,
  51. const libgav1::ObuFrameHeader& frame_header,
  52. const libgav1::ObuSequenceHeader& seq_header,
  53. const AV1ReferenceFrameVector& ref_frames,
  54. DXVA_PicParams_AV1* pp);
  55. raw_ptr<D3D11VideoDecoderClient> client_;
  56. std::unique_ptr<MediaLog> media_log_;
  57. ComD3D11VideoDecoder video_decoder_;
  58. ComD3D11VideoDevice video_device_;
  59. std::unique_ptr<VideoContextWrapper> video_context_;
  60. };
  61. } // namespace media
  62. #endif // MEDIA_GPU_WINDOWS_D3D11_AV1_ACCELERATOR_H_