// 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);
		}
	}
}