mathai 0.1.4__py3-none-any.whl → 0.1.5__py3-none-any.whl

mathai/__init__.py

@@ -1,10 +1,10 @@
 from .expand import expand
 from .parser import parse
-from .printeq import printeq, printeq_log
+from .printeq import printeq, printeq_log, printeq_str
 from .simplify import solve, simplify
-from .integrate import integrate, typesqint
+from .integrate import integrate, sqint, byparts
 from .diff import diff
-from .factor import factor
+from .factor import factor, factor2
 from .fraction import fraction
 from .inverse import inverse
 from .trig import trig0, trig1, trig2, trig3, trig4

mathai/apart.py

@@ -7,7 +7,7 @@ from .base import *
 import math
 from .tool import poly
 
-def apart(eq, v):
+def apart(eq, v="v_0"):
     eq = simplify(eq)
     if eq.name != "f_mul":
         return eq

mathai/console.py

@@ -3,7 +3,7 @@ from .expand import expand
 from .parser import parse
 from .printeq import printeq, printeq_log
 from .simplify import solve, simplify
-from .integrate import integrate, typesqint
+from .integrate import integrate, typesqint, typebyparts, typeintegrate
 from .diff import diff
 from .base import *
 from .factor import _factorconst, factor
@@ -58,9 +58,13 @@ def console():
                 eq = simplify(eq)
             elif command == "fraction":
                 eq = fraction(eq)
-            elif command.split(" ")[0] in ["integrate", "sqint"]:
+            elif command.split(" ")[0] in ["integrate", "sqint", "byparts"]:
                 if command.split(" ")[0] == "sqint":
                     typesqint()
+                elif command.split(" ")[0] == "byparts":
+                    typebyparts()
+                elif command.split(" ")[0] == "integrate":
+                    typeintegrate()
                 out = integrate(eq, parse(command.split(" ")[1]).name)
                 if out is None:
                     print("failed to integrate")

mathai/factor.py

@@ -79,8 +79,9 @@ def factor_quad_formula_init():
     return [formula_list, var, expr]
 
 def factor_cube_formula_init():
-    var = "x"
-    formula_list = [(f"D^3+E", f"(D+E^(1/3))*(D^2-D*E^(1/3)+E^(2/3))"), (f"D^3-E", f"(D-E^(1/3))*(D^2+D*E^(1/3)+E^(2/3))")]
+    var = ""
+    formula_list = [(f"D^3+E", f"(D+E^(1/3))*(D^2-D*E^(1/3)+E^(2/3))"), (f"D^3-E", f"(D-E^(1/3))*(D^2+D*E^(1/3)+E^(2/3))"),\
+                    (f"-D^3+E", f"(-D+E^(1/3))*(D^2+D*E^(1/3)+E^(2/3))")]
     formula_list = [[simplify(parse(y)) for y in x] for x in formula_list]
     expr = [[parse("A")], [parse("B")]]
     return [formula_list, var, expr]
@@ -101,8 +102,11 @@ def factor_helper(equation, complexnum, power=2):
         nonlocal maxnum
         if eq.name == "f_pow" and eq.children[1].name[:2] == "d_":
             n = int(eq.children[1].name[2:])
+            sgn = round(abs(n)/n)
+            n = abs(n)
             if n>power and n % power == 0 and maxnum==n:
-                return (eq.children[0]**tree_form("d_"+str(int(n/power))))**power
+                out= (eq.children[0]**tree_form("d_"+str(sgn*int(n/power))))**power
+                return out
         return TreeNode(eq.name, [helper(child) for child in eq.children])
     high(equation)
     out = None
@@ -114,7 +118,8 @@ def factor_helper(equation, complexnum, power=2):
         out = simplify(solve(out))
     if out is not None and (complexnum or (not complexnum and not contain(out, tree_form("s_i")))):
         return out
-    else:
-        return TreeNode(equation.name, [factor_helper(child, complexnum, power) for child in equation.children])
+    return TreeNode(equation.name, [factor_helper(child, complexnum, power) for child in equation.children])
 def factor(equation, complexnum=False):
