mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-12 21:30:45 +01:00
a2e945e042
This CL extracts the ivf file writer from `TestEncodedImageCallback` into separate .cc|.h files. Improve the `EncodedImageFileWriter` to support SVC that output ivf for all decode targets. EXAMPLE: Encode with VP9 L3T3_KEY, the outputs: output-VP9-L3T3_KEY-L0T0.ivf output-VP9-L3T3_KEY-L0T1.ivf output-VP9-L3T3_KEY-L0T2.ivf output-VP9-L3T3_KEY-L1T0.ivf output-VP9-L3T3_KEY-L1T1.ivf output-VP9-L3T3_KEY-L1T2.ivf output-VP9-L3T3_KEY-L2T0.ivf output-VP9-L3T3_KEY-L2T1.ivf output-VP9-L3T3_KEY-L2T2.ivf Bug: webrtc:15210 Change-Id: Iba46c897a7b783bb4b79ec18715e901476cb9f55 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/309280 Reviewed-by: Erik Språng <sprang@webrtc.org> Reviewed-by: Mirko Bonadei <mbonadei@webrtc.org> Commit-Queue: Jianhui J Dai <jianhui.j.dai@intel.com> Cr-Commit-Position: refs/heads/main@{#40363}
51 lines
1.6 KiB
C++
51 lines
1.6 KiB
C++
/*
|
|
* Copyright (c) 2023 The WebRTC project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
#ifndef RTC_TOOLS_VIDEO_ENCODER_ENCODED_IMAGE_FILE_WRITER_H_
|
|
#define RTC_TOOLS_VIDEO_ENCODER_ENCODED_IMAGE_FILE_WRITER_H_
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
#include "modules/video_coding/include/video_codec_interface.h"
|
|
#include "modules/video_coding/utility/ivf_file_writer.h"
|
|
|
|
namespace webrtc {
|
|
namespace test {
|
|
|
|
// The `EncodedImageFileWriter` writes the `EncodedImage` into ivf output. It
|
|
// supports SVC to output ivf for all decode targets.
|
|
class EncodedImageFileWriter final {
|
|
// The pair of writer and output file name.
|
|
using IvfWriterPair = std::pair<std::unique_ptr<IvfFileWriter>, std::string>;
|
|
|
|
public:
|
|
explicit EncodedImageFileWriter(const VideoCodec& video_codec_setting);
|
|
|
|
~EncodedImageFileWriter();
|
|
|
|
int Write(const EncodedImage& encoded_image);
|
|
|
|
private:
|
|
VideoCodec video_codec_setting_;
|
|
|
|
int spatial_layers_ = 0;
|
|
int temporal_layers_ = 0;
|
|
InterLayerPredMode inter_layer_pred_mode_ = InterLayerPredMode::kOff;
|
|
|
|
bool is_base_layer_key_frame = false;
|
|
std::vector<IvfWriterPair> decode_target_writers_;
|
|
};
|
|
|
|
} // namespace test
|
|
} // namespace webrtc
|
|
|
|
#endif // RTC_TOOLS_VIDEO_ENCODER_ENCODED_IMAGE_FILE_WRITER_H_
|