def zoom_snap2(self, snaps, field, sizes, base, sims):
nsizes = len(sizes)
ncols = np.ceil(np.sqrt(nsizes))
nrows = np.ceil(nsizes / ncols)
ncols, nrows = int(ncols), int(nrows)
left, right, bottom, top = 0.1, 0.9, 0.1, 0.9
width = (right - left) / ncols
height = (top - bottom) / nrows
for idx_size in range(nsizes):
sim = sims[idx_size]
snap = snaps[idx_size]
size = sizes[idx_size]
if idx_size == 0:
read_snap = True
else:
if sims[idx_size - 1] == sim and snaps[idx_size - 1] == snap:
read_snap = False
else:
read_snap = True
path = base + sim + '/snapdir_%03i/' % snap
file = sim + '_%03i' % snap
snapbase = path + file
if read_snap:
MySnap = Snap()
MySnap.read_header(snapbase)
MySnap.read_fields(snapbase)
MySnap.center_box()
MySnap.rotate_box()
MyImage = Image()
MyImage.calculate_image(MySnap, field, ImgWidth=size)
if idx_size == 0 or idx_size == 1 or idx_size == 2:
idx_subplot = idx_size + 1
elif idx_size == 3:
idx_subplot = 6
elif idx_size == 4:
idx_subplot = 5
elif idx_size == 5:
idx_subplot = 4
ax = self.fig.add_subplot(nrows, ncols, idx_subplot)
ax.set(aspect=1)
#vmin, vmax = np.min(MyImage.img), np.max(MyImage.img)
if idx_size == 0:
vmin, vmax = 3, 7
elif idx_size == 1:
vmin, vmax = 6, 9.5
elif idx_size == 2:
vmin, vmax = 8, 12
elif idx_size == 3:
vmin, vmax = 10, 14
elif idx_size == 4:
vmin, vmax = 12, 16
elif idx_size == 5:
vmin, vmax = 14, 19
im = ax.imshow(MyImage.img,
cmap=get_colormap(field),
extent=[0, MyImage.xbins, 0, MyImage.ybins],
vmin=vmin,
vmax=vmax)
if size > 1e4:
size_label = str(size / 2e5) + ' pc'
else:
size_label = '%i au' % size
ax.text(MyImage.xbins / 10.,
MyImage.ybins / 10.,
size_label,
fontsize=36,
color='w')
ax.set_xticklabels([])
ax.set_yticklabels([])
# Zoom effect
if idx_size != nsizes - 1:
dx, dy = float(
sizes[idx_size + 1]) / float(size) * MyImage.xbins, float(
sizes[idx_size + 1]) / float(size) * MyImage.ybins
xi, xf = MyImage.xbins / 2. - dx / 2., MyImage.xbins / 2. + dx / 2.
yi, yf = MyImage.ybins / 2. - dy / 2., MyImage.ybins / 2. + dy / 2.
# Square
square = pl.Rectangle((xi, yi),
dx,
dy,
facecolor='none',
edgecolor='k')
ax.add_patch(square)
if idx_size == 0 or idx_size == 1:
x1, y1 = np.array([xi, MyImage.xbins]), np.array([yi, 0])
x2, y2 = np.array([xi, MyImage.xbins
]), np.array([yf, MyImage.ybins])
elif idx_size == 2:
x1, y1 = np.array([xi, 0]), np.array([yf, 0])
x2, y2 = np.array([xf, MyImage.xbins]), np.array([yf, 0])
elif idx_size == 3 or idx_size == 4:
x1, y1 = np.array([xf, 0]), np.array([yi, 0])
x2, y2 = np.array([xf, 0]), np.array([yf, MyImage.ybins])
# Lines connecting edges
ax.plot(x1, y1, 'k')
ax.plot(x2, y2, 'k')
# Fixing limits
ax.set_xlim(0, MyImage.xbins)
ax.set_ylim(0, MyImage.ybins)
# Colorbar
if idx_size == 0 or idx_size == 1 or idx_size == 2:
cax = pl.axes([
left + (idx_subplot - 1) * width + 0.01,
bottom + nrows * height + 0.04, width - 0.02, 0.02
])
cb = pl.colorbar(im,
cax=cax,
orientation='horizontal',
ticks=np.arange(vmin, vmax + 1).astype(int))
cax.set_xlabel(get_label(field), fontsize=30, labelpad=-85)
cb.ax.tick_params(labelsize=24)
elif idx_size == 3 or idx_size == 4 or idx_size == 5:
cax = pl.axes([
left + (idx_subplot - 4) * width + 0.01, bottom - 0.07,
width - 0.02, 0.02
])
cb = pl.colorbar(im,
cax=cax,
orientation='horizontal',
ticks=np.arange(vmin, vmax + 1).astype(int))
cax.set_xlabel(get_label(field), fontsize=30, labelpad=-85)
cb.ax.tick_params(labelsize=24)
pl.subplots_adjust(left=left,
right=right,
bottom=bottom,
top=top,
wspace=0.0,
hspace=0.0)
self.fig.set_size_inches(ncols * 8, nrows * 8)
del MySnap, MyImage
def zoom_snap(self, snap, fields, sizes, base, sim):
path = base + sim + '/snapdir_%03i/' % snap
file = sim + '_%03i' % snap
snapbase = path + file
MySnap = Snap()
MySnap.read_header(snapbase)
MySnap.read_fields(snapbase)
MySnap.center_box()
#MySnap.rotate_box()
nsizes = len(sizes)
nfields = len(fields)
nrows, ncols = nfields, nsizes
vmin = np.zeros(nfields)
vmax = np.zeros(nfields)
cbar_mode = 'edge'
grid = ImageGrid(self.fig,
111,
nrows_ncols=[nrows, ncols],
axes_pad=0.0,
label_mode='all',
share_all=False,
cbar_location='right',
cbar_mode=cbar_mode,
cbar_pad='5%',
cbar_size='5%')
for idx_size in range(nsizes):
size = sizes[idx_size]
for idx_field in range(nfields):
field = fields[idx_field]
MyImage = Image()
MyImage.calculate_image(MySnap, field, ImgWidth=size)
if idx_size == 0:
vmin[idx_field] = np.min(MyImage.img)
vmax[idx_field] = np.max(MyImage.img)
#vmin[0] = 0.1
#vmax[0] = 10
idx_ax = nsizes * idx_field + idx_size
im = grid[idx_ax].imshow(
MyImage.img,
cmap=get_colormap(field),
extent=[0, MyImage.xbins, 0, MyImage.ybins],
vmin=vmin[idx_field],
vmax=vmax[idx_field])
grid[idx_ax].text(MyImage.xbins / 10.,
MyImage.ybins / 10.,
'%i pc' % (size * 1e3),
fontsize=36,
color='w')
grid[idx_ax].set_xticklabels([])
grid[idx_ax].set_yticklabels([])
if ((idx_ax + 1) % nsizes != 0):
dx, dy = float(sizes[idx_size + 1]) / float(
size) * MyImage.xbins, float(
sizes[idx_size + 1]) / float(size) * MyImage.ybins
xi, xf = MyImage.xbins / 2. - dx / 2., MyImage.xbins / 2. + dx / 2.
yi, yf = MyImage.ybins / 2. - dy / 2., MyImage.ybins / 2. + dy / 2.
# Square
square = pl.Rectangle((xi, yi),
dx,
dy,
facecolor='none',
edgecolor='w')
grid[idx_ax].add_patch(square)
# Lines connecting edges
grid[idx_ax].plot(np.array([xi, MyImage.xbins]),
np.array([yi, 0]), 'w')
grid[idx_ax].plot(np.array([xi, MyImage.xbins]),
np.array([yf, MyImage.ybins]), 'w')
# Fixing limits
grid[idx_ax].set_xlim(0, MyImage.xbins)
grid[idx_ax].set_ylim(0, MyImage.ybins)
# if idx_size == 1:
# circ = pl.Circle((MyImage.xbins/2, MyImage.ybins/2), radius=MyImage.xbins*0.18e-3/size, edgecolor='k', linestyle='dashed', fill=False)
# grid[idx_ax].add_patch(circ)
# if nfields == 1 and idx_size == 0:
# cb = grid.cbar_axes[idx_ax].colorbar(im)
# cb.set_label_text(get_label(field), fontsize = 52)
# cb.ax.tick_params(labelsize=40)
# #cb.set_clim()
# if ((idx_size == nsizes - 1) and (nfields != 1)):
# idx_cb = idx_ax - (idx_field+1) * (nsizes-1)
# cb = grid.cbar_axes[idx_cb].colorbar(im)
# cb.set_label_text(get_label(field), fontsize = 24)
# cb.ax.tick_params(labelsize=18)
# #cb.set_clim()
self.fig.set_size_inches(ncols * 8, nrows * 8)
del MySnap, MyImage
def multi_snap(self,
snaps,
fields,
base,
sim,
ImgWidth,
PlotTime=False,
PlotSize=False,
CenterProto=None,
PlotProto=False,
Rotate=False):
nsnaps = len(snaps)
nfields = len(fields)
if nfields == 1:
ncols = np.ceil(np.sqrt(nsnaps))
nrows = np.ceil(nsnaps / ncols)
ncols, nrows = int(ncols), int(nrows)
cbar_mode = 'single'
else:
nrows, ncols = nfields, nsnaps
cbar_mode = 'edge'
grid = ImageGrid(self.fig,
111,
nrows_ncols=[nrows, ncols],
axes_pad=0.0,
label_mode='L',
share_all=False,
cbar_location='right',
cbar_mode=cbar_mode,
cbar_pad='5%',
cbar_size='5%')
#vmin, vmax = 13, 20.5
for idx_snap in range(nsnaps):
snap = snaps[idx_snap]
path = base + sim + '/snapdir_%03i/' % snap
file = sim + '_%03i' % snap
snapbase = path + file
MySnap = Snap()
MySnap.read_header(snapbase)
MySnap.read_fields(snapbase)
if CenterProto:
if snap < 126:
f = h5py.File(
'/scratch/02563/fbecerra/paha/protostars/' + file +
'.hdf5', 'r')
try:
Center = np.array(f['Proto%i/Position' % CenterProto])
except:
f.close()
continue
f.close()
ImgWidth = 20
else:
ImgWidth = 50
MySnap.fields['x'] -= Center[0]
MySnap.fields['y'] -= Center[1]
MySnap.fields['z'] -= Center[2]
else:
MySnap.center_box()
if Rotate:
MySnap.rotate_box()
print MySnap.Center
for idx_field in range(nfields):
field = fields[idx_field]
MyImage = Image()
MyImage.calculate_image(MySnap, field, ImgWidth=ImgWidth)
idx_ax = nsnaps * idx_field + idx_snap
vmin, vmax = np.min(MyImage.img), np.max(MyImage.img)
#if field == 'nh':
# vmin, vmax = 4.6, 8
# vmin, vmax = 7.1, 10
#elif field == 'temp':
# vmin, vmax = 3.7, 4.5
# vmin, vmax = 3.7, 4.1
if field == 'phodens':
vmin, vmax = 0, 14
im = grid[idx_ax].imshow(
MyImage.img,
cmap=get_colormap(field),
extent=[0, MyImage.xbins, 0, MyImage.ybins],
vmin=vmin,
vmax=vmax)
try:
nsinks = len(MySnap.new_fields['sinks']['id'])
if nsinks > 0:
print MySnap.new_fields['sinks'][
'x'], MySnap.new_fields['sinks']['y']
x_sinks = MyImage.xbins / 2. + MyImage.xbins * MySnap.new_fields[
'sinks']['x'] / MyImage.width
y_sinks = MyImage.ybins / 2. + MyImage.ybins * MySnap.new_fields[
'sinks']['y'] / MyImage.height
for cx, cy in zip(x_sinks, y_sinks):
if ((cx > 0) & (cx < MyImage.xbins) & (cy > 0) &
(cy < MyImage.ybins)):
grid[idx_ax].plot(cx, cy, 'w+', ms=5, mew=3)
except:
print 'Snapshot does not have sink particles'
if PlotProto:
f = h5py.File(
'/scratch/02563/fbecerra/paha/protostars/' + file +
'.hdf5', 'r')
for key in f.keys():
protostar_center = np.array(f[key + '/Position'])
protostar_center -= Center
if Rotate:
x_proto = MySnap.params['rot12'][0][
0] * protostar_center[0] + MySnap.params[
'rot12'][0][1] * protostar_center[
1] + MySnap.params['rot12'][0][
2] * protostar_center[2]
y_proto = MySnap.params['rot12'][1][
0] * protostar_center[0] + MySnap.params[
'rot12'][1][1] * protostar_center[
1] + MySnap.params['rot12'][1][
2] * protostar_center[2]
else:
x_proto = protostar_center[0]
y_proto = protostar_center[1]
protostar_radius = np.float64(f[key + '/Radius'])
x_proto += MyImage.width / 2
y_proto += MyImage.height / 2
if ((x_proto > 0) & (x_proto < MyImage.width) &
(y_proto > 0) & (y_proto < MyImage.height)):
grid[idx_ax].plot(x_proto, y_proto, 'k+')
circ = pl.Circle((x_proto, y_proto),
radius=protostar_radius,
edgecolor='k',
fill=False)
grid[idx_ax].add_patch(circ)
f.close()
if PlotTime:
grid[idx_ax].text(
MyImage.xbins / 10.,
MyImage.ybins / 10.,
'%.2f yr' %
(MySnap.params['time'] * UNIT_TIME / SEC_PER_YEAR),
fontsize=36,
color='w')
if PlotSize:
grid[idx_ax].text(MyImage.xbins * (1 - 3. / 10),
MyImage.ybins * (1 - 1. / 10),
'%.2f pc' % (ImgWidth * 1e3),
fontsize=36,
color='w')
#grid[idx_ax].text(MyImage.xbins/10., MyImage.ybins*(1-1./10), sim, fontsize=36, color='w')
#circ = pl.Circle((MyImage.xbins/2, MyImage.ybins/2), radius=MyImage.xbins*2e-3/ImgWidth, edgecolor='white', linestyle='dashed', fill=False)
#grid[idx_ax].add_patch(circ)
grid[idx_ax].set_xlim(0, MyImage.xbins)
grid[idx_ax].set_ylim(0, MyImage.ybins)
grid[idx_ax].set_xticklabels([])
grid[idx_ax].set_yticklabels([])
if nfields == 1 and idx_snap == 0:
cb = grid.cbar_axes[idx_ax].colorbar(im)
cb.set_label_text(get_label(field), fontsize=nrows * 17)
cb.ax.tick_params(labelsize=nrows * 14)
#cb.set_clim()
if ((idx_snap == nsnaps - 1) and (nfields != 1)):
idx_cb = idx_ax - (idx_field + 1) * (nsnaps - 1)
cb = grid.cbar_axes[idx_cb].colorbar(im)
cb.set_label_text(get_label(field), fontsize=24)
cb.ax.tick_params(labelsize=18)
#cb.set_clim()
self.fig.set_size_inches(ncols * 8, nrows * 8)
del MySnap, MyImage
