511 lines
16 KiB
C++
511 lines
16 KiB
C++
#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;
|
|
for(typename SaveMeshType::VertexIterator it = m.vert.begin();it != m.vert.end();++it)
|
|
{
|
|
if (!(it->IsD()))
|
|
{
|
|
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;
|
|
}
|
|
}
|
|
|
|
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 = (*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 |