VisualMain.py

import time
import Reader
from Surface import Surface
from Segment import Segment

def chunk(segmentsAll, segmentsInOneIteration, iteration):
    segmentsTab = []
    for i in range(iteration):
        segments = []
        for j in range(segmentsInOneIteration):
            segments.append(segmentsAll[i * segmentsInOneIteration + j])
        segmentsTab.append(segments)

    return segmentsTab

def mainloop():
    diters = 0
    numberOfVertexs, radius, radiusNeighbour, fileWithVertexs, fileWithPath = Reader.readSettings()
    vertexs = Reader.readCoords(fileWithVertexs)
    countOfVertexs = len(vertexs)
    segmentsInOneIteration = numberOfVertexs * 3
    segments, distances = Reader.readSegments(fileWithPath, segmentsInOneIteration)
    iteration = int(len(segments) / segmentsInOneIteration)
    segmentsTab = chunk(segments, segmentsInOneIteration, iteration)
    sur = Surface()
    while True:
        sur.create()
        for point in vertexs:
            sur.drawCity(point.x, point.y, radiusNeighbour)
        i = 0
        for segment in segmentsTab[diters]:
            if (i % 3 == 0):
                sur.drawSegment(segment, radius, True)
            else:
                sur.drawSegment(segment, radius, False)
            i += 1
        sur.createResults(diters, countOfVertexs, numberOfVertexs, distances)
        sur.updateGraphicsOutput()
        diters += 1
        if (diters == iteration):
            print("KONIEC")
            break
        time.sleep(0)
    
    time.sleep(100000)

mainloop()