Skip to content

ImpedanCEI/Wakis

Folders and files

NameName
Last commit message
Last commit date

Latest commit

ย 

History

723 Commits
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 

wakis-logo-light-background

Open-source Wake and Impedance Solver

Documentation Status nightly_tests_CPU_py3.14 codecov Ruff LoC

PyPI - Version PyPI - Downloads PyPI - License DOI

wakis is a 3D Time-domain Electromagnetic solver that solves the Integral form of Maxwell's equations using the Finite Integration Technique (FIT) numerical method. It computes the longitudinal and transverse wake potential and beam-coupling impedance from the simulated electric and magnetic fields. It is also a multi-purpose solver, capable of simulating planewaves interaction with nano-structures, optical diffraction, and much more!

About

๐Ÿš€ Some of wakis features:

  • Wake potential and impedance calculations for particle beams with different relativistic $\beta$
  • Material tensors: permittivity $\varepsilon$, permeability $\mu$, conductivity $\sigma$. Possibility of anisotropy.
  • CAD geometry importer (STL & STEP format) for definition of embedded boundaries and material regions, based on pyvista
  • Boundary conditions: PEC, PMC, Periodic, ABC-FOEXTRAP, Perfect Matched Layers (PML)
  • Different time-domain sources: particle beam, planewave, gaussian wavepacket
  • 100% python, fully exposed API (material tensors, fields $E$, $H$, $J$). Matrix operators based on numpy and scipy.sparse routines ensure multithreaded calculations using Intel's mkl-service.
  • 1d, 2d, 3d built-in plotting on-the-fly
  • Optimized memory consumption & GPU acceleration using cupy/cupyx on double and single precision: added in#v0.6.1
  • CUDA-aware MPI parallelization with mpi4py and ipyparallel: added in#v0.6.0
  • Snappy Smart mesh added in #v0.6.2

๐Ÿงฉ Other complementary tools in the ecosystem:

  • Wakefield extrapolation via broadband resonator fitting with PIML iddefix evolutionary algorithms
  • Non-equidistant Filon Fourier integration with neffint
  • Beam-induced heating estimation due to impedance with bihc

๐Ÿ“ฃ Tag and version changes are decribed in each Wakis Github Release

How to use

๐Ÿ“– Documentation, powered by sphinx, is available at wakis.readthedocs.io

Check ๐Ÿ“ examples/ and notebooks/ for different physical applications:

  • Planewave interacting with a PEC or dielectric sphere
  • Gaussian wavepacket travelling through vacuum / dielectric
  • Custom perturbation interacting with PEC geometry
  • Wakefield simulation of accelerator cavity on CPU, GPU and with MPI

Check ๐ŸŒ๐Ÿ“ SWAN_tutorial/ for hands-on notebook examples ready to run on CERN's SWAN service's GPUs (A100, Tesla T4):

Open this Gallery in SWAN

Check ๐ŸŒ๐Ÿ“ wakis-benchmarks/ for beam-coupling impedance calculations & comparisons to the commercial tool CSTยฎ Wakefield solver:

  • PEC cubic cavity below cutoff (mm) and above cutoff (cm)
  • Conductive cubic cavity below cutoff
  • Lossy pillbox cavity (cylindrical) above cutoff
  • Simulations using beams with different relativistic $\beta$

Check ๐ŸŒ๐Ÿ“ BE-Seminar-demo/ for a complete demonstration of Wakis usage.

Installation

Wakis supports Python 3.9 - 3.14 and can be installed in any conda or venv environment.

๐Ÿ“– For a detailed installation guide (GPU, MPI setup, FAQs), check our documentation.

Install via PyPI

For basic usage, simply run:

pip install wakis

For additional features, including interactive 3D plots in Jupyter notebooks and Wakis' satellite packages, use:

pip install wakis['all']

To use the version-frozen environment specified in requirements.txt, with python 3.9-3.11 do:

conda create --name wakis-env python=3.11
conda activate wakis-env
pip install wakis['notebook']

To install Wakis from the source, clone the repository and install it in editable mode:

git clone https://github.com/ImpedanCEI/wakis.git
cd wakis
pip install -e .

Install via Docker

A pre-built Docker image of Wakis is available for easy setup and reproducibility.

sudo docker pull docker.io/edelafue/wakis:latest
sudo docker run --rm -it docker.io/edelafue/wakis:latest /bin/bash

๐Ÿ’ก Have a bug, feature request, or suggestion? Open a GitHub Issue so the community can track it.

๐Ÿ› ๏ธ Want to contribute? To merge your changes into main, create a Pull Request (PR) following our PR template.

Motivation

๐ŸŽฏ The determination of electromagnetic wakefields and their impact on accelerator performance is a significant issue in current accelerator components. These wakefields, which are generated within the accelerator vacuum chamber as a result of the interaction between the structure and a passing beam, can have significant effects on the machine. These effects can be characterized through the beam coupling impedance in the frequency domain and wake potential in the time domain. Accurate evaluation of these properties is essential for predicting dissipated power and maintaining beam stability. wakis is an open-source tool that can compute wake potential and impedance for both longitudinal and transverse planes for general 3D structures.

Citing Wakis

๐Ÿ”– Each Wakis release is linked to a Zenodo publication under a unique DOI. If you are using Wakis in your scientific research, please help our scientific visibility by citing this work:

[1] E. de la Fuente Garcia et. al., โ€œWakisโ€. Zenodo, 2025. doi: https://doi.org/10.5281/zenodo.15527405


Tests badges

nightly_tests_CPU_py3.10 nightly_tests_CPU_py3.11 nightly_tests_CPU_py3.12 nightly_tests_CPU_py3.13 nightly_tests_CPU_py3.14

About

3D electromagnetic time-domain solver, specialized in wake potential and beam-coupling impedance computation for particle accelerators

Topics

Resources

License

Code of conduct

Contributing

Security policy

Stars

Watchers

Forks

Contributors

Languages