66 lines
2.2 KiB
Python
66 lines
2.2 KiB
Python
from latex2sympy import process_sympy
|
|
from sympy import *
|
|
import sys
|
|
import hashlib
|
|
import time
|
|
|
|
sys.path.append("..")
|
|
|
|
|
|
M = Matrix([[1, 2], [3, 4]])
|
|
v = Matrix([1, 2])
|
|
|
|
# sub settings
|
|
sub_settings_symbols = {}
|
|
sub_settings_symbols[Symbol('M' + hashlib.md5('M'.encode()).hexdigest(), real=True)] = M
|
|
sub_settings_symbols[Symbol('v' + hashlib.md5('v'.encode()).hexdigest(), real=True)] = v
|
|
|
|
|
|
# one parameters
|
|
latex = "\\begin{matrix}1&2\\\\3&4\\end{matrix}\\cdot[!v!]"
|
|
equation_sympy_check = MatMul(M, Symbol('v' + hashlib.md5('v'.encode()).hexdigest(), real=True))
|
|
equation_sympy_subs_check = MatMul(M, v)
|
|
# placeholders
|
|
equation_sympy = process_sympy(latex)
|
|
print('latex = %s' % latex)
|
|
print('equation_sympy = %s' % equation_sympy)
|
|
print('equation_sympy_check = %s' % equation_sympy_check)
|
|
print('equation_sympy = %s' % (srepr(equation_sympy)))
|
|
|
|
equation_sympy_subs = equation_sympy.subs(sub_settings_symbols, evaluate=False)
|
|
print('equation_sympy_subs = %s' % equation_sympy_subs)
|
|
print('equation_sympy_subs_check = %s' % equation_sympy_subs_check)
|
|
|
|
|
|
# two parameters
|
|
|
|
# sub settings
|
|
print('')
|
|
print('============== Two Parameters -> M*v = Matrix*Vector =============')
|
|
sub_settings_symbols = {}
|
|
sub_settings_symbols[Symbol('M' + hashlib.md5('M'.encode()).hexdigest(), commutative=False)] = M
|
|
sub_settings_symbols[Symbol('v' + hashlib.md5('v'.encode()).hexdigest(), commutative=False)] = v
|
|
|
|
latex = "[!M!]\\cdot[!v!]"
|
|
math_check = Mul(Symbol('M' + hashlib.md5('M'.encode()).hexdigest(), commutative=False), Symbol('v' + hashlib.md5('v'.encode()).hexdigest(), commutative=False))
|
|
# placeholders
|
|
equation_sympy = process_sympy(latex)
|
|
print(latex)
|
|
print(math_check)
|
|
print(equation_sympy)
|
|
print(srepr(equation_sympy))
|
|
|
|
# performance
|
|
t0 = time.time()
|
|
|
|
# process_sympy and substitute at the same time
|
|
# Only needed for linalg input
|
|
placeholder_values = {'M': M, 'v': v}
|
|
equation_sympy_subs = process_sympy(latex, variable_values=placeholder_values)
|
|
|
|
t1 = time.time()
|
|
print('equation with substituted placeholders = %s' % (str(equation_sympy_subs)))
|
|
print('time to process to sympy with placeholders = %s s' % (t1 - t0))
|
|
print('')
|
|
print('============== Two Parameters -> M*v = Matrix*Vector =============')
|