ColumnDataArray.cpp

// ColumnDataArray.cpp: CColumnDataArray クラスのインプリメンテーション
//

#include "stdafx.h"
#include "nexpcsv.h"
#include "ColumnDataArray.h"

#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif

// 構築/消滅

CColumnDataArray::CColumnDataArray()
{
        m_szSplit = NULL;
        m_nQuote = 0;
        m_szAltSplit = NULL;
        m_szAltCrlf = NULL;
        m_szListSplit = NULL;
        m_session = NULL;
}

CColumnDataArray::~CColumnDataArray()
{

}

void CColumnDataArray::Copy(const CColumnDataArray &org)
{
        CArray<CColumnData, CColumnData&>::Copy(org);
        if (!org.m_strViewName.IsNull()) m_strViewName = org.m_strViewName;
}

void CColumnDataArray::SetView(LNViewFolder &view)
{
        if (!view.IsOpen()) view.Open();
        
        m_strViewName = view.GetName();
        LNINT colsize = view.GetColumnCount();
        for (LNINT n=0; n<colsize; n++) {
                LNVFColumn column;
                view.GetColumn(n, &column);
                CColumnData data;
                data.SetColumn(column);
                Add( data );
        }
        if (view.IsOpen()) view.Close();
}

void CColumnDataArray::SetOption(const LNNotesSession *session, LPCTSTR szListSplit, LPCTSTR szSplit, const int nQuote, LPCTSTR szAltSplit, LPCTSTR szAltCrlf)
{
        m_szSplit = szSplit;
        m_nQuote = nQuote;
        m_szAltSplit = szAltSplit;
        m_szAltCrlf = szAltCrlf;
        m_szListSplit = szListSplit;
        m_session = session;
}

LNString CColumnDataArray::ToCSVHeader()
{
        LNString strTemp;

        int nn = 0;
        for (int n=0; n<GetSize(); n++) {
                CColumnData &column = ElementAt(n);
                if (column.IsNull()) continue;
                if (!column.IsOutput()) continue;

                if (nn++>0) strTemp << m_szSplit;
                strTemp << ToCSVText(column.GetTitle(), m_nQuote, m_szSplit, m_szAltSplit, m_szAltCrlf);
        }
        return strTemp;
}

LNString CColumnDataArray::ToCSVData(LNVFEntry &entry, LNNote &note)
{
        LNString strTemp;

        // CSV1行の出力
        try {
                int nn = 0;
                for (int n=0; n<GetSize(); n++) {
                        // CSV1列の出力
                        CColumnData &column = ElementAt(n);
                        if (column.IsNull()) continue;
                        if (!column.IsOutput()) continue;

                        LNString &strText = LNString();
                        if (column.IsViewColumn()) {
                                strText = column.ToString(entry, m_session, m_szListSplit);
                        } else if (column.IsFormula()) {
                                if (!note.IsOpen()) note.Open(LNNOTEOPENFLAGS_DEFAULT | LNNOTEOPENFLAGS_NO_OBJECTS);
                                strText = column.ToString(note, m_session, m_szListSplit);
                        }
                        if (nn++>0) strTemp << m_szSplit;
                        strTemp << ToCSVText(strText, m_nQuote, m_szSplit, m_szAltSplit, m_szAltCrlf);
                }
        } catch (LNSTATUS) {
        }
        if (note.IsOpen()) note.Close();
        return strTemp;
}

LNString CColumnDataArray::ToCSVText(const LNString &strText, int nQuote, LPCTSTR szSplit, LPCTSTR szAltSplit, LPCTSTR szAltCrlf)
{
        LPCTSTR szCrlf = _T("\n");
        LPCTSTR szQuote = _T("\"");
        LNString strTemp(strText);

        // 代替分離記号への置換え
        if (szAltSplit != NULL) ReplaceAll(strTemp, szSplit, szAltSplit);

        // 代替改行記号への置換え
        if (szAltCrlf != NULL) ReplaceAll(strTemp, szCrlf, szAltCrlf);

        BOOL bQuote = FALSE;
        switch (nQuote) {
        case 0: // 囲み記号の処理(自動)
                {
                        LNINT loc;
                        if ( ( LNWARN_NOT_FOUND != strTemp.Find(szSplit, 0, &loc) )
                                || ( LNWARN_NOT_FOUND != strTemp.Find(szCrlf, 0, &loc) )
                                || ( LNWARN_NOT_FOUND != strTemp.Find(szQuote, 0, &loc) ) ) {
                                bQuote = TRUE;
                        }
                        break;
                }
        case 1: // 囲み記号の処理(する)
                bQuote = TRUE;
                break;
        }

        // データ内の改行コードを置き換える
        ReplaceAll(strTemp, szCrlf, "\x0A");

        // 囲み記号の処理
        if (bQuote) {
                ReplaceAll(strTemp, szQuote, "\"\"");
                strTemp.Insert(0, szQuote);
                strTemp.Append(szQuote);
        }

        return strTemp;
}

