openMVS/vcglib/wrap/io_trimesh/alnParser.h

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