zhipuzi_pay_plugin/zhipuzi_pay_plugin/wnd/CValueWnd.cpp

#include "../pch/pch.h"
#include "CValueWnd.h"

#include "../tool/CComHelper.h"

#include "../tool/CSerialPort.h"

void CValueWnd::Notify(TNotifyUI& msg)
{
    if(msg.sType == _T("click"))
    {
        if(msg.pSender->GetName() == _T("closebtn"))
        {
            PostQuitMessage(0);
            return;
        }
    }
}

void CValueWnd::Init()
{
    //初始化窗口位置
    InitWndPos();

    //添加托盘图标
    AddTrayIcon();

    //初始化菜单窗口
    pMenu = new CMenuWnd();
    pMenu->Create(m_hWnd, L"智铺子收银插件菜单", WS_POPUP, WS_EX_TOOLWINDOW);
    pMenu->ShowWindow(false);

    //再安装钩子
    //BOOL ret = InstallHook();

    //开始启动监听
    RestartWatch();

    /*
        std::thread(&CValueWnd::WatchWnd, this).detach();

        //开始测试ocr
        tesseract::TessBaseAPI tess;

        if(tess.Init("./tessdata", "eng"))
        {
            std::cout << "OCRTesseract: Could not initialize tesseract." << std::endl;
            return ;
        }

        // setup
        tess.SetPageSegMode(tesseract::PageSegMode::PSM_AUTO);
        tess.SetVariable("save_best_choices", "T");

        // read image
        auto pixs = pixRead("D:\\3.png");
        if(!pixs)
        {
            std::cout << "Cannot open input file: " << "D:\1.png" << std::endl;
            return ;
        }

        // recognize
        tess.SetImage(pixs);
        tess.Recognize(0);

        // get result and delete[] returned char* string
        std::string aaa = tess.GetUTF8Text();

        // cleanup
        tess.Clear();
        pixDestroy(&pixs);*/
}

void CValueWnd::RestartWatch()
{
	std::thread(&CValueWnd::StartWatchWork, this).detach();
}

/*
 *根据当前数据库的设置，重启监听
 **/
void CValueWnd::StartWatchWork()
{
	//先暂停已有的监控
	m_isWatchWork = false;

	//等待1秒，这1秒就是等已有的监控线程自动退出
	Sleep(1000);

	//然后开始工作
	m_isWatchWork = true;

    std::string system_setting_jinezhuaqu_setting_type = CSetting::GetParam("system_setting_jinezhuaqu_setting_type");

    m_nWatchType = atoi(system_setting_jinezhuaqu_setting_type.c_str());

    if(m_nWatchType == 1)
    {
        //更新监听的端口号
        std::string system_setting_jinezhuaqu_setting_chuankou_num = CSetting::GetParam("system_setting_jinezhuaqu_setting_chuankou_num");

        if(system_setting_jinezhuaqu_setting_chuankou_num == "")
        {
            //参数错误
            return;
        }

        std::wstring ws_system_setting_jinezhuaqu_setting_chuankou_num = CLewaimaiString::UTF8ToUnicode(system_setting_jinezhuaqu_setting_chuankou_num);

        //设置窗口句柄
        ::SetHwnd(m_hWnd);

        //设置串口端口号
        std::transform(system_setting_jinezhuaqu_setting_chuankou_num.begin(), system_setting_jinezhuaqu_setting_chuankou_num.end(), system_setting_jinezhuaqu_setting_chuankou_num.begin(), ::toupper);
        std::transform(ws_system_setting_jinezhuaqu_setting_chuankou_num.begin(), ws_system_setting_jinezhuaqu_setting_chuankou_num.end(), ws_system_setting_jinezhuaqu_setting_chuankou_num.begin(), ::toupper);

        ::SetComNum(system_setting_jinezhuaqu_setting_chuankou_num.c_str(), ws_system_setting_jinezhuaqu_setting_chuankou_num.c_str(), system_setting_jinezhuaqu_setting_chuankou_num.size());
    }
    else if(m_nWatchType == 2)
    {
        //虚拟串口的模式
        std::thread(&CValueWnd::ReadXunichuankouValue, this).detach();
    }
    else if(m_nWatchType == 3)
    {
        //窗口模式
        std::thread(&CValueWnd::WatchWnd, this).detach();
    }
}

/*
 *读取客显金额的线程，一旦m_nWatchType不为2，那么就直接退出这个线程，下次设置的时候再开启
 **/
