tests: add more tools to stl_utils

This commit is contained in:
Hugues Delorme 2015-09-11 16:35:04 +02:00
parent ac7fd554f8
commit 287ad5c13f
2 changed files with 102 additions and 5 deletions

View File

@ -15,7 +15,14 @@
#include "stl_utils.h"
#include "utils.h"
#include "../src/gmio_core/internal/min_max.h"
#include "../src/gmio_core/internal/safe_cast.h"
#include <stdlib.h>
#include <string.h>
#include <ctype.h>
void gmio_stl_nop_add_triangle(
void *cookie, uint32_t tri_id, const gmio_stl_triangle_t *triangle)
@ -25,23 +32,60 @@ void gmio_stl_nop_add_triangle(
GMIO_UNUSED(triangle);
}
gmio_stl_triangle_array_t gmio_stl_triangle_array_malloc(size_t tri_count)
{
gmio_stl_triangle_array_t array = {0};
if (tri_count > 0) {
array.ptr =
(gmio_stl_triangle_t*)malloc(tri_count * sizeof(gmio_stl_triangle_t));
}
array.count = gmio_size_to_uint32(tri_count);
array.capacity = array.count;
return array;
}
static void gmio_stl_data__ascii_begin_solid(
void* cookie, size_t stream_size, const char* solid_name)
{
gmio_stl_data_t* data = (gmio_stl_data_t*)cookie;
memset(&data->solid_name[0], 0, sizeof(data->solid_name));
if (solid_name != NULL) {
const size_t len =
GMIO_MIN(sizeof(data->solid_name), strlen(solid_name));
strncpy(&data->solid_name[0], solid_name, len);
}
/* Try to guess how many vertices we could have assume we'll need 200 bytes
* for each face */
{
const size_t facet_size = 200;
const size_t facet_count = GMIO_MAX(1, stream_size / facet_size);
data->tri_array = gmio_stl_triangle_array_malloc(facet_count);
}
}
static void gmio_stl_data__binary_begin_solid(
void* cookie, uint32_t tri_count, const gmio_stlb_header_t* header)
{
gmio_stl_data_t* data = (gmio_stl_data_t*)cookie;
memcpy(&data->header, header, GMIO_STLB_HEADER_SIZE);
data->tri_array.count = tri_count;
if (tri_count > 0) {
data->tri_array.ptr =
(gmio_stl_triangle_t*)malloc(tri_count * sizeof(gmio_stl_triangle_t));
}
data->tri_array = gmio_stl_triangle_array_malloc(tri_count);
}
static void gmio_stl_data__add_triangle(
void* cookie, uint32_t tri_id, const gmio_stl_triangle_t* triangle)
{
gmio_stl_data_t* data = (gmio_stl_data_t*)cookie;
if (tri_id >= data->tri_array.capacity) {
uint32_t cap = data->tri_array.capacity;
cap += cap >> 3; /* Add 12.5% more capacity */
data->tri_array.ptr =
realloc(data->tri_array.ptr, cap * sizeof(gmio_stl_triangle_t));
data->tri_array.capacity = cap;
}
data->tri_array.ptr[tri_id] = *triangle;
data->tri_array.count = GMIO_MAX(data->tri_array.count, tri_id + 1);
}
static void gmio_stl_data__get_triangle(
@ -55,6 +99,7 @@ gmio_stl_mesh_creator_t gmio_stl_data_mesh_creator(gmio_stl_data_t *data)
{
gmio_stl_mesh_creator_t creator = {0};
creator.cookie = data;
creator.func_ascii_begin_solid = &gmio_stl_data__ascii_begin_solid;
creator.func_binary_begin_solid = &gmio_stl_data__binary_begin_solid;
creator.func_add_triangle = &gmio_stl_data__add_triangle;
return creator;
@ -68,3 +113,37 @@ gmio_stl_mesh_t gmio_stl_data_mesh(const gmio_stl_data_t *data)
mesh.func_get_triangle = &gmio_stl_data__get_triangle;
return mesh;
}
void gmio_stlb_header_to_printable_string(
const gmio_stlb_header_t *header, char *str, char replacement)
{
size_t i = 0;
for (; i < GMIO_STLB_HEADER_SIZE; ++i) {
str[i] = isprint((int)header->data[i]) ?
(char) header->data[i] :
replacement;
}
str[GMIO_STLB_HEADER_SIZE] = 0;
}
gmio_bool_t gmio_stl_coords_equal(
const gmio_stl_coords_t *lhs,
const gmio_stl_coords_t *rhs,
uint32_t max_ulp_diff)
{
return gmio_float32_equals_by_ulp(lhs->x, rhs->x, max_ulp_diff)
&& gmio_float32_equals_by_ulp(lhs->y, rhs->y, max_ulp_diff)
&& gmio_float32_equals_by_ulp(lhs->z, rhs->z, max_ulp_diff);
}
gmio_bool_t gmio_stl_triangle_equal(
const gmio_stl_triangle_t *lhs,
const gmio_stl_triangle_t *rhs,
uint32_t max_ulp_diff)
{
return gmio_stl_coords_equal(&lhs->normal, &rhs->normal, max_ulp_diff)
&& gmio_stl_coords_equal(&lhs->v1, &rhs->v1, max_ulp_diff)
&& gmio_stl_coords_equal(&lhs->v2, &rhs->v2, max_ulp_diff)
&& gmio_stl_coords_equal(&lhs->v3, &rhs->v3, max_ulp_diff)
&& lhs->attribute_byte_count == rhs->attribute_byte_count;
}

View File

@ -24,6 +24,16 @@
#include <string.h>
gmio_bool_t gmio_stl_coords_equal(
const gmio_stl_coords_t* lhs,
const gmio_stl_coords_t* rhs,
uint32_t max_ulp_diff);
gmio_bool_t gmio_stl_triangle_equal(
const gmio_stl_triangle_t* lhs,
const gmio_stl_triangle_t* rhs,
uint32_t max_ulp_diff);
/*! Does binary STL header \p lhs compare equal to \p rhs ? */
GMIO_INLINE gmio_bool_t gmio_stlb_header_equal(
const gmio_stlb_header_t* lhs, const gmio_stlb_header_t* rhs)
@ -31,6 +41,9 @@ GMIO_INLINE gmio_bool_t gmio_stlb_header_equal(
return memcmp(lhs, rhs, GMIO_STLB_HEADER_SIZE) == 0;
}
void gmio_stlb_header_to_printable_string(
const gmio_stlb_header_t* header, char* str, char replacement);
/*! Callback for gmio_stl_mesh_creator::func_add_triangle that does
* nothing(ie "no operation") */
void gmio_stl_nop_add_triangle(
@ -41,13 +54,18 @@ struct gmio_stl_triangle_array
{
gmio_stl_triangle_t* ptr;
uint32_t count;
uint32_t capacity;
};
typedef struct gmio_stl_triangle_array gmio_stl_triangle_array_t;
/*! Returns an dynamically allocated array of gmio_stl_triangle_t */
gmio_stl_triangle_array_t gmio_stl_triangle_array_malloc(size_t tri_count);
/*! Holds complete STL data (usable for both binary and ascii formats) */
struct gmio_stl_data
{
gmio_stlb_header_t header;
char solid_name[1024];
gmio_stl_triangle_array_t tri_array;
};
typedef struct gmio_stl_data gmio_stl_data_t;