// cantestDlg.cpp : 實現文件
//

#include "stdafx.h"
#include "cantest.h"
#include "cantestDlg.h"
#include "settingdlg.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif

#define IOCTL_SET_CAN_BAUD			0x01
#define IOCTL_SET_CAN_TXMODE		0x02
#define IOCTL_SET_CAN_FILTER		0x03

typedef	 struct {	/* CAN_MSG */
	union {
		UINT8	id[4];
		UINT32	identifier;	
	};
	union {
		struct {
		UINT8	length	:4;		/* data length */
		UINT8	resbit	:2;
		UINT8	remote	:1;		/* remote transmission request */
		UINT8	format	:1;		/* frame format */
		};
		UINT8	info;
	};
	UINT8	reserve[3];
	UINT8	data[8];			/* data field */
} CAN_MSG;

// CAN 驅動程序注冊表信息
WCHAR *szRegKey = L"Drivers\\BuiltIn\\CAN";
WCHAR *szPrefix = L"CAN";
//WCHAR *szDLL = L"\\Program Files\\can\\can.dll";
WCHAR *szDLL = L"\\can.dll";
WCHAR *szFriendlyName = L"CAN Driver";
DWORD DrvIndex = 1;
DWORD DrvOrder = 0;

// 向注冊表寫入一個字符串
#define WRITE_REG_SZ(Name, Value) \
	RegSetValueEx(m_hDrvKey, Name, 0, REG_SZ, (LPBYTE)Value, \
				  (wcslen(Value)+1)*sizeof(WCHAR))

// 向注冊表寫入一個 DWORD 型的值
#define WRITE_REG_DWORD(Name, Value) { \
	DWORD dwValue = Value; \
	RegSetValueEx(m_hDrvKey, Name, 0, REG_DWORD, \
		          (LPBYTE)&dwValue, sizeof(DWORD)); \
}


// CCantestDlg 對話框

CCantestDlg::CCantestDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CCantestDlg::IDD, pParent)
	, m_strCanData(_T("00 01 02 03 04 05 06 07"))
	, m_nCanID(0)
{
	m_hDriver = INVALID_HANDLE_VALUE;
	m_hFile	= INVALID_HANDLE_VALUE;
	m_hDrvKey = NULL;
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
	m_dwTxMode = 2;
}

void CCantestDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	DDX_Text(pDX, IDC_CAN_DATA, m_strCanData);
	DDX_Text(pDX, IDC_CAN_ID, m_nCanID);
	DDV_MinMaxUInt(pDX, m_nCanID, 0, 536870911);
	DDX_Control(pDX, IDC_LIST1, m_msgList1);
}

BEGIN_MESSAGE_MAP(CCantestDlg, CDialog)
#if defined(_DEVICE_RESOLUTION_AWARE) && !defined(WIN32_PLATFORM_WFSP)
	ON_WM_SIZE()
#endif
	//}}AFX_MSG_MAP
	ON_BN_CLICKED(IDC_OPEN_DEVICE, OnBnClickedOpenDevice)
	ON_BN_CLICKED(IDC_CLOSE_DEVICE, OnBnClickedCloseDevice)
	ON_BN_CLICKED(IDC_SEND_MSG, OnBnClickedSendMsg)
	ON_BN_CLICKED(IDC_TXMODE, OnBnClickedTxmode)
	ON_COMMAND(ID_TX_NORMAL, OnTxNormal)
	ON_COMMAND(ID_TX_SINGLE, OnTxSingle)
	ON_COMMAND(ID_TX_SELF, OnTxSelf)
	ON_COMMAND(ID_TX_SINGLE_SELF, OnTxSingleSelf)
	ON_BN_CLICKED(IDC_CAN_CONFIG, OnBnClickedCanConfig)
END_MESSAGE_MAP()


// CCantestDlg 消息處理程序

BOOL CCantestDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	// 設置此對話框的圖標。當應用程序主窗口不是對話框時，框架將自動
	//  執行此操作
	SetIcon(m_hIcon, TRUE);			// 設置大圖標
	SetIcon(m_hIcon, FALSE);		// 設置小圖標

	// TODO: 在此添加額外的初始化代碼
	//if (!LoadDriver())
	//	return FALSE;

	CheckRadioButton(IDC_RADIO1, IDC_RADIO2, IDC_RADIO1);
	CheckRadioButton(IDC_RADIO3, IDC_RADIO4, IDC_RADIO3);

	return TRUE;  // 除非將焦點設置到控件，否則返回 TRUE
}

BOOL CCantestDlg::DestroyWindow()
{
	// TODO: 在此添加專用代碼和/或調用基類
	OnBnClickedCloseDevice();
	//UnloadDriver();

	return CDialog::DestroyWindow();
}

#if defined(_DEVICE_RESOLUTION_AWARE) && !defined(WIN32_PLATFORM_WFSP)
void CCantestDlg::OnSize(UINT /*nType*/, int /*cx*/, int /*cy*/)
{
	DRA::RelayoutDialog(
		AfxGetInstanceHandle(), 
		this->m_hWnd, 
		DRA::GetDisplayMode() != DRA::Portrait ? 
			MAKEINTRESOURCE(IDD_CANTEST_DIALOG_WIDE) : 
			MAKEINTRESOURCE(IDD_CANTEST_DIALOG));
}
#endif

