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
|
import util.ColorOps as color
from logger import main_log
import pdb
from pixelevents.StepEvent import *
import util.TimeOps as timeops
class Pixel:
"""Pixel keeps a queue of events (PixelEvent objects) (actually a dictionary
keyed by event time). Every time is state is
requested, it processes all the members of its queue. If a member returns none,
it is removed from the queue. Otherwise, its value added to the Pixels color
weighted by z-index. To get the current color of the pixel, call the state method."""
radius = 2
timeOff = -1
def __init__(self, location):
self.location = location
self.events = []
self.lastRenderTime = timeops.time()
self.lastRender = (0,0,0)
def turnOn(self):
self.turnOnFor(-1)
#Turn the light white for 'time' ms. Really only meant for testing. Use
#processInput instead. Also, you shouldn't use this anyway. You should be
#using the input method on the screen!
def turnOnFor(self, time):
event = StepEvent.generate(time, (255,255,255))
self.processInput(event, 0)
#Add a pixelEvent to the list of active events
def processInput(self,pixelEvent,zindex, scale=1,currentTime=None): #consider migrating arg to dict
if currentTime == None:
currentTime = timeops.time()
self.events.append((currentTime, zindex, scale, pixelEvent)) #TODO: clean this up, maybe?
def clearAllEvents(self):
self.events = []
def state(self, currentTime=None):
"""Combines all PixelEvents currently active and computes the current color of
the pixel."""
if currentTime == None:
currentTime = timeops.time()
if currentTime-self.lastRenderTime < 5:
return self.lastRender
if self.events == []:
self.lastRenderTime = currentTime
return (0,0,0)
deadEvents = []
resultingColor = (0,0,0)
colors = []
for eventObj in self.events: #TODO: right color weighting code
if len(self.events) > 50:
main_log.error('High pixel event count! Investigate!')
eventTime, zindex, scale, pixelEvent = eventObj
eventResult = pixelEvent.state(currentTime-eventTime)
if eventResult != None:
scaledEvent = color.multiplyColor(eventResult,scale)
if (scaledEvent[0] + scaledEvent[1] + scaledEvent[2]) < 5:
pass
#deadEvents.append(eventObj)
else:
colors.append(scaledEvent)
else:
deadEvents.append(eventObj)
resultingColor = color.combineColors(colors)
#if colors:
# resultingColor = [c / len(colors) for c in resultingColor]
[self.events.remove(event) for event in deadEvents]
resultingColor = [int(round(c)) for c in resultingColor]
self.lastRender = tuple(resultingColor)
self.lastRenderTime = currentTime
return tuple(resultingColor)
def __str__(self):
return 'Loc: ' + str(self.location)
|
