#############################################################################
## DreamScreen Commander
## By Taurean Dyer
## www.taureandyer.com
## v 1.01
## Based on example code from Dreamscreen SDK Docs and inspired by avwuff's DreamScreenControl.cs
## (use avwff's if you just want a C# app: https://github.com/avwuff/DreamScreenControl)
##
## modified 2021 by NeoRame
## update to v 1.1 (color mode and groups added)

import socket
import sys
import optparse
import time

##TODO
# Add colors chart and the ability to read from chart and put it in there.  Maybe have dec, hex, and human readible options


#####Edit these values as needed###############################################
#IP number here (see readme).  Touch anything else at your own risk!
#IP = "<enter IP here>" ##Put your IP address here
#####End Editabe Values########################################################

### NO TOUCHIE FROM HERE DOWN!!!!##############################################
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) # INTERNET, UDP datagram
crcTable = [0, 7, 4, 9, 28, 27, 18, 21, 56, 63, 54, 49, 36, 35, 42, 45, 112, 119, 126, 121, 108, 107, 98, 101, 72, 79, 70, 65, 84, 83, 90, 93, 224, 231, 238, 233, 252, 251, 242, 245, 216, 223, 214, 209, 196, 195, 202, 205, 144, 151, 158, 153, 140, 139, 130, 133, 168, 175, 166, 161, 180, 179, 186, 189, 199, 192, 201, 206, 219, 220, 213, 210, 255, 248, 241, 246, 227, 228, 237, 234, 183, 176, 185, 190, 171, 172, 165, 162, 143, 136, 129, 134, 147, 148, 157, 154, 39, 32, 41, 46, 59, 60, 53, 50, 31, 24, 17, 22, 3, 4, 13, 10, 87, 80, 89, 94, 75, 76, 69, 66, 111, 104, 97, 102, 115, 116, 125, 122, 137, 142, 135, 0, 149, 146, 155, 156, 177, 182, 191, 184, 173, 170, 163, 164, 249, 254, 247, 240, 229, 226, 235, 236, 193, 198, 207, 200, 221, 218, 211, 212, 105, 110, 103, 96, 117, 114, 123, 124, 81, 86, 95, 88, 77, 74, 67, 68, 25, 30, 23, 16, 5, 2, 11, 12, 33, 38, 47, 40, 61, 58, 51, 52, 78, 73, 64, 71, 82, 85, 92, 91, 118, 113, 120, 255, 106, 109, 100, 99, 62, 57, 48, 55, 34, 37, 44, 43, 6, 1, 8, 15, 26, 29, 20, 19, 174, 169, 160, 167, 178, 181, 188, 187, 150, 145, 152, 159, 138, 141, 132, 131, 222, 217, 208, 215, 194, 197, 204, 203, 230, 225, 232, 239, 250, 253, 244, 243]

#### Functions

def setIP(ip):
    ##check IP
    IP= ip
    global endpoint
    endpoint = (options.ip, 8888)

def setGroup(newgroup): #Groups in your DS App from Top to Bottom, for now limited to 3 Groups
    ##0 - No Group
    ##1 - Use Group 1
    ##2 - Use Group 2
    ##3 - Use Group 3

    global group
    group = newgroup

def setColor(color):
    setAmbience(0) #set to ambient scene mode
    ## decide whether to look up or split
    ##split

    red, green, blue = color.split(' ', 2)
    setMode(3)
    payload=[int(red), int(green), int(blue)]
    buildAndSendPacket(3, 5, payload) #set scene

def setAmbience(ambience):
    payload=[ambience]
    buildAndSendPacket(3,8, payload)

def setScene(scene): #this one is finicky at times.  Need feedback
    ##0x00 - Random Color
    ##0x01 - Fireside
    ##0x02 - Twinkle
    ##0x03 - Ocean
    ##0x04 - Rainbow
    ##0x05 - July 4th
    ##0x06 - Holiday
    ##0x07 - Pop
    ##0x08 - Enchanted Forest

    setAmbience(1)#set to ambient scene mode
    time.sleep(2) #delay to let previous packets get sent correctly.  Sometimes errored out with unexpected behavior...
    payload=[scene]
    buildAndSendPacket(3,13, payload) #set scene

def setMode(mode):
    ##0 - Sleep
    ##1 - Video
    ##2 - Music
    ##3 - Ambient

    payload=[mode]
    buildAndSendPacket(3,1, payload)

def setSource(source):
    # 0- input 1
    # 1- input 2
    # 2- input 3

    payload=[source]
    buildAndSendPacket(3,32, payload)

def setBrightness(brightness):
    payload=[brightness]
    buildAndSendPacket(3,2, payload)

