// TTYDoc.cpp : implementation of the CTTYDoc class
//

#include "stdafx.h"
#include "TTY.h"
#include "commsetDlg.h"
#include "TTYDoc.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CTTYDoc

IMPLEMENT_DYNCREATE(CTTYDoc, CDocument)

BEGIN_MESSAGE_MAP(CTTYDoc, CDocument)
	//{{AFX_MSG_MAP(CTTYDoc)
	ON_COMMAND(ID_API_CONNECT, OnApiConnect)
	ON_UPDATE_COMMAND_UI(ID_API_CONNECT, OnUpdateApiConnect)
	ON_COMMAND(ID_API_DISCONNECT, OnApiDisconnect)
	ON_UPDATE_COMMAND_UI(ID_API_DISCONNECT, OnUpdateApiDisconnect)
	ON_COMMAND(ID_API_SETTINGS, OnApiSettings)
	ON_UPDATE_COMMAND_UI(ID_API_SETTINGS, OnUpdateApiSettings)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
// CTTYDoc construction/destruction

CTTYDoc::CTTYDoc()
{
	// TODO: add one-time construction code here
//初始化操作
	m_bConnected=FALSE;//斷開連接菜單項無效
	m_pThread=NULL;

	m_nBaud = 9600;
	m_nDataBits = 8;
	m_bEcho = FALSE;
	m_bNewLine = FALSE;
	m_nParity = 0;
	m_sPort = "COM1";
	m_nStopBits = 0;
}

CTTYDoc::~CTTYDoc()
{
//程序結束時刪除線程、關閉串口的操作
	if(m_bConnected)
		CloseConnection();

	// 刪除事件句柄
	if(m_hPostMsgEvent)
		CloseHandle(m_hPostMsgEvent);

	if(m_osRead.hEvent)
		CloseHandle(m_osRead.hEvent);

	if(m_osWrite.hEvent)
		CloseHandle(m_osWrite.hEvent);
}

BOOL CTTYDoc::OnNewDocument()
{
	if (!CDocument::OnNewDocument())
		return FALSE;

	((CEditView*)m_viewList.GetHead())->SetWindowText(NULL);
	// TODO: add reinitialization code here
	// (SDI documents will reuse this document)
	
	// 為WM_COMMNOTIFY消息創建事件對象，手工重置，初始化為有信號的
	if((m_hPostMsgEvent=CreateEvent(NULL, TRUE, TRUE, NULL))==NULL)
		return FALSE;

	memset(&m_osRead, 0, sizeof(OVERLAPPED));
	memset(&m_osWrite, 0, sizeof(OVERLAPPED));

	// 為重疊讀創建事件對象，手工重置，初始化為無信號的
	if((m_osRead.hEvent=CreateEvent(NULL, TRUE, FALSE, NULL))==NULL)
		return FALSE;

	// 為重疊寫創建事件對象，手工重置，初始化為無信號的
	if((m_osWrite.hEvent=CreateEvent(NULL, TRUE, FALSE, NULL))==NULL)
		return FALSE;
	return TRUE;
}

/////////////////////////////////////////////////////////////////////////////
// CTTYDoc serialization

void CTTYDoc::Serialize(CArchive& ar)
{
	// CEditView contains an edit control which handles all serialization
	((CEditView*)m_viewList.GetHead())->SerializeRaw(ar);
}

/////////////////////////////////////////////////////////////////////////////
// CTTYDoc diagnostics

#ifdef _DEBUG
void CTTYDoc::AssertValid() const
{
	CDocument::AssertValid();
}

void CTTYDoc::Dump(CDumpContext& dc) const
{
	CDocument::Dump(dc);
}
#endif //_DEBUG

/////////////////////////////////////////////////////////////////////////////
// CTTYDoc commands

BOOL CTTYDoc::CanCloseFrame(CFrameWnd* pFrame) 
{
	// TODO: Add your specialized code here and/or call the base class
	
	SetModifiedFlag(FALSE); // 將文檔的修改標志設置成未修改
	return CDocument::CanCloseFrame(pFrame);
}

