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

Since https://webrtc-review.googlesource.com/c/src/+/267780 supported scalability modes are also used to compare for equality between SdpVideoFormats(?). Bug: webrtc:14267, webrtc:13573 Change-Id: I2f3c2fca93bac6fadd222f776f672c9bd3f1de0a Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/268304 Reviewed-by: Tomas Gunnarsson <tommi@webrtc.org> Commit-Queue: Philip Eliasson <philipel@webrtc.org> Cr-Commit-Position: refs/heads/main@{#37510}
134 lines
4.4 KiB
C++
134 lines
4.4 KiB
C++
/*
|
|
* Copyright (c) 2022 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 API_VIDEO_CODECS_VIDEO_ENCODER_FACTORY_TEMPLATE_H_
|
|
#define API_VIDEO_CODECS_VIDEO_ENCODER_FACTORY_TEMPLATE_H_
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "absl/algorithm/container.h"
|
|
#include "api/array_view.h"
|
|
#include "api/video_codecs/video_encoder.h"
|
|
#include "api/video_codecs/video_encoder_factory.h"
|
|
#include "modules/video_coding/svc/scalability_mode_util.h"
|
|
|
|
namespace webrtc {
|
|
// The VideoEncoderFactoryTemplate supports encoders implementations given as
|
|
// template arguments.
|
|
//
|
|
// To include an encoder in the factory it requires three static members
|
|
// functions to be defined:
|
|
//
|
|
// // Returns the supported SdpVideoFormats this encoder can produce.
|
|
// static std::vector<SdpVideoFormat> SupportedFormats();
|
|
//
|
|
// // Creates an encoder instance for the given format.
|
|
// static std::unique_ptr<VideoEncoder>
|
|
// CreateEncoder(const SdpVideoFormat& format);
|
|
//
|
|
// // Returns true if the encoder supports the given scalability mode.
|
|
// static bool
|
|
// IsScalabilityModeSupported(ScalabilityMode scalability_mode);
|
|
//
|
|
// Note that the order of the template arguments matter as the factory will
|
|
// query/return the first encoder implementation supporting the given
|
|
// SdpVideoFormat.
|
|
template <typename... Ts>
|
|
class VideoEncoderFactoryTemplate : public VideoEncoderFactory {
|
|
public:
|
|
std::vector<SdpVideoFormat> GetSupportedFormats() const override {
|
|
std::vector<SdpVideoFormat> formats;
|
|
GetSupportedFormatsInternal<Ts...>(formats);
|
|
return formats;
|
|
}
|
|
|
|
std::unique_ptr<VideoEncoder> CreateVideoEncoder(
|
|
const SdpVideoFormat& format) override {
|
|
return CreateVideoEncoderInternal<Ts...>(format);
|
|
}
|
|
|
|
CodecSupport QueryCodecSupport(
|
|
const SdpVideoFormat& format,
|
|
absl::optional<std::string> scalability_mode) const override {
|
|
return QueryCodecSupportInternal<Ts...>(format, scalability_mode);
|
|
}
|
|
|
|
private:
|
|
bool IsFormatInList(
|
|
const SdpVideoFormat& format,
|
|
rtc::ArrayView<const SdpVideoFormat> supported_formats) const {
|
|
return absl::c_any_of(
|
|
supported_formats, [&](const SdpVideoFormat& supported_format) {
|
|
return supported_format.name == format.name &&
|
|
supported_format.parameters == format.parameters;
|
|
});
|
|
}
|
|
|
|
template <typename V>
|
|
bool IsScalabilityModeSupported(
|
|
const absl::optional<std::string>& scalability_mode_string) const {
|
|
if (!scalability_mode_string.has_value()) {
|
|
return true;
|
|
}
|
|
absl::optional<ScalabilityMode> scalability_mode =
|
|
ScalabilityModeFromString(*scalability_mode_string);
|
|
return scalability_mode.has_value() &&
|
|
V::IsScalabilityModeSupported(*scalability_mode);
|
|
}
|
|
|
|
template <typename V, typename... Vs>
|
|
void GetSupportedFormatsInternal(std::vector<SdpVideoFormat>& formats) const {
|
|
auto supported_formats = V::SupportedFormats();
|
|
for (const auto& format : supported_formats) {
|
|
if (!IsFormatInList(format, formats)) {
|
|
formats.push_back(format);
|
|
}
|
|
}
|
|
|
|
if constexpr (sizeof...(Vs) > 0) {
|
|
return GetSupportedFormatsInternal<Vs...>(formats);
|
|
}
|
|
}
|
|
|
|
template <typename V, typename... Vs>
|
|
std::unique_ptr<VideoEncoder> CreateVideoEncoderInternal(
|
|
const SdpVideoFormat& format) {
|
|
if (IsFormatInList(format, V::SupportedFormats())) {
|
|
return V::CreateEncoder(format);
|
|
}
|
|
|
|
if constexpr (sizeof...(Vs) > 0) {
|
|
return CreateVideoEncoderInternal<Vs...>(format);
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
template <typename V, typename... Vs>
|
|
CodecSupport QueryCodecSupportInternal(
|
|
const SdpVideoFormat& format,
|
|
const absl::optional<std::string>& scalability_mode) const {
|
|
if (IsFormatInList(format, V::SupportedFormats())) {
|
|
return {.is_supported = IsScalabilityModeSupported<V>(scalability_mode)};
|
|
}
|
|
|
|
if constexpr (sizeof...(Vs) > 0) {
|
|
return QueryCodecSupportInternal<Vs...>(format, scalability_mode);
|
|
}
|
|
|
|
return {.is_supported = false};
|
|
}
|
|
};
|
|
|
|
} // namespace webrtc
|
|
|
|
#endif // API_VIDEO_CODECS_VIDEO_ENCODER_FACTORY_TEMPLATE_H_
|