#include "Ini.h"
/******************************************************************************
* 功 能：构造函数
* 参 数：无
* 返回值：无
* 备 注：
******************************************************************************/
CIni::CIni( )
{
memset( m_szKey,0,sizeof(m_szKey) );
m_fp = NULL;
m_bModify = false;
}
/******************************************************************************
* 功 能：析构函数
* 参 数：无
* 返回值：无
* 备 注：
******************************************************************************/
CIni::~CIni()
{
m_Map.clear();
}
/******************************************************************************
* 功 能：打开文件函数
* 参 数：无
* 返回值：
* 备 注：
******************************************************************************/
INI_RES CIni::OpenFile(const char* pathName, const char* type)
{
string szLine,szMainKey,szLastMainKey,szSubKey;
char strLine[ CONFIGLEN ] = { 0 };
KEYMAP mLastMap;
int nIndexPos = -1;
int nLeftPos = -1;
int nRightPos = -1;
m_fp = fopen(pathName, type);
if (m_fp == NULL)
{
printf( "open inifile %s error!\n",pathName );
return INI_OPENFILE_ERROR;
}
// fseek(m_fp, 0, SEEK_SET);
m_Map.clear();
while( fgets( strLine, CONFIGLEN,m_fp) )
{
szLine.assign( strLine );
//删除字符串中的非必要字符
nLeftPos = szLine.find("\n" );
if( string::npos != nLeftPos )
{
szLine.erase( nLeftPos,1 );
}
nLeftPos = szLine.find("\r" );
if( string::npos != nLeftPos )
{
szLine.erase( nLeftPos,1 );
}
//判断是否是主键
nLeftPos = szLine.find("[");
nRightPos = szLine.find("]");
if( nLeftPos != string::npos && nRightPos != string::npos )
{
szLine.erase( nLeftPos,1 );
nRightPos--;
szLine.erase( nRightPos,1 );
szLastMainKey = szLine ;
m_Map[ szLastMainKey ] = mLastMap;
mLastMap.clear();
}
else
{
//是否是子键
if( nIndexPos = szLine.find("=" ),string::npos != nIndexPos)
{
string szSubKey,szSubValue;
szSubKey = szLine.substr( 0,nIndexPos );
szSubValue = szLine.substr( nIndexPos+1,szLine.length()-nIndexPos-1);
//mLastMap[szSubKey] = szSubValue ;
MAINKEYMAP::iterator it = m_Map.find(szLastMainKey);
if (it != m_Map.end())
{
it->second[szSubKey] = szSubValue;
}
}
else
{
//TODO:不符合ini键值模板的内容 如注释等
}
}
}
//插入最后一次主键
// m_Map[ szLastMainKey ] = mLastMap;
return INI_SUCCESS;
}
/******************************************************************************
* 功 能：关闭文件函数
* 参 数：无
* 返回值：
* 备 注：
******************************************************************************/
INI_RES CIni::CloseFile()
{
if (m_bModify)
{
fseek(m_fp, 0, SEEK_SET);
//保存
for (MAINKEYMAP::iterator it = m_Map.begin(); it != m_Map.end(); ++it)
{
string AppName = "[";
AppName.append(it->first);
AppName.append("]\r\n");
fwrite(AppName.c_str(),sizeof(char), AppName.length(), m_fp);
for (KEYMAP::iterator it2 = it->second.begin(); it2 != it->second.end(); ++it2)
{
string KeyPair = it2->first;
KeyPair.append("=");
KeyPair.append(it2->second);
KeyPair.append("\r\n");
fwrite(KeyPair.c_str(), sizeof(char), KeyPair.length(), m_fp);
}
}
}
if (m_fp != NULL)
{
fclose(m_fp);
m_fp = NULL;
}
return INI_SUCCESS;
}
/******************************************************************************
* 功 能：获取[SECTION]下的某一个键值的字符串
* 参 数：
* char* mAttr 输入参数 主键
* char* cAttr 输入参数 子键
* char* value 输出参数 子键键值
* 返回值：
* 备 注：
******************************************************************************/
INI_RES CIni::GetKey(const char* mAttr, const char* cAttr, char* pValue)
{
/* KEYMAP mKey = m_Map[ mAttr ];
string sTemp = mKey[ cAttr ];
strcpy( pValue,sTemp.c_str() );*/
MAINKEYMAP::iterator it = m_Map.find(mAttr);
if (it == m_Map.end())
return INI_NO_ATTR;
KEYMAP::iterator it2 = it->second.find(cAttr);
if (it2 == it->second.end())
return INI_NO_ATTR;
string sTemp = it2->second;
strcpy( pValue,sTemp.c_str() );
return INI_SUCCESS;
}
void CIni::SetKey(const char* mAttr, const char* cAttr, char* value)
{
MAINKEYMAP::iterator it = m_Map.find(mAttr);
if (it != m_Map.end())
{
it->second[cAttr] = value;
}
else
{
KEYMAP keymp;
keymp[cAttr] = value;
m_Map[mAttr] = keymp;
}
}
/******************************************************************************
* 功 能：获取整形的键值
* 参 数：
* cAttr 主键
* cAttr 子键
* 返回值：正常则返回对应的数值 未读取成功则返回0(键值本身为0不冲突)
* 备 注：
******************************************************************************/
int CIni::GetInt(const char* mAttr, const char* cAttr, int iDefault)
{
int nRes = iDefault;
memset( m_szKey,0,sizeof(m_szKey) );
if( INI_SUCCESS == GetKey( mAttr,cAttr,m_szKey ) )
{
nRes = atoi( m_szKey );
}
return nRes;
}
/******************************************************************************
* 功 能：获取键值的字符串
* 参 数：
* cAttr 主键
* cAttr 子键
* 返回值：正常则返回读取到的子键字符串 未读取成功则返回"NULL"
* 备 注：
******************************************************************************/
char *CIni::GetStr(const char* mAttr, const char* cAttr )
{
memset( m_szKey,0,sizeof(m_szKey) );
if( INI_SUCCESS != GetKey( mAttr,cAttr,m_szKey ) )
{
strcpy( m_szKey,"NULL" );
}
return m_szKey;
}
string CIni::GetStr(const char* mAttr, const char* cAttr, string strDefault)
{
memset( m_szKey,0,sizeof(m_szKey) );
if( INI_SUCCESS != GetKey( mAttr,cAttr, m_szKey ) )
{
return strDefault;
}
else
return m_szKey;
}
void CIni::SetStr(const char* mAttr, const char* cAttr, char* str)
{
SetKey(mAttr, cAttr, str);
int size = m_Map.size();
m_bModify = true;
}
评论1