mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-13 13:50:40 +01:00

Mechanically generated by running this command: tools_webrtc/do-renames.sh update all-renames.txt && git cl format Then manually updating: tools_webrtc/sanitizers/tsan_suppressions_webrtc.cc Bug: webrtc:10159 No-Presubmit: true No-Tree-Checks: true No-Try: true Change-Id: I54824cd91dada8fc3ee3d098f971bc319d477833 Reviewed-on: https://webrtc-review.googlesource.com/c/115653 Reviewed-by: Karl Wiberg <kwiberg@webrtc.org> Cr-Commit-Position: refs/heads/master@{#26226}
89 lines
3 KiB
C++
89 lines
3 KiB
C++
/*
|
|
* Copyright (c) 2017 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 MODULES_VIDEO_CODING_CODECS_MULTIPLEX_INCLUDE_MULTIPLEX_ENCODER_ADAPTER_H_
|
|
#define MODULES_VIDEO_CODING_CODECS_MULTIPLEX_INCLUDE_MULTIPLEX_ENCODER_ADAPTER_H_
|
|
|
|
#include <map>
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
#include "api/video_codecs/sdp_video_format.h"
|
|
#include "api/video_codecs/video_encoder.h"
|
|
#include "api/video_codecs/video_encoder_factory.h"
|
|
#include "modules/video_coding/codecs/multiplex/multiplex_encoded_image_packer.h"
|
|
#include "modules/video_coding/include/video_codec_interface.h"
|
|
#include "rtc_base/critical_section.h"
|
|
|
|
namespace webrtc {
|
|
|
|
enum AlphaCodecStream {
|
|
kYUVStream = 0,
|
|
kAXXStream = 1,
|
|
kAlphaCodecStreams = 2,
|
|
};
|
|
|
|
class MultiplexEncoderAdapter : public VideoEncoder {
|
|
public:
|
|
// |factory| is not owned and expected to outlive this class.
|
|
MultiplexEncoderAdapter(VideoEncoderFactory* factory,
|
|
const SdpVideoFormat& associated_format,
|
|
bool supports_augmenting_data = false);
|
|
virtual ~MultiplexEncoderAdapter();
|
|
|
|
// Implements VideoEncoder
|
|
int InitEncode(const VideoCodec* inst,
|
|
int number_of_cores,
|
|
size_t max_payload_size) override;
|
|
int Encode(const VideoFrame& input_image,
|
|
const CodecSpecificInfo* codec_specific_info,
|
|
const std::vector<FrameType>* frame_types) override;
|
|
int RegisterEncodeCompleteCallback(EncodedImageCallback* callback) override;
|
|
int SetRateAllocation(const VideoBitrateAllocation& bitrate,
|
|
uint32_t new_framerate) override;
|
|
int Release() override;
|
|
EncoderInfo GetEncoderInfo() const override;
|
|
|
|
EncodedImageCallback::Result OnEncodedImage(
|
|
AlphaCodecStream stream_idx,
|
|
const EncodedImage& encodedImage,
|
|
const CodecSpecificInfo* codecSpecificInfo,
|
|
const RTPFragmentationHeader* fragmentation);
|
|
|
|
private:
|
|
// Wrapper class that redirects OnEncodedImage() calls.
|
|
class AdapterEncodedImageCallback;
|
|
|
|
VideoEncoderFactory* const factory_;
|
|
const SdpVideoFormat associated_format_;
|
|
std::vector<std::unique_ptr<VideoEncoder>> encoders_;
|
|
std::vector<std::unique_ptr<AdapterEncodedImageCallback>> adapter_callbacks_;
|
|
EncodedImageCallback* encoded_complete_callback_;
|
|
|
|
std::map<uint32_t /* timestamp */, MultiplexImage> stashed_images_
|
|
RTC_GUARDED_BY(crit_);
|
|
|
|
uint16_t picture_index_ = 0;
|
|
std::vector<uint8_t> multiplex_dummy_planes_;
|
|
|
|
int key_frame_interval_;
|
|
EncodedImage combined_image_;
|
|
|
|
rtc::CriticalSection crit_;
|
|
|
|
const bool supports_augmented_data_;
|
|
int augmenting_data_size_ = 0;
|
|
|
|
EncoderInfo encoder_info_;
|
|
};
|
|
|
|
} // namespace webrtc
|
|
|
|
#endif // MODULES_VIDEO_CODING_CODECS_MULTIPLEX_INCLUDE_MULTIPLEX_ENCODER_ADAPTER_H_
|