Skip to content

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

License

Notifications You must be signed in to change notification settings

ImpedanCEI/wakis

Repository files navigation

wakis-logo-light-background

Open-source Wake and Impedance Solver

Documentation Status Tests badge codecov

PyPI - Version PyPI - License Tokei - LOC Zenodo 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 fast calculations.
  • 1d, 2d, 3d built-in plotting on-the-fly
  • Optimized memory consumption
  • GPU acceleration using cupy/cupyx
  • CUDA-aware MPI parallelization with mpi4py and ipyparallel coming soon!

🧩 Other complementary tools in the ecosystem:

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

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 🌐📁 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$

🔁 To be informed of the latest features/bug fixes pushed to main, check the release.md

For specific needs, please contact the developer 👩‍💻👋

Installation

Wakis supports Python 3.9 - 3.11 and can be installed in any conda environment. 📖 For a detailed installation guide (GPU, MPI setup, FAQs), check our documentation.

Users: Install via PyPI

For basic usage, simply run:

pip install wakis

For additional features, including interactive 3D plots in Jupyter notebooks, use:

pip install wakis['notebook']

To install complementary tools in the Wakis ecosystem:

pip install neffint iddefix bihc

💡 Have a bug, feature request, or suggestion? Open a GitHub Issue so the community can track it.

Developers: Contribute to Wakis

First, Fork the repository and clone it from main:

# SSH:
git clone [email protected]:YourUserName/wakis.git

# or HTTPS:
git clone https://github.com/YourUserName/wakis.git

Create a dedicated conda environment and install dependencies:

cd wakis/
conda create --name wakis-env python=3.11
conda activate wakis-env
pip install -r requirements.txt
pip install neffint iddefix bihc  # Optional tools

🛠️ 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 bihc in your scientific research, please help our scientific visibility by citing this work:

[1] E. de la Fuente Garcia et. al., “Wakis”. Zenodo, Mar. 12, 2025. doi: 10.5281/zenodo.15011421.

About

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

Topics

Resources

License

Code of conduct

Stars

Watchers

Forks

Contributors 4

  •  
  •  
  •  
  •  

Languages