disc_golf_simulator.py

from logging import getLogger
from pathlib import Path
proj_dir = Path(__file__).parents[1]
import sys
sys.path.append(str(proj_dir))

import numpy as np
import streamlit as st

from shotshaper.projectile import DiscGolfDisc
from visualize import get_plot, get_stl, get_subplots, visualize_disc


# Define the default values
default_U = 24.2
default_omega = 116.8
default_z0 = 1.3
default_pitch = 15.5
default_nose = 0.0
default_roll = 14.7


def main():
    tab1, tab2 = st.tabs(['Simulator', 'FAQ'])
    with tab1:
        disc_names = {
            'Innova Wraith': 'dd2',
            'Innova Firebird': 'cd1',
            'Innova Roadrunner': 'cd5',
            'Innova Fairway Driver': 'fd2',
            }
        disc_selected = st.sidebar.selectbox("Disc Selection", disc_names.keys())
        disc_name = disc_names[disc_selected]

        # Create the sliders with the default values
        U = st.sidebar.slider("Throwing Velocity (m/s)", min_value=0.0, max_value=40.0, value=default_U, step=0.1,
                              help='Fastest Throw on record is ~40m/s by Simon Lizotte')
        omega = st.sidebar.slider("Omega", min_value=0.0, max_value=200.0, value=default_omega, step=0.1)
        z0 = st.sidebar.slider("Release Height (m)", min_value=0.0, max_value=2.0, value=default_z0, step=0.1)
        pitch = st.sidebar.slider("Pitch Angle (deg) | Release angle", min_value=0.0, max_value=90.0, value=default_pitch,
                                  step=0.1)
        nose = st.sidebar.slider("Nose Angle (deg) | Up/Down", min_value=0.0, max_value=90.0, value=default_nose,
                                 step=0.1)
        roll = st.sidebar.slider("Roll Angle (deg) | Tilt Left/Right", min_value=-90.0, max_value=90.0, value=default_roll,
                                 step=0.1)

        pos = np.array((0, 0, default_z0))
        disc_dict = DiscGolfDisc(disc_name)

        stl_mesh = get_stl(proj_dir / 'shotshaper' / 'discs' / (disc_name + '.stl'))
        fig = visualize_disc(stl_mesh, nose=nose, roll=roll)

        st.markdown("""## Disc orientation""")
        st.plotly_chart(fig)
        st.markdown("""## Flight Path""")
        shot = disc_dict.shoot(speed=U, omega=omega, pitch=pitch,
                               position=pos, nose_angle=nose, roll_angle=roll)

        # Plot trajectory
        x, y, z = shot.position
        x_new, y_new = -1 * y, x

        # Reversed x and y to mimic a throw
        fig = get_plot(x_new, y_new, z)
        st.plotly_chart(fig, True)

        st.markdown(
                f"""
        **Arrows in Blue** show you where your *s-turn* is.

        **Arrows in Red** show you your *max height* and *lateral deviance*.

        Hit Play to watch your animated throw.

        | Metric       | Value  |
        |--------------|--------|
        | Drift Left   | {round(min(x_new), 2)} |
        | Drift Right  | {round(max(x_new), 2)} |
        | Max Height   | {round(max(z), 2)}    |
        | Distance     | {round(max(y_new), 2)} |

        """
                )

        arc, alphas, betas, lifts, drags, moms, rolls = disc_dict.post_process(shot, omega)
        fig = get_subplots(arc, alphas, lifts, drags, moms, rolls, shot.velocity)
        st.plotly_chart(fig, True)

    with tab2:
        st.markdown("""
        # Motivation
        I saw some great work by [kegiljarhus](https://github.com/kegiljarhus) [repo](https://github.com/kegiljarhus/shotshaper)
        and wanted to make this available as an app so people could learn more about disc golf. I really want to commend
        the amazing idea of writing a [scientific article](https://link.springer.com/article/10.1007/s12283-022-00390-5)
         AND releasing code, and actually executing it well. This is what gets people excited about STEM.
        
        I originally saw this 
        [reddit post](https://www.reddit.com/r/discgolf/comments/yyhbcj/wrote_a_scientific_article_on_disc_golf_flight/)
        which really piqued my interest.
        
        # Questions
        - I imagine some of you will want to add your disc here, if you can convert your disc into an `.stl` then I will 
        add it to the database. If this gets common enough I will add an option to upload your own.
            - I imagine there will be a barrier to entry to do this.
            - If you have any ideas, just let me know in a discussion or in a pull request
        """)


if __name__ == "__main__":
    # Setting up Logger and proj_dir
    logger = getLogger(__name__)
    proj_dir = Path(__file__).parents[1]

    st.title("Disc Golf Simulator")

    # initialize_state()
    main()