~chenluhua/Bond2.git

// CControlJobManagerDlg.cpp: 实现文件
//
 
#include "stdafx.h"
#include "Servo.h"
#include "CControlJobManagerDlg.h"
#include "afxdialogex.h"
#include "ToolUnits.h"
#include "RecipeManager.h"
 
 
bool CControlJobManagerDlg::m_bHasState = false;
CControlJobManagerDlg::State CControlJobManagerDlg::m_state{};
 
// CControlJobManagerDlg 对话框
 
IMPLEMENT_DYNAMIC(CControlJobManagerDlg, CDialogEx)
 
CControlJobManagerDlg::CControlJobManagerDlg(CWnd* pParent /*=nullptr*/)
    : CDialogEx(IDD_DIALOG_CONTROL_JOB_MANAGER, pParent)
{
    m_pControlJob = nullptr;
}
 
CControlJobManagerDlg::~CControlJobManagerDlg()
{
 
}
 
void CControlJobManagerDlg::FreeState()
{
    if (!m_bHasState) return;
    for (auto item : m_state.pjWarps) {
        delete (SERVO::CProcessJob*)item.pj;
    }
    m_state.pjWarps.clear();
 
    if (m_state.pControlJob != nullptr) {
        delete m_state.pControlJob;
        m_state.pControlJob = nullptr;
    }
    m_bHasState = false;
}
 
void CControlJobManagerDlg::DoDataExchange(CDataExchange* pDX)
{
    CDialogEx::DoDataExchange(pDX);
    DDX_Control(pDX, IDC_TREE1, m_tree);    
}
 
 
BEGIN_MESSAGE_MAP(CControlJobManagerDlg, CDialogEx)
    ON_WM_SIZE()
    ON_WM_GETMINMAXINFO()
    ON_NOTIFY(TVN_ITEMCHANGED, IDC_TREE1, &CControlJobManagerDlg::OnTvnItemChangedTree)
    ON_NOTIFY(NM_CLICK, IDC_TREE1, &CControlJobManagerDlg::OnTreeClick)          // 新增
    ON_NOTIFY(TVN_KEYDOWN, IDC_TREE1, &CControlJobManagerDlg::OnTreeKeyDown)        // 新增
    ON_MESSAGE(WM_AFTER_TVCHECK, &CControlJobManagerDlg::OnAfterTvCheck)                // 新增
    ON_WM_DESTROY()
    ON_BN_CLICKED(IDC_BUTTON_APPLY, &CControlJobManagerDlg::OnBnClickedButtonApply)
    ON_NOTIFY(TVN_SELCHANGING, IDC_TREE1, &CControlJobManagerDlg::OnTvnSelchangingTree1)
    ON_BN_CLICKED(IDC_BUTTON_BATH_COMPLETION, &CControlJobManagerDlg::OnBnClickedButtonBathCompletion)
END_MESSAGE_MAP()
 
 
// CControlJobManagerDlg 消息处理程序
 
 
BOOL CControlJobManagerDlg::OnInitDialog()
{
    CDialogEx::OnInitDialog();
 
    auto onContentChanged = [&](void* pFrom, int code, void* pContext, int contextType) -> void {
        if (0 == code) {
            GetDlgItem(IDC_BUTTON_APPLY)->EnableWindow(TRUE);
        }
        else if (1 == code) {
            if (contextType == 1) {
                UpProcessJobId((PJWarp*)pContext);
            }
            else if (contextType == 2) {
                UpControlJobId((SERVO::CControlJob*)pContext);
            }
        }
    };
 
    // page1
    CCjPage1* pPage1 = new CCjPage1(this);
    pPage1->Create(IDD_CJ_PAGE1, this);
    pPage1->SetTitle(_T("未选择"));
    pPage1->SetOnContentChanged(onContentChanged);
    pPage1->ShowWindow(SW_SHOW);
    m_pages.push_back(pPage1);
 
    // page2
    CCjPage2* pPage2 = new CCjPage2(this);
    pPage2->Create(IDD_CJ_PAGE2, this);
    pPage2->SetTitle(_T("ProcessJob"));
    pPage2->SetOnContentChanged(onContentChanged);
    m_pages.push_back(pPage2);
 
    // page3
    CCjPage3* pPage3 = new CCjPage3(this);
    pPage3->Create(IDD_CJ_PAGE3, this);
    pPage3->SetTitle(_T("ControlJob"));
    pPage3->SetOnContentChanged(onContentChanged);
    m_pages.push_back(pPage3);
 
 
    // tree
    m_tree.ModifyStyle(0, TVS_CHECKBOXES);
 
 
    InitData();
    UpdateControlJob();
    UpdateCtrlState();
    Resize();
 
 
    return TRUE;  // return TRUE unless you set the focus to a control
                  // 异常: OCX 属性页应返回 FALSE
}
 
