From 1be12c2a97c21ae11bb84a41d61d2be038433c5b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Th=C3=A9o=20LUDWIG?= <contact@theoludwig.fr>
Date: Wed, 25 Sep 2024 14:34:33 +0200
Subject: [PATCH] perf: allow to init `array_list` with initial capacity

Fixes #8
---
 lib/array_list.c | 13 +++++++++----
 lib/array_list.h |  7 +++++++
 2 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/lib/array_list.c b/lib/array_list.c
index c475cda..519b373 100644
--- a/lib/array_list.c
+++ b/lib/array_list.c
@@ -1,25 +1,30 @@
 #include "array_list.h"
 
 struct array_list* array_list_initialization() {
+  return array_list_initialization_with_capacity(ARRAY_LIST_INITIAL_CAPACITY);
+}
+
+struct array_list* array_list_initialization_with_capacity(size_t capacity) {
   struct array_list* list = malloc(sizeof(struct array_list));
   if (list == NULL) {
     perror("Error (array_list_initialization)");
     exit(EXIT_FAILURE);
   }
-  list->data = malloc(sizeof(void*) * ARRAY_LIST_INITIAL_CAPACITY);
+  list->size = 0;
+  list->capacity = capacity;
+  list->capacity_step = capacity;
+  list->data = malloc(sizeof(void*) * list->capacity);
   if (list->data == NULL) {
     perror("Error (array_list_initialization)");
     exit(EXIT_FAILURE);
   }
-  list->size = 0;
-  list->capacity = ARRAY_LIST_INITIAL_CAPACITY;
   return list;
 }
 
 void array_list_add(struct array_list* list, void* element) {
   if (list->size >= list->capacity) {
     size_t previous_capacity = list->capacity;
-    list->capacity += ARRAY_LIST_INITIAL_CAPACITY;
+    list->capacity += list->capacity_step;
     list->data = realloc(list->data, sizeof(void*) * list->capacity);
     if (list->data == NULL) {
       perror("Error (array_list_add)");
diff --git a/lib/array_list.h b/lib/array_list.h
index 1a183a3..4189cbd 100644
--- a/lib/array_list.h
+++ b/lib/array_list.h
@@ -18,6 +18,7 @@ struct array_list {
   void** data;
   size_t size;
   size_t capacity;
+  size_t capacity_step;
 };
 
 /**
@@ -26,6 +27,12 @@ struct array_list {
  */
 struct array_list* array_list_initialization();
 
+/**
+ * @brief Initializes a new array list with a capacity.
+ * @since v5.0.0
+ */
+struct array_list* array_list_initialization_with_capacity(size_t capacity);
+
 /**
  * @brief Adds an element to the end of the array list.
  *