498 lines
15 KiB
C++
498 lines
15 KiB
C++
/*
|
|
Open Asset Import Library (assimp)
|
|
----------------------------------------------------------------------
|
|
|
|
Copyright (c) 2006-2016, assimp team
|
|
All rights reserved.
|
|
|
|
Redistribution and use of this software in source and binary forms,
|
|
with or without modification, are permitted provided that the
|
|
following conditions are met:
|
|
|
|
* Redistributions of source code must retain the above
|
|
copyright notice, this list of conditions and the
|
|
following disclaimer.
|
|
|
|
* Redistributions in binary form must reproduce the above
|
|
copyright notice, this list of conditions and the
|
|
following disclaimer in the documentation and/or other
|
|
materials provided with the distribution.
|
|
|
|
* Neither the name of the assimp team, nor the names of its
|
|
contributors may be used to endorse or promote products
|
|
derived from this software without specific prior
|
|
written permission of the assimp team.
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
----------------------------------------------------------------------
|
|
*/
|
|
|
|
#include <rapidjson/stringbuffer.h>
|
|
#include <rapidjson/writer.h>
|
|
#include <rapidjson/prettywriter.h>
|
|
|
|
namespace glTF {
|
|
|
|
using rapidjson::StringBuffer;
|
|
using rapidjson::PrettyWriter;
|
|
using rapidjson::Writer;
|
|
using rapidjson::StringRef;
|
|
using rapidjson::StringRef;
|
|
|
|
namespace {
|
|
|
|
template<size_t N>
|
|
inline Value& MakeValue(Value& val, float(&r)[N], MemoryPoolAllocator<>& al) {
|
|
val.SetArray();
|
|
val.Reserve(N, al);
|
|
for (int i = 0; i < N; ++i) {
|
|
val.PushBack(r[i], al);
|
|
}
|
|
return val;
|
|
};
|
|
|
|
template<class T>
|
|
inline void AddRefsVector(Value& obj, const char* fieldId, std::vector< Ref<T> >& v, MemoryPoolAllocator<>& al) {
|
|
if (v.empty()) return;
|
|
Value lst;
|
|
lst.SetArray();
|
|
lst.Reserve(v.size(), al);
|
|
for (size_t i = 0; i < v.size(); ++i) {
|
|
lst.PushBack(StringRef(v[i]->id), al);
|
|
}
|
|
obj.AddMember(StringRef(fieldId), lst, al);
|
|
};
|
|
|
|
|
|
}
|
|
|
|
inline void Write(Value& obj, Accessor& a, AssetWriter& w)
|
|
{
|
|
obj.AddMember("bufferView", Value(a.bufferView->id, w.mAl).Move(), w.mAl);
|
|
obj.AddMember("byteOffset", a.byteOffset, w.mAl);
|
|
obj.AddMember("byteStride", a.byteStride, w.mAl);
|
|
obj.AddMember("componentType", int(a.componentType), w.mAl);
|
|
obj.AddMember("count", a.count, w.mAl);
|
|
obj.AddMember("type", StringRef(AttribType::ToString(a.type)), w.mAl);
|
|
}
|
|
|
|
inline void Write(Value& obj, Animation& a, AssetWriter& w)
|
|
{
|
|
|
|
}
|
|
|
|
inline void Write(Value& obj, Buffer& b, AssetWriter& w)
|
|
{
|
|
std::string dataURI = "data:application/octet-stream;base64,";
|
|
Util::EncodeBase64(b.GetPointer(), b.byteLength, dataURI);
|
|
|
|
const char* type;
|
|
switch (b.type) {
|
|
case Buffer::Type_text:
|
|
type = "text"; break;
|
|
default:
|
|
type = "arraybuffer";
|
|
}
|
|
|
|
obj.AddMember("byteLength", b.byteLength, w.mAl);
|
|
obj.AddMember("type", StringRef(type), w.mAl);
|
|
obj.AddMember("uri", Value(dataURI, w.mAl).Move(), w.mAl);
|
|
}
|
|
|
|
inline void Write(Value& obj, BufferView& bv, AssetWriter& w)
|
|
{
|
|
obj.AddMember("buffer", Value(bv.buffer->id, w.mAl).Move(), w.mAl);
|
|
obj.AddMember("byteOffset", bv.byteOffset, w.mAl);
|
|
obj.AddMember("byteLength", bv.byteLength, w.mAl);
|
|
obj.AddMember("target", int(bv.target), w.mAl);
|
|
}
|
|
|
|
inline void Write(Value& obj, Camera& c, AssetWriter& w)
|
|
{
|
|
|
|
}
|
|
|
|
inline void Write(Value& obj, Image& img, AssetWriter& w)
|
|
{
|
|
std::string uri;
|
|
if (w.mAsset.extensionsUsed.KHR_binary_glTF && img.bufferView) {
|
|
Value exts, ext;
|
|
exts.SetObject();
|
|
ext.SetObject();
|
|
|
|
ext.AddMember("bufferView", StringRef(img.bufferView->id), w.mAl);
|
|
|
|
if (!img.mimeType.empty())
|
|
ext.AddMember("mimeType", StringRef(img.mimeType), w.mAl);
|
|
|
|
exts.AddMember("KHR_binary_glTF", ext, w.mAl);
|
|
obj.AddMember("extensions", exts, w.mAl);
|
|
return;
|
|
}
|
|
else if (img.HasData()) {
|
|
uri = "data:" + (img.mimeType.empty() ? "application/octet-stream" : img.mimeType);
|
|
uri += ";base64,";
|
|
Util::EncodeBase64(img.GetData(), img.GetDataLength(), uri);
|
|
}
|
|
else {
|
|
uri = img.uri;
|
|
}
|
|
|
|
obj.AddMember("uri", Value(uri, w.mAl).Move(), w.mAl);
|
|
}
|
|
|
|
namespace {
|
|
inline void WriteColorOrTex(Value& obj, TexProperty& prop, const char* propName, MemoryPoolAllocator<>& al)
|
|
{
|
|
if (prop.texture)
|
|
obj.AddMember(StringRef(propName), Value(prop.texture->id, al).Move(), al);
|
|
else {
|
|
Value col;
|
|
obj.AddMember(StringRef(propName), MakeValue(col, prop.color, al), al);
|
|
}
|
|
}
|
|
}
|
|
|
|
inline void Write(Value& obj, Material& m, AssetWriter& w)
|
|
{
|
|
Value v;
|
|
v.SetObject();
|
|
{
|
|
WriteColorOrTex(v, m.ambient, "ambient", w.mAl);
|
|
WriteColorOrTex(v, m.diffuse, "diffuse", w.mAl);
|
|
WriteColorOrTex(v, m.specular, "specular", w.mAl);
|
|
WriteColorOrTex(v, m.emission, "emission", w.mAl);
|
|
|
|
v.AddMember("shininess", m.shininess, w.mAl);
|
|
}
|
|
obj.AddMember("values", v, w.mAl);
|
|
}
|
|
|
|
namespace {
|
|
inline void WriteAttrs(AssetWriter& w, Value& attrs, Mesh::AccessorList& lst,
|
|
const char* semantic, bool forceNumber = false)
|
|
{
|
|
if (lst.empty()) return;
|
|
if (lst.size() == 1 && !forceNumber) {
|
|
attrs.AddMember(StringRef(semantic), Value(lst[0]->id, w.mAl).Move(), w.mAl);
|
|
}
|
|
else {
|
|
for (size_t i = 0; i < lst.size(); ++i) {
|
|
char buffer[32];
|
|
ai_snprintf(buffer, 32, "%s_%d", semantic, int(i));
|
|
attrs.AddMember(Value(buffer, w.mAl).Move(), Value(lst[i]->id, w.mAl).Move(), w.mAl);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
inline void Write(Value& obj, Mesh& m, AssetWriter& w)
|
|
{
|
|
Value primitives;
|
|
primitives.SetArray();
|
|
primitives.Reserve(m.primitives.size(), w.mAl);
|
|
|
|
for (size_t i = 0; i < m.primitives.size(); ++i) {
|
|
Mesh::Primitive& p = m.primitives[i];
|
|
Value prim;
|
|
prim.SetObject();
|
|
{
|
|
prim.AddMember("mode", Value(int(p.mode)).Move(), w.mAl);
|
|
|
|
if (p.material)
|
|
prim.AddMember("material", p.material->id, w.mAl);
|
|
|
|
if (p.indices)
|
|
prim.AddMember("indices", Value(p.indices->id, w.mAl).Move(), w.mAl);
|
|
|
|
Value attrs;
|
|
attrs.SetObject();
|
|
{
|
|
WriteAttrs(w, attrs, p.attributes.position, "POSITION");
|
|
WriteAttrs(w, attrs, p.attributes.normal, "NORMAL");
|
|
WriteAttrs(w, attrs, p.attributes.texcoord, "TEXCOORD", true);
|
|
WriteAttrs(w, attrs, p.attributes.color, "COLOR");
|
|
WriteAttrs(w, attrs, p.attributes.joint, "JOINT");
|
|
WriteAttrs(w, attrs, p.attributes.jointmatrix, "JOINTMATRIX");
|
|
WriteAttrs(w, attrs, p.attributes.weight, "WEIGHT");
|
|
}
|
|
prim.AddMember("attributes", attrs, w.mAl);
|
|
}
|
|
primitives.PushBack(prim, w.mAl);
|
|
}
|
|
|
|
obj.AddMember("primitives", primitives, w.mAl);
|
|
}
|
|
|
|
inline void Write(Value& obj, Node& n, AssetWriter& w)
|
|
{
|
|
|
|
if (n.matrix.isPresent) {
|
|
Value val;
|
|
obj.AddMember("matrix", MakeValue(val, n.matrix.value, w.mAl).Move(), w.mAl);
|
|
}
|
|
|
|
if (n.translation.isPresent) {
|
|
Value val;
|
|
obj.AddMember("translation", MakeValue(val, n.translation.value, w.mAl).Move(), w.mAl);
|
|
}
|
|
|
|
if (n.scale.isPresent) {
|
|
Value val;
|
|
obj.AddMember("scale", MakeValue(val, n.scale.value, w.mAl).Move(), w.mAl);
|
|
}
|
|
if (n.rotation.isPresent) {
|
|
Value val;
|
|
obj.AddMember("rotation", MakeValue(val, n.rotation.value, w.mAl).Move(), w.mAl);
|
|
}
|
|
|
|
AddRefsVector(obj, "children", n.children, w.mAl);
|
|
|
|
AddRefsVector(obj, "meshes", n.meshes, w.mAl);
|
|
}
|
|
|
|
inline void Write(Value& obj, Program& b, AssetWriter& w)
|
|
{
|
|
|
|
}
|
|
|
|
inline void Write(Value& obj, Sampler& b, AssetWriter& w)
|
|
{
|
|
|
|
}
|
|
|
|
inline void Write(Value& scene, Scene& s, AssetWriter& w)
|
|
{
|
|
AddRefsVector(scene, "nodes", s.nodes, w.mAl);
|
|
}
|
|
|
|
inline void Write(Value& obj, Shader& b, AssetWriter& w)
|
|
{
|
|
|
|
}
|
|
|
|
inline void Write(Value& obj, Skin& b, AssetWriter& w)
|
|
{
|
|
|
|
}
|
|
|
|
inline void Write(Value& obj, Technique& b, AssetWriter& w)
|
|
{
|
|
|
|
}
|
|
|
|
inline void Write(Value& obj, Texture& tex, AssetWriter& w)
|
|
{
|
|
if (tex.source) {
|
|
obj.AddMember("source", Value(tex.source->id, w.mAl).Move(), w.mAl);
|
|
}
|
|
}
|
|
|
|
inline void Write(Value& obj, Light& b, AssetWriter& w)
|
|
{
|
|
|
|
}
|
|
|
|
|
|
AssetWriter::AssetWriter(Asset& a)
|
|
: mDoc()
|
|
, mAsset(a)
|
|
, mAl(mDoc.GetAllocator())
|
|
{
|
|
mDoc.SetObject();
|
|
|
|
WriteMetadata();
|
|
WriteExtensionsUsed();
|
|
|
|
// Dump the contents of the dictionaries
|
|
for (size_t i = 0; i < a.mDicts.size(); ++i) {
|
|
a.mDicts[i]->WriteObjects(*this);
|
|
}
|
|
|
|
// Add the target scene field
|
|
if (mAsset.scene) {
|
|
mDoc.AddMember("scene", StringRef(mAsset.scene->id), mAl);
|
|
}
|
|
}
|
|
|
|
void AssetWriter::WriteFile(const char* path)
|
|
{
|
|
bool isBinary = mAsset.extensionsUsed.KHR_binary_glTF;
|
|
|
|
std::unique_ptr<IOStream> outfile
|
|
(mAsset.OpenFile(path, isBinary ? "wb" : "wt", true));
|
|
|
|
if (outfile == 0) {
|
|
throw DeadlyExportError("Could not open output file: " + std::string(path));
|
|
}
|
|
|
|
if (isBinary) {
|
|
// we will write the header later, skip its size
|
|
outfile->Seek(sizeof(GLB_Header), aiOrigin_SET);
|
|
}
|
|
|
|
StringBuffer docBuffer;
|
|
|
|
bool pretty = true;
|
|
if (!isBinary && pretty) {
|
|
PrettyWriter<StringBuffer> writer(docBuffer);
|
|
mDoc.Accept(writer);
|
|
}
|
|
else {
|
|
Writer<StringBuffer> writer(docBuffer);
|
|
mDoc.Accept(writer);
|
|
}
|
|
|
|
if (outfile->Write(docBuffer.GetString(), docBuffer.GetSize(), 1) != 1) {
|
|
throw DeadlyExportError("Failed to write scene data!");
|
|
}
|
|
|
|
if (isBinary) {
|
|
WriteBinaryData(outfile.get(), docBuffer.GetSize());
|
|
}
|
|
}
|
|
|
|
void AssetWriter::WriteBinaryData(IOStream* outfile, size_t sceneLength)
|
|
{
|
|
//
|
|
// write the body data
|
|
//
|
|
|
|
size_t bodyLength = 0;
|
|
if (Ref<Buffer> b = mAsset.GetBodyBuffer()) {
|
|
bodyLength = b->byteLength;
|
|
|
|
if (bodyLength > 0) {
|
|
size_t bodyOffset = sizeof(GLB_Header) + sceneLength;
|
|
bodyOffset = (bodyOffset + 3) & ~3; // Round up to next multiple of 4
|
|
|
|
outfile->Seek(bodyOffset, aiOrigin_SET);
|
|
|
|
if (outfile->Write(b->GetPointer(), b->byteLength, 1) != 1) {
|
|
throw DeadlyExportError("Failed to write body data!");
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
//
|
|
// write the header
|
|
//
|
|
|
|
GLB_Header header;
|
|
memcpy(header.magic, AI_GLB_MAGIC_NUMBER, sizeof(header.magic));
|
|
|
|
header.version = 1;
|
|
AI_SWAP4(header.version);
|
|
|
|
header.length = sizeof(header) + sceneLength + bodyLength;
|
|
AI_SWAP4(header.length);
|
|
|
|
header.sceneLength = sceneLength;
|
|
AI_SWAP4(header.sceneLength);
|
|
|
|
header.sceneFormat = SceneFormat_JSON;
|
|
AI_SWAP4(header.sceneFormat);
|
|
|
|
outfile->Seek(0, aiOrigin_SET);
|
|
|
|
if (outfile->Write(&header, 1, sizeof(header)) != sizeof(header)) {
|
|
throw DeadlyExportError("Failed to write the header!");
|
|
}
|
|
}
|
|
|
|
|
|
void AssetWriter::WriteMetadata()
|
|
{
|
|
Value asset;
|
|
asset.SetObject();
|
|
{
|
|
asset.AddMember("version", mAsset.asset.version, mAl);
|
|
|
|
asset.AddMember("generator", Value(mAsset.asset.generator, mAl).Move(), mAl);
|
|
}
|
|
mDoc.AddMember("asset", asset, mAl);
|
|
}
|
|
|
|
void AssetWriter::WriteExtensionsUsed()
|
|
{
|
|
Value exts;
|
|
exts.SetArray();
|
|
{
|
|
if (false)
|
|
exts.PushBack(StringRef("KHR_binary_glTF"), mAl);
|
|
|
|
if (false)
|
|
exts.PushBack(StringRef("KHR_materials_common"), mAl);
|
|
}
|
|
|
|
if (!exts.Empty())
|
|
mDoc.AddMember("extensionsUsed", exts, mAl);
|
|
}
|
|
|
|
template<class T>
|
|
void AssetWriter::WriteObjects(LazyDict<T>& d)
|
|
{
|
|
if (d.mObjs.empty()) return;
|
|
|
|
Value* container = &mDoc;
|
|
|
|
if (d.mExtId) {
|
|
Value* exts = FindObject(mDoc, "extensions");
|
|
if (!exts) {
|
|
mDoc.AddMember("extensions", Value().SetObject().Move(), mDoc.GetAllocator());
|
|
exts = FindObject(mDoc, "extensions");
|
|
}
|
|
|
|
if (!(container = FindObject(*exts, d.mExtId))) {
|
|
exts->AddMember(StringRef(d.mExtId), Value().SetObject().Move(), mDoc.GetAllocator());
|
|
container = FindObject(*exts, d.mExtId);
|
|
}
|
|
}
|
|
|
|
Value* dict;
|
|
if (!(dict = FindObject(*container, d.mDictId))) {
|
|
container->AddMember(StringRef(d.mDictId), Value().SetObject().Move(), mDoc.GetAllocator());
|
|
dict = FindObject(*container, d.mDictId);
|
|
}
|
|
|
|
for (size_t i = 0; i < d.mObjs.size(); ++i) {
|
|
if (d.mObjs[i]->IsSpecial()) continue;
|
|
|
|
Value obj;
|
|
obj.SetObject();
|
|
|
|
if (!d.mObjs[i]->name.empty()) {
|
|
obj.AddMember("name", StringRef(d.mObjs[i]->name.c_str()), mAl);
|
|
}
|
|
|
|
Write(obj, *d.mObjs[i], *this);
|
|
|
|
dict->AddMember(StringRef(d.mObjs[i]->id), obj, mAl);
|
|
}
|
|
}
|
|
|
|
template<class T>
|
|
struct LazyDictWriter< LazyDict<T> >
|
|
{
|
|
static void Write(LazyDict<T>& d, AssetWriter& w)
|
|
{
|
|
w.WriteObjects(d);
|
|
}
|
|
};
|
|
|
|
}
|
|
|
|
|