| Index: webrtc/modules/video_coding/codecs/test/videoprocessor.h
|
| diff --git a/webrtc/modules/video_coding/codecs/test/videoprocessor.h b/webrtc/modules/video_coding/codecs/test/videoprocessor.h
|
| index 242844387a77b6eada4d42d9ca5838acd0b5e483..cd4482aeff75dadcc7f7ee8a193bc0d7fbefdd49 100644
|
| --- a/webrtc/modules/video_coding/codecs/test/videoprocessor.h
|
| +++ b/webrtc/modules/video_coding/codecs/test/videoprocessor.h
|
| @@ -18,6 +18,7 @@
|
| #include "webrtc/api/video/video_frame.h"
|
| #include "webrtc/base/buffer.h"
|
| #include "webrtc/base/checks.h"
|
| +#include "webrtc/base/sequenced_task_checker.h"
|
| #include "webrtc/common_video/libyuv/include/webrtc_libyuv.h"
|
| #include "webrtc/modules/video_coding/include/video_codec_interface.h"
|
| #include "webrtc/modules/video_coding/codecs/test/packet_manipulator.h"
|
| @@ -128,16 +129,17 @@ struct TestConfig {
|
| // Video Engine, where signaling would request a retransmit of the lost packets,
|
| // since they're so important.
|
| //
|
| -// Note this class is not thread safe in any way and is meant for simple testing
|
| -// purposes.
|
| +// This class should be run on a single thread, or a single task queue.
|
| class VideoProcessor {
|
| public:
|
| - virtual ~VideoProcessor() {}
|
| + virtual ~VideoProcessor() = default;
|
|
|
| // Performs initial calculations about frame size, sets up callbacks etc.
|
| // Returns false if an error has occurred, in addition to printing to stderr.
|
| virtual bool Init() = 0;
|
|
|
| + virtual void DeregisterCallbacks() = 0;
|
| +
|
| // Processes a single frame. Returns true as long as there's more frames
|
| // available in the source clip.
|
| // Frame number must be an integer >= 0.
|
| @@ -172,8 +174,9 @@ class VideoProcessorImpl : public VideoProcessor {
|
| FrameWriter* source_frame_writer,
|
| IvfFileWriter* encoded_frame_writer,
|
| FrameWriter* decoded_frame_writer);
|
| - virtual ~VideoProcessorImpl();
|
| + ~VideoProcessorImpl() override;
|
| bool Init() override;
|
| + void DeregisterCallbacks() override;
|
| bool ProcessFrame(int frame_number) override;
|
|
|
| private:
|
| @@ -271,21 +274,22 @@ class VideoProcessorImpl : public VideoProcessor {
|
| // Return the number of spatial resizes.
|
| int NumberSpatialResizes() override;
|
|
|
| - webrtc::VideoEncoder* const encoder_;
|
| - webrtc::VideoDecoder* const decoder_;
|
| - const std::unique_ptr<VideoBitrateAllocator> bitrate_allocator_;
|
| + webrtc::VideoEncoder* const encoder_ PT_GUARDED_BY(task_checker_);
|
| + webrtc::VideoDecoder* const decoder_ PT_GUARDED_BY(task_checker_);
|
| + const std::unique_ptr<VideoBitrateAllocator> bitrate_allocator_
|
| + PT_GUARDED_BY(task_checker_);
|
|
|
| // Adapters for the codec callbacks.
|
| const std::unique_ptr<EncodedImageCallback> encode_callback_;
|
| const std::unique_ptr<DecodedImageCallback> decode_callback_;
|
|
|
| - PacketManipulator* const packet_manipulator_;
|
| + PacketManipulator* const packet_manipulator_ PT_GUARDED_BY(task_checker_);
|
| const TestConfig& config_;
|
|
|
| // These (mandatory) file manipulators are used for, e.g., objective PSNR and
|
| // SSIM calculations at the end of a test run.
|
| - FrameReader* const analysis_frame_reader_;
|
| - FrameWriter* const analysis_frame_writer_;
|
| + FrameReader* const analysis_frame_reader_ PT_GUARDED_BY(task_checker_);
|
| + FrameWriter* const analysis_frame_writer_ PT_GUARDED_BY(task_checker_);
|
| const int num_frames_;
|
|
|
| // These (optional) file writers are used for persistently storing the output
|
| @@ -294,31 +298,34 @@ class VideoProcessorImpl : public VideoProcessor {
|
| // experimenter an option to subjectively evaluate the quality of the
|
| // encoding, given the test settings. Each frame writer is enabled by being
|
| // non-null.
|
| - FrameWriter* const source_frame_writer_;
|
| - IvfFileWriter* const encoded_frame_writer_;
|
| - FrameWriter* const decoded_frame_writer_;
|
| + FrameWriter* const source_frame_writer_ PT_GUARDED_BY(task_checker_);
|
| + IvfFileWriter* const encoded_frame_writer_ PT_GUARDED_BY(task_checker_);
|
| + FrameWriter* const decoded_frame_writer_ PT_GUARDED_BY(task_checker_);
|
|
|
| - bool initialized_;
|
| + bool initialized_ GUARDED_BY(task_checker_);
|
|
|
| // Frame metadata for all frames that have been added through a call to
|
| // ProcessFrames(). We need to store this metadata over the course of the
|
| // test run, to support pipelining HW codecs.
|
| - std::vector<FrameInfo> frame_infos_;
|
| - int last_encoded_frame_num_;
|
| - int last_decoded_frame_num_;
|
| + std::vector<FrameInfo> frame_infos_ GUARDED_BY(task_checker_);
|
| + int last_encoded_frame_num_ GUARDED_BY(task_checker_);
|
| + int last_decoded_frame_num_ GUARDED_BY(task_checker_);
|
|
|
| // Keep track of if we have excluded the first key frame from packet loss.
|
| - bool first_key_frame_has_been_excluded_;
|
| + bool first_key_frame_has_been_excluded_ GUARDED_BY(task_checker_);
|
|
|
| // Keep track of the last successfully decoded frame, since we write that
|
| // frame to disk when decoding fails.
|
| - rtc::Buffer last_decoded_frame_buffer_;
|
| + rtc::Buffer last_decoded_frame_buffer_ GUARDED_BY(task_checker_);
|
|
|
| // Statistics.
|
| - Stats* stats_;
|
| - int num_dropped_frames_;
|
| - int num_spatial_resizes_;
|
| - double bit_rate_factor_; // Multiply frame length with this to get bit rate.
|
| + Stats* stats_ PT_GUARDED_BY(task_checker_);
|
| + int num_dropped_frames_ GUARDED_BY(task_checker_);
|
| + int num_spatial_resizes_ GUARDED_BY(task_checker_);
|
| + // Multiply frame length with this to get bit rate.
|
| + double bit_rate_factor_ GUARDED_BY(task_checker_);
|
| +
|
| + rtc::SequencedTaskChecker task_checker_;
|
| };
|
|
|
| } // namespace test
|
|
|