blob: 665d14eddbe1ae85fa8c246ee67480d9b826fa9f (
plain)
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
|
from operationscore.PixelAssembler import *
import pdb
class ZigzagLayout(PixelAssembler):
"""ZigZagLayout is a slightly more complex layout class that makes a zig-Zag Led Pattern
Inheriting classes must specify zigLength, the length in lights of a of a zig
and zig Axis, the direction of the long X axis (X or Y).
EG: zig length = 4, zig Axis = X would give:
X-X-X-X
|
X-X-X-X
|
X-X-X-X etc."""
def initLayout(self):
if not 'zigLength' in self.argDict:
raise Exception('zigLength must be defined in argDict')
if not 'zigAxis' in self.argDict:
raise Exception('zigAxis must be defined in argDict')
if not 'xDirection' in self.argDict:
self.argDict['xDirection'] = 1 #right
if not 'yDirection' in self.argDict:
self.argDict['yDirection'] = 1 #down
def layoutFunc(self, lastLocation):
if not 'buildQueue' in self.argDict:
self.argDict['buildQueue'] = self.argDict['zigLength']
newLoc = list(lastLocation)
if self.argDict['buildQueue'] > 1:
if self.argDict['zigAxis'] == 'X':
newLoc[0] += self.argDict['spacing'] * self.argDict['xDirection']
else:
newLoc[1] += self.argDict['spacing'] * self.argDict['yDirection']
self.argDict['buildQueue'] -= 1
else:
self.argDict['buildQueue'] = self.argDict['zigLength']
if self.argDict['zigAxis'] == 'X':
newLoc[1] += self.argDict['spacing'] * self.argDict['yDirection']
else:
newLoc[0] += self.argDict['spacing'] * self.argDict['xDirection']
if self.argDict['zigAxis'] == 'X':
self.argDict['xDirection'] *= -1
else:
self.argDict['yDirection'] *= -1
return newLoc
|
