save.h

/*****************************************************************************
 * VCLib                                                                     *
 * Visual Computing Library                                                  *
 *                                                                           *
 * Copyright(C) 2021-2025                                                    *
 * Visual Computing Lab                                                      *
 * ISTI - Italian National Research Council                                  *
 *                                                                           *
 * All rights reserved.                                                      *
 *                                                                           *
 * This program is free software; you can redistribute it and/or modify      *
 * it under the terms of the Mozilla Public License Version 2.0 as published *
 * by the Mozilla Foundation; either version 2 of the License, or            *
 * (at your option) any later version.                                       *
 *                                                                           *
 * This program is distributed in the hope that it will be useful,           *
 * but WITHOUT ANY WARRANTY; without even the implied warranty of            *
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the              *
 * Mozilla Public License Version 2.0                                        *
 * (https://www.mozilla.org/en-US/MPL/2.0/) for more details.                *
 ****************************************************************************/
 
#ifndef VCL_LOAD_SAVE_OFF_SAVE_H
#define VCL_LOAD_SAVE_OFF_SAVE_H
 
#include <vclib/exceptions/io.h>
#include <vclib/io/write.h>
#include <vclib/load_save/settings.h>
#include <vclib/misc/logger.h>
#include <vclib/space/complex/mesh_info.h>
 
namespace vcl {
 
template<MeshConcept MeshType, LoggerConcept LogType = NullLogger>
void saveOff(
    const MeshType&     m,
    std::ostream&       fp,
    LogType&            log      = nullLogger,
    const SaveSettings& settings = SaveSettings())
{
    MeshInfo meshInfo(m);
 
    // make sure that the given info contains only components that are actually
    // available in the mesh. meshInfo will contain the intersection between the
    // components that the user wants to save and the components that are
    // available in the mesh.
    if (!settings.info.isEmpty())
        meshInfo = settings.info.intersect(meshInfo);
 
    if (meshInfo.hasVertexNormals())
        fp << "N";
    if (meshInfo.hasVertexColors())
        fp << "C";
    if (meshInfo.hasVertexTexCoords())
        fp << "ST";
    fp << "OFF" << std::endl;
 
    uint vn = 0;
    uint fn = 0;
    uint en = 0;
    if constexpr (HasVertices<MeshType>) {
        vn = m.vertexNumber();
    }
    if constexpr (HasFaces<MeshType>) {
        fn = m.faceNumber();
    }
 
    io::writeInt(fp, vn, false);
    io::writeInt(fp, fn, false);
    io::writeInt(fp, en, false);
    fp << std::endl; // remove last char (a space) and add a newline
 
    // vertices
    if constexpr (HasVertices<MeshType>) {
        using VertexType = MeshType::VertexType;
        for (const VertexType& v : m.vertices()) {
            io::writeDouble(fp, v.coord().x(), false);
            io::writeDouble(fp, v.coord().y(), false);
            io::writeDouble(fp, v.coord().z(), false);
 
            if constexpr (HasPerVertexColor<MeshType>) {
                if (meshInfo.hasVertexColors()) {
                    io::writeInt(fp, v.color().red(), false);
                    io::writeInt(fp, v.color().green(), false);
                    io::writeInt(fp, v.color().blue(), false);
                    io::writeInt(fp, v.color().alpha(), false);
                }
            }
            if constexpr (HasPerVertexNormal<MeshType>) {
                if (meshInfo.hasVertexNormals()) {
                    io::writeDouble(fp, v.normal().x(), false);
                    io::writeDouble(fp, v.normal().y(), false);
                    io::writeDouble(fp, v.normal().z(), false);
                }
            }
            if constexpr (HasPerVertexTexCoord<MeshType>) {
                if (meshInfo.hasVertexTexCoords()) {
                    io::writeDouble(fp, v.texCoord().u(), false);
                    io::writeDouble(fp, v.texCoord().v(), false);
                }
            }
 
            fp << std::endl;
        }
    }
 
    // faces
    if constexpr (HasFaces<MeshType>) {
        using VertexType = MeshType::VertexType;
        using FaceType   = MeshType::FaceType;
 
        // indices of vertices that do not consider deleted vertices
        std::vector<uint> vIndices = m.vertexCompactIndices();
 
        for (const FaceType& f : m.faces()) {
            io::writeInt(fp, f.vertexNumber(), false);
            for (const VertexType* v : f.vertices()) {
                io::writeInt(fp, vIndices[m.index(v)], false);
            }
            if constexpr (HasPerFaceColor<MeshType>) {
                if (meshInfo.hasFaceColors()) {
                    io::writeInt(fp, f.color().red(), false);
                    io::writeInt(fp, f.color().green(), false);
                    io::writeInt(fp, f.color().blue(), false);
                    io::writeInt(fp, f.color().alpha(), false);
                }
            }
 
            fp << std::endl;
        }
    }
}
 
template<MeshConcept MeshType, LoggerConcept LogType = NullLogger>
void saveOff(
    const MeshType&     m,
    std::ostream&       fp,
    const SaveSettings& settings,
    LogType&            log = nullLogger)
{
    saveOff(m, fp, log, settings);
}
 
template<MeshConcept MeshType, LoggerConcept LogType = NullLogger>
void saveOff(
    const MeshType&     m,
    const std::string&  filename,
    LogType&            log      = nullLogger,
    const SaveSettings& settings = SaveSettings())
{
    std::ofstream fp = openOutputFileStream(filename, "off");
    saveOff(m, fp, log, settings);
    fp.close();
}
 
template<MeshConcept MeshType, LoggerConcept LogType = NullLogger>
void saveOff(
    const MeshType&     m,
    const std::string&  filename,
    const SaveSettings& settings,
    LogType&            log = nullLogger)
{
    saveOff(m, filename, log, settings);
}
 
} // namespace vcl
 
#endif // VCL_LOAD_SAVE_OFF_SAVE_H