import2tables.h 6.28 KB
Newer Older
Pierre Kraemer's avatar
Pierre Kraemer committed
1 2 3
/*******************************************************************************
* CGoGN: Combinatorial and Geometric modeling with Generic N-dimensional Maps  *
* version 0.1                                                                  *
4
* Copyright (C) 2009-2012, IGG Team, LSIIT, University of Strasbourg           *
Pierre Kraemer's avatar
Pierre Kraemer committed
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
*                                                                              *
* This library is free software; you can redistribute it and/or modify it      *
* under the terms of the GNU Lesser General Public License as published by the *
* Free Software Foundation; either version 2.1 of the License, or (at your     *
* option) any later version.                                                   *
*                                                                              *
* This library 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 Lesser General Public License  *
* for more details.                                                            *
*                                                                              *
* You should have received a copy of the GNU Lesser General Public License     *
* along with this library; if not, write to the Free Software Foundation,      *
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA.           *
*                                                                              *
20
* Web site: http://cgogn.unistra.fr/                                           *
Pierre Kraemer's avatar
Pierre Kraemer committed
21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36
* Contact information: cgogn@unistra.fr                                        *
*                                                                              *
*******************************************************************************/

#ifndef _IMPORT2TABLE_H
#define _IMPORT2TABLE_H

#include <fstream>
#include <iostream>
#include <vector>
#include <string>

//#include "Topology/generic/attributeHandler.h"
#include "Geometry/vector_gen.h"
#include "Geometry/matrix.h"

37 38

#include "Utils/gzstream.h"
Sylvain Thery's avatar
Sylvain Thery committed
39
#ifdef WITH_ASSIMP
Pierre Kraemer's avatar
Pierre Kraemer committed
40 41
#include "assimp.h"
#include "aiScene.h"
Sylvain Thery's avatar
Sylvain Thery committed
42
#endif
Pierre Kraemer's avatar
Pierre Kraemer committed
43 44 45 46 47 48 49 50 51 52 53 54

namespace CGoGN
{

namespace Algo
{

namespace Import
{

	namespace ImportSurfacique
	{
55
		enum ImportType { UNKNOWNSURFACE, TRIAN, TRIANBGZ, MESHBIN, PLY, /*PLYPTM, PLYSLFgeneric, PLYSLFgenericBin, */OFF, OBJ, VRML, AHEM };
Pierre Kraemer's avatar
Pierre Kraemer committed
56 57 58 59
	}

	namespace ImportVolumique
	{
60
		enum ImportType { UNKNOWNVOLUME , TET, OFF, TS, NODE};
Pierre Kraemer's avatar
Pierre Kraemer committed
61 62 63 64 65 66 67
	}


template <typename PFP>
class MeshTablesSurface
{
protected:
Pierre Kraemer's avatar
Pierre Kraemer committed
68 69
	typename PFP::MAP& m_map;

70
	unsigned int m_nbVertices;
Pierre Kraemer's avatar
Pierre Kraemer committed
71

72
	unsigned int m_nbFaces;
Pierre Kraemer's avatar
Pierre Kraemer committed
73 74

	unsigned int m_lab;
Pierre Kraemer's avatar
Pierre Kraemer committed
75

Pierre Kraemer's avatar
Pierre Kraemer committed
76 77 78 79 80 81 82 83 84 85 86 87
	/**
	* number of edges per face
	*/
	std::vector<short> m_nbEdges;

	/**
	* table of emb indice (for each edge, first vertex)
	*/
	std::vector<unsigned int> m_emb;

