1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63
   64
   65
   66
   67
   68
   69
   70
   71
   72
   73
   74
   75
   76
   77
   78
   79
   80
   81
   82
   83


media / base / mac / color_space_util_mac.mm [blame]

// Copyright 2020 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/mac/color_space_util_mac.h"

#include <CoreFoundation/CoreFoundation.h>
#include <CoreVideo/CoreVideo.h>
#include <simd/simd.h>
#include <vector>

#include "base/apple/foundation_util.h"
#include "base/apple/scoped_cftyperef.h"
#include "base/memory/scoped_policy.h"
#include "base/no_destructor.h"
#include "third_party/skia/modules/skcms/skcms.h"
#include "ui/gfx/mac/color_space_util.h"

namespace media {

gfx::ColorSpace GetImageBufferColorSpace(CVImageBufferRef image_buffer) {
  base::apple::ScopedCFTypeRef<CFTypeRef> color_primaries;
  base::apple::ScopedCFTypeRef<CFTypeRef> transfer_function;
  base::apple::ScopedCFTypeRef<CFTypeRef> gamma_level;
  base::apple::ScopedCFTypeRef<CFTypeRef> ycbcr_matrix;

  if (@available(macOS 12, iOS 15, *)) {
    color_primaries.reset(CVBufferCopyAttachment(
        image_buffer, kCVImageBufferColorPrimariesKey, nullptr));
    transfer_function.reset(CVBufferCopyAttachment(
        image_buffer, kCVImageBufferTransferFunctionKey, nullptr));
    gamma_level.reset(CVBufferCopyAttachment(
        image_buffer, kCVImageBufferGammaLevelKey, nullptr));
    ycbcr_matrix.reset(CVBufferCopyAttachment(
        image_buffer, kCVImageBufferYCbCrMatrixKey, nullptr));
  } else {
#if !defined(__IPHONE_15_0) || __IPHONE_OS_VERSION_MIN_REQUIRED < __IPHONE_15_0
    color_primaries.reset(
        CVBufferGetAttachment(image_buffer, kCVImageBufferColorPrimariesKey,
                              nullptr),
        base::scoped_policy::RETAIN);
    transfer_function.reset(
        CVBufferGetAttachment(image_buffer, kCVImageBufferTransferFunctionKey,
                              nullptr),
        base::scoped_policy::RETAIN);
    gamma_level.reset(CVBufferGetAttachment(
                          image_buffer, kCVImageBufferGammaLevelKey, nullptr),
                      base::scoped_policy::RETAIN);
    ycbcr_matrix.reset(CVBufferGetAttachment(
                           image_buffer, kCVImageBufferYCbCrMatrixKey, nullptr),
                       base::scoped_policy::RETAIN);
#endif
  }

  return gfx::ColorSpaceFromCVImageBufferKeys(
      color_primaries.get(), transfer_function.get(), gamma_level.get(),
      ycbcr_matrix.get());
}

gfx::ColorSpace GetFormatDescriptionColorSpace(
    CMFormatDescriptionRef format_description) {
  return gfx::ColorSpaceFromCVImageBufferKeys(
      CMFormatDescriptionGetExtension(
          format_description, kCMFormatDescriptionExtension_ColorPrimaries),
      CMFormatDescriptionGetExtension(
          format_description, kCMFormatDescriptionExtension_TransferFunction),
      CMFormatDescriptionGetExtension(format_description,
                                      kCMFormatDescriptionExtension_GammaLevel),
      CMFormatDescriptionGetExtension(
          format_description, kCMFormatDescriptionExtension_YCbCrMatrix));
}

// Converts a gfx::ColorSpace to individual kCVImageBuffer* keys.
bool GetImageBufferColorValues(const gfx::ColorSpace& color_space,
                               CFStringRef* out_primaries,
                               CFStringRef* out_transfer,
                               CFStringRef* out_matrix) {
  return gfx::ColorSpaceToCVImageBufferKeys(
      color_space,
      /*prefer_srgb_trfn=*/false, out_primaries, out_transfer, out_matrix);
}

}  // namespace media