Skip to content
Permalink
Branch: master
Find file Copy path
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
113 lines (93 sloc) 2.62 KB
package digispark
import (
"errors"
"fmt"
"strconv"
"gobot.io/x/gobot"
"gobot.io/x/gobot/drivers/i2c"
)
// ErrConnection is the error resulting of a connection error with the digispark
var ErrConnection = errors.New("connection error")
// Adaptor is the Gobot Adaptor for the Digispark
type Adaptor struct {
name string
littleWire lw
servo bool
pwm bool
i2c bool
connect func(*Adaptor) (err error)
}
// NewAdaptor returns a new Digispark Adaptor
func NewAdaptor() *Adaptor {
return &Adaptor{
name: gobot.DefaultName("Digispark"),
connect: func(d *Adaptor) (err error) {
d.littleWire = littleWireConnect()
if d.littleWire.(*littleWire).lwHandle == nil {
return ErrConnection
}
return
},
}
}
// Name returns the Digispark Adaptors name
func (d *Adaptor) Name() string { return d.name }
// SetName sets the Digispark Adaptors name
func (d *Adaptor) SetName(n string) { d.name = n }
// Connect starts a connection to the digispark
func (d *Adaptor) Connect() (err error) {
err = d.connect(d)
return
}
// Finalize implements the Adaptor interface
func (d *Adaptor) Finalize() (err error) { return }
// DigitalWrite writes a value to the pin. Acceptable values are 1 or 0.
func (d *Adaptor) DigitalWrite(pin string, level byte) (err error) {
p, err := strconv.Atoi(pin)
if err != nil {
return
}
if err = d.littleWire.pinMode(uint8(p), 0); err != nil {
return
}
return d.littleWire.digitalWrite(uint8(p), level)
}
// PwmWrite writes the 0-254 value to the specified pin
func (d *Adaptor) PwmWrite(pin string, value byte) (err error) {
if !d.pwm {
if err = d.littleWire.pwmInit(); err != nil {
return
}
if err = d.littleWire.pwmUpdatePrescaler(1); err != nil {
return
}
d.pwm = true
}
return d.littleWire.pwmUpdateCompare(value, value)
}
// ServoWrite writes the 0-180 degree val to the specified pin.
func (d *Adaptor) ServoWrite(pin string, angle uint8) (err error) {
if !d.servo {
if err = d.littleWire.servoInit(); err != nil {
return
}
d.servo = true
}
return d.littleWire.servoUpdateLocation(angle, angle)
}
// GetConnection returns an i2c connection to a device on a specified bus.
// Only supports bus number 0
func (d *Adaptor) GetConnection(address int, bus int) (connection i2c.Connection, err error) {
if bus != 0 {
return nil, fmt.Errorf("Invalid bus number %d, only 0 is supported", bus)
}
c := NewDigisparkI2cConnection(d, uint8(address))
if err := c.Init(); err != nil {
return nil, err
}
return i2c.Connection(c), nil
}
// GetDefaultBus returns the default i2c bus for this platform
func (d *Adaptor) GetDefaultBus() int {
return 0
}
You can’t perform that action at this time.