	static ImportSurfacique::ImportType getFileType(const std::string& filename);

Sylvain Thery's avatar
Sylvain Thery committed
88
#ifdef WITH_ASSIMP
89
	void extractMeshRec(AttributeContainer& container, AttributeHandler<typename PFP::VEC3, VERTEX>& positions, const struct aiScene* scene, const struct aiNode* nd, struct aiMatrix4x4* trafo);
Sylvain Thery's avatar
Sylvain Thery committed
90
#endif
Pierre Kraemer's avatar
Pierre Kraemer committed
91 92 93 94 95

public:
	typedef typename PFP::VEC3 VEC3 ;
	typedef typename VEC3::DATA_TYPE DATA_TYPE ;

Pierre Kraemer's avatar
Pierre Kraemer committed
96
	inline unsigned getNbFaces() const { return m_nbFaces; }
Pierre Kraemer's avatar
Pierre Kraemer committed
97

Pierre Kraemer's avatar
Pierre Kraemer committed
98
	inline unsigned getNbVertices() const { return m_nbVertices; }
Pierre Kraemer's avatar
Pierre Kraemer committed
99

Pierre Kraemer's avatar
Pierre Kraemer committed
100
	inline short getNbEdgesFace(int i) const  { return m_nbEdges[i]; }
Pierre Kraemer's avatar
Pierre Kraemer committed
101

Pierre Kraemer's avatar
Pierre Kraemer committed
102
	inline unsigned int getEmbIdx(int i) { return  m_emb[i]; }
Pierre Kraemer's avatar
Pierre Kraemer committed
103

Pierre Kraemer's avatar
Pierre Kraemer committed
104
	bool importMesh(const std::string& filename, std::vector<std::string>& attrNames);
Pierre Kraemer's avatar
Pierre Kraemer committed
105

Pierre Kraemer's avatar
Pierre Kraemer committed
106
	bool importTrian(const std::string& filename, std::vector<std::string>& attrNames);
Pierre Kraemer's avatar
Pierre Kraemer committed
107

Pierre Kraemer's avatar
Pierre Kraemer committed
108
	bool importTrianBinGz(const std::string& filename, std::vector<std::string>& attrNames);
Pierre Kraemer's avatar
Pierre Kraemer committed
109

Pierre Kraemer's avatar
Pierre Kraemer committed
110
	bool importOff(const std::string& filename, std::vector<std::string>& attrNames);
Pierre Kraemer's avatar
Pierre Kraemer committed
111

112 113
	bool importMeshBin(const std::string& filename, std::vector<std::string>& attrNames);

Pierre Kraemer's avatar
Pierre Kraemer committed
114
	bool importObj(const std::string& filename, std::vector<std::string>& attrNames);
Pierre Kraemer's avatar
Pierre Kraemer committed
115

Pierre Kraemer's avatar
Pierre Kraemer committed
116
	bool importPly(const std::string& filename, std::vector<std::string>& attrNames);
Pierre Kraemer's avatar
Pierre Kraemer committed
117

118
/*	bool importPlyPTM(const std::string& filename, std::vector<std::string>& attrNames);
119
	bool importPlySLFgeneric(const std::string& filename, std::vector<std::string>& attrNames);
Kenneth Vanhoey's avatar
Kenneth Vanhoey committed
120
	bool importPlySLFgenericBin(const std::string& filename, std::vector<std::string>& attrNames);
121
*/
Sylvain Thery's avatar
Sylvain Thery committed
122
#ifdef WITH_ASSIMP
Pierre Kraemer's avatar
Pierre Kraemer committed
123
	bool importASSIMP(const std::string& filename, std::vector<std::string>& attrNames);
Sylvain Thery's avatar
Sylvain Thery committed
124
#endif	
125
	bool mergeCloseVertices();
Pierre Kraemer's avatar
Pierre Kraemer committed
126

127
	bool importAHEM(const std::string& filename, std::vector<std::string>& attrNames);
Pierre Kraemer's avatar
Pierre Kraemer committed
128 129 130 131 132 133

	/**
	 * @param container container of vertex orbite
	 * @param idPositions id of position attribute in the container
	 * @param idLabels id of label attribute in the container
	 */
Pierre Kraemer's avatar
Pierre Kraemer committed
134 135
	MeshTablesSurface(typename PFP::MAP& map):
		m_map(map)
Pierre Kraemer's avatar
Pierre Kraemer committed
136 137 138 139 140 141 142 143 144
	{
	}
};


template <typename PFP>
class MeshTablesVolume
{
protected:
145 146
	typename PFP::MAP& m_map;

147
	unsigned int m_nbVertices;
Pierre Kraemer's avatar
Pierre Kraemer committed
148

149
	unsigned int m_nbFaces;
Pierre Kraemer's avatar
Pierre Kraemer committed
150

151
	unsigned int m_nbVolumes;
Pierre Kraemer's avatar
Pierre Kraemer committed
152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172

	/**
	* number of edges per face
	*/
	std::vector<short> m_nbEdges;

	/**
	* table of emb ptr (for each face, first vertex)
	*/
	std::vector<unsigned int> m_emb;

	static ImportVolumique::ImportType getFileType(const std::string& filename);

public:
	typedef typename PFP::VEC3 VEC3 ;
	typedef typename VEC3::DATA_TYPE DATA_TYPE ;

	typedef Geom::Vector<6,DATA_TYPE> VEC6;
	typedef Geom::Matrix<3,3,DATA_TYPE> MAT33;
	typedef Geom::Matrix<3,6,DATA_TYPE> MAT36;

173
	inline short getNbEdgesFace(int i) const  { return m_nbEdges[i]; }
Pierre Kraemer's avatar
Pierre Kraemer committed
174 175 176 177 178 179 180 181 182 183 184 185 186

	inline unsigned getNbVolumes() const { return m_nbVolumes; }

	inline unsigned getNbFaces() const { return m_nbFaces; }

	inline unsigned getNbVertices() const { return m_nbVertices; }

	//inline short getNbVerticesPerFace(int i) const  { return m_nbVerticesPerFace[i]; }

	//inline short getNbFacesPerVolume(int i) const { return m_nbFacesPerVolume[i]; }

	inline unsigned int getEmbIdx(int i) { return  m_emb[i]; }

Pierre Kraemer's avatar
Pierre Kraemer committed
187
	bool importMesh(const std::string& filename, std::vector<std::string>& attrNames, float scaleFactor = 1.0f);
Pierre Kraemer's avatar
Pierre Kraemer committed
188

189
	bool importTet(const std::string& filename, std::vector<std::string>& attrNames, float scaleFactor = 1.0f);
Pierre Kraemer's avatar
Pierre Kraemer committed
190

191 192
	MeshTablesVolume(typename PFP::MAP& map):
		m_map(map)
Pierre Kraemer's avatar
Pierre Kraemer committed
193 194 195 196 197 198 199 200 201 202 203 204 205 206
	{
	}
};

} // namespace Import

} // namespace Algo

} // namespace CGoGN

#include "Algo/Import/import2tablesSurface.hpp"
#include "Algo/Import/import2tablesVolume.hpp"

#endif