Generic.h 7.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124
  1. #ifndef CORE_GENERIC_H
  2. #define CORE_GENERIC_H
  3. #include "core/Matrix.h"
  4. // clang-format off
  5. #define GENERIC_PAIR(type, name) type: name, const type : name
  6. // clang-format on
  7. #define GENERIC_FLOAT_VECTOR(a, name) \
  8. _Generic((a), \
  9. GENERIC_PAIR(Vector2*, name##V2), \
  10. GENERIC_PAIR(Vector3*, name##V3), \
  11. GENERIC_PAIR(Vector4*, name##V4))
  12. #define GENERIC_VECTOR(a, name) \
  13. _Generic((a), \
  14. GENERIC_PAIR(Vector2*, name##V2), \
  15. GENERIC_PAIR(Vector3*, name##V3), \
  16. GENERIC_PAIR(Vector4*, name##V4), \
  17. GENERIC_PAIR(IntVector2*, name##IV2), \
  18. GENERIC_PAIR(IntVector3*, name##IV3), \
  19. GENERIC_PAIR(IntVector4*, name##IV4))
  20. #define ZERO_VECTOR(a) \
  21. _Generic((a), \
  22. GENERIC_PAIR(Matrix*, &(Vector3){0}), \
  23. GENERIC_PAIR(Vector2*, &(Vector2){0}), \
  24. GENERIC_PAIR(Vector3*, &(Vector3){0}), \
  25. GENERIC_PAIR(Vector4*, &(Vector4){0}), \
  26. GENERIC_PAIR(IntVector2*, &(IntVector2){0}), \
  27. GENERIC_PAIR(IntVector3*, &(IntVector3){0}), \
  28. GENERIC_PAIR(IntVector4*, &(IntVector4){0}))
  29. #define GENERIC_FACTOR(b, type, base, factor) \
  30. type: \
  31. _Generic((b), GENERIC_PAIR(type, base), default: factor)
  32. #define SELECT_OP(_1, _2, _3, name, ...) name
  33. #define addSet(a, b) GENERIC_VECTOR(a, addSet)(a, b)
  34. #define add3(a, b, c) GENERIC_VECTOR(a, add)(a, b, c)
  35. #define add2(a, b) add3(ZERO_VECTOR(a), a, b)
  36. #define add(...) SELECT_OP(__VA_ARGS__, add3, add2, 0)(__VA_ARGS__)
  37. #define subSet(a, b) GENERIC_VECTOR(a, subSet)(a, b)
  38. #define sub3(a, b, c) GENERIC_VECTOR(a, sub)(a, b, c)
  39. #define sub2(a, b) sub3(ZERO_VECTOR(a), a, b)
  40. #define sub(...) SELECT_OP(__VA_ARGS__, sub3, sub2, 0)(__VA_ARGS__)
  41. #define mulSet(a, b) \
  42. _Generic((a), \
  43. Matrix *: mulSetMatrix, \
  44. Quaternion *: mulSetQ, \
  45. GENERIC_FACTOR(b, Vector2*, mulSetV2, mulSetV2F), \
  46. GENERIC_FACTOR(b, Vector3*, mulSetV3, mulSetV3F), \
  47. GENERIC_FACTOR(b, Vector4*, mulSetV4, mulSetV4F), \
  48. GENERIC_FACTOR(b, IntVector2*, mulSetIV2, mulSetIV2F), \
  49. GENERIC_FACTOR(b, IntVector3*, mulSetIV3, mulSetIV3F), \
  50. GENERIC_FACTOR(b, IntVector4*, mulSetIV4, mulSetIV4F))(a, b)
  51. #define mul3(a, b, c) \
  52. _Generic((a), \
  53. Matrix *: mulMatrix, \
  54. Quaternion *: mulQ, \
  55. GENERIC_FACTOR(c, Vector2*, mulV2, mulV2F), \
  56. Vector3 *: _Generic((c), \
  57. Vector3 *: _Generic((b), \
  58. GENERIC_PAIR(Quaternion*, mulQV3), \
  59. GENERIC_PAIR(Matrix*, mulMatrixV3), \
  60. default: mulV3), \
  61. default: mulV3F), \
  62. GENERIC_FACTOR(c, Vector4*, mulV4, mulV4F), \
  63. GENERIC_FACTOR(c, IntVector2*, mulIV2, mulIV2F), \
  64. GENERIC_FACTOR(c, IntVector3*, mulIV3, mulIV3F), \
  65. GENERIC_FACTOR(c, IntVector4*, mulIV4, mulIV4F))(a, b, c)
  66. #define mul2(a, b) mul3(ZERO_VECTOR(a), a, b)
  67. #define mul(...) SELECT_OP(__VA_ARGS__, mul3, mul2, 0)(__VA_ARGS__)
  68. #define divSet(a, b) \
  69. _Generic((a), \
  70. GENERIC_FACTOR(b, Vector2*, divSetV2, divSetV2F), \
  71. GENERIC_FACTOR(b, Vector3*, divSetV3, divSetV3F), \
  72. GENERIC_FACTOR(b, Vector4*, divSetV4, divSetV4F), \
  73. GENERIC_FACTOR(b, IntVector2*, divSetIV2, divSetIV2F), \
  74. GENERIC_FACTOR(b, IntVector3*, divSetIV3, divSetIV3F), \
  75. GENERIC_FACTOR(b, IntVector4*, divSetIV4, divSetIV4F))(a, b)
  76. #define div3(a, b, c) \
  77. _Generic((a), \
  78. GENERIC_FACTOR(c, Vector2*, divV2, divV2F), \
  79. GENERIC_FACTOR(c, Vector3*, divV3, divV3F), \
  80. GENERIC_FACTOR(c, Vector4*, divV4, divV4F), \
  81. GENERIC_FACTOR(c, IntVector2*, divIV2, divIV2F), \
  82. GENERIC_FACTOR(c, IntVector3*, divIV3, divIV3F), \
  83. GENERIC_FACTOR(c, IntVector4*, divIV4, divIV4F))(a, b, c)
  84. #define div2(a, b) div3(ZERO_VECTOR(a), a, b)
  85. #define div(...) SELECT_OP(__VA_ARGS__, div3, div2, 0)(__VA_ARGS__)
  86. #define invertSet(a) GENERIC_VECTOR(a, invertSet)(a)
  87. #define invert2(a, b) GENERIC_VECTOR(a, invert)(a, b)
  88. #define invert1(a) invert2(ZERO_VECTOR(a), a)
  89. #define invert(...) SELECT_OP(0, __VA_ARGS__, invert2, invert1, 0)(__VA_ARGS__)
  90. #define dot(a, b) GENERIC_FLOAT_VECTOR(a, dot)(a, b)
  91. #define squareLength(a) GENERIC_FLOAT_VECTOR(a, squareLength)(a)
  92. #define length(a) GENERIC_FLOAT_VECTOR(a, length)(a)
  93. #define normalize(a) GENERIC_FLOAT_VECTOR(a, normalize)(a)
  94. #undef cross
  95. #define SELECT_OP(_1, _2, _3, name, ...) name
  96. #define cross3(a, b, c) cross(a, b, c)
  97. #define cross2(a, b) cross(&(Vector3){0}, a, b)
  98. #define cross(...) SELECT_OP(__VA_ARGS__, cross3, cross2, 0)(__VA_ARGS__)
  99. #define convert(a, b) \
  100. _Generic((a), \
  101. Vector2 *: convertIV2, \
  102. Vector3 *: convertIV3, \
  103. Vector4 *: convertIV4, \
  104. IntVector2 *: convertV2, \
  105. IntVector3 *: convertV3, \
  106. IntVector4 *: convertV4)(a, b)
  107. #endif