happymath 0.0.1__tar.gz

happymath-0.0.1/Differential_Equation/ODE/__init__.py

@@ -0,0 +1 @@
+from .utils import *

happymath-0.0.1/Differential_Equation/ODE/utils.py

@@ -0,0 +1,50 @@
+from sympy import dsolve
+from IPython.display import display
+from ...latex_trans import tex2sympy
+from ..utils import is_linear_de,de_var_order,de_func
+import re
+
+
+
+def is_ode(ode_sympy):
+    var_list, _ = de_var_order(ode_sympy)
+
+    if not de_func(ode_sympy):
+        return False
+    if len(var_list) == 1:
+        return True
+    elif len(var_list) > 1:
+        return False
+    else:
+        return False
+
+
+def ode_analyzer(ode_object):
+    if type(ode_object) == str:
+        ode_object = tex2sympy(ode_object)
+
+    bool_ode = is_ode(ode_object)
+    bool_linear = is_linear_de(ode_object)
+    _, older_ode = de_var_order(ode_object)
+
+    if not bool_ode:
+        raise ValueError("这似乎并不是一个常微分方程，请仔细检查输入表达式！")
+
+    if bool_linear:
+        linear = "线性"
+    else:
+        linear = "非线性"
+
+    try:
+        res_solve = dsolve(ode_object)
+        bool_solve = True
+    except Exception as e:
+        bool_solve = False
+
+    if bool_solve:
+        print(f"这是一个{older_ode}阶{linear}常微分方程，方程具有解析解，为:")
+        display(res_solve)
+        return res_solve
+    else:
+        print(f"这是一个{older_ode}阶{linear}常微分方程，方程不具有解析解，考虑数值方法求解！")
+        return None

happymath-0.0.1/Differential_Equation/__init__.py

@@ -0,0 +1,5 @@
+from .utils import *
+from .ODE import *
+
+from .utils import de_var_order
+from .utils import is_linear_de

happymath-0.0.1/Differential_Equation/utils.py

@@ -0,0 +1,156 @@
+import re
+import sympy
+from ..latex_trans import tex2sympy
+
+def de_func(de_sympy):
+    # 判断微分方程中未知函数
+    str_de = str(de_sympy)
+    pattern_func = "Derivative\((\S*), (\S*)"
+    cnt_find = re.findall("Derivative", str_de)
+
+    tmp_str = ""
+    tmp_de = str_de
+
+    for i in range(0, len(cnt_find)):
+        find_num = tmp_de.find("Derivative")
+        match_str = tmp_de[find_num + 10:]
+
+        cnt = 0
+        bool_str = []
+        check_sign = False
+        for check in list(match_str):
+            cnt += 1
+            if check == '(':
+                bool_tmp = True
+                bool_str.append(bool_tmp)
+            elif check == ')':
+                bool_str.pop()
+
+            if len(bool_str) == 0:
+                break
+
+        content = "Derivative" + tmp_de[find_num + 10:find_num + 10 + cnt]
+        match = re.search(pattern_func, content)
+
+        if tmp_str != "" and tmp_str != match.group(1):
+            return None
+
+        else:
+            tmp_str = match.group(1)
+            tmp_de = tmp_de.replace(content, "")
+
+    return tmp_str
+
+def de_var_order(ode_sympy):
+    tex_ode = str(ode_sympy)
+
+    # 返回Derivative字段的起始和结束位置
+    pattern = "Derivative"
+    matches = re.finditer(pattern, tex_ode)
+    pos_pattern_start = []
+    pos_pattern_end = []
+    for match in matches:
+        start = match.start()
+        end = match.end()
+        pos_pattern_start.append(start)
+        pos_pattern_end.append(end)
+
+        # 存储完整的Derivative函数内容
+    bool_str = []
+    end_pos = []
+    for i in pos_pattern_end:
+        for j in range(i, len(tex_ode)):
+            if tex_ode[j] == '(':
+                bool_tmp = True
+                bool_str.append(bool_tmp)
+            elif tex_ode[j] == ')':
+                bool_str.pop()
+
+            if len(bool_str) == 0:
+                end_pos.append(j + 1)
+                break
+
+    der_list = []
+    for idx, k in enumerate(pos_pattern_start):
+        der_list.append(tex_ode[k:end_pos[idx]])
+
+    # 根据函数内容分情况输出ode的变量与阶数
+    older_max = 0
+    var_list = []
+    for der in der_list:
+        var_older_list = []
+        split_res = der[11:-1].split(", ")
+        var_older_list = [i.replace("(", "").replace(")", "") for i in split_res]
+
+        if len(var_older_list) - 1 == 1:  ## 一阶单变量微分方程
+            older_max = max(1, older_max)
+            var_list.append(var_older_list[1])
+        else:
+            for var_oder in range(1, len(var_older_list)):
+                if var_older_list[var_oder].isdigit():
+                    older_max = max(int(var_older_list[var_oder]), older_max)
+                else:
+                    var_list.append(var_older_list[var_oder])
+
+    return list(set(var_list)), older_max
+
+
+def is_linear_de(de_sympy):
+    str_de = str(de_sympy)
+    fun_sym = de_func(de_sympy)
+
+    if not fun_sym:
+        raise ValueError("未知函数不存在，请检查方程定义是否正确！")
+
+    pattern = fun_sym + "*Derivative"
+
+    if str_de.find(pattern) == -1:
+        return True
+    else:
+        return False
+
+
+def apply_ics(res_od, ics: set, non_params: list = []):
+    # 根据解析解和初始条件求得特殊解
+
+    ics_pattern = r"(\S*)((\d+))"
+    var = list(res_od.lhs.free_symbols)[0]
+    trans_non = non_params
+
+    var_ics = []
+    trans_dic = {}
+    keys_list = list(ics.keys())
+    values_list = list(ics.values())
+
+    for i in range(0, len(keys_list)):
+        match = re.search(ics_pattern, str(keys_list[i]))
+        if match:
+            ics_value = match.group(2)
+
+        if type(keys_list[i]) == str:
+            tex_key = tex2sympy(keys_list[i])
+        else:
+            tex_key = keys_list[i]
+
+        values = values_list[i]
+        if type(values) == str:
+            value_tex = tex2sympy(values)
+            var_ics.append(value_tex)
+
+        var_ics.append(values)
+        trans_dic[tex_key] = values
+
+    if non_params:
+        trans_non = []
+        for j in non_params:
+            if type(j) == str:
+                tmp_tex = tex2sympy(j)
+            else:
+                tmp_tex = j
+            trans_non.append(tmp_tex)
+
+    apply_symbols = res_od.free_symbols - set(trans_non)
+    eqs = [(res_od.lhs.diff(var, n) - res_od.rhs.diff(var, n)).subs(var, ics_value).subs(trans_dic) for n in
+           range(len(ics))]
+    sol_params = sympy.solve(eqs, apply_symbols)
+    return res_od.subs(sol_params)

