struphy 2.3.0__tar.gz → 2.3.2__tar.gz

{struphy-2.3.0/src/struphy.egg-info → struphy-2.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: struphy
-Version: 2.3.0
+Version: 2.3.2
 Summary: Multi-model plasma physics package
 Author: Max Planck Institute for Plasma Physics
 Author-email: stefan.possanner@ipp.mpg.de, eric.sonnendruecker@ipp.mpg.de
@@ -27,33 +27,35 @@ Project-URL: changelog, https://gitlab.mpcdf.mpg.de/struphy/struphy/-/blob/devel
 Project-URL: Bug Tracker, https://gitlab.mpcdf.mpg.de/struphy/struphy/-/issues
 Keywords: plasma physics, fusion, numerical modeling, partial differential equations, energetic particles
 Classifier: Programming Language :: Python :: 3
-Requires-Python: <3.12,>=3.7
+Requires-Python: <3.12,>=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: PyYAML==6.0.1
-Requires-Dist: argcomplete==3.2.3
-Requires-Dist: docutils==0.20.1
-Requires-Dist: gvec-to-python==1.2.1
-Requires-Dist: h5py==3.10.0
-Requires-Dist: ipyparallel==8.7.0
-Requires-Dist: lxml_html_clean==0.1.0
+Requires-Dist: argcomplete==3.4.0
+Requires-Dist: desc-opt==0.12.0
+Requires-Dist: docutils==0.21.2
+Requires-Dist: gvec-to-python==1.2.2
+Requires-Dist: h5py==3.11.0
+Requires-Dist: ipyparallel==8.8.0
+Requires-Dist: lxml_html_clean==0.1.1
 Requires-Dist: m2r2==0.3.3.post2
-Requires-Dist: matplotlib==3.8.3
+Requires-Dist: matplotlib==3.9.0
 Requires-Dist: mistune==0.8.4
-Requires-Dist: mpi4py==3.1.5
-Requires-Dist: myst-parser==2.0.0
-Requires-Dist: nbsphinx==0.9.3
-Requires-Dist: notebook==7.1.2
+Requires-Dist: mpi4py==3.1.6
+Requires-Dist: myst-parser==3.0.1
+Requires-Dist: nbsphinx==0.9.4
+Requires-Dist: netCDF4==1.6.5
+Requires-Dist: notebook==7.2.1
 Requires-Dist: numpy==1.24.4
-Requires-Dist: pyccel==1.11.2
-Requires-Dist: pydata-sphinx-theme==0.15.2
-Requires-Dist: pytest==8.1.1
+Requires-Dist: pyccel==1.12.1
+Requires-Dist: pydata-sphinx-theme==0.15.3
+Requires-Dist: pytest==8.2.2
 Requires-Dist: pytest-monitor==1.6.6
 Requires-Dist: pytest-mpi==0.6
-Requires-Dist: scipy==1.12.0
-Requires-Dist: Sphinx==7.2.6
+Requires-Dist: scipy==1.13.1
+Requires-Dist: Sphinx==7.3.7
 Requires-Dist: sphinxcontrib-napoleon==0.7
-Requires-Dist: tqdm==4.66.2
+Requires-Dist: tqdm==4.66.4
 Requires-Dist: vtk==9.3.0
 Requires-Dist: wheel==0.43.0
 

{struphy-2.3.0 → struphy-2.3.2}/pyproject.toml

@@ -4,9 +4,9 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "struphy"
-version = "2.3.0"
+version = "2.3.2"
 readme = "README.md"
-requires-python = ">=3.7, <3.12"
+requires-python = ">=3.10, <3.12"
 license = {file = "LICENSE"}
 authors = [
   { name = "Max Planck Institute for Plasma Physics"},
@@ -20,29 +20,31 @@ classifiers = [
     ]
 dependencies = [
     'PyYAML==6.0.1',
-    'argcomplete==3.2.3',
-    'docutils==0.20.1',
-    'gvec-to-python==1.2.1',
-    'h5py==3.10.0',
-    'ipyparallel==8.7.0',
-    'lxml_html_clean==0.1.0',
+    'argcomplete==3.4.0',
+    'desc-opt==0.12.0',
+    'docutils==0.21.2',
+    'gvec-to-python==1.2.2',
+    'h5py==3.11.0',
+    'ipyparallel==8.8.0',
+    'lxml_html_clean==0.1.1',
     'm2r2==0.3.3.post2',
-    'matplotlib==3.8.3',
+    'matplotlib==3.9.0',
     'mistune==0.8.4',
-    'mpi4py==3.1.5',
-    'myst-parser==2.0.0',
-    'nbsphinx==0.9.3',
-    'notebook==7.1.2',
+    'mpi4py==3.1.6',
+    'myst-parser==3.0.1',
+    'nbsphinx==0.9.4',
+    'netCDF4==1.6.5',
+    'notebook==7.2.1',
     'numpy==1.24.4',
-    'pyccel==1.11.2',
-    'pydata-sphinx-theme==0.15.2',
-    'pytest==8.1.1',
+    'pyccel==1.12.1',
+    'pydata-sphinx-theme==0.15.3',
+    'pytest==8.2.2',
     'pytest-monitor==1.6.6',
     'pytest-mpi==0.6',
-    'scipy==1.12.0',
-    'Sphinx==7.2.6',
+    'scipy==1.13.1',
+    'Sphinx==7.3.7',
     'sphinxcontrib-napoleon==0.7',
-    'tqdm==4.66.2',
+    'tqdm==4.66.4',
     'vtk==9.3.0',
     'wheel==0.43.0',
 ]
@@ -80,5 +82,5 @@ struphy = "struphy.console.main:struphy"
                     ]
                     
 struphy = ['compile_struphy.mk',
-           'psydac-0.1.12-py3-none-any.whl',
+           'psydac-0.1.13-py3-none-any.whl',
            ]

{struphy-2.3.0 → struphy-2.3.2}/src/struphy/bsplines/evaluation_kernels_3d.py

@@ -24,9 +24,14 @@ from numpy import empty, zeros
 
 import struphy.bsplines.bsplines_kernels as bsplines_kernels
 
+import struphy.pic.pushing.pusher_args_kernels as pusher_args_kernels
+from struphy.pic.pushing.pusher_args_kernels import DerhamArguments, DomainArguments
+
 #################################
 ### Single process evaluation ###
 #################################
+
+
 def evaluation_kernel_3d(p1: int, p2: int, p3: int, basis1: 'float[:]', basis2: 'float[:]', basis3: 'float[:]', ind1: 'int[:]', ind2: 'int[:]', ind3: 'int[:]', coeff: 'float[:,:,:]') -> float:
     """
     Summing non-zero contributions of a spline function (serial, needs global arrays).
@@ -404,7 +409,19 @@ def evaluate_sparse(t1: 'float[:]', t2: 'float[:]', t3: 'float[:]', p1: int, p2:
 ##############################
 ### Distributed evaluation ###
 ##############################
-def eval_spline_mpi_kernel(p1: 'int', p2: 'int', p3: 'int', basis1: 'float[:]', basis2: 'float[:]', basis3: 'float[:]', span1: 'int', span2: 'int', span3: 'int', _data: 'float[:,:,:]', starts: 'int[:]') -> float:
+
+
+def eval_spline_mpi_kernel(p1: 'int',
+                           p2: 'int',
+                           p3: 'int',
+                           basis1: 'float[:]',
+                           basis2: 'float[:]',
+                           basis3: 'float[:]',
+                           span1: 'int',
+                           span2: 'int',
+                           span3: 'int',
+                           _data: 'float[:,:,:]',
+                           starts: 'int[:]') -> float:
     """
     Summing non-zero contributions of a spline function with distributed memory (domain decomposition).
 
@@ -767,7 +784,7 @@ def eval_spline_mpi_tensor_product_fixed(span1s: 'int[:]', span2s: 'int[:]', spa
             for k in range(nk):
                 span3 = span3s[k]
                 b3[:] = b3s[k, :]
-                
+
                 values[i, j, k] = eval_spline_mpi_kernel(pn[0] - kind[0],
                                                          pn[1] - kind[1],
                                                          pn[2] - kind[2],
@@ -876,4 +893,203 @@ def eval_spline_mpi_sparse_meshgrid(eta1: 'float[:,:,:]', eta2: 'float[:,:,:]',
                     continue  # point not in process domain
 
                 values[i, j, k] = eval_spline_mpi(
-                    eta1[i, 0, 0], eta2[0, j, 0], eta3[0, 0, k], _data, kind, pn, tn1, tn2, tn3, starts)
+                    eta1[i, 0, 0], eta2[0, j, 0], eta3[0, 0, k], _data, kind, pn, tn1, tn2, tn3, starts)
+
+
+###########################################
+### Distributed Derham field evaluation ###
+###########################################
+
+def get_spans(eta1: float,
+              eta2: float,
+              eta3: float,
+              args_derham: 'DerhamArguments'):
+    '''Compute the knot span index, 
+    the N-spline values (in bn) and the D-spline values (in bd) 
+    at (eta1, eta2, eta3).'''
+
+    # find spans
+    span1 = bsplines_kernels.find_span(
+        args_derham.tn1, args_derham.pn[0], eta1)
+    span2 = bsplines_kernels.find_span(
+        args_derham.tn2, args_derham.pn[1], eta2)
+    span3 = bsplines_kernels.find_span(
+        args_derham.tn3, args_derham.pn[2], eta3)
+
+    # get spline values at eta
+    bsplines_kernels.b_d_splines_slim(
+        args_derham.tn1, args_derham.pn[0], eta1, int(span1), args_derham.bn1, args_derham.bd1)
+    bsplines_kernels.b_d_splines_slim(
+        args_derham.tn2, args_derham.pn[1], eta2, int(span2), args_derham.bn2, args_derham.bd2)
+    bsplines_kernels.b_d_splines_slim(
+        args_derham.tn3, args_derham.pn[2], eta3, int(span3), args_derham.bn3, args_derham.bd3)
+
+    return span1, span2, span3
+
+
+def eval_0form_spline_mpi(span1: int,
+                          span2: int,
+                          span3: int,
+                          args_derham: 'DerhamArguments',
+                          form_coeffs: 'float[:,:,:]') -> float:
+    '''Single-point evaluation of Derham 0-form spline defined by form_coeffs,
+    given N-spline values (in bn) and knot span indices span.'''
+
+    out = eval_spline_mpi_kernel(args_derham.pn[0],
+                                 args_derham.pn[1],
+                                 args_derham.pn[2],
+                                 args_derham.bn1,
+                                 args_derham.bn2,
+                                 args_derham.bn3,
+                                 span1, span2, span3,
+                                 form_coeffs,
+                                 args_derham.starts)
+    return out
+
+
+def eval_1form_spline_mpi(span1: int,
+                          span2: int,
+                          span3: int,
+                          args_derham: 'DerhamArguments',
+                          form_coeffs_1: 'float[:,:,:]',
+                          form_coeffs_2: 'float[:,:,:]',
+                          form_coeffs_3: 'float[:,:,:]',
+                          out: 'float[:]'):
+    '''Single-point evaluation of Derham 1-form spline defined by form_coeffs,
+    given N-spline values (in bn), D-spline values (in bd)
+    and knot span indices span.'''
+
+    out[0] = eval_spline_mpi_kernel(args_derham.pn[0] - 1,
+                                    args_derham.pn[1],
+                                    args_derham.pn[2],
+                                    args_derham.bd1,
+                                    args_derham.bn2,
+                                    args_derham.bn3,
+                                    span1, span2, span3,
+                                    form_coeffs_1,
+                                    args_derham.starts)
+
+    out[1] = eval_spline_mpi_kernel(args_derham.pn[0],
+                                    args_derham.pn[1] - 1,
+                                    args_derham.pn[2],
+                                    args_derham.bn1,
+                                    args_derham.bd2,
+                                    args_derham.bn3,
+                                    span1, span2, span3,
+                                    form_coeffs_2,
+                                    args_derham.starts)
+
+    out[2] = eval_spline_mpi_kernel(args_derham.pn[0],
+                                    args_derham.pn[1],
+                                    args_derham.pn[2] - 1,
+                                    args_derham.bn1,
+                                    args_derham.bn2,
+                                    args_derham.bd3,
+                                    span1, span2, span3,
+                                    form_coeffs_3,
+                                    args_derham.starts)
+
+
+def eval_2form_spline_mpi(span1: int,
+                          span2: int,
+                          span3: int,
+                          args_derham: 'DerhamArguments',
+                          form_coeffs_1: 'float[:,:,:]',
+                          form_coeffs_2: 'float[:,:,:]',
+                          form_coeffs_3: 'float[:,:,:]',
+                          out: 'float[:]'):
+    '''Single-point evaluation of Derham 2-form spline defined by form_coeffs,
+    given N-spline values (in bn), D-spline values (in bd)
+    and knot span indices span.'''
+
+    out[0] = eval_spline_mpi_kernel(args_derham.pn[0],
+                                    args_derham.pn[1] - 1,
+                                    args_derham.pn[2] - 1,
+                                    args_derham.bn1,
+                                    args_derham.bd2,
+                                    args_derham.bd3,
+                                    span1, span2, span3,
+                                    form_coeffs_1,
+                                    args_derham.starts)
+
+    out[1] = eval_spline_mpi_kernel(args_derham.pn[0] - 1,
+                                    args_derham.pn[1],
+                                    args_derham.pn[2] - 1,
+                                    args_derham.bd1,
+                                    args_derham.bn2,
+                                    args_derham.bd3,
+                                    span1, span2, span3,
+                                    form_coeffs_2,
+                                    args_derham.starts)
+
+    out[2] = eval_spline_mpi_kernel(args_derham.pn[0] - 1,
+                                    args_derham.pn[1] - 1,
+                                    args_derham.pn[2],
+                                    args_derham.bd1,
+                                    args_derham.bd2,
+                                    args_derham.bn3,
+                                    span1, span2, span3,
+                                    form_coeffs_3,
+                                    args_derham.starts)
+
+
+def eval_3form_spline_mpi(span1: int,
+                          span2: int,
+                          span3: int,
+                          args_derham: 'DerhamArguments',
+                          form_coeffs: 'float[:,:,:]') -> float:
+    '''Single-point evaluation of Derham 0-form spline defined by form_coeffs,
+    given D-spline values (in bd) and knot span indices span.'''
+
+    out = eval_spline_mpi_kernel(args_derham.pn[0] - 1,
+                                 args_derham.pn[1] - 1,
+                                 args_derham.pn[2] - 1,
+                                 args_derham.bd1,
+                                 args_derham.bd2,
+                                 args_derham.bd3,
+                                 span1, span2, span3,
+                                 form_coeffs,
+                                 args_derham.starts)
+    return out
+
+
+def eval_vectorfield_spline_mpi(span1: int,
+                                span2: int,
+                                span3: int,
+                                args_derham: 'DerhamArguments',
+                                form_coeffs_1: 'float[:,:,:]',
+                                form_coeffs_2: 'float[:,:,:]',
+                                form_coeffs_3: 'float[:,:,:]',
+                                out: 'float[:]'):
+    '''Single-point evaluation of vector-field spline (H^1)^3 defined by form_coeffs,
+    given N-spline values (in bn) and knot span indices span.'''
+
+    out[0] = eval_spline_mpi_kernel(args_derham.pn[0],
+                                    args_derham.pn[1],
+                                    args_derham.pn[2],
+                                    args_derham.bn1,
+                                    args_derham.bn2,
+                                    args_derham.bn3,
+                                    span1, span2, span3,
+                                    form_coeffs_1,
+                                    args_derham.starts)
+
+    out[1] = eval_spline_mpi_kernel(args_derham.pn[0],
+                                    args_derham.pn[1],
+                                    args_derham.pn[2],
+                                    args_derham.bn1,
+                                    args_derham.bn2,
+                                    args_derham.bn3,
+                                    span1, span2, span3,
+                                    form_coeffs_2,
+                                    args_derham.starts)
+
+    out[2] = eval_spline_mpi_kernel(args_derham.pn[0],
+                                    args_derham.pn[1],
+                                    args_derham.pn[2],
+                                    args_derham.bn1,
+                                    args_derham.bn2,
+                                    args_derham.bn3,
+                                    span1, span2, span3,
+                                    form_coeffs_3,
+                                    args_derham.starts)

{struphy-2.3.0 → struphy-2.3.2}/src/struphy/conftest.py

@@ -1,5 +1,6 @@
 def pytest_addoption(parser):
     parser.addoption("--fast", action="store_true")
+    parser.addoption("--with-desc", action="store_true")
 
 
 def pytest_generate_tests(metafunc):
@@ -8,3 +9,7 @@ def pytest_generate_tests(metafunc):
     option_value = metafunc.config.option.fast
     if 'fast' in metafunc.fixturenames and option_value is not None:
         metafunc.parametrize("fast", [option_value])
+        
+    option_value = metafunc.config.option.with_desc
+    if 'with_desc' in metafunc.fixturenames and option_value is not None:
+        metafunc.parametrize("with_desc", [option_value])

{struphy-2.3.0 → struphy-2.3.2}/src/struphy/console/compile.py

@@ -28,7 +28,7 @@ def struphy_compile(language, compiler, omp_pic, omp_feec, delete, status, verbo
 
     dependencies : bool
         Whether to print Struphy kernels (to be compiled) and their dependencies on screen.
-        
+
     yes : bool
         Whether to say yes to prompt when changing the language.
     """
@@ -91,6 +91,10 @@ def struphy_compile(language, compiler, omp_pic, omp_feec, delete, status, verbo
                         '--status',
                         ], check=True, cwd=libpath)
 
+        print('\nDeleting state.yml ...')
+        os.remove(os.path.join(libpath, 'state.yml'))
+        print('Done.')
+
     elif status:
 
         # update status
@@ -183,7 +187,7 @@ def struphy_compile(language, compiler, omp_pic, omp_feec, delete, status, verbo
             else:
                 yesno = input(
                     f'Kernels compiled in language {state["last_used_language"]} exist, will be deleted, continue (Y/n)?')
-                
+
             if yesno in ('', 'Y', 'y', 'yes'):
                 subprocess.run(['struphy',
                                 'compile',
@@ -208,7 +212,7 @@ def struphy_compile(language, compiler, omp_pic, omp_feec, delete, status, verbo
             flags += ' --verbose'
 
         # install psydac from wheel if not there
-        current_ver = '0.1.12'
+        current_ver = '0.1.13'
         psydac_file = 'psydac-' + current_ver + '-py3-none-any.whl'
 
         try:
@@ -261,3 +265,9 @@ def struphy_compile(language, compiler, omp_pic, omp_feec, delete, status, verbo
                        'compile',
                         '--status',
                         ], check=True, cwd=libpath)
+
+        # collect available models
+        print('')
+        subprocess.run(['struphy',
+                        '--refresh-models',
+                        ], check=True, cwd=libpath)