ucx/iterator.c

Sun, 06 Oct 2024 12:00:31 +0200

author
Olaf Wintermann <olaf.wintermann@gmail.com>
date
Sun, 06 Oct 2024 12:00:31 +0200
branch
newapi
changeset 324
ce13a778654a
permissions
-rw-r--r--

update ucx

324
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
1 /*
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
2 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
3 *
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
4 * Copyright 2024 Mike Becker, Olaf Wintermann All rights reserved.
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
5 *
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
6 * Redistribution and use in source and binary forms, with or without
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
7 * modification, are permitted provided that the following conditions are met:
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
8 *
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
9 * 1. Redistributions of source code must retain the above copyright
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
10 * notice, this list of conditions and the following disclaimer.
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
11 *
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
12 * 2. Redistributions in binary form must reproduce the above copyright
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
13 * notice, this list of conditions and the following disclaimer in the
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
14 * documentation and/or other materials provided with the distribution.
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
15 *
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
16 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
17 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
19 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
20 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
26 * POSSIBILITY OF SUCH DAMAGE.
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
27 */
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
28
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
29 #include "cx/iterator.h"
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
30
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
31 #include <string.h>
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
32
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
33 static bool cx_iter_valid(const void *it) {
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
34 const struct cx_iterator_s *iter = it;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
35 return iter->index < iter->elem_count;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
36 }
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
37
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
38 static void *cx_iter_current(const void *it) {
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
39 const struct cx_iterator_s *iter = it;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
40 return iter->elem_handle;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
41 }
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
42
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
43 static void cx_iter_next_fast(void *it) {
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
44 struct cx_iterator_s *iter = it;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
45 if (iter->base.remove) {
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
46 iter->base.remove = false;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
47 iter->elem_count--;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
48 // only move the last element when we are not currently aiming
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
49 // at the last element already
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
50 if (iter->index < iter->elem_count) {
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
51 void *last = ((char *) iter->src_handle.m)
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
52 + iter->elem_count * iter->elem_size;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
53 memcpy(iter->elem_handle, last, iter->elem_size);
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
54 }
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
55 } else {
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
56 iter->index++;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
57 iter->elem_handle = ((char *) iter->elem_handle) + iter->elem_size;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
58 }
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
59 }
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
60
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
61 static void cx_iter_next_slow(void *it) {
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
62 struct cx_iterator_s *iter = it;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
63 if (iter->base.remove) {
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
64 iter->base.remove = false;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
65 iter->elem_count--;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
66
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
67 // number of elements to move
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
68 size_t remaining = iter->elem_count - iter->index;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
69 if (remaining > 0) {
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
70 memmove(
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
71 iter->elem_handle,
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
72 ((char *) iter->elem_handle) + iter->elem_size,
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
73 remaining * iter->elem_size
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
74 );
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
75 }
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
76 } else {
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
77 iter->index++;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
78 iter->elem_handle = ((char *) iter->elem_handle) + iter->elem_size;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
79 }
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
80 }
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
81
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
82 CxIterator cxMutIterator(
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
83 void *array,
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
84 size_t elem_size,
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
85 size_t elem_count,
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
86 bool remove_keeps_order
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
87 ) {
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
88 CxIterator iter;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
89
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
90 iter.index = 0;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
91 iter.src_handle.m = array;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
92 iter.elem_handle = array;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
93 iter.elem_size = elem_size;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
94 iter.elem_count = array == NULL ? 0 : elem_count;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
95 iter.base.valid = cx_iter_valid;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
96 iter.base.current = cx_iter_current;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
97 iter.base.next = remove_keeps_order ? cx_iter_next_slow : cx_iter_next_fast;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
98 iter.base.remove = false;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
99 iter.base.mutating = true;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
100
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
101 return iter;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
102 }
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
103
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
104 CxIterator cxIterator(
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
105 const void *array,
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
106 size_t elem_size,
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
107 size_t elem_count
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
108 ) {
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
109 CxIterator iter = cxMutIterator((void*)array, elem_size, elem_count, false);
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
110 iter.base.mutating = false;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
111 return iter;
ce13a778654a update ucx
Olaf Wintermann <olaf.wintermann@gmail.com>
parents:
diff changeset
112 }

mercurial