diff --git a/Include/Fledasty/Core/LinkedList.h b/Include/Fledasty/Core/LinkedList.h index 88a7c22..9b70f0e 100644 --- a/Include/Fledasty/Core/LinkedList.h +++ b/Include/Fledasty/Core/LinkedList.h @@ -46,6 +46,7 @@ FledastyError fledasty_linked_list_append(FledastyLinkedList *current_linked_lis FledastyError fledasty_linked_list_insert_at_index(FledastyLinkedList *current_linked_list, const size_t index, void *value); FledastyError fledasty_linked_list_insert_before_value(FledastyLinkedList *current_linked_list, void *before_value, void *value); +FledastyError fledasty_linked_list_insert_after_value(FledastyLinkedList *current_linked_list, void *after_value, void *value); static inline bool fledasty_linked_list_is_empty(FledastyLinkedList *current_linked_list) { return current_linked_list->size == 0; } diff --git a/Src/Core/LinkedList.c b/Src/Core/LinkedList.c index 740acf8..0d7fc11 100644 --- a/Src/Core/LinkedList.c +++ b/Src/Core/LinkedList.c @@ -165,10 +165,43 @@ FledastyError fledasty_linked_list_insert_before_value(FledastyLinkedList *curre new_node->next = current_linked_list->start; current_linked_list->start = new_node; } else { - new_node->next = previous_node->next; + new_node->next = current_node; previous_node->next = new_node; } current_linked_list->size += 1; return FLEDASTY_ERROR_NONE; } + +FledastyError fledasty_linked_list_insert_after_value(FledastyLinkedList *current_linked_list, void *after_value, void *value) { + if (current_linked_list == NULL || after_value == NULL || value == NULL) { + return FLEDASTY_ERROR_INVALID_POINTER; + } + + FledastyLinkedListNode *current_node = current_linked_list->start; + while (current_node != NULL && !hallocy_compare_memory(current_node->value, after_value, current_linked_list->element_byte_size)) { + current_node = current_node->next; + } + + if (current_node == NULL) { + return FLEDASTY_ERROR_VALUE_NOT_FOUND; + } + + FledastyLinkedListNode *new_node = (FledastyLinkedListNode*)hallocy_malloc(sizeof(FledastyLinkedListNode)); + + new_node->value = hallocy_malloc(current_linked_list->element_byte_size); + hallocy_copy_memory(new_node->value, value, current_linked_list->element_byte_size); + + if (current_node == current_linked_list->end) { + new_node->next = NULL; + + current_linked_list->end->next = new_node; + current_linked_list->end = new_node; + } else { + new_node->next = current_node->next; + current_node->next = new_node; + } + + current_linked_list->size += 1; + return FLEDASTY_ERROR_NONE; +} diff --git a/Tests/Main.c b/Tests/Main.c index 83c847f..3115622 100644 --- a/Tests/Main.c +++ b/Tests/Main.c @@ -115,6 +115,8 @@ int main() { insert_value = 28; insert_at_value = 35; fledasty_linked_list_insert_before_value(&test_linked_list, &insert_at_value, &insert_value); + insert_value = 90; + fledasty_linked_list_insert_after_value(&test_linked_list, &insert_at_value, &insert_value); FledastyLinkedListNode *test_linked_list_node = test_linked_list.start; for (int i = 0; i < test_linked_list.size; i += 1) {