package ptn

import (
	ta "github.com/banbox/banta"
)

func GetMains(e *ta.BarEnv, i int) (float64, float64, float64, float64) {
	o, h, l, c := e.Open.Get(i), e.High.Get(i), e.Low.Get(i), e.Close.Get(i)
	return o, h, l, c
}

type FuncPtnRecognize = func(e *ta.BarEnv) *PtnRes

var (
	recognizers = []FuncPtnRecognize{
		isEngulfing,
	}
)

func Recognize(e *ta.BarEnv) *PtnRes {
	for _, reg := range recognizers {
		res := reg(e)
		if res != nil {
			return res
		}
	}
	return nil
}