Find dispersion relation in PDE, systems of PDE, and discrete analogs
Project description
SymDR
SymDR is a library made to automate the process of finding dispersion relations.
It was created in The Great Mathematical Workshop 2024 by the command of young ambitious scientists.
Features
Finding dispersion relation in:
- Equations
- System of equations
- Discrete equations
- Systems of discrete equations
Quickstart
For the best experience, we highly recommend using Jupyter Notebook or Google Collab.
SymDR supports Python >=3.7 and only depends on Sympy. Install the Python package using pip.
pip install symdr
Example
- If you are new to symbolic mathematics in Python, read SymPy introduction first.
- For the end-to-end example of equation analysis, see notebook.
!!!!!!! IMPORTANT !!!!!!!! Variables
x
,t
,w
,k
must not be redefined. They are used by the algorithm. When working with discrete cases, the variablesh
,tau
,a
andn
are also added to this list.
SymDR has own objects for grid functions. To turn on pretty-printing for these functions, you need to add:
init_printing(latex_printer=discrete_latex_printer)
Creation of a grid function is the same as standart function:
u = DiscreteGrid('u')
Consider a discrete analog of Korteweg–De Vries equation:
$$u_{t}=6uu_{x}-u_{xxx}$$
For the third order derivative, we are going to use next scheme: $$ f^{(3)}(x)=\frac{f(x+2h)-2f(x+h)+2f(x-h)-f(x-2h)}{2h^3}$$
And we can write our equation
equation = u.diff(t) + (u.at_x(a+2) - 2 * u.at_x(a+1) + 2 * u.at_x(a-1) - u.at_x(a-2)) / (2 * h ** 3)
The class allows us to denote a derivative at (a, n) with "diff" method exactly the same way as SymPy functions do. Shift at a point is done by any of three methods:
- at_x - shift in space (i.e.
u.at_x(a+2)
means $u_{a+2}^n$)- at_t - shift in time (i.e.
u.at_t(n-2)
means $u_a^{n-2}$)- at - shift in both axes at once (i.e.
u.at(a+2, n-1)
means $u_{a+2}^{n-1}$)
Technical note: instead of moving the function into a subtree of
Derivative
object, as it is with SymPy functions, differentiation ofDiscreteGrid
object simply returns an object of the same class, but with different arguments.
Finally, let's find dispersion relation:
d_equation_dr(equation)
Also you can rewrite it with Euler formula:
d_equation_dr(equation, trig_rewrite=True)
And here's the full code:
from sympy import *
from symdr import *
u = DiscreteGrid('u')
equation = u.diff(t) + (u.at_x(a+2) - 2 * u.at_x(a+1) + 2 * u.at_x(a-1) - u.at_x(a-2)) / (2 * h ** 3)
d_equation_dr(equation)
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.