def get_wavelength_solutions_deprecated(thar_echellogram_products, echel,
new_orders):
"""
new_orders : output orders
"""
from ecfit import get_ordered_line_data, fit_2dspec, check_fit
from storage_descriptions import THAR_ALIGNED_JSON_DESC
affine_tr = thar_echellogram_products[THAR_ALIGNED_JSON_DESC]["affine_tr"]
d_x_wvl = {}
for order, z in echel.zdata.items():
xy_T = affine_tr.transform(np.array([z.x, z.y]).T)
x_T = xy_T[:,0]
d_x_wvl[order]=(x_T, z.wvl)
xl, yl, zl = get_ordered_line_data(d_x_wvl)
# xl : pixel
# yl : order
# zl : wvl * order
x_domain = [0, 2047]
orders_band = sorted(echel.zdata.keys())
#orders = igrins_orders[band]
#y_domain = [orders_band[0]-2, orders_band[-1]+2]
y_domain = [new_orders[0], new_orders[-1]]
p, m = fit_2dspec(xl, yl, zl, x_degree=4, y_degree=3,
x_domain=x_domain, y_domain=y_domain)
if 0:
import matplotlib.pyplot as plt
fig = plt.figure(figsize=(12, 7))
check_fit(fig, xl, yl, zl, p, orders_band, d_x_wvl)
fig.tight_layout()
xx = np.arange(2048)
wvl_sol = []
for o in new_orders:
oo = np.empty_like(xx)
oo.fill(o)
wvl = p(xx, oo) / o
wvl_sol.append(list(wvl))
if 0:
json.dump(wvl_sol,
open("wvl_sol_phase0_%s_%s.json" % \
(band, igrins_log.date), "w"))
from storage_descriptions import THAR_WVLSOL_JSON_DESC
r = PipelineProducts("wavelength solution from ThAr")
r.add(THAR_WVLSOL_JSON_DESC,
PipelineDict(orders=new_orders,
wvl_sol=wvl_sol))
return r
def get_wavelength_solutions(affine_tr_matrix, zdata, new_orders):
"""
new_orders : output orders
convert (x, y) of zdata (where x, y are pixel positions and z
is wavelength) with affine transform, then derive a new wavelength
solution.
"""
from ecfit import get_ordered_line_data, fit_2dspec, check_fit
affine_tr = matplotlib.transforms.Affine2D()
affine_tr.set_matrix(affine_tr_matrix)
d_x_wvl = {}
for order, z in zdata.items():
xy_T = affine_tr.transform(np.array([z.x, z.y]).T)
x_T = xy_T[:, 0]
d_x_wvl[order] = (x_T, z.wvl)
_xl, _ol, _wl = get_ordered_line_data(d_x_wvl)
# _xl : pixel
# _ol : order
# _wl : wvl * order
x_domain = [0, 2047]
#orders = igrins_orders[band]
#y_domain = [orders_band[0]-2, orders_band[-1]+2]
y_domain = [new_orders[0], new_orders[-1]]
p, m = fit_2dspec(_xl,
_ol,
_wl,
x_degree=4,
y_degree=3,
x_domain=x_domain,
y_domain=y_domain)
if 0:
import matplotlib.pyplot as plt
fig = plt.figure(figsize=(12, 7))
orders_band = sorted(zdata.keys())
check_fit(fig, xl, yl, zl, p, orders_band, d_x_wvl)
fig.tight_layout()
xx = np.arange(2048)
wvl_sol = []
for o in new_orders:
oo = np.empty_like(xx)
oo.fill(o)
wvl = p(xx, oo) / o
wvl_sol.append(list(wvl))
if 0:
json.dump(wvl_sol,
open("wvl_sol_phase0_%s_%s.json" % \
(band, igrins_log.date), "w"))
return wvl_sol
def get_wavelength_solutions(thar_echellogram_products, echel):
from ecfit import get_ordered_line_data, fit_2dspec, check_fit
from storage_descriptions import THAR_ALIGNED_JSON_DESC
affine_tr = thar_echellogram_products[THAR_ALIGNED_JSON_DESC]["affine_tr"]
d_x_wvl = {}
for order, z in echel.zdata.items():
xy_T = affine_tr.transform(np.array([z.x, z.y]).T)
x_T = xy_T[:,0]
d_x_wvl[order]=(x_T, z.wvl)
xl, yl, zl = get_ordered_line_data(d_x_wvl)
# xl : pixel
# yl : order
# zl : wvl * order
x_domain = [0, 2047]
orders_band = sorted(echel.zdata.keys())
#orders = igrins_orders[band]
y_domain = [orders_band[0]-2, orders_band[-1]+2]
p, m = fit_2dspec(xl, yl, zl, x_degree=4, y_degree=3,
x_domain=x_domain, y_domain=y_domain)
if 0:
import matplotlib.pyplot as plt
fig = plt.figure(figsize=(12, 7))
check_fit(fig, xl, yl, zl, p, orders_band, d_x_wvl)
fig.tight_layout()
xx = np.arange(2048)
wvl_sol = []
for o in orders_band:
oo = np.empty_like(xx)
oo.fill(o)
wvl = p(xx, oo) / o
wvl_sol.append(list(wvl))
if 0:
json.dump(wvl_sol,
open("wvl_sol_phase0_%s_%s.json" % \
(band, igrins_log.date), "w"))
from storage_descriptions import THAR_WVLSOL_JSON_DESC
r = PipelineProducts("wavelength solution from ThAr")
r.add(THAR_WVLSOL_JSON_DESC,
PipelineDict(orders=orders_band,
wvl_sol=wvl_sol))
return r
def get_wavelength_solutions2(affine_tr, zdata, new_orders):
"""
new_orders : output orders
"""
from ecfit import get_ordered_line_data, fit_2dspec, check_fit
d_x_wvl = {}
for order, z in zdata.items():
xy_T = affine_tr.transform(np.array([z.x, z.y]).T)
x_T = xy_T[:, 0]
d_x_wvl[order] = (x_T, z.wvl)
xl, yl, zl = get_ordered_line_data(d_x_wvl)
# xl : pixel
# yl : order
# zl : wvl * order
x_domain = [0, 2047]
orders_band = sorted(zdata.keys())
#orders = igrins_orders[band]
#y_domain = [orders_band[0]-2, orders_band[-1]+2]
y_domain = [new_orders[0], new_orders[-1]]
p, m = fit_2dspec(xl,
yl,
zl,
x_degree=4,
y_degree=3,
x_domain=x_domain,
y_domain=y_domain)
if 0:
import matplotlib.pyplot as plt
fig = plt.figure(figsize=(12, 7))
check_fit(fig, xl, yl, zl, p, orders_band, d_x_wvl)
fig.tight_layout()
xx = np.arange(2048)
wvl_sol = []
for o in new_orders:
oo = np.empty_like(xx)
oo.fill(o)
wvl = p(xx, oo) / o
wvl_sol.append(list(wvl))
if 0:
json.dump(wvl_sol,
open("wvl_sol_phase0_%s_%s.json" % \
(band, igrins_log.date), "w"))
return wvl_sol
def get_wavelength_solutions2(affine_tr, zdata,
new_orders):
"""
new_orders : output orders
"""
from ecfit import get_ordered_line_data, fit_2dspec, check_fit
d_x_wvl = {}
for order, z in zdata.items():
xy_T = affine_tr.transform(np.array([z.x, z.y]).T)
x_T = xy_T[:,0]
d_x_wvl[order]=(x_T, z.wvl)
xl, yl, zl = get_ordered_line_data(d_x_wvl)
# xl : pixel
# yl : order
# zl : wvl * order
x_domain = [0, 2047]
orders_band = sorted(zdata.keys())
#orders = igrins_orders[band]
#y_domain = [orders_band[0]-2, orders_band[-1]+2]
y_domain = [new_orders[0], new_orders[-1]]
p, m = fit_2dspec(xl, yl, zl, x_degree=4, y_degree=3,
x_domain=x_domain, y_domain=y_domain)
if 0:
import matplotlib.pyplot as plt
fig = plt.figure(figsize=(12, 7))
check_fit(fig, xl, yl, zl, p, orders_band, d_x_wvl)
fig.tight_layout()
xx = np.arange(2048)
wvl_sol = []
for o in new_orders:
oo = np.empty_like(xx)
oo.fill(o)
wvl = p(xx, oo) / o
wvl_sol.append(list(wvl))
if 0:
json.dump(wvl_sol,
open("wvl_sol_phase0_%s_%s.json" % \
(band, igrins_log.date), "w"))
return wvl_sol