from entity import *
import random
import numpy as np


def get_data():
    all_data = []
    for id in range(300000):
        day_data = []
        for i in range(random.randint(10,25)):
            d_x = random.randint(0,49)
            d_y = random.randint(0,49)
            driver = Driver(id,d_x,d_y)
            to_x = random.randint(0,49)
            to_y = random.randint(0,49)
            juli = 1000
            while juli > 6:
                o_x = random.randint(d_x-6,d_x+6)
                o_y = random.randint(d_y-6,d_y+6)
                juli = abs(o_x - d_x) + abs(o_y - d_y)
            order = Order(0,o_x,o_y,to_x,to_y,random.randint(1,144))
            match = Match(order,driver)
            day_data.append(match)
            print('\r'+str(id)+'  '+str(i),end='')

        all_data.append(day_data)
    save(all_data,"train_data_02.pkl")


if __name__ == '__main__':
    get_data()
 #    q = [[6.3724101e-02 ,1.0513991e+02],
 # [1.9977689e-03 ,9.5939758e+01],
 # [1.5849888e-02, 1.0017528e+02]]
 #    print(np.max(q, axis=1))