C/C++ INI文件操作實(shí)現(xiàn)代碼

BOOL WritePrivateProfileString(
 LPCTSTR lpAppName, // pointer to section name
 LPCTSTR lpKeyName, // pointer to key name
 LPCTSTR lpString, // pointer to string to add
 LPCTSTR lpFileName // pointer to initialization filename
);

DWORD GetPrivateProfileString(
 LPCTSTR lpAppName, // points to section name
 LPCTSTR lpKeyName, // points to key name
 LPCTSTR lpDefault, // points to default string
 LPTSTR lpReturnedString, // points to destination buffer
 DWORD nSize, // size of destination buffer
 LPCTSTR lpFileName // points to initialization filename
);

UINT GetPrivateProfileInt( LPCTSTR lpAppName, // address of section name
 LPCTSTR lpKeyName, // address of key name
 INT nDefault, // return value if key name is not found
 LPCTSTR lpFileName // address of initialization filename
);

DWORD GetPrivateProfileSectionNames(
 LPTSTR lpszReturnBuffer, // address of return buffer
 DWORD nSize, // size of return buffer
 LPCTSTR lpFileName // address of initialization filename
);

DWORD GetPrivateProfileSection(
 LPCTSTR lpAppName, // address of section name
 LPTSTR lpReturnedString, // address of return buffer
 DWORD nSize, // size of return buffer
 LPCTSTR lpFileName // address of initialization filename
);

#pragma once
#include <iostream>
#include <fstream>
#include <sstream>
#include <vector>
#include <cstdlib>
#include <map>
 
using namespace std;
 
//INI文件結(jié)點(diǎn)存儲(chǔ)結(jié)構(gòu)
class ININode
{
public:
	ININode(string root, string key, string value)
	{
		this->root = root;
		this->key = key;
		this->value = value;
	}
	string root;
	string key;
	string value;
};
 
//鍵值對(duì)結(jié)構(gòu)體
class SubNode
{
public:
	void InsertElement(string key, string value)
	{
		sub_node.insert(pair<string, string>(key, value));
	}
	map<string, string> sub_node;
};
 
//INI文件操作類
class CMyINI
{
public:
	CMyINI();
	~CMyINI();
 
public:
	int ReadINI(string path);													//讀取INI文件
	string GetValue(string root, string key);									//由根結(jié)點(diǎn)和鍵獲取值
	vector<ININode>::size_type GetSize(){ return map_ini.size(); }				//獲取INI文件的結(jié)點(diǎn)數(shù)
	vector<ININode>::size_type SetValue(string root, string key, string value);	//設(shè)置根結(jié)點(diǎn)和鍵獲取值
	int WriteINI(string path);			//寫入INI文件
	void Clear(){ map_ini.clear(); }	//清空
	void Travel();						//遍歷打印INI文件
private:
	map<string, SubNode> map_ini;		//INI文件內(nèi)容的存儲(chǔ)變量
};

#include "StdAfx.h"
#include "MyINI.h"
 
#define INIDEBUG
 
CMyINI::CMyINI()
{
}
 
 
CMyINI::~CMyINI()
{
}
 
//************************************************************************
// 函數(shù)名稱: 	TrimString
// 訪問(wèn)權(quán)限: 	public 
// 創(chuàng)建日期:		2017/01/05
// 創(chuàng) 建 人:		
// 函數(shù)說(shuō)明:		去除空格
// 函數(shù)參數(shù): 	string & str	輸入的字符串
// 返 回 值: 	std::string &	結(jié)果字符串
//************************************************************************
string &TrimString(string &str)
{
	string::size_type pos = 0;
	while (str.npos != (pos = str.find(" ")))
		str = str.replace(pos, pos + 1, "");
	return str;
}
 
//************************************************************************
// 函數(shù)名稱: 	ReadINI
// 訪問(wèn)權(quán)限: 	public 
// 創(chuàng)建日期:		2017/01/05
// 創(chuàng) 建 人:		
// 函數(shù)說(shuō)明:		讀取INI文件，并將其保存到map結(jié)構(gòu)中
// 函數(shù)參數(shù): 	string path	INI文件的路徑
// 返 回 值: 	int
//************************************************************************
int CMyINI::ReadINI(string path)
{
	ifstream in_conf_file(path.c_str());
	if (!in_conf_file) return 0;
	string str_line = "";
	string str_root = "";
	vector<ININode> vec_ini;
	while (getline(in_conf_file, str_line))
	{
		string::size_type left_pos = 0;
		string::size_type right_pos = 0;
		string::size_type equal_div_pos = 0;
		string str_key = "";
		string str_value = "";
		if ((str_line.npos != (left_pos = str_line.find("["))) && (str_line.npos != (right_pos = str_line.find("]"))))
		{
			//cout << str_line.substr(left_pos+1, right_pos-1) << endl;
			str_root = str_line.substr(left_pos + 1, right_pos - 1);
		}
 
		if (str_line.npos != (equal_div_pos = str_line.find("=")))
		{
			str_key = str_line.substr(0, equal_div_pos);
			str_value = str_line.substr(equal_div_pos + 1, str_line.size() - 1);
			str_key = TrimString(str_key);
			str_value = TrimString(str_value);
			//cout << str_key << "=" << str_value << endl;
		}
 
		if ((!str_root.empty()) && (!str_key.empty()) && (!str_value.empty()))
		{
			ININode ini_node(str_root, str_key, str_value);
			vec_ini.push_back(ini_node);
			//cout << vec_ini.size() << endl;
		}
	}
	in_conf_file.close();
	in_conf_file.clear();
 
	//vector convert to map
	map<string, string> map_tmp;
	for (vector<ININode>::iterator itr = vec_ini.begin(); itr != vec_ini.end(); ++itr)
	{
		map_tmp.insert(pair<string, string>(itr->root, ""));
	}	//提取出根節(jié)點(diǎn)
	for (map<string, string>::iterator itr = map_tmp.begin(); itr != map_tmp.end(); ++itr)
	{
#ifdef INIDEBUG
		cout << "根節(jié)點(diǎn)： " << itr->first << endl;
#endif	//INIDEBUG
		SubNode sn;
		for (vector<ININode>::iterator sub_itr = vec_ini.begin(); sub_itr != vec_ini.end(); ++sub_itr)
		{
			if (sub_itr->root == itr->first)
			{
#ifdef INIDEBUG
				cout << "鍵值對(duì)： " << sub_itr->key << "=" << sub_itr->value << endl;
#endif	//INIDEBUG
				sn.InsertElement(sub_itr->key, sub_itr->value);
			}
		}
		map_ini.insert(pair<string, SubNode>(itr->first, sn));
	}
	return 1;
}
 
