ACID-code-v2 0.2.1__tar.gz → 0.2.2a2__tar.gz

{acid_code_v2-0.2.1 → acid_code_v2-0.2.2a2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ACID_code_v2
-Version: 0.2.1
+Version: 0.2.2a2
 Summary: Returns line profiles from input spectra by fitting the stellar continuum and performing LSD
 Author: Lucy Dolan
 Author-email: Benjamin Cadell <bcadell01@qub.ac.uk>

{acid_code_v2-0.2.1 → acid_code_v2-0.2.2a2}/example/tutorial_code.py

@@ -5,6 +5,7 @@ import multiprocessing as mp
 import numpy as np
 import matplotlib.pyplot as plt
 try:
+    raise Exception("Force except block")
     import ACID_code_v2 as acid
 except:
     os.chdir(os.path.dirname(__file__))

{acid_code_v2-0.2.1 → acid_code_v2-0.2.2a2}/src/ACID_code_v2/ACID.py

@@ -201,38 +201,6 @@ def combine_spec(wavelengths_f, spectra_f, errors_f, sns_f):
     
     return reference_wave, spectrum_f, spec_errors_f, sn_f
 
-def model_func(inputs, x):
-    ## model for the mcmc - takes the profile(z) and the continuum coefficents(inputs[k_max:]) to create a model spectrum.
-    z = inputs[:k_max]
-
-    mdl = np.dot(alpha, z) ##alpha has been declared a global variable after LSD is run.
-
-    #converting model from optical depth to flux
-    mdl = np.exp(mdl)
-
-    ## these are used to adjust the wavelengths to between -1 and 1 - makes the continuum coefficents smaller and easier for emcee to handle.
-    a = 2/(np.max(x)-np.min(x))
-    b = 1 - a*np.max(x)
-
-    mdl1 = 0
-    for i in range(k_max, len(inputs) - 1):
-        mdl1 = mdl1 + (inputs[i] * ((x * a) + b) ** (i - k_max))
-
-    # coefs = np.asarray(inputs[k_max:-1], dtype=float) # Potential improvement - Ben
-    # X = (a * x) + b
-    # if coefs.size:
-    #     powers = np.arange(coefs.size)
-    #     # X[:, None] ** powers -> shape (len(x), coefs.size); dot with coefs -> (len(x),)
-    #     mdl1 = np.dot(X[:, None] ** powers[None, :], coefs)
-    # else:
-    #     mdl1 = 0.0
-
-    mdl1 = mdl1 * inputs[-1]
-    
-    mdl = mdl * mdl1
-   
-    return mdl
-
 def residual_mask(wavelengths, data_spec_in, data_err, initial_inputs, poly_ord, linelist,
                   velocities=np.arange(-25, 25, 0.82), pix_chunk=20, dev_perc=25, 
                   tell_lines=[3820.33, 3933.66, 3968.47, 4327.74, 4307.90, 4383.55,
@@ -240,7 +208,7 @@ def residual_mask(wavelengths, data_spec_in, data_err, initial_inputs, poly_ord,
                               7593.70, 8226.96], n_sig=1):
     ## iterative residual masking - mask continuous areas first - then possibly progress to masking the narrow lines
 
-    forward = model_func(initial_inputs, wavelengths)
+    forward = mcmc_utils.model_func(initial_inputs, wavelengths, k_max=k_max, alpha=alpha)
 
     a = 2 / (np.max(wavelengths) - np.min(wavelengths))
     b = 1 - a * np.max(wavelengths)
@@ -593,8 +561,8 @@ def ACID(input_wavelengths, input_spectra, input_spectral_errors, line, frame_sn
     pos = np.array(pos)
     pos = np.transpose(pos)
 
-    # Inistialise MCMC class:
-    mcmc_class = mcmc_utils.Model(model_func, x, y, yerr, velocities, k_max)
+    # Initialise MCMC class:
+    mcmc_class = mcmc_utils.Model(x, y, yerr, velocities, k_max, alpha)
 
     if verbose:
         t5 = time.time()
@@ -675,7 +643,7 @@ def ACID(input_wavelengths, input_spectra, input_spectral_errors, line, frame_sn
         inds = np.random.randint(len(flat_samples), size=100)
         for ind in inds:
             sample = flat_samples[ind]
-            mdl = model_func(sample, x)
+            mdl = mcmc_utils.model_func(sample, x, k_max=k_max, alpha=alpha)
             mdl1 = 0
             for i in np.arange(k_max, len(sample)-1):
                 mdl1 = mdl1+sample[i]*((a*x)+b)**(i-k_max)
@@ -729,7 +697,7 @@ def ACID(input_wavelengths, input_spectra, input_spectral_errors, line, frame_sn
         plt.savefig('figures/cont_%s'%(run_name))
     
         mcmc_inputs = np.concatenate((profile, poly_cos))
-        mcmc_mdl = model_func(mcmc_inputs, x)
+        mcmc_mdl = mcmc_utils.model_func(mcmc_inputs, x, k_max=k_max, alpha=alpha)
 
         residuals_2 = (y+1) - (mcmc_mdl+1)
 
@@ -773,7 +741,7 @@ def ACID(input_wavelengths, input_spectra, input_spectral_errors, line, frame_sn
     conts = []
     for ind in inds:
         sample = flat_samples[ind]
-        mdl = model_func(sample, wavelengths)
+        mdl = mcmc_utils.model_func(sample, wavelengths, k_max=k_max, alpha=alpha)
         #mdl = model_func(sample, x)
         #mdl = mdl[idx]
         mdl1_temp = 0

{acid_code_v2-0.2.1 → acid_code_v2-0.2.2a2}/src/ACID_code_v2/mcmc_utils.py

@@ -1,19 +1,49 @@
 import numpy as np
-from ACID_code_v2.ACID import model_func
 
+def model_func(inputs, x, k_max, alpha):
+    ## model for the mcmc - takes the profile(z) and the continuum coefficents(inputs[k_max:]) to create a model spectrum.
+    z = inputs[:k_max]
+
+    mdl = np.dot(alpha, z) ##alpha has been declared a global variable after LSD is run.
+
+    #converting model from optical depth to flux
+    mdl = np.exp(mdl)
+
+    ## these are used to adjust the wavelengths to between -1 and 1 - makes the continuum coefficents smaller and easier for emcee to handle.
+    a = 2/(np.max(x)-np.min(x))
+    b = 1 - a*np.max(x)
+
+    mdl1 = 0
+    for i in range(k_max, len(inputs) - 1):
+        mdl1 = mdl1 + (inputs[i] * ((x * a) + b) ** (i - k_max))
+
+    # coefs = np.asarray(inputs[k_max:-1], dtype=float) # Potential improvement - Ben
+    # X = (a * x) + b
+    # if coefs.size:
+    #     powers = np.arange(coefs.size)
+    #     # X[:, None] ** powers -> shape (len(x), coefs.size); dot with coefs -> (len(x),)
+    #     mdl1 = np.dot(X[:, None] ** powers[None, :], coefs)
+    # else:
+    #     mdl1 = 0.0
+
+    mdl1 = mdl1 * inputs[-1]
+    
+    mdl = mdl * mdl1
+   
+    return mdl
 
 class Model:
-    def __init__(self, model_func, x, y, yerr, velocities, k_max):
-        self.model_func = model_func
+    def __init__(self, x, y, yerr, velocities, k_max, alpha):
         self.x = x
         self.y = y
         self.yerr = yerr
         self.velocities = velocities
         self.k_max = k_max
+        self.alpha = alpha
 
     def log_likelihood(self, theta, x, y, yerr):
         ## maximum likelihood estimation for the mcmc model.
-        model = model_func(theta, x)
+        model = model_func(theta, x, k_max=self.k_max, alpha=self.alpha)
 
         lnlike = -0.5 * np.sum(((y) - (model)) ** 2 / yerr**2 + np.log(yerr**2)+ np.log(2*np.pi))
 

{acid_code_v2-0.2.1 → acid_code_v2-0.2.2a2}/src/ACID_code_v2.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ACID_code_v2
-Version: 0.2.1
+Version: 0.2.2a2
 Summary: Returns line profiles from input spectra by fitting the stellar continuum and performing LSD
 Author: Lucy Dolan
 Author-email: Benjamin Cadell <bcadell01@qub.ac.uk>