vcglib/wrap/io_trimesh/export_dae.h

#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