def __init__(self, mu, n_depth_pts, radial_grid): self.mu = mu self.I_lam = np.zeros(n_depth_pts) # outward pointing rays should have diffusion as their initial condition if self.mu > 0: self.I_lam[0] = planck_fn(1) # TODO: make this the diffusion condition # our object is not illuminated from some other source so inward pointing rays have zero intensity to start else: self.I_lam[0] = 0 self.tau_grid = np.zeros(n_depth_pts)
def calc_source_fn(source_fn, epsilon, J_n): source_fn[:] = epsilon * planck_fn(1) + (1 - epsilon) * J_n[:] return (source_fn)