stla: implement writing
This commit is contained in:
Hugues Delorme
parent
f866df2127
commit
e983afe159
@ -2,7 +2,212 @@
|
||||
|
||||
#include "../error.h"
|
||||
|
||||
int foug_stla_write(foug_stla_geom_output_t* geom, foug_transfer_t* trsf)
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
|
||||
/*
|
||||
* | 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;
|
||||
}
|
||||
|
||||
@ -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 */
|
||||
|
||||
Loading…
Reference in New Issue
Block a user