from Pile import Pile_chaine as Pile
from math import *
from infixe import estUnNombre

class Expression:
    def __init__(self, racine, gauche, droit):
        """initialisation d'un objet 'Expression'"""
        self.racine = racine
        self.gauche = gauche
        self.droit = droit
    
    def evalue(self):
        """renvoie la valeur de l'expression
        si 1 seul argument alors on prend le fils gauche"""
        #fonctions
        if self.racine == "sin": return sin(self.gauche.evalue())
        if self.racine == "cos": return cos(self.gauche.evalue())
        if self.racine == "tan": return tan(self.gauche.evalue())
        if self.racine == "opp": return - self.gauche.evalue()        
        #operateurs
        if self.racine == "+": return self.gauche.evalue() + self.droit.evalue()
        if self.racine == "-": return self.gauche.evalue() - self.droit.evalue()
        if self.racine == "*": return self.gauche.evalue() * self.droit.evalue()
        if self.racine == "/": return self.gauche.evalue() / self.droit.evalue()
        if self.racine == "^": return self.gauche.evalue() ** self.droit.evalue()
        
        
        if "." in self.racine:
            return float(self.racine)
        return int(self.racine)
   
    def __str__(self):
        """affiche l'expression"""
        priorite = {"opp":1,
                    "!":1,
                    "^":2,
                    "sin":3,
                    "cos":3,
                    "tan":3,
                    "*":4,
                    "/":4,
                    "+":5,
                    "-":5
                    }
        associativite = {"opp":"droit",
                        "^":"droit",
                         "*":"gauche",
                        "/":"gauche",
                        "+":"gauche",
                        "-":"gauche"
                        }
        pas_commutatif = ["^","/","-"]
        if self.racine in ["opp","!", "^", "*", "/", "+", "-"]:
            gauche = str(self.gauche)
            
            #definitivement à améliorer
            if self.droit == None:
                # racine est "-" (unaire) ou "opp"
                priorite_racine = priorite[self.racine]
                priorite_gauche = priorite.get(self.gauche.racine, -1)
                #test
                print(self.gauche.racine)
                #test
                if self.gauche.racine[0] == "-" or priorite_racine <= priorite_gauche:
                    gauche = parentheses(gauche)
                return "-" + gauche
            
            droit = str(self.droit)
            
            priorite_racine = priorite[self.racine]
            priorite_gauche = priorite.get(self.gauche.racine, -1)
            priorite_droit = priorite.get(self.droit.racine, -1)
            
            if self.gauche.racine == "opp" or self.gauche.racine != "-" and self.gauche.racine[0] == "-":
                gauche = parentheses(gauche)
            if self.droit.racine == "opp" or self.droit.racine != "-" and self.droit.racine[0] == "-":
                droit = parentheses(droit)
            
            if priorite_racine < priorite_gauche:
                gauche = parentheses(gauche)
            if priorite_racine < priorite_droit:
                droit = parentheses(droit)
            if self.racine in pas_commutatif:
                if priorite_racine == priorite_droit and associativite[self.racine] == "gauche":
                    droit = parentheses(droit)
                if priorite_racine == priorite_gauche and associativite[self.racine] == "droit":
                    gauche = parentheses(gauche)
                
            return gauche + self.racine + droit
        
        if self.racine in ["sin","cos","tan"]:
            return self.racine + "(" + str(self.gauche) + ")"
        return str(self.racine)
        
        
def parentheses(txt):
    """ajoute des parentheses à txt"""
    return "(" + txt + ")"
def npi2tree_original(liste_en_npi):
    """conversion d'une liste en NPI en un arbre"""
    pile_expr = Pile()
    for element in liste_en_npi:
        if element in ["+", "*","-","/", "^"]:
            fils_droit = pile_expr.depiler()
            fils_gauche = pile_expr.depiler()
            pile_expr.empiler(Expression(element, fils_gauche, fils_droit))
        elif element in ["sin","cos","tan", "opp"]:
            pile_expr.empiler(Expression(element, pile_expr.depiler(),None))
        else:
            pile_expr.empiler(Expression(element, None, None))
    return pile_expr.sommet()

#renommer la fonction !...
def npi2tree(liste, etat_affichage):
    """conversion d'une liste en NPI ou NP en un arbre"""
    pile_expr = Pile()
    for element in liste:
        if element in ["+", "*","/", "^"]:
            fils_gauche = pile_expr.depiler()
            fils_droit = pile_expr.depiler()
            
            if etat_affichage == "post-expr" or etat_affichage == "inf":
                fils_gauche, fils_droit = fils_droit, fils_gauche
            
            pile_expr.empiler(Expression(element, fils_gauche, fils_droit))
            
        elif element == "-":
            fils_gauche = pile_expr.depiler()
            if pile_expr.est_vide():
                # "-" est l'opposé
                pile_expr.empiler(Expression("-", fils_gauche, None))
            else:
                # "-" est la soustraction
                fils_droit = pile_expr.depiler()
                if etat_affichage == "post-expr" or etat_affichage == "inf":
                    fils_gauche, fils_droit = fils_droit, fils_gauche
                
                pile_expr.empiler(Expression(element, fils_gauche, fils_droit))
                
        elif element in ["sin","cos","tan", "opp"]:
            pile_expr.empiler(Expression(element, pile_expr.depiler(), None))
        else:
            pile_expr.empiler(Expression(element, None, None))
    return pile_expr.sommet()