main.py

from shapely.geometry import Polygon
import geometry
import a_star
import sys
import time

def read_data_from_file():
    with open(sys.argv[1]) as file:
        gallery_coordinates = [[float(x) for x in line.split()] for line in file]
    return gallery_coordinates

def print_results(cameras,node,a_star_time):
    print("LISTA WSZYSTKICH DOSTĘPNYCH KAMER: ")
    print_cameras_coordinates(cameras)
    if node is None:
        print("NIE DA SIĘ POKRYĆ WSZYSTKIMI KAMERAMI CAŁEGO OBSZARU!")
        return
    print("\nKAMERY, KTÓRE NALEŻY WŁĄCZYĆ: ")
    print_cameras_to_turn_on(node)
    print()
    print("Czas wykonania algorytmu: {}".format(a_star_time))

def print_cameras_coordinates(cameras):
    for (i, camera) in enumerate(geometry.Gallery.cameras):
        print("Kamera nr {} o współrzędnych {}".format(i,camera.point))

def print_cameras_to_turn_on(node):
    counter=0
    for (i, camera) in enumerate(node.camerasState):
        if node.camerasState[i]:
            counter=counter+1
            print("Kamera nr {} o współrzędnych {}".format(i,geometry.Gallery.cameras[i].point))
    print("\nMINIMALNA LICZBA KAMER DO POKRYCIA OBSZARU: {}".format(counter))

def print_experiment_results(a_star_time,cameras,node):
    amountOfTurnOnCameras = 0
    if node is None:
        print("{}\t{}\t{}\t{}\t{}".format(float(sys.argv[2]), float(sys.argv[3]), a_star_time, geometry.Gallery.camerasAmount,"-"))
        return
    for (i,camera) in enumerate(node.camerasState):
        if node.camerasState[i] == True:
            amountOfTurnOnCameras +=1
    print("{}\t{}\t{}\t{}\t{}".format(float(sys.argv[2]),float(sys.argv[3]),a_star_time,geometry.Gallery.camerasAmount,amountOfTurnOnCameras))


def main():
    radius, net_density=(float(sys.argv[2]), float(sys.argv[3]))
    gallery_coordinates = read_data_from_file()
    geometry.Gallery.gallery_polygon = Polygon(gallery_coordinates)
    cameras_coordinates=geometry.get_cameras_coordinates(net_density, gallery_coordinates)
    cameras = geometry.get_cameras_array(cameras_coordinates,radius)
    geometry.Gallery.cameras = cameras
    geometry.Gallery.camerasAmount = len(cameras)
    userFriendlyOutput=(sys.argv[4]=='t')
    start = time.time()
    node = a_star.aStar(cameras)
    end = time.time()
    a_star_time = round(end - start,4)
    if userFriendlyOutput:
        print_results(cameras,node,a_star_time)
    if not userFriendlyOutput:
        print_experiment_results(a_star_time,cameras,node)

if __name__ == "__main__":
    main()