Calculator-for-Matrix-and-Algebra/nessesary/equation/processing.py

from nessesary.polynomial import Polynomial
from nessesary.parser.parser import is_op, insert_mul_sign, is_num
import re

def change_side(equa:str):
    """
    Change all left side expression of = sign
    to right side.
    >>> change_side("x+2-x^2 = 32-x^3")
    'x+2-x^2-32+x^3'
    """
    temp = ''
    for i in equa:
        if i == " ":
            temp += ""
        else:
            temp += i
    
    equa = temp
            
    llist = equa.split("=")
    left_side, right_side = llist[0], llist[1]

    temp = ''
    if (right_side[0] != "-") or (right_side[0] != "+"):
        temp += '+'
    temp += right_side
    right_side = temp
    for i in right_side:
        if i == "+":
            left_side += "-"
        elif i == "-":
            left_side += "+"
        else:
            left_side += i
    return left_side

def simplify(expr:str):
    """
    Use to_str() method of polynomial.
    to sum and sub all coefficient.
    """
    poly = Polynomial(expr)
    try:
        result = poly.to_str()
        return result
    except:
        raise ValueError

def solver(expr:str):
    """
    Solve expression that using polynomial solve().
    """
    poly = Polynomial(expr)
    try:
        result = poly.solve()
        return result
    except:
        raise ValueError

def poly_expand(expr:str) -> str:
    """
    Add Polynomial Text to all element that
    can make polynomial.
    >>> poly_expand("(3x+2)^3+4x+5-2+(5x^2)^3")
    'Polynomial((3x+2")**3+Polynomial("4x")+Polynomial("5x^2")**3'
    """
    # i = 0
    # result = ""
    # found = False
    # added = False
    # while i < len(expr):
    #     char = expr[i]
    #     if char == "(":
    #         result += "Polynomial("
    #         i += 1
    #         found = True
    #     else:
    #         if not found:
    #             if added and is_op(char):
    #                 result += f"){char}"

    #                 added = False
    #             if is_op(char) and char != "^":
    #                 result += f"{char}Polynomial("
    #                 i += 1
    #                 added = True
    #                 found = True
    #                 continue
    #         result += char
    #         i += 1

    #     if char == ")":
    #         found = False

    # result = result.replace("++", "+").replace("--","-")
    # result = result.replace("^^", "^")
    # result = result.replace("))", ")")
    # result = result.replace("^", "**")
    # result = result.replace("Polynomial(Polynomial(", "Polynomial(")
    # # return result
    # result = ''
    # # ['(', '3x', '+', '2', ')', '^', '3', '+', '4x', '+', '5', '-', '2', '+', '(', '5x', '^', '2', ')', '^', '3']
    pattern = re.compile('\[[^\]]+\]|<[^>]+>|\{[^\}]+\}|\d+[eE][-+]\d+|\w+(?=\()|\w+|[-+\/*%]|.')
    coeff_var_list = pattern.findall(expr)
    found = False
    result = ''
    for char in coeff_var_list:
        if is_num(char):
            result += char
            continue

        if is_op(char):
            if char == '(':
                result += 'Polynomial('
                found = True
                continue
            elif char == '^':
                result += char
                continue
            else:
                result += char
                continue
        if char == ')':
            if found:
                result += char
                found = False
                continue
            else:
                raise ValueError

        if not is_num(char):
            if found:
                result += char
                continue
            else:
                result += f'Polynomial(\'{char}\')'
    
    # print(expr, result)
    # Match:Whole poly | Group1:inside Parentheses | Group2:Expo
    pattern = re.compile('([+-^$|])\w+(?=\()\(([^()]*)\)+(\^?\d?)')
    coeff_var_list = pattern.findall(result)
    collected_list = []
    for tup in coeff_var_list: # sign, poly, expo
        sign = tup[0]
        if sign == 'P':
            sign = '+'
        poly = tup[1].replace("\"","")
        expo = tup[2]
        if expo == '':
            expo = 1
        else:
            expo = int(expo[1:])
    if sign == '-':
        text = f"{sign}{poly}"
    elif sign == '+':
        text = f"{poly}"
    to_append = Polynomial(text)**expo
    collected_list.append(to_append)

    res = None
    for p in collected_list:
        if res == None:
            res = p
        else:
            res += p

    return res