def angledist(th, r1, r2, l): return np.array([ph.angleArea(thi, r1, r2, l)*np.cos(thi) for thi in th]) / (r1**2*np.pi)
# pl.figure(1, figsize=(8.27/1.5, 11.69/1.5)) fy = 5 fx = fy * 1.618 savefig = True pl.figure(1, figsize=(fx, fy)) styles = ['k-', 'r--', 'b:'] for i, r1 in enumerate([0.5, 1, 2]): r2 = 1 l = 10 maxth = 0.34 th = np.linspace(0, maxth, 1001) area = [] for thval in th: area += [ pinholes.angleArea(thval, r1, r2, l) ] area = np.array(area) / (r1*r1*np.pi) * np.cos(th) pl.plot(th, area, styles[i], linewidth=3, label=r"$r_1$ = %g" %(r1)) pl.xlim(0, maxth) pl.ylim(0, 1.05) pl.legend(loc='best') pl.title(r"Effect of entry pinhole size, $r_2$ = %g, $L$ = %g" %(r2, l)) pl.xlabel(r'Escape angle $\theta$', fontsize=14) pl.ylabel('Proportion of atoms escaping', fontsize=14) if savefig: pl.savefig('pinhole_entrysize.pdf') # pl.figure(2, figsize=(11.69/1.5, 8.27/1.5)) pl.figure(2, figsize=(fx, fy)) styles = ['k-', 'r--', 'b:'] for i, l in enumerate([5, 10, 20]):