import2tables.h 6.29 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-2011, 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.u-strasbg.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
	{
Sylvain Thery's avatar
Sylvain Thery committed
55
		enum ImportType { UNKNOWNSURFACE, TRIAN, TRIANBGZ, PLY, PLYPTM, PLYPTMgeneric, OFF, OBJ, CTM, VRML, AHEM };
Pierre Kraemer's avatar
Pierre Kraemer committed
56 57 58 59
	}

	namespace ImportVolumique
	{
Pierre Kraemer's avatar
Pierre Kraemer committed
60
		enum ImportType { UNKNOWNVOLUME ,TET ,TRIANBGZ ,PLY };
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>& 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, ImportSurfacique::ImportType kind);
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

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

Pierre Kraemer's avatar
Pierre Kraemer committed
114
	bool importPly(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 importPlyPTM(const std::string& filename, std::vector<std::string>& attrNames);
117
	bool importPlyPTMgeneric(const std::string& filename, std::vector<std::string>& attrNames);
Sylvain Thery's avatar
Sylvain Thery committed
118
#ifdef WITH_ASSIMP
Pierre Kraemer's avatar
Pierre Kraemer committed
119
	bool importASSIMP(const std::string& filename, std::vector<std::string>& attrNames);
Sylvain Thery's avatar
Sylvain Thery committed
120
#endif	
121
	bool mergeCloseVertices();
Pierre Kraemer's avatar
Pierre Kraemer committed
122

123
	bool importAHEM(const std::string& filename, std::vector<std::string>& attrNames);
Pierre Kraemer's avatar
Pierre Kraemer committed
124 125 126 127 128 129

	/**
	 * @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
130 131
	MeshTablesSurface(typename PFP::MAP& map):
		m_map(map)
Pierre Kraemer's avatar
Pierre Kraemer committed
132 133 134 135 136 137 138 139 140
	{
	}
};


template <typename PFP>
class MeshTablesVolume
{
protected:
141 142
	typename PFP::MAP& m_map;

143
	unsigned int m_nbVertices;
Pierre Kraemer's avatar
Pierre Kraemer committed
144

145
	unsigned int m_nbFaces;
Pierre Kraemer's avatar
Pierre Kraemer committed
146

147
	unsigned int m_nbVolumes;
Pierre Kraemer's avatar
Pierre Kraemer committed
148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168

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

169
	inline short getNbEdgesFace(int i) const  { return m_nbEdges[i]; }
Pierre Kraemer's avatar
Pierre Kraemer committed
170 171 172 173 174 175 176 177 178 179 180 181 182

	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]; }

183
	bool importMesh(const std::string& filename, std::vector<std::string>& attrNames, ImportVolumique::ImportType kind, float scaleFactor);
Pierre Kraemer's avatar
Pierre Kraemer committed
184

185
	bool importTet(const std::string& filename, std::vector<std::string>& attrNames, float scaleFactor);
Pierre Kraemer's avatar
Pierre Kraemer committed
186

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

189
	bool importTrianBinGz(const std::string& filename, std::vector<std::string>& attrNames);
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