Python binding ectrans4py

.gitignore

@@ -8,4 +8,4 @@ build/*
 install/*
 env.sh
 *.DS_Store
-
+*.py[co~]

CMakeLists.txt

@@ -181,6 +181,17 @@ ecbuild_add_option( FEATURE GPU_STATIC
                     DEFAULT ${GPU_STATIC_DEFAULT}
                     DESCRIPTION "Compile GPU library as static library")
 
+ecbuild_add_option( FEATURE ETRANS
+                    DEFAULT ON
+                    DESCRIPTION "Include Limited-Area-Model Transforms" )
+
+
+ecbuild_add_option( FEATURE ECTRANS4PY
+                    DEFAULT OFF
+		    CONDITION HAVE_ETRANS
+                    DESCRIPTION "Compile ectrans4py interface routines for python binding w/ ctypesForFortran" )
+
+
 ectrans_find_lapack()
 
 ecbuild_add_option( FEATURE TESTS

MANIFEST.in

@@ -0,0 +1,2 @@
+recursive-include cmake
+exclude MANIFEST.in

README.md

@@ -106,6 +106,28 @@ The benchmark drivers are found in the bin directory.
 A brief description of available command-line arguments can be obtained with e.g.
 ectrans-benchmark-cpu-sp --help
 
+Building `ectrans4py`
+---------------------
+
+The python wheel can be built from the root of the project, assuming above-mentioned variables are defined (`fiat_ROOT` etc...):
+```
+python -m build --wheel
+```
+and then:
+```
+python -m auditwheel
+```
+The built python wheel is then to be found in directory `wheelhouse/` and can be locally installed by pip:
+```
+pip install wheelhouse/ectrans4py-<x.y.z>(...).whl
+```
+The `_skbuild` and `dist` directories can be deleted.
+
+Tests can be run from `tests/test_ectrans4py/`:
+```
+python -m pytest
+```
+
 Reporting Bugs
 ==============
 

pyproject.toml

@@ -0,0 +1,23 @@
+[project]
+name = "ectrans4py"
+dynamic = ["version"]
+description = "ECTRANS interface for Python"
+readme = "README.md"
+requires-python = ">=3.10"
+dependencies=["numpy", "ctypesForFortran<2.0.0"]
+classifiers = [
+    'Development Status :: 3 - Alpha',
+    'Intended Audience :: Science/Research',
+    'Programming Language :: Python',
+    'Programming Language :: Python :: 3.10',
+    'Programming Language :: Python :: 3.11',
+    'Programming Language :: Python :: 3.12',
+    'Operating System :: Unix',
+]
+
+[build-system]
+requires = ["setuptools", "wheel", "scikit-build"]
+build-backend = "setuptools.build_meta"
+
+[tool.setuptools.dynamic]
+version = {attr = "ectrans4py.__version__"}

setup.py

@@ -0,0 +1,25 @@
+import os
+import ast
+from skbuild import setup
+
+_version_file = os.path.join(os.path.dirname(os.path.abspath(__file__)), "VERSION")
+with open(_version_file, "r") as f:
+    __version__ = f.read().strip()
+
+setup(
+    name="ectrans4py",
+    version=__version__,
+    packages=['ectrans4py'],
+    cmake_minimum_required_version="3.13",
+    cmake_args=[
+        '-DENABLE_ETRANS=ON',
+        '-DENABLE_ECTRANS4PY=ON',
+        '-DENABLE_SINGLE_PRECISION=OFF',
+        '-DENABLE_OMP=ON',
+        '-DFFTW_USE_STATIC_LIBS=ON',
+    ],
+    package_dir={"": "src"},
+    cmake_install_dir="src/ectrans4py",
+    setup_requires=["scikit-build", "setuptools"],
+    install_requires=["numpy", "ctypesforfortran==1.1.3"],
+)

src/CMakeLists.txt

@@ -11,3 +11,9 @@ add_subdirectory( programs )
 if( HAVE_TRANSI )
   add_subdirectory(transi)
 endif()
+if( HAVE_ETRANS )
+  add_subdirectory(etrans)
+endif()
+if(HAVE_ECTRANS4PY)
+  add_subdirectory(ectrans4py)
+endif()

src/ectrans4py/CMakeLists.txt

@@ -0,0 +1,20 @@
+if(HAVE_ETRANS)
+    # (using CMAKE_CURRENT_SOURCE_DIR is necessary because sources are in a different directory than the target library (trans_${prec}))
+    ecbuild_list_add_pattern(
+        LIST ectrans4py_src
+        GLOB ${CMAKE_CURRENT_SOURCE_DIR}/*.F90
+        QUIET
+    )
+
+    set(HAVE_dp ${HAVE_DOUBLE_PRECISION})
+    set(prec dp)
+
+    if(HAVE_${prec})
+        # Add sources
+        target_sources(trans_${prec} PRIVATE ${ectrans4py_src})
+    endif()
+
+else()
+    ecbuild_critical("To activate the ectrans Python interface, you must enable the ETRANS option.")
+endif()
+