void CValueWnd::ReadXunichuankouValue()
{
    //读取虚拟串口的模式
    std::string system_setting_jinezhuaqu_setting_xunichuankou_num_2 = CSetting::GetParam("system_setting_jinezhuaqu_setting_xunichuankou_num_2");
    std::string system_setting_jinezhuaqu_setting_xunichuankou_botelv = CSetting::GetParam("system_setting_jinezhuaqu_setting_xunichuankou_botelv");

    CSerialPort serial(CLewaimaiString::UTF8ToUnicode(system_setting_jinezhuaqu_setting_xunichuankou_num_2), atoi(system_setting_jinezhuaqu_setting_xunichuankou_botelv.c_str()));

    bool ret = serial.openComm();

    if(!ret)
    {
        LOG_INFO("open com failed!");
        return;
    }

    char a[1024] = { 0 };

	//检查监控模式和波特率，如果变了就关闭掉
    while(m_nWatchType == 2 && m_isWatchWork)
    {
        memset(a, 0, 1024);

        //开始读取串口的数据
        DWORD nReaded = 0;

        serial.readFromComm(a, 1024, &nReaded);

        if(nReaded > 0)
        {
            UpdateKexian(a, nReaded);
        }

        Sleep(200);
    }

    serial.closeComm();
}

void CValueWnd::WatchWnd()
{
	std::string system_setting_jinezhuaqu_setting_chuangkou_process_name = CSetting::GetParam("system_setting_jinezhuaqu_setting_chuangkou_process_name");
	std::string system_setting_jinezhuaqu_setting_chuangkou_title = CSetting::GetParam("system_setting_jinezhuaqu_setting_chuangkou_title");
	std::string system_setting_jinezhuaqu_setting_chuangkou_zorder = CSetting::GetParam("system_setting_jinezhuaqu_setting_chuangkou_zorder");

	//如果模式变了，或者参数变了，就退出（因为会重新启动一个新的监控）
    while(m_nWatchType == 3 && m_isWatchWork)
    {
        if(m_watchWnd != NULL)
        {
            char szTitle[MAX_PATH] = { 0 };

            LPSTR lpWindowName = szTitle;

            std::string value;

            int ret = ::SendMessageA(m_watchWnd, WM_GETTEXT, (WPARAM)(MAX_PATH), (LPARAM)lpWindowName);

            if(ret == 0)
            {
                //说明m_watchWnd已经失效了
                value = "0.00";

                m_watchWnd = NULL;
            }
            else
            {
                value = lpWindowName;
            }

            if(value != m_watchValue)
            {
                m_watchValue = value;

                SendMessage(WM_WATCH_WND_UPDATE, NULL, NULL);
            }
        }
        else
        {
            //获得进程名字
            std::wstring processName = CLewaimaiString::UTF8ToUnicode(system_setting_jinezhuaqu_setting_chuangkou_process_name);

            //根据进程名字，找到对应的主窗口
            HWND mainWnd = GetWndHwnd(processName.c_str(), system_setting_jinezhuaqu_setting_chuangkou_title);

            if(mainWnd ==  NULL)
            {
                //说明这个进程还没有启动
                m_watchWnd = NULL;
            }
            else
            {
                //根据Z序查找目标窗口               
                ValueWnd newWnd;
                newWnd.wndTitle = system_setting_jinezhuaqu_setting_chuangkou_title;
                newWnd.TopWnd = mainWnd;
                newWnd.z_order = atoi(system_setting_jinezhuaqu_setting_chuangkou_zorder.c_str());
                newWnd.processName = processName;
                newWnd.selfWnd = NULL;

                g_zorder = 0;

                EnumChildWindows(mainWnd, EnumMainwndChildProc, (LPARAM)(&newWnd));

                if(newWnd.selfWnd != NULL)
                {
                    //找到了
                    m_watchWnd = newWnd.selfWnd;
                }
            }
        }

        Sleep(200);
    }
}

LRESULT CValueWnd::OnCreate(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
    LONG styleValue = ::GetWindowLong(*this, GWL_STYLE);
    styleValue &= ~WS_CAPTION;
    ::SetWindowLong(*this, GWL_STYLE, styleValue | WS_CLIPSIBLINGS | WS_CLIPCHILDREN);

    // 把自己的窗口句柄与窗口绘制管理器挂接在一起
    m_pm.Init(m_hWnd);

    m_pm.AddPreMessageFilter(this);

    CDialogBuilder builder;

    CControlUI* pRoot = builder.Create(_T("value.xml"), (UINT)0, NULL, &m_pm);
    ASSERT(pRoot && "Failed to parse XML");

    // 把这些控件绘制到本窗口上
    m_pm.AttachDialog(pRoot);

    // 把自己加入到CPaintManagerUI的m_aNotifiers数组中，用于处理Notify函数
    //m_pm.AddNotifier(this);

    //进行一些初始化工作
    Init();

    return 0;
}

