#
# Simple test program for the Python MotionNode SDK.
#
# @file    tools/sdk/python/test.py
# @author  Luke Tokheim, luke@motionnode.com
# @version 1.2
#
# (C) Copyright GLI Interactive LLC 2009. All rights reserved.
#
# The coded instructions, statements, computer programs, and/or related
# material (collectively the "Data") in these files contain unpublished
# information proprietary to GLI Interactive LLC, which is protected by
# US federal copyright law and by international treaties.
#
# The Data may not be disclosed or distributed to third parties, in whole
# or in part, without the prior written consent of GLI Interactive LLC.
#
# The Data is provided "as is" without express or implied warranty, and
# with no claim as to its suitability for any purpose.
#
import sys
from MotionNodeSDK import *

PortPreview = 32079
PortSensor = 32078
PortRaw = 32077
PortConfigurable = 32076
PortConsole = 32075
NSample = 10


def test_Client(host, port):
    client = Client(host, port)

    print "Connected to " + str(host) + ":" + str(port)

    xml_string = "<?xml version=\"1.0\"?><configurable><preview><Gq/></preview><sensor><a/></sensor></configurable>"

    if client.writeData(xml_string):
        print "Sent active channel definition to Configurable service"

    if client.waitForData():
        sample_count = 0
        while sample_count < NSample:
            data = client.readData()
            if None == data:
                break
            
            if PortPreview == port:
                container = Format.Preview(data)
                for key in container:
                    q = container[key].getQuaternion(False)

                    print "q(" + str(key) + ") = (" + str(q[0]) + ", " + str(q[1]) + "i, " + str(q[2]) + "j, " + str(q[3]) + "k)"

            if PortSensor == port:
                container = Format.Sensor(data)
                for key in container:
                    a = container[key].getAccelerometer()

                    print "a(" + str(key) + ") = (" + str(a[0]) + ", " + str(a[1]) + ", " + str(a[2]) + ") g"

            if PortRaw == port:
                container = Format.Raw(data)
                for key in container:
                    a = container[key].getAccelerometer()

                    print "a(" + str(key) + ") = (" + str(a[0]) + ", " + str(a[1]) + ", " + str(a[2]) + ")"

            if PortConfigurable == port:
                container = Format.Configurable(data)
                for key in container:
                    line = "data(" + str(key) + ") = ("
                    for i in range(container[key].size()):
                        if i > 0:
                            line += ", "
                        line += str(container[key].value(i))
                    line += ")"

                    print line

            sample_count += 1


def test_LuaConsole(host, port):
    client = Client(host, port)

    print("Connected to " + str(host) + ":" + str(port))

    #
    # General Lua scripting interface.
    #
    lua_chunk = \
      "if not node.is_reading() then" \
      "   node.close()" \
      "   node.scan()" \
      "   node.start()" \
      " end" \
      " if node.is_reading() then" \
      "   print('Reading from ' .. node.num_reading() .. ' device(s)')" \
      " else" \
      "   print('Failed to start reading')" \
      " end"

    print LuaConsole.SendChunk(client, lua_chunk, 5)

    # Scripting language compatibility class. Translate
    # Python calls into Lua calls and send them to the
    # console service.
    node = LuaConsole.Node(client)
    print "node.is_reading() = " + str(node.is_reading())


def test_File():
    filename = "../../test_data/sensor.bin";
    
    print "reading take data file: \"" + filename + "\""

    take_file = File(filename)
    while True:
        data = take_file.readData(9, True)
        if None == data:
            break

        print Format.SensorElement(data).getAccelerometer()


def main(argv):
    # Set the default host name parameter. The SDK is
    # socket based so any networked MotionNode service is
    # available.
    host = ""
    if len(argv) > 1:
        host = argv[1]

    test_LuaConsole(host, PortConsole)

    test_Client(host, PortPreview)
    test_Client(host, PortSensor)
    test_Client(host, PortRaw)
    test_Client(host, PortConfigurable)

    test_File()

if __name__ == "__main__":
    sys.exit(main(sys.argv))