zcy
/
dust3d
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Export motions to glTF(.glb) 

Each motion will be exported as an animtion of glTF
master
Jeremy Hu 2018-10-27 10:29:51 +08:00
parent db33a5e662
commit d00c26ca08
3 changed files with 149 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/documentwindow.cpp
View File

@ -1275,21 +1275,16 @@ void SkeletonDocumentWindow::exportGlbResult()
}
QApplication::setOverrideCursor(Qt::WaitCursor);
Outcome skeletonResult = m_document->currentPostProcessedOutcome();
std::vector<std::pair<QString, std::vector<std::pair<float, JointNodeTree>>>> exportMotions;
for (const auto &motionId: m_document->motionIdList) {
const Motion *motion = m_document->findMotion(motionId);
if (nullptr == motion)
continue;
exportMotions.push_back({motion->name, motion->jointNodeTrees});
}
GlbFileWriter glbFileWriter(skeletonResult, m_document->resultRigBones(), m_document->resultRigWeights(), filename,
m_document->textureImage);
m_document->textureImage, exportMotions.empty() ? nullptr : &exportMotions);
glbFileWriter.save();
/*
QFileInfo nameInfo(filename);
QString borderFilename = nameInfo.path() + QDir::separator() + nameInfo.completeBaseName() + "-BORDER.png";
if (m_document->textureBorderImage)
m_document->textureBorderImage->save(borderFilename);
QString ambientOcclusionFilename = nameInfo.path() + QDir::separator() + nameInfo.completeBaseName() + "-AMBIENT-OCCLUSION.png";
if (m_document->textureAmbientOcclusionImage)
m_document->textureAmbientOcclusionImage->save(ambientOcclusionFilename);
QString colorFilename = nameInfo.path() + QDir::separator() + nameInfo.completeBaseName() + "-COLOR.png";
if (m_document->textureColorImage)
m_document->textureColorImage->save(colorFilename);
*/
QApplication::restoreOverrideCursor();
}

142
src/glbfile.cpp
View File

