In [1]:
import os
os.environ["SAS_OPENCL"] = "cuda"   # use CUDA GPU backend for sasmodels

import escape as esc
import numpy as np
esc.require("0.9.8")
Loading material database from C:\dev\escape-core\python\src\escape\scattering\..\data\mdb\materials.db

SAXS. Form-factors. Elliptical cylinder (SasView-aligned)¶

A right cylinder with an elliptical cross-section. The 1D powder average integrates over both the tilt angle $\alpha$ (axis vs $\mathbf{q}$) and the in-plane azimuthal angle $\psi$.

Reference: https://www.sasview.org/docs/user/models/elliptical_cylinder.html

Parameters (SasView defaults)¶

Parameter Variable Value
Scale scale 1
Background (cm⁻¹) background 0.001
Minor radius (Å) radius_minor 20
Axis ratio ν = major/minor axis_ratio 1.5
Length (Å) length 400
Contrast Δρ (10⁻⁶ Å⁻²) contrast 3 (= sld 4 − sld_solvent 1)
Theta (deg), 2D only theta 90
Phi (deg), 2D only phi 0

Form-factor (SasView elliptical_cylinder.c)¶

The effective radius depends on the in-plane azimuthal angle $\psi$:

$$r'(\psi) = \frac{r_{\min}}{\sqrt{2}}\sqrt{(1+\nu^2)+(1-\nu^2)\cos\psi}$$

The oriented amplitude:

