add kai_text_to_reals

3 files changed, 81 insertions, 0 deletions

diff --git a/src/util/util.c b/src/util/util.c
index c77b3f4..d3e8742 100644
--- a/src/util/util.c
+++ b/src/util/util.c
@@ -21,3 +21,22 @@ void *kai_expand_array(void **array, size_t *count, size_t element_size)
 	*count = new_count;
 	return new_element;
 }
+
+
+int kai_text_to_reals(ka_real_t *dest, const char *str, size_t count)
+{
+	char *nptr = (char*) str;
+	char *end;
+
+	int result = count;
+
+	for (int i=0; i<count; i++) {
+		dest[i] = KA_STR_TO_REAL(nptr, &end);
+		if (nptr == end) {
+			result -= 1;
+		}
+		nptr = end;
+	}
+
+	return result;
+}
diff --git a/src/util/util.h b/src/util/util.h
index c2f01eb..d6cdd20 100644
--- a/src/util/util.h
+++ b/src/util/util.h
@@ -1,6 +1,34 @@
 #ifndef KALMIA_UTIL_H
 #define KALMIA_UTIL_H
 
+#include <kalmia.h>
+
+
+/** @brief add a new element to an array
+ *
+ * @param array  Pointer to the array to expand
+ * @param len    Pointer to the current length of the array
+ * @param element_size  Size of each element in the array
+ *
+ * @returns  A pointer to the newly added element of the array, or NULL on failure.
+ */
 void * kai_expand_array(void **array, size_t *len, size_t element_size);
 
+
+/** @brief Convert a string to an array of reals
+ *
+ * This function will always fill the array with `count` elements. If there are not
+ * `count` elements to be converted in the string, it will fill the rest of the array
+ * with zeroes.
+ *
+ * @param dest  Array of reals in which to store the results
+ * @param str   The string to convert
+ * @param count The number of elements to convert
+ *
+ * @returns  The number of conversions performed. If there were at least `count` elements
+ * in the string to convert, this is the same as `count`. If not, it is the number of
+ * actual elements present in the string.
+ */
+int kai_text_to_reals(ka_real_t *dest, const char *str, size_t count);
+
 #endif
diff --git a/src/util/util.test.c b/src/util/util.test.c
index 9f64085..26aa3d1 100644
--- a/src/util/util.test.c
+++ b/src/util/util.test.c
@@ -57,5 +57,39 @@ LILY_TEST("expand array 100 times")
 #include LILY_PUSH_TEST()
 
 
+LILY_TEST("convert string to real array")
+{
+	const char *str = "0 0.5 1 1.5 2";
+	ka_real_t array[5];
+	int conv = kai_text_to_reals(array, str, 5);
+
+	CHECK_EQ(conv, 5, "%d");
+
+	CHECK_EQF(array[0], 0.0f, "%f");
+	CHECK_EQF(array[1], 0.5f, "%f");
+	CHECK_EQF(array[2], 1.0f, "%f");
+	CHECK_EQF(array[3], 1.5f, "%f");
+	CHECK_EQF(array[4], 2.0f, "%f");
+}
+#include LILY_PUSH_TEST()
+
+
+LILY_TEST("convert string to real array with fewer conversions than expected")
+{
+	const char *str = "0 0.5 1";
+	ka_real_t array[5];
+	int conv = kai_text_to_reals(array, str, 5);
+
+	CHECK_EQ(conv, 3, "%d");
+
+	CHECK_EQF(array[0], 0.0f, "%f");
+	CHECK_EQF(array[1], 0.5f, "%f");
+	CHECK_EQF(array[2], 1.0f, "%f");
+	CHECK_EQF(array[3], 0.0f, "%f");
+	CHECK_EQF(array[4], 0.0f, "%f");
+}
+#include LILY_PUSH_TEST()
+
+
 #define LILY_FILE_END
 #include LILY_REGISTER_TESTS()