im = ax.imshow(data,
cmap=cm.jet,
origin="lower",
norm=LogNorm(),
vmin=1e-3 * vmax,
vmax=vmax)
#t = [0.01, 0.1, 0.2, 0.4, 0.6, 0.8, 1.0]
fig.colorbar(im, cax=axcolor) #, ticks=t, format='$%.2f$')
# gc = aplpy.FITSFigure(cmap_file,figure=fig)
# gc.show_colorscale(aspect='auto')
if regions is not None:
r = pyregion.open(regions)
r2 = r.as_imagecoord(f[0].header)
patch_list, artist_list = r2.get_mpl_patches_texts()
for p in patch_list:
ax.add_patch(p)
pass
for t in artist_list:
ax.add_artist(t)
pass
pass
if pgwlist is not None:
_list = pgwave_list(pgwlist)
if len(_list.data) == 0:
print 'WARNING: pgwave candidate source list is empty.'
print 'Saving an empty output list...'
save_empty(args.pgwoutlist)
else:
l_b = _list.get_galactic()
ra_dec = _list.get_celestial()
snr, ksig, counts = _list.get_info()
def plot_state(directions_list, trim_names=True):
"""
Plots the facets of a run
"""
global midRA, midDec, fig, at, selected_direction, choose_from_list
selected_direction = None
choose_from_list = False
# Set up coordinate system and figure
points, midRA, midDec = factor.directions.getxy(directions_list)
fig = plt.figure(1, figsize=(10, 9))
if hasWCSaxes:
wcs = factor.directions.makeWCS(midRA, midDec)
ax = WCSAxes(fig, [0.16, 0.1, 0.8, 0.8], wcs=wcs)
fig.add_axes(ax)
else:
ax = plt.gca()
field_x = min(points[0])
field_y = max(points[1])
adjust_xy = True
while adjust_xy:
adjust_xy = False
for xy in points:
dist = np.sqrt((xy[0] - field_x)**2 + (xy[1] - field_y)**2)
if dist < 10.0:
field_x -= 1
field_y += 1
adjust_xy = True
break
field_ra, field_dec = factor.directions.xy2radec([field_x], [field_y],
refRA=midRA,
refDec=midDec)
field = Direction('field',
field_ra[0],
field_dec[0],
factor_working_dir=directions_list[0].working_dir)
directions_list.append(field)
ax.set_title('Overview of Factor run in\n{}'.format(
directions_list[0].working_dir))
# Plot facets
markers = []
for direction in directions_list:
if direction.name != 'field':
vertices = read_vertices(direction.vertices_file)
RAverts = vertices[0]
Decverts = vertices[1]
xverts, yverts = factor.directions.radec2xy(RAverts,
Decverts,
refRA=midRA,
refDec=midDec)
xyverts = [np.array([xp, yp]) for xp, yp in zip(xverts, yverts)]
mpl_poly = Polygon(np.array(xyverts),
edgecolor='#a9a9a9',
facecolor='#F2F2F2',
clip_box=ax.bbox,
picker=3.0,
linewidth=2)
else:
xverts = [field_x]
yverts = [field_y]
mpl_poly = Circle((field_x, field_y),
radius=5.0,
edgecolor='#a9a9a9',
facecolor='#F2F2F2',
clip_box=ax.bbox,
picker=3.0,
linewidth=2)
mpl_poly.facet_name = direction.name
mpl_poly.completed_ops = get_completed_ops(direction)
mpl_poly.started_ops = get_started_ops(direction)
mpl_poly.current_op = get_current_op(direction)
set_patch_color(mpl_poly, direction)
ax.add_patch(mpl_poly)
# Add facet names
if direction.name != 'field':
poly_tuple = tuple([(xp, yp) for xp, yp in zip(xverts, yverts)])
xmid = SPolygon(poly_tuple).centroid.x
ymid = SPolygon(poly_tuple).centroid.y
else:
xmid = field_x
ymid = field_y
if trim_names:
name = direction.name.split('_')[-1]
else:
name = direction.name
marker = ax.text(xmid,
ymid,
name,
color='k',
clip_on=True,
clip_box=ax.bbox,
ha='center',
va='bottom')
marker.set_zorder(1001)
markers.append(marker)
# Add info box
at = AnchoredText("Selected direction: None",
prop=dict(size=12),
frameon=True,
loc=3)
at.patch.set_boxstyle("round,pad=0.,rounding_size=0.2")
at.set_zorder(1002)
ax.add_artist(at)
ax.relim()
ax.autoscale()
ax.set_aspect('equal')
if hasWCSaxes:
RAAxis = ax.coords['ra']
RAAxis.set_axislabel('RA', minpad=0.75)
RAAxis.set_major_formatter('hh:mm:ss')
DecAxis = ax.coords['dec']
DecAxis.set_axislabel('Dec', minpad=0.75)
DecAxis.set_major_formatter('dd:mm:ss')
ax.coords.grid(color='black', alpha=0.5, linestyle='solid')
else:
plt.xlabel("RA (arb. units)")
plt.ylabel("Dec (arb. units)")
# Define coodinate formater to show RA and Dec under mouse pointer
ax.format_coord = formatCoord
# Show legend
not_processed_patch = plt.Rectangle((0, 0),
1,
1,
edgecolor='#a9a9a9',
facecolor='#F2F2F2',
linewidth=2)
processing_patch = plt.Rectangle((0, 0),
1,
1,
edgecolor='#a9a9a9',
facecolor='#F2F5A9',
linewidth=2)
selfcal_ok_patch = plt.Rectangle((0, 0),
1,
1,
edgecolor='#a9a9a9',
facecolor='#A9F5A9',
linewidth=2)
selfcal_not_ok_patch = plt.Rectangle((0, 0),
1,
1,
edgecolor='#a9a9a9',
facecolor='#A4A4A4',
linewidth=2)
processing_error = plt.Rectangle((0, 0),
1,
1,
edgecolor='#a9a9a9',
facecolor='#F5A9A9',
linewidth=2)
patch_list = [
not_processed_patch, processing_patch, processing_error,
selfcal_not_ok_patch, selfcal_ok_patch
]
label_list = [
'Unprocessed', 'Processing', 'Pipeline Error', 'Selfcal Failed',
'Selfcal OK'
]
for i in range(options['reimages']):
label_list.append('Image ' + str(i + 1))
color = (0.66 / (i + 2)**0.5, 0.96 / (i + 2)**0.5, 0.66 / (i + 2)**0.5,
1.0)
reimage_patch = plt.Rectangle((0, 0),
1,
1,
edgecolor='#a9a9a9',
facecolor=color,
linewidth=2)
patch_list.append(reimage_patch)
l = ax.legend(patch_list, label_list, loc="upper right")
l.set_zorder(1002)
# Add check for mouse clicks and key presses
fig.canvas.mpl_connect('pick_event', on_pick)
fig.canvas.mpl_connect('key_press_event', on_press)
# Add timer to update the plot every 60 seconds
timer = fig.canvas.new_timer(interval=60000)
timer.add_callback(update_plot)
timer.start()
# Show plot
plt.show()
plt.close(fig)
# Clean up any temp casacore images
if not hasaplpy:
if os.path.exists('/tmp/tempimage'):
try:
shutil.rmtree('/tmp/tempimage')
except OSError:
pass
ax.imshow(f_xray[0].data, cmap=cm.gray, vmin=0., vmax=0.00038, origin="lower")
reg_name = "test.reg"
r = pyregion.open(reg_name).as_imagecoord(header=f_xray[0].header)
from pyregion.mpl_helper import properties_func_default
# Use custom function for patch attribute
def fixed_color(shape, saved_attrs):
attr_list, attr_dict = saved_attrs
attr_dict["color"] = "red"
kwargs = properties_func_default(shape, (attr_list, attr_dict))
return kwargs
# select region shape with tag=="Group 1"
r1 = pyregion.ShapeList([rr for rr in r if rr.attr[1].get("tag") == "Group 1"])
patch_list1, artist_list1 = r1.get_mpl_patches_texts(fixed_color)
r2 = pyregion.ShapeList([rr for rr in r if rr.attr[1].get("tag") != "Group 1"])
patch_list2, artist_list2 = r2.get_mpl_patches_texts()
for p in patch_list1 + patch_list2:
ax.add_patch(p)
for t in artist_list1 + artist_list2:
ax.add_artist(t)
plt.show()
def plot_state(directions_list, trim_names=True):
"""
Plots the facets of a run
"""
global midRA, midDec, fig, at, selected_direction, choose_from_list
selected_direction = None
choose_from_list = False
# Set up coordinate system and figure
points, midRA, midDec = factor.directions.getxy(directions_list)
fig = plt.figure(1, figsize=(10,9))
if hasWCSaxes:
wcs = factor.directions.makeWCS(midRA, midDec)
ax = WCSAxes(fig, [0.16, 0.1, 0.8, 0.8], wcs=wcs)
fig.add_axes(ax)
else:
ax = plt.gca()
field_x = min(points[0])
field_y = max(points[1])
adjust_xy = True
while adjust_xy:
adjust_xy = False
for xy in points:
dist = np.sqrt( (xy[0] - field_x)**2 + (xy[1] - field_y)**2 )
if dist < 10.0:
field_x -= 1
field_y += 1
adjust_xy = True
break
field_ra, field_dec = factor.directions.xy2radec([field_x], [field_y],
refRA=midRA, refDec=midDec)
field = Direction('field', field_ra[0], field_dec[0],
factor_working_dir=directions_list[0].working_dir)
directions_list.append(field)
ax.set_title('Overview of Factor run in\n{}'.format(directions_list[0].working_dir))
# Plot facets
markers = []
for direction in directions_list:
if direction.name != 'field':
vertices = read_vertices(direction.vertices_file)
RAverts = vertices[0]
Decverts = vertices[1]
xverts, yverts = factor.directions.radec2xy(RAverts, Decverts,
refRA=midRA, refDec=midDec)
xyverts = [np.array([xp, yp]) for xp, yp in zip(xverts, yverts)]
mpl_poly = Polygon(np.array(xyverts), edgecolor='#a9a9a9', facecolor='#F2F2F2',
clip_box=ax.bbox, picker=3.0, linewidth=2)
else:
xverts = [field_x]
yverts = [field_y]
mpl_poly = Circle((field_x, field_y), radius=5.0, edgecolor='#a9a9a9', facecolor='#F2F2F2',
clip_box=ax.bbox, picker=3.0, linewidth=2)
mpl_poly.facet_name = direction.name
mpl_poly.completed_ops = get_completed_ops(direction)
mpl_poly.started_ops = get_started_ops(direction)
mpl_poly.current_op = get_current_op(direction)
set_patch_color(mpl_poly, direction)
ax.add_patch(mpl_poly)
# Add facet names
if direction.name != 'field':
poly_tuple = tuple([(xp, yp) for xp, yp in zip(xverts, yverts)])
xmid = SPolygon(poly_tuple).centroid.x
ymid = SPolygon(poly_tuple).centroid.y
else:
xmid = field_x
ymid = field_y
if trim_names:
name = direction.name.split('_')[-1]
else:
name = direction.name
marker = ax.text(xmid, ymid, name, color='k', clip_on=True,
clip_box=ax.bbox, ha='center', va='bottom')
marker.set_zorder(1001)
markers.append(marker)
# Add info box
at = AnchoredText("Selected direction: None", prop=dict(size=12), frameon=True,
loc=3)
at.patch.set_boxstyle("round,pad=0.,rounding_size=0.2")
at.set_zorder(1002)
ax.add_artist(at)
ax.relim()
ax.autoscale()
ax.set_aspect('equal')
if hasWCSaxes:
RAAxis = ax.coords['ra']
RAAxis.set_axislabel('RA', minpad=0.75)
RAAxis.set_major_formatter('hh:mm:ss')
DecAxis = ax.coords['dec']
DecAxis.set_axislabel('Dec', minpad=0.75)
DecAxis.set_major_formatter('dd:mm:ss')
ax.coords.grid(color='black', alpha=0.5, linestyle='solid')
else:
plt.xlabel("RA (arb. units)")
plt.ylabel("Dec (arb. units)")
# Define coodinate formater to show RA and Dec under mouse pointer
ax.format_coord = formatCoord
# Show legend
not_processed_patch = plt.Rectangle((0, 0), 1, 1, edgecolor='#a9a9a9',
facecolor='#F2F2F2', linewidth=2)
processing_patch = plt.Rectangle((0, 0), 1, 1, edgecolor='#a9a9a9',
facecolor='#F2F5A9', linewidth=2)
selfcal_ok_patch = plt.Rectangle((0, 0), 1, 1, edgecolor='#a9a9a9',
facecolor='#A9F5A9', linewidth=2)
selfcal_not_ok_patch = plt.Rectangle((0, 0), 1, 1, edgecolor='#a9a9a9',
facecolor='#A4A4A4', linewidth=2)
processing_error = plt.Rectangle((0, 0), 1, 1, edgecolor='#a9a9a9',
facecolor='#F5A9A9', linewidth=2)
patch_list=[not_processed_patch, processing_patch, processing_error, selfcal_not_ok_patch, selfcal_ok_patch]
label_list=['Unprocessed', 'Processing', 'Pipeline Error', 'Selfcal Failed', 'Selfcal OK']
for i in range(options['reimages']):
label_list.append('Image '+str(i+1))
color=(0.66/(i+2)**0.5, 0.96/(i+2)**0.5, 0.66/(i+2)**0.5, 1.0)
reimage_patch=plt.Rectangle((0, 0), 1, 1, edgecolor='#a9a9a9',
facecolor=color, linewidth=2)
patch_list.append(reimage_patch)
l = ax.legend(patch_list, label_list, loc="upper right")
l.set_zorder(1002)
# Add check for mouse clicks and key presses
fig.canvas.mpl_connect('pick_event', on_pick)
fig.canvas.mpl_connect('key_press_event', on_press)
# Add timer to update the plot every 60 seconds
timer = fig.canvas.new_timer(interval=60000)
timer.add_callback(update_plot)
timer.start()
# Show plot
plt.show()
plt.close(fig)
# Clean up any temp casacore images
if not hasaplpy:
if os.path.exists('/tmp/tempimage'):
try:
shutil.rmtree('/tmp/tempimage')
except OSError:
pass