void CControlJobManagerDlg::OnSize(UINT nType, int cx, int cy)
{
    CDialogEx::OnSize(nType, cx, cy);
    if (GetDlgItem(IDC_TREE1) == nullptr) return;
 
    Resize();
}
 
void CControlJobManagerDlg::Resize()
{
    CWnd* pItem;
    CRect rcClient, rcItem;
    GetClientRect(&rcClient);
 
    GetDlgItem(IDCANCEL)->GetWindowRect(&rcItem);
    ScreenToClient(&rcItem);
 
    const int LEFTWIDTH = 218;
    int x = 12, y = 12;
    int x2 = rcClient.right - 12;
    int y2 = rcClient.bottom - 12;
 
 
    // 先移动按钮
    pItem = GetDlgItem(IDC_BUTTON_APPLY);
    pItem->GetWindowRect(&rcItem);
    pItem->MoveWindow(x2 - rcItem.Width(), y2 - rcItem.Height(), rcItem.Width(), rcItem.Height());
    x2 -= rcItem.Width();
    x2 -= 8;
 
    pItem = GetDlgItem(IDC_BUTTON_BATH_COMPLETION);
    pItem->GetWindowRect(&rcItem);
    pItem->MoveWindow(x2 - rcItem.Width(), y2 - rcItem.Height(), rcItem.Width(), rcItem.Height());
    x += rcItem.Width();
    x += 8;
    
    y2 -= rcItem.Height();
    y2 -= 8;
 
 
    // 树控件
    x = 12;
    y = 12;
    pItem = GetDlgItem(IDC_TREE1);
    pItem->MoveWindow(x, y, LEFTWIDTH, y2 - y);
    x += LEFTWIDTH;
    x += 5;
 
 
    // 子页面
    x2 = rcClient.right - 12;
    for (auto page : m_pages) {
        page->MoveWindow(x, 12, x2 - x, y2 - 12);
    }
 
}
 
void CControlJobManagerDlg::OnGetMinMaxInfo(MINMAXINFO* lpMMI)
{
    CDialogEx::OnGetMinMaxInfo(lpMMI);
 
    // 设置最小宽高（比如 400x300）
    lpMMI->ptMinTrackSize.x = 600;
    lpMMI->ptMinTrackSize.y = 400;
 
    // 也可以顺便设置最大宽高
    // lpMMI->ptMaxTrackSize.x = 800;
    // lpMMI->ptMaxTrackSize.y = 600;
}
 
void CControlJobManagerDlg::UpdateCtrlState()
{
    GetDlgItem(IDC_BUTTON_BATH_COMPLETION)->EnableWindow(true);
}
 
void CControlJobManagerDlg::UpdateControlJob()
{
    m_tree.DeleteAllItems();
    if (m_pControlJob == nullptr) return;
 
    HTREEITEM hRoot = m_tree.InsertItem(m_pControlJob->id().c_str(), 0, 0);
    m_tree.SetItemData(hRoot, (DWORD_PTR)m_pControlJob);
    m_tree.SetItemState(hRoot, 0, TVIS_STATEIMAGEMASK);
    for (auto& item : m_pjWarps) {
        HTREEITEM hItem = m_tree.InsertItem(((SERVO::CProcessJob*)item.pj)->id().c_str(), 0, 0, hRoot);
        m_tree.SetItemData(hItem, (DWORD_PTR)&item);
        m_tree.SetItemState(hItem, INDEXTOSTATEIMAGEMASK(item.addToCj ? 2 : 1), TVIS_STATEIMAGEMASK);
    }
    m_tree.Expand(hRoot, TVE_EXPAND);
}
 
