def _make_report(self, image, qs):
d = Bunch.Bunch()
try:
x, y = qs.objx, qs.objy
equinox = float(image.get_keyword('EQUINOX', 2000.0))
try:
ra_deg, dec_deg = image.pixtoradec(x, y, coords='data')
ra_txt, dec_txt = wcs.deg2fmt(ra_deg, dec_deg, 'str')
except Exception as e:
self.logger.warning("Couldn't calculate sky coordinates: %s" %
(str(e)))
ra_deg, dec_deg = 0.0, 0.0
ra_txt = dec_txt = 'BAD WCS'
# Calculate star size from pixel pitch
try:
header = image.get_header()
((xrot, yrot),
(cdelt1, cdelt2)) = wcs.get_xy_rotation_and_scale(header)
starsize = self.iqcalc.starsize(qs.fwhm_x, cdelt1, qs.fwhm_y,
cdelt2)
except Exception as e:
self.logger.warning("Couldn't calculate star size: %s" %
(str(e)))
starsize = 0.0
rpt_x = x + self.pixel_coords_offset
rpt_y = y + self.pixel_coords_offset
# make a report in the form of a dictionary
d.setvals(
x=rpt_x,
y=rpt_y,
ra_deg=ra_deg,
dec_deg=dec_deg,
ra_txt=ra_txt,
dec_txt=dec_txt,
equinox=equinox,
fwhm=qs.fwhm,
fwhm_x=qs.fwhm_x,
fwhm_y=qs.fwhm_y,
ellipse=qs.elipse,
background=qs.background,
skylevel=qs.skylevel,
brightness=qs.brightness,
starsize=starsize,
time_local=time.strftime("%Y-%m-%d %H:%M:%S",
time.localtime()),
time_ut=time.strftime("%Y-%m-%d %H:%M:%S", time.gmtime()),
)
except Exception as e:
self.logger.error("Error making report: %s" % (str(e)))
return d
def make_report(self, image, qs):
d = Bunch.Bunch()
try:
x, y = qs.objx, qs.objy
equinox = float(image.get_keyword('EQUINOX', 2000.0))
try:
ra_deg, dec_deg = image.pixtoradec(x, y, coords='data')
ra_txt, dec_txt = wcs.deg2fmt(ra_deg, dec_deg, 'str')
except Exception as e:
self.logger.warning("Couldn't calculate sky coordinates: %s" % (str(e)))
ra_deg, dec_deg = 0.0, 0.0
ra_txt = dec_txt = 'BAD WCS'
# Calculate star size from pixel pitch
try:
header = image.get_header()
((xrot, yrot),
(cdelt1, cdelt2)) = wcs.get_xy_rotation_and_scale(header)
starsize = self.iqcalc.starsize(qs.fwhm_x, cdelt1,
qs.fwhm_y, cdelt2)
except Exception as e:
self.logger.warning("Couldn't calculate star size: %s" % (str(e)))
starsize = 0.0
rpt_x = x + self.pixel_coords_offset
rpt_y = y + self.pixel_coords_offset
# make a report in the form of a dictionary
d.setvals(x = rpt_x, y = rpt_y,
ra_deg = ra_deg, dec_deg = dec_deg,
ra_txt = ra_txt, dec_txt = dec_txt,
equinox = equinox,
fwhm = qs.fwhm,
fwhm_x = qs.fwhm_x, fwhm_y = qs.fwhm_y,
ellipse = qs.elipse, background = qs.background,
skylevel = qs.skylevel, brightness = qs.brightness,
starsize = starsize,
time_local = time.strftime("%Y-%m-%d %H:%M:%S",
time.localtime()),
time_ut = time.strftime("%Y-%m-%d %H:%M:%S",
time.gmtime()),
)
except Exception as e:
self.logger.error("Error making report: %s" % (str(e)))
return d
def update_pick(self, serialnum, objlist, qs, x1, y1, wd, ht, fig, msg):
if serialnum != self.get_serial():
return
try:
image = self.fitsimage.get_image()
point = fig.objects[1]
text = fig.objects[2]
text.text = "Pick"
if msg != None:
raise Exception(msg)
# Mark new peaks, if desired
if self.show_candidates:
for obj in objlist:
tag = self.fitsimage.add(self.dc.Point(
x1 + obj.objx,
y1 + obj.objy,
5,
linewidth=1,
color=self.candidate_color),
tagpfx='peak',
redraw=False)
# Add back in offsets into image to get correct values with respect
# to the entire image
qs.x += x1
qs.y += y1
qs.objx += x1
qs.objy += y1
# Calculate X/Y of center of star
obj_x = qs.objx
obj_y = qs.objy
self.logger.info("object center is x,y=%f,%f" % (obj_x, obj_y))
fwhm = qs.fwhm
fwhm_x, fwhm_y = qs.fwhm_x, qs.fwhm_y
point.x, point.y = obj_x, obj_y
text.color = 'cyan'
self.wdetail.fwhm_x.set_text('%.3f' % fwhm_x)
self.wdetail.fwhm_y.set_text('%.3f' % fwhm_y)
self.wdetail.fwhm.set_text('%.3f' % fwhm)
self.wdetail.object_x.set_text('%.3f' % (obj_x + 1))
self.wdetail.object_y.set_text('%.3f' % (obj_y + 1))
self.wdetail.sky_level.set_text('%.3f' % qs.skylevel)
self.wdetail.background.set_text('%.3f' % qs.background)
self.wdetail.brightness.set_text('%.3f' % qs.brightness)
self.w.btn_sky_cut.set_enabled(True)
self.w.btn_bright_cut.set_enabled(True)
# Mark center of object on pick image
i1 = point.x - x1
j1 = point.y - y1
self.pickcenter.x = i1
self.pickcenter.y = j1
self.pickcenter.color = 'cyan'
self.pick_qs = qs
self.pickimage.panset_xy(i1, j1, redraw=True)
# Mark object center on image
point.color = 'cyan'
#self.fitsimage.panset_xy(obj_x, obj_y, redraw=False)
equinox = float(image.get_keyword('EQUINOX', 2000.0))
# Calc RA, DEC, EQUINOX of X/Y center pixel
try:
ra_txt, dec_txt = image.pixtoradec(obj_x, obj_y, format='str')
self.last_rpt = self._mkreport(image, qs)
if self.do_record:
self.w.report.append_text(self.last_rpt)
except Exception as e:
ra_txt = 'WCS ERROR'
dec_txt = 'WCS ERROR'
self.wdetail.ra.set_text(ra_txt)
self.wdetail.dec.set_text(dec_txt)
self.wdetail.equinox.set_text(str(equinox))
# Calculate star size from pixel pitch
try:
#cdelt1, cdelt2 = image.get_keywords_list('CDELT1', 'CDELT2')
header = image.get_header()
((xrot, yrot),
(cdelt1, cdelt2)) = wcs.get_xy_rotation_and_scale(header)
starsize = self.iqcalc.starsize(fwhm_x, cdelt1, fwhm_y, cdelt2)
self.wdetail.star_size.set_text('%.3f' % starsize)
except Exception as e:
self.wdetail.star_size.set_text('ERROR')
self.fv.show_error("Couldn't calculate star size: %s" %
(str(e)),
raisetab=False)
self.update_status("Done")
self.plot_panx = float(i1) / wd
self.plot_pany = float(j1) / ht
if self.have_mpl:
self.plot_contours()
self.plot_fwhm(qs)
except Exception as e:
errmsg = "Error calculating quality metrics: %s" % (str(e))
self.logger.error(errmsg)
self.fv.show_error(errmsg, raisetab=False)
#self.update_status("Error")
for key in ('sky_level', 'background', 'brightness', 'star_size',
'fwhm_x', 'fwhm_y'):
self.wdetail[key].set_text('')
self.wdetail.fwhm.set_text('Failed')
self.w.btn_sky_cut.set_enabled(False)
self.w.btn_bright_cut.set_enabled(False)
self.pick_qs = None
text.color = 'red'
self.plot_panx = self.plot_pany = 0.5
#self.plot_contours()
# TODO: could calc background based on numpy calc
self.w.btn_intr_eval.set_enabled(False)
self.pickimage.redraw(whence=3)
self.canvas.redraw(whence=3)
self.fv.showStatus("Click left mouse button to reposition pick")
return True
def mosaic_inline(self,
imagelist,
bg_ref=None,
trim_px=None,
merge=False,
allow_expand=True,
expand_pad_deg=0.01,
max_expand_pct=None,
update_minmax=True,
suppress_callback=False):
"""Drops new images into the current image (if there is room),
relocating them according the WCS between the two images.
"""
# Get our own (mosaic) rotation and scale
header = self.get_header()
((xrot_ref, yrot_ref),
(cdelt1_ref, cdelt2_ref)) = wcs.get_xy_rotation_and_scale(header)
scale_x, scale_y = math.fabs(cdelt1_ref), math.fabs(cdelt2_ref)
# drop each image in the right place in the new data array
mydata = self._get_data()
count = 1
res = []
for image in imagelist:
name = image.get('name', 'image%d' % (count))
count += 1
data_np = image._get_data()
if 0 in data_np.shape:
self.logger.info("Skipping image with zero length axis")
continue
# Calculate sky position at the center of the piece
ctr_x, ctr_y = trcalc.get_center(data_np)
ra, dec = image.pixtoradec(ctr_x, ctr_y)
# User specified a trim? If so, trim edge pixels from each
# side of the array
ht, wd = data_np.shape[:2]
if trim_px:
xlo, xhi = trim_px, wd - trim_px
ylo, yhi = trim_px, ht - trim_px
data_np = data_np[ylo:yhi, xlo:xhi, ...]
ht, wd = data_np.shape[:2]
# If caller asked us to match background of pieces then
# get the median of this piece
if bg_ref is not None:
bg = iqcalc.get_median(data_np)
bg_inc = bg_ref - bg
data_np = data_np + bg_inc
# Determine max/min to update our values
if update_minmax:
maxval = numpy.nanmax(data_np)
minval = numpy.nanmin(data_np)
self.maxval = max(self.maxval, maxval)
self.minval = min(self.minval, minval)
# Get rotation and scale of piece
header = image.get_header()
((xrot, yrot), (cdelt1,
cdelt2)) = wcs.get_xy_rotation_and_scale(header)
self.logger.debug("image(%s) xrot=%f yrot=%f cdelt1=%f "
"cdelt2=%f" % (name, xrot, yrot, cdelt1, cdelt2))
# scale if necessary
# TODO: combine with rotation?
if (not numpy.isclose(math.fabs(cdelt1), scale_x)
or not numpy.isclose(math.fabs(cdelt2), scale_y)):
nscale_x = math.fabs(cdelt1) / scale_x
nscale_y = math.fabs(cdelt2) / scale_y
self.logger.debug("scaling piece by x(%f), y(%f)" %
(nscale_x, nscale_y))
data_np, (ascale_x, ascale_y) = trcalc.get_scaled_cutout_basic(
data_np,
0,
0,
wd - 1,
ht - 1,
nscale_x,
nscale_y,
logger=self.logger)
# Rotate piece into our orientation, according to wcs
rot_dx, rot_dy = xrot - xrot_ref, yrot - yrot_ref
flip_x = False
flip_y = False
# Optomization for 180 rotations
if (numpy.isclose(math.fabs(rot_dx), 180.0)
or numpy.isclose(math.fabs(rot_dy), 180.0)):
rotdata = trcalc.transform(data_np, flip_x=True, flip_y=True)
rot_dx = 0.0
rot_dy = 0.0
else:
rotdata = data_np
# Finish with any necessary rotation of piece
if not numpy.isclose(rot_dy, 0.0):
rot_deg = rot_dy
self.logger.debug("rotating %s by %f deg" % (name, rot_deg))
rotdata = trcalc.rotate(
rotdata,
rot_deg,
#rotctr_x=ctr_x, rotctr_y=ctr_y
logger=self.logger)
# Flip X due to negative CDELT1
if numpy.sign(cdelt1) != numpy.sign(cdelt1_ref):
flip_x = True
# Flip Y due to negative CDELT2
if numpy.sign(cdelt2) != numpy.sign(cdelt2_ref):
flip_y = True
if flip_x or flip_y:
rotdata = trcalc.transform(rotdata,
flip_x=flip_x,
flip_y=flip_y)
# Get size and data of new image
ht, wd = rotdata.shape[:2]
ctr_x, ctr_y = trcalc.get_center(rotdata)
# Find location of image piece (center) in our array
x0, y0 = self.radectopix(ra, dec)
# Merge piece as closely as possible into our array
# Unfortunately we lose a little precision rounding to the
# nearest pixel--can't be helped with this approach
x0, y0 = int(round(x0)), int(round(y0))
self.logger.debug("Fitting image '%s' into mosaic at %d,%d" %
(name, x0, y0))
# This is for useful debugging info only
my_ctr_x, my_ctr_y = trcalc.get_center(mydata)
off_x, off_y = x0 - my_ctr_x, y0 - my_ctr_y
self.logger.debug("centering offsets: %d,%d" % (off_x, off_y))
# Sanity check piece placement
xlo, xhi = x0 - ctr_x, x0 + wd - ctr_x
ylo, yhi = y0 - ctr_y, y0 + ht - ctr_y
assert (xhi - xlo == wd), \
Exception("Width differential %d != %d" % (xhi - xlo, wd))
assert (yhi - ylo == ht), \
Exception("Height differential %d != %d" % (yhi - ylo, ht))
mywd, myht = self.get_size()
if xlo < 0 or xhi > mywd or ylo < 0 or yhi > myht:
if not allow_expand:
raise Exception("New piece doesn't fit on image and "
"allow_expand=False")
# <-- Resize our data array to allow the new image
# determine amount to pad expansion by
expand_x = max(int(expand_pad_deg / scale_x), 0)
expand_y = max(int(expand_pad_deg / scale_y), 0)
nx1_off, nx2_off = 0, 0
if xlo < 0:
nx1_off = abs(xlo) + expand_x
if xhi > mywd:
nx2_off = (xhi - mywd) + expand_x
xlo, xhi = xlo + nx1_off, xhi + nx1_off
ny1_off, ny2_off = 0, 0
if ylo < 0:
ny1_off = abs(ylo) + expand_y
if yhi > myht:
ny2_off = (yhi - myht) + expand_y
ylo, yhi = ylo + ny1_off, yhi + ny1_off
new_wd = mywd + nx1_off + nx2_off
new_ht = myht + ny1_off + ny2_off
# sanity check on new mosaic size
old_area = mywd * myht
new_area = new_wd * new_ht
expand_pct = new_area / old_area
if ((max_expand_pct is not None)
and (expand_pct > max_expand_pct)):
raise Exception("New area exceeds current one by %.2f %%;"
"increase max_expand_pct (%.2f) to allow" %
(expand_pct * 100, max_expand_pct))
# go for it!
new_data = numpy.zeros((new_ht, new_wd))
# place current data into new data
new_data[ny1_off:ny1_off + myht, nx1_off:nx1_off + mywd] = \
mydata
self._data = new_data
mydata = new_data
if (nx1_off > 0) or (ny1_off > 0):
# Adjust our WCS for relocation of the reference pixel
crpix1, crpix2 = self.get_keywords_list('CRPIX1', 'CRPIX2')
kwds = dict(CRPIX1=crpix1 + nx1_off,
CRPIX2=crpix2 + ny1_off)
self.update_keywords(kwds)
# fit image piece into our array
try:
if merge:
mydata[ylo:yhi, xlo:xhi, ...] += rotdata[0:ht, 0:wd, ...]
else:
idx = (mydata[ylo:yhi, xlo:xhi, ...] == 0.0)
mydata[ylo:yhi, xlo:xhi, ...][idx] = \
rotdata[0:ht, 0:wd, ...][idx]
except Exception as e:
self.logger.error("Error fitting tile: %s" % (str(e)))
raise
res.append((xlo, ylo, xhi, yhi))
# TODO: recalculate min and max values
# Can't use usual techniques because it adds too much time to the
# mosacing
#self._set_minmax()
# Notify watchers that our data has changed
if not suppress_callback:
self.make_callback('modified')
return res
def mosaic_inline(self, imagelist, bg_ref=None, trim_px=None,
merge=False):
"""Drops new images into the current image (if there is room),
relocating them according the WCS between the two images.
"""
# Get our own (mosaic) rotation and scale
header = self.get_header()
((xrot_ref, yrot_ref),
(cdelt1_ref, cdelt2_ref)) = wcs.get_xy_rotation_and_scale(header)
ref_rot = yrot_ref
scale_x, scale_y = math.fabs(cdelt1_ref), math.fabs(cdelt2_ref)
# drop each image in the right place in the new data array
mydata = self._get_data()
count = 1
for image in imagelist:
name = image.get('name', 'image%d' % (count))
count += 1
data_np = image._get_data()
# Calculate sky position at the center of the piece
ctr_x, ctr_y = trcalc.get_center(data_np)
ra, dec = image.pixtoradec(ctr_x, ctr_y)
# User specified a trim? If so, trim edge pixels from each
# side of the array
ht, wd = data_np.shape[:2]
if trim_px:
xlo, xhi = trim_px, wd - trim_px
ylo, yhi = trim_px, ht - trim_px
data_np = data_np[ylo:yhi, xlo:xhi, ...]
ht, wd = data_np.shape[:2]
# If caller asked us to match background of pieces then
# get the median of this piece
if bg_ref != None:
bg = iqcalc.get_median(data_np)
bg_inc = bg_ref - bg
#print "bg=%f inc=%f" % (bg, bg_inc)
data_np = data_np + bg_inc
# Get rotation and scale of piece
header = image.get_header()
((xrot, yrot),
(cdelt1, cdelt2)) = wcs.get_xy_rotation_and_scale(header)
self.logger.debug("image(%s) xrot=%f yrot=%f cdelt1=%f cdelt2=%f" % (
name, xrot, yrot, cdelt1, cdelt2))
# scale if necessary
# TODO: combine with rotation?
if ((math.fabs(cdelt1) != scale_x) or
(math.fabs(cdelt2) != scale_y)):
nscale_x = math.fabs(cdelt1) / scale_x
nscale_y = math.fabs(cdelt2) / scale_y
self.logger.debug("scaling piece by x(%f), y(%f)" % (
nscale_x, nscale_y))
data_np, (ascale_x, ascale_y) = trcalc.get_scaled_cutout_basic(
data_np, 0, 0, wd-1, ht-1, nscale_x, nscale_y)
# Rotate piece into our orientation, according to wcs
rot_dx, rot_dy = xrot - xrot_ref, yrot - yrot_ref
flip_x = False
flip_y = False
## # Flip X due to negative CDELT1
## if numpy.sign(cdelt1) < 0:
## flip_x = True
## # Flip Y due to negative CDELT2
## if numpy.sign(cdelt2) < 0:
## flip_y = True
# Optomization for 180 rotations
if math.fabs(rot_dx) == 180.0:
flip_x = not flip_x
rot_dx = 0.0
if math.fabs(rot_dy) == 180.0:
flip_y = not flip_y
rot_dy = 0.0
self.logger.debug("flip_x=%s flip_y=%s" % (flip_x, flip_y))
rotdata = trcalc.transform(data_np, flip_x=flip_x, flip_y=flip_y)
# Finish with any necessary rotation of piece
if rot_dy != 0.0:
rot_deg = rot_dy
self.logger.debug("rotating %s by %f deg" % (name, rot_deg))
rotdata = trcalc.rotate(rotdata, rot_deg,
#rotctr_x=ctr_x, rotctr_y=ctr_y
)
# Get size and data of new image
ht, wd = rotdata.shape[:2]
ctr_x, ctr_y = trcalc.get_center(rotdata)
# Find location of image piece (center) in our array
x0, y0 = self.radectopix(ra, dec)
# Merge piece as closely as possible into our array
# Unfortunately we lose a little precision rounding to the
# nearest pixel--can't be helped with this approach
x0, y0 = int(round(x0)), int(round(y0))
self.logger.debug("Fitting image '%s' into mosaic at %d,%d" % (
name, x0, y0))
# This is for useful debugging info only
my_ctr_x, my_ctr_y = trcalc.get_center(mydata)
off_x, off_y = x0 - my_ctr_x, y0 - my_ctr_y
self.logger.debug("centering offsets: %d,%d" % (off_x, off_y))
# Sanity check piece placement
xlo, xhi = x0 - ctr_x, x0 + wd - ctr_x
ylo, yhi = y0 - ctr_y, y0 + ht - ctr_y
assert (xhi - xlo == wd), \
Exception("Width differential %d != %d" % (xhi - xlo, wd))
assert (yhi - ylo == ht), \
Exception("Height differential %d != %d" % (yhi - ylo, ht))
# fit image piece into our array, not overwriting any data
# already written
try:
if merge:
mydata[ylo:yhi, xlo:xhi, ...] += rotdata[0:ht, 0:wd, ...]
else:
idx = (mydata[ylo:yhi, xlo:xhi, ...] == 0.0)
#print idx.shape, rotdata.shape
mydata[ylo:yhi, xlo:xhi, ...][idx] = \
rotdata[0:ht, 0:wd, ...][idx]
except Exception as e:
self.logger.error("Error fitting tile: %s" % (str(e)))
raise
# TODO: recalculate min and max values
# Can't use usual techniques because it adds too much time to the
# mosacing
#self._set_minmax()
# Notify watchers that our data has changed
self.make_callback('modified')
return (xlo, ylo, xhi, yhi)
