fftvis 1.0.0

An FFT-based visibility simulator

fftvis: A Non-Uniform Fast Fourier Transform-based Visibility Simulator

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.

Features

• 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 matvis with a ~10x improvement in runtime
• Extensible design that allows for easy addition of new backends
• Support for polarized beam patterns and polarized sky models

Current Limitations

• No support for per-antenna beams
• GPU backend exists only as a stub implementation (coming soon!)
• Diffuse sky models must be pixelized

Installation

You can install fftvis via pip:

pip install fftvis

Basic Usage

from 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
)

Architecture

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.

Contributing

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

License

This project is licensed under the MIT License - see the LICENSE file for details.

Acknowledgments

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.