hybridgroup.gobot/drivers/i2c/lidarlite_driver.go

80 lines
1.7 KiB
Go

package i2c
import (
"github.com/hybridgroup/gobot"
"time"
)
var _ gobot.Driver = (*LIDARLiteDriver)(nil)
const lidarliteAddress = 0x62
type LIDARLiteDriver struct {
name string
connection I2c
}
// NewLIDARLiteDriver creates a new driver with specified i2c interface
func NewLIDARLiteDriver(a I2c) *LIDARLiteDriver {
return &LIDARLiteDriver{
name: "LIDARLite",
connection: a,
}
}
func (h *LIDARLiteDriver) Name() string { return h.name }
func (h *LIDARLiteDriver) SetName(n string) { h.name = n }
func (h *LIDARLiteDriver) Connection() gobot.Connection { return h.connection.(gobot.Connection) }
// Start initialized the LIDAR
func (h *LIDARLiteDriver) Start() (err error) {
if err := h.connection.I2cStart(lidarliteAddress); 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.I2cWrite(lidarliteAddress, []byte{0x00, 0x04}); err != nil {
return
}
time.Sleep(20 * time.Millisecond)
if err = h.connection.I2cWrite(lidarliteAddress, []byte{0x0F}); err != nil {
return
}
upper, err := h.connection.I2cRead(lidarliteAddress, 1)
if err != nil {
return
}
if len(upper) != 1 {
err = ErrNotEnoughBytes
return
}
if err = h.connection.I2cWrite(lidarliteAddress, []byte{0x10}); err != nil {
return
}
lower, err := h.connection.I2cRead(lidarliteAddress, 1)
if err != nil {
return
}
if len(lower) != 1 {
err = ErrNotEnoughBytes
return
}
distance = ((int(upper[0]) & 0xff) << 8) | (int(lower[0]) & 0xff)
return
}