vcglib/wrap/io_trimesh/export_dae.h

#ifndef __VCGLIB_EXPORTERDAE
#define __VCGLIB_EXPORTERDAE

#include <FCollada.h>
#include <FCDocument/FCDocument.h>
#include <FCDocument/FCDLibrary.h>
#include <FCDocument/FCDGeometry.h>
#include <FCDocument/FCDGeometryMesh.h>
#include <FCDocument/FCDGeometrySource.h>
#include <FCDocument/FCDGeometryPolygons.h>
#include <FCDocument/FCDSceneNode.h>
#include <FCDocument/FCDGeometryInstance.h>

namespace vcg {
namespace tri {
namespace io {

//FCollada Library assumes that SaveMeshType::ScalarType is always a float

template<typename SaveMeshType>
class ExporterDAE
{
public:
	static int Save(SaveMeshType &m, const char * filename)
	{
		unsigned int ncomp = sizeof(SaveMeshType::CoordType) / sizeof(SaveMeshType::ScalarType);
		FCDocument* doc = new FCDocument();
		FCDGeometryLibrary* geolib = doc->GetGeometryLibrary();
		FCDGeometry* geo = geolib->AddEntity();
		geo->SetDaeId("vcg-mesh");
		FCDGeometryMesh* mesh = geo->CreateMesh();

		FCDGeometrySource* vsource = mesh->AddVertexSource();
		vsource->SetDaeId("vcg-position");
		vsource->SetSourceType(FUDaeGeometryInput::POSITION);
		FCDGeometryPolygons* polyset = mesh->AddPolygons();
		//for(UInt32List::iterator it = polyinput->indices.begin();it != polyinput->indices.end();++it)
		//	*it = 100;
		
		FloatList plist(m.vert.size() * ncomp);
		
		SaveMeshType::CoordType point;
		unsigned int ii = 0;
		for(FloatList::iterator it = plist.begin();it != plist.end();++it)
		{
			if ((ii % ncomp) == 0)
			{
				point = m.vert[ii / ncomp].P();
			}
			(*it) = point[ii % ncomp];
			++ii;
		}
		
		vsource->SetSourceData(plist,ncomp);

		unsigned int jj = 0;
		FCDGeometryPolygonsInput* ps = mesh->GetPolygons(0)->FindInput(FUDaeGeometryInput::POSITION);
		//ps->indices.resize(m.face.size() * 3);
		for(SaveMeshType::FaceIterator itf = m.face.begin();itf != m.face.end();++itf)
		{
			if( !(*itf).IsD() )
			{
				polyset->AddFace(3);
				ps->indices[jj] = (itf->V(0) - &(m.vert[0]));
				ps->indices[jj + 1] = (itf->V(1) - &(m.vert[0]));
				ps->indices[jj + 2] = (itf->V(2) - &(m.vert[0]));
				jj += 3;
			}
		}

		FloatList nlist;

		FCDGeometrySource* nsource = mesh->AddSource();
		nsource->SetDaeId("vcg-wedgenormal");
		nsource->SetSourceType(FUDaeGeometryInput::NORMAL);
		//in the oldmesh format of vcg library normal is always a Point3
		nlist.resize(m.face.size() * 3 * 3);
		
		unsigned int cc = 0;
		jj = 0;
		
		FCDGeometryPolygonsInput* nm = polyset->AddInput(nsource,1);
		//nm->semantic = FUDaeGeometryInput::NORMAL;
		nm->indices.resize(m.face.size() * 3);
		
		for(SaveMeshType::FaceIterator fit = m.face.begin();fit != m.face.end();++fit)
		{
			if( !(*fit).IsD() )
			{
				vcg::Point3<SaveMeshType::ScalarType> norm = (((fit->V(1))->P() - (fit->V(0))->P()) ^ ((fit->V(2))->P() - (fit->V(0))->P())).Normalize();
				for(unsigned int vv = 0; vv < 3;++vv)
				{
					nm->indices[jj + vv] = cc / 3;
					for(unsigned int hh = 0; hh < 3;++hh)
					{
						nlist[cc] = norm[hh];
						++cc;
					}
				} 
				jj += 3;
			}
		}
		nsource->SetSourceData(nlist,3);
		
		FCDSceneNode* root = doc->AddVisualScene();
		root->SetDaeId("vcg-scenenode");
		FCDSceneNode* scenenod = root->AddChildNode();
		scenenod->AddInstance(geo);
		//root->
		doc->WriteToFile(FUStringConversion::ToFString(filename));
		return 0;
	}