//************************************************************************
// 函數(shù)名稱: 	GetValue
// 訪問(wèn)權(quán)限: 	public 
// 創(chuàng)建日期:		2017/01/05
// 創(chuàng) 建 人:		
// 函數(shù)說(shuō)明:		根據(jù)給出的根結(jié)點(diǎn)和鍵值查找配置項(xiàng)的值
// 函數(shù)參數(shù): 	string root		配置項(xiàng)的根結(jié)點(diǎn)
// 函數(shù)參數(shù): 	string key		配置項(xiàng)的鍵
// 返 回 值: 	std::string		配置項(xiàng)的值
//************************************************************************
string CMyINI::GetValue(string root, string key)
{
	map<string, SubNode>::iterator itr = map_ini.find(root);
	map<string, string>::iterator sub_itr = itr->second.sub_node.find(key);
	if (!(sub_itr->second).empty())
		return sub_itr->second;
	return "";
}
 
//************************************************************************
// 函數(shù)名稱: 	WriteINI
// 訪問(wèn)權(quán)限: 	public 
// 創(chuàng)建日期:		2017/01/05
// 創(chuàng) 建 人:		
// 函數(shù)說(shuō)明:		保存XML的信息到文件中
// 函數(shù)參數(shù): 	string path	INI文件的保存路徑
// 返 回 值: 	int
//************************************************************************
int CMyINI::WriteINI(string path)
{
	ofstream out_conf_file(path.c_str());
	if (!out_conf_file)
		return -1;
	//cout << map_ini.size() << endl;
	for (map<string, SubNode>::iterator itr = map_ini.begin(); itr != map_ini.end(); ++itr)
	{
		//cout << itr->first << endl;
		out_conf_file << "[" << itr->first << "]" << endl;
		for (map<string, string>::iterator sub_itr = itr->second.sub_node.begin(); sub_itr != itr->second.sub_node.end(); ++sub_itr)
		{
			//cout << sub_itr->first << "=" << sub_itr->second << endl;
			out_conf_file << sub_itr->first << "=" << sub_itr->second << endl;
		}
	}
 
	out_conf_file.close();
	out_conf_file.clear();
 
	return 1;
}
 
 
//************************************************************************
// 函數(shù)名稱: 	SetValue
// 訪問(wèn)權(quán)限: 	public 
// 創(chuàng)建日期:		2017/01/05
// 創(chuàng) 建 人:		
// 函數(shù)說(shuō)明:		設(shè)置配置項(xiàng)的值
// 函數(shù)參數(shù): 	string root		配置項(xiàng)的根節(jié)點(diǎn)
// 函數(shù)參數(shù): 	string key		配置項(xiàng)的鍵
// 函數(shù)參數(shù): 	string value	配置項(xiàng)的值
// 返 回 值: 	std::vector<ININode>::size_type	
//************************************************************************
vector<ININode>::size_type CMyINI::SetValue(string root, string key, string value)
{
	map<string, SubNode>::iterator itr = map_ini.find(root);	//查找
	if (map_ini.end() != itr)
	{
		//itr->second.sub_node.insert(pair<string, string>(key, value));
		itr->second.sub_node[key] = value;
	}	//根節(jié)點(diǎn)已經(jīng)存在了，更新值
	else
	{
		SubNode sn;
		sn.InsertElement(key, value);
		map_ini.insert(pair<string, SubNode>(root, sn));
	}	//根節(jié)點(diǎn)不存在，添加值
 
	return map_ini.size();
}
 
//************************************************************************
// 函數(shù)名稱: 	Travel
// 訪問(wèn)權(quán)限: 	public 
// 創(chuàng)建日期:		2017/01/05
// 創(chuàng) 建 人:		
// 函數(shù)說(shuō)明:		遍歷打印INI文件
// 返 回 值: 	void
//************************************************************************
void CMyINI::Travel()
{
	for (map<string, SubNode>::iterator itr = this->map_ini.begin(); itr!= this->map_ini.end(); ++itr)
	{
		//root
		cout << "[" << itr->first << "]" << endl;
		for (map<string, string>::iterator itr1=itr->second.sub_node.begin(); itr1!=itr->second.sub_node.end();
			++itr1)
		{
			cout << " " << itr1->first << " = " << itr1->second << endl;
		}
	}
	
}

CMyINI *p = new CMyINI();
p->ReadINI("Setting.ini");
cout << "\n原始INI文件內(nèi)容：" << std::endl;
p->Travel();
p->SetValue("setting", "hehe", "eheh");
cout << "\n增加節(jié)點(diǎn)之后的內(nèi)容：" << std::endl;
p->Travel();
cout << "\n修改節(jié)點(diǎn)之后的內(nèi)容：" << std::endl;
p->SetValue("kk", "kk", "2");
p->Travel();
p->WriteINI("Setting.ini");