vcglib/wrap/io_trimesh/export_dae.h

542 lines
18 KiB
C++

/****************************************************************************
* VCGLib o o *
* Visual and Computer Graphics Library o o *
* _ O _ *
* Copyright(C) 2004 \/)\/ *
* 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 GNU General Public License as published by *
* the Free Software 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 *
* GNU General Public License (http://www.gnu.org/licenses/gpl.txt) *
* for more details. *
* *
****************************************************************************/
/****************************************************************************
History
$Log: not supported by cvs2svn $
****************************************************************************/
#ifndef __VCGLIB_EXPORTERDAE
#define __VCGLIB_EXPORTERDAE
#include<wrap/io_trimesh/util_dae.h>
namespace vcg {
namespace tri {
namespace io {
//FCollada Library assumes that SaveMeshType::ScalarType is always a float
template<typename SaveMeshType>
class ExporterDAE : public UtilDAE
{
private:
//static void SaveTexture(QDomDocument& doc,QDomNode& n,const std::vector<QString>& stv)
//{
// removeChildNode(doc,"library_images");
// QDomElement el = doc.createElement("library_images");
// for(int img = 0;img < stv.size();++img)
// {
// QDomElement imgnode = doc.createElement("image");
// imgnode.setAttribute("id","file"+QString::number(img));
// imgnode.setAttribute("name","file"+QString::number(img));
// QDomElement tex = doc.createElement("init_from");
// QDomText txname = doc.createTextNode(stv[img]);
// tex.appendChild(txname);
// imgnode.appendChild(tex);
// el.appendChild(imgnode);
// }
// n.appendChild(el);
//}
static void SaveTextureName(QDomDocument doc,QDomNode lbim,const std::vector<std::string>& stv)
{
for(unsigned int img = 0;img < stv.size();++img)
{
QDomElement imgnode = doc.createElement("image");
imgnode.setAttribute("id","file"+QString::number(img));
imgnode.setAttribute("name","file"+QString::number(img));
QDomElement tex = doc.createElement("init_from");
QDomText txname = doc.createTextNode(QString(stv[img].c_str()));
tex.appendChild(txname);
imgnode.appendChild(tex);
lbim.appendChild(imgnode);
}
}
static void SaveTexture(QDomDocument doc,QDomNode n,const std::vector<std::string>& stv)
{
QDomElement el = doc.createElement("library_images");
SaveTextureName(doc,el,stv);
n.appendChild(el);
}
static void SaveTexture(QDomDocument doc,const std::vector<std::string>& stv)
{
QDomNodeList lbim = doc.elementsByTagName("library_images");
if (lbim.size() == 1)
removeChildNode(lbim.at(0));
SaveTextureName(doc,lbim.at(0),stv);
}
static void CreateVertInput(QDomDocument doc,QDomNode vert,const QString& attr,const QString& ref)
{
QDomElement vinp_pos = doc.createElement("input");
vinp_pos.setAttribute("semantic",attr);
vinp_pos.setAttribute("source",ref);
vert.appendChild(vinp_pos);
}
static void CreateFaceInput(QDomDocument doc,QDomNode tri,const QString& attr,const QString& ref,const int offset)
{
QDomElement tinp_vert = doc.createElement("input");
tinp_vert.setAttribute("offset",QString::number(offset));
tinp_vert.setAttribute("semantic",attr);
tinp_vert.setAttribute("source",ref);
tri.appendChild(tinp_vert);
}
static void CreateSource(QDomDocument doc,QDomNode meshnode,const QString& attr,const QDomText& val,int nvert)
{
int nel;
std::vector<QString> coord;
if ((attr == "positions") || (attr == "normals") || (attr == "wnornals"))
{
nel = 3;
coord.push_back("X");
coord.push_back("Y");
coord.push_back("Z");
}
else
{
if (attr == "colors")
{
nel = 4;
coord.push_back("R");
coord.push_back("G");
coord.push_back("B");
coord.push_back("A");
}
else
{
nel = 2;
coord.push_back("U");
coord.push_back("V");
}
}
QDomElement srcnmnode = doc.createElement("source");
srcnmnode.setAttribute("id","vcg-mesh-"+attr);
srcnmnode.setAttribute("name","vcg-mesh-"+attr);
QDomElement arraynmnode = doc.createElement("float_array");
arraynmnode.setAttribute("id","vcg-mesh-"+attr+"-array");
arraynmnode.setAttribute("count",QString::number(nvert * nel));
QDomElement technode2 = doc.createElement("technique_common");
QDomElement accnode2 = doc.createElement("accessor");
accnode2.setAttribute("source","#vcg-mesh-"+attr+"-array");
accnode2.setAttribute("count",QString::number(nvert));
accnode2.setAttribute("stride",QString::number(nel));
for(int jj = 0; jj < nel;++jj)
{
QDomElement parxnode2 = doc.createElement("param");
parxnode2.setAttribute("name",coord[jj]);
parxnode2.setAttribute("type","float");
accnode2.appendChild(parxnode2);
}
technode2.appendChild(accnode2);
arraynmnode.appendChild(val);
srcnmnode.appendChild(arraynmnode);
srcnmnode.appendChild(technode2);
meshnode.appendChild(srcnmnode);
}
static int SaveMesh(SaveMeshType& m,QDomDocument& doc,QDomNode meshnode,const int mask)
{
QString arrp;
arrp.reserve(10 * 3 * m.vert.size());
QString arrn;
if(mask & vcg::tri::io::Mask::IOM_VERTNORMAL | mask & vcg::tri::io::Mask::IOM_WEDGNORMAL)
arrn.reserve(10 * 3 * m.vert.size());
QString arrt;
if(mask & vcg::tri::io::Mask::IOM_VERTTEXCOORD)
arrt.reserve(10 * 2 * m.vert.size());
QString arrc;
if(mask & vcg::tri::io::Mask::IOM_VERTCOLOR)
arrc.reserve(5 * 4 * m.vert.size());
int nvert = 0;
std::vector<int> VertexInd(m.vert.size());
for(typename SaveMeshType::VertexIterator it = m.vert.begin();it != m.vert.end();++it)
{
if (!(it->IsD()))
{
VertexInd[it-m.vert.begin()]=nvert;
arrp.append(QString::number(float(it->P().X())).append(" ").append(QString::number(float(it->P().Y()))).append(" ").append(QString::number(float(it->P().Z()))).append(" "));
//if(mask & vcg::tri::io::Mask::IOM_VERTNORMAL)
arrn.append(QString::number(float(it->N().X())).append(" ").append(QString::number(float(it->N().Y()))).append(" ").append(QString::number(float(it->N().Z()))).append(" "));
if(mask & vcg::tri::io::Mask::IOM_VERTTEXCOORD)
arrt.append(QString::number(float(it->T().u())).append(" ").append(QString::number(float(it->T().v()))).append(" "));
if(mask & vcg::tri::io::Mask::IOM_VERTCOLOR)
arrc.append(QString::number(float(it->C().X())).append(" ").append(QString::number(float(it->C().Y()))).append(" ").append(QString::number(float(it->C().Z()))).append(" ").append(QString::number(float(it->C().W()))).append(" "));
++nvert;
}
}
assert(nvert==m.vn);
QDomText ap = doc.createTextNode(arrp);
CreateSource(doc,meshnode,"positions",ap,nvert);
/*if(mask & vcg::tri::io::Mask::IOM_VERTNORMAL | mask & vcg::tri::io::Mask::IOM_WEDGNORMAL)
{*/
QDomText an = doc.createTextNode(arrn);
CreateSource(doc,meshnode,"normals",an,nvert);
/*}*/
if(mask & vcg::tri::io::Mask::IOM_VERTTEXCOORD)
{
QDomText at = doc.createTextNode(arrt);
CreateSource(doc,meshnode,"textcoords",at,nvert);
}
if(mask & vcg::tri::io::Mask::IOM_VERTCOLOR)
{
QDomText ac = doc.createTextNode(arrc);
CreateSource(doc,meshnode,"colors",ac,nvert);
}
QDomElement vert = doc.createElement("vertices");
vert.setAttribute("id","vcg-mesh-vertices");
CreateVertInput(doc,vert,"POSITION","#vcg-mesh-positions");
//if(mask & vcg::tri::io::Mask::IOM_VERTNORMAL)
CreateVertInput(doc,vert,"NORMAL","#vcg-mesh-normals");
if(mask & vcg::tri::io::Mask::IOM_VERTCOLOR)
CreateVertInput(doc,vert,"COLOR","#vcg-mesh-colors");
if(mask & vcg::tri::io::Mask::IOM_VERTTEXCOORD)
CreateVertInput(doc,vert,"TEXCOORD","#vcg-mesh-normals");
meshnode.appendChild(vert);
QDomElement tri = doc.createElement("triangles");
CreateFaceInput(doc,tri,"VERTEX","#vcg-mesh-vertices",0);
QDomElement poly = doc.createElement("p");
int nface = 0;
int nattr = 1;
QString triangles_wn;
if (mask & MeshModel::IOM_WEDGNORMAL)
{
triangles_wn.reserve(3* 10 * m.face.size());
CreateFaceInput(doc,tri,"NORMAL","#vcg-mesh-wnormals",nattr);
++nattr;
}
QString triangles_wt;
if (mask & MeshModel::IOM_WEDGTEXCOORD)
{
triangles_wt.reserve(2 * 10 * m.face.size());
CreateFaceInput(doc,tri,"TEXCOORD","#vcg-mesh-wtext",nattr);
++nattr;
}
QString triangles_tess;
triangles_tess.reserve(nattr * 3 * 10 * m.face.size());
int wn = 0;
int wt = 0;
for(typename SaveMeshType::FaceIterator itf = m.face.begin();itf != m.face.end();++itf)
{
if (!(itf->IsD()))
{
for(unsigned int ii = 0;ii < 3;++ii)
{
int ind_v = VertexInd[(*itf).V(ii) - &(m.vert[0])];
if (triangles_tess == "")
triangles_tess = QString::number(ind_v);
else triangles_tess.append(" ").append(QString::number(ind_v));
if (mask & MeshModel::IOM_WEDGNORMAL)
{
triangles_tess.append(" ").append(QString::number(wn));
++wn;
triangles_wn.append(QString::number(float((*itf).WN(ii).X())).append(" ").append(QString::number(float((*itf).WN(ii).Y()))).append(" ").append(QString::number(float((*itf).WN(ii).Z()))).append(" "));
}
if (mask & MeshModel::IOM_WEDGTEXCOORD)
{
triangles_tess.append(" ").append(QString::number(wt));
++wt;
triangles_wt.append(QString::number((*itf).WT(ii).u()).append(" ").append(QString::number((*itf).WT(ii).v())).append(" "));
}
}
++nface;
}
}
tri.setAttribute("count",nface);
if (mask & MeshModel::IOM_WEDGNORMAL)
{
QDomText wnt = doc.createTextNode(triangles_wn);
CreateSource(doc,meshnode,"wnormals",wnt,nface * 3);
}
if (mask & MeshModel::IOM_WEDGTEXCOORD)
{
QDomText wtt = doc.createTextNode(triangles_wt);
CreateSource(doc,meshnode,"wtext",wtt,nface * 3);
}
QDomText tri_list = doc.createTextNode(triangles_tess);
poly.appendChild(tri_list);
tri.appendChild(poly);
meshnode.appendChild(tri);
return E_NOERROR;
}
public:
static int Save(SaveMeshType &m, const char * filename,const int mask)
{
QDomDocument doc("mydoc");
QDomElement coll = doc.createElement("COLLADA");
coll.setAttribute("version","1.4.1");
coll.setAttribute("xmlns","http://www.collada.org/2005/11/COLLADASchema");
doc.appendChild(coll);
QDomElement ass = doc.createElement("asset");
coll.appendChild(ass);
if (m.textures.size() != 0)
{
QDomNodeList cl = doc.elementsByTagName("COLLADA");
assert(cl.size() == 1);
SaveTexture(doc,cl.at(0),m.textures);
}
QDomElement geolib = doc.createElement("library_geometries");
QDomElement geonode = doc.createElement("geometry");
geonode.setAttribute("id","vcg-mesh-lib");
geonode.setAttribute("name","vcg-mesh");
QDomElement meshnode = doc.createElement("mesh");
int res = SaveMesh(m,doc,meshnode,mask);
if (res != 0)
return res;
geonode.appendChild(meshnode);
geolib.appendChild(geonode);
coll.appendChild(geolib);
QDomElement vslib = doc.createElement("library_visual_scenes");
QDomElement vslnode = doc.createElement("visual_scene");
vslnode.setAttribute("id","vcg-scene-node");
vslnode.setAttribute("name","vcg-untitled");
QDomElement vcgnode = doc.createElement("node");
vcgnode.setAttribute("id","vcg-node");
vcgnode.setAttribute("name","vcg-untitled");
QDomElement vcginst = doc.createElement("instance_geometry");
vcginst.setAttribute("url","#vcg-mesh-lib");
vcgnode.appendChild(vcginst);
vslnode.appendChild(vcgnode);
vslib.appendChild(vslnode);
coll.appendChild(vslib);
QDomElement vsscn = doc.createElement("scene");
QDomElement vsnode = doc.createElement("instance_visual_scene");
vsnode.setAttribute("url","#vcg-scene-node");
vsscn.appendChild(vsnode);
coll.appendChild(vsscn);
QString st = doc.toString();
QFile file(filename);
if (!file.open(QIODevice::ReadWrite | QIODevice::Truncate))
return E_CANTOPEN;
file.write(st.toAscii());
file.close();
return E_NOERROR;
}
static int Save(SaveMeshType &m, const char * filename,AdditionalInfo*& in, const int mask)
{
/*unsigned int ncomp = sizeof(SaveMeshType::CoordType) / sizeof(SaveMeshType::ScalarType);*/
assert(in != NULL);
AdditionalInfoDAE* inf = static_cast<AdditionalInfoDAE*>(in);
InfoDAE* info = inf->dae;
QDomNodeList scenelst = info->doc->elementsByTagName("scene");
//removeChildNode(scenelst,"instance_visual_scene");
//assert(scenelst.size() == 1);
if (m.textures.size() != 0)
SaveTexture(*(info->doc),m.textures);
QDomNodeList vsscene = info->doc->elementsByTagName("library_visual_scenes");
QDomNodeList geolib = info->doc->elementsByTagName("library_geometries");
assert(geolib.size() == 1);
if (info->doc->elementsByTagName("instance_geometry").size() != 0)
{
QDomNodeList invisscnlst = info->doc->elementsByTagName("instance_visual_scene");
assert(invisscnlst.size() == 1);
QString url;
referenceToANodeAttribute(invisscnlst.at(0),"url",url);
assert(vsscene.size() == 1);
QDomNode sc = findNodeBySpecificAttributeValue(vsscene.at(0),"visual_scene","id",url);
//removeChildNode(sc.childNodes(),"instance_geometry");
for(int no = 0;no < sc.childNodes().size();++no)
{
QDomNode oldnode = sc.childNodes().at(no);
QDomNodeList lst = sc.childNodes().at(no).toElement().elementsByTagName("instance_geometry");
int lst_size = lst.size();
if (lst_size != 0)
{
for(int ingeo = 0;ingeo < lst_size;++ingeo)
{
QString url;
referenceToANodeAttribute(lst.at(ingeo),"url",url);
QDomNode oldgeo = findNodeBySpecificAttributeValue(geolib.at(0),"geometry","id",url);
geolib.at(0).removeChild(oldgeo);
}
sc.removeChild(oldnode);
}
}
QDomElement vcgnode = info->doc->createElement("node");
vcgnode.setAttribute("id","vcg-node");
vcgnode.setAttribute("name","vcg-untitled");
QDomElement vcginst = info->doc->createElement("instance_geometry");
vcginst.setAttribute("url","#vcg-mesh-lib");
vcgnode.appendChild(vcginst);
sc.appendChild(vcgnode);
vsscene.at(0).appendChild(sc);
QDomNodeList geolib = info->doc->elementsByTagName("library_geometries");
assert(geolib.size() == 1);
//removeChildNode(geolib.at(0));
/*QDomElement mshnode;
mshnode.setTagName("mesh");*/
geolib.at(0).childNodes().size();
removeChildNode(geolib.at(0),QString("geometry"),QString("id"),QString("vcg-mesh-lib"));
QDomElement geonode = info->doc->createElement("geometry");
geonode.setAttribute("id","vcg-mesh-lib");
geonode.setAttribute("name","vcg-mesh");
QDomElement meshnode = info->doc->createElement("mesh");
int res = SaveMesh(m,*(info->doc),meshnode,mask);
if (res != 0) return res;
geonode.appendChild(meshnode);
geolib.at(0).appendChild(geonode);
}
else
{
/* removeChildNodeList(scenelst,QString("instance_visual_scene")); */
for(int vsscn = 0;vsscn < scenelst.size();++vsscn)
{
QString url = scenelst.at(vsscn).toElement().attribute("url");
}
QDomElement vsnode = info->doc->createElement("instance_visual_scene");
vsnode.setAttribute("url","#vcg-scene-node");
scenelst.at(0).appendChild(vsnode);
assert(vsscene.size() == 1);
removeChildNodeList(vsscene,QString("visual_scene"),QString("id"),QString("vcg-scene-node"));
QDomElement vslnode = info->doc->createElement("visual_scene");
vslnode.setAttribute("id","vcg-scene-node");
vslnode.setAttribute("name","vcg-untitled");
QDomElement vcgnode = info->doc->createElement("node");
vcgnode.setAttribute("id","vcg-node");
vcgnode.setAttribute("name","vcg-untitled");
QDomElement vcginst = info->doc->createElement("instance_geometry");
vcginst.setAttribute("url","#vcg-mesh-lib");
vcgnode.appendChild(vcginst);
vslnode.appendChild(vcgnode);
vsscene.at(0).appendChild(vslnode);
QDomNodeList geolib = info->doc->elementsByTagName("library_geometries");
assert(geolib.size() == 1);
removeChildNode(geolib.at(0),QString("geometry"),QString("id"),QString("vcg-mesh-lib"));
QDomElement geonode = info->doc->createElement("geometry");
geonode.setAttribute("id","vcg-mesh-lib");
geonode.setAttribute("name","vcg-mesh");
QDomElement meshnode = info->doc->createElement("mesh");
int res = SaveMesh(m,*(info->doc),meshnode,mask);
if (res != 0) return res;
geonode.appendChild(meshnode);
geolib.at(0).appendChild(geonode);
}
QString st = info->doc->toString();
QFile file(filename);
if (!file.open(QIODevice::ReadWrite | QIODevice::Truncate))
return E_CANTOPEN;
file.write(st.toAscii());
file.close();
return E_NOERROR;
}
static int GetExportMaskCapability()
{
int capability = 0;
//camera
//capability |= MeshModel::IOM_CAMERA;
//vert
capability |= MeshModel::IOM_VERTNORMAL;
capability |= MeshModel::IOM_VERTTEXCOORD;
capability |= MeshModel::IOM_VERTCOLOR;
//capability |= MeshModel::
////face
////capability |= MeshModel::IOM_FACEFLAGS;
////capability |= MeshModel::IOM_FACECOLOR;
//capability |= MeshModel::IOM_FACENORMAL;
////wedg
capability |= MeshModel::IOM_WEDGTEXCOORD;
capability |= MeshModel::IOM_WEDGNORMAL;
return capability;
}
};
}
}
}
#endif