def __init__(self):
self.camera = cv2.VideoCapture(0)
self.tracker = eyetracker.EyeTracker()
self.movAvgDict = dict()
self.lookingPointMovAvg = MovingAveragePoints(
Point(self.tracker.xScale / 2, self.tracker.yScale / 2), 5)
self.movAvgLength = 5
self.bottomRight = None
self.xMin = 0
self.yMin = 0
self.xMax = self.tracker.getXScale()
self.yMax = self.tracker.getYScale()
self.rangePadding = 10
self.headCenter = None
self.translationXMin = 0
self.translationYMin = 0
(self.translationXMax,
self.translationYMax) = (self.tracker.getXScale(),
self.tracker.getYScale())
self.canvasDim = Point(1280, 770)
self.colorPadding = 5
self.learningColors = dict()
def updateMovAvgDict(self, results):
newDict = dict()
for r in results:
if not r.getId() in self.movAvgDict.keys():
self.movAvgDict[r.getId()] = {
"centroid":
MovingAveragePoints(r.getPupil().getCentroid(),
self.movAvgLength),
"rVector":
MovingAveragePoints(
r.getResultantVector(self.tracker.getXScale(),
self.tracker.getYScale()),
self.movAvgLength)
}
self.updateExistingValue(r)
newDict[r.getId()] = self.movAvgDict[r.getId()]
return newDict