def buildAndSendPacket(upperC, lowerC, payload):
    resp=[] #starts the response object
    resp.append(252) #0xFC
    resp.append(len(payload) + 5)
    resp.append(group)
    resp.append(17)
    resp.append(upperC)
    resp.append(lowerC)
    for i in range(0,len(payload)): #for loop to handle colors.  I know that there are cleaner ways to write it, but I like controlling my iterators.
        resp.append(payload[i])
    crc = calcCRC8(resp) #calc the crc
    resp.append(crc)
    resp = bytearray(resp) #forms response object into a byte to be sent out
    sock.sendto(resp, endpoint)

def calcCRC8(resp):
    size = resp[1] + 1
    crc = 0
    for i in range(0,size):
        crc = crcTable[(resp[i] ^ crc) & 255]
        #print crc #just in for verification
    return crc

###OPTION HANDLING#####################################################
scenes = "Which scenes (use the number)?"\
         "\n0 - Random Color"\
         "\n1 - Fireside"\
         "\n2 - Twinkle"\
         "\n3 - Ocean"\
         "\n4 - Rainbow"\
         "\n5 - July 4th"\
         "\n6 - Holiday"\
         "\n7 - Pop"\
         "\n8 - Enchanted Forest"

modes = "Which mode? (use the number)"\
        "\n0 - Sleep"\
        "\n1 - Video"\
        "\n2 - Music"\
        "\n3 - Ambient"

sources = "Which mode? (use the number)"\
        "\n0 - input 1"\
        "\n1 - input 2"\
        "\n2 - input 3"

groups = "Use, if you have an group setup. (use the number)"\
        "\n0 - No Groups"\
        "\n1 - Use Group 1"\
        "\n2 - Use Group 2"\
        "\n3 - Use Group 3"

parser = optparse.OptionParser()
parser.add_option('-m', '--mode', dest='mode', help=modes)
parser.add_option('-s', '--source', dest='source', help=sources)
parser.add_option('-a', '--ambient scene', dest='scene', help=scenes)
parser.add_option('-b', '--brightness', dest='brightness', help="Enter from 0-100 to control brightness")
parser.add_option('-i', '--ip', dest='ip', help="Please enter your Dreamscreen's IP address")
parser.add_option('-c', '--color', dest='color', help='Which color? Use RGB values from 0-255 in this format: "255 255 255" for white.')
parser.add_option('-g', '--group', dest='group', help=groups)
options, args = parser.parse_args()

if options.ip:
        ##check IP's validity
        payload = options.ip.split('.')
        if len(payload) == 4:
            for i in range(len(payload)): ## check if all ip is valid numerically (you can add in more octets)
                int(payload[i])
            setIP(options.ip)
        else:
            print "There is an error in your IP's length"

if options.group:
    try:
        options.group = int(options.group)
        if options.group > 3:
                options.group = 3
        elif options.group < 0:
                options.group = 0
        setGroup(options.group)
    except:
        print "error: group not a valid number"
        sys.exit(0)

if options.mode:
    try:
        options.mode = int(options.mode) #options come in as strings, must cast as int
        #Data forming for integer edge cases
        if options.mode > 3:
            options.mode = 3
        elif options.mode < 0:
            options.mode = 0
        setMode(options.mode)
    except:
        print "error: mode not a valid number"
        sys.exit(0)

if options.color:
    try:
        options.color
        setColor(options.color)
    except:
        print "There was an error, please make sure you are using the right format: '255 255 255' for white"

if options.source:
    try:
        options.source = int(options.source) #options come in as strings, must cast as int
        if options.source > 2:
                options.source = 2
        elif options.source < 0:
                options.source = 0
        setSource(options.source)
    except:
        print "error: source not a valid number"
        sys.exit(0)

if options.scene:
    try:
        options.scene = int(options.scene) #options come in as strings, must cast as int
        if options.scene > 8:
                options.scene = 8
        elif options.scene < 0:
                options.scene = 0
        setScene(options.scene)
    except:
        print "error: scene not a valid number"
        sys.exit(0)

if options.brightness:
    try:
        options.brightness = int(options.brightness) #options come in as strings, must cast as int
        if options.brightness > 100:
                options.brightness = 100
        elif options.brightness < 0:
                options.brightness = 0
        setBrightness(options.brightness)
    except:
        print "error: brightness not a valid number"
        sys.exit(0)

if (options.ip is None) and (options.group is None) and (options.mode is None) and (options.color is None) and (options.source is None) and (options.scene is None) and (options.brightness is None):
    print "error: no options were selected"
    sys.exit(0)