lightboard2/master.py at master · dougnd/lightboard2 · 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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
import lights
import programs
import sys
import subprocess
import time
import array
import threading
#import pkgutil
#import ipdb as pdb
#import time


allLights = {
    'frontLeftMagic': lights.DMXMagic(53),
    'frontLeft': lights.DMXPar(4),
    'frontSpencer': lights.DMXPar(1),
    'frontTanner': lights.DMXPar(7),
    'frontRight': lights.DMXPar(10),
    'backSpencer': lights.DMXPar(37),
    'backTanner': lights.DMXPar(40),
    'backDoug': lights.DMXPar(43),
    'backTravis': lights.DMXPar7Ch(46),
    'rear1': lights.DMXPar6Ch(13),
    'rear2': lights.DMXPar6Ch(19),
    'rear3': lights.DMXPar6Ch(25),
    'rear4': lights.DMXPar6Ch(31)
}


class Master:
    def __init__(self):
        self.allPrograms = programs.programList
        self.currentProgram = None
        self.currentProgramIndex = 0
        self.loadProgram(0)
        self.lastFrontPress = 0


    def loadProgram(self, index):
        #print 'loading ' + str(self.allPrograms[index])

        #reset old program:
        if self.currentProgram:
            self.currentProgram.reset()

        self.currentProgramIndex = index
        currentProgramModule = __import__('programs.' +
                                          self.allPrograms[index][1],
                                          fromlist=[
                                              self.allPrograms[index][1]])
        programClass = getattr(currentProgramModule,
                               self.allPrograms[index][2])
        self.currentProgram = programClass(allLights)
        self.currentProgram.reset()
        print "------------------"
        print self.allPrograms[index][0]
        #self.currentProgramModule.setAllLights(allLights)

    def btn(self,number):
        #print "Button " + str(number) + " pressed"
        #testProgram.buttonPressed(number)
        self.currentProgram.buttonPressed(number)

    def nextProgram(self):
        self.loadProgram((self.currentProgramIndex+1)%len(self.allPrograms))

    def prevProgram(self):
        self.loadProgram((self.currentProgramIndex-1)%len(self.allPrograms))

    def darkBtn(self):
        print "Dark"

        self.currentProgram.reset()
        for name, l in allLights.items():
            l.setController(lights.FadeInController(
                lights.ConstantRGBController(0,0,0), 0.1
            ))

    def frontBtn(self):
        if time.time() - self.lastFrontPress > 1.0:
            print "Front Spencer"
            frontLights = ['frontSpencer']
        else:
            print "All Front"
            frontLights = ['frontSpencer', 'frontTanner', 'frontLeft', 'frontRight']

        self.lastFrontPress = time.time()
        for name, l in allLights.items():
            l.setController(lights.FadeInController(
                lights.ConstantRGBController(0,0,0), 0.1
            ))
        for l in frontLights:
            allLights[l].setController(lights.FadeInController(
                lights.ConstantRGBController(255,255,255), 0.3
            ))


    def getProgramName(self):
        return self.allPrograms[self.currentProgramIndex][0]


    def runReal(self):
        self.dmxArray = array.array('B', [0]*100)
        self.dmxArray[0] = 255 # needed for spencer's second 4channel front light

        for name, l in allLights.items():
            l.dmxArray = self.dmxArray

        from ola.ClientWrapper import ClientWrapper
        targetFps = 24.0
        TICK_INTERVAL = 1000.0/targetFps
        wrapper = None

        def DmxSent(state):
            if not state.Succeeded():
                print "Error!"

        def SendDMXFrame():
            wrapper.AddEvent(TICK_INTERVAL, SendDMXFrame)

            for name, l in allLights.items():
                l.update()

            wrapper.Client().SendDmx(0, self.dmxArray, DmxSent)

        wrapper = ClientWrapper()
        wrapper.AddEvent(TICK_INTERVAL, SendDMXFrame)
        wrapper.Run()
        print "What??"