// 工作者線程，負責監視串行口
UINT CommProc(LPVOID pParam)
{
	OVERLAPPED os;
	DWORD dwMask, dwTrans;
	COMSTAT ComStat;
	DWORD dwErrorFlags;

	CTTYDoc *pDoc=(CTTYDoc*)pParam;

	memset(&os, 0, sizeof(OVERLAPPED));
	os.hEvent=CreateEvent(NULL, TRUE, FALSE, NULL);

	if(os.hEvent==NULL)
	{
		AfxMessageBox("Can't create event object!");
		return (UINT)-1;
	}

	// Clear comm buffers at startup
	if (pDoc->m_hCom)		// check if the port is opened
		PurgeComm(pDoc->m_hCom, PURGE_RXCLEAR | PURGE_TXCLEAR | PURGE_RXABORT | PURGE_TXABORT);

	while(pDoc->m_bConnected)
	{
		ClearCommError(pDoc->m_hCom,&dwErrorFlags,&ComStat);

		if(ComStat.cbInQue)
		{
			// 無限等待WM_COMMNOTIFY消息被處理完
			WaitForSingleObject(pDoc->m_hPostMsgEvent, INFINITE);
			ResetEvent(pDoc->m_hPostMsgEvent);

			// 通知視圖
			PostMessage(pDoc->m_hTermWnd, WM_COMMNOTIFY, EV_RXCHAR, 0);

			continue;
		}

		dwMask=0;

		if(!WaitCommEvent(pDoc->m_hCom, &dwMask, &os)) // 重疊操作
		{

			if(GetLastError()==ERROR_IO_PENDING)
			// 無限等待重疊操作結果
				GetOverlappedResult(pDoc->m_hCom, &os, &dwTrans, TRUE);
			else
			{
				CloseHandle(os.hEvent);
				return (UINT)-1;
			}
		}

	}

	CloseHandle(os.hEvent);
	return 0;
}

// 將指定數量的字符從串行口輸出
DWORD CTTYDoc::WriteComm(unsigned char *buf, DWORD dwLength)
{
	BOOL fState;
	DWORD length=dwLength;
	COMSTAT ComStat;
	DWORD dwErrorFlags;

	ClearCommError(m_hCom,&dwErrorFlags,&ComStat);
	fState=WriteFile(m_hCom,buf,length,&length,&m_osWrite);

	if(!fState)
	{

		if(GetLastError()==ERROR_IO_PENDING)
		{
			GetOverlappedResult(m_hCom,&m_osWrite,&length,TRUE);// 等待
		}
		else
			length=0;
	}

	return length;
}

// 從串行口輸入緩沖區中讀入指定數量的字符
DWORD CTTYDoc::ReadComm(unsigned char *buf, DWORD dwLength)
{
	DWORD length=0;
	COMSTAT ComStat;
	DWORD dwErrorFlags;

	ClearCommError(m_hCom,&dwErrorFlags,&ComStat);
	length=min(dwLength, ComStat.cbInQue);
	ReadFile(m_hCom,buf,length,&length,&m_osRead);

	return length;
}

//菜單項"串口連接"的響應
void CTTYDoc::OnApiConnect() 
{
	// TODO: Add your command handler code here
	if(!OpenConnection())
		AfxMessageBox("Can't open connection");

}

//更新"串口連接"菜單項
void CTTYDoc::OnUpdateApiConnect(CCmdUI* pCmdUI) 
{
	// TODO: Add your command update UI handler code here
	pCmdUI->Enable(!m_bConnected);
}

//菜單項"斷開連接"的響應函數
void CTTYDoc::OnApiDisconnect() 
{
	// TODO: Add your command handler code here
	CloseConnection(); 
}

//更新菜單項"斷開連接"
void CTTYDoc::OnUpdateApiDisconnect(CCmdUI* pCmdUI) 
{
	// TODO: Add your command update UI handler code here
	pCmdUI->Enable(m_bConnected);
}

//菜單項"串行口設置" 的響應函數
void CTTYDoc::OnApiSettings() 
{
	// TODO: Add your command handler code here
	CCommSetDlg dlg;
	CString str;

	dlg.m_bConnected = m_bConnected;

	dlg.m_sPort = m_sPort;

	str.Format("%d", m_nBaud);
	dlg.m_sBaud = str;

	str.Format("%d", m_nDataBits);
	dlg.m_sDataBits = str;

	dlg.m_nParity = m_nParity;
	
	dlg.m_nStopBits = m_nStopBits;
	
	dlg.m_bEcho = m_bEcho;
	dlg.m_bNewLine = m_bNewLine;

	if(dlg.DoModal() == IDOK)
	{
		m_sPort = dlg.m_sPort;
		m_nBaud = atoi(dlg.m_sBaud);
		m_nDataBits = atoi(dlg.m_sDataBits);
		m_nParity = dlg.m_nParity;
		m_nStopBits = dlg.m_nStopBits;
		
		m_bEcho=dlg.m_bEcho;
		m_bNewLine=dlg.m_bNewLine;

		if(m_bConnected)
			if(!ConfigConnection())
				AfxMessageBox("Can't realize the settings!");
	}
}

