hybridgroup.gobot/platforms/mavlink/mavlink_adaptor.go

65 lines
1.4 KiB
Go

package mavlink
import (
"io"
"github.com/tarm/serial"
"gobot.io/x/gobot"
common "gobot.io/x/gobot/platforms/mavlink/common"
)
// Adaptor is a Mavlink transport adaptor.
type BaseAdaptor interface {
gobot.Connection
io.Writer
ReadMAVLinkPacket() (*common.MAVLinkPacket, error)
}
// Adaptor is a Mavlink-over-serial adaptor.
type Adaptor struct {
name string
port string
sp io.ReadWriteCloser
connect func(string) (io.ReadWriteCloser, error)
}
// NewAdaptor creates a new mavlink adaptor with specified port
func NewAdaptor(port string) *Adaptor {
return &Adaptor{
name: "Mavlink",
port: port,
connect: func(port string) (io.ReadWriteCloser, error) {
return serial.OpenPort(&serial.Config{Name: port, Baud: 57600})
},
}
}
func (m *Adaptor) Name() string { return m.name }
func (m *Adaptor) SetName(n string) { m.name = n }
func (m *Adaptor) Port() string { return m.port }
// Connect returns true if connection to device is successful
func (m *Adaptor) Connect() (err error) {
if sp, e := m.connect(m.Port()); e != nil {
return e
} else {
m.sp = sp
}
return
}
// Finalize returns true if connection to devices is closed successfully
func (m *Adaptor) Finalize() (err error) {
err = m.sp.Close()
return
}
func (m *Adaptor) ReadMAVLinkPacket() (*common.MAVLinkPacket, error) {
return common.ReadMAVLinkPacket(m.sp)
}
func (m *Adaptor) Write(b []byte) (int, error) {
return m.sp.Write(b)
}