结构体数组保存进二进制文件的简单做法

clever101

于 2023-11-17 20:01:09 发布

阅读量274

点赞数

分类专栏： C/C++ 文章标签：结构体数组保存二进制文件

本文链接：https://blog.csdn.net/clever101/article/details/134469921

版权

C/C++ 专栏收录该内容

132 篇文章 5 订阅

订阅专栏

作者：朱金灿
来源：clever101的专栏

为什么大多数人学不会人工智能编程？>>>

最近面临这样一个需求：以比较节省存储空间的存储一组坐标点到文件，要求程序能够跨平台读写这种文件。思考了一下，比较简单的方式是以二进制文件的方式保存，先存入点的个数，然后把点结构体逐个存储进去。因为要节省存储空间，得保证结构体是以单字节对齐的。代码如下：



#include <stdio.h>
#include <string>
#include <vector>


/* 对齐结构成员到1字节 */
#ifdef __GNUC__
#define GNUC_PACKED __attribute__((packed))
#else
#define GNUC_PACKED
#endif
#ifdef __arm 
#define ARM_PACKED  __packed 
#else 
#define ARM_PACKED 
#endif
#if defined(_WIN32) || defined(_WIN64)
#pragma pack(1)
#endif
ARM_PACKED

/*
坐标点结构体
*/
struct  stPointCoodinate
{
	double m_CoodinateX;  //x坐标
	double m_CoodinateY;  //y坐标
};

#if defined(_WIN32) || defined(_WIN64)
#pragma pack()
#endif

/*
将点结构体转化为二进制文件
具体是先写入点的数目，然后再逐个写入点
*/
int WritePointsToFile(const std::vector<stPointCoodinate>& vecTarget, const std::string& binFile)
{
	if (vecTarget.empty())
	{
		printf("point number is 0\n");
		return -1;
	}

	FILE* fp = fopen(binFile.c_str(), "wb");
	if (NULL == fp)
	{
		printf("create point file failed!\n");
		return -2;
	}
	int nTargetNum = vecTarget.size();
	fwrite(&nTargetNum, sizeof(int), 1, fp);
	fwrite(&vecTarget[0], sizeof(stPointCoodinate), nTargetNum, fp);

	fclose(fp);

	return 0;
}

/*
从文件中读入点数据
*/
int ReadPointsFromFile(const std::string& binFile, std::vector<stPointCoodinate>& vecTarget)
{
	FILE* fp = fopen(binFile.c_str(), "rb");
	if (NULL == fp)
	{
		printf("point file open failed!\n");
		return -1;
	}
	int nTargetNum = 0;
	fread(&nTargetNum, sizeof(int), 1, fp);
	vecTarget.resize(nTargetNum);
	fread(&vecTarget[0], sizeof(stPointCoodinate), nTargetNum, fp);

	fclose(fp);

	return 0;
}

/*
测试点文件的读写
*/
void TestPointFileRW()
{
	std::vector<stPointCoodinate> vecTarget;
	for (int i = 0; i < 5; i++)
	{
		stPointCoodinate pt;
		pt.m_CoodinateX = 100.0 + i * 10.0;
		pt.m_CoodinateY = 100.0 + i * 10.0;
		vecTarget.push_back(pt);
	}

#if defined(_WIN32) || defined(_WIN64)

	std::string binFile = "D:\\TestData\\test_point.pt";

#else

	std::string binFile = "/home/lx/TestData/test_point.pt";

#endif

	int ret = WritePointsToFile(vecTarget, binFile);

	vecTarget.clear();

	ret = ReadPointsFromFile(binFile, vecTarget);

	if (!vecTarget.empty())
	{
		int num = vecTarget.size();
		for (int i = 0; i < num; i++)
		{
			printf("%f,%f \n", vecTarget[i].m_CoodinateX, vecTarget[i].m_CoodinateY);
		}
	}
}

int main(int argc, char* argv[])
{
	TestPointFileRW();

	getchar();
    return 0;
}