ledcontroller/test_led.py at master · maddoxlab/ledcontroller · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
# -*- coding: utf-8 -*-
"""
Created on Fri May 31 10:37:22 2019

@author: mcappelloni
"""
from _led import (DotStrip, _LightShape, Dot, Line, Gaussian, Tukey, PixelArray)
import numpy as np
import matplotlib.pyplot as plt
from expyfun import ExperimentController

n_led = 1091

import socket
HOST = '169.254.150.219'  # The server's hostname or IP address
PORT = 5005        # The port used by the server
client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
client.connect((HOST, PORT))

dots = DotStrip(client, n_led)

with ExperimentController('test_led', output_dir=None, version='dev',
                          participant='foo', session='foo',
                          full_screen=False) as ec:
    dots.clear_strip()
    dots.send()

    # %%%%%%%%%%%%%%%
    test_holes = [Dot(ec, dots, 4 * [.5], int(i), 'deg') for i in np.arange(-102, 103, 4)]
    [t.draw() for t in test_holes]
    dots.send()
    ec.screen_prompt('Lights will line up with between speaker holes')
    dots.clear_strip()
    dots.send()
    ec.wait_secs(1)
    test_center = [Dot(ec, dots, 4 * [.5], int(dots.get_speaker_location(i, 'deg')), 'deg') for i in range(53)]
    [t.draw() for t in test_center]
    dots.send()
    ec.screen_prompt('Lights will line up with center of speakers')
    # %%%%%%%%%%%%%%%
    ec.screen_prompt('There are three blending modes')
    dots.clear_strip()
    dots.send()
    blend1 = Gaussian(ec, dots, [.4, .8, 0, .3], 600, 10)
    blend1.draw()
    dots.send()
    ec.screen_prompt('Background Shape')

    blend2 = Line(ec, dots, [.6, .6, .6, .4], [590, 595])
    blend3 = Line(ec, dots, [.6, .6, .6, .4], [597, 602])
    blend4 = Line(ec, dots, [.6, .6, .6, .4], [605, 610])
    blend3.draw('max')
    blend2.draw('occlude')
    blend4.draw('add')
    dots.send()
    ec.screen_prompt('White line on top with occlude, max, and add modes')

    dots.clear_strip()
    dots.send()
    ec.screen_prompt('Now some simple shapes')

    test1 = Dot(ec, dots, [.3, .3, .1, .8], 590)
    test1.draw()
    dots.send()
    ec.screen_prompt('Dot')

    test2 = Line(ec, dots, [.7, .2, .3, .3], [595, 600])
    test2.draw()
    dots.send()
    ec.screen_prompt('Line')

    test3 = Gaussian(ec, dots, [.3, .4, .7, .5], 610, 3)
    test3.draw()
    dots.send()
    ec.screen_prompt('Gaussian')

    test4 = Tukey(ec, dots, [.9, .6, .1, .5], 620, 10, .9)
    test4.draw()
    dots.send()
    ec.screen_prompt('Tukey with alpha=.9')

    dots.clear_strip()
    dots.send()
    ec.screen_prompt('Now use any arbitrary array')

    pressed = None
    ec.screen_text('Press any key to exit')
    ec.flip()
    cm = plt.get_cmap('hsv')
    colors = cm(np.linspace(0, 1, 40, dtype=float))
    colors[:, -1] = .3
    images = [PixelArray(ec, dots, colors, [a+300, b+300]) for a, b in
                  zip(np.arange(120, 400, 10), np.arange(160, 440, 10))]

    while not pressed:
        ec.listen_presses()
        for i in images:
            i.draw()
            dots.send()
            dots.clear_strip()
            ec.wait_secs(0)
            pressed = ec.get_presses()

    client.sendall(bytes(1))
    client.close()