transfer of Export_3ds.h, Export_obj.h, Io_3ds_obj_material.h from Meshlab to vcg

This commit is contained in:
Paolo Cignoni 2006-02-16 19:28:36 +00:00
parent 947ecf364e
commit 897721f451
4 changed files with 1140 additions and 1 deletions

View File

@ -25,6 +25,9 @@
History History
$Log: not supported by cvs2svn $ $Log: not supported by cvs2svn $
Revision 1.2 2006/01/13 15:47:42 cignoni
Uniformed return type to the style of Open. Now every export function returns 0 in case of success.
Revision 1.1 2005/11/12 18:34:17 cignoni Revision 1.1 2005/11/12 18:34:17 cignoni
Initial Commit Initial Commit
@ -38,6 +41,7 @@ Initial Commit
#include <wrap/io_trimesh/export_stl.h> #include <wrap/io_trimesh/export_stl.h>
#include <wrap/io_trimesh/export_off.h> #include <wrap/io_trimesh/export_off.h>
#include <wrap/io_trimesh/export_dxf.h> #include <wrap/io_trimesh/export_dxf.h>
#include <wrap/io_trimesh/export_obj.h>
#include <locale> #include <locale>
@ -54,7 +58,7 @@ template <class OpenMeshType>
class Exporter class Exporter
{ {
private: private:
enum KnownTypes { KT_UNKNOWN, KT_PLY, KT_STL, KT_DXF, KT_OFF }; enum KnownTypes { KT_UNKNOWN, KT_PLY, KT_STL, KT_DXF, KT_OFF, KT_OBJ};
static int &LastType() static int &LastType()
{ {
static int lastType= KT_UNKNOWN; static int lastType= KT_UNKNOWN;
@ -101,6 +105,11 @@ static int Save(OpenMeshType &m, const char *filename, const int mask, CallBackP
err = ExporterDXF<OpenMeshType>::Save(m,filename); err = ExporterDXF<OpenMeshType>::Save(m,filename);
LastType()=KT_DXF; LastType()=KT_DXF;
} }
else if(FileExtension(filename,"obj"))
{
err = ExporterOBJ<OpenMeshType>::Save(m,filename,mask,cb);
LastType()=KT_OBJ;
}
else { else {
err=1; err=1;
LastType()=KT_UNKNOWN; LastType()=KT_UNKNOWN;
@ -117,6 +126,7 @@ static const char *ErrorMsg(int error)
case KT_STL : return ExporterSTL<OpenMeshType>::ErrorMsg(error); break; case KT_STL : return ExporterSTL<OpenMeshType>::ErrorMsg(error); break;
case KT_OFF : return ExporterOFF<OpenMeshType>::ErrorMsg(error); break; case KT_OFF : return ExporterOFF<OpenMeshType>::ErrorMsg(error); break;
case KT_DXF : return ExporterDXF<OpenMeshType>::ErrorMsg(error); break; case KT_DXF : return ExporterDXF<OpenMeshType>::ErrorMsg(error); break;
case KT_OBJ : return ExporterOBJ<OpenMeshType>::ErrorMsg(error); break;
} }
return "Unknown type"; return "Unknown type";
} }

View File

@ -0,0 +1,572 @@
/****************************************************************************
* VCGLib o o *
* Visual and Computer Graphics Library o o *
* _ O _ *
* Copyright(C) 2004 \/)\/ *
* Visual Computing Lab /\/| *
* ISTI - Italian National Research Council | *
* \ *
* All rights reserved. *
* *
* This program is free software; you can redistribute it and/or modify *
* it under the terms of the GNU General Public License as published by *
* the Free Software Foundation; either version 2 of the License, or *
* (at your option) any later version. *
* *
* This program 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 General Public License (http://www.gnu.org/licenses/gpl.txt) *
* for more details. *
* *
****************************************************************************/
/****************************************************************************
History
$Log: not supported by cvs2svn $
Revision 1.14 2006/02/06 11:26:26 fmazzant
bug-fix comment
Revision 1.13 2006/02/06 11:04:39 fmazzant
added file material.h. it include struct Material, CreateNewMaterial(...) and MaterialsCompare(...)
Revision 1.12 2006/02/04 10:18:46 fmazzant
clean code
Revision 1.11 2006/02/03 12:36:53 fmazzant
not significant updating
Revision 1.10 2006/02/03 10:04:41 fmazzant
no significant updated
Revision 1.9 2006/02/02 16:08:27 fmazzant
updated comment code
Revision 1.8 2006/02/02 16:05:18 fmazzant
deleted bug when saving mesh without textures.
Revision 1.7 2006/02/02 15:35:15 fmazzant
updated comment code
Revision 1.6 2006/02/02 13:08:10 fmazzant
cleaned & commented[italian] code
Revision 1.5 2006/02/02 10:50:46 fmazzant
deleted a big bug of exporter 3ds
Revision 1.4 2006/01/31 09:34:30 fmazzant
bug-fix on savemaskexporter, when press cancel returns -1.
Revision 1.3 2006/01/29 23:52:43 fmazzant
correct a small bug
Revision 1.2 2006/01/29 18:33:42 fmazzant
added some comment to the code
Revision 1.1 2006/01/29 16:33:03 fmazzant
moved export_obj and export_3ds from test/io into meshio/
Revision 1.30 2006/01/26 15:57:20 fmazzant
deleted a small bug
Revision 1.29 2006/01/26 15:48:49 fmazzant
added control on the maximum number of vertices allowed by the 3DS file format
Revision 1.28 2006/01/23 14:07:39 fmazzant
deleted bug when saving a face color.
Revision 1.27 2006/01/22 23:59:01 fmazzant
changed default value of diffuse. 1.0 -> 0.8
Revision 1.26 2006/01/22 10:42:18 fmazzant
cleaned code & optimized code of material's name 3ds
Revision 1.25 2006/01/22 01:26:52 fmazzant
deleted bug on saving name material 3ds
Revision 1.24 2006/01/21 15:19:51 fmazzant
changed:
inserting coord texture in to map from key = index, value=coord
to key=coord, value=index_of_vertix.
Revision 1.23 2006/01/20 14:15:52 fmazzant
added texture filename on material 3ds and coordtexture on face
Revision 1.22 2006/01/19 09:36:29 fmazzant
cleaned up history log
Revision 1.21 2006/01/18 16:14:52 fmazzant
update small error
Revision 1.20 2006/01/18 16:12:58 fmazzant
updated material
Revision 1.19 2006/01/18 15:09:45 fmazzant
added texture base
Revision 1.18 2006/01/18 14:57:26 fmazzant
added Lib3dsNode in export_3ds
****************************************************************************/
#ifndef __VCGLIB_EXPORT_3DS
#define __VCGLIB_EXPORT_3DS
#include <wrap/callback.h>
#include <vcg/complex/trimesh/allocate.h>
#include <wrap/io_trimesh/io_mask.h>
#include "io_3ds_obj_material.h"
/*
3DS export needs the Lib3ds library.
lib3ds is a free ANSI-C library for working with the popular "3ds" 3D model format.
Supported platforms include GNU (autoconf, automake, libtool, make, GCC) on Unix and
Cygwin, and MS Visual C++ 6.0. lib3ds loads and saves Atmosphere settings, Background
settings, Shadow map settings, Viewport setting, Materials, Cameras, Lights, Meshes,
Hierarchy, Animation keyframes. It also contains useful matrix, vector and quaternion
mathematics tools. lib3ds usually integrates well with OpenGL. In addition, some
diagnostic and conversion tools are included.
lib3ds is distributed under the terms of the GNU Lesser General Public License.
this information has been taken by the official site.
runable from http://lib3ds.sourceforge.net/
*/
#include <lib3ds/file.h>
#include <lib3ds/io.h>
#include <lib3ds/mesh.h>
#include <lib3ds/types.h>
#include <lib3ds/material.h>
#include <vector>
#include <iostream>
#include <fstream>
#include <QString>
#include <QMessageBox>
#define MAX_POLYGONS 65535
namespace vcg {
namespace tri {
namespace io {
template <class SaveMeshType>
class Exporter3DS
{
public:
typedef typename SaveMeshType::FaceIterator FaceIterator;
typedef typename SaveMeshType::VertexIterator VertexIterator;
typedef typename SaveMeshType::VertexType VertexType;
//int: old index vertex
//TCoord2: textcoord with vertex's index i
typedef std::pair<int,vcg::TCoord2<float> > Key;
/*
enum of all the types of error
*/
enum SaveError
{
E_NOERROR, // 0
E_CANTOPENFILE, // 1
E_CANTCLOSEFILE, // 2
E_UNESPECTEDEOF, // 3
E_ABORTED, // 4
E_NOTDEFINITION, // 5
E_NOTVEXTEXVALID, // 6
E_NOTFACESVALID, // 7
E_NOTEXCOORDVALID, // 8
E_NOTNUMBERVERTVALID // 9
};
/*
this function takes an index and the relative error message gets back
*/
static const char* ErrorMsg(int error)
{
static const char* obj_error_msg[] =
{
"No errors", // 0
"Can't open file", // 1
"can't close file", // 2
"Premature End of file", // 3
"File saving aborted", // 4
"Function not defined", // 5
"Vertices not valid", // 6
"Faces not valid", // 7
"Texture Coord not valid", // 8
"You cannot save more than 65535 vertices for the 3DS format" // 9
};
if(error>9 || error<0) return "Unknown error";
else return obj_error_msg[error];
};
/*
returns mask of capability one define with what are the saveable information of the format.
*/
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;
}
/*
function which saves in 3DS file format
*/
static int SaveASCII(SaveMeshType &m, const char * filename)
{
return E_NOTDEFINITION;
}
/*
function which saves in 3DS file format
*/
static int SaveBinary(SaveMeshType &m, const char * filename, const int &mask, CallBackPos *cb=0)
{
if(m.vert.size() > MAX_POLYGONS)//check max polygons
return E_NOTNUMBERVERTVALID;
if(m.vert.size() == 0)
return E_NOTVEXTEXVALID;
if(m.face.size() == 0)
return E_NOTFACESVALID;
/*
<<concetto:>>
si tiene in considerazione una mappa ListOfDuplexVert<Key,int>, alla quale gli viene associato il seguente significato:
Key:è una coppia (int,TexCoord) formata da un int che rappresenta l'indice del vettore nella mesh originale e la sua
coordinata di texture. tale coppia rappresenta una chiave, essendo univoca in tutta la mesh. Non è possibile che
si incontrino due vertici che hanno solito indice di vertice e solite coordinate di texture, se un vertice di
questo tipo esistesse allora i due vertici rappresenterebbero lo stesso vertice.
int:è l'indice del vertice inserito all'interno del vettore VectorOfVertexType<VertexType>
Nel vertice VectorOfVertexType vengono inseriti tutti i vertici appartenenti alla mesh + i k vertici dublicati. la scelta
di tali vertici va in base alla seguente regola:
se un vertice con indice x(originale) ha più di una coordinata di texture allora tale vertice viene duplicato e
inserito in ListOfDuplexVert e in VectorOfVertexType(in VectorOfVertexType, l'inserimento del doppio vertice non
sarebbe necessario, però viene fatto per comodità, in caso contrario dovremmo cercare il vertice dentro il vettore).
rappresentazione grafica:
ListOfDuplexVert VectorOfVertexType
------ ---------
|key1| -> index1 --------- |vertex1|
------ | ---------
|key2| -> index2 ---- -------------> |vertex2|
------ | ---------
|key3| | |vertex3|
------ | ---------
|key4| ------------------> |vertex4|
------ ---------
|key5| --------------> |vertex5|
------ | ---------
. | .
. | .
. | .
------ | ---------
|keyn| -> indexn -------- |vertexn|
------ ---------
questo tipo di struttura permette di selezionare l'indice del vertice in VectorOfVertexType con costo O(1).
<<code:>>
questo pezzo di codice itera su tutte le facce della mesh per riempire la mappa e il vettore.
per ogni faccia e per ogni vertice di faccia costruisce la coppia (indice,texture), controlla se
all'interno di ListOfDuplexVert esiste già in tal caso non fa niente, in caso contrario aggiunte la
coppia in ListOfDuplexVert e l'oggetto VertexType in VectorOfVertexType associando al valore della
chiave (indice,texture) l'indice del vertice a cui punta.
alla fine vengono duplicati solamente quei vertici che hanno più coordinate di texture.
c'e' da tenere presente che il codice appena descritto viene eseguito SOLAMENTE se la mesh contiene texture e
se dalla dialog di exporter viene spuntato il salvataggio delle texture. In caso contrario non esegue niente e tratta
solamente i vertici che sono presenti nella mesh senza creare duplicati. Le informazioni presenti in assenza di
texture sono piu' che sufficienti.
Federico Mazzanti
*/
std::map<Key,int> ListOfDuplexVert;
std::vector<VertexType> VectorOfVertexType;
int count = 1;
int nface = 0;
if(m.HasPerWedgeTexture() && mask & vcg::tri::io::Mask::IOM_WEDGTEXCOORD )
{
FaceIterator fi;
for(fi=m.face.begin(); fi!=m.face.end(); ++fi) if( !(*fi).IsD() )
{
for(unsigned int k=0;k<3;k++)
{
int i = GetIndexVertex(m, (*fi).V(k));
vcg::TCoord2<float> t = (*fi).WT(k);
if(!m.vert[i].IsD())
{
if(AddDuplexVertexCoord(ListOfDuplexVert,Key(i,t)))
{
VectorOfVertexType.push_back((*(*fi).V(k)));
ListOfDuplexVert[Key(i,t)] = VectorOfVertexType.size()-1;
count++;
}
}
}
if (cb !=NULL)
(*cb)(100.0 * (float)++nface/(float)m.face.size(), "calc duplex vertex ...");
else
return E_ABORTED;
}
}
int number_vertex_to_duplicate = 0;
if(m.HasPerWedgeTexture() && mask & MeshModel::IOM_WEDGTEXCOORD )
number_vertex_to_duplicate = (count-1) - m.vert.size();
Lib3dsFile *file = lib3ds_file_new();//creates new file
Lib3dsMesh *mesh = lib3ds_mesh_new("mesh");//creates a new mesh with mesh's name "mesh"
QString qnamematerial = "Material - %1";
std::vector<Material> materials;
int current = 0;
int max = m.vert.size()+m.face.size()+ number_vertex_to_duplicate;
lib3ds_mesh_new_point_list(mesh, m.vert.size() + number_vertex_to_duplicate);// set number of vertexs
if(m.HasPerWedgeTexture() && mask & vcg::tri::io::Mask::IOM_WEDGTEXCOORD )
lib3ds_mesh_new_texel_list(mesh,m.vert.size() + number_vertex_to_duplicate); //set number of textures
int v_index = 0;
VertexIterator vi;
//saves vert
if(m.HasPerWedgeTexture() && mask & MeshModel::IOM_WEDGTEXCOORD )
{
for(unsigned int i=0; i< VectorOfVertexType.size();i++)
{
Lib3dsPoint point;
point.pos[0] = VectorOfVertexType[i].P()[0];
point.pos[1] = VectorOfVertexType[i].P()[1];
point.pos[2] = VectorOfVertexType[i].P()[2];
mesh->pointL[i] = point;
if (cb !=NULL)
(*cb)(100.0 * (float)++current/(float)max, "writing vertices ");
else
return E_ABORTED;
}
}
else
{
for(vi=m.vert.begin(); vi!=m.vert.end(); ++vi) if( !(*vi).IsD() )
{
Lib3dsPoint point;
point.pos[0] = (*vi).P()[0];
point.pos[1] = (*vi).P()[1];
point.pos[2] = (*vi).P()[2];
mesh->pointL[v_index] = point;
if (cb !=NULL)
(*cb)(100.0 * (float)++current/(float)max, "writing vertices ");
else
return E_ABORTED;
v_index++;
}
}
lib3ds_mesh_new_face_list (mesh, m.face.size());//set number of faces
int f_index = 0;//face index
int t_index = 0;//texture index
FaceIterator fi;
for(fi=m.face.begin(); fi!=m.face.end(); ++fi) if( !(*fi).IsD() )
{
int i0 = GetIndexVertex(m, (*fi).V(0));
vcg::TCoord2<float> t0;
int i1 = GetIndexVertex(m, (*fi).V(1));
vcg::TCoord2<float> t1;
int i2 = GetIndexVertex(m, (*fi).V(2));
vcg::TCoord2<float> t2;
if(m.HasPerWedgeTexture() && mask & MeshModel::IOM_WEDGTEXCOORD )
{
t0 = (*fi).WT(0);
t1 = (*fi).WT(1);
t2 = (*fi).WT(2);
}
Lib3dsFace face;
if(m.HasPerWedgeTexture() && mask & MeshModel::IOM_WEDGTEXCOORD )
{
face.points[0] = GetIndexDuplexVertex(ListOfDuplexVert,Key(i0,t0));
face.points[1] = GetIndexDuplexVertex(ListOfDuplexVert,Key(i1,t1));
face.points[2] = GetIndexDuplexVertex(ListOfDuplexVert,Key(i2,t2));
}
else
{
face.points[0] = i0;
face.points[1] = i1;
face.points[2] = i2;
}
//saves coord textures
if(m.HasPerWedgeTexture() && mask & MeshModel::IOM_WEDGTEXCOORD )
{
mesh->texelL[face.points[0]][0] = t0.u();
mesh->texelL[face.points[0]][1] = t0.v();
mesh->texelL[face.points[1]][0] = t1.u();
mesh->texelL[face.points[1]][1] = t1.v();
mesh->texelL[face.points[2]][0] = t2.u();
mesh->texelL[face.points[2]][1] = t2.v();
}
if(mask & MeshModel::IOM_FACEFLAGS)
face.flags = 0;
face.smoothing = 10;
if(mask & MeshModel::IOM_FACENORMAL | mask & MeshModel::IOM_WEDGNORMAL)
{
face.normal[0] = (*fi).N()[0];
face.normal[1] = (*fi).N()[1];
face.normal[2] = (*fi).N()[2];
}
if(mask & MeshModel::IOM_FACECOLOR | mask & MeshModel::IOM_WEDGTEXCOORD)
{
int material_index = vcg::tri::io::Materials<SaveMeshType>::CreateNewMaterial(m, materials, 0, fi);
if(material_index == materials.size())
{
Lib3dsMaterial *material = lib3ds_material_new();//creates a new material
std::string name = qnamematerial.arg(material_index-1).toStdString();
strcpy(material->name,name.c_str());//copy new name of material
if(mask & MeshModel::IOM_FACECOLOR)
{
//ambient
material->ambient[0] = materials[materials.size()-1].Ka[0];
material->ambient[1] = materials[materials.size()-1].Ka[1];
material->ambient[2] = materials[materials.size()-1].Ka[2];
material->ambient[3] = materials[materials.size()-1].Tr;
//diffuse
material->diffuse[0] = materials[materials.size()-1].Kd[0];
material->diffuse[1] = materials[materials.size()-1].Kd[1];
material->diffuse[2] = materials[materials.size()-1].Kd[2];
material->diffuse[3] = materials[materials.size()-1].Tr;
//specular
material->specular[0] = materials[materials.size()-1].Ks[0];
material->specular[1] = materials[materials.size()-1].Ks[1];
material->specular[2] = materials[materials.size()-1].Ks[2];
material->specular[3] = materials[materials.size()-1].Tr;
//shininess
material->shininess = materials[materials.size()-1].Ns;
}
//texture
if(mask & MeshModel::IOM_WEDGTEXCOORD)
strcpy(material->texture1_map.name,materials[materials.size()-1].map_Kd.c_str());
lib3ds_file_insert_material(file,material);//inserts the material inside the file
strcpy(face.material,name.c_str());
}
else
{
std::string name = qnamematerial.arg(material_index).toStdString();
strcpy(face.material,name.c_str());//set name of material
}
}
mesh->faceL[f_index]=face;
if (cb !=NULL)
(*cb)(100.0 * (float)++current/(float)max, "writing faces ");
else
return E_ABORTED;
f_index++;
}
lib3ds_file_insert_mesh(file, mesh);//inserts the Mesh into file
Lib3dsNode *node = lib3ds_node_new_object();//creates a new node
strcpy(node->name,mesh->name);
node->parent_id = LIB3DS_NO_PARENT;
lib3ds_file_insert_node(file,node);//inserts the node into file
bool result = lib3ds_file_save(file, filename); //saves the file
if(result)
return E_NOERROR;
else
return E_ABORTED;
}
/*
function which saves in 3DS format
*/
static int Save(SaveMeshType &m, const char * filename, const int &mask, CallBackPos *cb=0)
{
return SaveBinary(m,filename,mask,cb);
}
/*
returns index of the vertex
*/
inline static int GetIndexVertex(SaveMeshType &m, VertexType *p)
{
return p-&*(m.vert.begin());
}
/*
added pair Key,int into map
*/
inline static bool AddDuplexVertexCoord(std::map<Key,int> &m,Key key)
{
int index = m[key];
if(index==0)
return true;
return false;
}
/*
returns value of key key into map. this value is vertex's index into list all duplicate vertex
*/
inline static int GetIndexDuplexVertex(std::map<Key,int> &m,Key key)
{
return m[key];
}
}; // end class
} // end Namespace tri
} // end Namespace io
} // end Namespace vcg
#endif

