Video-Summary/LayerFactory.py

from Layer import Layer

class LayerFactory:
    data = {}
    layers = []
    def __init__(self, data=None):
        print("LayerFactory constructed")
        self.data = data
        if data is not None:
            self.extractLayers(data)

    def extractLayers(self, data = None):
        if self.data is None:
            if data is None:
                print("LayerFactory data was none")
                return None
            else:
                self.data = data

        layers = []
        frameNumber = min(data)
        contours = data[frameNumber]

        for contour in contours:
            layers.append(Layer(frameNumber, contour, None))

        for frameNumber, contours in data.items():
            layerNum = len(layers)
            i = 0
            while i < layerNum:
                layer = layers[i]
                if frameNumber - layer.lastFrame < 5:
                    for contour, (x,y,w,h) in contours:
                        if len(layer.data[-1][1]) != 4:
                            print("LayerFactory: Layer knew no bounds")
                            continue
                        (x2,y2,w2,h2) = layer.data[-1][1]
                        
                        if self.contoursOverlay((x,y+h), (x+w,y), (x2,y2+h2), (x2+w2,y2)):
                            layer.add(frameNumber, (contour, (x,y,w,h)))
                        else:
                            layers.append(Layer(frameNumber, contour, None))
                            layerNum = len(layers)
                i+=1

        self.layers = layers


    def contoursOverlay(self, l1, r1, l2, r2): 
      
        # If one rectangle is on left side of other 
        if(l1[0] >= r2[0] or l2[0] >= r1[0]): 
            return False
    
        # If one rectangle is above other 
        if(l1[1] <= r2[1] or l2[1] <= r1[1]): 
            return False
    
        return True