gmio/tests/bench_libstl/main.c
2013-04-27 00:23:00 +02:00

88 lines
2.3 KiB
C

#include <datax/c/libstl/stla_read.h>
#include <datax/c/libstl/stlb_read.h>
#include <datax/c/error.h>
#include "../commons/bench_tools.h"
#include <stdio.h>
#include <stdlib.h>
typedef struct my_igeom
{
uint32_t facet_count;
} my_igeom_t;
static void dummy_process_next_triangle(foug_stlb_geom_input_t* igeom,
const foug_stlb_triangle_t* triangle)
{
my_igeom_t* my_igeom = (my_igeom_t*)(igeom->cookie);
if (my_igeom != NULL)
++(my_igeom->facet_count);
}
static void libstl_foug_stlb_read(const char* filepath)
{
FILE* file = fopen(filepath, "rb");
if (file == NULL) {
fprintf(stderr, "Failed to open binary STL file %s\n", filepath);
return;
}
my_igeom_t cookie;
cookie.facet_count = 0;
foug_stlb_geom_input_t geom;
memset(&geom, 0, sizeof(foug_stlb_geom_input_t));
geom.cookie = &cookie;
geom.process_next_triangle_func = dummy_process_next_triangle;
foug_transfer_t trsf;
memset(&trsf, 0, sizeof(foug_transfer_t));
foug_stream_set_stdio(&trsf.stream, file);
trsf.buffer = (uint8_t*)malloc(512 * 1024);
trsf.buffer_size = 512 * 1024;
const int result = foug_stlb_read(&geom, &trsf, FOUG_LITTLE_ENDIAN);
if (foug_datax_error(result))
fprintf(stderr, "foug_stlb_read() error %i", result);
fprintf(stdout, "Facet count: %i\n", cookie.facet_count);
free(trsf.buffer);
fclose(file);
}
static void libstl_foug_stla_read(const char* filepath)
{
FILE* file = fopen(filepath, "r");
if (file == NULL) {
fprintf(stderr, "Failed to open ascii STL file %s\n", filepath);
return;
}
foug_transfer_t trsf;
memset(&trsf, 0, sizeof(foug_transfer_t));
foug_stream_set_stdio(&trsf.stream, file);
trsf.buffer = (char*)malloc(512 * 1024);
trsf.buffer_size = 512 * 1024;
const int result = foug_stla_read(NULL, &trsf, 0);
if (foug_datax_error(result))
fprintf(stderr, "foug_stla_read() error %i", result);
/* fprintf(stdout, "Facet count: %i\n", igeom.facet_count); */
fclose(file);
}
int main(int argc, char** argv)
{
if (argc < 3)
return -1;
if (strcmp(argv[1], "--stla") == 0)
benchmark(&libstl_foug_stla_read, "foug_stla_read()", argc - 2, argv + 2);
else if (strcmp(argv[1], "--stlb") == 0)
benchmark(&libstl_foug_stlb_read, "foug_stlb_read()", argc - 2, argv + 2);
return 0;
}