View File

@ -0,0 +1,424 @@
/****************************************************************************
* VCGLib o o *
* Visual and Computer Graphics Library o o *
* _ O _ *
* Copyright(C) 2004 \/)\/ *
* Visual Computing Lab /\/| *
* ISTI - Italian National Research Council | *
* \ *
* All rights reserved. *
* *
* This program is free software; you can redistribute it and/or modify *
* it under the terms of the GNU General Public License as published by *
* the Free Software Foundation; either version 2 of the License, or *
* (at your option) any later version. *
* *
* This program 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 General Public License (http://www.gnu.org/licenses/gpl.txt) *
* for more details. *
* *
****************************************************************************/
/****************************************************************************
History
$Log: not supported by cvs2svn $
Revision 1.7 2006/02/06 11:04:40 fmazzant
added file material.h. it include struct Material, CreateNewMaterial(...) and MaterialsCompare(...)
Revision 1.6 2006/02/04 10:18:46 fmazzant
clean code
Revision 1.5 2006/02/03 10:04:41 fmazzant
no significant updated
Revision 1.4 2006/01/30 14:02:05 fmazzant
bug-fix
Revision 1.3 2006/01/29 23:52:43 fmazzant
correct a small bug
Revision 1.2 2006/01/29 18:33:42 fmazzant
added some comment to the code
Revision 1.1 2006/01/29 16:33:03 fmazzant
moved export_obj and export_3ds from test/io into meshio/
Revision 1.34 2006/01/22 23:59:01 fmazzant
changed default value of diffuse. 1.0 -> 0.8
Revision 1.33 2006/01/19 09:36:29 fmazzant
cleaned up history log
Revision 1.32 2006/01/18 00:45:56 fmazzant
added control on face's diffuse
Revision 1.31 2006/01/17 13:48:54 fmazzant
added capability mask on export file format
Revision 1.30 2006/01/15 00:45:40 fmazzant
extend mask exporter for all type file format +
Revision 1.29 2006/01/14 00:03:26 fmazzant
added more controls
****************************************************************************/
#ifndef __VCGLIB_EXPORT_OBJ
#define __VCGLIB_EXPORT_OBJ
#include <wrap/callback.h>
#include <vcg/complex/trimesh/allocate.h>
#include <wrap/io_trimesh/io_mask.h>
#include "io_3ds_obj_material.h"
#include <iostream>
#include <fstream>
#include <map>
#include <QMessageBox>
namespace vcg {
namespace tri {
namespace io {
template <class SaveMeshType>
class ExporterOBJ
{
public:
typedef typename SaveMeshType::FaceIterator FaceIterator;
typedef typename SaveMeshType::VertexIterator VertexIterator;
typedef typename SaveMeshType::VertexType VertexType;
/*
enum of all the types of error
*/
enum SaveError
{
E_NOERROR, // 0
E_CANTOPENFILE, // 1
E_CANTCLOSEFILE, // 2
E_UNESPECTEDEOF, // 3
E_ABORTED, // 4
E_NOTDEFINITION, // 5
E_NOTVEXTEXVALID, // 6
E_NOTFACESVALID // 7
};
/*
this function takes an index and the relative error message gets back
*/
static const char* ErrorMsg(int error)
{
static const char* obj_error_msg[] =
{
"No errors", // 0
"Can't open file", // 1
"can't close file", // 2
"Premature End of file", // 3
"File saving aborted", // 4
"Function not defined", // 5
"Vertices not valid", // 6
"Faces not valid" // 7
};
if(error>7 || error<0) return "Unknown error";
else return obj_error_msg[error];
};
/*
returns mask of capability one define with what are the saveable information of the format.
*/
static int GetExportMaskCapability()
{
int capability = 0;
//vert
capability |= vcg::tri::io::Mask::IOM_VERTNORMAL;
//face
capability |= vcg::tri::io::Mask::IOM_FACECOLOR;
//wedg
capability |= vcg::tri::io::Mask::IOM_WEDGTEXCOORD;
capability |= vcg::tri::io::Mask::IOM_WEDGNORMAL;
return capability;
}
/*
function which saves in OBJ file format
*/
static int SaveASCII(SaveMeshType &m, const char * filename, ObjInfo &oi)
{
CallBackPos *cb = oi.cb;
if(m.vert.size() == 0)
return E_NOTVEXTEXVALID;
if(m.face.size() == 0)
return E_NOTFACESVALID;
int current = 0;
int max = m.vert.size()+ m.face.size();
std::vector<Material> materials;
std::string fn(filename);
int i=fn.size()-1;
while(fn[--i]!='/');
FILE *fp;
fp = fopen(filename,"w");
if(fp == NULL)return E_CANTOPENFILE;
fprintf(fp,"####\n#\n# OBJ File Generated by Meshlab\n#\n####\n");
fprintf(fp,"# Object %s\n#\n# Vertices: %d\n# Faces: %d\n#\n####\n",fn.substr(i+1).c_str(),m.vert.size(),m.face.size());
//library materials
if(oi.mask & vcg::tri::io::Mask::IOM_FACECOLOR)
fprintf(fp,"mtllib ./%s.mtl\n\n",fn.substr(i+1).c_str());
//vertexs + normal
VertexIterator vi;
std::map<Point3f,int> NormalVertex;
int numvert = 0;
int value = 1;
for(vi=m.vert.begin(); vi!=m.vert.end(); ++vi) if( !(*vi).IsD() )
{
//saves normal per vertex
if(oi.mask & vcg::tri::io::Mask::IOM_VERTNORMAL | oi.mask & vcg::tri::io::Mask::IOM_WEDGNORMAL)
{
if(AddNewNormalVertex(NormalVertex,(*vi).N(),value))
{
fprintf(fp,"vn %f %f %f\n",(*vi).N()[0],(*vi).N()[1],(*vi).N()[2]);
value++;
}
}
//saves vertex
fprintf(fp,"v %f %f %f\n",(*vi).P()[0],(*vi).P()[1],(*vi).P()[2]);
if (cb !=NULL)
(*cb)(100.0 * (float)++current/(float)max, "writing vertices ");
else
{ fclose(fp); return E_ABORTED;}
}
fprintf(fp,"# %d vertices, %d vertices normals\n\n",m.vert.size(),NormalVertex.size());
//faces + texture coords
FaceIterator fi;
std::map<vcg::TCoord2<float>,int> CoordIndexTexture;
unsigned int material_num = 0;
int mem_index = 0; //var temporany
/*int*/ value = 1;//tmp
for(fi=m.face.begin(); fi!=m.face.end(); ++fi) if( !(*fi).IsD() )
{
if(oi.mask & vcg::tri::io::Mask::IOM_FACECOLOR)
{
int index = vcg::tri::io::Materials<SaveMeshType>::CreateNewMaterial(m,materials,material_num,fi);
if(index == materials.size())//inserts a new element material
{
material_num++;
fprintf(fp,"\nusemtl material_%d\n",materials[index-1].index);
mem_index = index-1;
}
else
{
if(index != mem_index)//inserts old name elemente material
{
fprintf(fp,"\nusemtl material_%d\n",materials[index].index);
mem_index=index;
}
}
}
//saves texture coord
unsigned int MAX = 3;
for(unsigned int k=0;k<MAX;k++)
{
if(m.HasPerWedgeTexture() && oi.mask & vcg::tri::io::Mask::IOM_WEDGTEXCOORD)
{
if(AddNewTextureCoord(CoordIndexTexture,(*fi).WT(k),value))
{
fprintf(fp,"vt %f %f\n",(*fi).WT(k).u(),(*fi).WT(k).v());
value++;//ncreases the value number to be associated to the Texture
}
}
}
fprintf(fp,"f ");
for(unsigned int k=0;k<MAX;k++)
{
int v = -1;
// +1 because Obj file format begins from index = 1 but not from index = 0.
v = GetIndexVertex(m, (*fi).V(k)) + 1;//index of vertex per face
int vt = -1;
if(oi.mask & vcg::tri::io::Mask::IOM_WEDGTEXCOORD)
vt = GetIndexVertexTexture(CoordIndexTexture,(*fi).WT(k));//index of vertex texture per face
int vn = -1;
if(oi.mask & vcg::tri::io::Mask::IOM_VERTNORMAL | oi.mask & vcg::tri::io::Mask::IOM_WEDGNORMAL)
vn = GetIndexVertexNormal(m, NormalVertex, v);//index of vertex normal per face.
//writes elements on file obj
WriteFacesElement(fp,v,vt,vn);
if(k!=MAX-1)
fprintf(fp," ");
else
fprintf(fp,"\n");
}
if (cb !=NULL)
(*cb)(100.0 * (float)++current/(float)max, "writing faces ");
else
{ fclose(fp); return E_ABORTED;}
}//for
fprintf(fp,"# %d faces, %d coords texture\n\n",m.face.size(),CoordIndexTexture.size());
fprintf(fp,"# End of File");
fclose(fp);
int r = 0;
if(oi.mask & vcg::tri::io::Mask::IOM_WEDGTEXCOORD | oi.mask & vcg::tri::io::Mask::IOM_FACECOLOR)
r = WriteMaterials(materials, filename,cb);//write material
if(r!= E_NOERROR)
return r;
return E_NOERROR;
}
/*
function which saves in OBJ file format
*/
static int SaveBinary(SaveMeshType &m, const char * filename, ObjInfo &oi)
{
return E_NOTDEFINITION;
}
/*
function which saves in OBJ file format
*/
static int Save(SaveMeshType &m, const char * filename, const int &mask, CallBackPos *cb=0)
{
ObjInfo oi;
oi.cb=cb;
oi.mask=mask;
return SaveASCII(m,filename,oi);
}
/*
returns index of the vertex
*/
inline static int GetIndexVertex(SaveMeshType &m, VertexType *p)
{
return p-&*(m.vert.begin());
}
/*
returns index of the texture coord
*/
inline static int GetIndexVertexTexture(std::map<vcg::TCoord2<float>,int> &m, const vcg::TCoord2<float> &wt)
{
int index = m[wt];
if(index!=0){return index;}
return -1;
}
/*
returns index of the vertex normal
*/
inline static int GetIndexVertexNormal(SaveMeshType &m, std::map<Point3f,int> &ma, unsigned int iv )
{
int index = ma[m.vert[iv].N()];
if(index!=0){return index;}
return -1;
}
/*
write elements on file
*/
inline static void WriteFacesElement(FILE *fp,int v,int vt, int vn)
{
fprintf(fp,"%d",v);
if(vt!=-1)
{
fprintf(fp,"/%d",vt);
if(vn!=-1)
fprintf(fp,"/%d",vn);
}
else if(vn!=-1)
fprintf(fp,"//%d",vn);
}
/*
adds a new index to the coordinate of Texture if it is the first time
which is otherwise met not execute anything
*/
inline static bool AddNewTextureCoord(std::map<vcg::TCoord2<float>,int> &m, const vcg::TCoord2<float> &wt,int value)
{
int index = m[wt];
if(index==0){m[wt]=value;return true;}
return false;
}
/*
adds a new index to the normal per vertex if it is the first time
which is otherwise met does not execute anything
*/
inline static bool AddNewNormalVertex(std::map<Point3f,int> &m, Point3f &n ,int value)
{
int index = m[n];
if(index==0){m[n]=value;return true;}
return false;
}
/*
writes material into file
*/
inline static int WriteMaterials(std::vector<Material> &materials, const char * filename, CallBackPos *cb=0)
{
std::string fileName = std::string(filename);
fileName+=".mtl";
if(materials.size() > 0)
{
FILE *fp;
fp = fopen(fileName.c_str(),"w");
if(fp==NULL)return E_ABORTED;
fprintf(fp,"#\n# Wavefront material file\n# Converted by Meshlab Group\n#\n\n");
int current = 0;
for(unsigned int i=0;i<materials.size();i++)
{
if (cb !=NULL)
(*cb)(100.0 * (float)++current/(float)materials.size(), "saving material file ");
else
{ fclose(fp); return E_ABORTED;}
fprintf(fp,"newmtl material_%d\n",materials[i].index);
fprintf(fp,"Ka %f %f %f\n",materials[i].Ka[0],materials[i].Ka[1],materials[i].Ka[2]);
fprintf(fp,"Kd %f %f %f\n",materials[i].Kd[0],materials[i].Kd[1],materials[i].Kd[2]);
fprintf(fp,"Ks %f %f %f\n",materials[i].Ks[0],materials[i].Ks[1],materials[i].Ks[2]);
fprintf(fp,"Tr %f\n",materials[i].Tr);
fprintf(fp,"illum %d\n",materials[i].illum);
fprintf(fp,"Ns %f\n",materials[i].Ns);
if(materials[i].map_Kd.size()>0)
fprintf(fp,"map_Kd %s\n",materials[i].map_Kd.c_str());
fprintf(fp,"\n");
}
fclose(fp);
}
return E_NOERROR;
}
}; // end class
} // end Namespace tri
} // end Namespace io
} // end Namespace vcg
#endif

