def __init__(self, master=None):
"""
initialize Galaxy photo-z controller
:param master: master application
"""
if master is None:
master_flag = True
master = Tk.Tk()
master.withdraw()
else:
master_flag = False
self.model = GalModel()
self.viewer = GalViewer(master)
# Initial Data Frame
self.viewer.button_load.config(command=self.load_catalog)
self.cluster_id = None
self.cmr_line = dict(gr=None, gi=None, ri=None, rz=None, iz=None)
self.cmr_dots = dict(gr=None, gi=None, ri=None, rz=None, iz=None)
self.sketch_dots = dict(all=None, gr=None, gi=None, ri=None, rz=None, iz=None)
# Initial Sketch
self.viewer.button_sketch.config(command=self.create_sketch)
# Initial selection
self.viewer.button_select_sketch.config(command=lambda: self.select_galaxy_sketch(self.viewer.sketch.figure))
self.viewer.button_update_select.config(command=self.update_selection)
self.viewer.button_resample.config(command=self.resample)
self.viewer.z_scale.config(command=self.update_cmr_z)
self.viewer.z_scale.bind_all("<Left>", self.key_cmr_left)
self.viewer.z_scale.bind_all("<Right>", self.key_cmr_right)
# Save part
self.viewer.button_clean.config(command=self.clean_selection)
self.viewer.button_save_cat.config(command=self.save_cat)
self.viewer.button_save_reg.config(command=self.save_reg)
# Initial Cluster information frame
for j in range(len(P_method)):
self.viewer.button_method[j].config(state=Tk.DISABLED,
command=lambda m=P_method[j]: self.update_candidate_sketch(m))
if master_flag:
self.viewer.protocol("WM_DELETE_WINDOW", master.destroy)
self.viewer.mainloop()
class GalController():
"""
Controller of MVC
"""
def __init__(self, master=None):
"""
initialize Galaxy photo-z controller
:param master: master application
"""
if master is None:
master_flag = True
master = Tk.Tk()
master.withdraw()
else:
master_flag = False
self.model = GalModel()
self.viewer = GalViewer(master)
# Initial Data Frame
self.viewer.button_load.config(command=self.load_catalog)
self.cluster_id = None
self.cmr_line = dict(gr=None, gi=None, ri=None, rz=None, iz=None)
self.cmr_dots = dict(gr=None, gi=None, ri=None, rz=None, iz=None)
self.sketch_dots = dict(all=None, gr=None, gi=None, ri=None, rz=None, iz=None)
# Initial Sketch
self.viewer.button_sketch.config(command=self.create_sketch)
# Initial selection
self.viewer.button_select_sketch.config(command=lambda: self.select_galaxy_sketch(self.viewer.sketch.figure))
self.viewer.button_update_select.config(command=self.update_selection)
self.viewer.button_resample.config(command=self.resample)
self.viewer.z_scale.config(command=self.update_cmr_z)
self.viewer.z_scale.bind_all("<Left>", self.key_cmr_left)
self.viewer.z_scale.bind_all("<Right>", self.key_cmr_right)
# Save part
self.viewer.button_clean.config(command=self.clean_selection)
self.viewer.button_save_cat.config(command=self.save_cat)
self.viewer.button_save_reg.config(command=self.save_reg)
# Initial Cluster information frame
for j in range(len(P_method)):
self.viewer.button_method[j].config(state=Tk.DISABLED,
command=lambda m=P_method[j]: self.update_candidate_sketch(m))
if master_flag:
self.viewer.protocol("WM_DELETE_WINDOW", master.destroy)
self.viewer.mainloop()
def clean_selection(self):
"""
Clean current selections
"""
if self.model.gid is not None:
self.model.gid = None
for i in self.viewer.cmr_window.keys():
if self.cmr_dots[i] is not None:
print i
self.cmr_dots[i].pop(0).remove()
self.cmr_dots[i] = None
if self.sketch_dots[i] is not None:
self.sketch_dots[i].pop(0).remove()
self.sketch_dots[i] = None
if self.viewer.cmr_window[i] is not None:
self.viewer.cmr_window[i].update_canvas()
self.model.cgid[i] = None
self.viewer.sketch.update_canvas()
def create_sketch(self):
"""
create sketch to show galaxy distribution
"""
if self.viewer.sketch is not None:
self.destroy_sketch()
self.viewer.sketch = PlotWindow(master=self.viewer, title="Sketch")
self.view_sketch(self.viewer.sketch.axis)
self.viewer.sketch.protocol("WM_DELETE_WINDOW", self.destroy_sketch)
method = self.model.get_method(P_method)
for j in range(len(P_method)):
if method[j] == 1:
self.viewer.button_method[j].config(state=Tk.NORMAL)
else:
self.viewer.button_method[j].config(state=Tk.DISABLED)
self.viewer.button_select_sketch.config(state=Tk.NORMAL)
self.viewer.button_update_select.config(state=Tk.NORMAL)
self.viewer.button_save_cat.config(state=Tk.NORMAL)
def destroy_sketch(self):
"""
disable corresponding button when the sketch window is closed
"""
self.viewer.sketch.destroy()
self.viewer.sketch = None
for j in range(len(P_method)):
self.viewer.button_method[j].config(state=Tk.DISABLED)
self.viewer.button_select_sketch.config(state=Tk.DISABLED)
self.viewer.button_update_select.config(state=Tk.DISABLED)
def initialize_galaxy(self, cid):
"""
load galaxy data file
:param cid: cluster ID to load
:return: True/False
"""
try:
assert isinstance(cid, int)
self.cluster_id = cid
self.model.load_gal(self.cluster_id)
return True
except ValueError:
self.cluster_id = None
Message("Invalid Input of cluster ID: "+str(cid), master=self.viewer)
return False
except IOError:
self.cluster_id = None
Message("Cluster file is not available!", master=self.viewer)
return False
def key_cmr_left(self, event):
"""
Move CMR to lower redshift
"""
self.viewer.redshift.set(self.viewer.redshift.get()-0.01)
self.update_cmr_z(self.viewer.redshift.get())
def key_cmr_right(self, event):
"""
Move CMR to higher redshift
"""
self.viewer.redshift.set(self.viewer.redshift.get()+0.01)
self.update_cmr_z(self.viewer.redshift.get())
def load_catalog(self):
"""
Load cluster galaxy catalog
"""
flag = self.initialize_galaxy(int(self.viewer.entry_id.get()))
if not flag:
self.viewer.button_sketch.config(state=Tk.DISABLED)
self.viewer.button_fits_gri.config(state=Tk.DISABLED)
self.viewer.button_fits_riz.config(state=Tk.DISABLED)
self.viewer.button_fits_grz.config(state=Tk.DISABLED)
self.viewer.button_sketch.config(state=Tk.DISABLED)
for i in CMR_combination:
self.viewer.button_cmr[i].config(state=Tk.DISABLED)
self.viewer.button_save_reg.config(state=Tk.DISABLED)
return
# set cluster information
self.viewer.label_cluster_info.config(text=self.model.get_info())
# enable button when data is ready
self.viewer.button_sketch.config(state=Tk.NORMAL)
self.viewer.button_fits_gri.config(state=Tk.NORMAL, command=lambda: self.view_fits('gri'))
self.viewer.button_fits_riz.config(state=Tk.NORMAL, command=lambda: self.view_fits('riz'))
self.viewer.button_fits_grz.config(state=Tk.NORMAL, command=lambda: self.view_fits('grz'))
self.viewer.button_sketch.config(state=Tk.NORMAL)
for i in CMR_combination:
self.viewer.button_cmr[i].config(state=Tk.NORMAL, command=lambda a=i: self.view_cmr(a[0], a[1]))
self.viewer.button_save_reg.config(state=Tk.NORMAL)
def resample(self):
"""
Resample the sketch based on selection in sketch
"""
if self.model.gid is not None:
self.model.galaxy.data = self.model.galaxy.data[self.model.gid, :]
self.model.gid = [i for i in range(len(self.model.gid))] # recreate the gid
for ic in self.model.cgid.keys():
self.model.cgid[ic] = None
self.viewer.button_select_sketch.config(state=Tk.DISABLED)
def save_cat(self):
"""
save select in sketch galaxies into new cat
"""
self.model.save_cat(OUTPUT_CAT_PATTERN.format(self.cluster_id))
def save_reg(self):
"""
output to region file for DS9 to load
And close the save window
"""
self.viewer.save_window()
def save_reg_confirm():
save_id = map(int, self.viewer.multi_list.curselection())
for i in save_id:
if i == 0: # save full region
self.model.galaxy.save_reg(OUTPUT_REGION_PATTERN.format("{0:d}".format(self.cluster_id)), self.model.gid)
else: # save color magnitude selected region
j = i - 1
self.model.galaxy.save_reg(OUTPUT_REGION_PATTERN.format("{0:d}_{1:d}".format(self.cluster_id, j)),
self.model.cgid[CMR_combination[j]],
color=CMR_COLOR[CMR_combination[j]])
self.viewer.close_save_window()
self.viewer.button_save_reg_confirm.config(command=save_reg_confirm)
def select_galaxy_sketch(self, figure):
"""
active galaxy selection function
this will set the model.gid
:param figure: figure object for selecting galaxy
"""
self.model.draw_in_circle(figure)
self.viewer.button_update_select.config(state=Tk.NORMAL)
self.viewer.button_resample.config(state=Tk.NORMAL)
self.viewer.sketch.update_canvas()
def update_candidate_sketch(self, method):
"""
Add candidate information
:param method: Add candidate radius with given method
"""
self.model.draw_candidate(self.viewer.sketch.axis, method=method)
self.viewer.sketch.update_canvas()
def update_cmr(self):
"""
update the galaxy of the cmr plot based on sketch selection
"""
for icmr in self.viewer.cmr_window.keys():
self.model.galaxy.plot_cmr(icmr[0], icmr[1],
self.viewer.cmr_window[icmr].axis, gid=self.model.gid,
marker='x', color="r", ls='None')
self.viewer.cmr_window[icmr].axis.set_xlim([16,24])
self.viewer.cmr_window[icmr].axis.set_ylim([0,3])
self.viewer.cmr_window[icmr].update_canvas()
def update_cmr_sketch(self, zid, icmr, **args):
"""
Update galaxies selected from CMR_combination plot in sketch plot
:param zid: redshift id
:param icmr: cmr model color combination
:param args: arguments to plotting
"""
if self.viewer.sketch is None:
return
band1 = icmr[0]
band2 = icmr[1]
self.model.cgid[icmr] = self.model.galaxy.in_cmr_band(band1, band2,
self.model.cmr.get_magnitude(zid, band1),
self.model.cmr.get_magnitude(zid, band2))
if self.sketch_dots[icmr] is not None:
self.sketch_dots[icmr].pop(0).remove()
self.sketch_dots[icmr] = self.model.galaxy.plot_sky(self.viewer.sketch.axis, gid=self.model.cgid[icmr],
color=CMR_COLOR[icmr], marker='+', ls='None', alpha=0.8)
self.viewer.sketch.update_canvas()
def update_cmr_z(self, z):
"""
update the cmr plot to highlight the redshift
also update the sketch if available
:param z: redshift to plot (str or float)
"""
if type(z) is str:
z = float(z)
self.viewer.str_redshift.set("Redshift: {0:4.2f}".format(z))
zid = self.model.cmr.get_zid(z)
for icmr in self.viewer.cmr_window.keys():
if (icmr in self.cmr_line) and (self.cmr_line[icmr] is not None):
self.cmr_line[icmr].pop(0).remove()
self.cmr_line[icmr] = self.model.cmr.plot(zid, icmr[0], icmr[1], self.viewer.cmr_window[icmr].axis,
color='g', lw=2)
self.update_cmr_sketch(zid, icmr)
if self.cmr_dots[icmr] is not None:
self.cmr_dots[icmr].pop(0).remove()
self.cmr_dots[icmr] = self.model.galaxy.plot_cmr(icmr[0], icmr[1],
self.viewer.cmr_window[icmr].axis,
gid=self.model.cgid[icmr],
marker='+', color=CMR_COLOR[icmr], ls='None')
self.viewer.cmr_window[icmr].axis.set_xlim([16,24])
self.viewer.cmr_window[icmr].axis.set_ylim([0,3])
self.viewer.cmr_window[icmr].update_canvas()
def update_galaxy_sketch(self):
"""
update the sketch figure for galaxies distribution according to manual circle selection
"""
if self.sketch_dots['all'] is not None:
self.sketch_dots['all'].pop(0).remove()
self.sketch_dots['all'] = self.model.galaxy.plot_sky(self.viewer.sketch.axis, gid=self.model.gid,
color='r', marker='x', ls='None', alpha=0.6)
self.viewer.sketch.update_canvas()
def update_selection(self):
"""
Update sketch selection
"""
if self.model.gid is not None:
self.update_galaxy_sketch()
self.update_cmr()
def view_cmr(self, band1, band2):
"""
create Color-magnitude relation plot
plot both the galaxies and the model
:param band1: red band
:param band2: blue band
"""
ax = self.viewer.create_cmr(band1 + band2)
self.model.galaxy.plot_cmr(band1, band2, ax, all=True,
marker='.', ms=2 , ls='None', color='b', alpha=0.3)
self.model.cmr.plot_all(band1, band2, ax, color='c', alpha=0.5)
def view_fits(self, band):
"""
call DS9 to show pseudo color image
:param band: call ds9 to plot given band
"""
call_ds9(self.cluster_id, band)
if band=='gri':
self.viewer.button_fits_gri.config(state=Tk.DISABLED)
elif band=='riz':
self.viewer.button_fits_riz.config(state=Tk.DISABLED)
else: #grz
self.viewer.button_fits_grz.config(state=Tk.DISABLED)
def view_sketch(self, axis):
"""
create sketch plot to show galaxy positions
:param axis: axis for plotting
:return: None
"""
self.model.galaxy.plot_sky(axis, all=True,
color='b', alpha=0.3, marker='.', ms=2, ls='None')