~chenluhua/Bond2.git

#include "stdafx.h"
#include "Model.h"
#include "Log.h"
#include "Common.h"
#include "ToolUnits.h"
 
// ³£Á¿
 
 
CModel* g_pModel = NULL;
void CALLBACK ModerTimerProc(HWND hWnd, UINT nMsg, UINT nTimerid, DWORD dwTime)
{
    if (g_pModel != NULL) {
        g_pModel->onTimer(nTimerid);
    }
}
 
 
CModel::CModel()
{
    m_pObservableEmitter = nullptr;
    m_pObservable = nullptr;
    m_nTimerID = 0;
}
 
 
CModel::~CModel()
{
    for (auto item : m_mapPlc) {
        delete item.second;
    }
    m_mapPlc.clear();
}
 
IObservable* CModel::getObservable()
{
    return m_pObservable;
}
 
void CModel::setWorkDir(const char* pszWorkDir)
{
    m_strWorkDir = pszWorkDir;
}
 
int CModel::init()
{
    CString strIniFile;
    CString strUnitId;
    strIniFile.Format(_T("%s\\Configuration.ini"), (LPTSTR)(LPCTSTR)m_strWorkDir);
    m_configuration.setFilepath((LPTSTR)(LPCTSTR)strIniFile);
    m_configuration.getUnitId(strUnitId);
 
 
    // Log
    CString strLogDir;
    strLogDir.Format(_T("%s\\BoounionPLCLog"), (LPTSTR)(LPCTSTR)m_strWorkDir);
    ::CreateDirectory(strLogDir, NULL);
    CLog::GetLog()->SetOnLogCallback([&](int level, const char* pszMessage) -> void {
        notifyTextAndInt(RX_CODE_LOG, pszMessage, level);
    });
    CLog::GetLog()->SetAutoAppendTimeString(TRUE);
    CLog::GetLog()->SetOutputTarget(OT_FILE);
    CLog::GetLog()->SetLogsDir(strLogDir);
    CLog::GetLog()->SetEquipmentId((LPTSTR)(LPCTSTR)strUnitId);
    LOGI("\r\n\r\n~~~ Prog Start! ~~~");
 
 
 
    m_pObservable = RX_AllocaObservable([&](IObservableEmitter* e) -> void {
        m_pObservableEmitter = e;            // ±£´æ·¢ÉäÆ÷
    });
 
 
    m_nTimerID = (int)SetTimer(NULL, 1, 1000, (TIMERPROC)ModerTimerProc);
    g_pModel = this;
 
 
    // »ñÈ¡ËùÓÐPLCÐÅÏ¢
    std::vector<PlcInfo> plcList;
    m_configuration.getAllPLCInfo(plcList);
    for (const auto& plc : plcList) {
        addPlc(plc.strName, plc.strIp, plc.nPort);
    }
 
    return 0;
}
 
int CModel::term()
{
    for (auto item : m_mapPlc) {
        item.second->term();
    }
    CLog::GetLog()->SetOnLogCallback(nullptr);
    return 0;
}
 
int CModel::notify(int code)
{
    /* code */
    if (m_pObservableEmitter != NULL) {
        IAny* pAny = RX_AllocaAny();
        if (pAny != NULL) {
            pAny->addRef();
            pAny->setCode(code);
            m_pObservableEmitter->onNext(pAny);
            pAny->release();
        }
    }
 
    return 1;
}
 
int CModel::notifyPtr(int code, void* ptr/* = NULL*/)
{
    /* code */
    if (m_pObservableEmitter != NULL) {
        IAny* pAny = RX_AllocaAny();
        if (pAny != NULL) {
            pAny->addRef();
            pAny->setCode(code);
            pAny->setPtrValue("ptr", ptr);
            m_pObservableEmitter->onNext(pAny);
            pAny->release();
        }
    }
 
    return 1;
}
 
int CModel::notifyObj(int code, IRxObject* pObj)
{
    /* code */
    if (m_pObservableEmitter != NULL) {
        IAny* pAny = RX_AllocaAny();
        if (pAny != NULL) {
            pAny->addRef();
            pAny->setCode(code);
            pAny->setObject("obj", pObj);
            m_pObservableEmitter->onNext(pAny);
            pAny->release();
        }
    }
 
    return 1;
}
 
