2 files changed, 217 insertions, 0 deletions
diff --git a/source/blender/io/common/intern/string_utils.cc b/source/blender/io/common/intern/string_utils.cc
new file mode 100644
index 00000000000..01107b0866e
--- /dev/null
+++ b/source/blender/io/common/intern/string_utils.cc
@@ -0,0 +1,99 @@
+/* SPDX-License-Identifier: GPL-2.0-or-later */
+
+#include "IO_string_utils.hh"
+
+/* Note: we could use C++17 <charconv> from_chars to parse
+ * floats, but even if some compilers claim full support,
+ * their standard libraries are not quite there yet.
+ * LLVM/libc++ only has a float parser since LLVM 14,
+ * and gcc/libstdc++ since 11.1. So until at least these are
+ * the mininum spec, use an external library. */
+#include "fast_float.h"
+#include <charconv>
+
+namespace blender::io {
+
+StringRef read_next_line(StringRef &buffer)
+{
+  const char *start = buffer.begin();
+  const char *end = buffer.end();
+  size_t len = 0;
+  char prev = 0;
+  const char *ptr = start;
+  while (ptr < end) {
+    char c = *ptr++;
+    if (c == '\n' && prev != '\\') {
+      break;
+    }
+    prev = c;
+    ++len;
+  }
+
+  buffer = StringRef(ptr, end);
+  return StringRef(start, len);
+}
+
+static bool is_whitespace(char c)
+{
+  return c <= ' ' || c == '\\';
+}
+
+StringRef drop_whitespace(StringRef str)
+{
+  while (!str.is_empty() && is_whitespace(str[0])) {
+    str = str.drop_prefix(1);
+  }
+  return str;
+}
+
+StringRef drop_non_whitespace(StringRef str)
+{
+  while (!str.is_empty() && !is_whitespace(str[0])) {
+    str = str.drop_prefix(1);
+  }
+  return str;
+}
+
+static StringRef drop_plus(StringRef str)
+{
+  if (!str.is_empty() && str[0] == '+') {
+    str = str.drop_prefix(1);
+  }
+  return str;
+}
+
+StringRef parse_float(StringRef str, float fallback, float &dst, bool skip_space)
+{
+  if (skip_space) {
+    str = drop_whitespace(str);
+  }
+  str = drop_plus(str);
+  fast_float::from_chars_result res = fast_float::from_chars(str.begin(), str.end(), dst);
+  if (res.ec == std::errc::invalid_argument || res.ec == std::errc::result_out_of_range) {
+    dst = fallback;
+  }
+  return StringRef(res.ptr, str.end());
+}
+
+StringRef parse_floats(StringRef str, float fallback, float *dst, int count)
+{
+  for (int i = 0; i < count; ++i) {
+    str = parse_float(str, fallback, dst[i]);
+  }
+  return str;
+}
+
+StringRef parse_int(StringRef str, int fallback, int &dst, bool skip_space)
+{
+  if (skip_space) {
+    str = drop_whitespace(str);
+  }
+  str = drop_plus(str);
+  std::from_chars_result res = std::from_chars(str.begin(), str.end(), dst);
+  if (res.ec == std::errc::invalid_argument || res.ec == std::errc::result_out_of_range) {
+    dst = fallback;
+  }
+  return StringRef(res.ptr, str.end());
+}
+
+}  // namespace blender::io
diff --git a/source/blender/io/common/intern/string_utils_test.cc b/source/blender/io/common/intern/string_utils_test.cc
new file mode 100644
index 00000000000..a78bd7ab8a3
--- /dev/null
+++ b/source/blender/io/common/intern/string_utils_test.cc
@@ -0,0 +1,118 @@
+/* SPDX-License-Identifier: Apache-2.0 */
+
+#include "IO_string_utils.hh"
+
+#include "testing/testing.h"
+
+namespace blender::io {
+
+#define EXPECT_STRREF_EQ(str1, str2) EXPECT_STREQ(str1, std::string(str2).c_str())
+
+TEST(string_utils, read_next_line)
+{
+  std::string str = "abc\n  \n\nline with \\\ncontinuation\nCRLF ending:\r\na";
+  StringRef s = str;
+  EXPECT_STRREF_EQ("abc", read_next_line(s));
+  EXPECT_STRREF_EQ("  ", read_next_line(s));
+  EXPECT_STRREF_EQ("", read_next_line(s));
+  EXPECT_STRREF_EQ("line with \\\ncontinuation", read_next_line(s));
+  EXPECT_STRREF_EQ("CRLF ending:\r", read_next_line(s));
+  EXPECT_STRREF_EQ("a", read_next_line(s));
+  EXPECT_TRUE(s.is_empty());
+}
+
+TEST(string_utils, drop_whitespace)
+{
+  /* Empty */
+  EXPECT_STRREF_EQ("", drop_whitespace(""));
+  /* Only whitespace */
+  EXPECT_STRREF_EQ("", drop_whitespace(" "));
+  EXPECT_STRREF_EQ("", drop_whitespace("   "));
+  EXPECT_STRREF_EQ("", drop_whitespace(" \t\n\r "));
+  /* Drops leading whitespace */
+  EXPECT_STRREF_EQ("a", drop_whitespace(" a"));
+  EXPECT_STRREF_EQ("a b", drop_whitespace("   a b"));
+  EXPECT_STRREF_EQ("a b   ", drop_whitespace(" a b   "));
+  /* No leading whitespace */
+  EXPECT_STRREF_EQ("c", drop_whitespace("c"));
+  /* Case with backslash, should be treated as whitespace */
+  EXPECT_STRREF_EQ("d", drop_whitespace(" \\ d"));
+}
+
+TEST(string_utils, parse_int_valid)
+{
+  std::string str = "1 -10 \t  1234 1234567890 +7 123a";
+  StringRef s = str;
+  int val;
+  s = parse_int(s, 0, val);
+  EXPECT_EQ(1, val);
+  s = parse_int(s, 0, val);
+  EXPECT_EQ(-10, val);
+  s = parse_int(s, 0, val);
+  EXPECT_EQ(1234, val);
+  s = parse_int(s, 0, val);
+  EXPECT_EQ(1234567890, val);
+  s = parse_int(s, 0, val);
+  EXPECT_EQ(7, val);
+  s = parse_int(s, 0, val);
+  EXPECT_EQ(123, val);
+  EXPECT_STRREF_EQ("a", s);
+}
+
+TEST(string_utils, parse_int_invalid)
+{
+  int val;
+  /* Invalid syntax */
+  EXPECT_STRREF_EQ("--123", parse_int("--123", -1, val));
+  EXPECT_EQ(val, -1);
+  EXPECT_STRREF_EQ("foobar", parse_int("foobar", -2, val));
+  EXPECT_EQ(val, -2);
+  /* Out of integer range */
+  EXPECT_STRREF_EQ(" a", parse_int("1234567890123 a", -3, val));
+  EXPECT_EQ(val, -3);
+  /* Has leading white-space when we don't expect it */
+  EXPECT_STRREF_EQ(" 1", parse_int(" 1", -4, val, false));
+  EXPECT_EQ(val, -4);
+}
+
+TEST(string_utils, parse_float_valid)
+{
+  std::string str = "1 -10 123.5 -17.125 0.1 1e6 50.0e-1";
+  StringRef s = str;
+  float val;
+  s = parse_float(s, 0, val);
+  EXPECT_EQ(1.0f, val);
+  s = parse_float(s, 0, val);
+  EXPECT_EQ(-10.0f, val);
+  s = parse_float(s, 0, val);
+  EXPECT_EQ(123.5f, val);
+  s = parse_float(s, 0, val);
+  EXPECT_EQ(-17.125f, val);
+  s = parse_float(s, 0, val);
+  EXPECT_EQ(0.1f, val);
+  s = parse_float(s, 0, val);
+  EXPECT_EQ(1.0e6f, val);
+  s = parse_float(s, 0, val);
+  EXPECT_EQ(5.0f, val);
+  EXPECT_TRUE(s.is_empty());
+}
+
+TEST(string_utils, parse_float_invalid)
+{
+  float val;
+  /* Invalid syntax */
+  EXPECT_STRREF_EQ("_0", parse_float("_0", -1.0f, val));
+  EXPECT_EQ(val, -1.0f);
+  EXPECT_STRREF_EQ("..5", parse_float("..5", -2.0f, val));
+  EXPECT_EQ(val, -2.0f);
+  /* Out of float range. Current float parser (fast_float)
+   * clamps out of range numbers to +/- infinity, so this
+   * one gets a +inf instead of fallback -3.0. */
+  EXPECT_STRREF_EQ(" a", parse_float("9.0e500 a", -3.0f, val));
+  EXPECT_EQ(val, std::numeric_limits<float>::infinity());
+  /* Has leading white-space when we don't expect it */
+  EXPECT_STRREF_EQ(" 1", parse_float(" 1", -4.0f, val, false));
+  EXPECT_EQ(val, -4.0f);
+}
+
+}  // namespace blender::io