hybridgroup.gobot/drivers/i2c/lidarlite_driver.go

108 lines
2.3 KiB
Go

package i2c
import (
"gobot.io/x/gobot"
"time"
)
const lidarliteAddress = 0x62
// LIDARLiteDriver is the Gobot driver for the LIDARLite I2C LIDAR device.
type LIDARLiteDriver struct {
name string
connector Connector
connection Connection
Config
}
// NewLIDARLiteDriver creates a new driver for the LIDARLite I2C LIDAR device.
//
// Params:
// conn Connector - the Adaptor to use with this Driver
//
// Optional params:
// i2c.WithBus(int): bus to use with this driver
// i2c.WithAddress(int): address to use with this driver
//
func NewLIDARLiteDriver(a Connector, options ...func(Config)) *LIDARLiteDriver {
l := &LIDARLiteDriver{
name: gobot.DefaultName("LIDARLite"),
connector: a,
Config: NewConfig(),
}
for _, option := range options {
option(l)
}
// TODO: add commands to API
return l
}
// Name returns the Name for the Driver
func (h *LIDARLiteDriver) Name() string { return h.name }
// SetName sets the Name for the Driver
func (h *LIDARLiteDriver) SetName(n string) { h.name = n }
// Connection returns the connection for the Driver
func (h *LIDARLiteDriver) Connection() gobot.Connection { return h.connector.(gobot.Connection) }
// Start initialized the LIDAR
func (h *LIDARLiteDriver) Start() (err error) {
bus := h.GetBusOrDefault(h.connector.GetDefaultBus())
address := h.GetAddressOrDefault(lidarliteAddress)
h.connection, err = h.connector.GetConnection(address, bus)
if err != nil {
return err
}
return
}
// Halt returns true if devices is halted successfully
func (h *LIDARLiteDriver) Halt() (err error) { return }
// Distance returns the current distance in cm
func (h *LIDARLiteDriver) Distance() (distance int, err error) {
if _, err = h.connection.Write([]byte{0x00, 0x04}); err != nil {
return
}
time.Sleep(20 * time.Millisecond)
if _, err = h.connection.Write([]byte{0x0F}); err != nil {
return
}
upper := []byte{0}
bytesRead, err := h.connection.Read(upper)
if err != nil {
return
}
if bytesRead != 1 {
err = ErrNotEnoughBytes
return
}
if _, err = h.connection.Write([]byte{0x10}); err != nil {
return
}
lower := []byte{0}
bytesRead, err = h.connection.Read(lower)
if err != nil {
return
}
if bytesRead != 1 {
err = ErrNotEnoughBytes
return
}
distance = ((int(upper[0]) & 0xff) << 8) | (int(lower[0]) & 0xff)
return
}