LRESULT CValueWnd::OnClose(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
    bHandled = FALSE;
    return 0;
}

/*
 *这个是窗口被销毁的时候调用的
 **/
LRESULT CValueWnd::OnDestroy(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
    bHandled = FALSE;
    return 0;
}

LRESULT CValueWnd::OnNcActivate(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
    if(::IsIconic(*this))
    {
        bHandled = FALSE;
    }

    return (wParam == 0) ? TRUE : FALSE;
}

LRESULT CValueWnd::OnNcCalcSize(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
    return 0;
}

LRESULT CValueWnd::OnNcPaint(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
    //在这里设置焦点才有用
    CEditUI* pAccountEdit = static_cast<CEditUI*>(m_pm.FindControl(_T("accountedit")));

    if(pAccountEdit)
    {
        pAccountEdit->SetFocus();
    }

    return 0;
}

LRESULT CValueWnd::OnNcHitTest(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
    POINT pt;
    pt.x = GET_X_LPARAM(lParam);
    pt.y = GET_Y_LPARAM(lParam);
    ::ScreenToClient(*this, &pt);

    RECT rcClient;
    ::GetClientRect(*this, &rcClient);

    RECT rcCaption = m_pm.GetCaptionRect();

    if(pt.x >= rcClient.left + rcCaption.left && pt.x < rcClient.right - rcCaption.right \
            && pt.y >= rcCaption.top && pt.y < rcCaption.bottom)
    {
        CControlUI* pControl = static_cast<CControlUI*>(m_pm.FindControl(pt));

        if(pControl && _tcscmp(pControl->GetClass(), DUI_CTR_BUTTON) != 0)
        {
            return HTCAPTION;
        }
    }

    return HTCLIENT;
}

LRESULT CValueWnd::OnLButtonDown(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
    /*
    	//获取鼠标点击位置
    	int xPos = LOWORD(lParam);
    	int yPos = HIWORD(lParam);

    	//將位置转换为屏幕坐标
    	POINT pt = { xPos,yPos };
    	ClientToScreen(m_hWnd, &pt);

    	RECT menuWndRect = { 0 };
    	::GetWindowRect(pMenu->GetHWND(), &menuWndRect);

    	if (pt.x <= menuWndRect.left || pt.x >= menuWndRect.right
    		|| pt.y <= menuWndRect.top || pt.y >= menuWndRect.bottom)
    	{
    		pMenu->ShowWindow(false);
    	}

    	bHandled = FALSE;
    	return 0;*/

    bHandled = FALSE;
    return 0;
}

LRESULT CValueWnd::OnRButtonDown(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
    /*
    	int xPos = LOWORD(lParam);
    	int yPos = HIWORD(lParam);

    	POINT pt = { xPos,yPos };

    	//可能是因为整个窗口都是标题区域，这里不需要做转换
    	//ClientToScreen(m_hWnd, &pt);

    	pMenu->ShowWindow(true);

    	::SetWindowPos(pMenu->GetHWND(), NULL, pt.x, pt.y, 0, 0, SWP_NOSIZE | SWP_NOACTIVATE);

    	bHandled = FALSE;
    	return 0;*/

    ShowMenu();

    bHandled = true;
    return 0;
}

LRESULT CValueWnd::OnKexianUpdate(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
    //如果当前不是串口模式，直接忽略
    if(m_nWatchType != 1)
    {
        bHandled = TRUE;
        return true;
    }

    //收到了客显数据消息，读取客显数据
    char data[MAX_DATA_LENGTH] = { 0 };
    int data_length = 0;

    HookGetData(data, &data_length);

    if(data_length == 0)
    {
        bHandled = TRUE;
        return true;
    }

    UpdateKexian(data, data_length);

    bHandled = TRUE;
    return true;
}

LRESULT CValueWnd::OnWatchWndUpdate(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
    CLabelUI* valueLabel = static_cast<CLabelUI*>(m_pm.FindControl(_T("value")));

    std::string money_show = CLewaimaiString::DoubleToString(atof(m_watchValue.c_str()), 2);
    valueLabel->SetText(CLewaimaiString::UTF8ToUnicode(money_show).c_str());

    return TRUE;
}

