DArray.h 7.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155
  1. #ifndef SCUFCONT_DARRAY_H_
  2. #define SCUFCONT_DARRAY_H_
  3. #include <stdbool.h>
  4. #include <stddef.h>
  5. #include <stdlib.h> // IWYU pragma: keep
  6. #include <string.h>
  7. #include "scufcont.h" // IWYU pragma: keep
  8. // #define SCUFCONT_BOUNDS_CHECK
  9. #ifdef SCUFCONT_BOUNDS_CHECK
  10. #include <stdio.h>
  11. #endif // SCUFCONT_BOUNDS_CHECK
  12. #define SCDARRAY_EXPANDER(macro, type, type_name, func_name) macro(type, SCDArray_##type_name, SCDArray_##type_name##_##func_name)
  13. #define SCDARRAY_IMPL(type, type_name) \
  14. SCDARRAY_EXPANDER(SCDARRAY_ALLOC, type, type_name, alloc) \
  15. SCDARRAY_EXPANDER(SCDARRAY_FREE, type, type_name, free) \
  16. SCDARRAY_EXPANDER(SCDARRAY_IS_NULL, type, type_name, is_null) \
  17. SCDARRAY_EXPANDER(SCDARRAY_PUSH_BACK, type, type_name, push_back) \
  18. SCDARRAY_EXPANDER(SCDARRAY_PUSH_BACK_MULTIPLE, type, type_name, push_back_multiple) \
  19. SCDARRAY_EXPANDER(SCDARRAY_AT, type, type_name, at) \
  20. SCDARRAY_EXPANDER(SCDARRAY_REMOVE, type, type_name, remove)
  21. #define SCDARRAY_HEADER(type, type_name) \
  22. SCDARRAY_STRUCT(type, type_name) \
  23. SCDARRAY_SIGS(type, type_name)
  24. #define SCDARRAY_SIGS(type, type_name) \
  25. SCDARRAY_EXPANDER(SCDARRAY_ALLOC_SIG, type, type_name, alloc); \
  26. SCDARRAY_EXPANDER(SCDARRAY_FREE_SIG, type, type_name, free); \
  27. SCDARRAY_EXPANDER(SCDARRAY_IS_NULL_SIG, type, type_name, is_null); \
  28. SCDARRAY_EXPANDER(SCDARRAY_PUSH_BACK_SIG, type, type_name, push_back); \
  29. SCDARRAY_EXPANDER(SCDARRAY_PUSH_BACK_MULTIPLE_SIG, type, type_name, push_back_multiple); \
  30. SCDARRAY_EXPANDER(SCDARRAY_AT_SIG, type, type_name, at); \
  31. SCDARRAY_EXPANDER(SCDARRAY_REMOVE_SIG, type, type_name, remove);
  32. // SCDARRAY_ALLOC_SIG(type, SCDArray_##type_name, SCDArray_##type_name##_alloc); \
  33. // SCDARRAY_PUSH_BACK_SIG(type, SCDArray_##type_name, SCDArray_##type_name##_push_back);
  34. #define SCDARRAY_STRUCT(type, type_name) \
  35. typedef struct { \
  36. type* data; \
  37. size_t len; \
  38. size_t capacity; \
  39. } SCDArray_##type_name;
  40. // SCDARRAY_STRUCT(int);
  41. #define SCDARRAY_ALLOC_SIG(type, type_name, func_name) ScufContResult func_name(type_name* darr)
  42. #define SCDARRAY_ALLOC(type, type_name, func_name) \
  43. SCDARRAY_ALLOC_SIG(type, type_name, func_name) { \
  44. darr->data = (type*) calloc(1, sizeof(type)); \
  45. if (darr->data == NULL) { \
  46. return SCUFCONT_OUT_OF_MEMORY; \
  47. } \
  48. darr->len = 0; \
  49. darr->capacity = 1; \
  50. return SCUFCONT_SUCCESS; \
  51. }
  52. #define SCDARRAY_FREE_SIG(type, type_name, func_name) void func_name(type_name* darr)
  53. #define SCDARRAY_FREE(type, type_name, func_name) \
  54. SCDARRAY_FREE_SIG(type, type_name, func_name) { \
  55. free(darr->data); \
  56. }
  57. #define SCDARRAY_IS_NULL_SIG(type, type_name, func_name) bool func_name(const type_name* darr)
  58. #define SCDARRAY_IS_NULL(type, type_name, func_name) \
  59. SCDARRAY_IS_NULL_SIG(type, type_name, func_name) { \
  60. return darr->data == NULL; \
  61. }
  62. // SCDARRAY_ALLOC(int, SCDArray_int, SCDArray_int_alloc);
  63. #define SCDARRAY_PUSH_BACK_SIG(type, type_name, func_name) ScufContResult func_name(type_name* darr, const type* new_elem)
  64. #define SCDARRAY_PUSH_BACK(type, type_name, func_name) \
  65. SCDARRAY_PUSH_BACK_SIG(type, type_name, func_name) { \
  66. if (darr->len + 1 > darr->capacity) { \
  67. darr->capacity *= 2; \
  68. darr->data = (type*) realloc(darr->data, darr->capacity * sizeof(*new_elem)); \
  69. if (darr->data == NULL) { \
  70. return SCUFCONT_OUT_OF_MEMORY; \
  71. } \
  72. } \
  73. memcpy(darr->data + darr->len, new_elem, sizeof(*new_elem)); \
  74. darr->len++; \
  75. return SCUFCONT_SUCCESS; \
  76. }
  77. #define SCDARRAY_PUSH_BACK_MULTIPLE_SIG(type, type_name, func_name) ScufContResult func_name(type_name* darr, const type* new_elems, const size_t new_elems_n)
  78. #define SCDARRAY_PUSH_BACK_MULTIPLE(type, type_name, func_name) \
  79. SCDARRAY_PUSH_BACK_MULTIPLE_SIG(type, type_name, func_name) { \
  80. if (darr->len + new_elems_n > darr->capacity) { \
  81. darr->data = (type*) realloc(darr->data, (darr->capacity + new_elems_n) * sizeof(type)); \
  82. if (darr->data == NULL) { \
  83. return SCUFCONT_OUT_OF_MEMORY; \
  84. } \
  85. darr->capacity += new_elems_n; \
  86. } \
  87. memcpy(darr->data + darr->len, new_elems, sizeof(type) * new_elems_n); \
  88. darr->len += new_elems_n; \
  89. return SCUFCONT_SUCCESS; \
  90. }
  91. // SCDARRAY_PUSH_BACK_MULTIPLE(int, SCDArray_int, SCDArray_int_push_back_multiple)
  92. #define SCDARRAY_AT_SIG(type, type_name, func_name) type* func_name(type_name* darr, const size_t i)
  93. // SCDARRAY_PUSH_BACK(int, SCDArray_int, SCDArray_int_push_back);
  94. #ifdef SCUFCONT_BOUNDS_CHECK
  95. #define SCDARRAY_AT(type, type_name, func_name) \
  96. SCDARRAY_AT_SIG(type, type_name, func_name) { \
  97. if (i < 0 || i > darr->len) { \
  98. fprintf(stderr, #func_name " out-of-bounds. i: %zu, len: %zu", i, darr->len); \
  99. exit(1); \
  100. } \
  101. \
  102. return darr->data + i; \
  103. }
  104. #else
  105. #define SCDARRAY_AT(type, type_name, func_name) \
  106. SCDARRAY_AT_SIG(type, type_name, func_name) { \
  107. return darr->data + i; \
  108. }
  109. #endif // SCUFCONT_BOUNDS_CHECK
  110. // SCDARRAY_AT(char*, SCDArray_charp, SCDArray_charp_at)
  111. #define SCDARRAY_REMOVE_SIG(type, type_name, func_name) void func_name(type_name* darr, const size_t i)
  112. #define SCDARRAY_REMOVE(type, type_name, func_name) \
  113. SCDARRAY_REMOVE_SIG(type, type_name, func_name) { \
  114. memmove(darr->data + i, darr->data + i + 1, (darr->len - i) * sizeof(type)); \
  115. darr->len--; \
  116. }
  117. // SCDARRAY_REMOVE(char*, SCDArray_charp, SCDArray_charp_remove)
  118. #endif // SCUFCONT_DARRAY_H_