/***************************************************************************
 *   Copyright (c) 2022 Werner Mayer <wmayer[at]users.sourceforge.net>     *
 *                                                                         *
 *   This file is part of the FreeCAD CAx development system.              *
 *                                                                         *
 *   This library is free software; you can redistribute it and/or         *
 *   modify it under the terms of the GNU Library General Public           *
 *   License as published by the Free Software Foundation; either          *
 *   version 2 of the License, or (at your option) any later version.      *
 *                                                                         *
 *   This library  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         *
 *   GNU Library General Public License for more details.                  *
 *                                                                         *
 *   You should have received a copy of the GNU Library General Public     *
 *   License along with this library; see the file COPYING.LIB. If not,    *
 *   write to the Free Software Foundation, Inc., 59 Temple Place,         *
 *   Suite 330, Boston, MA  02111-1307, USA                                *
 *                                                                         *
 ***************************************************************************/

#include "PreCompiled.h"
#ifndef _PreComp_
#include <QBuffer>
#include <QByteArray>

#include <Inventor/SbRotation.h>
#include <Inventor/SbViewportRegion.h>
#include <Inventor/nodes/SoCoordinate3.h>
#include <Inventor/nodes/SoDirectionalLight.h>
#include <Inventor/nodes/SoIndexedFaceSet.h>
#include <Inventor/nodes/SoOrthographicCamera.h>
#include <Inventor/nodes/SoSeparator.h>
#endif

#include <Gui/SoFCOffscreenRenderer.h>

#include "ThumbnailExtension.h"
#include "ViewProvider.h"


using namespace MeshGui;

Mesh::Extension3MF::Resource ThumbnailExtension3MF::addMesh(const Mesh::MeshObject& mesh)
{
    SoCoordinate3* coord = new SoCoordinate3();
    SoIndexedFaceSet* faces = new SoIndexedFaceSet();

    SoOrthographicCamera* cam = new SoOrthographicCamera();
    SoSeparator* root = new SoSeparator();
    root->ref();
    root->addChild(cam);
    root->addChild(new SoDirectionalLight);
    root->addChild(coord);
    root->addChild(faces);

    ViewProviderMeshBuilder().createMesh(mesh.getKernel(), coord, faces);

    SbRotation rot(-0.35355f, -0.14644f, -0.35355f, -0.85355f);
    cam->orientation.setValue(rot);
    SbViewportRegion vpr(256, 256);
    cam->viewAll(root, vpr);

    Gui::SoQtOffscreenRenderer renderer(vpr);
    renderer.setBackgroundColor(SbColor4f(1.0f, 1.0f, 1.0f, 0.0f));
    QImage img;
    renderer.render(root);
    renderer.writeToImage(img);
    root->unref();

    QByteArray data;
    QBuffer buffer(&data);
    buffer.open(QIODevice::WriteOnly);
    img.save(&buffer, "PNG");
    buffer.close();

    Mesh::Extension3MF::Resource res;
    res.extension = "png";
    res.contentType = "image/png";
    res.relationshipType =
        "http://schemas.openxmlformats.org/package/2006/relationships/metadata/thumbnail";
    res.fileContent = std::string(data.data(), data.size());
    setContentName(res);

    index++;
    return res;
}

void ThumbnailExtension3MF::setContentName(Mesh::Extension3MF::Resource& res)
{
    if (index == 0) {
        res.relationshipTarget = "/Metadata/thumbnail.png";
        res.fileNameInZip = "Metadata/thumbnail.png";
    }
    else {
        std::string suf = std::to_string(index);
        res.relationshipTarget = "/Metadata/thumbnail" + suf + ".png";
        res.fileNameInZip = "Metadata/thumbnail" + suf + ".png";
    }
}