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