aboutsummaryrefslogtreecommitdiff
path: root/pixelmappers/GaussianMapper.py
blob: 686ebcd4866b34d5c8c540cd78522539e4dc8c8c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
from operationscore.PixelMapper import *
import util.Geo as Geo
class GaussianMapper(PixelMapper):
    def mappingFunction(self, eventLocation, screen):
        returnPixels = [] #TODO: consider preallocation and trimming
        [x,y] = eventLocation
        for (x,pixel) in screen.pixelsInRange(x-self.CutoffDist, \
                x+self.CutoffDist):
            pixelDist = Geo.dist(pixel.location, eventLocation)
            if pixelDist < self.CutoffDist:
                w = Geo.gaussian(pixelDist, self.Height, 0, self.Width)
                if w > self.MinWeight:
                    returnPixels.append((pixel, w))
        return returnPixels