Mapytex/mapytex/calculus/core/renders/tree2tex.py

238 lines
5.8 KiB
Python
Raw Normal View History

2018-03-10 13:08:28 +00:00
#! /usr/bin/env python
# -*- coding: utf-8 -*-
# vim:fenc=utf-8
#
# Copyright © 2017 lafrite <lafrite@Poivre>
#
# Distributed under terms of the MIT license.
from mapytex.calculus.core.operator import OPERATORS
2019-05-14 04:55:56 +00:00
__all__ = ["tree2tex"]
2018-03-10 13:08:28 +00:00
def plus2tex(left, right):
r""" + rendering
>>> from ..MO import MO
>>> plus2tex(MO.factory(2), MO.factory(3))
2018-03-10 13:08:28 +00:00
'2 + 3'
>>> from ..tree import Tree
>>> t = Tree.from_str("1+2")
>>> plus2tex(t, MO.factory(3))
2018-03-10 13:08:28 +00:00
'1 + 2 + 3'
>>> plus2tex(t, MO.factory(-3))
2018-03-10 13:08:28 +00:00
'1 + 2 - 3'
>>> plus2tex(MO.factory(-3), t)
2018-03-10 13:08:28 +00:00
'- 3 + 1 + 2'
>>> t = Tree.from_str("-2*3")
>>> plus2tex(MO.factory(3), t)
2018-03-10 13:08:28 +00:00
'3 - 2 \\times 3'
"""
left_ = render_with_parenthesis(left, "+")
right_ = render_with_parenthesis(right, "+")
display_plus = True
if right_.startswith("-"):
display_plus = False
2018-03-10 13:08:28 +00:00
if display_plus:
return f"{left_} + {right_}"
return f"{left_} {right_}"
2019-05-14 04:55:56 +00:00
2018-03-10 13:08:28 +00:00
def minus2tex(left, right):
r""" - rendering
>>> from ..MO import MO
>>> minus2tex(None, MO.factory(3))
2018-03-10 13:08:28 +00:00
'- 3'
>>> from ..tree import Tree
>>> t = Tree.from_str("1+2")
>>> minus2tex(None, t)
'- (1 + 2)'
"""
try:
right_need_parenthesis = False
2019-05-14 04:55:56 +00:00
if OPERATORS[right.node]["precedence"] < OPERATORS["-"]["precedence"]:
2018-03-10 13:08:28 +00:00
right_need_parenthesis = True
except AttributeError:
right_ = right.__tex__
else:
if right_need_parenthesis:
right_ = f"({tree2tex(right)})"
else:
right_ = tree2tex(right)
return f"- {right_}"
2019-05-14 04:55:56 +00:00
2018-03-10 13:08:28 +00:00
def mul2tex(left, right):
r""" * rendering
>>> from ..MO import MO
>>> mul2tex(MO.factory(2), MO.factory(3))
2018-03-10 13:08:28 +00:00
'2 \\times 3'
>>> from ..tree import Tree
>>> t = Tree.from_str("1*2")
>>> mul2tex(t, MO.factory(3))
2018-03-10 13:08:28 +00:00
'1 \\times 2 \\times 3'
>>> t = Tree.from_str("1+2")
>>> mul2tex(t, MO.factory(3))
2018-03-10 13:08:28 +00:00
'(1 + 2) \\times 3'
>>> mul2tex(MO.factory(3), t)
2018-03-10 13:08:28 +00:00
'3(1 + 2)'
>>> a = MO.factory('x')
>>> mul2tex(MO.factory(3), a)
2018-03-10 13:08:28 +00:00
'3x'
>>> mul2tex(MO.factory(-3), a)
'- 3x'
>>> mul2tex(a, a)
'x \\times x'
2018-03-10 13:08:28 +00:00
"""
left_ = render_with_parenthesis(left, "*")
right_ = render_with_parenthesis(right, "*")
2018-03-10 13:08:28 +00:00
display_time = True
# if (right_[0].isalpha() and (left_.isnumeric() or left_.isdecimal())) or right_[
# 0
# ] == "(":
# display_time = False
if right_[0].isalpha():
# TODO: C'est bien beurk en dessous... |ven. déc. 21 12:03:07 CET 2018
if type(left).__name__ == "MOnumber":
display_time = False
elif right_[0] == "(":
2019-05-14 04:55:56 +00:00
display_time = False
2018-03-10 13:08:28 +00:00
if display_time:
return f"{left_} \\times {right_}"
2018-03-13 11:43:48 +00:00
return f"{left_}{right_}"
2018-03-10 13:08:28 +00:00
2019-05-14 04:55:56 +00:00
2018-03-10 13:08:28 +00:00
def div2tex(left, right):
r""" / rendering
>>> from ..MO import MO
>>> div2tex(MO.factory(2), MO.factory(3))
2018-03-10 13:08:28 +00:00
'\\frac{2}{3}'
>>> from ..tree import Tree
>>> t = Tree.from_str("1/2")
>>> div2tex(t, MO.factory(3))
2018-03-10 13:08:28 +00:00
'\\frac{\\frac{1}{2}}{3}'
>>> t = Tree.from_str("1+2")
>>> div2tex(t, MO.factory(3))
2018-03-10 13:08:28 +00:00
'\\frac{1 + 2}{3}'
>>> t = Tree.from_str("1*2")
>>> div2tex(MO.factory(3), t)
2018-03-10 13:08:28 +00:00
'\\frac{3}{1 \\times 2}'
"""
try:
left_ = tree2tex(left)
except (AttributeError, ValueError):
2018-03-10 13:08:28 +00:00
left_ = left.__tex__
try:
right_ = tree2tex(right)
except (AttributeError, ValueError):
2018-03-10 13:08:28 +00:00
right_ = right.__tex__
return "\\frac{" + left_ + "}{" + right_ + "}"
2019-05-14 04:55:56 +00:00
2018-03-10 13:08:28 +00:00
def pow2tex(left, right):
r""" ^ rendering
>>> from ..MO import MO
>>> pow2tex(MO.factory(2), MO.factory(3))
2018-03-14 08:40:57 +00:00
'2^{3}'
2018-03-10 13:08:28 +00:00
>>> from ..tree import Tree
>>> t = Tree.from_str("1^2")
>>> pow2tex(t, MO.factory(3))
2018-03-14 08:40:57 +00:00
'1^{2}^{3}'
>>> t = Tree.from_str("1^2")
>>> pow2tex(MO.factory(3), t)
2018-03-14 08:40:57 +00:00
'3^{1^{2}}'
2018-03-10 13:08:28 +00:00
>>> t = Tree.from_str("1+2")
>>> pow2tex(t, MO.factory(3))
2018-03-14 08:40:57 +00:00
'(1 + 2)^{3}'
2018-03-10 13:08:28 +00:00
>>> t = Tree.from_str("1*2")
>>> pow2tex(MO.factory(3), t)
2018-03-14 08:40:57 +00:00
'3^{1 \\times 2}'
2018-03-10 13:08:28 +00:00
"""
try:
left_need_parenthesis = False
2019-05-14 04:55:56 +00:00
if OPERATORS[left.node]["precedence"] < OPERATORS["^"]["precedence"]:
2018-03-10 13:08:28 +00:00
left_need_parenthesis = True
except AttributeError:
left_ = left.__tex__
else:
if left_need_parenthesis:
left_ = f"({tree2tex(left)})"
else:
left_ = tree2tex(left)
2018-03-14 08:40:57 +00:00
2018-03-10 13:08:28 +00:00
try:
2018-03-14 08:40:57 +00:00
right.node
2018-03-10 13:08:28 +00:00
except AttributeError:
right_ = right.__tex__
else:
2018-03-14 08:40:57 +00:00
right_ = tree2tex(right)
2018-03-10 13:08:28 +00:00
2018-03-14 08:40:57 +00:00
return f"{left_}^{{{right_}}}"
2018-03-10 13:08:28 +00:00
def render_with_parenthesis(subtree, operator):
subtree_need_parenthesis = False
try:
subtree.node
except AttributeError:
try:
2019-05-14 04:55:56 +00:00
if (
OPERATORS[subtree.MAINOP]["precedence"]
< OPERATORS[operator]["precedence"]
):
subtree_need_parenthesis = True
except (AttributeError, KeyError):
pass
2019-07-11 16:03:49 +00:00
subtree_ = subtree.__tex__
else:
if OPERATORS[subtree.node]["precedence"] < OPERATORS[operator]["precedence"]:
subtree_need_parenthesis = True
subtree_ = tree2tex(subtree)
if subtree_need_parenthesis:
return f"({subtree_})"
return subtree_
2019-05-14 04:55:56 +00:00
OPERATOR2TEX = {"+": plus2tex, "-": minus2tex, "*": mul2tex, "/": div2tex, "^": pow2tex}
2018-03-10 13:08:28 +00:00
def tree2tex(tree):
r""" Convert a tree into its tex version
It calls __tex__ to render MOs.
2018-09-17 14:27:20 +00:00
:param tree: tree to render
:example:
2018-03-10 13:08:28 +00:00
>>> from ..tree import Tree
>>> t = Tree.from_str("2+3*4")
>>> tree2tex(t)
'2 + 3 \\times 4'
"""
from ..tree import Tree
if not isinstance(tree, Tree):
raise ValueError(f"Can only render a Tree (got {type(tree).__name__}: {tree})")
2018-03-13 11:43:48 +00:00
return OPERATOR2TEX[tree.node](tree.left_value, tree.right_value)
2018-03-10 13:08:28 +00:00
# -----------------------------
# Reglages pour 'vim'
# vim:set autoindent expandtab tabstop=4 shiftwidth=4:
# cursor: 16 del