package mainAxle

import "time"

// 主轴正常状态
type axleOkState struct {
	speed int //转速
	FSMState
}

func NewAxleOkState() *axleOkState {
	return &axleOkState{speed: 0}
}

func (a *axleOkState) Enter(endTime time.Time) {
	a.isOk = true
	a.speed = 0
	a.startTime = time.Now()
	a.endTime = time.Now()
}

func (a *axleOkState) Exit() {
	a.isOk = false
	a.startTime = time.Now()
}

func (a *axleOkState) V() int {
	return a.speed
}

func (a *axleOkState) CheckTransition() bool {
	//如果为正常状态，或者非正常状态且结束时间没到
	if a.isOk  || a.endTime.After(time.Now()) {
		return false
	}
	return true
}

// 主轴停止状态（非正常状态）
type axleStopState struct {
	speed int //转速
	FSMState
}

func NewAxleStopState() *axleStopState {
	return &axleStopState{speed: 0}
}

func (a *axleStopState) Enter(endTime time.Time) {
	a.isOk = false
	a.speed = 0
	a.startTime = time.Now()
	a.endTime = endTime
}

func (a *axleStopState) Exit() {
	a.isOk = true
	a.startTime = time.Now()
}

func (a *axleStopState) V() int {
	return a.speed
}

func (a *axleStopState) CheckTransition() bool {
	//如果为正常状态，或者非正常状态且结束时间没到
	if !a.isOk && a.endTime.After(time.Now()) {
		return false
	}
	return true
}