#include <gmio_core/error.h>
#include <gmio_stl/stl_io.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_triangle(void* cookie,
                                   uint32_t triangle_id,
                                   const gmio_stl_triangle_t* triangle)
{
  my_igeom_t* my_igeom = (my_igeom_t*)(cookie);
  if (my_igeom != NULL)
    ++(my_igeom->facet_count);
}

static void libstl_gmio_stlb_read(const char* filepath)
{
  /* uint8_t stack_buff[30 * 1024]; */

  my_igeom_t cookie;
  gmio_transfer_t trsf;
  gmio_stl_mesh_creator_t mesh_creator;
  int result;

  FILE* file = fopen(filepath, "rb");
  if (file == NULL) {
    fprintf(stderr, "Failed to open binary STL file %s\n", filepath);
    return;
  }

  cookie.facet_count = 0;
  memset(&mesh_creator, 0, sizeof(gmio_stl_mesh_creator_t));
  mesh_creator.cookie = &cookie;
  mesh_creator.add_triangle_func = dummy_process_triangle;

  memset(&trsf, 0, sizeof(gmio_transfer_t));
  gmio_stream_set_stdio(&trsf.stream, file);
  trsf.buffer = (uint8_t*)malloc(512 * 1024);
  trsf.buffer_size = 512 * 1024;
  /* trsf.buffer = stack_buff;
  trsf.buffer_size = 24 * 1024; */

  result = gmio_stlb_read(&mesh_creator, &trsf, GMIO_LITTLE_ENDIAN);
  if (gmio_error(result))
    fprintf(stderr, "gmio_stlb_read() error %i", result);

  fprintf(stdout, "Facet count: %i\n", cookie.facet_count);

  free(trsf.buffer);
  fclose(file);
}

static void libstl_gmio_stla_read(const char* filepath)
{
  gmio_transfer_t trsf;
  int result;
  FILE* file;

  file = fopen(filepath, "r");
  if (file == NULL) {
    fprintf(stderr, "Failed to open ascii STL file %s\n", filepath);
    return;
  }

  memset(&trsf, 0, sizeof(gmio_transfer_t));
  gmio_stream_set_stdio(&trsf.stream, file);
  trsf.buffer = (char*)malloc(512 * 1024);
  trsf.buffer_size = 512 * 1024;

  result = gmio_stla_read(NULL, &trsf, 0);
  if (gmio_error(result))
    fprintf(stderr, "gmio_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_gmio_stla_read, "gmio_stla_read()", argc - 2, argv + 2);
  else if (strcmp(argv[1], "--stlb") == 0)
    benchmark(&libstl_gmio_stlb_read, "gmio_stlb_read()", argc - 2, argv + 2);

  return 0;
}