vcglib/wrap/io_trimesh/alnParser.h

155 lines
5.4 KiB
C++

/****************************************************************************
* MeshLab o o *
* An extendible mesh processor o o *
* _ O _ *
* Copyright(C) 2005, 2021 \/)\/ *
* 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. *
* *
****************************************************************************/
#ifndef __OM_ALNPARSER_
#define __OM_ALNPARSER_
#include <vector>
#include <string>
#include <vcg/math/matrix44.h>
struct RangeMap
{
RangeMap() : quality(1.0f) {}
std::string filename;
vcg::Matrix44d transformation;
float quality;
};
class ALNParser
{
public:
enum ALNParserCodes {NoError, CantOpen, UnexpectedEOF, ExpectingComment};
static const char* ErrorMsg(int message_code)
{
static const char* error_msg[4] = {"No errors", "Can't open file", "Premature End of file", "I was expecting a comment"};
if(message_code>3 || message_code<0)
return "Unknown error";
else
return error_msg[message_code];
};
static int BuildALN(std::vector<RangeMap> &rangemaps, std::vector< std::string > &files)
{
rangemaps.clear();
rangemaps.resize( files.size() );
std::vector< RangeMap >::iterator rm = rangemaps.begin();
std::vector< std::string >::iterator it = files.begin();
std::vector< std::string >::iterator end = files.end();
for ( ; it!=end; it++, rm++)
{
(*rm).filename = (*it);
(*rm).quality = 1.0f;
(*rm).transformation.SetIdentity();
}
files.clear();
return NoError;
};
static int ParseALN(std::vector<RangeMap> &rangemaps, const char *ALNname)
{
rangemaps.clear();
FILE *stream=fopen(ALNname, "rt");
if(stream==NULL)
return CantOpen;
int mesh_number;
fscanf(stream, "%i\n", &mesh_number);
char buffer[1024];
for (int m=0; m<mesh_number; m++)
{
RangeMap rm;
fgets(buffer, 1024, stream);
*strchr(buffer, '\n')=0;
if(strchr(buffer,'\r')) *strchr(buffer,'\r')=0;
rm.filename = buffer;
fgets(buffer, 1024, stream);
if(buffer[0]!='#')
return ExpectingComment;
*strchr(buffer,'\n')=0;
if(strchr(buffer,'\r')) *strchr(buffer,'\r')=0;
char *occurrence = strchr(buffer, 'W');
if(occurrence!=NULL && occurrence[1]==':')
rm.quality = (float) atof(occurrence+2);
assert(rm.quality>0);
fscanf(stream,"%lf %lf %lf %lf \n",&(rm.transformation[0][0]),&(rm.transformation[0][1]),&(rm.transformation[0][2]),&(rm.transformation[0][3]));
fscanf(stream,"%lf %lf %lf %lf \n",&(rm.transformation[1][0]),&(rm.transformation[1][1]),&(rm.transformation[1][2]),&(rm.transformation[1][3]));
fscanf(stream,"%lf %lf %lf %lf \n",&(rm.transformation[2][0]),&(rm.transformation[2][1]),&(rm.transformation[2][2]),&(rm.transformation[2][3]));
fscanf(stream,"%lf %lf %lf %lf \n",&(rm.transformation[3][0]),&(rm.transformation[3][1]),&(rm.transformation[3][2]),&(rm.transformation[3][3]));
rangemaps.push_back(rm);
}
fclose(stream);
return NoError;
} // end of ParseALN
static bool SaveALN(const char *alnfile, std::vector<std::string> &names)
{
std::vector<vcg::Matrix44f> Tr(names.size());
for(int i=0; i < static_cast<int>(Tr.size()); ++i) Tr[i].SetIdentity();
return SaveALN(alnfile,names, Tr);
}
template <class matrixfloat>
static bool SaveALN(const char *alnfile, std::vector<std::string> &names, std::vector<vcg::Matrix44<matrixfloat> > &Tr)
{
// printf("Saving aln file %s\n",alnfile);
FILE *fp=fopen(alnfile,"w");
if(!fp)
{
printf("unable to open file %s\n",alnfile);
return false;
}
fprintf(fp,"%i\n",(int)names.size());
for(int i=0;i < static_cast<int>(names.size());++i)
{
fprintf(fp,"%s\n",names[i].c_str());
fprintf(fp,"#\n");
fprintf(fp,"%lf %lf %lf %lf \n",(Tr[i][0][0]),(Tr[i][0][1]),(Tr[i][0][2]),(Tr[i][0][3]));
fprintf(fp,"%lf %lf %lf %lf \n",(Tr[i][1][0]),(Tr[i][1][1]),(Tr[i][1][2]),(Tr[i][1][3]));
fprintf(fp,"%lf %lf %lf %lf \n",(Tr[i][2][0]),(Tr[i][2][1]),(Tr[i][2][2]),(Tr[i][2][3]));
fprintf(fp,"%lf %lf %lf %lf \n",(Tr[i][3][0]),(Tr[i][3][1]),(Tr[i][3][2]),(Tr[i][3][3]));
}
fprintf(fp,"0\n");
fclose(fp);
return true;
}
};
#endif //__OM_ALNPARSER_