// DIODlg.cpp : implementation file
//

#include "stdafx.h"
#include "DIO.h"
#include "DIODlg.h"
#include "afxdialogex.h"

#include "AIOWDM.h"  //WDM Driver

#ifdef _DEBUG
#define new DEBUG_NEW
#endif

#define InsCR(x) x += 0x0d; x += 0x0a;

// CAboutDlg dialog used for App About

class CAboutDlg : public CDialogEx
{
public:
	CAboutDlg();

// Dialog Data
	enum { IDD = IDD_ABOUTBOX };

	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support

// Implementation
protected:
	DECLARE_MESSAGE_MAP()
public:
//	afx_msg void OnTimer(UINT_PTR nIDEvent);
};

CAboutDlg::CAboutDlg() : CDialogEx(CAboutDlg::IDD)
{
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialogEx::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx)
//	ON_WM_TIMER()
END_MESSAGE_MAP()


// CDIODlg dialog

CDIODlg::CDIODlg(CWnd* pParent /*=NULL*/)
	: CDialogEx(CDIODlg::IDD, pParent)
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
	m_AddressSelection = _T("");
	m_InputA = _T("");
	m_InputB = _T("");
	m_OutputA = _T("");
	m_StatusEdit = _T("");
	m_CardName = _T("");
}

void CDIODlg::DoDataExchange(CDataExchange* pDX)
{
	CDialogEx::DoDataExchange(pDX);
	DDX_Control(pDX, IDC_COMBO1, m_AddressCombo);
	DDX_CBString(pDX, IDC_COMBO1, m_AddressSelection);
	DDX_Text(pDX, IDC_EDIT_INPUTA, m_InputA);
	DDX_Text(pDX, IDC_EDIT_INPUTB, m_InputB);
	DDX_Text(pDX, IDC_EDIT_OUTPUTA, m_OutputA);
	DDX_Text(pDX, IDC_EDIT_STATUS, m_StatusEdit);
	//  DDX_Control(pDX, IDC_STATIC_CARD_NAME, m_StatusControl);
	DDX_Control(pDX, IDC_BUTTON_PERFORMIO, m_TimerButton);
	DDX_Control(pDX, IDC_EDIT_STATUS, m_StatusControl);
	DDX_Text(pDX, IDC_STATIC_CARD_NAME, m_CardName);
}

BEGIN_MESSAGE_MAP(CDIODlg, CDialogEx)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_BUTTON_PERFORMIO, &CDIODlg::OnBnClickedPerformIO)
	ON_WM_TIMER()
	ON_CBN_SELCHANGE(IDC_COMBO1, &CDIODlg::OnCbnSelchangeCombo1)
END_MESSAGE_MAP()


// CDIODlg message handlers

BOOL CDIODlg::OnInitDialog()
{
	CDialogEx::OnInitDialog();

	// Add "About..." menu item to system menu.

	// IDM_ABOUTBOX must be in the system command range.
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != NULL)
	{
		BOOL bNameValid;
		CString strAboutMenu;
		bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX);
		ASSERT(bNameValid);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

	// Set the icon for this dialog.  The framework does this automatically
	//  when the application's main window is not a dialog
	SetIcon(m_hIcon, TRUE);			// Set big icon
	SetIcon(m_hIcon, FALSE);		// Set small icon

	// TODO: Add extra initialization here
	
	// Ping the driver AIOWDM.SYS:
	if (InPortB(0x61) == 0xAA55) 
	{
		AfxMessageBox(_T("AIOWDM.SYS not detected.\n Please copy AIOWDM.SYS into [Windows]/system32/drivers and re-run this sample.\n Ensure that a board is installed properly."));
	}

	FindCardsWDM();
	
	m_StatusEdit = _T("This sample outputs a walking bit on Port A of the selected group. It then reads back the value from Port A.  This should be identical to the output value due to on-chip readback. It also displays the Input from Port B. If a wrap plug is installed, Port B should read the same value. Otherwise, ground individual bits on Port B to see them show zeroes on the display.");
	
	UpdateData(false);

	return TRUE;  // return TRUE  unless you set the focus to a control
}

void CDIODlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialogEx::OnSysCommand(nID, lParam);
	}
}

// If you add a minimize button to your dialog, you will need the code below
//  to draw the icon.  For MFC applications using the document/view model,
//  this is automatically done for you by the framework.

void CDIODlg::OnPaint()
{
	if (IsIconic())
	{
		CPaintDC dc(this); // device context for painting

		SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

		// Center icon in client rectangle
		int cxIcon = GetSystemMetrics(SM_CXICON);
		int cyIcon = GetSystemMetrics(SM_CYICON);
		CRect rect;
		GetClientRect(&rect);
		int x = (rect.Width() - cxIcon + 1) / 2;
		int y = (rect.Height() - cyIcon + 1) / 2;

		// Draw the icon
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialogEx::OnPaint();
	}
}

// The system calls this function to obtain the cursor to display while the user drags
//  the minimized window.
HCURSOR CDIODlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}


void CDIODlg::OnBnClickedPerformIO()
{
	// TODO: Add your control notification handler code here
	static bool TimerEnabled = false;

	if (TimerEnabled)
	{
		TimerEnabled = false;
		KillTimer(0);
		m_TimerButton.SetWindowText(_T("Perform I/O"));
	}
	else
	{
		TimerEnabled = true;
		RelOutPortB(CardNum, Offset + 3, 0x82); // Driver call
		SetTimer(0, 1000, NULL);
		m_TimerButton.SetWindowText(_T("Stop I/O"));
	}
}


void CDIODlg::OnTimer(UINT_PTR nIDEvent)
{
	// TODO: Add your message handler code here and/or call default
	static int value[3] = {0,0,0}, i = 0;
	int j;
	char msg[9] = "00000000";

	value[0] = 1 << i++;
	RelOutPortB(CardNum, Offset, value[0]); // Driver Calls
	value[1] = RelInPortB(CardNum, Offset);
	value[2] = RelInPortB(CardNum, Offset+1);

	i %= 8;
	for (j=0; j<8; j++)
        msg[7-j] = (((value[0] & (1 << j)) > 0) + 0x30);
    m_OutputA = msg;
	for (j=0; j<8; j++)
        msg[7-j] = (((value[1] & (1 << j)) > 0) + 0x30);
	m_InputA = msg;
    for (j=0; j<8; j++)
        msg[7-j] = (((value[2] & (1 << j)) > 0) + 0x30);
	m_InputB = msg;

	UpdateData(false);


	CDialogEx::OnTimer(nIDEvent);
}


void CDIODlg::OnCbnSelchangeCombo1()
{
	// TODO: Add your control notification handler code here
	CString SelStr;
	UpdateData(true);
	m_AddressCombo.GetLBText(m_AddressCombo.GetCurSel(), SelStr);
	
	// Select and set new address data:
	if(CardData.IsValid == true)
	{
		Offset = BaseOffset[m_AddressCombo.GetCurSel()];	
		CardData.IsSelected = true;
	}

	m_StatusEdit += "\r\nNew Port Selected: ";
	m_StatusEdit += SelStr;
	
	m_AddressSelection = SelStr;
	
	UpdateData(false);

}

