def plot_phase(epicID,
campaign,
period,
t0,
directory="/k2_data/",
tail="",
initial_time=0,
fn='',
raw=False,
save_path=''):
'''see phase'''
time, flux = data.retrieve(epicID,
campaign,
directory,
tail,
fn=fn,
raw=raw)
if initial_time != 0:
time, flux = time[time > initial_time], flux[time > initial_time]
plot_phase_work(time,
period,
t0,
flux,
"EPIC " + str(epicID) +
":period={0:.3f}, t0={1:.6f}".format(period, t0),
save_path=save_path)
def retrieve_random_quote():
"""Fetches a random quote from the database.
Returns:
- result (string) the quote retrieved from the database, or an error
message
"""
result = data.retrieve()
if result == -1:
QUOTE_REQUESTS.labels(command_type="quote", result="failure").inc()
return ":( there was an error getting a quote."
QUOTE_REQUESTS.labels(command_type="quote", result="success").inc()
return result
def plot_lc(epicID, campaign, directory="/k2_data/",mark_list=[],tail="",injected=False,ylimtype="med",xlim=[0,0],initial_time = 0,fn='',raw=False,save_path=''):
'''see lightcurve'''
epicID,campaign = str(epicID),str(campaign)
if not injected:
time, flux = data.retrieve(epicID,campaign,directory,tail,fn=fn,raw=raw)
else:
time, flux, transits = data.retrieve(epicID,campaign,directory,tail,injected=True,fn=fn,raw=raw)
if initial_time != 0:
time,flux = time[time>initial_time], flux[time>initial_time]
fig = pl.figure(figsize=(10 * golden,10))
pl.title("EPIC " + epicID, **title_font)
pl.plot(time,flux,'k',lw=0.3)####################
pl.scatter(time,flux,c='darkslategrey',lw=0,s=30)#######################
pl.xlabel("Time (BJD)", **axis_font)
pl.ylabel("Flux", **axis_font)
pl.plot(mark_list, np.median(flux) + np.zeros(len(mark_list)),'r*',markersize=20)
if injected:
pl.plot(transits, np.median(flux) + np.zeros(len(transits)),'r*',markersize=20)
if xlim != [0,0]:
pl.xlim(xlim)
flux = flux[((time>xlim[0])*(time<xlim[1]))]
else:
xlim = [min(time),max(time)]
flux = flux[((time>xlim[0])*(time<xlim[1]))]
if ylimtype=="med":
pl.ylim(np.median(flux)-0.5*np.std(flux),np.median(flux)+0.5*np.std(flux))
elif ylimtype=="minmax":
pl.ylim(np.min(flux),np.max(flux))
elif ylimtype=="pick":
pl.ylim(min([np.min(flux),np.median(flux)-0.5*np.std(flux)]),
max([np.max(flux),np.median(flux)+0.5*np.std(flux)]))
if save_path == '':
pl.show()
else:
pl.savefig(save_path)
pl.close()
del mark_list
def blswrap(epicid, campaign, initial_time):
global nbin
try:
start = time.time()
initial_time = float(initial_time)
t, f = data.retrieve(epicid, campaign, directory=directory)
if initial_time != 0:
t, f = t[t>initial_time], f[t>initial_time]
u, v = np.zeros(len(t)), np.zeros(len(f))
#minfreq, dfreq, nfreq = 1/70., 4.082799167108228e-06, 1000000
minfreq, dfreq, nfreq = 0.015, 2.0437359493152146e-05,100000
#nbin = 100
minduration, maxduration = 0.01, 0.05
results = bls.eebls(t, f, u, v, nfreq, minfreq, dfreq, nbin, minduration, maxduration)
end = time.time()
print(epicid, end - start)
return epicid, results[1:]
except:
print("SKIPPING " + str(epicid))
def retrieve():
global state, profile, conn
cmd.print_credentials(profile, data.retrieve(conn),
data.column_names(conn))
state = states.DISPLAY
def sintatic():
queue.append('$' + ',' + str(0))
stack = [0]
begin = 0
a, pos = queue[begin].split(',')
while True:
s = stack[0]
if 'S' in data.retrieve(s, a):
null = data.retrieve(s, a)
t = int(null[1:]) - 1
stack.insert(0, t)
semstack(a, begin)
begin += 1
a, pos = queue[begin].split(',')
elif 'R' in data.retrieve(s, a):
null = data.retrieve(s, a)
t = int(null[1:])
A = data.goto(t)
for i in range(data.rules(t, 0) - 2):
stack.pop(0)
s = stack[0]
aux = data.retrieve(s, A)
if type(aux) == float:
s = int(aux) - 1
elif type(aux) == int:
s = aux - 1
else:
s = int(aux[1:]) - 1
stack.insert(0, s)
semantic(t)
elif 'ACC' in data.retrieve(s, a):
semantic(12)
semantic(12)
semantic(23)
break
elif 'E' in data.retrieve(s, a):
null = data.retrieve(s, a)
t = int(null[1:])
data.handle_error(t, pos)
begin += 1
if t == 4:
begin -= 2
queue.insert(-1, 'fim,' + pos)
a, pos = queue[begin].split(',')
elif t == 6:
begin -= 1
queue.insert(0, 'inicio,' + pos)
a, pos = queue[begin].split(',')
elif t == 7:
begin -= 1
queue.insert(1, 'varinicio,' + pos)
a, pos = queue[begin].split(',')
elif t == 8:
stack.insert(0, 56)
elif t == 9:
begin -= 3
a, pos = queue[begin].split(',')
elif t == 10:
stack.insert(0, 1)
a, pos = queue[begin].split(',')
elif t == 11:
begin += 1
a, pos = queue[begin].split(',')
else:
a, pos = queue[begin].split(',')
else:
code = data.show_error(a)
if code == 1:
begin += 1
a, pos = queue[begin].split(',')
def plot_phase(epicID,campaign,period, t0, directory ="/k2_data/",tail="",initial_time=0,fn='',raw=False, save_path=''):
'''see phase'''
time, flux = data.retrieve(epicID,campaign,directory,tail,fn=fn,raw=raw)
if initial_time != 0:
time,flux = time[time>initial_time], flux[time>initial_time]
plot_phase_work(time, period, t0, flux, "EPIC " + str(epicID) + ":period={0:.3f}, t0={1:.6f}".format(period,t0),save_path=save_path)
def test_retrieve_fails(self):
expected_response = -1
data.connect = MagicMock(return_value=[self.conn, self.conn])
self.conn.fetchone = MagicMock(side_effect=psycopg2.Error)
received_response = data.retrieve()
self.assertEqual(expected_response, received_response)
def test_retrieve(self):
expected_response = "I am a quote"
data.connect = MagicMock(return_value=[self.conn, self.conn])
received_response = data.retrieve()
self.assertEqual(expected_response, received_response)
def plot_lc(epicID,
campaign,
directory="/k2_data/",
mark_list=[],
tail="",
injected=False,
ylimtype="med",
xlim=[0, 0],
initial_time=0,
fn='',
raw=False,
save_path=''):
'''see lightcurve'''
epicID, campaign = str(epicID), str(campaign)
if not injected:
time, flux = data.retrieve(epicID,
campaign,
directory,
tail,
fn=fn,
raw=raw)
else:
time, flux, transits = data.retrieve(epicID,
campaign,
directory,
tail,
injected=True,
fn=fn,
raw=raw)
if initial_time != 0:
time, flux = time[time > initial_time], flux[time > initial_time]
fig = pl.figure(figsize=(10 * golden, 10))
pl.title("EPIC " + epicID, **title_font)
pl.plot(time, flux, 'k', lw=0.3) ####################
pl.scatter(time, flux, c='darkslategrey', lw=0,
s=30) #######################
pl.xlabel("Time (BJD)", **axis_font)
pl.ylabel("Flux", **axis_font)
pl.plot(mark_list,
np.median(flux) + np.zeros(len(mark_list)),
'r*',
markersize=20)
if injected:
pl.plot(transits,
np.median(flux) + np.zeros(len(transits)),
'r*',
markersize=20)
if xlim != [0, 0]:
pl.xlim(xlim)
flux = flux[((time > xlim[0]) * (time < xlim[1]))]
else:
xlim = [min(time), max(time)]
flux = flux[((time > xlim[0]) * (time < xlim[1]))]
if ylimtype == "med":
pl.ylim(
np.median(flux) - 0.5 * np.std(flux),
np.median(flux) + 0.5 * np.std(flux))
elif ylimtype == "minmax":
pl.ylim(np.min(flux), np.max(flux))
elif ylimtype == "pick":
pl.ylim(min([np.min(flux),
np.median(flux) - 0.5 * np.std(flux)]),
max([np.max(flux),
np.median(flux) + 0.5 * np.std(flux)]))
if save_path == '':
pl.show()
else:
pl.savefig(save_path)
pl.close()
del mark_list