-    return solve(take_common2(solve(factor_helper(simplify(solve(factor_helper(simplify(equation), complexnum, 2))), complexnum, 3))))
+    return solve(take_common2(simplify(factor_helper(simplify(equation), complexnum, 2))))
+def factor2(equation, complexnum=False):
+    return solve(factor_helper(simplify(factor_helper(simplify(equation), complexnum, 2)), complexnum, 3))

mathai/integrate.py

@@ -9,8 +9,11 @@ from .printeq import printeq_str
 from .structure import transform_formula
 from .inverse import inverse
 from .tool import poly
+from fractions import Fraction
+from .printeq import printeq
 def integrate_summation(equation, wrt, tab, inf):
     logs= []
+    orig = copy.deepcopy(equation)
     for i in range(2):
         
         if equation.name == "f_add":
@@ -25,17 +28,23 @@ def integrate_summation(equation, wrt, tab, inf):
             return summation(answer), logs
         if i == 0:
             
-            tmp = expand(equation)
+            tmp = expand(simplify(fraction(simplify(equation))))
+            
             logs += [(tab, f"integrating {printeq_str(simplify(equation))} will be the same thing as integrating {printeq_str(simplify(tmp))}")]
+            
             equation = tmp
+            if equation.name != "f_add" and orig != equation:
+                out = integrate(equation, wrt, tab+1, inf)
+                if out is None:
+                    return None
+                return out[0], logs+out[1]
     return None
 
 def subs_heuristic(eq, var):
     output = []
     def collect2(eq):
-        if eq.name == "f_pow" and eq.children[0] == tree_form(var) and eq.children[1].name[:2]=="d_":
-            if int(eq.children[1].name[2:])==6:
-                output.append(str_form( tree_form(var)**tree_form("d_3") ))
+        if eq.name == "f_pow" and frac(eq.children[1]) is not None and abs(frac(eq.children[1])) == Fraction(1,2):
+            output.append(str_form(eq.children[0].fx("sqrt")))
         if eq.name in ["f_pow", "f_sin", "f_cos", "f_arcsin"] and eq.children[0].name[:2] != "v_" and var in str_form(eq.children[0]):
             output.append(str_form(eq.children[0]))
         if eq.name == "f_pow" and eq.children[0].name == "s_e" and "v_" in str_form(eq):
@@ -54,10 +63,12 @@ def subs_heuristic(eq, var):
     if output == []:
         collect3(eq)
     tmp = sorted(output, key=lambda x: len(x))
-    tmp = [solve(tree_form(x)) for x in tmp]
+    tmp = [simplify(tree_form(x)) for x in tmp]
+    
     return tmp
 
 def integrate_subs(equation, term, v1, v2, tab, inf):
+    
     origv2 = copy.deepcopy(v2)
     equation = solve(equation)
     eq = equation
@@ -83,6 +94,7 @@ def integrate_subs(equation, term, v1, v2, tab, inf):
             continue
         
         eq = expand(simplify(eq))
+        
         out = integrate(eq, origv2, tab+1, inf)
        
         if out is None:
@@ -103,7 +115,7 @@ def integrate_subs_main(equation, wrt, tab, inf):
         if tmp3 is not None:
             return tmp3[0], tmp3[1]
     return None
-def sqint(equation, var, depth, inf):
+def sqint(equation, var="v_0", depth=0, inf=0):
     
     logs = []
     def sgn(eq):
@@ -148,6 +160,7 @@ def sqint(equation, var, depth, inf):
         t1, t2 = sgn(const)
         la = s2**root
         lb = b*s2**root/(two*a)
+        
         if mode:
             if s1 == one:
                 if t1 == one:
@@ -162,9 +175,10 @@ def sqint(equation, var, depth, inf):
                     return -k*((la*x+lb)/t2**root).fx("arctan")/(la * t2**root), logs
         if s1 == one:
             if t1 == one:
-                return k*(la*x + lb + ((la*x + lb)**two + t2)**root).fx("abs").fx("log")/la, logs
+                return simplify(k*(la*x + lb + ((la*x + lb)**two + t2)**root).fx("abs").fx("log")/la), logs
             else:
