#include "image_common.h"

#ifdef cl_khr_fp16
#pragma OPENCL EXTENSION cl_khr_fp16 : enable
#endif

// From
// https://github.com/ROCm/clr/tree/amd-staging/hipamd/include/hip/amd_detail/texture_fetch_functions.h
_CLC_CONST_AS const unsigned int SAMPLER_OBJECT_OFFSET_DWORD = 12;

// Using the builtin as_type() and as_typen() functions to reinterpret types.
// The restriction being is that element "type"s need to be of the same size.
#define _CLC_DEFINE_BUILTIN_CAST_VEC4_TO_VEC3(vec4_elem_t, to_t)               \
  _CLC_DEF to_t##3 __clc_cast_from_##vec4_elem_t##4_to_##to_t##3(              \
      vec4_elem_t##4 from) {                                                   \
    vec4_elem_t##3 casted = as_##vec4_elem_t##3(from);                         \
    return as_##to_t##3(casted);                                               \
  }
#define _CLC_DEFINE_BUILTIN_CAST_VEC4_TO_VEC2(vec4_elem_t, to_t)               \
  _CLC_DEF to_t##2 __clc_cast_from_##vec4_elem_t##4_to_##to_t##2(              \
      vec4_elem_t##4 from) {                                                   \
    vec4_elem_t##4 casted = as_##vec4_elem_t##4(from);                         \
    return as_##to_t##2((vec4_elem_t##2)(casted.x, casted.y));                 \
  }
#define _CLC_DEFINE_BUILTIN_CAST_VEC4_TO_SCALAR(vec4_elem_t, to_t)             \
  _CLC_DEF to_t __clc_cast_from_##vec4_elem_t##4_to_##to_t(                    \
      vec4_elem_t##4 from) {                                                   \
    vec4_elem_t##4 casted = as_##vec4_elem_t##4(from);                         \
    return as_##to_t(casted.x);                                                \
  }
#define _CLC_DEFINE_BUILTIN_CAST_VEC3_TO_VEC4(from_t, vec4_elem_t)             \
  _CLC_DEF vec4_elem_t##4 __clc_cast_from_##from_t##3_to_##vec4_elem_t##4(     \
      from_t##3 from) {                                                        \
    vec4_elem_t##3 casted = as_##vec4_elem_t##3(from);                         \
    return as_##vec4_elem_t##4(casted);                                        \
  }
#define _CLC_DEFINE_BUILTIN_CAST_VEC2_TO_VEC4(from_t, vec4_elem_t)             \
  _CLC_DEF vec4_elem_t##4 __clc_cast_from_##from_t##2_to_##vec4_elem_t##4(     \
      from_t##2 from) {                                                        \
    vec4_elem_t##2 casted = as_##vec4_elem_t##2(from);                         \
    return (vec4_elem_t##4)(casted.x, casted.y, 0, 0);                         \
  }
#define _CLC_DEFINE_BUILTIN_CAST_SCALAR_TO_VEC4(from_t, vec4_elem_t)           \
  _CLC_DEF vec4_elem_t##4 __clc_cast_from_##from_t##_to_##vec4_elem_t##4(      \
      from_t from) {                                                           \
    vec4_elem_t casted = as_##vec4_elem_t(from);                               \
    return (vec4_elem_t##4)(casted, 0, 0, 0);                                  \
  }

// Generic casts between builtin types.
#define _CLC_DEFINE_CAST_VEC4(vec4_elem_t, to_t)                               \
  _CLC_DEF to_t##4 __clc_cast_from_##vec4_elem_t##4_to_##to_t##4(              \
      vec4_elem_t##4 from) {                                                   \
    return (to_t##4)(from.x, from.y, from.z, from.w);                          \
  }
#define _CLC_DEFINE_CAST_VEC4_TO_VEC3(vec4_elem_t, to_t)                       \
  _CLC_DEF to_t##3 __clc_cast_from_##vec4_elem_t##4_to_##to_t##3(              \
      vec4_elem_t##4 from) {                                                   \
    return (to_t##3)(from.x, from.y, from.z);                                  \
  }
#define _CLC_DEFINE_CAST_VEC4_TO_VEC2(vec4_elem_t, to_t)                       \
  _CLC_DEF to_t##2 __clc_cast_from_##vec4_elem_t##4_to_##to_t##2(              \
      vec4_elem_t##4 from) {                                                   \
    return (to_t##2)(from.x, from.y);                                          \
  }
#define _CLC_DEFINE_CAST_VEC4_TO_SCALAR(vec4_elem_t, to_t)                     \
  _CLC_DEF to_t __clc_cast_from_##vec4_elem_t##4_to_##to_t(                    \
      vec4_elem_t##4 from) {                                                   \
    return (to_t)from.x;                                                       \
  }
