Deepak Mallubhotla
6231ecc203
All checks were successful
gitea-physics/pynam/pipeline/head This commit looks good
22 lines
491 B
Python
22 lines
491 B
Python
from typing import Callable
|
|
|
|
import numpy as np
|
|
import scipy.integrate
|
|
|
|
import pynam.noise.im_ref
|
|
|
|
|
|
def get_chi_zz_e(eps: Callable[[float], complex]) -> Callable[[float], float]:
|
|
im_ref_p = pynam.noise.im_ref.get_im_ref_p(eps)
|
|
|
|
def chi_zz_e(z: float) -> float:
|
|
def integrand(y: float) -> float:
|
|
return (y**2) * im_ref_p(y / z) * np.exp(-2 * y)
|
|
|
|
integral = scipy.integrate.quad(
|
|
integrand, 0, np.inf, epsabs=1e-10, epsrel=1e-10
|
|
)
|
|
return integral[0] / (z**3)
|
|
|
|
return chi_zz_e
|