-                return k*(la*x + lb + ((la*x + lb)**two - t2)**root).fx("abs").fx("log")/la, logs
+                return simplify(k*(la*x + lb + ((la*x + lb)**two - t2)**root).fx("abs").fx("log")/la), logs
+                
         else:
             if t1 == one:
                 return k*((la*x + lb)/t2**root).fx("arcsin")/la, logs
@@ -191,7 +205,6 @@ def sqint(equation, var, depth, inf):
             else:
                 log2 = tmp[1]
                 tmp = tmp[0]
-                
             return A*two*t**root + tmp*B, logs+log2
         else:
             tmp = sqint(simplify(one/t), var, depth, inf)
@@ -206,28 +219,65 @@ def sqint(equation, var, depth, inf):
             else:
                 log2 = tmp[1]
                 tmp = tmp[0]
-                
             return A*t.fx("abs").fx("log") + tmp*B, logs+log2
     return None
+typeint = "integrate"
+def typeintegrate():
+    global typeint
+    typeint=  "integrate"
+def typesqint():
+    global typeint
+    typeint=  "sqint"
+def typebyparts():
+    global typeint
+    typeint=  "byparts"
+def byparts(eq, wrt="v_0", tab=0, inf=0):
+    
+    lst = factor_generation(eq)
+    if len(lst) == 1:
+        lst += [tree_form("d_1")]
+    if len(lst) == 2:
+        for i in range(2):
+            
+            f, g = copy.deepcopy([lst[i], lst[1-i]])
+            
+            logs = [(tab, f"trying integration by parts, f = {printeq_str(simplify(f))} and g = {printeq_str(simplify(g))}")]
+            typeintegrate()
+            out1 = integrate(copy.deepcopy(g), wrt, tab+1, inf)
+            typebyparts()
+            
+            if out1 is None:
+                continue
+            
+            typeintegrate()
+            out2 = integrate(simplify(diff(copy.deepcopy(f), wrt)*out1[0]), wrt, tab+1, inf)
+            
+            typebyparts()
+            if out2 is None:
+                continue
+            
+            return copy.deepcopy([simplify(copy.deepcopy(f) * out1[0] - out2[0]), logs+out1[1]+out2[1]])
+    return None
 def integration_formula_init():
     var = "x"
     formula_list = [(f"(A*{var}+B)^C", f"(A*{var}+B)^(C+1)/(A*(C+1))"),\
                     (f"sin(A*{var}+B)", f"-cos(A*{var}+B)/A"),\
                     (f"cos(A*{var}+B)", f"sin(A*{var}+B)/A"),\
-                    (f"1/(A*{var}+B)", f"log(abs(A*{var}+B))/A")]
+                    (f"1/(A*{var}+B)", f"log(abs(A*{var}+B))/A"),\
+                    (f"e^(A*{var}+B)", f"e^(A*{var}+B)/A"),\
+                    (f"abs(A*{var}+B)", f"(A*{var}+B)*abs(A*{var}+B)/(2*A)")]
     formula_list = [[simplify(parse(y)) for y in x] for x in formula_list]
     expr = [[parse("A"), parse("1")], [parse("B"), parse("0")]]
     return [formula_list, var, expr]
 formula_gen = integration_formula_init()
-typeint = "integrate"
-def typesqint():
-    global typeint
-    typeint=  "sqint"
+
+
 def integrate(equation, wrt="v_0", tab=0, inf=0):
     global formula_list, var, expr
     global typeint
 
     equation = simplify(equation)
+    
     logs = []
     if tab == 0:
         logs += [(tab, f"the given question is to integrate {printeq_str(simplify(equation))} wrt to {str(tree_form(wrt))}")]
@@ -245,9 +295,10 @@ def integrate(equation, wrt="v_0", tab=0, inf=0):
     if lst_const != []:
         equation = product([item for item in lst if contain(item, tree_form(wrt))])
         const = product(lst_const)
-        if simplify(const) != 1:
+        const = simplify(const)
+        if const != 1 and not contain(const, tree_form("s_i")):
             