int CModel::notifyObjAndPtr(int code, IRxObject* pObj, void* ptr)
{
    /* code */
    if (m_pObservableEmitter != NULL) {
        IAny* pAny = RX_AllocaAny();
        if (pAny != NULL) {
            pAny->addRef();
            pAny->setCode(code);
            pAny->setObject("obj", pObj);
            pAny->setPtrValue("ptr", ptr);
            m_pObservableEmitter->onNext(pAny);
            pAny->release();
        }
    }
 
    return 1;
}
 
int CModel::notifyInt(int code, int exCode)
{
    if (m_pObservableEmitter != NULL) {
        IAny* pAny = RX_AllocaAny();
        if (pAny != NULL) {
            pAny->addRef();
            pAny->setCode(code);
            pAny->setIntValue("exCode", exCode);
            m_pObservableEmitter->onNext(pAny);
            pAny->release();
        }
    }
 
    return 0;
}
 
int CModel::notifyInt2(int code, int exCode, int exCode2)
{
    if (m_pObservableEmitter != NULL) {
        IAny* pAny = RX_AllocaAny();
        if (pAny != NULL) {
            pAny->addRef();
            pAny->setCode(code);
            pAny->setIntValue("exCode", exCode);
            pAny->setIntValue("exCode2", exCode2);
            m_pObservableEmitter->onNext(pAny);
            pAny->release();
        }
    }
 
    return 0;
}
 
int CModel::notifyDouble(int code, double dValue)
{
    if (m_pObservableEmitter != NULL) {
        IAny* pAny = RX_AllocaAny();
        if (pAny != NULL) {
            pAny->addRef();
            pAny->setCode(code);
            pAny->setDoubleValue("value", dValue);
            m_pObservableEmitter->onNext(pAny);
            pAny->release();
        }
    }
 
    return 0;
}
 
int CModel::notifyObjAndInt(int code, IRxObject* pObj1, IRxObject* pObj2, int exCode)
{
    if (m_pObservableEmitter != NULL) {
        IAny* pAny = RX_AllocaAny();
        if (pAny != NULL) {
            pAny->addRef();
            pAny->setCode(code);
            if (pObj1 != nullptr) pAny->setObject("obj", pObj1);
            if (pObj2 != nullptr) pAny->setObject("obj2", pObj2);
            pAny->setIntValue("exCode", exCode);
            m_pObservableEmitter->onNext(pAny);
            pAny->release();
        }
    }
 
    return 0;
}
 
int CModel::notifyText(int code, const char* pszText)
{
    if (m_pObservableEmitter != NULL) {
        IAny* pAny = RX_AllocaAny();
        if (pAny != NULL) {
            pAny->addRef();
            pAny->setCode(code);
            pAny->setStringValue("text", pszText);
            m_pObservableEmitter->onNext(pAny);
            pAny->release();
        }
    }
 
    return 0;
}
 
int CModel::notifyTextAndInt(int code, const char* pszText, int exCode)
{
    if (m_pObservableEmitter != NULL) {
        IAny* pAny = RX_AllocaAny();
        if (pAny != NULL) {
            pAny->addRef();
            pAny->setCode(code);
            pAny->setStringValue("text", pszText);
            pAny->setIntValue("exCode", exCode);
            m_pObservableEmitter->onNext(pAny);
            pAny->release();
        }
    }
 
    return 0;
}
 
int CModel::notifyPtrAndInt(int code, void* ptr1, void* ptr2, int exCode)
{
    if (m_pObservableEmitter != NULL) {
        IAny* pAny = RX_AllocaAny();
        if (pAny != NULL) {
            pAny->addRef();
            pAny->setCode(code);
            pAny->setPtrValue("ptr", ptr1);
            pAny->setPtrValue("ptr1", ptr1);
            pAny->setPtrValue("ptr2", ptr2);
            pAny->setIntValue("exCode", exCode);
            m_pObservableEmitter->onNext(pAny);
            pAny->release();
        }
    }
 
    return 0;
}
 
