Skip to content
Snippets Groups Projects
Commit eae07bb7 authored by Delorme, Maxime Dr (Physics)'s avatar Delorme, Maxime Dr (Physics)
Browse files

Decoupled data and subtracted data

parent 533ca7b7
No related branches found
No related tags found
No related merge requests found
...@@ -19,6 +19,7 @@ import os, sys, warnings ...@@ -19,6 +19,7 @@ import os, sys, warnings
# Objects # Objects
data = None data = None
data_sub = None
h0 = None h0 = None
h1 = None h1 = None
zs = ZScaleInterval() zs = ZScaleInterval()
...@@ -66,16 +67,21 @@ def get_parameter(param_name): ...@@ -66,16 +67,21 @@ def get_parameter(param_name):
return None return None
def plot_data(zscale=True): def plot_data(zscale=True):
if bg_sub:
pdata = data_sub
else:
pdata = data
if zscale: if zscale:
lims = zs.get_limits(data) lims = zs.get_limits(pdata)
else: else:
lims = [data.min(), data.max()] lims = [pdata.min(), pdata.max()]
plt.imshow(data, cmap='Greys_r', vmin=lims[0], vmax=lims[1], origin='lower') plt.imshow(pdata, cmap='Greys_r', vmin=lims[0], vmax=lims[1], origin='lower')
plt.show() plt.show()
def subtract_background(bg_wsize=50, sclip=3.0, plot=False): def subtract_background(bg_wsize=50, sclip=3.0, plot=False):
global data, error, bg_sub global data, error, data_sub, bg_sub
sigma_clip = SigmaClip(sigma=sclip) sigma_clip = SigmaClip(sigma=sclip)
bkg_estimator = MedianBackground() bkg_estimator = MedianBackground()
bkg = Background2D(data, (bg_wsize, bg_wsize), sigma_clip=sigma_clip, bkg_estimator=bkg_estimator) bkg = Background2D(data, (bg_wsize, bg_wsize), sigma_clip=sigma_clip, bkg_estimator=bkg_estimator)
...@@ -83,7 +89,7 @@ def subtract_background(bg_wsize=50, sclip=3.0, plot=False): ...@@ -83,7 +89,7 @@ def subtract_background(bg_wsize=50, sclip=3.0, plot=False):
plt.imshow(bkg.background, origin='lower', cmap='Greys_r') plt.imshow(bkg.background, origin='lower', cmap='Greys_r')
plt.show() plt.show()
error = calc_total_error(data, bkg.background_rms, eff_gain) error = calc_total_error(data, bkg.background_rms, eff_gain)
data = data - bkg.background data_sub = data - bkg.background
bg_sub = True bg_sub = True
print('Background successfully subtracted') print('Background successfully subtracted')
...@@ -95,10 +101,10 @@ def find_center(x, y, mod_fit_size=10, plot=True, contour=True): ...@@ -95,10 +101,10 @@ def find_center(x, y, mod_fit_size=10, plot=True, contour=True):
y_min = y - mod_fit_size y_min = y - mod_fit_size
y_max = y + mod_fit_size y_max = y + mod_fit_size
window = data[y_min:y_max+1, x_min:x_max+1] window = data_sub[y_min:y_max+1, x_min:x_max+1]
# Initial guess # Initial guess
z0 = data[y, x] z0 = data_sub[y, x]
m_init = models.Moffat2D(z0, x, y) m_init = models.Moffat2D(z0, x, y)
manual_pick = False manual_pick = False
...@@ -161,7 +167,7 @@ def compute_photometry(x, y, aperture_r=3.0, sky_in=6.0, sky_out=8.0): ...@@ -161,7 +167,7 @@ def compute_photometry(x, y, aperture_r=3.0, sky_in=6.0, sky_out=8.0):
print(' Performing photometry measures without error model') print(' Performing photometry measures without error model')
phot_table = aperture_photometry(data, apers) phot_table = aperture_photometry(data, apers)
else: else:
phot_table = aperture_photometry(data, apers, error) phot_table = aperture_photometry(data_sub, apers, error)
# Mean sky subtraction # Mean sky subtraction
bkg_mean = phot_table['aperture_sum_1'] / annulus_apertures.area() bkg_mean = phot_table['aperture_sum_1'] / annulus_apertures.area()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment