main.cpp

This code supplements the tutorial in Finite State Machines

/****************************************
*	Author:		Nathaniel Meyer			*
*	E-Mail:		[e-mail]	*
*	Website:	http://www.nutty.ca		*
*										*
*   You are free to use, redistribute,  *
*   and alter this file in anyway, so   *
*   long as credit is given where due.	*
****************************************/


#include "Main.h"
#include "resource.h"


// Global Variables
HWND gHWnd;

char selEvent[50],
	 lastState[50];
unsigned int eventID;


/******************
	STATE ACTIONS
******************/
// State: TurnedOn: On Entry
void sIdle_OnExit (char *args)
{
	SetDlgItemText(gHWnd, IDC_RESPONSE, "Powering up...");
	Sleep(1000);	// Give some time for the user to read the response
}

void sIdle_TurnOff (char *args)
{
	SetDlgItemText(gHWnd, IDC_RESPONSE, "Bender has shut down");
}

// State: TurnedOn: On Entry
void sTurnedOn_OnEntry (char *args)
{
	SetDlgItemText(gHWnd, IDC_RESPONSE, "Bender is now online");
}

// State: Activity: On Event: Walk
void sActivity_Walk (char *args)
{
	SetDlgItemText(gHWnd, IDC_RESPONSE, "Activity: Walking");
}

// State: Activity: On Event: Run
void sActivity_Run (char *args)
{
	SetDlgItemText(gHWnd, IDC_RESPONSE, "Activity: Running");
}

// State: Activity: On Event: Talk
void sActivity_Talk (char *args)
{
	SetDlgItemText(gHWnd, IDC_RESPONSE, "Activity: Talking");
}



void init ()
{
	// Idle state
	cState[0].setName("Idle");
	cState[0].addAction(2, 0, "Action: TurnOn", sIdle_OnExit);
	cState[0].addAction(3, 0, "Action: TurnOff", "TurnOff", sIdle_TurnOff);
	cState[0].addTransition("TurnOn", &cState[1]);

	// TurnedOn state
	cState[1].setName("TurnedOn");
	cState[1].addAction(0, 0, "Action: TurnOn", sTurnedOn_OnEntry);
	cState[1].addTransition("TurnOff", &cState[0]);
	cState[1].addTransition("GoToActivity", &cState[2]);

	// Activity state
	cState[2].setName("Activity");
	cState[2].addAction(3, 0, "Action: Walk", "Walk", sActivity_Walk);
	cState[2].addAction(3, 0, "Action: Run", "Run", sActivity_Run);
	cState[2].addAction(3, 0, "Action: Talk", "Talk", sActivity_Talk);
	cState[2].addTransition("RequestTurnOff", &cState[1]);
	cState[2].addTransition("Walk", &cState[2]);
	cState[2].addTransition("Run", &cState[2]);
	cState[2].addTransition("Talk", &cState[2]);

	// Add the states to the FSM controller
	cFSM.addStates(cState);
}

/* CarFSM Dialog Box */
BOOL CALLBACK MainDlgProc(HWND hWnd, UINT Message, WPARAM wParam, LPARAM lParam)
{
	switch(Message)
	{
		case WM_INITDIALOG:
			gHWnd = hWnd;

			// Initialize states and FSM
			init();

			strcpy(selEvent, "");

			// Set combo box events
			SendDlgItemMessage(hWnd, IDC_EVENTLIST, CB_ADDSTRING, 0, (LPARAM) (LPCTSTR)"TurnOn");
			SendDlgItemMessage(hWnd, IDC_EVENTLIST, CB_ADDSTRING, 0, (LPARAM) (LPCTSTR)"TurnOff");
			SendDlgItemMessage(hWnd, IDC_EVENTLIST, CB_ADDSTRING, 0, (LPARAM) (LPCTSTR)"GoToActivity");
			SendDlgItemMessage(hWnd, IDC_EVENTLIST, CB_ADDSTRING, 0, (LPARAM) (LPCTSTR)"Walk");
			SendDlgItemMessage(hWnd, IDC_EVENTLIST, CB_ADDSTRING, 0, (LPARAM) (LPCTSTR)"Run");
			SendDlgItemMessage(hWnd, IDC_EVENTLIST, CB_ADDSTRING, 0, (LPARAM) (LPCTSTR)"Talk");
			SendDlgItemMessage(hWnd, IDC_EVENTLIST, CB_ADDSTRING, 0, (LPARAM) (LPCTSTR)"RequestTurnOff");

			// Set default titles
			SetDlgItemText(hWnd, IDC_LASTSTATE, "none");
			SetDlgItemText(hWnd, IDC_CURRENTSTATE, "Idle");
			SetDlgItemText(hWnd, IDC_STATUS, "Ready");
			SetDlgItemText(hWnd, IDC_RESPONSE, "none");
		break;
		
		case WM_CLOSE:
			EndDialog(hWnd, 0);
			return 0;
		break;
		
		case WM_DESTROY:
			EndDialog(hWnd, 0);
			return 0;
		break;

		case WM_COMMAND:
			switch(LOWORD(wParam))
			{
				case IDC_EVENTLIST:
					switch(HIWORD(wParam))
					{
						case CBN_SELCHANGE:
							// Get the selected event
							eventID = SendDlgItemMessage(hWnd, IDC_EVENTLIST, CB_GETCURSEL, 0, 0);
							SendDlgItemMessage(hWnd, IDC_EVENTLIST, CB_GETLBTEXT, eventID, (LPARAM)(LPCTSTR)selEvent);
						break;
					}
				break;

				case IDC_SENDEVENT:
					/*
						Validate the event
						If it exists then process it, otherwise report an unknown event
					*/
					if ( strcmp(selEvent, "") != 0 )
					{
						// args is left blank, optionally you can add whatever you like
						if ( cFSM.outEvent(selEvent, "") )
						{
							GetDlgItemText(hWnd, IDC_CURRENTSTATE, lastState, 50);
							SetDlgItemText(hWnd, IDC_LASTSTATE, lastState);
							SetDlgItemText(hWnd, IDC_CURRENTSTATE, cFSM.getCurrentStateName());
							SetDlgItemText(hWnd, IDC_STATUS, selEvent);
						}
						else
						{
							SetDlgItemText(hWnd, IDC_STATUS, "Unknown Event");
						}
					}
				break;
			}
		break;
	}
	return FALSE;
}

int APIENTRY WinMain (HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow)
{
	// Display the dialog box
	DialogBox(hInstance, MAKEINTRESOURCE(IDD_MAIN), NULL, MainDlgProc);

	return FALSE;
}