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

#include "media/base/format_utils.h"
#include "base/logging.h"

namespace media {

std::optional<VideoPixelFormat> SharedImageFormatToVideoPixelFormat(
    viz::SharedImageFormat format) {
  if (format == viz::SinglePlaneFormat::kBGRX_8888) {
    return PIXEL_FORMAT_XRGB;
  } else if (format == viz::SinglePlaneFormat::kBGRA_8888) {
    return PIXEL_FORMAT_ARGB;
  } else if (format == viz::SinglePlaneFormat::kRGBX_8888) {
    return PIXEL_FORMAT_XBGR;
  } else if (format == viz::SinglePlaneFormat::kRGBA_8888) {
    return PIXEL_FORMAT_ABGR;
  } else if (format == viz::SinglePlaneFormat::kRGBA_1010102) {
    return PIXEL_FORMAT_XB30;
  } else if (format == viz::SinglePlaneFormat::kBGRA_1010102) {
    return PIXEL_FORMAT_XR30;
  } else if (format == viz::SinglePlaneFormat::kRGBA_F16) {
    return PIXEL_FORMAT_RGBAF16;
  } else if (format == viz::MultiPlaneFormat::kI420) {
    return PIXEL_FORMAT_I420;
  } else if (format == viz::MultiPlaneFormat::kI420A) {
    return PIXEL_FORMAT_I420A;
  } else if (format == viz::MultiPlaneFormat::kYV12) {
    return PIXEL_FORMAT_YV12;
  } else if (format == viz::MultiPlaneFormat::kNV12) {
    return PIXEL_FORMAT_NV12;
  } else if (format == viz::MultiPlaneFormat::kNV16) {
    return PIXEL_FORMAT_NV16;
  } else if (format == viz::MultiPlaneFormat::kNV24) {
    return PIXEL_FORMAT_NV24;
  } else if (format == viz::MultiPlaneFormat::kNV12A) {
    return PIXEL_FORMAT_NV12A;
  } else if (format == viz::MultiPlaneFormat::kP010) {
    return PIXEL_FORMAT_P010LE;
  } else if (format == viz::MultiPlaneFormat::kP210) {
    return PIXEL_FORMAT_P210LE;
  } else if (format == viz::MultiPlaneFormat::kP410) {
    return PIXEL_FORMAT_P410LE;
  } else {
    DLOG(WARNING) << "Unsupported SharedImageFormat: " << format.ToString();
    return std::nullopt;
  }
}

std::optional<viz::SharedImageFormat> VideoPixelFormatToSharedImageFormat(
    VideoPixelFormat pixel_format) {
  switch (pixel_format) {
    case PIXEL_FORMAT_ARGB:
      return viz::SinglePlaneFormat::kBGRA_8888;
    case PIXEL_FORMAT_XRGB:
      return viz::SinglePlaneFormat::kBGRX_8888;
    case PIXEL_FORMAT_ABGR:
      return viz::SinglePlaneFormat::kRGBA_8888;
    case PIXEL_FORMAT_XBGR:
      return viz::SinglePlaneFormat::kRGBX_8888;
    case PIXEL_FORMAT_XB30:
      return viz::SinglePlaneFormat::kRGBA_1010102;
    case PIXEL_FORMAT_XR30:
      return viz::SinglePlaneFormat::kBGRA_1010102;
    case PIXEL_FORMAT_RGBAF16:
      return viz::SinglePlaneFormat::kRGBA_F16;
    case PIXEL_FORMAT_YV12:
      return viz::MultiPlaneFormat::kYV12;
    case PIXEL_FORMAT_NV12:
      return viz::MultiPlaneFormat::kNV12;
    case PIXEL_FORMAT_NV16:
      return viz::MultiPlaneFormat::kNV16;
    case PIXEL_FORMAT_NV24:
      return viz::MultiPlaneFormat::kNV24;
    case PIXEL_FORMAT_NV12A:
      return viz::MultiPlaneFormat::kNV12A;
    case PIXEL_FORMAT_P010LE:
      return viz::MultiPlaneFormat::kP010;
    case PIXEL_FORMAT_P210LE:
      return viz::MultiPlaneFormat::kP210;
    case PIXEL_FORMAT_P410LE:
      return viz::MultiPlaneFormat::kP410;
    case PIXEL_FORMAT_I420:
      return viz::MultiPlaneFormat::kI420;
    case PIXEL_FORMAT_I420A:
      return viz::MultiPlaneFormat::kI420A;
    default:
      DLOG(WARNING) << "Unsupported VideoPixelFormat: " << pixel_format;
      return std::nullopt;
  }
}

}  // namespace media