Add rgb driver

Author: xanecs

driver/direct.go

@@ -30,13 +30,13 @@ func (d *directDriver) write() error {
 
 func (d *directDriver) HandleMessage(action string, params map[string]interface{}) error {
 	switch action {
-	case "on":
+	case cmdOn:
 		d.state = true
 
-	case "off":
+	case cmdOff:
 		d.state = false
 
-	case "write":
+	case cmdWrite:
 		val := params["power"]
 		if val == nil {
 			return errors.New("Missing parameter 'power'")

driver/driver.go

@@ -5,6 +5,13 @@ import (
 
 	"github.com/xanecs/lighthouse/config"
 	"gobot.io/x/gobot"
+	"gobot.io/x/gobot/drivers/gpio"
+)
+
+const (
+	cmdOn    = "on"
+	cmdOff   = "off"
+	cmdWrite = "write"
 )
 
 // Device represents a device that can handle messages
@@ -23,6 +30,13 @@ func NewDriver(cfg config.DeviceConfig, connection gobot.Connection) (Device, er
 	case "pwm":
 		return newPwmDriver(cfg, connection)
 
+	case "rgb":
+		dd, ok := connection.(gpio.DigitalWriter)
+		if !ok {
+			return nil, errors.New("rgb mode is not supported on this board")
+		}
+		return newRgbDriver(cfg, dd)
+
 	default:
 		return nil, errors.New("Invalid mode")
 	}

driver/pwm.go

@@ -18,6 +18,9 @@ type pwmDriver struct {
 
 func (p *pwmDriver) write() error {
 	if !p.power {
+		if p.inverted {
+			return p.driver.PwmWrite(255)
+		}
 		return p.driver.Off()
 	}
 	brightness := p.brightness
@@ -29,10 +32,10 @@ func (p *pwmDriver) write() error {
 
 func (p *pwmDriver) HandleMessage(action string, params map[string]interface{}) error {
 	switch action {
-	case "on":
+	case cmdOn:
 		p.power = true
 
-	case "off":
+	case cmdOff:
 		p.power = false
 
 	case "brightness":
@@ -46,7 +49,7 @@ func (p *pwmDriver) HandleMessage(action string, params map[string]interface{})
 		}
 		p.brightness = uint8(255 * brightness)
 
-	case "write":
+	case cmdWrite:
 		val := params["power"]
 		if val == nil {
 			return errors.New("Missing parameter 'power'")

driver/rgb.go

@@ -0,0 +1,112 @@
+package driver
+
+import (
+	"errors"
+
+	"github.com/xanecs/lighthouse/config"
+
+	"gobot.io/x/gobot"
+	"gobot.io/x/gobot/drivers/gpio"
+)
+
+type rgbDriver struct {
+	driver   *gpio.RgbLedDriver
+	inverted bool
+	power    bool
+	color    color
+}
+
+type color struct {
+	red   byte
+	green byte
+	blue  byte
+}
+
+func (r *rgbDriver) write() error {
+	if !r.power {
+		if r.inverted {
+			return r.driver.SetRGB(255, 255, 255)
+		}
+		return r.driver.Off()
+	}
+	if r.inverted {
+		return r.driver.SetRGB(255-r.color.red, 255-r.color.green, 255-r.color.blue)
+	}
+	return r.driver.SetRGB(r.color.red, r.color.green, r.color.blue)
+}
+
+func (r *rgbDriver) HandleMessage(action string, params map[string]interface{}) error {
+	switch action {
+	case cmdOn:
+		r.power = true
+
+	case cmdOff:
+		r.power = false
+
+	case "color":
+		newColor, err := parseColor(params)
+		if err != nil {
+			return err
+		}
+		r.color = newColor
+
+	case cmdWrite:
+		val := params["power"]
+		if val == nil {
+			return errors.New("Missing parameter 'power'")
+		}
+		power, ok := val.(bool)
+		if !ok {
+			return errors.New("Invalid parameter 'power'")
+		}
+
+		newColor, err := parseColor(params)
+		if err != nil {
+			return err
+		}
+
+		r.power = power
+		r.color = newColor
+	}
+	return r.write()
+}
+
+func (r *rgbDriver) Status() map[string]interface{} {
+	return map[string]interface{}{"power": r.power, "color": r.color}
+}
+
+func (r *rgbDriver) Driver() gobot.Device {
+	return r.driver
+}
+
+func parseColor(params map[string]interface{}) (color, error) {
+	valR := params["red"]
+	valG := params["green"]
+	valB := params["blue"]
+
+	if valR == nil || valG == nil || valB == nil {
+		return color{}, errors.New("Missing color parameters")
+	}
+
+	red, okR := valR.(byte)
+	green, okG := valG.(byte)
+	blue, okB := valB.(byte)
+
+	if !(okR && okG && okB) {
+		return color{}, errors.New("Invalid color parameters")
+	}
+
+	return color{red, green, blue}, nil
+}
+
+func newRgbDriver(cfg config.DeviceConfig, connection gpio.DigitalWriter) (*rgbDriver, error) {
+	if len(cfg.Pins) != 3 {
+		return nil, errors.New("Invalid number of pins")
+	}
+	return &rgbDriver{
+		driver:   gpio.NewRgbLedDriver(connection, cfg.Pins[0], cfg.Pins[1], cfg.Pins[2]),
+		inverted: cfg.Inverted,
+		power:    false,
+		color:    color{255, 255, 255},
+	}, nil
+}