#!/usr/bin/env python3
"""Figure ch07 — répartition des charges communes annuelles entre les huit fractions (DONNÉES SYNTHÉTIQUES).
Sortie : assets/figures/ch07/ch07-fig-repartition-charges.pdf"""
import sys, pathlib
ROOT = pathlib.Path(__file__).resolve().parents[3]
sys.path.insert(0, str(ROOT / "analysis/python")); sys.path.insert(0, str(ROOT / "analysis/python/cases"))
import numpy as np, matplotlib.pyplot as plt
from traite.style import setup, palette, savefig, LARGEUR, fmt_dollars
import ch07_servitudes_copro as cas

def main():
    setup()
    qp = cas.quotes_parts(); A = cas.cotisation_seuil(); rep = cas.repartition_charges(A, qp)
    noms = [f"{n} ({s:.0f} m², q = {q:.4f})".replace(".", ",") for n, s, q in zip(qp["noms"], qp["superficies"], qp["q"])]
    blocs = [("Charges courantes (q_i × budget)", rep["par_fraction_courantes"], palette.cat[0]),
             ("Fonds de prévoyance (q_i × A*)", rep["par_fraction_fonds"], palette.cat[1]),
             ("Fonds d'auto assurance (q_i)", rep["par_fraction_auto"], palette.cat[2]),
             ("Stationnement à usage restreint (part égale)", rep["par_fraction_restreint"], palette.cat[3])]
    y = np.arange(len(noms)); gauche = np.zeros(len(noms))
    fig, ax = plt.subplots(figsize=(LARGEUR, 3.4))
    for lab, vals, col in blocs:
        ax.barh(y, vals, left=gauche, color=col, height=0.66, label=lab); gauche += vals
    for yi, tot in zip(y, rep["par_fraction_total"]):
        ax.text(tot + 150, yi, f"{tot:,.0f} $/an".replace(",", " "), va="center", fontsize=7.5, color=palette.ink)
    ax.set_yticks(y); ax.set_yticklabels(noms); ax.invert_yaxis()
    ax.xaxis.set_major_formatter(plt.FuncFormatter(fmt_dollars)); ax.set_xlim(0, rep["par_fraction_total"].max() * 1.25)
    ax.set_xlabel("Charges annuelles de la fraction (dollars de 2026, données synthétiques)")
    ax.grid(axis="y", visible=False)
    ax.legend(loc="upper center", bbox_to_anchor=(0.45, -0.2), ncol=2, fontsize=7.5)
    out = savefig(fig, "ch07/ch07-fig-repartition-charges.pdf"); print("OK", out)

if __name__ == "__main__":
    main()