/*
 *  Flo's Open libRary (floor)
 *  Copyright (C) 2004 - 2026 Florian Ziesche
 *  
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; version 2 of the License only.
 *  
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *  
 *  You should have received a copy of the GNU General Public License along
 *  with this program; if not, write to the Free Software Foundation, Inc.,
 *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

#pragma once

#if defined(FLOOR_DEVICE_METAL)

#pragma OPENCL EXTENSION cl_khr_fp64 : enable
#pragma OPENCL EXTENSION cl_khr_fp16 : enable
#pragma OPENCL EXTENSION cl_khr_gl_msaa_sharing : enable

#define global __attribute__((global_as))
#define constant __attribute__((constant_as))
#define local __attribute__((local_as))
#define kernel_1d(... /* x */) extern "C" __attribute__((compute_kernel, kernel_dim(1), kernel_work_group_size(__VA_ARGS__)))
#define kernel_2d(... /* x, y */) extern "C" __attribute__((compute_kernel, kernel_dim(2), kernel_work_group_size(__VA_ARGS__)))
#define kernel_3d(... /* x, y, z */) extern "C" __attribute__((compute_kernel, kernel_dim(3), kernel_work_group_size(__VA_ARGS__)))
#define kernel_1d_simd(req_simd_width, ... /* x */) extern "C" __attribute__((compute_kernel, kernel_dim(1), kernel_simd_width(req_simd_width), kernel_work_group_size(__VA_ARGS__)))
#define kernel_2d_simd(req_simd_width, ... /* x, y */) extern "C" __attribute__((compute_kernel, kernel_dim(2), kernel_simd_width(req_simd_width), kernel_work_group_size(__VA_ARGS__)))
#define kernel_3d_simd(req_simd_width, ... /* x, y, z */) extern "C" __attribute__((compute_kernel, kernel_dim(3), kernel_simd_width(req_simd_width), kernel_work_group_size(__VA_ARGS__)))
#define kernel kernel_1d()
#define vertex extern "C" __attribute__((vertex_shader))
#define fragment extern "C" __attribute__((fragment_shader))
#define tessellation_control extern "C" __attribute__((tessellation_control_shader, kernel_dim(1)))
#define tessellation_evaluation extern "C" __attribute__((tessellation_evaluation_shader))
#if FLOOR_DEVICE_INFO_MESH_SHADING_SUPPORT
#define task_shader_1d(... /* x */) extern "C" __attribute__((task_shader, kernel_dim(1), kernel_work_group_size(__VA_ARGS__)))
#define task_shader_2d(... /* x, y */) extern "C" __attribute__((task_shader, kernel_dim(2), kernel_work_group_size(__VA_ARGS__)))
#define task_shader_3d(... /* x, y, z */) extern "C" __attribute__((task_shader, kernel_dim(3), kernel_work_group_size(__VA_ARGS__)))
#define task_shader_1d_simd(req_simd_width, ... /* x */) extern "C" __attribute__((task_shader, kernel_dim(1), kernel_simd_width(req_simd_width), kernel_work_group_size(__VA_ARGS__)))
#define task_shader_2d_simd(req_simd_width, ... /* x, y */) extern "C" __attribute__((task_shader, kernel_dim(2), kernel_simd_width(req_simd_width), kernel_work_group_size(__VA_ARGS__)))
#define task_shader_3d_simd(req_simd_width, ... /* x, y, z */) extern "C" __attribute__((task_shader, kernel_dim(3), kernel_simd_width(req_simd_width), kernel_work_group_size(__VA_ARGS__)))
#define mesh_shader_1d(... /* x */) extern "C" __attribute__((mesh_shader, kernel_dim(1), kernel_work_group_size(__VA_ARGS__)))
#define mesh_shader_2d(... /* x, y */) extern "C" __attribute__((mesh_shader, kernel_dim(2), kernel_work_group_size(__VA_ARGS__)))
#define mesh_shader_3d(... /* x, y, z */) extern "C" __attribute__((mesh_shader, kernel_dim(3), kernel_work_group_size(__VA_ARGS__)))
#define mesh_shader_1d_simd(req_simd_width, ... /* x */) extern "C" __attribute__((mesh_shader, kernel_dim(1), kernel_simd_width(req_simd_width), kernel_work_group_size(__VA_ARGS__)))
#define mesh_shader_2d_simd(req_simd_width, ... /* x, y */) extern "C" __attribute__((mesh_shader, kernel_dim(2), kernel_simd_width(req_simd_width), kernel_work_group_size(__VA_ARGS__)))
#define mesh_shader_3d_simd(req_simd_width, ... /* x, y, z */) extern "C" __attribute__((mesh_shader, kernel_dim(3), kernel_simd_width(req_simd_width), kernel_work_group_size(__VA_ARGS__)))
#define task_max_mesh_work_groups(x) [[mesh_max_work_groups(x)]]
#endif

#define metal_func inline __attribute__((always_inline))

// misc types
using int8_t = char;
using int16_t = short int;
using int32_t = int;
using int64_t = long int;
using uint8_t = unsigned char;
using uint16_t = unsigned short int;
using uint32_t = unsigned int;
using uint64_t = unsigned long int;

using size_t = __SIZE_TYPE__;
using ssize_t = __PTRDIFF_TYPE__;

using uintptr_t = __SIZE_TYPE__;
using intptr_t = __PTRDIFF_TYPE__;
using ptrdiff_t = __PTRDIFF_TYPE__;

// memory and synchronization scopes
#define FLOOR_METAL_SYNC_SCOPE_THREAD 0
#define FLOOR_METAL_SYNC_SCOPE_LOCAL 1
#define FLOOR_METAL_SYNC_SCOPE_GLOBAL 2
#define FLOOR_METAL_SYNC_SCOPE_SUB_GROUP 4

#define FLOOR_METAL_MEM_FLAGS_NONE 0
#define FLOOR_METAL_MEM_FLAGS_GLOBAL 1
#define FLOOR_METAL_MEM_FLAGS_LOCAL 2
#define FLOOR_METAL_MEM_FLAGS_TEXTURE 4
#define FLOOR_METAL_MEM_FLAGS_LOCAL_IMAGE_BLOCK 8
#define FLOOR_METAL_MEM_FLAGS_TASK_PAYLOAD 16
#define FLOOR_METAL_MEM_FLAGS_ALL (FLOOR_METAL_MEM_FLAGS_GLOBAL | FLOOR_METAL_MEM_FLAGS_LOCAL | FLOOR_METAL_MEM_FLAGS_TEXTURE | \
								   FLOOR_METAL_MEM_FLAGS_LOCAL_IMAGE_BLOCK | FLOOR_METAL_MEM_FLAGS_TASK_PAYLOAD)

// NOTE: with Metal 4.1 these are now properly supported (on Apple GPUs), but there was support since at least Metal 2.4,
//       so we make use of these in earlier versions as well
#define FLOOR_METAL_MEM_ORDER_RELAXED 0
#define FLOOR_METAL_MEM_ORDER_ACQUIRE 2
#define FLOOR_METAL_MEM_ORDER_RELEASE 3
#define FLOOR_METAL_MEM_ORDER_ACQ_REL 4
#define FLOOR_METAL_MEM_ORDER_SEQ_CST 5

// use acquire-release ordering by default
#define FLOOR_METAL_MEM_ORDER_DEFAULT FLOOR_METAL_MEM_ORDER_ACQ_REL

#endif
