fftvis is a fast Python package designed for simulating interferometric visibilities using the Non-Uniform Fast Fourier Transform (NUFFT). It provides a convenient and efficient way to generate simulated visibilities.
- Utilizes the Flatiron Institute NUFFT (finufft) algorithm for fast visibility simulations that agree with similar methods (
matvis) to high precision. - Designed to be a near drop-in replacement to
matviswith a ~10x improvement in runtime - Extensible design that allows for easy addition of new backends
- Support for polarized beam patterns and polarized sky models
- No support for per-antenna beams
- GPU backend exists only as a stub implementation (coming soon!)
- Diffuse sky models must be pixelized
You can install fftvis via pip:
pip install fftvisfrom fftvis import simulate_vis
# Simulate visibilities with the CPU backend (default)
vis = simulate_vis(
ants=antenna_positions,
fluxes=source_fluxes,
ra=source_ra,
dec=source_dec,
freqs=frequencies,
times=observation_times,
beam=beam_model,
polarized=True,
backend="cpu" # Use "gpu" for GPU acceleration when implemented
)fftvis is structured with a modular design:
- Core: Contains abstract interfaces and base classes that define the API
- CPU: Contains the CPU-specific implementation
- GPU: Contains the GPU implementation (currently stubbed for future development)
- Wrapper: Provides a high-level API for backward compatibility
This modular design makes the package more maintainable and extensible, allowing for the addition of new backends and optimizations without affecting the user API.
Contributions to fftvis are welcome! If you find any issues, have feature requests, or want to contribute improvements, please open an issue or submit a pull request on the GitHub repository: fftvis on GitHub
This project is licensed under the MIT License - see the LICENSE file for details.
This package relies on the finufft implementation provided by finufft library. Special thanks to the contributors and maintainers of open-source libraries used in this project.