77 lines
2.3 KiB
C++
77 lines
2.3 KiB
C++
|
#include <set>
|
||
|
|
#include <QGuiApplication>
|
||
|
|
#include <QtCore/qbuffer.h>
|
||
|
|
#include <dust3d/base/ds3_file.h>
|
||
|
|
#include <dust3d/base/snapshot_xml.h>
|
||
|
|
#include "document_saver.h"
|
||
|
|
#include "image_forever.h"
|
||
|
|
|
||
|
|
DocumentSaver::DocumentSaver(const QString *filename,
|
||
|
|
dust3d::Snapshot *snapshot,
|
||
|
|
QByteArray *turnaroundPngByteArray) :
|
||
|
|
m_filename(filename),
|
||
|
|
m_snapshot(snapshot),
|
||
|
|
m_turnaroundPngByteArray(turnaroundPngByteArray)
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
DocumentSaver::~DocumentSaver()
|
||
|
|
{
|
||
|
|
delete m_snapshot;
|
||
|
|
delete m_turnaroundPngByteArray;
|
||
|
|
}
|
||
|
|
|
||
|
|
void DocumentSaver::process()
|
||
|
|
{
|
||
|
|
save(m_filename,
|
||
|
|
m_snapshot,
|
||
|
|
m_turnaroundPngByteArray);
|
||
|
|
emit finished();
|
||
|
|
}
|
||
|
|
|
||
|
|
void DocumentSaver::collectUsedResourceIds(const dust3d::Snapshot *snapshot,
|
||
|
|
std::set<dust3d::Uuid> &imageIds)
|
||
|
|
{
|
||
|
|
for (const auto &material: snapshot->materials) {
|
||
|
|
for (auto &layer: material.second) {
|
||
|
|
for (auto &mapItem: layer.second) {
|
||
|
|
auto findImageIdString = mapItem.find("linkData");
|
||
|
|
if (findImageIdString == mapItem.end())
|
||
|
|
continue;
|
||
|
|
dust3d::Uuid imageId = dust3d::Uuid(findImageIdString->second);
|
||
|
|
imageIds.insert(imageId);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
bool DocumentSaver::save(const QString *filename,
|
||
|
|
dust3d::Snapshot *snapshot,
|
||
|
|
const QByteArray *turnaroundPngByteArray)
|
||
|
|
{
|
||
|
|
dust3d::Ds3FileWriter ds3Writer;
|
||
|
|
|
||
|
|
{
|
||
|
|
std::string modelXml;
|
||
|
|
saveSnapshotToXmlString(*snapshot, modelXml);
|
||
|
|
if (modelXml.size() > 0) {
|
||
|
|
ds3Writer.add("model.xml", "model", modelXml.c_str(), modelXml.size());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (nullptr != turnaroundPngByteArray && turnaroundPngByteArray->size() > 0)
|
||
|
|
ds3Writer.add("canvas.png", "asset", turnaroundPngByteArray->data(), turnaroundPngByteArray->size());
|
||
|
|
|
||
|
|
std::set<dust3d::Uuid> imageIds;
|
||
|
|
collectUsedResourceIds(snapshot, imageIds);
|
||
|
|
|
||
|
|
for (const auto &imageId: imageIds) {
|
||
|
|
const QByteArray *pngByteArray = ImageForever::getPngByteArray(imageId);
|
||
|
|
if (nullptr == pngByteArray)
|
||
|
|
continue;
|
||
|
|
if (pngByteArray->size() > 0)
|
||
|
|
ds3Writer.add("images/" + imageId.toString() + ".png", "asset", pngByteArray->data(), pngByteArray->size());
|
||
|
|
}
|
||
|
|
|
||
|
|
return ds3Writer.save(filename->toUtf8().constData());
|
||
|
|
}
|