98 lines
3.2 KiB
C++
98 lines
3.2 KiB
C++
#include "stdafx.h"
|
||
#include "CSVWriter.h"
|
||
#include "StringUtil.h"
|
||
|
||
|
||
// 초반 네이밍 칼럼 설정
|
||
// 칼럼 추가는 AddColoum 으로 설정
|
||
// 필드 추가는 AddField로 설정
|
||
|
||
|
||
CSVWriter::CSVWriter()
|
||
{
|
||
m_stream = NULL;
|
||
m_bEnableWrite = false;
|
||
}
|
||
|
||
CSVWriter::~CSVWriter()
|
||
{
|
||
}
|
||
|
||
bool CSVWriter::Initialize( const wchar_t *FileName , std::wstring strColumname )
|
||
{
|
||
#ifndef _FINAL_BUILD
|
||
_wsetlocale(LC_ALL,L"kor"); // 로케일 지정을 해주어야 국가별 언어가 정상적으로 표현됩니다.
|
||
#endif
|
||
|
||
m_strFileName = FileName;
|
||
m_strColumname = strColumname;
|
||
|
||
return CheckExistFile();
|
||
}
|
||
|
||
bool CSVWriter::CheckExistFile()
|
||
{
|
||
if( _wfopen_s( &m_stream, m_strFileName.c_str(), L"r" ) != 0 )
|
||
{
|
||
if( _wfopen_s( &m_stream, m_strFileName.c_str(), L"w" ) != 0 )
|
||
{
|
||
return false;
|
||
}
|
||
else
|
||
{
|
||
fclose(m_stream);
|
||
WriteColumnName();
|
||
}
|
||
}
|
||
else
|
||
{
|
||
fclose(m_stream);
|
||
}
|
||
|
||
return true;
|
||
}
|
||
|
||
void CSVWriter::WriteColumnName()
|
||
{
|
||
StartWrite();
|
||
std::vector<std::wstring> tokens;
|
||
TokenizeW( m_strColumname , tokens, L";" );
|
||
for ( int j = 0 ; j < (int)tokens.size() ; j++ )
|
||
AddColumn(tokens[j].c_str());
|
||
AddField();
|
||
EndWrite();
|
||
}
|
||
|
||
void CSVWriter::StartWrite()
|
||
{
|
||
if( _wfopen_s( &m_stream, m_strFileName.c_str(), L"a" ) != 0 )
|
||
return;
|
||
|
||
m_bEnableWrite = true;
|
||
}
|
||
void CSVWriter::EndWrite()
|
||
{
|
||
if(!m_bEnableWrite)
|
||
return;
|
||
|
||
m_bEnableWrite = false;
|
||
fclose(m_stream);
|
||
}
|
||
|
||
|
||
void CSVWriter::AddColumn(const wchar_t *str)
|
||
{
|
||
if(m_bEnableWrite)
|
||
fwprintf_s( m_stream, L"%s,", str );
|
||
}
|
||
|
||
|
||
|
||
void CSVWriter::AddField( int nCount )
|
||
{
|
||
if(m_bEnableWrite)
|
||
for( int i = 0; i < nCount; i ++ )
|
||
fwprintf_s( m_stream, L"\n" );
|
||
}
|
||
|
||
|