gmio/tests/stream_buffer.c
2015-03-02 17:38:33 +01:00

86 lines
3.1 KiB
C

/****************************************************************************
**
** GeomIO Library
** Copyright FougSys (2 Mar. 2015)
** contact@fougsys.fr
**
** This software is a reusable library whose purpose is to provide complete
** I/O support for various CAD file formats (eg. STL)
**
** This software is governed by the CeCILL-B license under French law and
** abiding by the rules of distribution of free software. You can use,
** modify and/ or redistribute the software under the terms of the CeCILL-B
** license as circulated by CEA, CNRS and INRIA at the following URL
** "http://www.cecill.info".
**
****************************************************************************/
#include "stream_buffer.h"
#include <string.h>
static gmio_bool_t gmio_stream_buffer_at_end(void* cookie)
{
const gmio_buffer_t* buff = (const gmio_buffer_t*)cookie;
return buff->pos >= buff->len;
}
static int gmio_stream_buffer_error(void* cookie)
{
const gmio_buffer_t* buff = (const gmio_buffer_t*)cookie;
return buff == NULL || buff->pos > buff->len;
}
static size_t gmio_stream_buffer_read(void* cookie,
void* ptr,
size_t item_size,
size_t item_count)
{
if (item_size > 0 && item_count > 0) {
gmio_buffer_t* buff = (gmio_buffer_t*)cookie;
const size_t buff_remaining_size = buff->len - buff->pos;
const size_t wanted_read_size = item_size * item_count;
const size_t next_read_size = wanted_read_size <= buff_remaining_size ? wanted_read_size :
buff_remaining_size;
const size_t next_read_item_count = next_read_size / item_size;
memcpy(ptr, (const char*)buff->ptr + buff->pos, next_read_item_count * item_size);
buff->pos += next_read_item_count * item_size;
return next_read_item_count;
}
else {
return 0;
}
}
static size_t gmio_stream_buffer_write(void* cookie,
const void* ptr,
size_t item_size,
size_t item_count)
{
if (item_size > 0 && item_count > 0) {
gmio_buffer_t* buff = (gmio_buffer_t*)cookie;
const size_t buff_remaining_size = buff->len - buff->pos;
const size_t wanted_write_size = item_size * item_count;
const size_t next_write_size = wanted_write_size <= buff_remaining_size ? wanted_write_size :
buff_remaining_size;
const size_t next_write_item_count = next_write_size / item_size;
memcpy((char*)buff->ptr + buff->pos, ptr, next_write_item_count * item_size);
buff->pos += next_write_item_count * item_size;
return next_write_item_count;
}
else {
return 0;
}
}
void gmio_stream_set_buffer(gmio_stream_t *stream, gmio_buffer_t *buff)
{
stream->cookie = buff;
stream->at_end_func = gmio_stream_buffer_at_end;
stream->error_func = gmio_stream_buffer_error;
stream->read_func = gmio_stream_buffer_read;
stream->write_func = gmio_stream_buffer_write;
}