eventloop_gen.py

import random
from random import randint
import requests
import pandas as pd
import numpy as np
from sklearn.utils import shuffle

# with open('./racing_maps.txt') as f:
#     racing_maps = f.read().splitlines()
#
# with open('./arena_maps.txt') as f:
#     arena_maps = f.read().splitlines()
#
# with open('./oval_maps.txt') as f:
#     oval_maps = f.read().splitlines()
#
# with open('./figure_8_maps.txt') as f:
#     figure_8_maps = f.read().splitlines()


def refetch():
    url = f'https://docs.google.com/spreadsheets/d/10tpRET18K7rIlvYC4An90y5KzeuSfEiBv5muAjFJHuk/export?format=csv'
    response = requests.get(url)

    data = response.content.decode()
    with open('tracks_dev.csv', 'w') as fh:
        for row in data.splitlines():
            fh.writelines(f'{row.strip()}\n')
    print('Refetched')

def gen_event(rmap, laps, class_res, special, rtype):
    res = f'''
# {rtype} / {rmap} / {laps} / {class_res} / {special}
el_add={rmap}
el_gamemode=racing
el_laps={laps}
el_bots=0
el_car_class_restriction={class_res}
el_car_restriction={special}
'''
    return res


def parse():
    with open('tracks_wiki.csv') as fh:
        out_maps = fh.read().splitlines()
        out_maps = [i.split(",") for i in out_maps]
        out_maps_cols = out_maps.pop(0)
        out_maps_df = pd.DataFrame().from_records(out_maps, columns=out_maps_cols)

    print(out_maps_df)

    with open('tracks_dev.csv', 'r') as fh:
        data = fh.read().splitlines()[2:]

    templ = {
        'envir': [],
        'event': [],
        'code': []
    }
    envir = None
    for row in data:
        print(row)
        recs = row.split(',')
        recs = [i.strip() for i in recs]
        if len(recs) < 2:
            continue
        # code = None
        if recs[1]:
            envir = recs[1]

        if not recs[2]:
            continue

        event = recs[2]
        templ['envir'].append(envir)
        templ['event'].append(event)
        templ['code'].append(recs[3])
        # print(row)
    out = pd.DataFrame().from_dict(templ)
    # print(out)
    out.loc[out['envir'] == 'Drytown Desert', 'envir'] = 'Drytown Desert Circuit'
    out.loc[out['envir'] == 'Finncross Circuit', 'envir'] = 'FinnCross Circuit'
    out.loc[out['envir'] == 'Firwood Motorcenter', 'envir'] = 'Firwood Motocenter'
    out.loc[out['envir'] == 'Maasten Motorcenter', 'envir'] = 'Maasten Motocenter'
    out.loc[(out['envir'] == 'Northfolk Ring') & (out['event'] == 'Main Route Reverse'), 'event'] = 'Main Circuit Reverse'
    out.loc[(out['envir'] == 'Hillstreet Circuit') & (out['event'] == 'Racing Track'), 'event'] = 'Race Track'
    out.loc[(out['envir'] == 'Hillstreet Circuit') & (out['event'] == 'Racing Track Reverse'), 'event'] = 'Race Track Reverse'
    out.loc[(out['event'] == 'Alt Route Revese'), 'event'] = 'Alt Route Reverse'
    out.loc[(out['envir'] == 'Tribend Raceway') & (out['event'] == 'Main Circuit Reverse'), 'event'] = 'Reverse Circuit'
    out.loc[out['envir'] == 'Tribend Raceway', 'envir'] = 'Tribend Speedway'
    out.loc[out['envir'] == 'Vale Falls', 'envir'] = 'Vale Falls Circuit'
    out.loc[(out['envir'] == 'Wrecknado') & (out['event'] == 'Racing Track'), 'event'] = 'Main Circuit'

    out = pd.merge(out, out_maps_df, how='left', left_on=['envir', 'event'], right_on=['Map', 'Track'])
    for col in ['rev', 'alt', 'oval', 'arena', 'fig8', 'roam']:
        out[col] = 0

    # mark
    out['rev'] = np.where(out['code'].str.endswith('_rev'), 1, out['rev'])
    out['rev'] = np.where(out['event'].str.endswith('Reverse'), 1, out['rev'])
    out['alt'] = np.where(out['event'].str.startswith('Alt'), 1, out['alt'])
    out['oval'] = np.where(out['event'].str.endswith('Oval'), 1, out['oval'])
    out['oval'] = np.where(out['event'].str.endswith('Oval Loop'), 1, out['oval'])
    out['arena'] = np.where(out['event'].str.endswith('Arena'), 1, out['arena'])
    out['arena'] = np.where(out['event'] == 'Mud Pit', 1, out['arena'])
    out['arena'] = np.where(out['envir'] == 'Wrecking Playground', 1, out['arena'])
    out['arena'] = np.where(out['code'].str.endswith('_arena'), 1, out['arena'])
    out['fig8'] = np.where(out['code'].str.endswith('_figure_8'), 1, out['fig8'])
    out['roam'] = np.where(out['event'].str.startswith('Free Roam'), 1, out['roam'])

    print(out)

    # filter
    out = out[out['roam'] == 0]
    out = out[out['rev'] == 0]

    print(out)
    # views
    races_oval = out[out['oval'] == 1]
    races_fig8 = out[out['fig8'] == 1]
    races_arena = out[out['arena'] == 1]
    races_circ = out[(out['fig8'] == 0) & (out['oval'] == 0) & (out['arena'] == 0)]

    races_oval = shuffle(races_oval).reset_index(drop=True)
    races_fig8 = shuffle(races_fig8).reset_index(drop=True)
    races_arena = shuffle(races_arena).reset_index(drop=True)
    races_circ = shuffle(races_circ).reset_index(drop=True)
    idx = {
        'a': 0,
        'o': 0,
        'f': 0,
        'c': 0
    }

    prep = []
    cycle = 0
    for idx['c'] in range(len(races_circ)):
        if cycle == 5:
            cycle = 0
        cycle += 1
        race = races_circ.iloc[idx['c']]
        print(race)
        laps = randint(int(race['min_laps']), int(race['max_laps']))
        class_res = random.choice(['a', 'a', 'b', 'b', 'c', 'c', 'special'])
        special = 'school bus' if class_res == 'special' else ''
        event = gen_event(race['code'], laps, class_res, special, 'Circ')
        prep.append(event)

        if cycle in [1, 3]:
            # Fo8
            race = races_fig8.iloc[idx['f']]
            laps = randint(int(race['min_laps']), int(race['max_laps']))
            class_res = random.choice(['a', 'b', 'c', 'special'])
            specials = ['school bus', 'motor home', 'sofa car']
            special = random.choice(specials) if class_res == 'special' else ''
            event = gen_event(race['code'], laps, class_res, special, 'Fo8')
            prep.append(event)
            idx['f'] += 1
            if idx['f'] >= len(races_fig8):
                idx['f'] = 0
            pass
        elif cycle in [2, 4]:
            # Oval
            race = races_oval.iloc[idx['o']]
            laps = randint(int(race['min_laps']), int(race['max_laps']))
            class_res = random.choice(['a', 'b', 'c', 'special'])
            specials = ['school bus', 'motor home', 'sofa car']
            special = random.choice(specials) if class_res == 'special' else ''
            event = gen_event(race['code'], laps, class_res, special, 'Oval')
            prep.append(event)
            idx['o'] += 1
            if idx['o'] >= len(races_oval):
                idx['o'] = 0
            pass
        elif cycle in [5]:
            # Arena
            race = races_arena.iloc[idx['a']]
            laps = 0
            class_res = random.choice(['a', 'b', 'c', 'special'])
            specials = ['school bus', 'motor home', 'sofa car']
            special = random.choice(specials) if class_res == 'special' else ''
            event = gen_event(race['code'], laps, class_res, special, 'Arena')
            prep.append(event)
            idx['a'] += 1
            if idx['a'] >= len(races_arena):
                idx['a'] = 0
            pass

    print(prep)
    with open('server.cfg', 'w') as fh:
        fh.writelines(prep)


if __name__ == '__main__':
    refetch()
    parse()