192 lines
6.0 KiB
Plaintext
192 lines
6.0 KiB
Plaintext
|
#!/usr/bin/env python
|
||
|
# encoding: utf-8
|
||
|
|
||
|
|
||
|
class Expression(object):
|
||
|
"""A calculus expression. Today it can andle only expression with numbers later it will be able to manipulate unknown"""
|
||
|
|
||
|
priority = {"*" : 3, "/": 3, "+": 2, "-":2, "(": 1}
|
||
|
|
||
|
def __init__(self, exp):
|
||
|
""" Initiate the expression.
|
||
|
|
||
|
:param exp: the expression. It can be a string or a list of tokens. It can be infix or postfix expression
|
||
|
"""
|
||
|
self._tokens = {}
|
||
|
self._strings = {}
|
||
|
if type(exp) == list:
|
||
|
fix = self.get_fix(exp)
|
||
|
self._tokens[fix] = exp
|
||
|
elif type(exp) == str:
|
||
|
tokens = self.parseExp(exp)
|
||
|
fix = self.get_fix(tokens)
|
||
|
self._tokens[fix] = tokens
|
||
|
self._strings[fix] = exp
|
||
|
else:
|
||
|
raise ValueError("The expression needs to be a string or a list of token")
|
||
|
|
||
|
# ---------------------
|
||
|
# String parsing
|
||
|
|
||
|
# @classmethod ????
|
||
|
def parseExp(self, exp):
|
||
|
""" Parse the expression, ie tranform a string into a list of tokens
|
||
|
|
||
|
:param exp: The expression (a string)
|
||
|
:returns: list of token
|
||
|
|
||
|
"""
|
||
|
return exp.split(" ")
|
||
|
|
||
|
# ---------------------
|
||
|
# "fix" recognition
|
||
|
|
||
|
def get_fix(self, tokens):
|
||
|
""" Give the "fix" of an expression
|
||
|
infix -> A + B
|
||
|
prefix -> + A B
|
||
|
postfix -> A B +
|
||
|
|
||
|
:param exp: the expression (list of token)
|
||
|
:returns: the "fix" (infix, postfix, prefix)
|
||
|
|
||
|
"""
|
||
|
if tokens[0] in "+-*/":
|
||
|
return "prefix"
|
||
|
elif token[0] not in "+-*/" ans token[1] not in "+-*/":
|
||
|
return "postfix"
|
||
|
else:
|
||
|
return "infix"
|
||
|
|
||
|
# ----------------------
|
||
|
# Expressions - tokens getters
|
||
|
|
||
|
def tokens(self, fix = "infix"):
|
||
|
"""Get the list of tokens with the wanted fix
|
||
|
|
||
|
:param fix: the fix wanted (infix default)
|
||
|
:returns: list of tokens
|
||
|
|
||
|
"""
|
||
|
# Si ce fix n'a pas encore été enregistré
|
||
|
if fix not in self._tokens:
|
||
|
# Il peut venir de la version string
|
||
|
if fix in self._string:
|
||
|
self._tokens[fix] = self.parseExp(self._string[fix])
|
||
|
# Sinon il faut le calculer à partir d'une autre forme
|
||
|
else:
|
||
|
fix_transfo = "to" + fix.capitalize()
|
||
|
getattr(self,fix_transfo)()
|
||
|
|
||
|
return self._tokens[fix]
|
||
|
|
||
|
##### On en est là,il faudra faire attention à bien vérifier ce que les "to..." enregistre (string ou tokens)
|
||
|
|
||
|
def string(self, fix = "infix"):
|
||
|
"""Get the string with the wanted fix
|
||
|
|
||
|
:param fix: the fix wanted (infix default)
|
||
|
:returns: the string representing the expression
|
||
|
|
||
|
"""
|
||
|
# Si ce fix n'a pas encore été enregistré
|
||
|
if fix not in self._string:
|
||
|
# Il peut venir de la version string
|
||
|
if fix in self._tokens:
|
||
|
self._string[fix] = self.parseExp(self._string[fix])
|
||
|
# Sinon il faut le calculer à partir d'une autre forme
|
||
|
else:
|
||
|
fix_transfo = "to" + fix.capitalize()
|
||
|
getattr(self,fix_transfo)()
|
||
|
|
||
|
return self._string[fix]
|
||
|
|
||
|
# ----------------------
|
||
|
# "fix" tranformations
|
||
|
|
||
|
def toPostfix(self):
|
||
|
""" Transorm the expression into postfix form using the infix form"""
|
||
|
pass
|
||
|
|
||
|
def toInfix(self):
|
||
|
""" Tranform the expression into infix form using postfix form"""
|
||
|
pass
|
||
|
|
||
|
# ---------------------
|
||
|
# Tools for placing parenthesis in infix notation
|
||
|
|
||
|
def needPar(operande, operator, posi = "after"):
|
||
|
""" Says whether or not the operande needs parenthesis
|
||
|
|
||
|
:param operande: the operande
|
||
|
:param operator: the operator
|
||
|
:param posi: "after"(default) if the operande will be after the operator, "before" othewise
|
||
|
:returns: bollean
|
||
|
"""
|
||
|
if isNumber(operande) and "-" in operande:
|
||
|
return 1
|
||
|
elif not isNumber(operande):
|
||
|
# Si c'est une grande expression ou un chiffre négatif
|
||
|
stand_alone = get_main_op(operande)
|
||
|
# Si la priorité de l'operande est plus faible que celle de l'opérateur
|
||
|
minor_priority = self.priority[get_main_op(operande)] < self.priority[operator]
|
||
|
# Si l'opérateur est -/ pour after ou juste / pour before
|
||
|
special = (operator in "-/" and posi == "after") or (operator in "/" and posi == "before")
|
||
|
|
||
|
return stand_alone and (minor_priority or special)
|
||
|
else:
|
||
|
return 0
|
||
|
|
||
|
# J'aime pas bien cette endroit faudrait que ce soit une méthode qui s'applique uniquement à l'expression en question (self) pas à n'importe quel string, ça serait plus propre.
|
||
|
def get_main_op(exp):
|
||
|
""" Gives the main operation of the expression
|
||
|
|
||
|
:param exp: the expression
|
||
|
:returns: the main operation (+, -, * or /) or 0 if the expression is only one element
|
||
|
|
||
|
"""
|
||
|
parStack = Stack()
|
||
|
tokenList = exp.split(" ")
|
||
|
|
||
|
if len(tokenList) == 1:
|
||
|
# Si l'expression n'est qu'un élément
|
||
|
return 0
|
||
|
|
||
|
main_op = []
|
||
|
|
||
|
for token in tokenList:
|
||
|
if token == "(":
|
||
|
parStack.push(token)
|
||
|
elif token == ")":
|
||
|
parStack.pop()
|
||
|
elif token in "+-*/" and parStack.isEmpty():
|
||
|
main_op.append(token)
|
||
|
|
||
|
return min(main_op, key = lambda s: priority[s])
|
||
|
|
||
|
# ---------------------
|
||
|
# Computing the expression
|
||
|
|
||
|
def compute(self):
|
||
|
""" Recursive method for computing as a student the expression
|
||
|
:returns: list of steps needed to reach the result
|
||
|
|
||
|
"""
|
||
|
pass
|
||
|
|
||
|
def doMath(op, op1, op2):
|
||
|
"""Compute "op1 op op2"
|
||
|
|
||
|
:param op: operator
|
||
|
:param op1: first operande
|
||
|
:param op2: second operande
|
||
|
:returns: string representing the result
|
||
|
|
||
|
"""
|
||
|
return str(eval(op1 + op + op2))
|
||
|
|
||
|
# -----------------------------
|
||
|
# Reglages pour 'vim'
|
||
|
# vim:set autoindent expandtab tabstop=4 shiftwidth=4:
|
||
|
# cursor: 16 del
|