# Fonction pour afficher le menu
from kandinsky import *

fill_rect(0,0,320,222,color(255,255,255))
def afficher_menu():
    print("Menu:")
    print("1. Calculer l'énergie (P = U * I)")
    print("2. Calculer la loi d'Ohm (U = R * I)")
    print("3. Calculer l'énergie et le travail")
    print("4. Quitter")

# Fonction pour traiter l'option sélectionnée
fill_rect(0,0,320,222,color(255,255,255))
def traiter_option(option):
    if option == "1":
        print("Vous avez sélectionné l'option 1.")
        fill_rect(0,0,320,222,color(255,255,255))
        def calculer_energie():
            choix_formule = input("Choisissez une formule de puissance :"
                                  "\n1. P = U * I (Tension * Courant)"
                                  "\n2. P = F * V (Force * Vitesse)"
                                  "\n3. P = C * v (Couple * Vitesse angulaire)"
                                  "\n4. P = Q * p (Charge * Pression)"
                                  "\nEntrez le numéro de la formule choisie : ")

            if choix_formule == "1":
                U = float(input("Entrez la valeur de la tension (U) : "))
                I = float(input("Entrez la valeur du courant (I) : "))
                P = U * I
                print("La puissance (P) est :", U, "V *", I, "A =", P, "W")
            elif choix_formule == "2":
                F = float(input("Entrez la valeur de la force (F) : "))
                V = float(input("Entrez la valeur de la vitesse (V) : "))
                P = F * V
                print("La puissance (P) est :", F, "N *", V, "m/s =", P, "W")
            elif choix_formule == "3":
                C = float(input("Entrez la valeur du couple (C) : "))
                v = float(input("Entrez la valeur de la vitesse angulaire (v) : "))
                P = C * v
                print("La puissance (P) est :", C, "N.m *", v, "rad/s =", P, "W")
            elif choix_formule == "4":
                Q = float(input("Entrez la valeur de la charge (Q) : "))
                p = float(input("Entrez la valeur de la pression (p) : "))
                P = Q * p
                print("La puissance (P) est :", Q, "C *", p, "Pa =", P, "W")
            else:
                print("Formule invalide. Veuillez choisir une formule valide.")

        calculer_energie()

    elif option == "2":
        print("Vous avez sélectionné l'option 2.")
        fill_rect(0,0,320,222,color(255,255,255))
        def calculer_loi_ohm():
            choix_calcul = input("Choisissez le calcul à effectuer :"
                                 "\n1. Calculer la tension (U)"
                                 "\n2. Calculer le courant (I)"
                                 "\nEntrez le numéro du calcul choisi : ")

            if choix_calcul == "1":
                R = float(input("Entrez la valeur de la résistance (R) : "))
                I = float(input("Entrez la valeur du courant (I) : "))
                U = R * I
                print("La tension (U) est :", R, "Ω *", I, "A =", U, "V")
            elif choix_calcul == "2":
                U = float(input("Entrez la valeur de la tension (U) : "))
                R = float(input("Entrez la valeur de la résistance (R) : "))
                I = U / R
                print("Le courant (I) est :", U, "V /", R, "Ω =", I, "A")
            else:
                print("Calcul invalide. Veuillez choisir un calcul valide.")

        calculer_loi_ohm()

    elif option == "3":
        print("Vous avez sélectionné l'option 3.")
        fill_rect(0,0,320,222,color(255,255,255))
        def calculer_energie_travail():
            choix_calcul = input("Choisissez le calcul à effectuer :"
                                 "\n1. Calculer l'énergie potentielle (E_pot)"
                                 "\n2. Calculer l'énergie cinétique (E_cin)"
                                 "\n3. Calculer l'énergie mécanique (E_mec)"
                                 "\n4. Calculer le travail (W)"
                                 "\nEntrez le numéro du calcul choisi : ")

            if choix_calcul == "1":
                m = float(input("Entrez la valeur de la masse (m) : "))
                g = float(input("Entrez la valeur de l'accélération due à la gravité (g) : "))
                h = float(input("Entrez la valeur de la hauteur (h) : "))
                E_pot = m * g * h
                print("L'énergie potentielle (E_pot) est :", m, "kg *", g, "m/s^2 *", h, "m =", E_pot, "J")
            elif choix_calcul == "2":
                m = float(input("Entrez la valeur de la masse (m) : "))
                v = float(input("Entrez la valeur de la vitesse (v) : "))
                E_cin = 0.5 * m * v**2
                print("L'énergie cinétique (E_cin) est :", 0.5, "*", m, "kg *", v, "m/s^2 =", E_cin, "J")
            elif choix_calcul == "3":
                E_pot = float(input("Entrez la valeur de l'énergie potentielle (E_pot) : "))
                E_cin = float(input("Entrez la valeur de l'énergie cinétique (E_cin) : "))
                E_mec = E_pot + E_cin
                print("L'énergie mécanique (E_mec) est :", E_pot, "J +", E_cin, "J =", E_mec, "J")
            elif choix_calcul == "4":
                F = float(input("Entrez la valeur de la force (F) : "))
                d = float(input("Entrez la valeur de la distance (d) : "))
                W = F * d
                print("Le travail (W) est :", F, "N *", d, "m =", W, "J")
            else:
                print("Calcul invalide. Veuillez choisir un calcul valide.")

        calculer_energie_travail()

    elif option == "4":
        print("Au revoir !")
        exit()

    else:
        print("Option invalide. Veuillez sélectionner une option valide.")


# Boucle principale du programme
while True:
    # Afficher le menu
    afficher_menu()

    # Demander à l'utilisateur de saisir une option
    choix = input("Entrez le numéro de l'option choisie : ")

    # Traiter l'option sélectionnée
    traiter_option(choix)