gmio/src/gmio_stl/internal/stlb_write.c

/****************************************************************************
** gmio
** Copyright Fougue (2 Mar. 2015)
** contact@fougue.pro
**
** 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/licences/Licence_CeCILL-B_V1-en.html".
****************************************************************************/

#include "stlb_write.h"

#include "stl_rw_common.h"
#include "stlb_byte_swap.h"
#include "../stl_error.h"
#include "../stl_io.h"

#include "../../gmio_core/error.h"
#include "../../gmio_core/internal/byte_codec.h"
#include "../../gmio_core/internal/min_max.h"
#include "../../gmio_core/internal/helper_stream.h"
#include "../../gmio_core/internal/helper_transfer.h"
#include "../../gmio_core/internal/safe_cast.h"

#include <string.h>

GMIO_INLINE void write_triangle_memcpy(
        const gmio_stl_triangle_t* triangle, uint8_t* mblock)
{
    memcpy(mblock, triangle, GMIO_STLB_TRIANGLE_RAWSIZE);
}

static void gmio_stlb_write_facets(
        const gmio_stl_mesh_t* mesh,
        uint8_t* mblock,
        const gmio_stlb_readwrite_helper_t* wparams)
{
    const uint32_t facet_count = wparams->facet_count;
    const uint32_t i_facet_offset = wparams->i_facet_offset;
    gmio_stl_triangle_t triangle;
    uint32_t mblock_offset = 0;
    uint32_t i_facet = 0;

    if (mesh == NULL || mesh->func_get_triangle == NULL)
        return;

    triangle.attribute_byte_count = 0;
    for (i_facet = i_facet_offset;
         i_facet < (i_facet_offset + facet_count);
         ++i_facet)
    {
        mesh->func_get_triangle(mesh->cookie, i_facet, &triangle);

        if (wparams->func_fix_endian != NULL)
            wparams->func_fix_endian(&triangle);

        write_triangle_memcpy(&triangle, mblock + mblock_offset);

        mblock_offset += GMIO_STLB_TRIANGLE_RAWSIZE;
    } /* end for */
}

int gmio_stlb_write(
        gmio_transfer_t* trsf,
        const gmio_stl_mesh_t* mesh,
        const gmio_stl_write_options_t* options,
        gmio_endianness_t byte_order)
{
    /* Constants */
    const uint32_t facet_count =
            mesh != NULL ? mesh->triangle_count : 0;
    const gmio_bool_t write_triangles_only =
            options != NULL ? options->stl_write_triangles_only : GMIO_FALSE;
    /* Variables */
    void* mblock_ptr = trsf != NULL ? trsf->memblock.ptr : NULL;
    gmio_stlb_readwrite_helper_t wparams = {0};
    uint32_t i_facet = 0;
    int error = GMIO_ERROR_OK;

    /* Check validity of input parameters */
    gmio_stl_check_mesh(&error, mesh);
    gmio_stlb_check_params(&error, trsf, byte_order);
    if (gmio_error(error))
        return error;

    /* Initialize wparams */
    if (byte_order != GMIO_ENDIANNESS_HOST)
        wparams.func_fix_endian = gmio_stl_triangle_bswap;
    /* Note: trsf != NULL  certified by gmio_stlb_check_params() */
    /* coverity[var_deref_op : FALSE] */
    wparams.facet_count = gmio_size_to_uint32(
                trsf->memblock.size / GMIO_STLB_TRIANGLE_RAWSIZE);

    if (!write_triangles_only) {
        error = gmio_stlb_write_header(
                    &trsf->stream,
                    byte_order,
                    options != NULL ? options->stlb_header_data : NULL,
                    facet_count);
        if (gmio_error(error))
            return error;
    }

    /* Write triangles */
    for (i_facet = 0;
         i_facet < facet_count && gmio_no_error(error);
         i_facet += wparams.facet_count)
    {
        gmio_transfer_handle_progress(trsf, i_facet, facet_count);

        /* Write to memory block */
        wparams.facet_count = GMIO_MIN(wparams.facet_count,
                                       facet_count - wparams.i_facet_offset);

        gmio_stlb_write_facets(mesh, mblock_ptr, &wparams);
        wparams.i_facet_offset += wparams.facet_count;

        /* Write memory block to stream */
        if (gmio_stream_write(
                    &trsf->stream,
                    mblock_ptr,
                    GMIO_STLB_TRIANGLE_RAWSIZE,
                    wparams.facet_count)
                != wparams.facet_count)
        {
            error = GMIO_ERROR_STREAM;
        }

        /* Handle stop request */
        if (gmio_no_error(error) && gmio_transfer_is_stop_requested(trsf))
            error = GMIO_ERROR_TRANSFER_STOPPED;
    } /* end for */

    return error;
}