BOOL CCantestDlg::LoadDriver() 
{
	LONG nErr;
	DWORD dwDisp;

	// 創建注冊表鍵 HKEY_LOCAL_MACHINE\Drivers\BuiltIn\SPL
    nErr = RegCreateKeyEx(HKEY_LOCAL_MACHINE, szRegKey, 0, L"", 0, 0, NULL, &m_hDrvKey, &dwDisp);
	if (nErr == ERROR_SUCCESS)
	{	
        WRITE_REG_SZ( L"Prefix", szPrefix);				// 寫入 "Prefix"="SPL" 
        WRITE_REG_SZ( L"Dll", szDLL);					// 寫入 "Dll"="\\SimpleDriver.dll"
        WRITE_REG_SZ( L"FriendlyName", szFriendlyName);	// 寫入 "FriendlyName"="Simple Driver"
        WRITE_REG_DWORD( L"Order", DrvOrder);			// 寫入 "Index"=dword:0
        WRITE_REG_DWORD( L"Index", DrvIndex);			// 寫入 "Order"=dword:0
 	}
	else
	{
		AfxMessageBox(_T("創建驅動注冊表子鍵失敗!"), MB_ICONERROR);
		return FALSE;
	}

	m_hDriver = ActivateDeviceEx(szRegKey, NULL, 0, NULL);
	if ((m_hDriver == INVALID_HANDLE_VALUE) || (m_hDriver == NULL))
	{
		//TRACE0("加載 CAN.DLL 失敗.");
		AfxMessageBox(_T("加載驅動程序失敗!"), MB_ICONERROR);
		return FALSE;
	}
	else
	{
		//TRACE0("加載 CAN.DLL 成功.");
		return TRUE;
	}
}

void CCantestDlg::UnloadDriver() 
{
	if ((m_hDriver != INVALID_HANDLE_VALUE) && (m_hDriver != NULL))
	{
		//OnCloseDriver();								// 關閉驅動

		if (DeactivateDevice(m_hDriver))
		{	
			RegCloseKey(m_hDrvKey);						// 關閉注冊表鍵
			RegDeleteKey(HKEY_LOCAL_MACHINE, szRegKey);	// 刪除注冊表鍵的內容
			m_hDriver = INVALID_HANDLE_VALUE;
			//TRACE0("卸載 CAN.DLL 成功.");
		}
		else
		{
			//TRACE0("卸載 CAN.DLL 失敗.");
			AfxMessageBox(_T("卸載驅動程序失敗."), MB_ICONERROR);
		}
	}
}

void CCantestDlg::OnBnClickedOpenDevice()
{
	DWORD dwReturn;

	m_hFile = CreateFile(TEXT("CAN1:"), GENERIC_READ | GENERIC_WRITE, 0, 
						NULL, OPEN_EXISTING, 0, 0);
	if (m_hFile == INVALID_HANDLE_VALUE)
	{
		AfxMessageBox(_T("不能打開 CAN1!"), MB_ICONERROR); 
		return;
	}

	GetDlgItem(IDC_OPEN_DEVICE)->EnableWindow(FALSE);
	GetDlgItem(IDC_CLOSE_DEVICE)->EnableWindow();
	//GetDlgItem(IDC_BUS_ON)->EnableWindow();
	//GetDlgItem(IDC_BUS_OFF)->EnableWindow();
	GetDlgItem(IDC_CAN_CONFIG)->EnableWindow();
	GetDlgItem(IDC_SEND_MSG)->EnableWindow();
	GetDlgItem(IDC_TXMODE)->EnableWindow();

	m_bKillThread = FALSE;
	AfxBeginThread(RxThreadProc, this);

	DeviceIoControl(m_hFile, IOCTL_SET_CAN_TXMODE,
		&m_dwTxMode, 1, NULL, 0, &dwReturn, NULL);
}

void CCantestDlg::OnBnClickedCloseDevice()
{
	if (m_hFile != INVALID_HANDLE_VALUE)
	{
		m_bKillThread = TRUE;
		Sleep(200);
		CloseHandle(m_hFile);
		m_hFile = INVALID_HANDLE_VALUE;

		GetDlgItem(IDC_OPEN_DEVICE)->EnableWindow();
		GetDlgItem(IDC_CLOSE_DEVICE)->EnableWindow(FALSE);
		//GetDlgItem(IDC_BUS_ON)->EnableWindow(FALSE);
		//GetDlgItem(IDC_BUS_OFF)->EnableWindow(FALSE);
		GetDlgItem(IDC_CAN_CONFIG)->EnableWindow(FALSE);
		GetDlgItem(IDC_SEND_MSG)->EnableWindow(FALSE);
		GetDlgItem(IDC_TXMODE)->EnableWindow(FALSE);
	}
}

