View Code? Open in Web Editor NEW

This project forked from omaskery/fsm-rs

A simple Finite State Machine library in Rust, define a couple of enums, construct a machine, define some transition behaviours and voila!

License: MIT License

Rust 100.00%

fsm-rs's Introduction

A simple Finite State Machine library in Rust. Provide State and Event types (usually enums), then generate a machine with an initial state, give it some transition behaviours and you have your state machine!

Usage

Using the simple coin-operated turnstyle example from the FSM wikipedia entry:

Define your states and events:

// states and events must be C-like enums (copyable and easily converted to primitives)
#[derive(Copy, Clone)]
enum TurnStyleState {
	Locked,
	Unlocked,
}

#[derive(Copy, Clone)]
enum TurnStyleEvent {
	Push,
	InsertCoin,
}

// implement the EnumTag trait for states and events
impl EnumTag for TurnStyleState {
	fn tag_number(&self) -> usize {
		*self as usize
	}
	fn max_tag_number() -> usize {
		TurnStyleState::Unlocked as usize
	}
}

impl EnumTag for TurnStyleEvent {
	fn tag_number(&self) -> usize {
		*self as usize
	}
	fn max_tag_number() -> usize {
		TurnStyleEvent::InsertCoin as usize
	}
}

Create your machine and define your transitions:

// create the machine initially in the Locked state
let mut machine = Machine::new(TurnStyleState::Locked);
// create the transition from Locked -> Unlocked upon inserting a coin
machine.add_transition(
	TurnStyleState::Locked, TurnStyleEvent::InsertCoin,
	TurnStyleState::Unlocked, |_,_| println!("unlocked")
);
// create the transition from Unlocked -> Locked upon pushing the turnstyle
machine.add_transition(
	TurnStyleState::Unlocked, TurnStyleEvent::Push,
	TurnStyleState::Locked, |_,_| println!("locked")
);

Trigger events as needed and huzzah, off you go:

// initially we're in the Locked state
machine.on_event(TurnStyleEvent::InsertCoin);
// now we're Unlocked, ("unlocked" was just printed)
machine.on_event(TurnStyleEvent::Push);
// now we're Locked again, ("locked" was just printed)