	static int Save(SaveMeshType &m, const char * filename,AdditionalInfo*& in)
	{
		unsigned int ncomp = sizeof(SaveMeshType::CoordType) / sizeof(SaveMeshType::ScalarType);
		assert(in != NULL);

		AdditionalInfoDAE* inf = static_cast<AdditionalInfoDAE*>(in);
		InfoDAE* info = inf->dae;
		FCDGeometryLibrary* geolib = info->doc->GetGeometryLibrary();
		
		size_t dim = geolib->GetEntityCount();

		for(unsigned int cont = 0;cont < dim;++cont)
		{
				delete geolib->GetEntity(cont);
		}

		FCDGeometry* geo = geolib->AddEntity();
		geo->SetDaeId("vcg-mesh");
		FCDGeometryMesh* mesh = geo->CreateMesh();

		FCDGeometrySource* vsource = mesh->AddVertexSource();
		vsource->SetDaeId("vcg-position");
		vsource->SetSourceType(FUDaeGeometryInput::POSITION);
		FCDGeometryPolygons* polyset = mesh->AddPolygons();
		FloatList plist(m.vert.size() * ncomp);
		
		SaveMeshType::CoordType point;
		unsigned int ii = 0;
		for(FloatList::iterator it = plist.begin();it != plist.end();++it)
		{
			if ((ii % ncomp) == 0)
			{
				point = m.vert[ii / ncomp].P();
			}
			(*it) = point[ii % ncomp];
			++ii;
		}
		
		vsource->SetSourceData(plist,ncomp);

		unsigned int jj = 0;
		FCDGeometryPolygonsInput* ps = mesh->GetPolygons(0)->FindInput(FUDaeGeometryInput::POSITION);
		//ps->indices.resize(m.face.size() * 3);
		for(SaveMeshType::FaceIterator itf = m.face.begin();itf != m.face.end();++itf)
		{
			if( !(*itf).IsD() )
			{
				polyset->AddFace(3);
				ps->indices[jj] = (itf->V(0) - &(m.vert[0]));
				ps->indices[jj + 1] = (itf->V(1) - &(m.vert[0]));
				ps->indices[jj + 2] = (itf->V(2) - &(m.vert[0]));
				jj += 3;
			}
		}

		FloatList nlist;

		FCDGeometrySource* nsource = mesh->AddSource();
		nsource->SetDaeId("vcg-wedgenormal");
		nsource->SetSourceType(FUDaeGeometryInput::NORMAL);
		//in the oldmesh format of vcg library normal is always a Point3
		nlist.resize(m.face.size() * 3 * 3);
		
		unsigned int cc = 0;
		jj = 0;
		
		FCDGeometryPolygonsInput* nm = polyset->AddInput(nsource,1);
		//nm->semantic = FUDaeGeometryInput::NORMAL;
		nm->indices.resize(m.face.size() * 3);
		
		for(SaveMeshType::FaceIterator fit = m.face.begin();fit != m.face.end();++fit)
		{
			if( !(*fit).IsD() )
			{
				vcg::Point3<SaveMeshType::ScalarType> norm = (((fit->V(1))->P() - (fit->V(0))->P()) ^ ((fit->V(2))->P() - (fit->V(0))->P())).Normalize();
				for(unsigned int vv = 0; vv < 3;++vv)
				{
					nm->indices[jj + vv] = cc / 3;
					for(unsigned int hh = 0; hh < 3;++hh)
					{
						nlist[cc] = norm[hh];
						++cc;
					}
				} 
				jj += 3;
			}
		}
		nsource->SetSourceData(nlist,3);
			

		FCDSceneNode* root = info->doc->GetVisualSceneRoot ();
		FCDSceneNodeTrackList& listchild = root->GetChildren();
		std::vector<FCDSceneNodeTrackList::iterator> ve;
		//size_t n = listchild.size();
		for(FCDSceneNodeTrackList::iterator itlc = listchild.begin();itlc != listchild.end();++itlc)
		{
			FCDEntityInstanceContainer& li = (*itlc)->GetInstances();
			for(FCDEntityInstanceContainer::iterator itli = li.begin();itli != li.end();++itli)
			{
				if ((*itli)->GetEntityType() == FCDEntity::GEOMETRY)
				{
					ve.push_back(itlc);
				}
			}
		}
		for(std::vector<FCDSceneNodeTrackList::iterator>::iterator itld = ve.begin();itld != ve.end();++itld)
			listchild.erase((*itld));
		ve.clear();
		root->SetDaeId("vcg-scenenode");
		FCDSceneNode* scenenod = root->AddChildNode();
		scenenod->AddInstance(geo);
		//root->
		info->doc->WriteToFile(FUStringConversion::ToFString(filename));
		return 0;
	}

	static int GetExportMaskCapability()
	{
		int capability = 0;

		//camera
		//capability |= MeshModel::IOM_CAMERA;

		//vert
		//capability |= MeshModel::IOM_VERTTEXCOORD;

		//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