Add [raw]memelem{cpy,move}

Signed-off-by: Mattias Andrée <maandree@kth.se>

1 files changed, 94 insertions, 0 deletions

diff --git a/memelemcpy.c b/memelemcpy.c
new file mode 100644
index 0000000..da32946
--- /dev/null
+++ b/memelemcpy.c
@@ -0,0 +1,94 @@
+/* See LICENSE file for copyright and license details. */
+#include "libsimple.h"
+#ifndef TEST
+
+
+static inline uint8_t *
+memelemcpy8(uint8_t *restrict d, const uint8_t *restrict s, uint8_t elem, size_t n)
+{
+	for (; n; s++, n--)
+		if ((*d++ = *s) == elem)
+			return d;
+	return NULL;
+}
+
+
+static inline uint16_t *
+memelemcpy16(uint16_t *restrict d, const uint16_t *restrict s, uint16_t elem, size_t n)
+{
+	for (; n; s++, n--)
+		if ((*d++ = *s) == elem)
+			return d;
+	return NULL;
+}
+
+
+static inline uint32_t *
+memelemcpy32(uint32_t *restrict d, const uint32_t *restrict s, uint32_t elem, size_t n)
+{
+	for (; n; s++, n--)
+		if ((*d++ = *s) == elem)
+			return d;
+	return NULL;
+}
+
+
+static inline uint64_t *
+memelemcpy64(uint64_t *restrict d, const uint64_t *restrict s, uint64_t elem, size_t n)
+{
+	for (; n; s++, n--)
+		if ((*d++ = *s) == elem)
+			return d;
+	return NULL;
+}
+
+
+static inline char *
+memelemcpyx(char *restrict d, const char *restrict s, const char *restrict elem, size_t size, size_t n)
+{
+	size_t i;
+	for (; n; s += size, n--) {
+		for (i = 0; i < size; i++)
+			d[i] = s[i];
+		for (i = 0; i < size; i++)
+			if (d[i] != elem[i])
+				goto next;
+		d += size;
+		return d;
+	next:
+		d += size;
+	}
+	return NULL;
+}
+
+
+void *
+libsimple_memelemcpy(void *restrict d, const void *restrict s, const void *restrict elem, size_t size, size_t n) /* TODO test, man */
+{
+	switch (size) {
+	case 0:
+		return d;
+	case 1:
+		return memelemcpy8(d, s, *(const uint8_t *)elem, n);
+	case 2:
+		return memelemcpy16(d, s, *(const uint16_t *)elem, n);
+	case 4:
+		return memelemcpy32(d, s, *(const uint32_t *)elem, n);
+	case 8:
+		return memelemcpy64(d, s, *(const uint64_t *)elem, n);
+	default:
+		return memelemcpyx(d, s, elem, size, n);
+	}
+}
+
+
+#else
+#include "test.h"
+
+int
+main(void)
+{
+	return 0;
+}
+
+#endif