LRESULT CValueWnd::HandleMessage(UINT uMsg, WPARAM wParam, LPARAM lParam)
{
    LRESULT lRes = 0;
    BOOL bHandled = TRUE;

    switch(uMsg)
    {
    case WM_CREATE:
        lRes = OnCreate(uMsg, wParam, lParam, bHandled);
        break;

    case WM_CLOSE:
        lRes = OnClose(uMsg, wParam, lParam, bHandled);
        break;

    case WM_DESTROY:
        lRes = OnDestroy(uMsg, wParam, lParam, bHandled);
        break;

    case WM_NCACTIVATE:
        lRes = OnNcActivate(uMsg, wParam, lParam, bHandled);
        break;

    case WM_NCLBUTTONDOWN:
        lRes = OnLButtonDown(uMsg, wParam, lParam, bHandled);
        break;

    case WM_NCRBUTTONDOWN:
        lRes = OnRButtonDown(uMsg, wParam, lParam, bHandled);
        break;

    case WM_NCCALCSIZE:
        lRes = OnNcCalcSize(uMsg, wParam, lParam, bHandled);
        break;

    case WM_NCPAINT:
        lRes = OnNcPaint(uMsg, wParam, lParam, bHandled);
        break;

    case WM_NCHITTEST:
        lRes = OnNcHitTest(uMsg, wParam, lParam, bHandled);
        break;

    case WM_SHOWTASK:
        lRes = OnTrayIcon(uMsg, wParam, lParam, bHandled);
        break;

    case WM_HOOK_MESSAGE:
        lRes = OnKexianUpdate(uMsg, wParam, lParam, bHandled);
        break;

    case WM_WATCH_WND_UPDATE:
        lRes = OnWatchWndUpdate(uMsg, wParam, lParam, bHandled);
        break;

    default:
        bHandled = FALSE;
    }

    if(bHandled)
    {
        return lRes;
    }

    if(m_pm.MessageHandler(uMsg, wParam, lParam, lRes))
    {
        return lRes;
    }

    return CWindowWnd::HandleMessage(uMsg, wParam, lParam);
}

LRESULT CValueWnd::MessageHandler(UINT uMsg, WPARAM wParam, LPARAM lParam, bool& bHandled)
{
    return false;
}

void CValueWnd::AddTrayIcon()
{
    memset(&m_trayIcon, 0, sizeof(NOTIFYICONDATA));

    m_trayIcon.cbSize = sizeof(NOTIFYICONDATA);
    m_trayIcon.hIcon = ::LoadIcon(GetModuleHandle(NULL), MAKEINTRESOURCE(IDI_ICON_DUILIB));
    m_trayIcon.hWnd = m_hWnd;
    lstrcpy(m_trayIcon.szTip, _T("智铺子收银插件"));
    m_trayIcon.uCallbackMessage = WM_SHOWTASK;
    m_trayIcon.uFlags = NIF_ICON | NIF_MESSAGE | NIF_TIP;

    Shell_NotifyIcon(NIM_ADD, &m_trayIcon);
}

LRESULT CValueWnd::OnTrayIcon(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
    //如果在图标中单击左键则还原
    if(lParam == WM_LBUTTONDOWN)
    {

    }

    //如果在图标中单击右键则弹出声明式菜单
    if(lParam == WM_RBUTTONDOWN)
    {
        ShowMenu();
    }

    bHandled = true;
    return 0;
}

void CValueWnd::UpdateKexian(const char* data, int length)
{
    std::string new_data = data;

    //清屏命令
    std::string clear_command = "\f";

    //4个信号灯全灭
    std::string quanmie_command = "\x1bs0";


    std::string show_command = "\x1bQA";
    std::string type_command = "\x1bs";

    if(new_data == clear_command || new_data == quanmie_command)
    {
        //清屏命令，或者是熄灭灯的命令
        return;
    }

    if(new_data.find(show_command) == new_data.npos)
    {
        return;
    }

    std::string money;

    int nPos = new_data.find(show_command);

    int nTypePos = new_data.find(type_command, nPos + 1);

    if(nTypePos == new_data.npos)
    {
        //没有类型信息，有些收银机（比如思迅天店）就是不传类型信息，这种只有传什么显示什么了
        int nTabPos = new_data.find("\r", nPos + 1);

        if(nTabPos == new_data.npos)
        {
            //制表符也没有
            return;
        }

        money = new_data.substr(nPos + 3, nTabPos - nPos - 3);

        CLabelUI* valueLabel = static_cast<CLabelUI*>(m_pm.FindControl(_T("value")));

        std::string money_show = CLewaimaiString::DoubleToString(atof(money.c_str()), 2);
        valueLabel->SetText(CLewaimaiString::UTF8ToUnicode(money_show).c_str());
    }
    else
    {
        std::string value_type = new_data.substr(nTypePos + 2, 1);
        money = new_data.substr(nPos + 3, nTypePos - nPos - 3 - 1);

        if(value_type == "2")
        {
            //这个表示是总计的，那么就要刷新金额显示
            CLabelUI* valueLabel = static_cast<CLabelUI*>(m_pm.FindControl(_T("value")));

            std::string money_show = CLewaimaiString::DoubleToString(atof(money.c_str()), 2);
            valueLabel->SetText(CLewaimaiString::UTF8ToUnicode(money_show).c_str());
        }
        else
        {
            //这种情况传输的是一些找零、单价之类的信息，不更新客显金额
        }
    }

    return;
}

