def plotPer(time, y1, y2, y3 = None, figno = 2, \
savefile = None, period = None, fmp = 8):
'''Plot light curve and RV amplitude spectra'''
M, N = np.shape(y1)
pmax = 2* (np.nanmax(time) - np.nanmin(time))
if period is None:
dt = np.median(time[1:]-time[:N-1])
pmin = dt * 2.
else:
pmin = period / fmp
nper = 1000
if period is None:
fac = 1.0
else:
fac = period
if y3 is None:
ny = 2
else:
ny = 3
y = np.zeros((M*ny, N))
y[:M,:] = y1
y[M:2*M,:] = y2
if not (y3 is None):
y[2*M:,:] = y3
res = sinefitm(time, y, fmin = 1./pmax, fmax = 1./pmin, \
nfreq = nper)
freq, amps, ampc = res[0], res[2], res[3]
pers = 1.0 / freq
amp = np.sqrt(amps**2 + ampc**2)
amp1 = amp[:M,:]
amp2 = amp[M:2*M,:]
if not (y3 is None):
amp3 = amp[2*M:,:]
ee = dofig(figno, 1, ny, aspect = 1)
ax1 = doaxes(ee, 1, ny, 0, 0)
pl.setp(ax1.get_xticklabels(), visible = False)
pl.ylabel(r"$A_F$ (\%)")
for i in np.arange(M):
pl.plot(fac / pers, amp1[i,:] * fac1, ls[i], c = col[i])
pl.ylim(0, 1.1 * np.nanmax(amp1) * fac1)
ax2 = doaxes(ee, 1, ny, 0, 1, sharex = ax1)
pl.ylabel(r"$A_V$ (m/s)")
for i in np.arange(M):
pl.plot(fac / pers, amp2[i,:] * fac2, ls[i], c = col[i])
pl.ylim(0, 1.1 * np.nanmax(amp2) * fac2)
if not(y3 is None):
ax3 = doaxes(ee, 1, ny, 0, 2, sharex = ax1)
pl.ylabel(r"$A_{\mathrm{bis}}$ (m/s)")
for i in np.arange(M):
pl.plot(fac / pers, amp3[i,:] * fac3, ls[i], c = col[i])
pl.ylim(0, 1.1 * np.nanmax(amp3) * fac3)
if period is None:
pl.xlabel(r"Frequency (cycles/day)")
else:
pl.xlabel(r"Frequency (cycles/$P_{\mathrm{rot}}^{-1}$)")
if savefile:
pl.savefig(savefile)
return
def plotPer(time, y1, y2, y3=None, figno=2, savefile=None, period=None, fmp=8):
"""Plot light curve and RV amplitude spectra"""
M, N = np.shape(y1)
pmax = 2 * (np.nanmax(time) - np.nanmin(time))
if period == None:
dt = np.median(time[1:] - time[: N - 1])
pmin = dt * 2.0
else:
pmin = period / fmp
nper = 1000
if period == None:
fac = 1.0
else:
fac = period
if y3 is None:
ny = 2
else:
ny = 3
y = np.zeros((M * ny, N))
y[:M, :] = y1
y[M : 2 * M, :] = y2
if not y3 is None:
y[2 * M :, :] = y3
res = sinefitm(time, y, fmin=1.0 / pmax, fmax=1.0 / pmin, nfreq=nper)
freq, amps, ampc = res[0], res[2], res[3]
pers = 1.0 / freq
amp = np.sqrt(amps ** 2 + ampc ** 2)
amp1 = amp[:M, :]
amp2 = amp[M : 2 * M, :]
if not y3 is None:
amp3 = amp[2 * M :, :]
ee = dofig(figno, 1, ny, aspect=1)
ax1 = doaxes(ee, 1, ny, 0, 0)
pl.setp(ax1.get_xticklabels(), visible=False)
pl.ylabel(r"$A_F$ (\%)")
for i in np.arange(M):
pl.plot(fac / pers, amp1[i, :] * fac1, ls[i], c=col[i])
pl.ylim(0, 1.1 * np.nanmax(amp1) * fac1)
ax2 = doaxes(ee, 1, ny, 0, 1, sharex=ax1)
pl.ylabel(r"$A_V$ (m/s)")
for i in np.arange(M):
pl.plot(fac / pers, amp2[i, :] * fac2, ls[i], c=col[i])
pl.ylim(0, 1.1 * np.nanmax(amp2) * fac2)
if y3 != None:
ax3 = doaxes(ee, 1, ny, 0, 2, sharex=ax1)
pl.ylabel(r"$A_{\mathrm{bis}}$ (m/s)")
for i in np.arange(M):
pl.plot(fac / pers, amp3[i, :] * fac3, ls[i], c=col[i])
pl.ylim(0, 1.1 * np.nanmax(amp3) * fac3)
if period == None:
pl.xlabel(r"Frequency (cycles/day)")
else:
pl.xlabel(r"Frequency (cycles/$P_{\mathrm{rot}}^{-1}$)")
if savefile:
pl.savefig(savefile)
return
def plotTS(time, y1, y2, y3 = None, figno = 1, discrete = True, \
savefile = None, period = None, xper = False):
'''Plot light and RV curve(s)'''
M, N = np.shape(y1)
if discrete == True:
m1 = np.copy(mrk)
else:
m1 = np.copy(ls)
if (xper == True) * (not(period is None)):
tt = time / period - 0.5
xr = [-0.5,0.5]
xttl = 'phase'
else:
tt = time
xr = np.nanmin(time), np.nanmax(time)
xttl = 'time (days)'
if y3 is None:
ny = 2
else:
ny = 3
ee = dofig(figno, 1, ny, aspect = 1)
ax1 = doaxes(ee, 1, ny, 0, 0)
for i in np.arange(M):
pl.plot(tt, y1[i,:] * fac1, m1[i], c = col[i])
pl.ylabel(r"$\Delta F$ (\%)")
ymin = np.nanmin(y1) * fac1
ymax = np.nanmax(y1) * fac1
yr = ymax - ymin
pl.ylim(ymin - 0.1 * yr, ymax + 0.1 * yr)
ax2 = doaxes(ee, 1, ny, 0, 1, sharex = ax1)
for i in np.arange(M):
pl.plot(tt, y2[i,:] * fac2, m1[i], c = col[i])
pl.ylabel(r"$\Delta V$ (m/s)")
ymin = np.nanmin(y2) * fac2
ymax = np.nanmax(y2) * fac2
yr = ymax - ymin
pl.ylim(ymin - 0.1 * yr, ymax + 0.1 * yr)
if not(y3 is None):
ax3 = doaxes(ee, 1, ny, 0, 2, sharex = ax1)
for i in np.arange(M):
pl.plot(tt, y3[i,:] * fac2, m1[i], c = col[i])
pl.ylabel(r"$V_{\rm{bis}}$ (m/s)")
ymin = np.nanmin(y3) * fac2
ymax = np.nanmax(y3) * fac2
yr = ymax - ymin
pl.ylim(ymin - 0.1 * yr, ymax + 0.1 * yr)
pl.xlabel(xttl)
pl.xlim(xr[0], xr[1])
if savefile: pl.savefig(savefile)
return
def plotTS(time, y1, y2, y3=None, figno=1, discrete=True, savefile=None, period=None, xper=False):
"""Plot light and RV curve(s)"""
M, N = np.shape(y1)
if discrete == True:
m1 = np.copy(mrk)
else:
m1 = np.copy(ls)
if (xper == True) * (period != None):
tt = time / period - 0.5
xr = [-0.5, 0.5]
xttl = "phase"
else:
tt = time
xr = np.nanmin(time), np.nanmax(time)
xttl = "time (days)"
if y3 == None:
ny = 2
else:
ny = 3
ee = dofig(figno, 1, ny, aspect=1)
ax1 = doaxes(ee, 1, ny, 0, 0)
for i in np.arange(M):
pl.plot(tt, y1[i, :] * fac1, m1[i], c=col[i])
pl.ylabel(r"$\Delta F$ (\%)")
ymin = np.nanmin(y1) * fac1
ymax = np.nanmax(y1) * fac1
yr = ymax - ymin
pl.ylim(ymin - 0.1 * yr, ymax + 0.1 * yr)
ax2 = doaxes(ee, 1, ny, 0, 1, sharex=ax1)
for i in np.arange(M):
pl.plot(tt, y2[i, :] * fac2, m1[i], c=col[i])
pl.ylabel(r"$\Delta V$ (m/s)")
ymin = np.nanmin(y2) * fac2
ymax = np.nanmax(y2) * fac2
yr = ymax - ymin
pl.ylim(ymin - 0.1 * yr, ymax + 0.1 * yr)
if y3 != None:
ax3 = doaxes(ee, 1, ny, 0, 2, sharex=ax1)
for i in np.arange(M):
pl.plot(tt, y3[i, :] * fac2, m1[i], c=col[i])
pl.ylabel(r"$V_{\rm{bis}}$ (m/s)")
ymin = np.nanmin(y3) * fac2
ymax = np.nanmax(y3) * fac2
yr = ymax - ymin
pl.ylim(ymin - 0.1 * yr, ymax + 0.1 * yr)
pl.xlabel(xttl)
pl.xlim(xr[0], xr[1])
if savefile:
pl.savefig(savefile)
return