View File

@ -0,0 +1,133 @@
/****************************************************************************
* VCGLib o o *
* Visual and Computer Graphics Library o o *
* _ O _ *
* Copyright(C) 2004 \/)\/ *
* Visual Computing Lab /\/| *
* ISTI - Italian National Research Council | *
* \ *
* All rights reserved. *
* *
* This program is free software; you can redistribute it and/or modify *
* it under the terms of the GNU General Public License as published by *
* the Free Software Foundation; either version 2 of the License, or *
* (at your option) any later version. *
* *
* This program 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 General Public License (http://www.gnu.org/licenses/gpl.txt) *
* for more details. *
* *
****************************************************************************/
/****************************************************************************
History
$Log: not supported by cvs2svn $
Revision 1.1 2006/02/06 11:04:40 fmazzant
added file material.h. it include struct Material, CreateNewMaterial(...) and MaterialsCompare(...)
****************************************************************************/
#ifndef __VCGLIB_MATERIAL
#define __VCGLIB_MATERIAL
namespace vcg {
namespace tri {
namespace io {
/*
structures material
*/
struct Material
{
unsigned int index;//index of material
Point3f Ka;//ambient
Point3f Kd;//diffuse
Point3f Ks;//specular
float d;//alpha
float Tr;//alpha
int illum;//specular illumination
float Ns;
std::string map_Kd; //filename texture
};
template <class SaveMeshType>
class Materials
{
public:
typedef typename SaveMeshType::FaceIterator FaceIterator;
typedef typename SaveMeshType::VertexIterator VertexIterator;
typedef typename SaveMeshType::VertexType VertexType;
/*
creates a new meterial
*/
inline static int CreateNewMaterial(SaveMeshType &m, std::vector<Material> &materials, unsigned int index, FaceIterator &fi)
{
unsigned char r = (*fi).C()[0];
unsigned char g = (*fi).C()[1];
unsigned char b = (*fi).C()[2];
unsigned char alpha = (*fi).C()[3];
Point3f diffuse = Point3f((float)r/255.0,(float)g/255.0,(float)b/255.0);//diffuse
float Tr = (float)alpha/255.0;//alpha
int illum = 2; //default not use Ks!
float ns = 0.0; //default
Material mtl;
mtl.index = index;//index of materials
mtl.Ka = Point3f(0.2,0.2,0.2);//ambient
mtl.Kd = diffuse;//diffuse
mtl.Ks = Point3f(1.0,1.0,1.0);//specular
mtl.Tr = Tr;//alpha
mtl.Ns = ns;
mtl.illum = illum;//illumination
if(m.textures.size() && (*fi).WT(0).n() >=0 )
mtl.map_Kd = m.textures[(*fi).WT(0).n()];
else
mtl.map_Kd = "";
int i = -1;
if((i = MaterialsCompare(materials,mtl)) == -1)
{
materials.push_back(mtl);
return materials.size();
}
return i;
}
/*
returns the index of the material if it exists inside the list of the materials,
otherwise it returns -1.
*/
inline static int MaterialsCompare(std::vector<Material> &materials, Material mtl)
{
for(int i=0;i<materials.size();i++)
{
bool ka = materials[i].Ka == mtl.Ka;
bool kd = materials[i].Kd == mtl.Kd;
bool ks = materials[i].Ks == mtl.Ks;
bool tr = materials[i].Tr == mtl.Tr;
bool illum = materials[i].illum == mtl.illum;
bool ns = materials[i].Ns == mtl.Ns;
bool map = materials[i].map_Kd == mtl.map_Kd;
if(ka & kd & ks & tr & illum & ns & map){return i;}
}
return -1;
}
};
}
}
}
#endif __VCGLIB_MATERIAL