-            equation = solve(equation)
+            equation = simplify(equation)
             out = integrate(equation, wrt, tab+1, inf)
             
             if out is None:
@@ -261,14 +312,18 @@ def integrate(equation, wrt="v_0", tab=0, inf=0):
     if out is not None:
         return out[0], logs+out[1]+[(tab, f"result is {printeq_str(simplify(out[0]))}")]
     out = None
-    if typeint == "sqint":
+    if typeint in ["integrate", "sqint"]:
         out = sqint(equation, wrt, tab+1, inf)
         if out is not None:
             return out[0], logs+out[1]+[(tab, f"result is {printeq_str(simplify(out[0]))}")]
-    elif typeint == "integrate":
+    if typeint == "integrate":
         if inf==0:
             out = integrate_subs_main(equation, wrt, tab, inf+1)
         if out is not None:
             return out[0], logs+out[1]+[(tab, f"result is {printeq_str(simplify(out[0]))}")]
-    
+    if typeint == "byparts":
+        
+        out = byparts(copy.deepcopy(equation), wrt, tab, inf)
+        if out is not None:
+            return out[0], logs+out[1]+[(tab, f"result is {printeq_str(simplify(out[0]))}")]
     return None

mathai/simplify.py

@@ -244,6 +244,12 @@ def simplify(eq):
                     out = perfect_nth_root_value(n, r)
                     if out is not None:
                         return pp( tree_form("d_"+str(out)), f)
+        if eq.name == "f_mul" and len(eq.children)== 2:
+            for i in range(2):
+                if eq.children[i].name[:2] == "d_" and eq.children[1-i].name == "f_log":
+                    return (eq.children[1-i].children[0]**eq.children[i]).fx("log")
+        if eq.name == "f_pow" and eq.children[0] == tree_form("s_e") and eq.children[1].name == "f_log":
+            return eq.children[1].children[0]
         if eq.name == "f_pow" and eq.children[0] == tree_form("d_1"):
             eq = tree_form("d_1")
         if eq.name == "f_pow" and eq.children[0] == tree_form("d_0"):
@@ -297,7 +303,12 @@ def simplify(eq):
         if eq.name == "f_pow" and eq.children[1].name[:2] == "d_" and abs(int(eq.children[1].name[2:]))%2==0 and eq.children[0].name == "f_abs":
             return eq.children[0].children[0]**eq.children[1]
         return TreeNode(eq.name, [helper5(child) for child in eq.children])
-    
+    def helper8(eq):
+        if eq.name == "f_abs" and eq.children[0].name == "f_abs":
+            return eq.children[0]
+        if eq.name == "f_cos" and eq.children[0].name == "f_abs":
+            return eq.children[0].children[0].fx("cos")
+        return TreeNode(eq.name, [helper8(child) for child in eq.children])
     def fx1(eq):
         for item in [helper, helper3, helper4, helper6,solve,helper2,helper5]:
             eq = dowhile(eq, item)
@@ -311,5 +322,5 @@ def simplify(eq):
             eq = dowhile(eq, item)
         return eq
     eq = dowhile(eq, fx3)
-    
+    eq = dowhile(eq, helper8)
     return solve(eq)

