Ray Diagrams with Matplotlib

Model TEM

The same code which creates an interactive model via pyqtgraph of the specified microscope components can also be used to create a static ray diagram in matplotlib. The highlighted lines show the changes that need to me made to create a matplotlib diagram rather than in an interactive pyqtgraph model.

from temgymbasic import components as comp
from temgymbasic.model import Model
from temgymbasic.run import show_matplotlib

components = [comp.Lens(name = 'Electrostatic Lens', z = 3, f = -0.2),
              comp.DoubleDeflector(name = 'Gun Beam Deflectors', z_up = 2.8, z_low = 2.7),
              comp.Lens(name = '1st Condenser Lens', z = 2.6, f = -0.2),
              comp.Lens(name = '2nd Condenser Lens', z = 2.5, f = -0.2),
              comp.Aperture(name = 'Condenser Aperture', z = 2.3, aperture_radius_inner=0.05),
              comp.Quadrupole(name = 'Condenser Stig', z = 2.2),
              comp.DoubleDeflector(name = 'Condenser Deflectors', z_up = 2.1, z_low = 2.0),
              comp.Lens(name = 'Condenser Mini Lens', z = 1.8, f = -0.2),
              comp.Aperture(name = 'Objective Aperture', z = 1.7, aperture_radius_inner=0.05),
              comp.Lens(name = 'Objective Lens', z = 1.5, f = -0.2),
              comp.Quadrupole(name = 'Objective Stig', z = 1.4),
              comp.Lens(name = 'Objective Mini Lens', z = 1.3, f = -0.2),
              comp.DoubleDeflector(name = 'Image Shifts', z_up = 1.1, z_low = 1.0),
              comp.Aperture(name = 'Selected Area Aperture', z = 0.9, aperture_radius_inner=0.05),
              comp.Quadrupole(name = 'Intermediate Lens Stigmator', z = 0.8),
              comp.Lens(name = '1st Intermediate Lens', z = 0.7, f = -0.2),
              comp.Lens(name = '2nd Intermediate Lens', z = 0.6, f = -0.2),
              comp.Lens(name = '3rd Intermediate Lens', z = 0.5, f = -0.2),
              comp.DoubleDeflector(name = 'Projector Lens Deflectors', z_up = 0.4, z_low = 0.3),
              comp.Lens(name = 'Projector Lens', z =0.2, f = -0.2)
              ]

axis_view = 'x_axial'
model_ = Model(components, beam_z=3.5, beam_type='x_axial',
                   num_rays=32, beam_semi_angle=0.15)

name = 'model_tem.svg'
fig, ax = show_matplotlib(model_, name = name, label_fontsize = 14)
fig.suptitle('TEM Model', fontsize=32)
fig.savefig(name, dpi = 500)

which produces this model: