ex01 single lens

import%20marimo%0A%0A__generated_with%20%3D%20%220.23.8%22%0Aapp%20%3D%20marimo.App(width%3D%22medium%22)%0A%0A%0A%40app.cell%0Adef%20intro()%3A%0A%20%20%20%20import%20marimo%20as%20mo%0A%20%20%20%20mo.md(%0A%20%20%20%20%20%20%20%20%22%22%22%0A%20%20%20%20%20%20%20%20%23%2001%20Ex%20Single%20Thin%20Lens%20(f%20%3D%2050%20mm)%20%0A%0A%20%20%20%20%20%20%20%20*Replicates%20DCC-Lab%2FRayTracing%20example%201%20using%20Optiverse.*%0A%0A%20%20%20%20%20%20%20%20A%20collimated%20beam%20enters%20a%20thin%20lens%20of%20focal%20length%20**f**.%0A%20%20%20%20%20%20%20%20Rays%20converge%20at%20the%20focal%20point%20on%20the%20other%20side.%0A%0A%20%20%20%20%20%20%20%20**Concepts%3A**%20%60LensElement%60%2C%20%60SourceParams%60%2C%20%60trace_rays_polymorphic%60%0A%20%20%20%20%20%20%20%20%22%22%22%0A%20%20%20%20)%0A%20%20%20%20return%20(mo%2C)%0A%0A%0A%40app.cell%0Adef%20controls(mo)%3A%0A%20%20%20%20focal_length%20%3D%20mo.ui.slider(start%3D20%2C%20stop%3D200%2C%20step%3D5%2C%20value%3D50%2C%20label%3D%22Focal%20length%20(mm)%22)%0A%20%20%20%20n_rays%20%3D%20mo.ui.slider(start%3D3%2C%20stop%3D21%2C%20step%3D2%2C%20value%3D11%2C%20label%3D%22Number%20of%20rays%22)%0A%20%20%20%20beam_size%20%3D%20mo.ui.slider(start%3D5%2C%20stop%3D40%2C%20step%3D5%2C%20value%3D20%2C%20label%3D%22Beam%20size%20(mm)%22)%0A%20%20%20%20mo.md(%0A%20%20%20%20%20%20%20%20f%22%22%22%0A%20%20%20%20%20%20%20%20%23%23%20Optical%20Setup%0A%20%20%20%20%20%20%20%20%7Bfocal_length%7D%0A%20%20%20%20%20%20%20%20%7Bn_rays%7D%0A%20%20%20%20%20%20%20%20%7Bbeam_size%7D%0A%20%20%20%20%20%20%20%20%22%22%22%0A%20%20%20%20)%0A%20%20%20%20return%20beam_size%2C%20focal_length%2C%20n_rays%0A%0A%0A%40app.cell%0Adef%20trace_and_plot(beam_size%2C%20focal_length%2C%20n_rays)%3A%0A%20%20%20%20import%20matplotlib.pyplot%20as%20plt%0A%20%20%20%20import%20numpy%20as%20np%0A%20%20%20%20from%20optiverse.raytracing.elements.lens%20import%20LensElement%0A%20%20%20%20from%20optiverse.raytracing.engine%20import%20trace_rays_polymorphic%0A%20%20%20%20from%20optiverse.core.models%20import%20SourceParams%0A%0A%20%20%20%20f%20%3D%20float(focal_length.value)%0A%20%20%20%20lens_half%20%3D%2030.0%0A%0A%20%20%20%20lens%20%3D%20LensElement(%0A%20%20%20%20%20%20%20%20p1%3Dnp.array(%5B0.0%2C%20-lens_half%5D)%2C%0A%20%20%20%20%20%20%20%20p2%3Dnp.array(%5B0.0%2C%20lens_half%5D)%2C%0A%20%20%20%20%20%20%20%20efl_mm%3Df%2C%0A%20%20%20%20)%0A%20%20%20%20elements%20%3D%20%5Blens%5D%0A%0A%20%20%20%20source%20%3D%20SourceParams(%0A%20%20%20%20%20%20%20%20x_mm%3D-150.0%2C%20y_mm%3D0.0%2C%20angle_deg%3D0.0%2C%0A%20%20%20%20%20%20%20%20size_mm%3Dfloat(beam_size.value)%2C%0A%20%20%20%20%20%20%20%20n_rays%3Dint(n_rays.value)%2C%0A%20%20%20%20%20%20%20%20ray_length_mm%3D400.0%2C%20spread_deg%3D0.0%2C%0A%20%20%20%20%20%20%20%20color_hex%3D%22%23DC143C%22%2C%20wavelength_nm%3D633.0%2C%0A%20%20%20%20)%0A%20%20%20%20paths%20%3D%20trace_rays_polymorphic(elements%2C%20%5Bsource%5D%2C%20parallel%3DFalse)%0A%0A%20%20%20%20fig%2C%20ax%20%3D%20plt.subplots(figsize%3D(10%2C%204))%0A%20%20%20%20ax.plot(%5B0%2C%200%5D%2C%20%5B-lens_half%2C%20lens_half%5D%2C%20%22b-%22%2C%20linewidth%3D3%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20label%3Df%22Lens%20(f%3D%7Bf%3A.0f%7D%20mm)%22)%0A%20%20%20%20for%20path%20in%20paths%3A%0A%20%20%20%20%20%20%20%20xs%20%3D%20%5Bfloat(p%5B0%5D)%20for%20p%20in%20path.points%5D%0A%20%20%20%20%20%20%20%20ys%20%3D%20%5Bfloat(p%5B1%5D)%20for%20p%20in%20path.points%5D%0A%20%20%20%20%20%20%20%20r%2C%20g%2C%20b%2C%20a%20%3D%20path.rgba%0A%20%20%20%20%20%20%20%20ax.plot(xs%2C%20ys%2C%20color%3D(r%2F255%2C%20g%2F255%2C%20b%2F255%2C%20a%2F255)%2C%20linewidth%3D0.8)%0A%20%20%20%20ax.plot(-150%2C%200%2C%20%22r*%22%2C%20markersize%3D12%2C%20label%3D%22Source%22)%0A%20%20%20%20ax.axvline(x%3Df%2C%20color%3D%22blue%22%2C%20linestyle%3D%22%3A%22%2C%20alpha%3D0.4%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20label%3Df%22Focal%20plane%20(x%3D%7Bf%3A.0f%7D)%22)%0A%20%20%20%20ax.set_xlabel(%22x%20(mm)%22)%0A%20%20%20%20ax.set_ylabel(%22y%20(mm)%22)%0A%20%20%20%20ax.set_title(f%22Single%20Thin%20Lens%20(f%20%3D%20%7Bf%3A.0f%7D%20mm)%22)%0A%20%20%20%20ax.legend(loc%3D%22upper%20right%22%2C%20fontsize%3D8)%0A%20%20%20%20ax.set_aspect(%22equal%22)%0A%20%20%20%20ax.grid(True%2C%20alpha%3D0.3)%0A%20%20%20%20plt.tight_layout()%0A%20%20%20%20fig%0A%20%20%20%20return%20(paths%2C)%0A%0A%0A%40app.cell%0Adef%20summary(mo%2C%20paths)%3A%0A%20%20%20%20n_paths%20%3D%20len(paths)%0A%20%20%20%20total_pts%20%3D%20sum(len(p.points)%20for%20p%20in%20paths)%0A%20%20%20%20mo.md(%0A%20%20%20%20%20%20%20%20f%22%22%22%0A%20%20%20%20%20%20%20%20%23%23%20Trace%20Summary%0A%20%20%20%20%20%20%20%20%7C%20Metric%20%7C%20Value%20%7C%0A%20%20%20%20%20%20%20%20%7C--------%7C-------%7C%0A%20%20%20%20%20%20%20%20%7C%20Ray%20paths%20%7C%20%7Bn_paths%7D%20%7C%0A%20%20%20%20%20%20%20%20%7C%20Total%20vertices%20%7C%20%7Btotal_pts%7D%20%7C%0A%20%20%20%20%20%20%20%20%7C%20Wavelength%20%7C%20633%20nm%20(HeNe)%20%7C%0A%20%20%20%20%20%20%20%20%22%22%22%0A%20%20%20%20)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_()%3A%0A%20%20%20%20%23%20SELF-CHECK%20%E2%80%94%20thin-lens%20image%20distance%20(object%20at%20150%20mm)%0A%20%20%20%20_f_mm%20%3D%2050.0%20%20%23%20SELF-CHECK-PARAM%0A%20%20%20%20_i%20%3D%201.0%20%2F%20(1.0%2F_f_mm%20-%201.0%2F150.0)%0A%20%20%20%20assert%2070.0%20%3C%20_i%20%3C%2080.0%2C%20f%22lens%20image-distance%20self-check%20failed%3A%20%7B_i%3A.1f%7D%20mm%22%0A%20%20%20%20return%0A%0A%0Aif%20__name__%20%3D%3D%20%22__main__%22%3A%0A%20%20%20%20app.run()%0A