$$F(q,\alpha,\psi) = V\cdot\frac{2J_1(r'\,q_\perp)}{r'\,q_\perp}\;\mathrm{sinc}\!\left(\tfrac{L}{2}q_\parallel\right)$$

$$I(q) = \frac{\mathrm{scale}}{V}\cdot\frac{2}{\pi}\int_0^{\pi/2}d\psi\int_0^{\pi/2}F^2\,\sin\alpha\,d\alpha + \mathrm{background}$$

In [2]:
# ── Variables ──────────────────────────────────────────────────────────────
q     = esc.var("Q")
alpha = esc.var("alpha")   # tilt angle between cylinder axis and q
psi   = esc.var("psi")     # in-plane azimuthal angle (averages ellipse orientation)

# ── Parameters ─────────────────────────────────────────────────────────────
scale        = esc.par("Scale",        1.0,  scale=1e8, fixed=True)
radius_minor = esc.par("Radius minor", 20.0, units=esc.angstr)
axis_ratio   = esc.par("Axis ratio",   1.5,  userlim=[1.0, 10.0])
length       = esc.par("Length",      400.0, units=esc.angstr)
contrast     = esc.par("Contrast",     3.0,  scale=1e-6, units=f"{esc.angstr}^-2")
background   = esc.par("Background",   0.001, userlim=[0.0, 0.03])

# ── Geometry ───────────────────────────────────────────────────────────────
# Geometric-mean radius: r_eff = sqrt(a*b) = r_minor * sqrt(axis_ratio)
r_eff  = radius_minor * esc.sqrt(axis_ratio)
volume = np.pi * esc.pow(r_eff, 2) * length

# ψ-dependent effective radius for the Bessel argument
r_psi = (radius_minor / esc.sqrt(2.0)) * esc.sqrt(
    (1.0 + esc.pow(axis_ratio, 2)) + (1.0 - esc.pow(axis_ratio, 2)) * esc.cos(psi))

# ── Oriented amplitude ─────────────────────────────────────────────────────
q_axial  = q * esc.cos(alpha)
q_radial = q * esc.sin(alpha)

# F = V * 2*J1(r_psi*q_radial)/(r_psi*q_radial) * sinc(L/2*q_axial)
F_psi = volume * 2.0 * esc.j1_over_x(r_psi * q_radial) * esc.sinc(0.5 * length * q_axial)

# ── Powder average: integrate over alpha then psi ──────────────────────────
# Inner integral over alpha (orientation average)
alpha_integral = esc.integral(esc.pow(F_psi, 2) * esc.sin(alpha),
                               alpha, 0.0, np.pi / 2.0,
                               numpoints=61, maxiter=5, epsabs=1e-5)

# Outer integral over psi (ellipse orientation average), normalised by 1/pi
i1d = (scale * esc.pow(contrast, 2) / volume
       * esc.integral(alpha_integral, psi, 0.0, np.pi,
                      numpoints=61, maxiter=5, epsabs=1e-5)
       * (1.0 / np.pi)
       + background)
In [3]:
i1d.device = "gpu"

qs = np.linspace(0.001, 1.0, 300)
i1d.show(coordinates=qs).config(
    title="Elliptical cylinder — powder average (1D)",
    xlog=True, ylog=True,
    xlabel=f"Q [{esc.angstr}^-1]", ylabel="I(q) [cm^-1]")
Out[3]:

2D oriented scattering (qx, qy)¶

For a fixed orientation $(\theta, \phi)$ the geometric-mean radius $r_{\mathrm{eff}} = \sqrt{ab}$ is used for the Bessel argument (the ψ average is replaced by a fixed effective radius).

$$I_{\mathrm{2D}}(q_x,q_y) = \frac{\mathrm{scale}}{V}\,F^2(q_\parallel, q_\perp) + \mathrm{background}$$

In [4]:
qx = esc.var("qx")
qy = esc.var("qy")

theta = esc.par("Theta", 90.0, userlim=[0.0, 180.0], units="deg")
phi   = esc.par("Phi",    0.0, userlim=[0.0, 360.0], units="deg")

deg = np.pi / 180.0
sin_t = esc.sin(theta * deg)
ux = sin_t * esc.cos(phi * deg)
uy = sin_t * esc.sin(phi * deg)

q_sq      = esc.pow(qx, 2) + esc.pow(qy, 2)
q_par_2d  = qx * ux + qy * uy
q_perp_2d = esc.sqrt(q_sq - esc.pow(q_par_2d, 2))

# Use geometric-mean radius r_eff for the 2D oriented case
F_2d = volume * 2.0 * esc.j1_over_x(r_eff * q_perp_2d) * esc.sinc(0.5 * length * q_par_2d)
i2d  = scale * esc.pow(contrast, 2) / volume * esc.pow(F_2d, 2) + background

i2d.device = "gpu"

xs = np.linspace(-1.0, 1.0, 300); ys = np.linspace(-1.0, 1.0, 300)
xv, yv = np.meshgrid(xs, ys)
coords_2d = np.column_stack([xv.flatten(), yv.flatten()]).flatten()
i2d.show(coordinates=coords_2d).config(
    title="Elliptical cylinder — oriented 2D SAXS (qx, qy)",
    xlabel=f"qx [{esc.angstr}^-1]", ylabel=f"qy [{esc.angstr}^-1]",
    cblog=True, colorscale="jet")
Out[4]:

SasView reference model & comparison¶

ESCAPE parameter SasView parameter Notes
contrast * 1e-6 sld - sld_solvent contrast in Å⁻²
radius_minor r_minor minor semi-axis (Å)
axis_ratio r_ratio major/minor ratio
length length cylinder length (Å)
In [5]:
import time
import matplotlib.pyplot as plt
from sasmodels.core import load_model
from sasmodels.data import empty_data1D
from sasmodels.direct_model import DirectModel

qs = np.linspace(0.001, 1.0, 300).copy()

kernel = load_model("elliptical_cylinder")
f_sas  = DirectModel(empty_data1D(qs), kernel)
sas_pars = dict(scale=1.0, background=0.001,
                sld=4.0, sld_solvent=1.0,
                radius_minor=20.0, axis_ratio=1.5, length=400.0)

f_sas(**sas_pars)
i1d.device = "gpu"; i1d(qs[:5])

def timeit(fn, n=5):
    t0 = time.perf_counter()
    for _ in range(n): result = fn()
    return (time.perf_counter() - t0) / n * 1e3, result

t_sas, Iq_sas = timeit(lambda: f_sas(**sas_pars))

i1d.device = "gpu"
t_gpu, Iq_gpu = timeit(lambda: i1d(qs), n=3)
i1d.device = "cpu"
t_cpu, Iq_cpu = timeit(lambda: i1d(qs))
i1d.device = "gpu"

print(f"SASView GPU : {t_sas:.0f} ms")
print(f"ESCAPE GPU : {t_gpu:.0f} ms")
print(f"ESCAPE CPU : {t_cpu:.0f} ms  ({len(qs)} q-pts)")
rel = np.max(np.abs((Iq_gpu - Iq_sas) / Iq_sas)) * 100
print(f"Max relative diff vs SasView: {rel:.2f}%")


esc.overlay(Iq_sas, Iq_gpu, Iq_cpu, coordinates=qs).config(
    xlabel="Q (1/A)", ylabel="I(q) (1/cm)", 
    xlog=True, ylog=True,
    fig_title=f"Elliptical cylinder I(q) — {len(qs)} pts",
    labels=["SASView", "ESCAPE GPU", "ESCAPE CPU"],
    line_styles=["solid", "dash", "dot"],
    line_widths=[2, 3, 3]
)
C:\Users\User\AppData\Local\Temp\ipykernel_49232\923603957.py:16: UserWarning:

Input array does not own its data (e.g. it is a view or slice); data will be copied

SASView GPU : 10 ms
ESCAPE GPU : 168 ms
ESCAPE CPU : 144 ms  (300 q-pts)
Max relative diff vs SasView: 0.37%
Out[5]:
In [ ]: