| OLD | NEW |
| (Empty) |
| 1 // Copyright 2016 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 | |
| 5 #ifndef MEDIA_GPU_MEDIA_FOUNDATION_VIDEO_ENCODE_ACCELERATOR_WIN_H_ | |
| 6 #define MEDIA_GPU_MEDIA_FOUNDATION_VIDEO_ENCODE_ACCELERATOR_WIN_H_ | |
| 7 | |
| 8 #include <mfapi.h> | |
| 9 #include <mfidl.h> | |
| 10 #include <stdint.h> | |
| 11 #include <strmif.h> | |
| 12 | |
| 13 #include <deque> | |
| 14 #include <memory> | |
| 15 | |
| 16 #include "base/memory/weak_ptr.h" | |
| 17 #include "base/sequenced_task_runner.h" | |
| 18 #include "base/single_thread_task_runner.h" | |
| 19 #include "base/threading/thread_checker.h" | |
| 20 #include "base/win/scoped_comptr.h" | |
| 21 #include "media/gpu/media_gpu_export.h" | |
| 22 #include "media/video/video_encode_accelerator.h" | |
| 23 | |
| 24 namespace media { | |
| 25 | |
| 26 // Media Foundation implementation of the VideoEncodeAccelerator interface for | |
| 27 // Windows. | |
| 28 // This class saves the task runner on which it is constructed and returns | |
| 29 // encoded data to the client using that same task runner. This class has | |
| 30 // DCHECKs to makes sure that methods are called in sequence. It starts an | |
| 31 // internal encoder thread on which VideoEncodeAccelerator implementation tasks | |
| 32 // are posted. | |
| 33 class MEDIA_GPU_EXPORT MediaFoundationVideoEncodeAccelerator | |
| 34 : public VideoEncodeAccelerator { | |
| 35 public: | |
| 36 MediaFoundationVideoEncodeAccelerator(); | |
| 37 | |
| 38 // VideoEncodeAccelerator implementation. | |
| 39 VideoEncodeAccelerator::SupportedProfiles GetSupportedProfiles() override; | |
| 40 bool Initialize(VideoPixelFormat input_format, | |
| 41 const gfx::Size& input_visible_size, | |
| 42 VideoCodecProfile output_profile, | |
| 43 uint32_t initial_bitrate, | |
| 44 Client* client) override; | |
| 45 void Encode(const scoped_refptr<VideoFrame>& frame, | |
| 46 bool force_keyframe) override; | |
| 47 void UseOutputBitstreamBuffer(const BitstreamBuffer& buffer) override; | |
| 48 void RequestEncodingParametersChange(uint32_t bitrate, | |
| 49 uint32_t framerate) override; | |
| 50 void Destroy() override; | |
| 51 | |
| 52 // Preload dlls required for encoding. | |
| 53 static void PreSandboxInitialization(); | |
| 54 | |
| 55 protected: | |
| 56 ~MediaFoundationVideoEncodeAccelerator() override; | |
| 57 | |
| 58 private: | |
| 59 // Holds output buffers coming from the client ready to be filled. | |
| 60 struct BitstreamBufferRef; | |
| 61 | |
| 62 // Holds output buffers coming from the encoder. | |
| 63 class EncodeOutput; | |
| 64 | |
| 65 // Initializes and allocates memory for input and output samples. | |
| 66 bool InitializeInputOutputSamples(); | |
| 67 | |
| 68 // Initializes encoder parameters for real-time use. | |
| 69 bool SetEncoderModes(); | |
| 70 | |
| 71 // Helper function to notify the client of an error on |client_task_runner_|. | |
| 72 void NotifyError(VideoEncodeAccelerator::Error error); | |
| 73 | |
| 74 // Encoding tasks to be run on |encoder_thread_|. | |
| 75 void EncodeTask(const scoped_refptr<VideoFrame>& frame, bool force_keyframe); | |
| 76 | |
| 77 // Checks for and copies encoded output on |encoder_thread_|. | |
| 78 void ProcessOutput(); | |
| 79 | |
| 80 // Inserts the output buffers for reuse on |encoder_thread_|. | |
| 81 void UseOutputBitstreamBufferTask( | |
| 82 std::unique_ptr<BitstreamBufferRef> buffer_ref); | |
| 83 | |
| 84 // Copies EncodeOutput into a BitstreamBuffer and returns it to the |client_|. | |
| 85 void ReturnBitstreamBuffer( | |
| 86 std::unique_ptr<EncodeOutput> encode_output, | |
| 87 std::unique_ptr<MediaFoundationVideoEncodeAccelerator::BitstreamBufferRef> | |
| 88 buffer_ref); | |
| 89 | |
| 90 // Changes encode parameters on |encoder_thread_|. | |
| 91 void RequestEncodingParametersChangeTask(uint32_t bitrate, | |
| 92 uint32_t framerate); | |
| 93 | |
| 94 // Destroys encode session on |encoder_thread_|. | |
| 95 void DestroyTask(); | |
| 96 | |
| 97 // Bitstream buffers ready to be used to return encoded output as a FIFO. | |
| 98 std::deque<std::unique_ptr<BitstreamBufferRef>> bitstream_buffer_queue_; | |
| 99 | |
| 100 // EncodeOutput needs to be copied into a BitstreamBufferRef as a FIFO. | |
| 101 std::deque<std::unique_ptr<EncodeOutput>> encoder_output_queue_; | |
| 102 | |
| 103 gfx::Size input_visible_size_; | |
| 104 size_t bitstream_buffer_size_; | |
| 105 int32_t frame_rate_; | |
| 106 int32_t target_bitrate_; | |
| 107 size_t u_plane_offset_; | |
| 108 size_t v_plane_offset_; | |
| 109 | |
| 110 base::win::ScopedComPtr<IMFTransform> encoder_; | |
| 111 base::win::ScopedComPtr<ICodecAPI> codec_api_; | |
| 112 | |
| 113 DWORD input_stream_count_min_; | |
| 114 DWORD input_stream_count_max_; | |
| 115 DWORD output_stream_count_min_; | |
| 116 DWORD output_stream_count_max_; | |
| 117 | |
| 118 base::win::ScopedComPtr<IMFSample> input_sample_; | |
| 119 base::win::ScopedComPtr<IMFSample> output_sample_; | |
| 120 | |
| 121 // To expose client callbacks from VideoEncodeAccelerator. | |
| 122 // NOTE: all calls to this object *MUST* be executed on |client_task_runner_|. | |
| 123 base::WeakPtr<Client> client_; | |
| 124 std::unique_ptr<base::WeakPtrFactory<Client>> client_ptr_factory_; | |
| 125 | |
| 126 // Our original calling task runner for the child thread. | |
| 127 const scoped_refptr<base::SequencedTaskRunner> client_task_runner_; | |
| 128 // Sequence checker to enforce that the methods of this object are called in | |
| 129 // sequence. | |
| 130 base::SequenceChecker sequence_checker_; | |
| 131 | |
| 132 // This thread services tasks posted from the VEA API entry points by the | |
| 133 // GPU child thread and CompressionCallback() posted from device thread. | |
| 134 base::Thread encoder_thread_; | |
| 135 scoped_refptr<base::SingleThreadTaskRunner> encoder_thread_task_runner_; | |
| 136 | |
| 137 // Declared last to ensure that all weak pointers are invalidated before | |
| 138 // other destructors run. | |
| 139 base::WeakPtrFactory<MediaFoundationVideoEncodeAccelerator> | |
| 140 encoder_task_weak_factory_; | |
| 141 | |
| 142 DISALLOW_COPY_AND_ASSIGN(MediaFoundationVideoEncodeAccelerator); | |
| 143 }; | |
| 144 | |
| 145 } // namespace media | |
| 146 | |
| 147 #endif // MEDIA_GPU_MEDIA_FOUNDATION_VIDEO_ENCODE_ACCELERATOR_WIN_H_ | |
| OLD | NEW |