bool CControlJobManagerDlg::AddPorcessJob(SERVO::CProcessJob* pj)
{
    if (m_pControlJob == nullptr) return false;
    return m_pControlJob->addPjPointer(pj);
}
 
bool CControlJobManagerDlg::RemovePorcessJob(SERVO::CProcessJob* pj)
{
    if (m_pControlJob == nullptr) return false;
    return m_pControlJob->removePjPointer(pj->id());
}
 
void CControlJobManagerDlg::OnTvnItemChangedTree(NMHDR* pNMHDR, LRESULT* pResult)
{
    auto* p = reinterpret_cast<LPNMTREEVIEW>(pNMHDR);
    UINT oldState = p->itemOld.state, newState = p->itemNew.state;
    HTREEITEM hItem = p->itemNew.hItem;
 
    if (((oldState ^ newState) & TVIS_STATEIMAGEMASK) != 0) {
        const int idx = (newState & TVIS_STATEIMAGEMASK) >> 12; // 1=未选,2=已选
        const bool checked = (idx == 2);
 
        PJWarp* pjWarp = (PJWarp*)m_tree.GetItemData(hItem);
        if (pjWarp != nullptr) {
            CString s; s.Format("%s %d", ((SERVO::CProcessJob*)pjWarp->pj)->id().c_str(), 
                checked ? "" : "");
            AfxMessageBox(s);
        }
    }
 
 
    *pResult = 0;
}
 
// 命中复选框：用 NM_CLICK 做命中测试，然后“滞后”读取新状态
void CControlJobManagerDlg::OnTreeClick(NMHDR* pNMHDR, LRESULT* pResult)
{
    *pResult = 0;
 
    DWORD pos = ::GetMessagePos();
    CPoint pt(GET_X_LPARAM(pos), GET_Y_LPARAM(pos));
    m_tree.ScreenToClient(&pt);
 
    TVHITTESTINFO ht{}; ht.pt = pt;
    HTREEITEM hItem = m_tree.HitTest(&ht);
    if (hItem && (ht.flags & TVHT_ONITEMSTATEICON)) {
        // 让 TreeView 先切换，再异步读取最终状态
        PostMessage(WM_AFTER_TVCHECK, (WPARAM)hItem, 0);
    }
}
 
// 空格键也会切换复选框
void CControlJobManagerDlg::OnTreeKeyDown(NMHDR* pNMHDR, LRESULT* pResult)
{
    *pResult = 0;
    auto* p = reinterpret_cast<LPNMTVKEYDOWN>(pNMHDR);
    if (p->wVKey == VK_SPACE) {
        HTREEITEM hItem = m_tree.GetSelectedItem();
        if (hItem) PostMessage(WM_AFTER_TVCHECK, (WPARAM)hItem, 0);
    }
}
 
// 统一处理（读最终状态 + 你的业务）
LRESULT CControlJobManagerDlg::OnAfterTvCheck(WPARAM wParam, LPARAM /*lParam*/)
{
    HTREEITEM hItem = (HTREEITEM)wParam;
    if (!hItem) return 0;
 
    // 只处理第二层：根的直接子节点（可选）
    auto getLevel = [&](HTREEITEM h) {
        int lv = 0; for (HTREEITEM p = m_tree.GetParentItem(h); p; p = m_tree.GetParentItem(p)) ++lv; return lv;
    };
    if (getLevel(hItem) != 1) return 0;
 
    BOOL checked = m_tree.GetCheck(hItem);
 
    // 你的业务逻辑（修正了 CString::Format 的参数类型）
    auto* pjWarp = reinterpret_cast<PJWarp*>(m_tree.GetItemData(hItem));
    if (pjWarp) {
        pjWarp->addToCj = checked;
    }
 
    return 0;
}
 