void CCantestDlg::OnBnClickedSendMsg()
{
	DWORD dwWrite;
	CAN_MSG msg;

	GetDlgItem(IDC_SEND_MSG)->EnableWindow(FALSE);
	GetDlgItem(IDC_TXMODE)->EnableWindow(FALSE);

	int nMsgType = GetCheckedRadioButton(IDC_RADIO1, IDC_RADIO2);
	int nMsgFormat = GetCheckedRadioButton(IDC_RADIO3, IDC_RADIO4);
	int nData[8];
	UpdateData();
	_stscanf(m_strCanData, _T("%2X %2X %2X %2X %2X %2X %2X %2X"), 
		&nData[0], &nData[1], &nData[2], &nData[3],
		&nData[4], &nData[5], &nData[6], &nData[7]);
	UpdateData(FALSE);

	msg.identifier = m_nCanID;
	msg.length = 8;
	msg.remote = nMsgFormat - IDC_RADIO3;
	msg.format = nMsgType - IDC_RADIO1;
	for (int i = 0; i < 8; i++)
		msg.data[i] = nData[i] & 0xff;

	WriteFile(m_hFile, &msg, sizeof(msg), &dwWrite, NULL);
	if (dwWrite < sizeof(CAN_MSG))
		AfxMessageBox(_T("發送失敗"), MB_ICONEXCLAMATION);

	GetDlgItem(IDC_SEND_MSG)->EnableWindow();
	GetDlgItem(IDC_TXMODE)->EnableWindow();
}

UINT CCantestDlg::RxThreadProc(LPVOID pParam)
{
	DWORD dwRead;
	CAN_MSG msg;
	CCantestDlg* pDlg = (CCantestDlg*)pParam;
	TCHAR szMsg[512];

	while (!pDlg->m_bKillThread)
	{
		ReadFile(pDlg->m_hFile, &msg, sizeof(msg), &dwRead, NULL);
		if (dwRead == sizeof(msg))
		{
			int nLen = _stprintf(szMsg, _T("%9.9u, %s, %s, %u, "),
				msg.identifier,
				msg.remote ? _T("R") : _T("D"),
				msg.format ? _T("E") : _T("S"),
				msg.length); 
			for (int i = 0; i < msg.length; i++)
				nLen += _stprintf(szMsg + nLen, _T("%2.2X "), msg.data[i]); 

			int nItem = pDlg->m_msgList1.GetCount();
			if (nItem > 64)
			{
				pDlg->m_msgList1.DeleteString(0);
				nItem--;
			}
			pDlg->m_msgList1.InsertString(nItem, szMsg);

			pDlg->m_msgList1.SetCurSel(nItem);
		}
	}
	return 0;
}

void CCantestDlg::OnBnClickedTxmode()
{
	CRect rect;
	GetDlgItem(IDC_TXMODE)->GetWindowRect(rect);

	CMenu menu;
	menu.LoadMenu(IDR_MENU_TXMODE);
	menu.GetSubMenu(0)->CheckMenuRadioItem(ID_TX_NORMAL, ID_TX_SINGLE_SELF, 
		ID_TX_NORMAL + m_dwTxMode, 0);
	menu.GetSubMenu(0)->TrackPopupMenu(TPM_LEFTALIGN, rect.right, rect.top, this);
}


void CCantestDlg::OnTxNormal()
{
	m_dwTxMode = 0;
	DWORD dwReturn;
	DeviceIoControl(m_hFile, IOCTL_SET_CAN_TXMODE,
		&m_dwTxMode, 1, NULL, 0, &dwReturn, NULL);
}

void CCantestDlg::OnTxSingle()
{
	m_dwTxMode = 1;
	DWORD dwReturn;
	DeviceIoControl(m_hFile, IOCTL_SET_CAN_TXMODE,
		&m_dwTxMode, 1, NULL, 0, &dwReturn, NULL);
}

void CCantestDlg::OnTxSelf()
{
	m_dwTxMode = 2;
	DWORD dwReturn;
	DeviceIoControl(m_hFile, IOCTL_SET_CAN_TXMODE,
		&m_dwTxMode, 1, NULL, 0, &dwReturn, NULL);
}

void CCantestDlg::OnTxSingleSelf()
{
	m_dwTxMode = 3;
	DWORD dwReturn;
	DeviceIoControl(m_hFile, IOCTL_SET_CAN_TXMODE,
		&m_dwTxMode, 1, NULL, 0, &dwReturn, NULL);
}

void CCantestDlg::OnBnClickedCanConfig()
{
	DWORD dwReturn;
	UINT32 filter[2];
	CSettingDlg dlg;

	INT_PTR nResponse = dlg.DoModal();
	if (nResponse == IDOK)
	{
		DeviceIoControl(m_hFile, IOCTL_SET_CAN_BAUD,
			&dlg.m_btr, 2, NULL, 0, &dwReturn, NULL);

		filter[0] = dlg.m_acr;
		filter[1] = dlg.m_amr;
		DeviceIoControl(m_hFile, IOCTL_SET_CAN_FILTER,
			&filter, 4, NULL, 0, &dwReturn, NULL);
	}
}