vcglib/wrap/io_trimesh/export_vrml.h

#ifndef __VCGLIB_EXPORT_WRL
#define __VCGLIB_EXPORT_WRL

#include <stdio.h>
#include <wrap/io_trimesh/io_mask.h>

#include "io_material.h"

namespace vcg {
	namespace tri {
		namespace io {

			template <class SaveMeshType>
			class ExporterWRL
			{
			public:
				typedef typename SaveMeshType::VertexPointer VertexPointer;
				typedef typename SaveMeshType::ScalarType ScalarType;
				typedef typename SaveMeshType::VertexType VertexType;
				typedef typename SaveMeshType::FaceType FaceType;
				typedef typename SaveMeshType::VertexIterator VertexIterator;
				typedef typename SaveMeshType::FaceIterator FaceIterator;
				
				static int Save(SaveMeshType &m, const char * filename, const int &mask, CallBackPos *cb=0)
				{					
					FILE *fp;
					fp = fopen(filename,"wb");
					if(fp==NULL)
						return 1;

					// Header					
					fprintf(fp,
						 "#VRML V2.0 utf8\n"  
   	 					"\n" 	 					
						"# Generated by VCGLIB, (C)Copyright 1999-2001 VCG, IEI-CNR\n"
						"\n"						
						"NavigationInfo {\n"						
						"	type [ \"EXAMINE\", \"ANY\" ]\n"						
						"}\n"						
						);					

					// Tranche principale
					double ss = 8.0/m.bbox.Diag();					
					m.bbox.Diag();
					
					fprintf(fp,
					"Transform {\n"
					"  scale %g %g %g\n"
					"  translation %g %g %g\n"
					"  children\n"
					"  [\n"
					,ss
					,ss
					,ss
					,  -m.bbox.Center()[0]*ss
					,  -m.bbox.Center()[1]*ss
					,-3-m.bbox.Center()[2]*ss
						);

					// Start Shape
					fprintf(fp,
					"    Shape\n"
					"    {\n"
					"      geometry IndexedFaceSet\n"
					"      {\n"
					"        creaseAngle .5\n"
					"        solid FALSE\n"
					"        coord Coordinate\n"
					"        {\n"
					"          point\n"
					"          ["
					);
					FaceIterator fi;	
					VertexIterator vi;	
					std::map<VertexPointer,int> index;
					int ind;

					// Vertici
					for(ind=0,vi=m.vert.begin(); vi!=m.vert.end(); ++vi, ++ind)	
					{
						if(ind%4==0)
							fprintf(fp,
									"\n            "
									);
						fprintf(fp,
							"%g %g %g, "
							,(*vi).P()[0]
							,(*vi).P()[1]
							,(*vi).P()[2]
							);
						index[&*vi] = ind;

					}
					fprintf(fp,"\n"
								"          ]\n"
								"        }\n"
								);
					
					
					if( m.HasPerVertexColor() && (mask & vcg::tri::io::Mask::IOM_VERTCOLOR))
					{
						fprintf(fp,
							"        color Color\n"
							"        {\n"
							"          color\n"
							"          ["
							);
						for(ind=0,vi=m.vert.begin();vi!=m.vert.end();++vi,++ind)
						{
							float r = float(vi->C()[0])/255;
							float g = float(vi->C()[1])/255;
							float b = float(vi->C()[2])/255;

							if(ind%4==0)
								fprintf(fp,"\n            ");
							fprintf(fp,"%g %g %g,",r,g,b);
						}
						fprintf(fp,	
							"\n"
							"          ]\n"							
							"        }\n"
							"        colorIndex\n"
							"        ["    						
							);
						
						for(ind=0,fi=m.face.begin(); fi!=m.face.end(); ++fi,++ind)
						{
							if(ind%4==0)
								fprintf(fp,"\n          ");
							for (int j = 0; j < 3; j++)
								fprintf(fp,"%i,",index[(*fi).V(j)]);	
							fprintf(fp,"-1,");						
						}						
						fprintf(fp,
							"\n"
							"        ]\n"
							);
					}
					else if( m.HasPerWedgeColor() &&  (mask & vcg::tri::io::Mask::IOM_WEDGCOLOR ))
					{						
						fprintf(fp,
							"        color Color\n"	
							"        {\n"							
							"          color\n"							
							"          ["							);
						for(ind=0,fi=m.face.begin();fi!=m.face.end();++fi,++ind)
						{							
							for(int z=0;z<3;++z)							
							{								
								float r = float(fi->WC(z)[0])/255;
								float g = float(fi->WC(z)[1])/255;
								float b = float(fi->WC(z)[2])/255;
								if(ind%4==0)
									fprintf(fp,"\n            ");
								fprintf(fp,"%g %g %g,",r,g,b);
							}
						}						
						fprintf(fp,							
							"\n"							
							"          ]\n"							
							"        }\n"						
							"        colorIndex\n"							
							"        ["							);						
						int nn = 0;						
						for(ind=0,fi=m.face.begin(); fi!=m.face.end(); ++fi,++ind)
						{							
							if(ind%4==0)							
								fprintf(fp,"\n          ");							
							for (int j = 0; j < 3; j++)							
							{								
								fprintf(fp,"%i,",nn);								
								++nn;							
							}							
							fprintf(fp,"-1,");						
						}						
						fprintf(fp,							
							"\n"							
							"       ]\n"							
							);					
					}					
					else if (HasPerWedgeTexture(m) &&(mask &  vcg::tri::io::Mask::IOM_WEDGTEXCOORD))
					{
						fprintf(fp,
							"\n"
							"        texCoord TextureCoordinate\n"
							"        {\n"
							"          point\n"
							"          [\n"
							);
						for(ind=0,fi=m.face.begin(); fi!=m.face.end(); ++fi,++ind)
						{
							if(ind%4==0)
								fprintf(fp,"\n          ");
							for (int j = 0; j < 3; j++)
							{
								fprintf(fp,"%g %g, ",fi->WT(j).u(),fi->WT(j).v());
							}
						}
						fprintf(fp,
							"\n"
							"          ]\n"
							"        }\n"
							"        texCoordIndex\n"
							"        [\n"
							);
						int nn = 0;
						for(ind=0,fi=m.face.begin(); fi!=m.face.end(); ++fi,++ind)
						{
							if(ind%4==0)
								fprintf(fp,"\n          ");
 							for (int j = 0; j < 3; j++)
							{
								fprintf(fp,"%d,",nn);
								++nn;
							}
							fprintf(fp,"-1,");
						}
						fprintf(fp,
							"\n"
							"        ]\n"
							);
					}
					fprintf(fp,
							"        coordIndex\n"
							"        ["
							);
					// Facce
					for(ind=0,fi=m.face.begin(); fi!=m.face.end(); ++fi,++ind)
					{
						if(ind%6==0)
						{
							fprintf(fp,"\n          ");
						}
						for (int j = 0; j < 3; j++)
							fprintf(fp,"%i,",index[(*fi).V(j)]);
						fprintf(fp,"-1,");	
					}
					fprintf(fp,
							"\n"
							"        ]\n"
							"      }\n"
							"      appearance Appearance\n"
							"      {\n"						
							"        material Material\n"
							"        {\n"
							"	       ambientIntensity 0.2\n"
							"	       diffuseColor 0.9 0.9 0.9\n"
							"	       specularColor .1 .1 .1\n"
							"	       shininess .5\n"
							"        }\n"
							);
					if(m.textures.size())
					{
						fprintf(fp,
							"        texture ImageTexture { url \"  %s  \" }\n"
							,m.textures[0].c_str()
							);
					}
					fprintf(fp,
						"      }\n"
						"    }\n"
						"  ]\n"
						"}\n"
						);
					fclose(fp);
					return 0;
	}
	static int GetExportMaskCapability()
	{
		int capability = 0;

		//vert
		capability |= MeshModel::IOM_VERTCOLOR;
		
		//wedg
		capability |= MeshModel::IOM_WEDGTEXCOORD;
		capability |= MeshModel::IOM_WEDGCOLOR;

		return capability;
	}

	static const char *ErrorMsg(int error)
	{
		static std::vector<std::string> wrl_error_msg;
		if(wrl_error_msg.empty())
		{
			wrl_error_msg.resize(2 );
			wrl_error_msg[0]="No errors";
			wrl_error_msg[1]="Can't open file";
		}
		if(error>1 || error<0) return "Unknown error";
		else return wrl_error_msg[error].c_str();
		}

}; // end class
} // end Namespace io
} // end Namespace tri
} // end Namespace vcg

#endif