void CControlJobManagerDlg::OnTvnSelchangingTree1(NMHDR* pNMHDR, LRESULT* pResult)
{
    LPNMTREEVIEW pNMTreeView = reinterpret_cast<LPNMTREEVIEW>(pNMHDR);
 
    bool allow = FALSE;
    HTREEITEM hOldSel = pNMTreeView->itemOld.hItem;
    HTREEITEM hSel = pNMTreeView->itemNew.hItem;
    if (hSel != nullptr) {
        HTREEITEM hParent = m_tree.GetParentItem(hSel);
        if (hParent == nullptr) {
            SERVO::CControlJob* cj = (SERVO::CControlJob*)m_tree.GetItemData(hSel);
            ASSERT(m_pages.size() == 3);
            if (0 == ShowPage(2)) {
                SERVO::CControlJob* pControlJob = (SERVO::CControlJob*)m_tree.GetItemData(hSel);
                m_pages[2]->SetContext(pControlJob, 2);
            }
        }
        else if (m_tree.GetParentItem(hParent) == nullptr) {
            if (0 == ShowPage(1)) {
                PJWarp* pjWarp = (PJWarp*)m_tree.GetItemData(hSel);
                ((CCjPage2*)m_pages[1])->SetPjWarps(m_pjWarps);
                m_pages[1]->SetContext(pjWarp, 1);
            }
            else {
                allow = TRUE;
            }
        }
        else {
            // 有祖先 → 第三层及以下 → Glass
        }
    }
 
    *pResult = allow;
}
 
int CControlJobManagerDlg::ShowPage(int index)
{
    ASSERT(0 <= index && index <= 2);
 
    for (int i = 0; i < 3; i++) {
        if (m_pages[i]->IsWindowVisible()) {
            int ret = m_pages[i]->OnApply();
            if (ret != 0) return -1;
        }
 
        m_pages[i]->ShowWindow(index == i ? SW_SHOW : SW_HIDE);
    }
 
    return 0;
}
 
void CControlJobManagerDlg::OnDestroy()
{
    CDialogEx::OnDestroy();
    SaveState();
 
    for (auto page : m_pages) {
        page->DestroyWindow();
        delete page;
    }
}
 
void CControlJobManagerDlg::InitData()
{
    LoadState();
    if (m_pControlJob != nullptr) return;
 
    m_pControlJob = new SERVO::CControlJob("CJ" + CToolUnits::NowStrSec());
 
    char szBuffer[256];
    for (int i = 0; i < 4; i++) {
        sprintf_s(szBuffer, 256, "PJ%03d", i + 1);
        SERVO::CProcessJob* pj = new SERVO::CProcessJob(std::string(szBuffer));
        PJWarp pjWarp = {};
        pjWarp.pj = pj;
        pjWarp.port = -1;
        m_pjWarps.push_back(pjWarp);
    }
}
 
void CControlJobManagerDlg::OnBnClickedButtonApply()
{
    for (auto item : m_pages) {
        if (item->IsWindowVisible()) {
            if (0 == item->OnApply()) {
                GetDlgItem(IDC_BUTTON_APPLY)->EnableWindow(FALSE);
            }
        }
    }
}
 
void CControlJobManagerDlg::UpProcessJobId(PJWarp* pjWarp)
{
    // 更新树控件
    // 遍历根节点
    HTREEITEM hRoot = m_tree.GetRootItem();
    while (hRoot) {
        // 遍历第二层子节点
        HTREEITEM hChild = m_tree.GetChildItem(hRoot);
        while (hChild) {
            DWORD_PTR data = m_tree.GetItemData(hChild);
            if ((void*)data == pjWarp) {
                SERVO::CProcessJob* pj = (SERVO::CProcessJob*)pjWarp->pj;
                m_tree.SetItemText(hChild, pj->id().c_str());
                return; // 找到就返回
            }
            hChild = m_tree.GetNextSiblingItem(hChild);
        }
 
        hRoot = m_tree.GetNextSiblingItem(hRoot);
    }
}
 
void CControlJobManagerDlg::UpControlJobId(SERVO::CControlJob* pControlJob)
{
    // 更新树控件
    // 遍历根节点
    HTREEITEM hRoot = m_tree.GetRootItem();
    if (hRoot != nullptr) {
        DWORD_PTR data = m_tree.GetItemData(hRoot);
        if ((void*)data == pControlJob) {
            m_tree.SetItemText(hRoot, pControlJob->id().c_str());
            return; // 找到就返回
        }
    }
}
 
void CControlJobManagerDlg::LoadState()
{
    if (!m_bHasState) return;
 
    // 把 s_state -> 成员变量
    m_pControlJob = m_state.pControlJob;
    m_pjWarps = m_state.pjWarps;
}
 
