// Copyright 2014 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef MEDIA_BASE_MEDIA_CLIENT_H_
#define MEDIA_BASE_MEDIA_CLIENT_H_

#include <memory>
#include <optional>
#include <vector>

#include "media/base/audio_codecs.h"
#include "media/base/audio_parameters.h"
#include "media/base/decoder_buffer.h"
#include "media/base/key_system_info.h"
#include "media/base/media_export.h"
#include "media/base/media_types.h"
#include "media/base/video_codecs.h"
#include "media/base/video_color_space.h"
#include "ui/gfx/color_space.h"

namespace media {

class MediaClient;

// Setter for the client. If a customized client is needed, it should be set
// early, before the client could possibly be used.
MEDIA_EXPORT void SetMediaClient(MediaClient* media_client);

// Media's embedder API should only be used by media.
#if defined(IS_MEDIA_IMPL) || defined(MEDIA_BLINK_IMPLEMENTATION)
// Getter for the client. Returns NULL if no customized client is needed.
MEDIA_EXPORT MediaClient* GetMediaClient();
#endif

class MEDIA_EXPORT ExternalMemoryAllocator {
 public:
  virtual ~ExternalMemoryAllocator() = default;
  virtual std::unique_ptr<DecoderBuffer::ExternalMemory> CopyFrom(
      base::span<const uint8_t> span) = 0;
};

// A client interface for embedders (e.g. content/renderer) to provide
// customized key systems and decoders.
class MEDIA_EXPORT MediaClient {
 public:
  MediaClient();
  virtual ~MediaClient();

  // Returns true if the given audio decoder config is supported.
  virtual bool IsDecoderSupportedAudioType(const AudioType& type) = 0;

  // Returns true if the given video decoder config is supported.
  virtual bool IsDecoderSupportedVideoType(const VideoType& type) = 0;

  // Returns true if the given video encoder config is supported.
  virtual bool IsEncoderSupportedVideoType(const VideoType& type) = 0;

  // Returns true if the compressed audio |codec| format is supported by the
  // audio sink.
  virtual bool IsSupportedBitstreamAudioCodec(AudioCodec codec) = 0;

  // Returns true if audio tracks should be suppressed (neither decoded nor
  // rendered).
  virtual bool ShouldSuppressAudioTracks() = 0;

  // Optionally returns audio renderer algorithm parameters.
  virtual std::optional<::media::AudioRendererAlgorithmParameters>
  GetAudioRendererAlgorithmParameters(AudioParameters audio_parameters) = 0;

  // Returns custom allocator if embedder has provided one, else nullptr.
  // When an allocator is provided, the MSE pipeline will use the provided
  // allocator during StreamParserBuffer creation. The SRC playback pipeline
  // does not currently use the allocator, but could be updated to do so.
  virtual ExternalMemoryAllocator* GetMediaAllocator() = 0;
};

}  // namespace media

#endif  // MEDIA_BASE_MEDIA_CLIENT_H_