/* -*- 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()->mBitDepth = 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

  // bitdepth

  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