commit b469f7f4660e95148b88fe9c9d29c38a658d8eed
parent 97ddca2218eddeaa639baa4880bd72a1b1dc0f4c
Author: Vincent Forest <vincent.forest@meso-star.com>
Date: Tue, 22 Apr 2025 17:47:50 +0200
Start the writer tests
Test the API and write ASCII data
Diffstat:
2 files changed, 221 insertions(+), 1 deletion(-)
diff --git a/Makefile b/Makefile
@@ -147,7 +147,8 @@ lint:
TEST_SRC =\
src/test_sstl.c\
src/test_sstl_load_ascii.c\
- src/test_sstl_load_binary.c
+ src/test_sstl_load_binary.c\
+ src/test_sstl_write_ascii.c
TEST_OBJ = $(TEST_SRC:.c=.o)
TEST_DEP = $(TEST_SRC:.c=.d)
TEST_TGT = $(TEST_SRC:.c=.t)
@@ -180,6 +181,7 @@ $(TEST_OBJ): config.mk sstl-local.pc
test_sstl \
test_sstl_load_ascii \
test_sstl_load_binary \
+test_sstl_write_ascii \
: config.mk sstl-local.pc $(LIBNAME)
$(CC) $(CFLAGS_TEST) -o $@ src/$@.o $(LDFLAGS_TEST)
diff --git a/src/test_sstl_write_ascii.c b/src/test_sstl_write_ascii.c
@@ -0,0 +1,218 @@
+/* Copyright (C) 2015, 2016, 2019, 2021, 2023, 2025 |Méso|Star> (contact@meso-star.com)
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>. */
+
+#include "sstl.h"
+
+#include <rsys/float3.h>
+#include <rsys/logger.h>
+#include <rsys/mem_allocator.h>
+
+/*******************************************************************************
+ * Helper functions
+ ******************************************************************************/
+static void
+log_stream(const char* msg, void* ctx)
+{
+ ASSERT(msg);
+ (void)msg, (void)ctx;
+ printf("%s\n", msg);
+}
+
+static void
+check_api(void)
+{
+ struct logger logger;
+ struct mem_allocator allocator;
+ struct sstl_writer_create_args args = SSTL_WRITER_CREATE_ARGS_DEFAULT;
+ struct sstl_writer* writer = NULL;
+ const char* filename = "test.stl";
+
+ CHK(sstl_writer_create(NULL, &writer) == RES_BAD_ARG);
+ CHK(sstl_writer_create(&args, NULL) == RES_BAD_ARG);
+ CHK(sstl_writer_create(&args, &writer) == RES_BAD_ARG);
+
+ args.filename = filename;
+ CHK(sstl_writer_create(&args, &writer) == RES_OK);
+
+ CHK(sstl_write_facet(NULL, &SSTL_FACET_NULL) == RES_BAD_ARG);
+ CHK(sstl_write_facet(writer, NULL) == RES_BAD_ARG);
+ CHK(sstl_write_facet(writer, &SSTL_FACET_NULL) == RES_OK);
+
+ CHK(sstl_writer_ref_get(NULL) == RES_BAD_ARG);
+ CHK(sstl_writer_ref_get(writer) == RES_OK);
+ CHK(sstl_writer_ref_put(NULL) == RES_BAD_ARG);
+ CHK(sstl_writer_ref_put(writer) == RES_OK);
+ CHK(sstl_writer_ref_put(writer) == RES_OK);
+
+ args.triangles_count = 0;
+ CHK(sstl_writer_create(&args, &writer) == RES_OK);
+ CHK(sstl_write_facet(writer, &SSTL_FACET_NULL) == RES_BAD_OP);
+ CHK(sstl_writer_ref_put(writer) == RES_OK);
+
+ mem_init_proxy_allocator(&allocator, &mem_default_allocator);
+ CHK(logger_init(&allocator, &logger) == RES_OK);
+ logger_set_stream(&logger, LOG_OUTPUT, log_stream, NULL);
+ logger_set_stream(&logger, LOG_ERROR, log_stream, NULL);
+ logger_set_stream(&logger, LOG_WARNING, log_stream, NULL);
+
+ args.allocator = &allocator;
+ args.logger = &logger;
+ args.triangles_count = 1;
+ CHK(sstl_writer_create(&args, &writer) == RES_OK);
+ CHK(sstl_write_facet(writer, &SSTL_FACET_NULL) == RES_OK);
+ CHK(sstl_write_facet(writer, &SSTL_FACET_NULL) == RES_BAD_OP);
+ CHK(sstl_writer_ref_put(writer) == RES_OK);
+
+ logger_release(&logger);
+ CHK(MEM_ALLOCATED_SIZE(&allocator) == 0);
+ mem_shutdown_proxy_allocator(&allocator);
+}
+
+static void
+check_1_triangle(struct sstl* sstl)
+{
+ struct sstl_desc desc = SSTL_DESC_NULL;
+ struct sstl_facet facet = SSTL_FACET_NULL;
+ struct sstl_writer_create_args args = SSTL_WRITER_CREATE_ARGS_DEFAULT;
+ struct sstl_writer* writer = NULL;
+
+ float v[3] = {0,0,0};
+ const char* filename = "1_triangle.stl";
+
+ args.filename = filename;
+ args.triangles_count = 1;
+ CHK(sstl_writer_create(&args, &writer) == RES_OK);
+
+ f3_splat(facet.normal, 0); /* Let sstl calculate it automatically */
+ f3(facet.vertices[0], 0.f, 0.f, 0.f);
+ f3(facet.vertices[1], 1.f, 0.f, 0.f);
+ f3(facet.vertices[2], 0.f, 0.f, 1.f);
+ CHK(sstl_write_facet(writer, &facet) == RES_OK);
+ CHK(sstl_write_facet(writer, &facet) == RES_BAD_OP);
+
+ CHK(sstl_writer_ref_put(writer) == RES_OK);
+
+ CHK(sstl_load(sstl, filename) == RES_OK);
+ CHK(sstl_get_desc(sstl, &desc) == RES_OK);
+
+ CHK(desc.type == SSTL_ASCII);
+ CHK(!strcmp(desc.filename, filename));
+ CHK(desc.solid_name == NULL);
+ CHK(desc.vertices_count == 3);
+ CHK(desc.triangles_count == 1);
+ CHK(desc.indices[0] == 0);
+ CHK(desc.indices[1] == 1);
+ CHK(desc.indices[2] == 2);
+ CHK(f3_eq(desc.vertices + 0*3, f3(v, 0.f, 0.f, 0.f)) == 1);
+ CHK(f3_eq(desc.vertices + 1*3, f3(v, 1.f, 0.f, 0.f)) == 1);
+ CHK(f3_eq(desc.vertices + 2*3, f3(v, 0.f, 0.f, 1.f)) == 1);
+ CHK(f3_eq(desc.normals, f3(v, 0.f, -1.f, 0.f)) == 1);
+}
+
+static void
+check_tetrahedron(struct sstl* sstl)
+{
+ struct sstl_desc desc = SSTL_DESC_NULL;
+ struct sstl_facet facet = SSTL_FACET_NULL;
+ struct sstl_writer_create_args args = SSTL_WRITER_CREATE_ARGS_DEFAULT;
+ struct sstl_writer* writer = NULL;
+
+ float v[3] = {0,0,0};
+ FILE* fp = NULL;
+ const char* filename = "Tetrahedron";
+
+ CHK((fp = tmpfile()) != NULL);
+ args.filename = filename;
+ args.stream = fp;
+ args.solid_name = "cube corner";
+ CHK(sstl_writer_create(&args, &writer) == RES_OK);
+
+ f3(facet.normal, 0.f,-1.f, 0.f);
+ f3(facet.vertices[0], 0.f, 0.f, 0.f);
+ f3(facet.vertices[1], 1.f, 0.f, 0.f);
+ f3(facet.vertices[2], 0.f, 0.f, 1.f);
+ CHK(sstl_write_facet(writer, &facet) == RES_OK);
+
+ f3(facet.normal, 0.f, 0.f, -1.f);
+ f3(facet.vertices[0], 0.f, 0.f, 0.f);
+ f3(facet.vertices[1], 0.f, 1.f, 0.f);
+ f3(facet.vertices[2], 1.f, 0.f, 0.f);
+ CHK(sstl_write_facet(writer, &facet) == RES_OK);
+
+ f3(facet.normal,-1.f, 0.f, 0.f);
+ f3(facet.vertices[0], 0.f, 0.f, 0.f);
+ f3(facet.vertices[1], 0.f, 0.f, 1.f);
+ f3(facet.vertices[2], 0.f, 1.f, 0.f);
+ CHK(sstl_write_facet(writer, &facet) == RES_OK);
+
+
+ f3(facet.normal, 0.577f, 0.577f, 0.577f);
+ f3(facet.vertices[0], 1.f, 0.f, 0.f);
+ f3(facet.vertices[1], 0.f, 1.f, 0.f);
+ f3(facet.vertices[2], 0.f, 0.f, 1.f);
+ CHK(sstl_write_facet(writer, &facet) == RES_OK);
+
+ CHK(sstl_writer_ref_put(writer) == RES_OK);
+
+ rewind(fp);
+ CHK(sstl_load_stream_ascii(sstl, fp, filename) == RES_OK);
+ CHK(fclose(fp) == 0);
+
+ CHK(sstl_get_desc(sstl, &desc) == RES_OK);
+ CHK(!strcmp(desc.filename, filename));
+ CHK(!strcmp(desc.solid_name, "cube corner"));
+ CHK(desc.type == SSTL_ASCII);
+ CHK(desc.vertices_count == 4);
+ CHK(desc.triangles_count == 4);
+
+ CHK(f3_eq(desc.vertices + desc.indices[0]*3, f3(v, 0.f, 0.f, 0.f)) == 1);
+ CHK(f3_eq(desc.vertices + desc.indices[1]*3, f3(v, 1.f, 0.f, 0.f)) == 1);
+ CHK(f3_eq(desc.vertices + desc.indices[2]*3, f3(v, 0.f, 0.f, 1.f)) == 1);
+ CHK(f3_eq(desc.vertices + desc.indices[3]*3, f3(v, 0.f, 0.f, 0.f)) == 1);
+ CHK(f3_eq(desc.vertices + desc.indices[4]*3, f3(v, 0.f, 1.f, 0.f)) == 1);
+ CHK(f3_eq(desc.vertices + desc.indices[5]*3, f3(v, 1.f, 0.f, 0.f)) == 1);
+ CHK(f3_eq(desc.vertices + desc.indices[6]*3, f3(v, 0.f, 0.f, 0.f)) == 1);
+ CHK(f3_eq(desc.vertices + desc.indices[7]*3, f3(v, 0.f, 0.f, 1.f)) == 1);
+ CHK(f3_eq(desc.vertices + desc.indices[8]*3, f3(v, 0.f, 1.f, 0.f)) == 1);
+ CHK(f3_eq(desc.vertices + desc.indices[9]*3, f3(v, 1.f, 0.f, 0.f)) == 1);
+ CHK(f3_eq(desc.vertices + desc.indices[10]*3, f3(v, 0.f, 1.f, 0.f)) == 1);
+ CHK(f3_eq(desc.vertices + desc.indices[11]*3, f3(v, 0.f, 0.f, 1.f)) == 1);
+
+ CHK(f3_eq(desc.normals + 0*3, f3(v, 0.f,-1.f, 0.f)) == 1);
+ CHK(f3_eq(desc.normals + 1*3, f3(v, 0.f, 0.f,-1.f)) == 1);
+ CHK(f3_eq(desc.normals + 2*3, f3(v,-1.f, 0.f, 0.f)) == 1);
+ f3_normalize(v, f3(v, 1.f, 1.f, 1.f));
+ CHK(f3_eq_eps(desc.normals + 3*3, v, 1.e-6f) == 1);
+}
+
+/*******************************************************************************
+ * The test
+ ******************************************************************************/
+int
+main(int argc, char** argv)
+{
+ struct sstl* sstl = NULL;
+ (void)argc, (void)argv;
+
+ CHK(sstl_create(NULL, NULL, 3, &sstl) == RES_OK);
+
+ check_api();
+ check_1_triangle(sstl);
+ check_tetrahedron(sstl);
+
+ CHK(sstl_ref_put(sstl) == RES_OK);
+ CHK(mem_allocated_size() == 0);
+ return 0;
+}
