#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Copyright © 2019 Michael J. Hayford
""" Interactive layout figure with paraxial editing
.. Created on Thu Mar 14 10:20:33 2019
.. codeauthor: Michael J. Hayford
"""
import numpy as np
from rayoptics.gui.util import bbox_from_poly, scale_bounds
from rayoptics.mpl.interactivefigure import InteractiveFigure
import rayoptics.elem.layout
[docs]class InteractiveLayout(InteractiveFigure):
""" Editable version of optical system layout, aka Live Layout
Attributes:
opt_model: parent optical model
refresh_gui: function to be called on refresh_gui event
offset_factor: how much to draw rays before first surface
do_draw_rays: if True, draw edge rays
do_paraxial_layout: if True, draw editable paraxial axial and chief ray
entity_factory_list: list of drawable entity factories. Allows new
drawables to be added to the layout.
"""
def __init__(self, opt_model, refresh_gui=None,
offset_factor=0.05,
do_draw_rays=True,
do_draw_beams=True,
do_draw_edge_rays=True,
do_draw_ray_fans=False,
num_rays_in_fan=11,
do_paraxial_layout=False,
entity_factory_list=None,
**kwargs):
self.refresh_gui = refresh_gui
is_dark = kwargs['is_dark'] if 'is_dark' in kwargs else False
self.layout = rayoptics.elem.layout.LensLayout(opt_model,
is_dark=is_dark)
if do_draw_rays:
self.do_draw_beams = do_draw_beams
self.do_draw_edge_rays = do_draw_rays
else:
self.do_draw_beams = False
self.do_draw_edge_rays = False
self.do_draw_edge_rays = do_draw_edge_rays
self.do_draw_ray_fans = do_draw_ray_fans
self.num_rays_in_fan = num_rays_in_fan
self.do_paraxial_layout = do_paraxial_layout
self.offset_factor = offset_factor
if entity_factory_list is None:
self.entity_factory_list = []
else:
self.entity_factory_list = entity_factory_list
super().__init__(**kwargs)
[docs] def sync_light_or_dark(self, is_dark, **kwargs):
self.layout.sync_light_or_dark(is_dark)
super().sync_light_or_dark(is_dark, **kwargs)
[docs] def update_data(self, **kwargs):
self.artists = []
concat_bbox = []
layout = self.layout
self.ele_shapes = layout.create_element_entities(self)
self.ele_bbox = self.update_patches(self.ele_shapes)
concat_bbox.append(self.ele_bbox)
# if self.do_draw_beams or self.do_draw_edge_rays or self.do_draw_ray_fans:
self.sl_so = layout.system_length(self.ele_bbox,
offset_factor=self.offset_factor)
system_length, start_offset = self.sl_so
if self.do_draw_beams or self.do_draw_edge_rays:
self.ray_shapes = layout.create_ray_entities(self, start_offset)
self.ray_bbox = self.update_patches(self.ray_shapes)
if self.do_draw_ray_fans:
self.rayfan_shapes = layout.create_ray_fan_entities(
self, start_offset,
num_rays=self.num_rays_in_fan
)
self.rayfan_bbox = self.update_patches(self.rayfan_shapes)
if self.do_paraxial_layout:
self.parax_shapes = layout.create_paraxial_ray_entities(self)
self.parax_bbox = self.update_patches(self.parax_shapes)
for ef in self.entity_factory_list:
# each entity factory is a tuple of callable, args, and kwargs
# the callable signature includes the current figure.
ef_fct, ef_args, ef_kwargs = ef
ef_shapes = ef_fct(self, *ef_args, **ef_kwargs)
ef_bbox = self.update_patches(ef_shapes)
sys_bbox = np.concatenate(concat_bbox)
self.sys_bbox = scale_bounds(bbox_from_poly(sys_bbox),
self.offset_factor)
return self
[docs] def action_complete(self):
super().action_complete()
self.do_action = self.do_shape_action
[docs] def fit_axis_limits(self):
return self.sys_bbox