void CDIODlg::FindCardsWDM(void)
{
	bool found = false;
	char AddStr[256];
	
	// Vars for QueryCardInfo() :
    unsigned long DeviceID, Base;
    unsigned long NameSize;
    UCHAR Name[256];
	CString strname;

	// Init:
    RunFlag = true;
	NumCards = 0;
	CardNum = 0;
	int DataIndex = 0;
	Offset =0;

	// Set CardData to 0 as needed:
    CardData.IsValid = false;
    CardData.IsSelected = false;
    CardData.DeviceID = 0;
    CardData.Base = 0;	  		
	
	BaseOffset[0] = 0;
	BaseOffset[1] = 4;
	BaseOffset[2] = 8;
	BaseOffset[3] = 12;
	BaseOffset[4] = 16;

	// Get the total number of cards installed:
	NumCards = GetNumCards();
	
	if (NumCards == 0) // no cards present
	{	    		
		m_CardName = "No Card Found";

		m_StatusEdit =  "No cards were found!";
		m_StatusEdit += "This may mean the card is not installed. ";
		m_StatusEdit += "Check Device Manager for a card and its status";
		m_StatusEdit += "You may consider rebooting your system.";
		m_AddressCombo.EnableWindow(false);
		m_StatusControl.LineScroll(999);

		RunFlag = false;
	}
	else 
	{
		// validate and store card data:
		NameSize = 256;
		QueryCardInfo(CardNum, &DeviceID, &Base, &NameSize, Name);
						
		// only need to do this once:
		CardData.DeviceID = DeviceID;
		CardData.Base = Base;
		CardData.IsValid = true;

		// Populate list box with addresses found set flags:
		switch (DeviceID) 
		{
			case 0xECA9:
				m_CardName = "PCI-A12-16 Analog I/O Card with Digital";
				sprintf(AddStr, "Port 0: %4X", (Base & 0xFFF8) + 0x10);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
				found = true;
				break;
			case 0xECAA:
				m_CardName = "PCI-A12-16A Analog I/O Card with Digital";
				sprintf(AddStr, "Port 0: %4X", (Base & 0xFFF8) + 0x10);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
				found = true;
				break;
			case 0x2C58:
				m_CardName = "PCI-DIO-96C3 Parallel Digital I/O Card";
				sprintf(AddStr, "Port 0: %4X", Base & 0xFFF8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
				sprintf(AddStr, "Port 1: %4X", (Base & 0xFFF8) + 4);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
				sprintf(AddStr, "Port 2: %4X", (Base & 0xFFF8) + 8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
				sprintf(AddStr, "Port 3: %4X", (Base & 0xFFF8) + 12);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
				found = true;
				break;
			case 0x2C70:
			case 0x2C50:
				m_CardName = "PCI-DIO-96CT Parallel Digital I/O Card";
					sprintf(AddStr, "Port 0: %4X", Base & 0xFFF8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
				sprintf(AddStr, "Port 1: %4X", (Base & 0xFFF8) + 4);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
				sprintf(AddStr, "Port 2: %4X", (Base & 0xFFF8) + 8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
				sprintf(AddStr, "Port 3: %4X", (Base & 0xFFF8) + 12);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
				found = true;
				break;
			case 0x0C50:
				m_CardName = "PCI-DIO-24H Parallel Digital I/O Card";
				sprintf(AddStr, "Port 0: %4X", Base & 0xFFF8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
				found = true;
				break;
			case 0x0C51:
				m_CardName = "PCI-DIO-24D Parallel Digital I/O Card";
				sprintf(AddStr, "Port 0: %4X", Base & 0xFFF8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
				found = true;
				break;
			case 0x0E50:
				m_CardName = "PCI-DIO-24S Parallel Digital I/O Card";
				sprintf(AddStr, "Port 0: %4X", Base & 0xFFF8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
				found = true;
				break;
			case 0x0E51:
				m_CardName = "PCI-DIO-24H(C) Parallel Digital I/O Card w/Counter";
				sprintf(AddStr, "Port 0: %4X", Base & 0xFFF8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
				found = true;
				break;
			case 0x0E52:
				m_CardName = "PCI-DIO-24D(C) Parallel Digital I/O Card w/Counter";
				sprintf(AddStr, "Port 0: %4X", Base & 0xFFF8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
				found = true;
				break;
			case 0x0C60:
				m_CardName = "PCI-DIO-48 Parallel Digital I/O Card";
					sprintf(AddStr, "Port 0: %4X", Base & 0xFFF8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
					sprintf(AddStr, "Port 1: %4X", (Base & 0xFFF8) + 4);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
				found = true;
				break;
			case 0x0E60:
				m_CardName = "PCI-DIO-48S Parallel Digital I/O Card";
					sprintf(AddStr, "Port 0: %4X", Base & 0xFFF8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
					sprintf(AddStr, "Port 1: %4X", (Base & 0xFFF8) + 4);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
				found = true;
				break;
				
			// Will cover PCIe-DIO-24 here:
			case 0x0C52:
				m_CardName = "PCIe-DIO-24 Parallel Digital I/O Card";
					sprintf(AddStr, "Port 0: %4X", Base & 0xFFF8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
					CardData.IsValid = true;
					CardData.DeviceID = DeviceID;
					CardData.Base = Base;
					DataIndex++;
				found = true;
				break;
			case 0x0E53:
				m_CardName = "PCIe-DIO-24S Parallel Digital I/O Card";
					sprintf(AddStr, "Port 0: %4X", Base & 0xFFF8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
					CardData.IsValid = true;
					CardData.DeviceID = DeviceID;
					CardData.Base = Base;
					DataIndex++;
				found = true;
				break;
			case 0x0C53:
				m_CardName = "PCIe-DIO-24D Parallel Digital I/O Card";
					sprintf(AddStr, "Port 0: %4X", Base & 0xFFF8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
					CardData.IsValid = true;
					CardData.DeviceID = DeviceID;
					CardData.Base = Base;
					DataIndex++;
				found = true;
				break;
			case 0x0E55:
				m_CardName = "PCIe-DIO-24DC Parallel Digital I/O Card w/Counter";
					sprintf(AddStr, "Port 0: %4X", Base & 0xFFF8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
					CardData.IsValid = true;
					CardData.DeviceID = DeviceID;
					CardData.Base = Base;
					DataIndex++;
				found = true;
				break;
			case 0x0E54:
				m_CardName = "PCIe-DIO-24DS Parallel Digital I/O Card";
					sprintf(AddStr, "Port 0: %4X", Base & 0xFFF8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
					CardData.IsValid = true;
					CardData.DeviceID = DeviceID;
					CardData.Base = Base;
					DataIndex++;
				found = true;
				break;
			case 0x0E56:
				m_CardName = "PCIe-DIO-24DCS Parallel Digital I/O Card w/Counter";
					sprintf(AddStr, "Port 0: %4X", Base & 0xFFF8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
					CardData.IsValid = true;
					CardData.DeviceID = DeviceID;
					CardData.Base = Base;
					DataIndex++;
				found = true;
				break;

			// Will cover PCIe-DIO-48 here:
			case 0x0C61:
				m_CardName = "PCIe-DIO-48 Parallel Digital I/O Card";
					sprintf(AddStr, "Port 0: %4X", Base & 0xFFF8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
					sprintf(AddStr, "Port 1: %4X", (Base & 0xFFF8) + 4);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
				found = true;
				break;
			case 0x0E61:
				m_CardName = "PCIe-DIO-48S Parallel Digital I/O Card";
				sprintf(AddStr, "Port 0: %4X", Base & 0xFFF8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
					sprintf(AddStr, "Port 1: %4X", (Base & 0xFFF8) + 4);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
				found = true;
				break;
			case 0x0C68:
				m_CardName = "PCI-DIO-72 Parallel Digital I/O Card";
					sprintf(AddStr, "Port 0: %4X", Base & 0xFFF8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
					sprintf(AddStr, "Port 1: %4X", (Base & 0xFFF8) + 4);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
					sprintf(AddStr, "Port 2: %4X", (Base & 0xFFF8) + 8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
				found = true;
				break;
			case 0x0C70:
				m_CardName = "PCI-DIO-96 Parallel Digital I/O Card";
					sprintf(AddStr, "Port 0: %4X", Base & 0xFFF8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
					sprintf(AddStr, "Port 1: %4X", (Base & 0xFFF8) + 4);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
					sprintf(AddStr, "Port 2: %4X", (Base & 0xFFF8) + 8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
					sprintf(AddStr, "Port 3: %4X", (Base & 0xFFF8) + 12);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
				found = true;
				break;
			case 0x0C69:
				m_CardName = "P104-DIO-96 Parallel Digital I/O Card";
					sprintf(AddStr, "Port 0: %4X", Base & 0xFFF8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
					sprintf(AddStr, "Port 1: %4X", (Base & 0xFFF8) + 4);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
					sprintf(AddStr, "Port 2: %4X", (Base & 0xFFF8) + 8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
					sprintf(AddStr, "Port 3: %4X", (Base & 0xFFF8) + 12);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
				found = true;
				break;
			case 0x0C78:
				m_CardName = "PCI-DIO-120 Parallel Digital I/O Card";
					sprintf(AddStr, "Port 0: %4X", Base & 0xFFF8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
					sprintf(AddStr, "Port 1: %4X", (Base & 0xFFF8) + 4);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
					sprintf(AddStr, "Port 2: %4X", (Base & 0xFFF8) + 8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
					sprintf(AddStr, "Port 3: %4X", (Base & 0xFFF8) + 12);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
					sprintf(AddStr, "Port 4: %4X", (Base & 0xFFF8) + 16);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
				found = true;
				break;
            case 0x2EE0:
				m_CardName = "PCIe-DIO24S-CTR12 Digital/Counter Card";
					sprintf(AddStr, "Port 0: %4X", Base & 0xFFF8);
				strname = AddStr;
				m_AddressCombo.AddString(strname);
				found = true;
				break;

			// default:
				break;
		}//end switch			
	}//end else card present

	// If card(s) are present but no valid cards were found:
	if ((NumCards != 0) && (found == false) ) 
	{	    		
		m_CardName = "No Valid Card Found.";

		m_StatusEdit =  "No valid card was found!";
		m_StatusEdit += "This may mean the card is not installed. ";
		m_StatusEdit += "Check Device Manager for a card and its status";
		m_StatusEdit += "You may consider rebooting your system.";
		m_AddressCombo.EnableWindow(false);
		m_StatusControl.LineScroll(999);

		RunFlag = false;
	}

   UpdateData(false);
   m_AddressCombo.SetCurSel(0);
   if (RunFlag) 
   {
      UpdateData(true);
	  
	  // This is the result setting the active address offset to be used:	  
	  Offset = 0;  //default to fist valid
	  CardData.IsSelected = true;
   }

}