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
|
from operationscore.PixelMapper import *
import util.Geo as Geo
import sys
class C5SignMapper(PixelMapper):
"""C5SignMapper is a modification to SimpleMapper which maps events to the
nearest Pixel. In addtion, it also maps sign artifacts (letters, logo, etc)
to their representative locations if given in the form "ts rs :: conditions"
It also supports strings of the form: {x}>5, {y}<10, {x}*{y}<{x}, etc.
(Conditons, separated by commas. and and or may also be used)."""
signPosition = {
"ls" : [(2,8), (2,14), (2,20)],
"ts" : [(4,22), (10,22), (16,22), (22,22), (27, 22), (33, 22), (39,22),
(44, 22)],
"rs" : [(45,2), (45, 8), (45,14), (45,20)],
"bs" : [(4,2), (10,2), (16,2), (22, 2), (27,2), (34,2), (39,2), (44,2)],
"wt" : [(12,5), (13, 5), (16,5), (18,5), (21,5), (23,5), (26,5), (27,5),
(30,5), (34,5), (37,5)],
"cl" : [(17,8), (21,10), (24,10), (26,12), (31,12)],
"c5" : [(6,17), (11,17), (15,17), (19,17), (22, 17), (27,17), (33,16),
(34, 16), (38,17), (42,17)]}
def mappingFunction(self, eventLocation, screen):
if type(eventLocation) == type(tuple()):
bestDist = sys.maxint
bestPixel = None
[x,y] = eventLocation
for (x,pixel) in screen.pixelsInRange(x-self['CutoffDist'], \
x+self['CutoffDist']):
pixelDist = Geo.dist(pixel.location, eventLocation)
if pixelDist < bestDist:
bestPixel = pixel
bestDist = pixelDist
if bestPixel != None:
return [(bestPixel,1)]
else:
return []
elif type(eventLocation) == type
else:
eventLocSplit = eventLocation.split(' :: ')
if len(eventLocSplit) == 2:
[signPart, eventLocation] = eventLocSplit
signParts = [signPosition[x] for x in signPart.split(' ')]
#{x}>5,{y}<k
ret = []
eventLocation = eventLocation.replace('{x}', 'pixel.location[0]')
eventLocation = eventLocation.replace('{y}', 'pixel.location[1]')
conditions = eventLocation.split(',')
conditionLambdas = [eval('lambda pixel:'+condition) for condition in conditions]
for pixel in screen:
try:
pixelValid = True
for p in conditionLambdas:
if p(pixel) == False:
pixelValid = False
continue
if pixelValid:
ret.append((pixel, 1))
except Exception as exp:
raise Exception('Bad event condition')
return ret
|
