diff --git a/src/c/libstl/stla_write.c b/src/c/libstl/stla_write.c index 948c797..ea99922 100644 --- a/src/c/libstl/stla_write.c +++ b/src/c/libstl/stla_write.c @@ -2,7 +2,212 @@ #include "../error.h" -int foug_stla_write(foug_stla_geom_output_t* geom, foug_transfer_t* trsf) +#include +#include + +/* + * | 14 || 17 || 17 || 15 | -> 73 + * facet normal 0.986544556E+00 0.986544556E+00 0.986544556E+00 + * | 11 | + * outer loop + * | 10 || 17 || 17 || 15 | -> 69 + * vertex 0.167500112E+02 0.505000112E+02 0.000000000E+00 + * vertex 0.164599000E+02 0.505000111E+02 0.221480112E+01 + * vertex 0.166819000E+02 0.483135112E+02 0.221480112E+01 + * | 8 | + * endloop + * | 8 | + * endfacet + * + * Total without EOL = 73 + 11 + 3*69 + 8 + 8 = 307 + * Total with EOL(2 chars) = 307 + 7*2 = 321 + */ + +#define FOUG_STLA_FACET_SIZE 321 +#define FOUG_STLA_FACET_SIZE_P2 512 +#define FOUG_STLA_SOLID_NAME_MAX_LEN 512 + +static char* foug_write_string(char* buffer, const char* str) { - return FOUG_DATAX_NO_ERROR; + strcpy(buffer, str); + return buffer + strlen(str); +} + +static char* foug_write_string_eol(char* buffer, const char* str) +{ + const size_t len = strlen(str); + strncpy(buffer, str, len); + buffer[len] = '\n'; + return buffer + len + 1; +} + +static char* foug_write_eol(char* buffer) +{ + *buffer = '\n'; + return buffer + 1; +} + +/*static char* foug_write_space(char* buffer) +{ + *buffer = ' '; + return buffer + 1; +}*/ + +static char* foug_write_nspaces(char* buffer, size_t n) +{ + while (n > 0) + buffer[--n] = ' '; + return buffer + n; +} + +static char* foug_write_stdio_format(char* buffer, uint8_t prec) +{ + size_t prec_len = 0 ; + + buffer[0] = '%'; + buffer[1] = '.'; + prec_len = sprintf(buffer + 2, "%i", prec); + buffer[2 + prec_len] = 'E'; + return buffer + 3 + prec_len; +} + +/*static char* foug_write_real32_string(char* buffer, const char* format, foug_real32_t val) +{ + return buffer + sprintf(buffer, format, val); +}*/ + +static char* foug_write_coords(char* buffer, + const char* coords_format, + const foug_stl_coords_t* coords) +{ + return buffer + sprintf(buffer, coords_format, coords->x, coords->y, coords->z); +} + +static foug_bool_t foug_tansfer_flush_buffer(foug_transfer_t* trsf, size_t n) +{ + return foug_stream_write(&trsf->stream, trsf->buffer, sizeof(char), n) == n; +} + +int foug_stla_write(foug_stla_geom_output_t* geom, + foug_transfer_t* trsf, + uint8_t real32_precision) +{ + size_t solid_count = 0; + size_t total_facet_count = 0; + size_t written_facet_count = 0; + size_t isolid = 0; + const size_t buffer_facet_count = trsf != NULL ? trsf->buffer_size / FOUG_STLA_FACET_SIZE_P2 : 0; + char* buffer_iterator = trsf != NULL ? trsf->buffer : NULL; + char coords_format[64]; + int error = FOUG_DATAX_NO_ERROR; + + if (real32_precision == 0 || real32_precision > 9) + return FOUG_STLA_WRITE_INVALID_REAL32_PRECISION; + if (buffer_iterator == NULL) + return FOUG_DATAX_NULL_BUFFER_ERROR; + if (trsf->buffer_size < FOUG_STLA_FACET_SIZE_P2) + return FOUG_DATAX_INVALID_BUFFER_SIZE_ERROR; + if (geom->get_triangle_count_func == NULL) + return FOUG_STLA_WRITE_NULL_GET_TRIANGLE_COUNT_FUNC; + if (geom->get_triangle_func == NULL) + return FOUG_STLA_WRITE_NULL_GET_TRIANGLE_FUNC; + + { + char* coords_format_iterator = coords_format; + coords_format_iterator = foug_write_stdio_format(coords_format_iterator, real32_precision); + coords_format_iterator = foug_write_nspaces(coords_format_iterator, 2); + coords_format_iterator = foug_write_stdio_format(coords_format_iterator, real32_precision); + coords_format_iterator = foug_write_nspaces(coords_format_iterator, 2); + coords_format_iterator = foug_write_stdio_format(coords_format_iterator, real32_precision); + /* TODO: check the "format" string can contain the given precision */ + } + + /* Get solid count */ + if (geom->get_solid_count_func != NULL) + solid_count = geom->get_solid_count_func(geom); + else + solid_count = 1; + + /* Compute total facet count */ + if (trsf->task_control.handle_progress_func != NULL) { + for (isolid = 0; isolid < solid_count; ++isolid) + total_facet_count += geom->get_triangle_count_func(geom, isolid); + } + + /* Write solids */ + for (isolid = 0; isolid < solid_count && foug_datax_no_error(error); ++isolid) { + char solid_name[FOUG_STLA_SOLID_NAME_MAX_LEN]; + const size_t facet_count = geom->get_triangle_count_func(geom, isolid); + size_t ifacet = 0; + + /* Get current solid name */ + if (geom->get_solid_name != NULL) + geom->get_solid_name(geom, isolid, solid_name); + else + sprintf(solid_name, "solid_%u", (unsigned int)isolid); + + /* Write solid declaration */ + solid_name[FOUG_STLA_SOLID_NAME_MAX_LEN - 1] = 0; /* Ensure string is null-terminated */ + buffer_iterator = foug_write_string(buffer_iterator, "solid "); + buffer_iterator = foug_write_string_eol(buffer_iterator, solid_name); + if (!foug_tansfer_flush_buffer(trsf, buffer_iterator - (char*)trsf->buffer)) + return FOUG_DATAX_STREAM_ERROR; + + /* Write current solid's facets */ + for (ifacet = 0; + ifacet < facet_count && foug_datax_no_error(error); + ifacet += buffer_facet_count) + { + foug_stl_triangle_t tri; + size_t ibuffer_facet; + + /* Writing of facets is buffered */ + buffer_iterator = trsf->buffer; + for (ibuffer_facet = ifacet; + ibuffer_facet < (ifacet + buffer_facet_count); + ++ibuffer_facet) + { + geom->get_triangle_func(geom, isolid, ibuffer_facet, &tri); + buffer_iterator = foug_write_string(buffer_iterator, "facet normal "); + buffer_iterator = foug_write_coords(buffer_iterator, coords_format, &tri.normal); + buffer_iterator = foug_write_eol(buffer_iterator); + + buffer_iterator = foug_write_string_eol(buffer_iterator, " outer loop"); + buffer_iterator = foug_write_string(buffer_iterator, " vertex "); + buffer_iterator = foug_write_coords(buffer_iterator, coords_format, &tri.v1); + buffer_iterator = foug_write_eol(buffer_iterator); + buffer_iterator = foug_write_string(buffer_iterator, " vertex "); + buffer_iterator = foug_write_coords(buffer_iterator, coords_format, &tri.v2); + buffer_iterator = foug_write_eol(buffer_iterator); + buffer_iterator = foug_write_string(buffer_iterator, " vertex "); + buffer_iterator = foug_write_coords(buffer_iterator, coords_format, &tri.v3); + buffer_iterator = foug_write_eol(buffer_iterator); + buffer_iterator = foug_write_string_eol(buffer_iterator, " endloop"); + + buffer_iterator = foug_write_string_eol(buffer_iterator, "endfacet"); + } /* end for (ibuffer_facet) */ + + if (!foug_tansfer_flush_buffer(trsf, buffer_iterator - (char*)trsf->buffer)) + error = FOUG_DATAX_STREAM_ERROR; + + /* Task control */ + if (foug_datax_no_error(error) && trsf->task_control.handle_progress_func != NULL) { + size_t percentage = 0; + written_facet_count += buffer_facet_count; + percentage = foug_percentage(0, total_facet_count, written_facet_count); + if (!foug_task_control_handle_progress(&trsf->task_control, percentage)) + error = FOUG_DATAX_TASK_STOPPED_ERROR; + } + } /* end for (ifacet) */ + + /* Write end of solid */ + if (foug_datax_no_error(error)) { + buffer_iterator = foug_write_string(trsf->buffer, "endsolid "); + buffer_iterator = foug_write_string_eol(buffer_iterator, solid_name); + if (!foug_tansfer_flush_buffer(trsf, buffer_iterator - (char*)trsf->buffer)) + error = FOUG_DATAX_STREAM_ERROR; + } + } /* end for (isolid) */ + + return error; } diff --git a/src/c/libstl/stla_write.h b/src/c/libstl/stla_write.h index c8135af..f34c365 100644 --- a/src/c/libstl/stla_write.h +++ b/src/c/libstl/stla_write.h @@ -10,18 +10,20 @@ typedef struct foug_stla_geom_output foug_stla_geom_output_t; struct foug_stla_geom_output { void* cookie; - size_t (*get_solid_count_func) (foug_stla_geom_output_t*); - void (*get_solid_name) (foug_stla_geom_output_t*, size_t, char*); + size_t (*get_solid_count_func) (foug_stla_geom_output_t*); /* Optional */ + void (*get_solid_name) (foug_stla_geom_output_t*, size_t, char*); /* Optional */ size_t (*get_triangle_count_func)(foug_stla_geom_output_t*, size_t); void (*get_triangle_func) (foug_stla_geom_output_t*, size_t, size_t, foug_stl_triangle_t*); }; /* foug_stla_write() */ FOUG_DATAX_LIBSTL_EXPORT int foug_stla_write(foug_stla_geom_output_t* geom, - foug_transfer_t* trsf); + foug_transfer_t* trsf, + uint8_t real32_precision); /* Specific error codes returned by foug_stla_write() */ #define FOUG_STLA_WRITE_NULL_GET_TRIANGLE_COUNT_FUNC 1 -#define FOUG_STLB_WRITE_NULL_GET_TRIANGLE_FUNC 2 +#define FOUG_STLA_WRITE_NULL_GET_TRIANGLE_FUNC 2 +#define FOUG_STLA_WRITE_INVALID_REAL32_PRECISION 3 #endif /* FOUG_DATAX_C_LIBSTL_STLA_WRITE_H */