mycroft-core/mycroft/client/enclosure/eyes.py

103 lines
3.3 KiB
Python

# Copyright 2016 Mycroft AI, Inc.
#
# This file is part of Mycroft Core.
#
# Mycroft Core is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Mycroft Core is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Mycroft Core. If not, see <http://www.gnu.org/licenses/>.
from mycroft.util.log import getLogger
__author__ = 'jdorleans'
LOGGER = getLogger(__name__)
class EnclosureEyes:
"""
Listens to enclosure commands for Mycroft's Eyes.
Performs the associated command on Arduino by writing on the Serial port.
"""
def __init__(self, ws, writer):
self.ws = ws
self.writer = writer
self.__init_events()
def __init_events(self):
self.ws.on('enclosure.eyes.on', self.on)
self.ws.on('enclosure.eyes.off', self.off)
self.ws.on('enclosure.eyes.blink', self.blink)
self.ws.on('enclosure.eyes.narrow', self.narrow)
self.ws.on('enclosure.eyes.look', self.look)
self.ws.on('enclosure.eyes.color', self.color)
self.ws.on('enclosure.eyes.level', self.brightness)
self.ws.on('enclosure.eyes.volume', self.volume)
self.ws.on('enclosure.eyes.spin', self.spin)
self.ws.on('enclosure.eyes.timedspin', self.timed_spin)
self.ws.on('enclosure.eyes.reset', self.reset)
def on(self, event=None):
self.writer.write("eyes.on")
def off(self, event=None):
self.writer.write("eyes.off")
def blink(self, event=None):
side = "b"
if event and event.data:
side = event.data.get("side", side)
self.writer.write("eyes.blink=" + side)
def narrow(self, event=None):
self.writer.write("eyes.narrow")
def look(self, event=None):
if event and event.data:
side = event.data.get("side", "")
self.writer.write("eyes.look=" + side)
def color(self, event=None):
r, g, b = 255, 255, 255
if event and event.data:
r = int(event.data.get("r"), r)
g = int(event.data.get("g"), g)
b = int(event.data.get("b"), b)
color = (r * 65536) + (g * 256) + b
self.writer.write("eyes.color=" + str(color))
def brightness(self, event=None):
level = 30
if event and event.data:
level = event.data.get("level", level)
self.writer.write("eyes.level=" + str(level))
def volume(self, event=None):
volume = 4
if event and event.data:
volume = event.data.get("volume", volume)
self.writer.write("eyes.volume=" + str(volume))
def reset(self, event=None):
self.writer.write("eyes.reset")
def spin(self, event=None):
self.writer.write("eyes.spin")
def timed_spin(self, event=None):
length = 5000
if event and event.data:
length = event.data.get("length", length)
self.writer.write("eyes.spin=" + str(length))