FSM/FSM_OOP/baseFSM/FSM.c

129 lines
3.5 KiB
C

#include <stdlib.h>
#include <stdio.h>
#include <assert.h>
#include "FSM_private.h"
#include "FSM_protected.h"
#include "FSM_public.h"
/*
protected 函数
*/
void resetBaseFSM(FSM *pFSM)
{
if (pFSM->privateVars.fcns.exitActionTable[pFSM->privateVars.curState] != NULL)
pFSM->privateVars.fcns.exitActionTable[pFSM->privateVars.curState](pFSM);
pFSM->privateVars.curState = 0;
}
void stepBaseFSM(FSM *pFSM)
{
int index = pFSM->privateVars.curState * pFSM->privateVars.numEvent + pFSM->privateVars.curEvent;
if (pFSM->privateVars.fcns.transitionTable[index] != NULL)
{
if (pFSM->privateVars.fcns.exitActionTable[pFSM->privateVars.curState] != NULL)
pFSM->privateVars.fcns.exitActionTable[pFSM->privateVars.curState](pFSM);
pFSM->privateVars.curState = (*pFSM->privateVars.fcns.transitionTable[index])(pFSM);
if (pFSM->privateVars.fcns.enterActionTable[pFSM->privateVars.curState] != NULL)
pFSM->privateVars.fcns.enterActionTable[pFSM->privateVars.curState](pFSM);
}
else if(pFSM->privateVars.curState == 0){
pFSM->privateVars.curState = pFSM->privateVars.defaultState;
if (pFSM->privateVars.fcns.enterActionTable[pFSM->privateVars.curState] != NULL)
pFSM->privateVars.fcns.enterActionTable[pFSM->privateVars.curState](pFSM);
}
else
{
if (pFSM->privateVars.fcns.duringActionTable[pFSM->privateVars.curState] != NULL)
pFSM->privateVars.fcns.duringActionTable[pFSM->privateVars.curState](pFSM);
}
}
FSM* newBaseFSM(int numState, int numEvent, int defaultState){
typedef void (*StateFuncPtr)(FSM* pFSM);
typedef int (*TransitionFuncPtr)(FSM* pFSM);
FSM *pFSM = calloc(1, sizeof(FSM));
pFSM->privateVars.curEvent = 0;
pFSM->privateVars.curState = 0;
pFSM->privateVars.numEvent = numEvent;
pFSM->privateVars.defaultState = defaultState;
pFSM->privateVars.numChild = 0;
pFSM->privateVars.fcns.duringActionTable = calloc(numState, sizeof(StateFuncPtr));
pFSM->privateVars.fcns.enterActionTable = calloc(numState, sizeof(StateFuncPtr));
pFSM->privateVars.fcns.exitActionTable = calloc(numState, sizeof(StateFuncPtr));
pFSM->privateVars.fcns.transitionTable = calloc(numState * numEvent, sizeof(TransitionFuncPtr));
FSMVtbl vtbl = {
.reset = resetBaseFSM,
.step = stepBaseFSM,
.selectEvent = NULL,
.setupHandler = NULL,
.initData = NULL,
};
pFSM->vtbl = vtbl;
return pFSM;
}
void constructFSM(FSM* pFSM){
assert(pFSM->vtbl.setupHandler);
assert(pFSM->vtbl.initData);
pFSM->vtbl.setupHandler(&pFSM->privateVars.fcns);
pFSM->vtbl.initData(pFSM);
}
/*
public 函数
*/
int getFSMCurState(FSM *pFSM){
return pFSM->privateVars.curState;
}
int getFSMCurEvent(FSM *pFSM){
return pFSM->privateVars.curEvent;
}
void setEvent(FSM* pFSM){
assert(pFSM->vtbl.selectEvent);
pFSM->privateVars.curEvent = pFSM->vtbl.selectEvent(pFSM->data);
if(0 == pFSM->privateVars.numChild){
return;
}
for (int i = 0; i < pFSM->privateVars.numChild; i++)
{
FSM *cFSM = pFSM->privateVars.childFSM[i];
setEvent(cFSM);
}
}
void setChildNum(FSM *pFSM, int num){
pFSM->privateVars.numChild = num;
pFSM->privateVars.childFSM = (FSM **)calloc(num, sizeof(FSM *));
}
void registerChildFSM(FSM *parent, FSM *child, int index){
parent->privateVars.childFSM[index] = child;
}
void *getData(FSM* pFSM){
return pFSM->data;
}
FSMVtbl *vptrFSM(FSM* pFSM){
return &pFSM->vtbl;
}