main.py

from OutputParser import OutputParser
from InputParser import InputParser
import numpy as np
import math
import matplotlib.pyplot as plt
import pandas as pd
from shared import *

PATH = ""
FILE = "c_no_hurry"


# ROWS: index, xstart, ystart, xend, yend, start, stop

class Auto:
    def __init__(self, id):
        self.id = id
        self.ticksToFree = 0
        self.x = 0
        self.y = 0
        self.served = []

    def goRide(self, ride, current_tick):
        self.ticksToFree = abs(self.x - ride[1]) + abs(self.y - ride[2])
        if 0 < ride[-2] - abs(self.y - ride[3]) - abs(self.x - ride[2]) - current_tick:
            self.ticksToFree += ride[-2] - abs(self.x - ride[2]) + abs(self.y - ride[3]) - current_tick
        self.ticksToFree += abs(ride[3] - ride[1]) + abs(ride[4] - ride[2])

        self.served.append(ride[0])
        self.x = ride[3]
        self.y = ride[4]

    def tick(self):
        # Decrease ticks of the car
        if self.ticksToFree > 0:
            self.ticksToFree -= 1


def get_free_cars():
    result = []
    for car in cars:
        if car.ticksToFree == 0:
            result.append(car)
    return result


possibleRides = []
impossibleRides = []

inputData = InputParser(FILE + ".in")
rows, columns, vehicles, numebrOfRides, bonusOnTime, steps = inputData.readLine()
inputData.close()

rides = pd.read_table(FILE + ".in", sep=' ')

counter = 0
for i in rides.values:
    if (abs(i[2] - i[0]) + abs(i[3] - i[1])) >= abs(i[5] - i[4]):
        impossibleRides.append(np.insert(i, 0, counter))
    else:
        possibleRides.append(np.insert(i, 0, counter))
    counter += 1

# Sorting
possibleRides = pd.DataFrame(possibleRides, columns=[str(x) for x in range(7)])
impossibleRides = pd.DataFrame(impossibleRides, columns=[str(x) for x in range(7)])
possibleRides = possibleRides.sort_values(by='5', ascending=False)
if len(impossibleRides) > 0:
    impossibleRides = impossibleRides.sort_values(by='5')

cars = []
for i in range(vehicles):
    cars.append(Auto(i + 1))
cars = np.array(cars)

# MAIN LOOP
possibleRides = possibleRides.values

for i in range(steps):
    freeCars = get_free_cars()

    if len(possibleRides) > 0:
        for car in freeCars:
            possibleRides = decide_on_job(car, possibleRides, i)
            if len(possibleRides) == 0:
                break
    if len(possibleRides) > 0:
        new_free_cars = get_free_cars()
        extra = 1
        while get_free_cars():
            extra += 1
            for car in get_free_cars():
                possibleRides = leftover_tracks(car, possibleRides, i, extra)
                if len(possibleRides) == 0:
                    break
    for car in cars:
        car.tick()

# Save output
outputData = OutputParser(PATH + FILE + ".out")
for i in cars:
    outputData.write([len(i.served)] + i.served)

outputData.close()