int CModel::notifyMesMsg(int code, int stream, int function, const char* pszText)
{
    if (m_pObservableEmitter != NULL) {
        IAny* pAny = RX_AllocaAny();
        if (pAny != NULL) {
            pAny->addRef();
            pAny->setCode(code);
            pAny->setIntValue("stream", stream);
            pAny->setIntValue("function", function);
            pAny->setStringValue("text", pszText);
            m_pObservableEmitter->onNext(pAny);
            pAny->release();
        }
    }
 
    return 0;
}
 
void CModel::onTimer(UINT nTimerid)
{
    static int ii = 0;
    ii++;
 
 
    // ¼ì²âÊÇ·ñ¶Ï¿ª£¬ÖØÁ¬
    if (ii % 5 == 0) {
        for (auto item : m_mapPlc) {
            if (!item.second->isConnected()) {
                item.second->connect();
            }
        }
    }
 
 
    // ¶ÁÈ¡²úÄÜÐÅÏ¢
    CPLC* pPlc = getCurrentPlc();
    if (pPlc != nullptr) {
        pPlc->readPLCDataRegularly();
    }
}
 
std::map<std::string, CPLC*>& CModel::getPlcMap()
{
    return m_mapPlc;
}
 
int CModel::addPlc(const char* pszName, const char* pszIp, const unsigned int port)
{
    auto iter = m_mapPlc.find(pszName);
    if (iter != m_mapPlc.end()) return -1;
    CString strPlcDir;
    strPlcDir.Format(_T("%s\\Plcs\\%s"), (LPTSTR)(LPCTSTR)m_strWorkDir, pszName);
    CPLC* pPLC = new CPLC(pszName, pszIp, port);
    pPLC->setWorkDir((LPTSTR)(LPCTSTR)strPlcDir);
    PLCListener listener;
    listener.onStateChanged = [&](void* pFrom, int state) -> void {
        LOGD("PLC×´Ì¬¸Ä±ä£¬%d", state);
    };
    listener.onMonitorData = [&](void* pFrom, int id) -> void {
        LOGD("PLConMonitorData£¬%d", id);
    };
    listener.onAlarm = [&](void* pFrom, CAlarm* pAlarm, int flag) -> void {
        LOGE("onAlarm£¬%d %s", pAlarm->getId(), flag != 0 ? "ON" : "Off");
        if (flag == 1) {
            pAlarm->addRef();
            notifyObjAndPtr(RX_CODE_ALARM_ON, pAlarm, pFrom);
            pAlarm->release();
        }
        else {
            pAlarm->addRef();
            notifyObjAndPtr(RX_CODE_ALARM_OFF, pAlarm, pFrom);
            pAlarm->release();
        }
 
    };
    pPLC->setListener(listener);
    pPLC->init();
    m_mapPlc[pszName] = pPLC;
 
    CString strDir;
    strDir.Format(_T("%s\\PLCs\\%s"), (LPTSTR)(LPCTSTR)m_strWorkDir, (LPTSTR)(LPCTSTR)pszName);
    CToolUnits::createDir(strDir);
    m_configuration.addPLC(pszName, pszIp, port);
 
    notifyPtr(RX_CODE_ADD_PLC, pPLC);
    return 0;
}
 
int CModel::removePlc(const char* pszName)
{
    auto iter = m_mapPlc.find(pszName);
    if (iter == m_mapPlc.end()) return -1;
 
    CString strDir;
    strDir.Format(_T("%s\\PLCs\\%s"), (LPTSTR)(LPCTSTR)m_strWorkDir, (LPTSTR)(LPCTSTR)pszName);
    CToolUnits::deleteDir(strDir);
    m_configuration.removePLC(pszName);
 
    notifyPtr(RX_CODE_REMOVE_PLC, iter->second);
    delete iter->second;
    m_mapPlc.erase(iter);
 
    return 0;
}
 
void CModel::setCurrentPlc(CPLC* pPlc)
{
    if (pPlc != nullptr) {
        m_strCurrPlc = pPlc->getName();
    } else {
        m_strCurrPlc = "";
    }
}
 
CPLC* CModel::getCurrentPlc()
{
    auto item = m_mapPlc.find(m_strCurrPlc);
    if (item != m_mapPlc.end()) {
        return item->second;
    }
 
    return nullptr;
}