#include "IniFile.h"
#include <fstream>
#include <strstream>
namespace fish
{
IniFile::IniFile(const string &fileName)
:m_fileName(fileName)
{
ReadFile();
}
IniFile::~IniFile(void)
{
WriteFile();
}
bool IniFile::ReadFile( void )
{
ifstream in(m_fileName.c_str());
bool bopen = in.is_open();
string line;
while( getline(in,line) )
{
m_vctLine.push_back(line);
}
return true;
}
string IniFile::ReadString( const string §ion, const string &key, const string &value )
{
for( size_t i = 0;i < m_vctLine.size(); ++i )
{
string §ion_line = m_vctLine[i];
size_t sec_begin_pos = section_line.find('[');
if( sec_begin_pos == string::npos || sec_begin_pos != 0 )
{
continue;
}
size_t sec_end_pos = section_line.find( ']', sec_begin_pos );
if( sec_end_pos == string::npos )
{
continue;
}
if( section != Trim(section_line.substr(sec_begin_pos+1, sec_end_pos-sec_begin_pos-1) ) )
{
continue;
}
//find key, ÕâÀïµÄϱ껹ÊÇi
for( ++i; i < m_vctLine.size(); ++i )
{
string &key_line = m_vctLine[i];
size_t sec_pos = key_line.find('[');
if( sec_pos != string::npos && sec_pos == 0 )
{
--i; //reback a step,find again
break;//the line is section line
}
size_t begin_key_pos = key_line.find(key);
if( begin_key_pos == string::npos )
{
continue;
}
size_t equal_pos = key_line.find( '=', begin_key_pos );
if( equal_pos == string::npos )
{
continue;
}
if( key != Trim(key_line.substr( begin_key_pos, equal_pos - begin_key_pos) ) )
{
continue;
}
size_t comment_pos = key_line.find( "#", equal_pos + 1 );
if( comment_pos != string::npos )
{
//²»Òª×¢ÊÍ
return Trim(key_line.substr( equal_pos + 1, comment_pos - equal_pos - 1 ));
}
return Trim(key_line.substr( equal_pos + 1 ));
}
}
return value;
}
bool IniFile::WriteString( const string §ion, const string &key, const string &value )
{
for( size_t i = 0;i < m_vctLine.size(); ++i )
{
string §ion_line = m_vctLine[i];
size_t sec_begin_pos = section_line.find('[');
if( sec_begin_pos == string::npos || sec_begin_pos != 0 )
{
continue;
}
size_t sec_end_pos = section_line.find( ']', sec_begin_pos );
if( sec_end_pos == string::npos )
{
continue;
}
if( section != Trim(section_line.substr(sec_begin_pos+1, sec_end_pos-sec_begin_pos-1)) )
{
continue;
}
//find key, ÕâÀïµÄϱ껹ÊÇi
for( ++i; i < m_vctLine.size(); ++i )
{
string &key_line = m_vctLine[i];
size_t sec_pos = key_line.find('[');
if( sec_pos != string::npos && sec_pos == 0 )
{
--i; //reback a step,find again
break;//the line is section line
}
size_t begin_key_pos = key_line.find(key);
if( begin_key_pos == string::npos )
{
continue;
}
size_t equal_pos = key_line.find( '=', begin_key_pos );
if( equal_pos == string::npos )
{
continue;
}
if( key != Trim(key_line.substr( begin_key_pos, equal_pos - begin_key_pos )) )
{
continue;
}
size_t comment_pos = key_line.find( "#", equal_pos + 1 );
string new_line = key_line.substr( 0, equal_pos + 1 ) + value;
if( comment_pos != string::npos )
{
new_line += key_line.substr( comment_pos );
}
key_line = new_line;
return true;
}
//δÕÒµ½key
m_vctLine.insert( m_vctLine.begin() + i, key + "=" + value );
return true;
}
//sectionºÍkey¶¼²»´æÔÚ
m_vctLine.insert( m_vctLine.end(), "" );
m_vctLine.insert( m_vctLine.end(), "[" + section + "]" );
m_vctLine.insert( m_vctLine.end(), key + "=" + value );
return true;
}
bool IniFile::RemoveSection( const string §ion )
{
for( size_t i = 0;i < m_vctLine.size(); ++i )
{
string §ion_line = m_vctLine[i];
size_t sec_begin_pos = section_line.find('[');
if( sec_begin_pos == string::npos || sec_begin_pos != 0 )
{
continue;
}
size_t sec_end_pos = section_line.find( ']', sec_begin_pos );
if( sec_end_pos == string::npos )
{
continue;
}
if( section != Trim(section_line.substr(sec_begin_pos+1, sec_end_pos-sec_begin_pos-1)) )
{
continue;
}
//ÕÒµ½ÁËsection
size_t del_begin = i;
for( ++i ; i < m_vctLine.size(); ++i )
{
string &next_section = m_vctLine[i];
size_t next_pos = next_section.find('[');
if( next_pos == string::npos || next_pos != 0 )
{
continue;
}
break;
}
m_vctLine.erase( m_vctLine.begin() + del_begin, m_vctLine.begin()+i );
return true;
}
return false;
}
bool IniFile::RemoveKey( const string §ion, const string &key )
{
for( size_t i = 0;i < m_vctLine.size(); ++i )
{
string §ion_line = m_vctLine[i];
size_t sec_begin_pos = section_line.find('[');
if( sec_begin_pos == string::npos || sec_begin_pos != 0 )
{
continue;
}
size_t sec_end_pos = section_line.find( ']', sec_begin_pos );
if( sec_end_pos == string::npos )
{
continue;
}
if( section != Trim(section_line.substr(sec_begin_pos+1, sec_end_pos-sec_begin_pos-1)) )
{
continue;
}
//ÕÒµ½ÁËsection
for( ++i ; i < m_vctLine.size(); ++i )
{
string &key_line = m_vctLine[i];
size_t key_pos = key_line.find( key );
if( key_pos == string::npos || key_pos != 0 )
{
continue;
}
else
{
m_vctLine.erase( m_vctLine.begin() + i );
return true;
}
}
}
return false;
}
bool IniFile::WriteFile( void )
{
ofstream out(m_fileName.c_str());
for( size_t i = 0; i < m_vctLine.size(); ++i )
{
out<<m_vctLine[i]<<endl;
}
return true;
}
int IniFile::ReadInt( const string §ion, const string &key, int value )
{
string str = ReadString( section, key, "" );
if( "" == str )
{
return value;
}
istrstream in( str.c_str() );
int ret = 0;
in>>ret;
return ret;
}
bool IniFile::WriteInt( const string §ion, const string &key, int value )
{
ostrstream out;
out<<value;
//char buffer[64] = "";
//itoa( value, buffer, 10 );
return WriteString( section, key, out.str() );
}
string IniFile::LTrim( const string &str )
{
size_t pos = 0;
while( pos != str.size() )
{
if( ' ' == str[pos] )
{
++pos;
}
else
{
break;
}
}
return str.substr(pos);
}
string IniFile::RTrim( const string &str )
{
size_t pos = str.size() - 1;
while( pos >= 0 )
{
if(' ' == str[pos])
{
--pos;
}
else
{
break;
}
}
return str.substr( 0, pos + 1 );
}
string IniFile::Trim(const string &str)
{
return LTrim( RTrim(str) );
}
}
- 1
- 2
- 3
- 4
- 5
前往页