vectorbeam 0.1.0

A minimal core for vector-beam simulation

VectorBeam

vectorbeam is a minimal, elegant Python library for vector-beam simulation and propagation.

Features

• Vector Fields: Easy creation and manipulation of spatially varying polarization fields (e.g., Full Poincaré beams, vector vortex beams).
• Propagation: Fast scalar and vector propagation algorithms (e.g., Fresnel, Richards-Wolf for high-NA focusing).
• Optical Elements: Simulation of Waveplates, Q-Plates, SLMs, and more.
• Precision Toggle: Support for both single (complex64) and double (complex128) precision.

Installation

You can install vectorbeam directly via pip:

pip install vectorbeam

Or install from source:

git clone https://github.com/yourusername/vectorbeam.git
cd vectorbeam
pip install -e .

Quickstart

import numpy as np
import matplotlib.pyplot as plt
from vectorbeam import fpb_lambert, propagate_field, plotting

# Create a Full Poincaré beam
extent = 10e-3 # 10 mm
N = 512
lambda0 = 632.8e-9 # 632.8 nm
beam = fpb_lambert(N, extent, lambda0, w0=1e-3)

# Propagate the beam
z = 0.5 # 50 cm
propagated_beam = beam.propagate(z)

# Plot the intensity and polarization ellipses
fig, axes = plt.subplots(1, 2, figsize=(10, 4))
plotting.plot_intensity(propagated_beam, ax=axes[0], title="Intensity")
plotting.plot_polarization_ellipses(propagated_beam, ax=axes[1], step=16)
plt.show()

Gallery

License

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