from dataclasses import dataclass

import param_
import streamlit as st
import numpy as np


@dataclass
class Settings:

    perimeter: int = param_.PERIMETER
    perimeter_z: int = param_.PERIMETER_Z
    groundzone: int = param_.GROUNDZONE

    latlon = param_.LATLON['Paris']['lat'], param_.LATLON['Paris']['lon']

    blues: int = param_.BLUES

    blues_per_circle = np.array(param_.BLUES_PER_CIRCLE)
    blue_circles_rho = param_.BLUE_CIRCLES_RHO
    blue_circles_theta = param_.BLUE_CIRCLES_THETA
    blue_circles_zed = param_.BLUE_CIRCLES_ZED
    blue_distance_factor: float = param_.BLUE_DISTANCE_FACTOR

    is_unkillable: bool = param_.BLUE_IS_UNKILLABLE

    blue_speed_init: int = param_.BLUE_SPEED_INIT

    reds: int = param_.REDS

    red_squads = param_.RED_SQUADS
    red_squads_rho = np.array(param_.RED_SQUADS_RHO)
    red_squads_theta = param_.RED_SQUADS_THETA
    red_squads_zed = param_.RED_SQUADS_ZED
    red_distance_factor: float = param_.RED_DISTANCE_FACTOR

    red_rho_noise = np.array(param_.RED_RHO_NOISE)
    red_theta_noise = np.array(param_.RED_THETA_NOISE)
    red_zed_noise = np.array(param_.RED_ZED_NOISE)

    red_speed_init: int = param_.RED_SPEED_INIT

    policy_folder: str = param_.POLICY_FOLDER


def define_(with_streamlit: bool = True, blues: int = Settings.blues, reds: int = Settings.reds):
    """"
        shows the blue and red swarms in Streamlit
        :return:
        """
    blues = blues
    reds = reds

    if with_streamlit:
        st.title('Blues against Reds by hexamind.ai')
        st.write('controlled by Reinforcement Learning.')
        st.text('<- Set parameters')

        st.sidebar.subheader("Define the battlefield")
        blues = st.sidebar.slider("how many blues on defense?", 1, 20, 6)
        Settings.blues = blues
        blue_dispersion = st.sidebar.slider("set the average blue dispersion", 0.3, 1.0, 0.8)
        Settings.reds = reds
        reds = st.sidebar.slider("how many reds are on the attack?", 1, 20, 6)
        Settings.reds = reds
        red_dispersion = st.sidebar.slider("set the average red dispersion", 0.3, 1.0, 0.7)

        Settings.blue_distance_factor = 3 * blue_dispersion
        Settings.red_distance_factor = 3 * red_dispersion

        location = st.sidebar.radio("Location", ['Paris', 'Puilaurens', 'San Francisco'])

        lat_tg = param_.LATLON[location]['lat']
        lon_tg = param_.LATLON[location]['lon']

        Settings.latlon = lat_tg, lon_tg

        st.sidebar.write(
            'you probably need more drones '
            'No worries, we have plenty at www.hexamind.ai ')

    return blues, reds