#define _CLC_DEFINE_CAST_VEC3_TO_VEC4(from_t, vec4_elem_t)                     \
  _CLC_DEF vec4_elem_t##4 __clc_cast_from_##from_t##3_to_##vec4_elem_t##4(     \
      from_t##3 from) {                                                        \
    return (vec4_elem_t##4)(from.x, from.y, from.z, 0);                        \
  }
#define _CLC_DEFINE_CAST_VEC2_TO_VEC4(from_t, vec4_elem_t)                     \
  _CLC_DEF vec4_elem_t##4 __clc_cast_from_##from_t##2_to_##vec4_elem_t##4(     \
      from_t##2 from) {                                                        \
    return (vec4_elem_t##4)(from.x, from.y, 0, 0);                             \
  }
#define _CLC_DEFINE_CAST_SCALAR_TO_VEC4(from_t, vec4_elem_t)                   \
  _CLC_DEF vec4_elem_t##4 __clc_cast_from_##from_t##_to_##vec4_elem_t##4(      \
      from_t from) {                                                           \
    return (vec4_elem_t##4)(from, 0, 0, 0);                                    \
  }

// Helpers to extract N channel(s) from a four-channel (RGBA/XYZW) color type.

#define _CLC_DEFINE_EXTRACT_SAME_SIZE_COLOR_HELPERS(from_t, to_t)              \
  _CLC_DEFINE_CAST_VEC4(from_t, to_t)                                          \
  _CLC_DEFINE_BUILTIN_CAST_VEC4_TO_VEC3(from_t, to_t)                          \
  _CLC_DEFINE_BUILTIN_CAST_VEC4_TO_VEC2(from_t, to_t)                          \
  _CLC_DEFINE_BUILTIN_CAST_VEC4_TO_SCALAR(from_t, to_t)                        \
  _CLC_DEFINE_BUILTIN_CAST_VEC2_TO_VEC4(from_t, to_t)                          \
  _CLC_DEFINE_BUILTIN_CAST_VEC3_TO_VEC4(from_t, to_t)                          \
  _CLC_DEFINE_BUILTIN_CAST_SCALAR_TO_VEC4(from_t, to_t)

#define _CLC_DEFINE_EXTRACT_COLOR_HELPERS(from_t, to_t)                        \
  _CLC_DEFINE_CAST_VEC4(from_t, to_t)                                          \
  _CLC_DEFINE_CAST_VEC4_TO_VEC3(from_t, to_t)                                  \
  _CLC_DEFINE_CAST_VEC4_TO_VEC2(from_t, to_t)                                  \
  _CLC_DEFINE_CAST_VEC4_TO_SCALAR(from_t, to_t)                                \
  _CLC_DEFINE_CAST_VEC2_TO_VEC4(from_t, to_t)                                  \
  _CLC_DEFINE_CAST_VEC3_TO_VEC4(from_t, to_t)                                  \
  _CLC_DEFINE_CAST_SCALAR_TO_VEC4(from_t, to_t)

// Define casts between supported builtin types for image color

_CLC_DEFINE_EXTRACT_SAME_SIZE_COLOR_HELPERS(float, float)
_CLC_DEFINE_EXTRACT_SAME_SIZE_COLOR_HELPERS(float, int)
_CLC_DEFINE_EXTRACT_SAME_SIZE_COLOR_HELPERS(int, float)
_CLC_DEFINE_EXTRACT_SAME_SIZE_COLOR_HELPERS(float, uint)
_CLC_DEFINE_EXTRACT_SAME_SIZE_COLOR_HELPERS(uint, float)
#ifdef cl_khr_fp16
_CLC_DEFINE_EXTRACT_SAME_SIZE_COLOR_HELPERS(half, half)
_CLC_DEFINE_EXTRACT_SAME_SIZE_COLOR_HELPERS(half, short)
_CLC_DEFINE_EXTRACT_SAME_SIZE_COLOR_HELPERS(short, half)
_CLC_DEFINE_EXTRACT_SAME_SIZE_COLOR_HELPERS(half, ushort)
_CLC_DEFINE_EXTRACT_SAME_SIZE_COLOR_HELPERS(ushort, half)
#endif

_CLC_DEFINE_EXTRACT_COLOR_HELPERS(float, short)
_CLC_DEFINE_EXTRACT_COLOR_HELPERS(short, float)
_CLC_DEFINE_EXTRACT_COLOR_HELPERS(float, ushort)
_CLC_DEFINE_EXTRACT_COLOR_HELPERS(ushort, float)
_CLC_DEFINE_EXTRACT_COLOR_HELPERS(float, char)
_CLC_DEFINE_EXTRACT_COLOR_HELPERS(char, float)
_CLC_DEFINE_EXTRACT_COLOR_HELPERS(float, uchar)
_CLC_DEFINE_EXTRACT_COLOR_HELPERS(uchar, float)
#ifdef cl_khr_fp16
_CLC_DEFINE_EXTRACT_COLOR_HELPERS(float, half)
_CLC_DEFINE_EXTRACT_COLOR_HELPERS(half, float)
_CLC_DEFINE_EXTRACT_COLOR_HELPERS(half, int)
_CLC_DEFINE_EXTRACT_COLOR_HELPERS(int, half)
_CLC_DEFINE_EXTRACT_COLOR_HELPERS(half, uint)
_CLC_DEFINE_EXTRACT_COLOR_HELPERS(uint, half)
_CLC_DEFINE_EXTRACT_COLOR_HELPERS(half, char)
_CLC_DEFINE_EXTRACT_COLOR_HELPERS(char, half)
_CLC_DEFINE_EXTRACT_COLOR_HELPERS(half, uchar)
_CLC_DEFINE_EXTRACT_COLOR_HELPERS(uchar, half)
#endif

#undef _CLC_DEFINE_EXTRACT_COLOR_HELPERS
#undef _CLC_DEFINE_EXTRACT_SAME_SIZE_COLOR_HELPERS

#undef _CLC_DEFINE_CAST_SCALAR_TO_VEC4
#undef _CLC_DEFINE_CAST_VEC2_TO_VEC4
#undef _CLC_DEFINE_CAST_VEC3_TO_VEC4
#undef _CLC_DEFINE_CAST_VEC4_TO_SCALAR
#undef _CLC_DEFINE_CAST_VEC4_TO_VEC3
#undef _CLC_DEFINE_CAST_VEC4_TO_VEC2
#undef _CLC_DEFINE_CAST_VEC4
#undef _CLC_DEFINE_BUILTIN_CAST_SCALAR_TO_VEC4
#undef _CLC_DEFINE_BUILTIN_CAST_VEC2_TO_VEC4
#undef _CLC_DEFINE_BUILTIN_VEC3_TO_VEC4
#undef _CLC_DEFINE_BUILTIN_VEC4_TO_SCALAR
#undef _CLC_DEFINE_BUILTIN_VEC4_TO_VEC3
#undef _CLC_DEFINE_BUILTIN_VEC4_TO_VEC2
#undef _CLC_DEFINE_BUILTIN_VEC4