from UI.Objects.Multiscene import Multiscene
from UI.Scenes.BasicScene import BasicScene
from numpy import array as a

from UI.Objects.nr_objects.Line import Line


class BasicMultiscene(Multiscene):
    def __init__(self, scene_size):

        self.s__size = [600, 400]

        super().__init__(self.s__size[0], self.s__size[1], scene_size)

        s0 = BasicScene([200, 400])

        s1 = BasicScene([400, 400])
        s1.position = a([200, 0])

        l0 = Line(a([200, 0]), a([200, 400]), 10, (255, 100, 100), self)

        self.sort_objects(s0, s1, l0)

        self.i = 0

    def progress(self):
        if self.i == 1:
            self.subscenes[0].subscene_size[0] += 1
            self.subscenes[1].subscene_size[0] -= 1
            self.subscenes[1].position[0] += 1

            self.nr_objects[0].a[0] += 1
            self.nr_objects[0].b[0] += 1

            self.resize_subscenes()
            self.i = 0
        else:
            self.i += 1