void CControlJobManagerDlg::SaveState()
{
    m_state.pControlJob = m_pControlJob;
    m_state.pjWarps = m_pjWarps;
    m_bHasState = true;
}
 
void CControlJobManagerDlg::OnBnClickedButtonBathCompletion()
{
    // 先检查当前master
    auto& master = theApp.m_model.getMaster();
    if (!master.canCreateControlJob()) {
        AfxMessageBox("当前Master有未结批的Job, 请先结批处理");
        return;
    }
 
 
    // 先应用
    for (int i = 0; i < 3; i++) {
        if (m_pages[i]->IsWindowVisible()) {
            int ret = m_pages[i]->OnApply();
            if (ret != 0) return ;
        }
    }
    GetDlgItem(IDC_BUTTON_APPLY)->EnableWindow(FALSE);
 
 
    // 先检查数据正确性
    int checkCount = 0;
    for (auto item : m_pjWarps) {
        if (!item.addToCj) continue;
        checkCount++;
    }
    if (checkCount == 0) {
        AfxMessageBox(_T("您没有选择要进行工艺处理的Process Job!\n请在要进行工艺处理的Process Job前打勾。"));
        return;
    }
 
 
    SERVO::CLoadPort* pPorts[4];
    pPorts[0] = (SERVO::CLoadPort*)master.getEquipment(EQ_ID_LOADPORT1);
    pPorts[1] = (SERVO::CLoadPort*)master.getEquipment(EQ_ID_LOADPORT2);
    pPorts[2] = (SERVO::CLoadPort*)master.getEquipment(EQ_ID_LOADPORT3);
    pPorts[3] = (SERVO::CLoadPort*)master.getEquipment(EQ_ID_LOADPORT4);
 
    bool bProcessStart[] = {false, false, false, false};
    std::vector<SERVO::CProcessJob*> pjs;
    for (auto item : m_pjWarps) {
        if (!item.addToCj) continue;
        if (item.port == -1) continue;
        BOOL bCheck = FALSE;
        for (int i = 0; i < 8; i++) {
            if (item.checkSlot[i]) {
                bCheck = TRUE;
                break;
            }
        }
        if (!bCheck) continue;
 
        SERVO::CProcessJob* pScr = (SERVO::CProcessJob*)item.pj;
        pScr->setPjWarp(item);
        pScr->setLotId("LotID1");
        pScr->setProductId("ProductId1");
        pScr->setOperationId("OperationId");
        pScr->setRecipe(SERVO::RecipeMethod::NoTuning, pScr->recipeSpec());
 
        SERVO::CProcessJob * pj = new SERVO::CProcessJob(pScr->id());
        pj->setPjWarp(item);
        pj->setLotId("LotID1");
        pj->setProductId("ProductId1");
        pj->setOperationId("OperationId");
        pj->setRecipe(SERVO::RecipeMethod::NoTuning, pScr->recipeSpec());
 
        std::vector<SERVO::CarrierSlotInfo> carriers;
        SERVO::CarrierSlotInfo csi;
        csi.carrierId = pPorts[item.port]->getCassetteId();
        for (int i = 0; i < 8; i++) {
            if (item.checkSlot[i]) {
                SERVO::CGlass* pGlass = pPorts[item.port]->getGlassFromSlot(i+1);
                if (pGlass != nullptr) {
                    csi.slots.push_back(i + 1);
                }
            }
        }
        carriers.push_back(csi);
        pj->setCarriers(carriers);
        pjs.push_back(pj);
        bProcessStart[item.port] = true;
 
        m_pControlJob->addPJ(pScr->id());
    }
 
 
    if (pjs.empty()) {
        AfxMessageBox(_T("没有需要进行工艺处理的Process Job!\n可能未选择Port或选择任何物料。"));
        return;
    }
 
    m_pControlJob->setPJs(pjs);
    m_pControlJob->clearIssues();
    int nRet = master.setProcessJobs(pjs);
    if (nRet <= 0) {
        std::string msg("同步Process Job失败!");
        for (auto pj : pjs) {
            auto& issues = pj->issues();
            if (!issues.empty()) {
                msg.append("\n");
                msg.append(pj->id());
                msg.append(":\n");
                for (auto i : issues) {
                    msg.append("[");
                    msg.append(std::to_string(i.code));
                    msg.append("]");
                    msg.append(i.text);
                    msg.append("\n");
                }
            }
            delete pj;
        }
        pjs.clear();
        AfxMessageBox(msg.c_str());
        return;
    }
 
    // 继续释放有问题的 ProcessJob
    for (auto pj : pjs) {
        if (!pj->issues().empty()) {
            delete pj;
        }
    }
    pjs.clear();
 
    nRet = master.setControlJob(*m_pControlJob);
    if (nRet != 0) {
        std::string msg("同步ControlJob失败!");
        auto& issues = m_pControlJob->issues();
        if (!issues.empty()) {
            msg.append("\n");
            for (auto i : issues) {
                msg.append("[");
                msg.append(std::to_string(i.code));
                msg.append("]");
                msg.append(i.text);
                msg.append("\n");
            }
        }
        AfxMessageBox(msg.c_str());
        return;
    }
 
 
    // 成功，要判断，同步到slot的glass中，类型等
    for (int p = 0; p < 4; p++) {
        if (m_pjWarps[p].port == -1) continue;
        ASSERT(0 <= m_pjWarps[p].port && m_pjWarps[p].port <= 3);
 
        SERVO::CLoadPort* pLoadPort = pPorts[m_pjWarps[p].port];
        for (int i = 0; i < SLOT_MAX; ++i) {
            SERVO::CSlot* pSlot = pLoadPort->getSlot(i);
            if (!pSlot) continue;
 
            SERVO::CGlass* pGlass = dynamic_cast<SERVO::CGlass*>(pSlot->getContext());
            if (pGlass == nullptr) continue;
 
            SERVO::CJobDataS* pJobDataS = pGlass->getJobDataS();
            if (pJobDataS == nullptr) continue;
 
 
            // 设置 Panel ID 和勾选框
            SERVO::CProcessJob* pj = (SERVO::CProcessJob*)m_pjWarps[p].pj;
            int nRecipeID = RecipeManager::getInstance().getIdByPPID(pj->recipeSpec());
            RecipeInfo stRecipeInfo = RecipeManager::getInstance().getRecipeByPPID(pj->recipeSpec());
            std::vector<DeviceRecipe> vecRecipeInfo = stRecipeInfo.vecDeviceList;
 
            pGlass->setScheduledForProcessing(m_pjWarps[p].checkSlot[i]);
            pGlass->setType(static_cast<SERVO::MaterialsType>(m_pjWarps[p].material[i]));
 
            pJobDataS->setLotId(pj->getLotId().c_str());
                pJobDataS->setProductId(pj->getProductId().c_str());
                pJobDataS->setOperationId(pj->getOperationId().c_str());
                pJobDataS->setMaterialsType(m_pjWarps[p].material[i]);
                pJobDataS->setMasterRecipe(nRecipeID);
 
            for (const auto& info : vecRecipeInfo) {
                const std::string& name = info.strDeviceName;
                short nRecipeID = (short)info.nRecipeID;
                
                if (name == EQ_NAME_EFEM) {
                    pJobDataS->setDeviceRecipeId(0, nRecipeID);
                }
                else if (name == EQ_NAME_BONDER1) {
                    pJobDataS->setDeviceRecipeId(1, nRecipeID);
                }
                else if (name == EQ_NAME_BONDER2) {
                    pJobDataS->setDeviceRecipeId(2, nRecipeID);
                }
                else if (name == EQ_NAME_BAKE_COOLING) {
                    pJobDataS->setDeviceRecipeId(3, nRecipeID);
                }
                else if (name == EQ_NAME_VACUUMBAKE) {
                    pJobDataS->setDeviceRecipeId(4, nRecipeID);
                }
                else if (name == EQ_NAME_MEASUREMENT) {
                    pJobDataS->setDeviceRecipeId(5, nRecipeID);
                }
            }
        }
    }
 
    // process start
    for (int p = 0; p < 4; p++) {
        if (bProcessStart[p]) {
            pPorts[p]->sendCassetteCtrlCmd(CCC_PROCESS_START, nullptr, 0, 0, 0, nullptr, nullptr);
            Sleep(100);
        }
    }
}