"""
Annotate stacked bar
====================

"""
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
import marsilea as ma
import marsilea.plotter as mp

# sphinx_gallery_start_ignore
import mpl_fontkit as fk
fk.install("Lato", verbose=False)
# sphinx_gallery_end_ignore

np.random.seed(0)
data = np.random.randint(1, 100, (5, 10))
data = pd.DataFrame(data=data, index=list("abcde"))

bar = mp.StackBar(data / data.sum(), width=.9,
                  legend_kws=dict(title="Stacked Bar"))
top_colors = mp.Colors([1, 1, 2, 2, 4], cmap="Set2", label="Category")
top_mesh = mp.ColorMesh(data.sum(), cmap="cool", label="Total",
                        cbar_kws=dict(orientation="horizontal"))

wb = ma.WhiteBoard(width=3, height=3, margin=.2)
wb.add_layer(bar)
wb.add_top(top_colors, size=.2)
wb.add_top(top_mesh, size=.2, pad=.1)
wb.add_legends()
wb.set_margin(.2)
wb.render()


# sphinx_gallery_start_ignore
if '__file__' in globals():
    from pathlib import Path
    import matplotlib.pyplot as plt
    save_path = Path(__file__).parent / "imgs"
    plt.savefig(save_path / "stacked_bar.svg", bbox_inches="tight")
# sphinx_gallery_end_ignore