//更新"串行口設置"菜單項
void CTTYDoc::OnUpdateApiSettings(CCmdUI* pCmdUI) 
{
	// TODO: Add your command update UI handler code here
	
}
 //配置串口連接
BOOL CTTYDoc::ConfigConnection()
{
	DCB dcb;

	if(!GetCommState(m_hCom, &dcb))
		return FALSE;

	dcb.fBinary=TRUE;
	dcb.BaudRate = m_nBaud; // 數據傳輸速率
	dcb.ByteSize = m_nDataBits; // 每字節位數
	dcb.fParity = TRUE;

	switch(m_nParity) // 校驗設置
	{
	case 0: 
		dcb.Parity=NOPARITY;
		break;

	case 1: 
		dcb.Parity=EVENPARITY;
		break;

	case 2: 
		dcb.Parity=ODDPARITY;
		break;

	default:;
	}

	switch(m_nStopBits) // 停止位
	{
	case 0: 
		dcb.StopBits=ONESTOPBIT;
		break;

	case 1: 
		dcb.StopBits=ONE5STOPBITS;
		break;

	case 2:
		dcb.StopBits=TWOSTOPBITS;
		break;

	default:;
	}

	// 硬件流控制設置
	dcb.fOutxCtsFlow = FALSE;  //如果僅使用TX、RX、GND三根線連接設置為FALSE,否則為TRUE
	dcb.fRtsControl = RTS_CONTROL_ENABLE;

	// XON/XOFF流控制設置
	dcb.fInX=dcb.fOutX = FALSE; //如果僅使用TX、RX、GND三根線連接設置為FALSE,否則為TRUE
	dcb.XonChar = XON;
	dcb.XoffChar = XOFF;
	dcb.XonLim = 50;
	dcb.XoffLim = 50;

	return SetCommState(m_hCom, &dcb);
}

// 打開并配置串行口，建立工作者線程
BOOL CTTYDoc::OpenConnection()
{
	COMMTIMEOUTS TimeOuts;
	POSITION firstViewPos;
	CView *pView;

	firstViewPos=GetFirstViewPosition();
	pView=GetNextView(firstViewPos);

	m_hTermWnd=pView->GetSafeHwnd();

	if(m_bConnected)
		return FALSE;

	m_hCom=CreateFile(m_sPort, GENERIC_READ | GENERIC_WRITE, 0, NULL,
		OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL | FILE_FLAG_OVERLAPPED, 
		NULL); // 重疊方式

	if(m_hCom==INVALID_HANDLE_VALUE)
		return FALSE;

	SetupComm(m_hCom,MAXBLOCK,MAXBLOCK);
	SetCommMask(m_hCom, EV_RXCHAR);


	// 把間隔超時設為最大，把總超時設為0將導致ReadFile立即返回并完成操作
	TimeOuts.ReadIntervalTimeout=MAXDWORD; 
	TimeOuts.ReadTotalTimeoutMultiplier=0; 
	TimeOuts.ReadTotalTimeoutConstant=0; 

	/* 設置寫超時以指定WriteComm成員函數中的GetOverlappedResult函數的等待時間*/
	TimeOuts.WriteTotalTimeoutMultiplier=50;
	TimeOuts.WriteTotalTimeoutConstant=2000;

	SetCommTimeouts(m_hCom, &TimeOuts);

	if(ConfigConnection())
	{

		m_pThread=AfxBeginThread(CommProc, this, THREAD_PRIORITY_NORMAL, 
			0, CREATE_SUSPENDED, NULL); // 創建并掛起線程

		if(m_pThread==NULL)
		{
			CloseHandle(m_hCom);
			return FALSE;
		}
		else

		{
			m_bConnected=TRUE;
			m_pThread->ResumeThread(); // 恢復線程運行
		}
	}
	else
	{
		CloseHandle(m_hCom);
		return FALSE;
	}
	
	return TRUE;
}

//關閉連接，關閉工作線程
void CTTYDoc::CloseConnection()
{
	if(!m_bConnected)
		return;

	m_bConnected=FALSE;

	//結束CommProc線程中WaitSingleObject函數的等待
	SetEvent(m_hPostMsgEvent); 

	//結束CommProc線程中WaitCommEvent的等待
	SetCommMask(m_hCom, 0); 

	//等待輔助線程終止
	WaitForSingleObject(m_pThread->m_hThread, INFINITE);
	m_pThread=NULL;
	CloseHandle(m_hCom);

}