vcglib/wrap/io_trimesh/import_dae.h

#ifndef __VCGLIB_IMPORTERDAE
#define __VCGLIB_IMPORTERDAE

//importer for collada's files

#include <FCollada.h>
#include <FUtils/FUStringConversion.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 <wrap/gl/trimesh.h>
#include <vcg/complex/trimesh/update/normal.h>
#include <vcg/complex/trimesh/allocate.h>


namespace vcg {
namespace tri {
namespace io {

template<typename OpenMeshType>
class ImporterDAE
{
public:
	//merge all meshes in the collada's file in the templeted mesh m
	//I assume the mesh 

	enum DAEError 
	{
		E_NOERROR,				// 0
		E_CANTOPEN,				// 1
		E_NOGEOMETRYLIBRARY,     // 2 
		E_NOMESH,      // 3
		E_NOVERTEXPOSITION,            // 4
		E_NO3DVERTEXPOSITION			// 5
	};

	static const char *ErrorMsg(int error)
	{
		static const char * dae_error_msg[] =
		{
			"No errors",
			"Can't open file",
			"File without a geometry library",
			"There isn't mesh in file",
			"The meshes in file haven't the vertex position attribute",
			"The importer assumes that the OpenMeshType uses a 3D point for the vertex position"
		};

		if(error>5 || error<0) return "Unknown error";
		else return dae_error_msg[error];
	};

