#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""Import files from `OpticalBenchHub` web page
This module implements lens import from the `OpticalBenchHub <https://www.photonstophotos.net/GeneralTopics/Lenses/OpticalBench/OpticalBenchHub.htm>`_
portion of Bill Claff's `PhotonsToPhotos <https://www.photonstophotos.net/>`_
website.
To import a file from the website, navigate to the lens you wish to import
and select the entire web address of the page. Paste this into the url
argument of the :func:`~.read_obench_url` function.
.. Created on Sat Jul 24 21:34:49 2021
.. codeauthor: Michael J. Hayford
"""
import requests
from rayoptics.optical.opticalmodel import OpticalModel
from rayoptics.elem.profiles import (EvenPolynomial, RadialPolynomial)
from rayoptics.oprops import doe
from rayoptics.oprops.doe import DiffractiveElement
from rayoptics.raytr.opticalspec import WvlSpec
from rayoptics.util.misc_math import isanumber, is_kinda_big
from opticalglass import util
_track_contents: dict|None = None
[docs]
def read_obench_url(url, **kwargs) -> tuple["OpticalModel", tuple[dict, dict]]:
''' given a url to a OpticalBench file, return an OpticalModel and info. '''
global _track_contents
url1 = url.replace('OpticalBench.htm#', '')
url2 = url1.partition(',')[0]
r = requests.get(url2, allow_redirects=True)
apparent_encoding = r.apparent_encoding
r.encoding = r.apparent_encoding
inpt = r.text
lines = inpt.splitlines()
inpt = [l.split('\t') for l in lines]
obench_dict = {}
for line in inpt:
if line[0][0] == '[':
# process new section header, initialize input list
key = line[0][1:-1]
obench_dict[key] = []
else:
# add input to the currect section's list of inputs
obench_dict[key].append(line)
opt_model = read_lens(obench_dict, **kwargs)
_track_contents['obench db'] = obench_dict # type: ignore
_track_contents['encoding'] = apparent_encoding # type: ignore
return opt_model, (_track_contents, {}) # type: ignore
[docs]
def read_lens(inpts, opt_model=None) -> OpticalModel:
global _track_contents
def read_float(s):
if s == 'Infinity':
return float('inf')
elif isanumber(s):
return float(s)
else:
if s == 'undefined':
return float('nan')
elif s == 'AS':
return 0.
elif s == '':
return 0.
else:
try:
return float(read_float(var_dists[s][0]))
except:
return 0.
_track_contents = util.Counter()
constants_inpt = inpts['constants']
constants = {c_item[0]: c_item[1:] for c_item in constants_inpt}
var_dists_inpt = inpts['variable distances']
var_dists = {var_dist[0]: var_dist[1:] for var_dist in var_dists_inpt}
thi_obj = 0.
if 'd0' in var_dists:
thi_obj = read_float(var_dists['d0'][0])
if thi_obj == float('inf'):
thi_obj = 1.0e10
conj_type = 'finite'
if is_kinda_big(thi_obj):
conj_type = 'infinite'
_track_contents['conj type'] = conj_type
if opt_model is None:
opt_model = OpticalModel(do_init=True)
opt_model.radius_mode = True
sm = opt_model['sm']
sm.do_apertures = False
sm.gaps[0].thi = thi_obj
osp = opt_model['osp']
osp['pupil'].key = ('image', 'f/#')
osp['pupil'].value = read_float(var_dists['F-Number'][0])
angle_of_view = read_float(var_dists['Angle of View'][0])
osp['fov'].is_wide_angle = True if angle_of_view/2 > 45. else False
osp['fov'].key = ('image', 'real height')
osp['fov'].value = read_float(var_dists['Image Height'][0])/2
osp['fov'].is_relative = True
osp['fov'].set_from_list([0., .707, 1.])
osp['wvls'] = WvlSpec(wlwts=[('F', .5), ('d', 1.), ('C', .5)], ref_wl=1)
if 'lens data' in inpts:
input_lines = inpts['lens data']
_track_contents['# surfs'] = len(input_lines)
for line in input_lines:
inpt = []
inpt.append(read_float(line[1])) # radius
inpt.append(read_float(line[2])) # thi
if line[3] == '':
inpt.append('')
inpt.append('')
else:
inpt.append(read_float(line[3])) # nd
if line[5] != '':
inpt.append(read_float(line[5])) # vd
diam = read_float(line[4])
sm.add_surface(inpt, sd=diam/2)
if line[1] == 'AS':
sm.set_stop()
if 'aspherical data' in inpts:
if 'AsphericalOddCount' in constants:
typ = 'AsphericalOddCount'
elif 'AsphericalA2' in constants:
typ = 'AsphericalA2'
else:
typ = 'Aspherical'
input_lines = inpts['aspherical data']
_track_contents[typ] = len(input_lines)
for line in input_lines:
if typ == 'AsphericalOddCount':
asp_coefs = [read_float(item) for item in line[3:]]
asp_coefs = [0., 0.] + asp_coefs
elif typ == 'AsphericalA2':
asp_coefs = [read_float(item) for item in line[3:]]
else:
asp_coefs = [read_float(item) for item in line[2:]]
asp_coefs[0] = 0.
if typ == 'AsphericalOddCount':
asp = RadialPolynomial(r=read_float(line[1]),
cc=read_float(line[2]),
coefs=asp_coefs)
else:
asp = EvenPolynomial(r=read_float(line[1]),
cc=read_float(line[2]),
coefs=asp_coefs)
idx = int(line[0])
sm.ifcs[idx].profile = asp
if 'diffractive data' in inpts:
input_lines = inpts['diffractive data']
_track_contents['# doe'] = len(input_lines)
for line in input_lines:
coefs = [read_float(item) for item in line[3:]]
dif_elem = DiffractiveElement(coefficients=coefs,
ref_wl=read_float(line[1]),
order=read_float(line[2]),
phase_fct=doe.radial_phase_fct)
idx = int(line[0])
sm.ifcs[idx].phase_element = dif_elem
if 'descriptive data' in inpts:
input_lines = inpts['descriptive data']
descripts = {input_line[0]: input_line[1:]
for input_line in input_lines}
if 'title' in descripts:
opt_model['sys'].title = descripts['title'][0]
opt_model.update_model()
return opt_model