C++读入"N,X,Y,Z"格式文本文件到Eigen3 Matrix

C++读入"N,X,Y,Z"格式文本文件到Eigen3 Matrix,以及相同格式输出方法

很多数据资料的格式类似这样:

1,-2085738.7757,5503702.8697,2892977.6829
2,-2071267.5135,5520926.7235,2883341.8135
3,-2079412.5535,5512450.8800,2879771.2119
4,-2093693.1744,5511218.2651,2869861.8947
5,-2113681.5062,5491864.0382,2896934.4852
6,-2100573.2849,5496675.0138,2894377.6030

其中数据按照N(点号),X,Y,Z(三维坐标)排序。
这里提供一种C++读入"N,X,Y,Z"格式文本文件到Eigen3 Matrix的方法,以及对应的同格式输出方法

`

#pragma once
#include <fstream>
#include <iostream>
#include <string>
#include <Eigen/Dense>
#include <vector>
#include <cmath>
#include <iomanip>

using namespace std;
using namespace Eigen;

// 字符串分割
void SplitString(const std::string& s, std::vector<std::string>& v, const std::string& c)
{
	std::string::size_type pos1, pos2;
	pos2 = s.find(c);
	pos1 = 0;
	while (std::string::npos != pos2)
	{
		v.push_back(s.substr(pos1, pos2 - pos1));

		pos1 = pos2 + c.size();
		pos2 = s.find(c, pos1);
	}
	if (pos1 != s.length())
		v.push_back(s.substr(pos1));
}
// 读入相应格式的xyz文件
void ReadXYZFile(string filepath, MatrixXd& origin_data)
{
	ifstream infile;
	infile.open(filepath);
	cout << "Reading XYZ File: " << filepath << endl;
	if (!infile.is_open())
	{
		cout << "File Cannot Open" << endl;
		exit(1);
	}

	int r = 0; // 逐行加载数据
	char buffer[100];
	while (!infile.eof())
	{
		// getline只能读成char*,
		// 而SplitString只能切割string,
		// 而atof又只能转化char*到double
		infile.getline(buffer, 100);
		// cout << buffer << endl;
		string line = buffer;
		if (line == "")
		{
			continue;
		}
		vector<string> vector_data;
		SplitString(line, vector_data, ",");
		for (int c = 0; c < origin_data.cols(); c++)
		{
			origin_data(r, c) = atof(vector_data[c].c_str());
		}
		r++;

	}
	return;


}
// 将矩阵按读入的相同格式保存至相应路径
void WriteXYZFile(string filepath, MatrixXd& trans_data)
{
	ofstream outfile;
	outfile.open(filepath, ios::out | ios::trunc);
	for (int r = 0; r < trans_data.rows(); r++)
	{
		for (int c = 0; c < trans_data.cols(); c++)
		{
			if (c < trans_data.cols() - 1)
			{
				outfile << trans_data(r, c) << ',';
			}
			if (c == trans_data.cols() - 1)
			{
				outfile << trans_data(r, c);
			}

		}
		outfile << endl;
	}
	cout << "Write XYZ File: " << filepath << endl;
	outfile.close();
	return;
}
发布了10 篇原创文章 · 获赞 8 · 访问量 1597

猜你喜欢

转载自blog.csdn.net/Canvaskan/article/details/105215077