import numpy as np
import sympy as s
from sympy import pprint, Symbol, sin, exp, sqrt, series, latex, symbols, simplify, expand
from sympy import init_printing

cols = 2
rows = 5

def pl(expression):
    print(latex(expression))


def pls(expression):
    print(latex(simplify(expression)))


def ple(expression):
    print(latex(expand(expression)))


def l(expression):
    return latex(expression)


def ls(expression):
    return latex(simplify(expression))


def le(expression):
    return latex(expand(expression))

def a2list(list):
    temp = []
    for i in range(len(list)):
        temp.append(list[i])
    return temp

def a2l(list):
    temp = []
    for i in range(len(list)):
        temp.append(l(list[i]))
    return temp
def a2le(list):
    temp = []
    for i in range(len(list)):
        temp.append(le(list[i]))
    return temp

def binomischeform1(test1):
    out_ang = "\\begin{equation*} \n"
    for i in range(len(a2l(test1))):
        if i%rows is 0:
            out_ang += "\\begin{aligned}[c]\n"+a2l(test1)[i]+"\\\\"
        if i%rows is not 0 and i%rows is not rows-1:
            out_ang += a2l(test1)[i] + " \\\\ "
        if i%rows is rows-1:
            out_ang += a2l(test1)[i]+"\n\\end{aligned} \n\\qquad\\qquad \n"
        #out_ang = " \\\\ ".join(a2l(test1))+""
    out_ang += "\\end{equation*}"
    out_los = "\\section{Lösung:}\n\\begin{equation*} \n \hspace{-" + str(cols * 20) + "px}"
    for i in range(len(a2l(test1))):
        if i % rows is 0:
            out_los += "\\begin{aligned}[c]\n" + a2le(test1)[i] + "\\\\"
        if i % rows is not 0 and i % rows is not rows - 1:
            out_los += a2le(test1)[i] + " \\\\ "
        if i % rows is rows - 1:
            out_los += a2le(test1)[i] + "\n\\end{aligned} \n\\qquad \n"
        # out_ang = " \\\\ ".join(a2l(test1))+""
    out_los += "\\end{equation*}"
    return out_ang, out_los
x, y, z, a, b, c = symbols('x y z a b c')

test1 = (np.random.randint(1, 11, cols * rows)*a+np.random.randint(1, 11, cols*rows)*b)**2


#print(out_ang)

#print("{0}\n{1}".format(binomischeform1(test1)))
#print(len(binomischeform1(test1)))

b = binomischeform1(test1)

# print(b[0])
# print(b[1])

f = open("latexout2.tex", "w")
f.write("{}\n{}".format(b[0],b[1]))

#print("{}\n{}".format(b[0],b[1]))