hybridgroup.gobot/platforms/neurosky/neurosky_adaptor.go

package neurosky

import (
	"io"

	"github.com/hybridgroup/gobot"
	"github.com/tarm/goserial"
)

var _ gobot.AdaptorInterface = (*NeuroskyAdaptor)(nil)

type NeuroskyAdaptor struct {
	gobot.Adaptor
	sp      io.ReadWriteCloser
	connect func(*NeuroskyAdaptor) (err error)
}

// NewNeuroskyAdaptor creates a neurosky adaptor with specified name
func NewNeuroskyAdaptor(name string, port string) *NeuroskyAdaptor {
	return &NeuroskyAdaptor{
		Adaptor: *gobot.NewAdaptor(
			name,
			"NeuroskyAdaptor",
			port,
		),
		connect: func(n *NeuroskyAdaptor) (err error) {
			sp, err := serial.OpenPort(&serial.Config{Name: n.Port(), Baud: 57600})
			if err != nil {
				return err
			}
			n.sp = sp
			return
		},
	}
}

// Connect returns true if connection to device is successful
func (n *NeuroskyAdaptor) Connect() (errs []error) {
	if err := n.connect(n); err != nil {
		return []error{err}
	}
	return
}

// Finalize returns true if device finalization is successful
func (n *NeuroskyAdaptor) Finalize() (errs []error) {
	if err := n.sp.Close(); err != nil {
		return []error{err}
	}
	return
}