happymath-0.0.1/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) [year] [fullname]
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

happymath-0.0.1/PKG-INFO

@@ -0,0 +1,16 @@
+Metadata-Version: 2.1
+Name: happymath
+Version: 0.0.1
+Summary: Test vsrsion for happymath
+Author: Zou
+Author-email: 1514117376@qq.com
+License: MIT
+Classifier: Framework :: Pytest
+Classifier: Programming Language :: Python
+Classifier: Topic :: Software Development :: Testing
+Classifier: Programming Language :: Python :: 3.8
+License-File: LICENSE
+Requires-Dist: sympy
+Requires-Dist: IPython
+Requires-Dist: latex2sympy2
+Requires-Dist: importlib

happymath-0.0.1/README.md

@@ -0,0 +1 @@
+This just a test module for happymath, coming soon...

happymath-0.0.1/happymath.egg-info/PKG-INFO

@@ -0,0 +1,16 @@
+Metadata-Version: 2.1
+Name: happymath
+Version: 0.0.1
+Summary: Test vsrsion for happymath
+Author: Zou
+Author-email: 1514117376@qq.com
+License: MIT
+Classifier: Framework :: Pytest
+Classifier: Programming Language :: Python
+Classifier: Topic :: Software Development :: Testing
+Classifier: Programming Language :: Python :: 3.8
+License-File: LICENSE
+Requires-Dist: sympy
+Requires-Dist: IPython
+Requires-Dist: latex2sympy2
+Requires-Dist: importlib

happymath-0.0.1/happymath.egg-info/SOURCES.txt

@@ -0,0 +1,12 @@
+LICENSE
+README.md
+setup.py
+Differential_Equation/__init__.py
+Differential_Equation/utils.py
+Differential_Equation/ODE/__init__.py
+Differential_Equation/ODE/utils.py
+happymath.egg-info/PKG-INFO
+happymath.egg-info/SOURCES.txt
+happymath.egg-info/dependency_links.txt
+happymath.egg-info/requires.txt
+happymath.egg-info/top_level.txt

happymath-0.0.1/happymath.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

happymath-0.0.1/happymath.egg-info/requires.txt

@@ -0,0 +1,4 @@
+sympy
+IPython
+latex2sympy2
+importlib

happymath-0.0.1/happymath.egg-info/top_level.txt

@@ -0,0 +1 @@
+Differential_Equation

happymath-0.0.1/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

happymath-0.0.1/setup.py

@@ -0,0 +1,20 @@
+from setuptools import setup,find_packages
+
+setup(
+    name='happymath',
+    version='0.0.1',
+    author="Zou",
+    author_email='1514117376@qq.com',
+    description='Test vsrsion for happymath',
+    classifiers=[# 分类索引 ，pip 对所属包的分类
+        'Framework :: Pytest',
+        'Programming Language :: Python',
+        'Topic :: Software Development :: Testing',
+        'Programming Language :: Python :: 3.8',
+    ],
+    license='MIT',
+    # 需要安装的依赖
+    install_requires=[
+        'sympy', 'IPython', 'latex2sympy2', 'importlib',
+    ]
+)