@ -26,7 +26,8 @@ GlbFileWriter::GlbFileWriter(Outcome &outcome,
const std::vector<RiggerBone> *resultRigBones,
const std::map<int, RiggerVertexWeights> *resultRigWeights,
const QString &filename,
QImage *textureImage) :
QImage *textureImage,
const std::vector<std::pair<QString, std::vector<std::pair<float, JointNodeTree>>>> *motions) :
m_filename(filename),
m_outputNormal(true),
m_outputAnimation(true),
@ -42,6 +43,10 @@ GlbFileWriter::GlbFileWriter(Outcome &outcome,
m_outputUv = nullptr != triangleVertexUvs;
}
if (m_outputAnimation) {
m_outputAnimation = nullptr != motions && !motions->empty();
}
QDataStream binStream(&m_binByteArray, QIODevice::WriteOnly);
binStream.setFloatingPointPrecision(QDataStream::SinglePrecision);
binStream.setByteOrder(QDataStream::LittleEndian);
@ -72,10 +77,9 @@ GlbFileWriter::GlbFileWriter(Outcome &outcome,
m_json["asset"]["generator"] = APP_NAME " " APP_HUMAN_VER;
m_json["scenes"][0]["nodes"] = {0};
if (resultRigBones && resultRigWeights && !resultRigBones->empty()) {
constexpr int skeletonNodeStartIndex = 2;
if (resultRigBones && resultRigWeights && !resultRigBones->empty()) {
m_json["nodes"][0]["children"] = {
1,
skeletonNodeStartIndex
@ -348,6 +352,136 @@ GlbFileWriter::GlbFileWriter(Outcome &outcome,
}
}
if (m_outputAnimation) {
for (int animationIndex = 0; animationIndex < (int)motions->size(); ++animationIndex) {
const auto &motion = (*motions)[animationIndex];
m_json["animations"][animationIndex]["name"] = motion.first.toUtf8().constData();
int input = bufferViewIndex;
bufferViewFromOffset = (int)m_binByteArray.size();
m_json["bufferViews"][bufferViewIndex]["buffer"] = 0;
m_json["bufferViews"][bufferViewIndex]["byteOffset"] = bufferViewFromOffset;
float minTime = 1000000.0;
float maxTime = 0.0;
QStringList timeList;
float timePoint = 0;
for (const auto &keyframe: motion.second) {
binStream << (float)timePoint;
if (timePoint < minTime)
minTime = timePoint;
if (timePoint > maxTime)
maxTime = timePoint;
if (m_enableComment)
timeList.append(QString::number(timePoint));
timePoint += keyframe.first;
}
m_json["bufferViews"][bufferViewIndex]["byteLength"] = m_binByteArray.size() - bufferViewFromOffset;
alignBin();
if (m_enableComment)
m_json["accessors"][bufferViewIndex]["__comment"] = QString("/accessors/%1: times %2").arg(QString::number(bufferViewIndex)).arg(timeList.join(" ")).toUtf8().constData();
m_json["accessors"][bufferViewIndex]["bufferView"] = bufferViewIndex;
m_json["accessors"][bufferViewIndex]["byteOffset"] = 0;
m_json["accessors"][bufferViewIndex]["componentType"] = 5126;
m_json["accessors"][bufferViewIndex]["count"] = motion.second.size();
m_json["accessors"][bufferViewIndex]["type"] = "SCALAR";
m_json["accessors"][bufferViewIndex]["max"][0] = maxTime;
m_json["accessors"][bufferViewIndex]["min"][0] = minTime;
bufferViewIndex++;
std::set<int> rotatedJoints;
std::set<int> translatedJoints;
for (const auto &keyframe: motion.second) {
for (int i = 0; i < (int)keyframe.second.nodes().size() && i < (int)boneNodes.size(); ++i) {
const auto &src = boneNodes[i];
const auto &dest = keyframe.second.nodes()[i];
if (!qFuzzyCompare(src.rotation, dest.rotation))
rotatedJoints.insert(i);
if (!qFuzzyCompare(src.translation, dest.translation))
translatedJoints.insert(i);
}
}
int sampler = 0;
int channel = 0;
for (const auto &jointIndex: rotatedJoints) {
int output = bufferViewIndex;
bufferViewFromOffset = (int)m_binByteArray.size();
m_json["bufferViews"][bufferViewIndex]["buffer"] = 0;
m_json["bufferViews"][bufferViewIndex]["byteOffset"] = bufferViewFromOffset;
QStringList rotationList;
for (int frame = 0; frame < (int)motion.second.size(); frame++) {
const auto &keyframe = motion.second[frame];
const auto &rotation = keyframe.second.nodes()[jointIndex].rotation;
float x = rotation.x();
float y = rotation.y();
float z = rotation.z();
float w = rotation.scalar();
binStream << (float)x << (float)y << (float)z << (float)w;
if (m_enableComment)
rotationList.append(QString("%1:<%2,%3,%4,%5>").arg(QString::number(frame)).arg(QString::number(x)).arg(QString::number(y)).arg(QString::number(z)).arg(QString::number(w)));
}
m_json["bufferViews"][bufferViewIndex]["byteLength"] = m_binByteArray.size() - bufferViewFromOffset;
alignBin();
if (m_enableComment)
m_json["accessors"][bufferViewIndex]["__comment"] = QString("/accessors/%1: rotation %2").arg(QString::number(bufferViewIndex)).arg(rotationList.join(" ")).toUtf8().constData();
m_json["accessors"][bufferViewIndex]["bufferView"] = bufferViewIndex;
m_json["accessors"][bufferViewIndex]["byteOffset"] = 0;
m_json["accessors"][bufferViewIndex]["componentType"] = 5126;
m_json["accessors"][bufferViewIndex]["count"] = motion.second.size();
m_json["accessors"][bufferViewIndex]["type"] = "VEC4";
bufferViewIndex++;
m_json["animations"][animationIndex]["samplers"][sampler]["input"] = input;
m_json["animations"][animationIndex]["samplers"][sampler]["interpolation"] = "LINEAR";
m_json["animations"][animationIndex]["samplers"][sampler]["output"] = output;
m_json["animations"][animationIndex]["channels"][channel]["sampler"] = sampler;
m_json["animations"][animationIndex]["channels"][channel]["target"]["node"] = skeletonNodeStartIndex + jointIndex;
m_json["animations"][animationIndex]["channels"][channel]["target"]["path"] = "rotation";
sampler++;
channel++;
for (const auto &jointIndex: translatedJoints) {
int output = bufferViewIndex;
bufferViewFromOffset = (int)m_binByteArray.size();
m_json["bufferViews"][bufferViewIndex]["buffer"] = 0;
m_json["bufferViews"][bufferViewIndex]["byteOffset"] = bufferViewFromOffset;
for (int frame = 0; frame < (int)motion.second.size(); frame++) {
const auto &keyframe = motion.second[frame];
const auto &translation = keyframe.second.nodes()[jointIndex].translation;
binStream << (float)translation.x() << (float)translation.y() << (float)translation.z();
}
m_json["bufferViews"][bufferViewIndex]["byteLength"] = m_binByteArray.size() - bufferViewFromOffset;
alignBin();
if (m_enableComment)
m_json["accessors"][bufferViewIndex]["__comment"] = QString("/accessors/%1: translation").arg(QString::number(bufferViewIndex)).toUtf8().constData();
m_json["accessors"][bufferViewIndex]["bufferView"] = bufferViewIndex;
m_json["accessors"][bufferViewIndex]["byteOffset"] = 0;
m_json["accessors"][bufferViewIndex]["componentType"] = 5126;
m_json["accessors"][bufferViewIndex]["count"] = motion.second.size();
m_json["accessors"][bufferViewIndex]["type"] = "VEC3";
bufferViewIndex++;
m_json["animations"][animationIndex]["samplers"][sampler]["input"] = input;
m_json["animations"][animationIndex]["samplers"][sampler]["interpolation"] = "LINEAR";
m_json["animations"][animationIndex]["samplers"][sampler]["output"] = output;
m_json["animations"][animationIndex]["channels"][channel]["sampler"] = sampler;
m_json["animations"][animationIndex]["channels"][channel]["target"]["node"] = skeletonNodeStartIndex + jointIndex;
m_json["animations"][animationIndex]["channels"][channel]["target"]["path"] = "translation";
sampler++;
channel++;
}
}
}
}
// Images should be put in the end of the buffer, because we are not using accessors
if (nullptr != textureImage) {
m_json["textures"][0]["sampler"] = 0;
m_json["textures"][0]["source"] = 0;
@ -373,7 +507,7 @@ GlbFileWriter::GlbFileWriter(Outcome &outcome,
m_json["buffers"][0]["byteLength"] = m_binByteArray.size();
auto jsonString = m_json.dump(4);
auto jsonString = m_enableComment ? m_json.dump(4) : m_json.dump();
jsonStream.writeRawData(jsonString.data(), jsonString.size());
alignJson();
}

3
src/glbfile.h
View File

@ -19,7 +19,8 @@ public:
const std::vector<RiggerBone> *resultRigBones,
const std::map<int, RiggerVertexWeights> *resultRigWeights,
const QString &filename,
QImage *textureImage=nullptr);
QImage *textureImage=nullptr,
const std::vector<std::pair<QString, std::vector<std::pair<float, JointNodeTree>>>> *motions=nullptr);
bool save();
private:
QString m_filename;