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

#ifndef UI_GFX_MAC_IO_SURFACE_H_
#define UI_GFX_MAC_IO_SURFACE_H_

#include <IOSurface/IOSurfaceRef.h>
#include <mach/mach.h>

#include "base/apple/scoped_cftyperef.h"
#include "base/component_export.h"
#include "base/memory/unsafe_shared_memory_region.h"
#include "ui/gfx/buffer_types.h"
#include "ui/gfx/color_space.h"
#include "ui/gfx/generic_shared_memory_id.h"
#include "ui/gfx/geometry/size.h"

namespace viz {
class SharedImageFormat;
}

namespace gfx {

namespace internal {

struct IOSurfaceMachPortTraits {
  COMPONENT_EXPORT(GFX) static mach_port_t InvalidValue() {
    return MACH_PORT_NULL;
  }
  COMPONENT_EXPORT(GFX) static mach_port_t Retain(mach_port_t);
  COMPONENT_EXPORT(GFX) static void Release(mach_port_t);
};

struct ScopedInUseIOSurfaceTraits {
  static IOSurfaceRef InvalidValue() { return nullptr; }
  static IOSurfaceRef Retain(IOSurfaceRef io_surface) {
    CFRetain(io_surface);
    IOSurfaceIncrementUseCount(io_surface);
    return io_surface;
  }
  static void Release(IOSurfaceRef io_surface) {
    IOSurfaceDecrementUseCount(io_surface);
    CFRelease(io_surface);
  }
};

}  // namespace internal

using IOSurfaceId = GenericSharedMemoryId;

// A scoper for handling Mach port names that are send rights for IOSurfaces.
// This scoper is both copyable and assignable, which will increase the kernel
// reference count of the right. On destruction, the reference count is
// decremented.
using ScopedRefCountedIOSurfaceMachPort =
    base::apple::ScopedTypeRef<mach_port_t, internal::IOSurfaceMachPortTraits>;

// A scoper for holding a reference to an IOSurface and also incrementing its
// in-use counter while the scoper exists.
using ScopedInUseIOSurface =
    base::apple::ScopedTypeRef<IOSurfaceRef,
                               internal::ScopedInUseIOSurfaceTraits>;

// A scoper for holding a reference to an IOSurface.
using ScopedIOSurface = base::apple::ScopedCFTypeRef<IOSurfaceRef>;

static constexpr size_t kMaxIOSurfacePlanes = 3;

// Helper function to create an IOSurface with a specified size and format.
// The surface is zero-initialized if |should_clear| is true. This is not
// necessary for anonymous surfaces that are not exported to renderers and used
// as render targets only. If |override_rgba_to_bgra| is true (default) a BGRA
// IOSurface is created for RGBA/X_8888 format. This is needed for GL
// usage since neither ANGLE Metal nor CGL support importing RGBA IOSurfaces,
// whereas for non-GL backends (Dawn and Metal) we want the formats to match.
// TODO(sunnyps): Revisit this when we switch to ANGLE Metal completely since
// wrapping RGBA_8888 can be implemented with Metal quite easily.
COMPONENT_EXPORT(GFX)
ScopedIOSurface CreateIOSurface(const Size& size,
                                viz::SharedImageFormat format,
                                bool should_clear = true,
#if BUILDFLAG(IS_IOS)
                                bool override_rgba_to_bgra = false
#else
                                bool override_rgba_to_bgra = true
#endif
);

#if BUILDFLAG(IS_IOS)
COMPONENT_EXPORT(GFX)
void ExportIOSurfaceSharedMemoryRegion(
    IOSurfaceRef io_surface,
    base::UnsafeSharedMemoryRegion& shared_memory_region,
    std::array<uint32_t, kMaxIOSurfacePlanes>& plane_strides,
    std::array<uint32_t, kMaxIOSurfacePlanes>& plane_offsets);
#endif

// Return true if there exists a value for IOSurfaceColorSpace or
// IOSurfaceICCProfile that will make CoreAnimation render using |color_space|.
COMPONENT_EXPORT(GFX)
bool IOSurfaceCanSetColorSpace(const gfx::ColorSpace& color_space);

// Set color space for given IOSurface. IOSurfaceCanSetColorSpace must return
// true for |color_space| otherwise this does nothing.
COMPONENT_EXPORT(GFX)
void IOSurfaceSetColorSpace(IOSurfaceRef io_surface,
                            const gfx::ColorSpace& color_space);

// Return an IOSurface consuming |io_surface_mach_port|.
COMPONENT_EXPORT(GFX)
ScopedIOSurface IOSurfaceMachPortToIOSurface(
    ScopedRefCountedIOSurfaceMachPort io_surface_mach_port);

}  // namespace gfx

#endif  // UI_GFX_MAC_IO_SURFACE_H_