void CValueWnd::InitWndPos()
{
    ASSERT(::IsWindow(m_hWnd));
    ASSERT((GetWindowStyle(m_hWnd)&WS_CHILD) == 0);
    RECT rcDlg = { 0 };
    ::GetWindowRect(m_hWnd, &rcDlg);
    RECT rcArea = { 0 };
    RECT rcCenter = { 0 };
    HWND hWnd = *this;
    HWND hWndParent = ::GetParent(m_hWnd);
    HWND hWndCenter = ::GetWindowOwner(m_hWnd);

    if(hWndCenter != NULL)
    {
        hWnd = hWndCenter;
    }

    // 处理多显示器模式下屏幕居中
    MONITORINFO oMonitor = {};
    oMonitor.cbSize = sizeof(oMonitor);
    ::GetMonitorInfo(::MonitorFromWindow(hWnd, MONITOR_DEFAULTTONEAREST), &oMonitor);
    rcArea = oMonitor.rcWork;

    if(hWndCenter == NULL)
    {
        rcCenter = rcArea;
    }
    else
    {
        ::GetWindowRect(hWndCenter, &rcCenter);
    }

    int DlgWidth = rcDlg.right - rcDlg.left;
    int DlgHeight = rcDlg.bottom - rcDlg.top;

    // Find dialog's upper left based on rcCenter
    int xLeft = rcCenter.right - DlgWidth - 60;
    int yTop = 100;

    // The dialog is outside the screen, move it inside
    if(xLeft < rcArea.left)
    {
        xLeft = rcArea.left;
    }
    else if(xLeft + DlgWidth > rcArea.right)
    {
        xLeft = rcArea.right - DlgWidth;
    }

    if(yTop < rcArea.top)
    {
        yTop = rcArea.top;
    }
    else if(yTop + DlgHeight > rcArea.bottom)
    {
        yTop = rcArea.bottom - DlgHeight;
    }

    ::SetWindowPos(m_hWnd, HWND_TOPMOST, xLeft, yTop, 0, 0, SWP_NOSIZE | SWP_NOACTIVATE);
}

void CValueWnd::ShowMenu()
{
    //获取鼠标坐标
    POINT pt;
    GetCursorPos(&pt);

    //右击后点别地可以清除“右击出来的菜单”
    SetForegroundWindow(m_hWnd);

    //托盘菜单,win32程序使用的是HMENU，如果是MFC程序可以使用CMenu
    HMENU hMenu;

    //生成托盘菜单
    hMenu = CreatePopupMenu();

    //添加菜单,关键在于设置的一个标识符  WM_ONCLOSE 点击后会用到
    AppendMenu(hMenu, MF_STRING, WM_MENU_SYSTEM_SETTING, _T("系统设置"));
    AppendMenu(hMenu, MF_STRING, WM_ONCLOSE, _T("退出"));

    //弹出菜单,并把用户所选菜单项的标识符返回
    int cmd = TrackPopupMenu(hMenu, TPM_RETURNCMD, pt.x, pt.y, NULL, m_hWnd, NULL);

    //如果标识符是WM_ONCLOSE则关闭
    if(cmd == WM_ONCLOSE)
    {
        //退出程序
        ::PostQuitMessage(0);
    }

    if(cmd == WM_MENU_SYSTEM_SETTING)
    {
        if(m_is_show_setting == true)
        {
            return;
        }

        CSystemSettingWnd* pSystemSettingFrame = new CSystemSettingWnd(this);

        if(pSystemSettingFrame != NULL)
        {
            m_is_show_setting = true;

            pSystemSettingFrame->Create(NULL, _T(""), UI_WNDSTYLE_DIALOG, WS_EX_TOOLWINDOW);
            pSystemSettingFrame->SetIcon(IDI_ICON_DUILIB);
            pSystemSettingFrame->CenterWindow();
            UINT ret = pSystemSettingFrame->ShowModal();

            m_is_show_setting = false;

            if(ret == IDOK)
            {

            }
            else if(ret == IDCANCEL)
            {

            }
        }
    }
}