#define FILE_VERSION                    _T("nExpCsv1.5")
#define COLUMNTYPE_VIEWCOLUMN   _T("ViewColumn")
#define COLUMNTYPE_FORMULA              _T("Formula")

BOOL CColumnDataArray::Save(LPCTSTR szFilepath)
{
        LPCTSTR szSplit = _T(",");

        int nOutputMode = ios::out | ios::binary;
        ofstream fofs(szFilepath, nOutputMode);
        if ( !fofs ) return FALSE;

        fofs << FILE_VERSION << szSplit;
        fofs << (LPCTSTR)ToCSVText(m_strViewName, 0, szSplit, NULL, NULL) << endl;

        int nn = 0;
        for (int n=0; n<GetSize(); n++) {
                // CSV1列の出力
                CColumnData &column = ElementAt(n);
                if (column.IsNull()) continue;

                LNString strTitle = column.GetTitle();
                fofs << (LPCTSTR)ToCSVText(strTitle, 0, szSplit, NULL, NULL) << szSplit;
                if (column.IsViewColumn()) {
                        fofs << COLUMNTYPE_VIEWCOLUMN << szSplit;
                        fofs << column.GetViewColumnNumber() << szSplit;
                } else {
                        fofs << COLUMNTYPE_FORMULA << szSplit;
                        LNString strFormula = column.GetFormula();
                        fofs << (LPCTSTR)ToCSVText(strFormula, 0, szSplit, NULL, NULL) << szSplit;
                }
                LPCTSTR szOutput = column.IsOutput() ? _T("TRUE"):_T("FALSE");
                fofs << szOutput << endl;
        }
        fofs.close();
        return FALSE;
}

BOOL CColumnDataArray::Load(LPCTSTR szFilepath)
{
        ifstream ifs(szFilepath);
        if ( !ifs ) {
                CString strMsg;
                strMsg.Format(_T("ファイルを開けませんでした。\nFile=%s"), szFilepath);
                AfxMessageBox(strMsg);
                return FALSE;
        }

        CCsvFile csv(ifs);
        csv.settrim();
        if (!csv.getline() || stricmp(csv[0], FILE_VERSION) != 0) {
                CString strMsg;
                strMsg.Format(_T("CSVエクスポート定義ファイルではありません。\nFile=%s"), szFilepath);
                AfxMessageBox(strMsg);
                return FALSE;
        }

        BOOL bViewCheck = TRUE;
        if ( stricmp(csv[1], m_strViewName) != 0) {
                CString strMsg;
                strMsg.Format(_T("異なるビュー\"%s\"で保存されたファイルです。\n保存している式の読み込みを継続しますか?"), csv[1]);
                int nRet = AfxMessageBox(strMsg, MB_YESNO);
                if (nRet == IDNO) return FALSE;
                bViewCheck = FALSE;
        }

        CColumnDataArray arrData;
        for (int n=0; n<GetSize(); n++) {
                CColumnData &column = ElementAt(n);
                if (!column.IsViewColumn()) continue;
                LNINT nCol = column.GetViewColumnNumber();
                if (nCol < 0) continue;
                arrData.InsertAt(nCol, column);
        }

        if (bViewCheck) {
                RemoveAll();
        } else {
                Copy(arrData);
                arrData.RemoveAll();
        }
        
        while (csv.getline())
        {
                CColumnData column;
                column.SetTitle(csv[0]);
                if (bViewCheck && stricmp(csv[1], COLUMNTYPE_VIEWCOLUMN) == 0) {
                        int n = atoi(csv[2]);
                        if (n < arrData.GetSize() && arrData[n].IsViewColumn()) {
                                column = arrData[n];
                                arrData[n] = CColumnData();     // クリア
                        }
                } else if (stricmp(csv[1], COLUMNTYPE_FORMULA) == 0) {
                        try {
                                column.SetColumn(csv[2]);
                        } catch (LNSTATUS) {
                        }
                }
                if (!column.IsNull()) Add(column);
        }
        ifs.close();

        // 利用されなかったビュー列情報を追加
        for ( n=0; n<arrData.GetSize(); n++) {
                CColumnData &column = arrData.ElementAt(n);
                if (!column.IsViewColumn()) continue;
//              column.SetOutput(FALSE);
                Add(column);
        }
        
        return TRUE;
}

---