{mathai-0.1.4.dist-info → mathai-0.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: mathai
-Version: 0.1.4
+Version: 0.1.5
 Summary: Mathematics solving Ai tailored to NCERT
 Home-page: https://github.com/infinity390/mathai4
 Author: educated indians are having a low iq and are good for nothing

{mathai-0.1.4.dist-info → mathai-0.1.5.dist-info}/RECORD

@@ -1,22 +1,22 @@
-mathai/__init__.py,sha256=xOcZ1nVRQPWmKoW9qGtTbazsKhDON3-rWiWlRBPLG8A,477
-mathai/apart.py,sha256=FHbSwoqTA_8ZRhXuFzxlfB-OvaOucaTNQAIlyD428tM,3124
+mathai/__init__.py,sha256=CIxRQncAAbHhI6YtrOI2HAiTyJX8reAoqnu7qfonVDc,504
+mathai/apart.py,sha256=P0B21wHUFitc5RGbNJlwSeOdseOMcav4I7teFNcZmFA,3130
 mathai/base.py,sha256=dnOCgCquNeSZY5tnd0aeXwOf1VxFrVMipdF4M92Bd-M,12223
-mathai/console.py,sha256=f_4GAJU3BL--gh3pECqlz86zp6zbd1WxiV-TPHDccx8,2890
+mathai/console.py,sha256=FsEMqDsgFeBSUiPCm9sfReZp56NSyoFVldqmVwvW4lg,3119
 mathai/diff.py,sha256=Ia3ndCOa0c8qqcMEWBIrsXco942dcNfFLpu_T6TeEZA,2788
 mathai/expand.py,sha256=zI2QDjjPSP5dqzgTVhuAMnBlohApL1MRjPZe0v-ccTw,1753
-mathai/factor.py,sha256=bbkoJBzfqDAzqeQoF6z-iu25qfpHwe_ZVtHTrb2ka1w,5061
+mathai/factor.py,sha256=1-Bxl3MX6P1mYM-CX0hRskCrTZWHFdaUL-V1OqX48hY,5316
 mathai/fraction.py,sha256=KGldpi8mndp5cvp6qbN7ZdMZyVmK6popa3AwkWmHXl4,2573
-mathai/integrate.py,sha256=GN0ue3LPFLfINuAQTaF7iNTVPVX1wDKjzsNjGGnHGgE,10049
+mathai/integrate.py,sha256=bt03xpe5dTGJiIIB--lvKZluLy1NAwBQNfTIDfjRxo4,11984
 mathai/inverse.py,sha256=jd3YMqFgeHBUh9JoZa-1Wqh5lXZNB_UcrSD1paZYn-Q,2578
 mathai/linear.py,sha256=eVnDbJYC1TWwg4J7ovBKsaHYlSoDmXk5jQpsqwtVPyI,5481
 mathai/logic.py,sha256=UvHzRmKcO9AD51tRzHmpNSEhgW5gmaf4XPaQKFjGfC4,9653
 mathai/parser.py,sha256=ENtMMm7Q_5QZL6cjbwsXGVJv-H53ueCPGEosJy4LcjY,6809
 mathai/printeq.py,sha256=JTB0_RBcgt1yFviqlHtXGXuSXcpKrxzSxj5WHkOHon4,1318
-mathai/simplify.py,sha256=p1iKCY0OR4EknsJ0dkUg6lZqtb8P9Dn1po0CKN_xric,12960
+mathai/simplify.py,sha256=niSfqqzg44YGVuRRrQ9XG_K7Ak0Wj0WrRxQjkM0KBrw,13721
 mathai/structure.py,sha256=4lKbRNRAkLTZjvnWEeKLXg_tTeBPgc0VIz0evQq4-K8,4069
 mathai/tool.py,sha256=0WLsZR-LsXaPzrSt49ijvJImnqqW1_ArY4s1-V5o_74,1041
 mathai/trig.py,sha256=gvwDjid8_O6XNnWkXXf0oGvkkqtbKg7_XMmwHwK9sjc,6681
-mathai-0.1.4.dist-info/METADATA,sha256=EUGHRz2i1CYWDiQgj7PesPcc7xmlp2YK5aHG6-FC7MQ,741
-mathai-0.1.4.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
-mathai-0.1.4.dist-info/top_level.txt,sha256=ROP4l3OhGYw3ihkQGASr18xM9GsK4z3_6whV5AyXLwE,7
-mathai-0.1.4.dist-info/RECORD,,
+mathai-0.1.5.dist-info/METADATA,sha256=c9zZ2vTB5uPeEZ3XSZnbH4ui69XnimwtJCPlitvbXXM,741
+mathai-0.1.5.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
+mathai-0.1.5.dist-info/top_level.txt,sha256=ROP4l3OhGYw3ihkQGASr18xM9GsK4z3_6whV5AyXLwE,7
+mathai-0.1.5.dist-info/RECORD,,