Solverz 0.0.1rc1__py3-none-any.whl

Solverz/__init__.py

@@ -0,0 +1,15 @@
+from Solverz.equation.eqn import Eqn, Ode, HyperbolicPde
+from Solverz.equation.equations import AE, FDAE, DAE
+from Solverz.equation.param import Param, IdxParam, TimeSeriesParam
+from Solverz.sym_algebra.symbols import idx, Para, iVar, iAliasVar
+from Solverz.sym_algebra.functions import Sign, Abs, transpose, exp, Diag, Mat_Mul, sin, cos, Min, AntiWindUp, Saturation
+from Solverz.num_api.custom_function import minmod_flag, minmod
+from Solverz.variable.variables import Vars, TimeVars, as_Vars
+from Solverz.solvers import *
+from Solverz.code_printer.make_pyfunc import made_numerical
+from Solverz.code_printer.py_printer import render_modules
+from Solverz.code_printer.make_module import module_printer
+from Solverz.utilities.io import save, load, save_result
+from Solverz.utilities.profile import count_time
+from Solverz.variable.ssymbol import Var, AliasVar
+from Solverz.model.basic import Model

Solverz/code_printer/make_module.py

@@ -0,0 +1,34 @@
+from typing import List
+
+from Solverz.code_printer.py_printer import render_modules as py_module_renderer
+from Solverz.equation.equations import AE, FDAE, DAE
+from Solverz.variable.variables import Vars
+
+
+class module_printer:
+    def __init__(self,
+                 mdl: AE | FDAE | DAE,
+                 variables: Vars | List[Vars],
+                 name: str,
+                 lang='python',
+                 directory=None,
+                 jit=False):
+        self.name = name
+        self.lang = lang
+        self.mdl = mdl
+        if isinstance(variables, Vars):
+            self.variables = [variables]
+        else:
+            self.variables = variables
+        self.directory = directory
+        self.jit = jit
+
+    def render(self):
+        if self.lang == 'python':
+            py_module_renderer(self.mdl,
+                               *self.variables,
+                               name=self.name,
+                               directory=self.directory,
+                               numba=self.jit)
+        else:
+            raise NotImplemented(f"{self.lang} module renderer not implemented!")

Solverz/code_printer/make_pyfunc.py

@@ -0,0 +1,5 @@
+import numpy as np
+
+from Solverz.code_printer.py_printer import made_numerical
+
+