	static int Open(OpenMeshType& m,const char* filename)
	{
		assert(filename!=0);
		FCDocument* doc = new FCDocument();
		
		FUStatus st = doc->LoadFromFile(FUStringConversion::ToFString(filename));
		if (st.IsFailure()) 
			return E_CANTOPEN;
		FCDGeometryLibrary* geolib = doc->GetGeometryLibrary();
		if (geolib->IsEmpty()) return E_NOGEOMETRYLIBRARY;
		size_t n = geolib->GetEntityCount();
		std::vector<FCDGeometryMesh*> geomsh(n);
		
		//for any mesh in the collada file

		for(unsigned int ii = 0;ii < geomsh.size();++ii)
		{
			if (!geolib->GetEntity(ii)->IsMesh())
			{
				return E_NOMESH;
			}
			else
			{
				geomsh[ii] = geolib->GetEntity(ii)->GetMesh();
				if (geomsh[ii]->GetFaceCount() > 0)
				{
					geomsh[ii]->Triangulate(); 
					//geomsh[ii]->Get
					size_t dim = geomsh[ii]->GetFaceVertexCount() / geomsh[ii]->GetFaceCount();
					assert(dim == 3);
					//MyMesh* msh = new MyMesh();
					//size_t  nattr = geomsh[ii]->GetSourceCount();
					//FCDGeometrySourceList& srclst = geomsh[ii]->GetVertexSources(); 
						
					FCDGeometrySource* src;
					if ((src = geomsh[ii]->GetPositionSource()) != NULL)
					{
						FloatList& flst = src->GetSourceData();
						unsigned int str = src->GetSourceStride();
						assert(flst.size() % str == 0);
						for(unsigned int cont = 0;cont < flst.size();cont += str)
						{
							OpenMeshType::VertexIterator vi=vcg::tri::Allocator<OpenMeshType>::AddVertices(m,1);
							vi->P()= vcg::Point3f(flst[cont],flst[cont + 1],flst[cont + 2]);
							vi->N() = vcg::Point3f(0.0,0.0,0.0);
						}
					}
					else return E_NOVERTEXPOSITION;

					//a single mesh may be composed by a variable numbers of polygons' subsets
					size_t pol = geomsh[ii]->GetPolygonsCount();

					//for any polygons' subset in a single mesh
					for(unsigned int pset = 0; pset < pol;++pset)
					{
						FCDGeometryMesh* tmp = geomsh[ii];
						FCDGeometryPolygonsInput* pos = tmp->GetPolygons(pset)->FindInput(FUDaeGeometryInput::POSITION);
						if ((pos == NULL) || (pos->source->GetSourceStride() != 3)) return E_NO3DVERTEXPOSITION;
						//unsigned int hi = pos->indices[1];
						FCDGeometryPolygonsInput* norm = tmp->GetPolygons(pset)->FindInput(FUDaeGeometryInput::NORMAL);
						//unsigned int li = norm->indices[1];
						FCDGeometryPolygonsInput* text = tmp->GetPolygons(pset)->FindInput(FUDaeGeometryInput::TEXCOORD);
						
						bool isvalidwnorm = (m.HasPerWedgeNormal()) && (norm != NULL) && (norm->source->GetSourceStride() == 3); 
						bool isvalidnorm = (m.HasPerVertexNormal()) && (norm != NULL) && (norm->source->GetSourceStride() == 3);
						bool isvalidtext = (HasPerWedgeTexture(m)) && (text != NULL) && (text->source->GetSourceStride() == 2);
						
						FCDGeometryPolygonsInputList tet; 
						tmp->GetPolygons(pset)->FindInputs(FUDaeGeometryInput::TEXCOORD,tet);

						for(unsigned int ind = 0;ind < pos->indices.size();++ind)
						{
							OpenMeshType::FaceIterator fi=vcg::tri::Allocator<OpenMeshType>::AddFaces(m,1);		
							assert(pos->indices[ind] < m.vert.size());
							fi->V(0) = &m.vert[pos->indices[ind]];
							
							if (isvalidnorm) 
							{
								assert(norm->indices[ind]  * 3  < norm->source->GetSourceData().size());
								fi->V(0)->N() += vcg::Point3f(norm->source->GetSourceData()[norm->indices[ind] * 3],norm->source->GetSourceData()[norm->indices[ind]  * 3 + 1],norm->source->GetSourceData()[norm->indices[ind]  * 3 + 2]).Normalize();
								//++fi->V(0)->incidentfaces;
							}

							if (isvalidtext)
							{
								for(unsigned int hh = 0; hh < tet.size();++hh)
								{
									//NON CAMBIARE!!!!E' L'unico modo in cui restituisce gli indici corretti quando c'e' piu' di un insieme con la stessa semantica!!
									UInt32List* ls = tmp->GetPolygons(pset)->FindIndices(tet[hh]);
									fi->WT(0).t(hh) = vcg::Point2f(tet[hh]->source->GetSourceData()[(*ls)[ind] * 2],tet[hh]->source->GetSourceData()[(*ls)[ind] * 2 + 1]);
								}
							}

							if (isvalidwnorm)
							{
								fi->WN(0) = vcg::Point3f(norm->source->GetSourceData()[norm->indices[ind] * 3],norm->source->GetSourceData()[norm->indices[ind]  * 3 + 1],norm->source->GetSourceData()[norm->indices[ind]  * 3 + 2]).Normalize();
								fi->WN(1) = vcg::Point3f(norm->source->GetSourceData()[norm->indices[ind + 1] * 3],norm->source->GetSourceData()[norm->indices[ind + 1]  * 3 + 1],norm->source->GetSourceData()[norm->indices[ind + 1]  * 3 + 2]).Normalize();
								fi->WN(2) = vcg::Point3f(norm->source->GetSourceData()[norm->indices[ind + 2] * 3],norm->source->GetSourceData()[norm->indices[ind + 2]  * 3 + 1],norm->source->GetSourceData()[norm->indices[ind + 2]  * 3 + 2]).Normalize();
							}

							++ind;
							assert(pos->indices[ind] < m.vert.size());
							fi->V(1) = &m.vert[pos->indices[ind]];
							
							if (isvalidnorm) 
							{	
								assert(norm->indices[ind]  * 3 < norm->source->GetSourceData().size());
								fi->V(1)->N() += vcg::Point3f(norm->source->GetSourceData()[norm->indices[ind] * 3],norm->source->GetSourceData()[norm->indices[ind]  * 3 + 1],norm->source->GetSourceData()[norm->indices[ind]  * 3 + 2]).Normalize();
								//++fi->V(1)->incidentfaces;
							}

							if (isvalidtext)
							{
								for(unsigned int hh = 0; hh < tet.size();++hh)
								{
									UInt32List* ls = tmp->GetPolygons(pset)->FindIndices(tet[hh]);
									fi->WT(1).t(hh) = vcg::Point2f(tet[hh]->source->GetSourceData()[(*ls)[ind] * 2],tet[hh]->source->GetSourceData()[(*ls)[ind] * 2 + 1]);
								}
							}

							++ind;
							assert(pos->indices[ind] < m.vert.size());
							fi->V(2) = &m.vert[pos->indices[ind]];
							
							if (isvalidnorm)
							{
								assert(norm->indices[ind]  * 3 < norm->source->GetSourceData().size());
								fi->V(2)->N() += vcg::Point3f(norm->source->GetSourceData()[norm->indices[ind]  * 3],norm->source->GetSourceData()[norm->indices[ind]  * 3 + 1],norm->source->GetSourceData()[norm->indices[ind]  * 3 + 2]).Normalize();
								//++fi->V(2)->incidentfaces;
							}
							
							if (isvalidtext)
							{
								for(unsigned int hh = 0; hh < tet.size();++hh)
								{
									UInt32List* ls = tmp->GetPolygons(pset)->FindIndices(tet[hh]);
									fi->WT(2).t(hh) = vcg::Point2f(tet[hh]->source->GetSourceData()[(*ls)[ind] * 2],tet[hh]->source->GetSourceData()[(*ls)[ind] * 2 + 1]);
								}
							}

							if (isvalidnorm) fi->N() = ((fi->V(1)->P() - fi->V(0)->P()) ^ (fi->V(2)->P() - fi->V(0)->P())).Normalize();

							/*FCDGeometryPolygonsInput* posa = tmp->GetPolygons(pset)->FindInput(FUDaeGeometryInput::POSITION);
							FloatList& list = posa->source->GetSourceData();
							int dim = list.size();
							list[0] = -100.0;*/
						}
						//vm.push_back(msh);
						
						if (isvalidnorm)
						{
							vcg::tri::UpdateNormals<OpenMeshType>::PerVertexNormalized(m);
							/*for(MyMesh::VertexIterator vit = msh->vert.begin(); vit != msh->vert.end();++vit)
								vit->N() = (vit->N() / vit->incidentfaces).Normalize();*/
						}
					}
				}
			}
		}

		//doc->WriteToFile("PincoPalla.dae");
		delete doc;
		return E_NOERROR;
	}
};

}
}
}

#endif