WuDevice

from twisted.internet import reactor
from udpwkpf import WuClass, Device
import sys
from udpwkpf_io_interface import *

Button_Pin = 2
Light_Actuator_Pin = 4
Smoke_Sensor_Pin = 7
R_LED_Pin = 3
G_LED_Pin = 5
B_LED_Pin = 6
Magnetic_Pin = 8
Slider_Pin = 14
Light_Sensor_Pin = 15

class Button(WuClass):
    def __init__(self):
        WuClass.__init__(self)
        self.loadClass('Button')
        self.IO = pin_mode(Button_Pin, PIN_TYPE_DIGITAL, PIN_MODE_INPUT)
        self.prev_value = -1

    def update(self,obj,pID=None,val=None):
        try:
            current_value = digital_read(self.IO)
            if current_value != self.prev_value:
                self.prev_value = current_value
                print "Button value: %d" % current_value
                obj.setProperty(0, current_value)
        except IOError:
            print "Error"

class Light_Actuator(WuClass):
    def __init__(self):
        WuClass.__init__(self)
        self.loadClass('Light_Actuator')
        self.light_actuator_gpio = pin_mode(Light_Actuator_Pin, PIN_TYPE_DIGITAL, PIN_MODE_OUTPUT)

    def update(self,obj,pID=None,val=None):
        try:
            if pID == 0:
                if val == True:
                    digital_write(self.light_actuator_gpio, 1)
                    print "Light Actuator On"
                else:
                    digital_write(self.light_actuator_gpio, 0)
                    print "Light Actuator Off"
        except IOError:
            print ("Error")

class Smoke_Sensor(WuClass):
    def __init__(self):
        WuClass.__init__(self)
        self.loadClass('Smoke_Sensor')
        self.IO = pin_mode(Smoke_Sensor_Pin, PIN_TYPE_DIGITAL, PIN_MODE_INPUT)
        self.prev_value = -1

    def update(self,obj,pID=None,val=None):
        try:
            current_value = digital_read(self.IO)
            if current_value != self.prev_value:
                self.prev_value = current_value
                print "Smoke sensor value: %d" % current_value
                obj.setProperty(0, current_value)
        except IOError:
            print "Error"

class RGB_LED(WuClass):
    def __init__(self):
        WuClass.__init__(self)
        self.loadClass('RGB_LED')
        self.r_led_gpio = pin_mode(R_LED_Pin, PIN_TYPE_DIGITAL, PIN_MODE_OUTPUT)
        self.g_led_gpio = pin_mode(G_LED_Pin, PIN_TYPE_DIGITAL, PIN_MODE_OUTPUT)
        self.b_led_gpio = pin_mode(B_LED_Pin, PIN_TYPE_DIGITAL, PIN_MODE_OUTPUT)

    def update(self,obj,pID=None,val=None):
        try:
            if pID == 0:
                analog_write(self.r_led_gpio, val)
                print "set R:", val
            elif pID == 1:
                analog_write(self.g_led_gpio, val)
                print "set G:", val
            elif pID == 2:
                analog_write(self.b_led_gpio, val)
                print "set B:", val
        except IOError:
            print ("Error")

class Magnetic_Sensor(WuClass):
    def __init__(self):
        WuClass.__init__(self)
        self.loadClass('Magnetic_Sensor')
        self.IO = pin_mode(Magnetic_Pin, PIN_TYPE_DIGITAL, PIN_MODE_INPUT)

    def update(self,obj,pID=None,val=None):
        try:
            current_value = digital_read(self.IO)
            print "Magnetic value: %d" % current_value
            obj.setProperty(0, current_value)
        except IOError:
            print "Error"

class Slider(WuClass):
    def __init__(self):
        WuClass.__init__(self)
        self.loadClass('Slider')
        self.slider_aio = pin_mode(Slider_Pin, PIN_TYPE_ANALOG)
        print "Slider init success"
        self.prev_value = -1

    def update(self,obj,pID=None,val=None):
        try:
            current_value = analog_read(self.slider_aio)/4 #4 is divisor value which depends on the slider.
            if current_value != self.prev_value:
                self.prev_value = current_value
                obj.setProperty(0, current_value)
                print "Slider analog pin: ", Slider_Pin, ", value: ", current_value
        except IOError:
            print ("Error")

class Light_Sensor(WuClass):
    def __init__(self):
        WuClass.__init__(self)
        self.loadClass('Light_Sensor')
        self.light_sensor_aio = pin_mode(Light_Sensor_Pin, PIN_TYPE_ANALOG)
        self.prev_value = -1

    def update(self,obj,pID=None,val=None):
        try:
            current_value = analog_read(self.light_sensor_aio)/4 # 4 is divisor value which depends on the light sensor
            if current_value != self.prev_value:
                self.prev_value = current_value
                obj.setProperty(0, current_value)
                print "Light sensor analog pin: ", Light_Sensor_Pin, ", value: ", current_value
        except IOError:
            print ("Error")

if __name__ == "__main__":
    class MyDevice(Device):
        def __init__(self,addr,localaddr):
            Device.__init__(self,addr,localaddr)

        def init(self):
            m = Button()
            self.addClass(m,0)
            self.addObject(m.ID)

            m2 = Light_Actuator()
            self.addClass(m2,0)
            self.addObject(m2.ID)

            m3 = Smoke_Sensor()
            self.addClass(m3,0)
            self.addObject(m3.ID)

            m4 = RGB_LED()
            self.addClass(m4,0)
            self.addObject(m4.ID)

            m5 = Magnetic_Sensor()
            self.addClass(m5,0)
            self.addObject(m5.ID)

            m6 = Slider()
            self.addClass(m6,0)
            self.addObject(m6.ID)

            m7 = Light_Sensor()
            self.addClass(m7,0)
            self.addObject(m7.ID)

    if len(sys.argv) <= 2:
        print 'python %s <gip> <dip>:<port>' % sys.argv[0]
        print '      <gip>: IP addrees of gateway'
        print '      <dip>: IP address of Python device'
        print '      <port>: An unique port number'
        print ' ex. python %s 192.168.4.7 127.0.0.1:3000' % sys.argv[0]
        sys.exit(-1)

    d = MyDevice(sys.argv[1],sys.argv[2])
    reactor.run()
    device_cleanup()