#tutorial 
import matplotlib.pyplot as plt
from matplotlib.offsetbox import OffsetImage, AnnotationBbox
import numpy as np
import pandas as pd


color_dict = {"Norway": "#2B314D", "Denmark": "#A54836", "Sweden": "#5375D4" }

data = {
    "year": [2004, 2022, 2004, 2022, 2004, 2022],
    "countries" : [ "Denmark", "Denmark", "Norway", "Norway","Sweden", "Sweden"],
    "sites": [4,10,5,8,13,15]
}
df= pd.DataFrame(data)

df['ctry_code'] = df.countries.astype(str).str[:2].astype(str).str.upper()
df['year_lbl'] ="'"+df['year'].astype(str).str[-2:].astype(str)
df = df.sort_values(['countries' ,'sites' ], ascending=False ).reset_index(drop=True)
#map the colors of a dict to a dataframe
df['color']= df.countries.map(color_dict)
df

# Generate 10 plots in a grid with 3 rows, 5 columns
x_coord = [-0.1,0.4,0.55]
y_coord = [0.6,0.3, 1.2]
width = [1,0.4,0.6]
height = [1,0.4,0.6]

direction_sites = [-1, 0, 0]           #direction of the data labels
direction_lables = [1, -1, 1]          #direction of the year labels

axes_coord = pd.DataFrame({
    'x': x_coord,
    'y': y_coord,
    'w': width,
    'h': height,
    'dir_lbl': direction_lables,
    'dir_sites': direction_sites

})
axes_coord

fig = plt.figure(figsize=(5, 5))


for i, ((country, group), row) in enumerate(zip(df.groupby("countries", sort = False), axes_coord.itertuples())):

    #add the axes
    ax= fig.add_axes([row.x, row.y, row.w, row.h], polar=True)

    max_sites = group.sites.max()
    #add the data labels
    ax.text( 
        np.pi * row.dir_sites,
        max_sites + 2, 
        max_sites,
        color = group['color'].iloc[0], 
        size = 12, 
        weight = "bold", 
        ha="center", 
        va= "center" 
    )

    # add the flags
    image_box = OffsetImage(img[i], zoom=0.02)  # container for the image
    ab = AnnotationBbox(
        image_box, 
        (0, 0),
        frameon=False
        )
    ax.add_artist(ab)

    for g in group.itertuples():
        #add the cirlces    
        ax.plot(
            np.linspace(0, 2*np.pi, 100), 
            np.ones(100)*g.sites, 
            color=g.color, 
            linestyle='-',
            lw=2
            )
        ax.set_rgrids(range(0,max_sites))
        ax.xaxis.grid(False)
        ax.set_frame_on(False)
        ax.yaxis.grid(
            True,
            color=g.color,
            linestyle='-'
            )
        ax.patch.set_alpha(0.1) #transparent axis
        ax.tick_params(labelleft= False, labelbottom = False)

        #add the year lables
        ax.annotate(
            g.year_lbl, 
            xy = (np.pi/2 * row.dir_lbl, g.sites-0.1),
            color = g.color,
            ha="center", 
            va= "center" ,
            bbox=dict(
                facecolor='w', 
                edgecolor='w', 
                boxstyle='round,pad=0.2'
                )
            )