CMarkup 解析XML

#include "StdAfx.h"
#include "CIMFile.h"
#include "Markup.h"

#ifdef  UNICODE 
    #define _tcout(x) std::wcout << x;
#else
    #define _tcout(x) std::cout << x;
#endif

char* Wide2Byte(const TCHAR* pWide)
{
    char *pBytes = NULL;
    int nLen = WideCharToMultiByte(CP_ACP,0,pWide,-1,NULL,0,NULL,NULL);
    // get need convert length
    int i = (int)wcslen(pWide)*sizeof(TCHAR);
    // get need new memory size
    pBytes = new char[i+1];
    if (!pBytes) // if fail then new again
    {
            pBytes = new char[i+1];
    }
    WideCharToMultiByte(CP_ACP,0,pWide,-1,pBytes,nLen,NULL,NULL);
    pBytes[nLen]='\0';

    return pBytes;
}

TCHAR* Byte2Wide(const char* pBytes)
{
    TCHAR *pWide = NULL;

    int nLen = MultiByteToWideChar(CP_ACP,0,pBytes,strlen(pBytes),NULL,0);
    pWide = new wchar_t[nLen+1];    
    if (!pWide) // if fail then new again
    {
        pWide = new TCHAR[nLen+1];
    }
    MultiByteToWideChar(CP_ACP,0,pBytes,strlen(pBytes),pWide,nLen); 
    pWide[nLen]='\0';

    return pWide;
}

void _FindDirFiles(tstring dirPath, tstring suffix, vector<tstring> &files)
{
    TCHAR *_dirPath = new TCHAR[MAX_PATH];
    int len = dirPath.length() + 4;
    _tcscpy_s(_dirPath, len, dirPath.c_str());
    _tcscat_s(_dirPath, len, _T("*.*"));

    _finddata_t file;

    char *pdirPath = Wide2Byte(_dirPath);

    long lf;
    if((lf = _findfirst(pdirPath, &file))==-1l)
        cout<<"文件没有找到!\n";
    else
    {
        files.clear();
        while( _findnext( lf, &file ) == 0 )
        {
            TCHAR *pName = Byte2Wide(file.name);
            tstring str(pName);
            int te = str.find(suffix);
            if(te != -1){
                //cout<<file.name<<endl;
                files.push_back(str);
            }
            if(file.attrib == _A_NORMAL)cout<<"  普通文件  ";
            else if(file.attrib == _A_RDONLY)cout<<"  只读文件  ";
            else if(file.attrib == _A_HIDDEN )cout<<"  隐藏文件  ";
            else if(file.attrib == _A_SYSTEM )cout<<"  系统文件  ";
            else if(file.attrib == _A_SUBDIR){

                if(strcmp(file.name,"..")==0 || strcmp(file.name,".svn")==0){}
            }
            
            delete[] pName;
        }
    }
    delete[] pdirPath;
    delete[] _dirPath;

    _findclose(lf);

}

void FindDirFiles(string dirPath, string suffix, vector<string> &files)
{
    char *pchar = new char[MAX_PATH];
    int len = dirPath.length() + 4;
    strcpy_s(pchar, len, dirPath.c_str());
    strcat_s(pchar, len, "*.*");

    _finddata_t file;

    long lf;
    if((lf = _findfirst(pchar, &file))==-1l)
        cout<<"文件没有找到!\n";
    else
    {
        files.clear();
        while( _findnext( lf, &file ) == 0 )
        {

            string str(file.name);
            int te = str.find(suffix.c_str());
            if(te != -1){
                //cout<<file.name<<endl;
                files.push_back(str);
            }
            if(file.attrib == _A_NORMAL)cout<<"  普通文件  ";
            else if(file.attrib == _A_RDONLY)cout<<"  只读文件  ";
            else if(file.attrib == _A_HIDDEN )cout<<"  隐藏文件  ";
            else if(file.attrib == _A_SYSTEM )cout<<"  系统文件  ";
            else if(file.attrib == _A_SUBDIR){

                if(strcmp(file.name,"..")==0 || strcmp(file.name,".svn")==0){}
            }
        }
    }
    delete[] pchar;

    _findclose(lf);

}

CCIMFile::CCIMFile(void) :
    m_arSubstation(10000)
    , m_arPDLoadSWDet(10000)
{
    m_pXML = new CMarkup();
}

CCIMFile::~CCIMFile(void)
{
    CleanMem();
    DELETE_POINTER(m_pXML);
}

void CCIMFile::CleanMem()
{
    m_arSubstation.ReleaseAll();
    m_arSWGear.ReleaseAll();
    m_arPDRoom.ReleaseAll();
    m_arPDSite.ReleaseAll();
    m_arPDHVLine.ReleaseAll();
    m_arPDLeadLineDet.ReleaseAll();
    m_arPDMainBusDet.ReleaseAll();
    m_arPDBreaker.ReleaseAll();
    m_arPDSwitch.ReleaseAll();
    m_arPDSubSWDet.ReleaseAll();
    m_arPDLoadSW.ReleaseAll();
    m_arPDLoadSWDet.ReleaseAll();
    m_arPDFusePR.ReleaseAll();
    m_arPDFuse.ReleaseAll();
    m_arPDTran.ReleaseAll();
    m_arPDXFmrDet.ReleaseAll();

    //DeleteMap(&mpTbl);
    mpTbl.clear();

}

BOOL CCIMFile::ParseDir(LPCTSTR strDir)
{
    CleanMem();

    tstring suffix = _T(".xml");
    vector<tstring> files;
    _FindDirFiles(strDir, suffix, files);
    int num = files.size();
    cout << "    文件总数：" << num << endl << endl;
    
    for (int i = 0; i < (int)files.size(); i++)
    {
        tstring filePath = strDir + files[i];
        int nLen = files[i].length();
        TCHAR *strTemp = new TCHAR[nLen + 1];
        _tcscpy_s(strTemp, nLen + 1, files[i].c_str());
        wcstok(strTemp, _T("_"));
        CIM_NODE::nParseTblId = _ttoi(wcstok(NULL, _T("_")));
        DELETE_POINTER(strTemp);
        
        wcout << "  ["<< i << "/" << num << "] 正在解析：" << files[i].c_str();
        ParseFile(filePath.c_str());
    }

    wcout << "    解析完毕！" << endl;

    return TRUE;
}

BOOL CCIMFile::ParseFile(LPCTSTR strFilePath)
{
    //string path = "D:\\iStudio\\svn\\SZ_REVA\\PowerES\\prog\\data\\GISCIM\\";

    int numsOfPSR = 0;

    if(!m_pXML->Load((MCD_CSTR)strFilePath))
        return FALSE;

    switch (CIM_NODE::nParseTblId)
    {
    case 76:    // CIM_DIST_LOADSWITCHDET
        {
            m_pXML->ResetMainPos();
            CIM_NODE::strParseTblName = CIM_DIST_LOADSWITCHDET::strTblName;
            while (m_pXML->FindChildElem(MCD_CSTR(_T("cim:") + CIM_DIST_LOADSWITCHDET::strTblName)))
            {
                CIM_DIST_LOADSWITCHDET *pNode = m_arPDLoadSWDet.AddItem();
                pNode->Parse(m_pXML);
            
                mpNode.insert(pair<int, CIM_NODE *>(pNode->nID, pNode));
                mpTbl.insert(pair<int, CIM_NODE *>(CIM_NODE::nParseTblId, pNode));

                numsOfPSR++;
            }

        }
        break;
    }

    wcout << "\t 新增解析对象个数：" << numsOfPSR << "  ,共 " << mpNode.size() << "个" << endl << endl;

    return TRUE;
}