validator.py

# Based on: https://github.com/sarful/People-counter-opencv-python3/blob/master/Person.py

import math

class MyValidator:
    tracks = []

    def __init__(self, i, xi, yi, max_age):
        self.i = i
        self.x = xi
        self.y = yi
        self.tracks = []
        self.done = False
        self.state = '0'
        self.age = 0
        self.max_age = max_age
        self.dir = None

    def getTracks(self):
        return self.tracks

    def getId(self):
        return self.i

    def getState(self):
        return self.state

    def getDir(self):
        return self.dir

    def getX(self):
        return self.x

    def getY(self):
        return self.y

    def updateCoords(self, xn, yn):
        self.age = 0
        self.tracks.append([self.x, self.y])
        self.x = xn
        self.y = yn

    def setDone(self):
        self.done = True

    def timedOut(self):
        return self.done

    def going_DOWN(self, mid_start):

        # Check if you have at least 2 coordinates of stored objects
        if len(self.tracks) >= 2:

            # Check if the condition of the object is zero
            # The state of the object will only change when crossing the input threshold
            if self.state == '0':

                # Euclidian distance calculation
                distance = math.sqrt(float((self.tracks[-1][1] - self.tracks[-2][1])**2) + float(
                    (self.tracks[-1][1] - self.tracks[-2][1])**2))
                if distance < 10:
                    # [-2] are two previous positions of the vector record and [1] is the column containing the
                    # vertical values (y) of each object
                    # If the anterior vertical position of the object is greater than the input threshold and if in
                    # two anterior vertical positions the value is less than or equal to the input threshold
                    # we updated the state of the object to 1 and indicated that it moved downwards (down)
                    # We do this to make sure that the object crossed the entrance line, moving from top to bottom
                    if self.tracks[-1][1] > mid_start and self.tracks[-2][1] <= mid_start:
                        state = '1'
                        self.dir = 'down'
                        return True
            else:
                return False
        else:
            return False

    def going_UP(self, mid_end):
        if len(self.tracks) >= 2:
            if self.state == '0':
                distance = math.sqrt(float((self.tracks[-1][1] - self.tracks[-2][1])**2) + float(
                    (self.tracks[-1][1] - self.tracks[-2][1])**2))
                if distance < 10:
                    if self.tracks[-1][1] < mid_end and self.tracks[-2][1] >= mid_end:
                        state = '1'
                        self.dir = 'up'
                        return True
            else:
                return False
        else:
            return False

    def age_one(self):
        self.age += 1
        if self.age > self.max_age:
            self.done = True
        return True