gecko-dev/dom/media/webm/WebMDecoder.cpp

124 строки
3.6 KiB
C++

/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim:set ts=2 sw=2 sts=2 et cindent: */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#include "WebMDecoder.h"
#include "mozilla/Move.h"
#include "mozilla/Preferences.h"
#include "mozilla/StaticPrefs.h"
#ifdef MOZ_AV1
# include "AOMDecoder.h"
#endif
#include "MediaContainerType.h"
#include "PDMFactory.h"
#include "VideoUtils.h"
namespace mozilla {
/* static */
nsTArray<UniquePtr<TrackInfo>> WebMDecoder::GetTracksInfo(
const MediaContainerType& aType, MediaResult& aError) {
nsTArray<UniquePtr<TrackInfo>> tracks;
const bool isVideo = aType.Type() == MEDIAMIMETYPE("video/webm");
if (aType.Type() != MEDIAMIMETYPE("audio/webm") && !isVideo) {
aError = MediaResult(
NS_ERROR_DOM_MEDIA_FATAL_ERR,
RESULT_DETAIL("Invalid type:%s", aType.Type().AsString().get()));
return tracks;
}
aError = NS_OK;
const MediaCodecs& codecs = aType.ExtendedType().Codecs();
if (codecs.IsEmpty()) {
return tracks;
}
for (const auto& codec : codecs.Range()) {
if (codec.EqualsLiteral("opus") || codec.EqualsLiteral("vorbis")) {
tracks.AppendElement(
CreateTrackInfoWithMIMETypeAndContainerTypeExtraParameters(
NS_LITERAL_CSTRING("audio/") + NS_ConvertUTF16toUTF8(codec),
aType));
continue;
}
if (isVideo) {
UniquePtr<TrackInfo> trackInfo;
if (IsVP9CodecString(codec)) {
trackInfo = CreateTrackInfoWithMIMETypeAndContainerTypeExtraParameters(
NS_LITERAL_CSTRING("video/vp9"), aType);
} else if (IsVP8CodecString(codec)) {
trackInfo = CreateTrackInfoWithMIMETypeAndContainerTypeExtraParameters(
NS_LITERAL_CSTRING("video/vp8"), aType);
}
if (trackInfo) {
uint8_t profile = 0;
uint8_t level = 0;
uint8_t bitDepth = 0;
if (ExtractVPXCodecDetails(codec, profile, level, bitDepth)) {
trackInfo->GetAsVideoInfo()->mColorDepth =
gfx::ColorDepthForBitDepth(bitDepth);
}
tracks.AppendElement(std::move(trackInfo));
continue;
}
}
#ifdef MOZ_AV1
if (StaticPrefs::MediaAv1Enabled() && IsAV1CodecString(codec)) {
tracks.AppendElement(
CreateTrackInfoWithMIMETypeAndContainerTypeExtraParameters(
NS_LITERAL_CSTRING("video/av1"), aType));
continue;
}
#endif
// Unknown codec
aError = MediaResult(
NS_ERROR_DOM_MEDIA_FATAL_ERR,
RESULT_DETAIL("Unknown codec:%s", NS_ConvertUTF16toUTF8(codec).get()));
}
return tracks;
}
/* static */
bool WebMDecoder::IsSupportedType(const MediaContainerType& aContainerType) {
if (!StaticPrefs::MediaWebMEnabled()) {
return false;
}
MediaResult rv = NS_OK;
auto tracks = GetTracksInfo(aContainerType, rv);
if (NS_FAILED(rv)) {
return false;
}
if (tracks.IsEmpty()) {
// WebM guarantees that the only codecs it contained are vp8, vp9, opus or
// vorbis.
return true;
}
// Verify that we have a PDM that supports the whitelisted types, include
// color depth
RefPtr<PDMFactory> platform = new PDMFactory();
for (const auto& track : tracks) {
if (!track || !platform->Supports(*track, nullptr /* diagnostic */)) {
return false;
}
}
return true;
}
/* static */
nsTArray<UniquePtr<TrackInfo>> WebMDecoder::GetTracksInfo(
const MediaContainerType& aType) {
MediaResult rv = NS_OK;
return GetTracksInfo(aType, rv);
}
} // namespace mozilla