DiadFit 1.0.0__py3-none-any.whl → 1.0.1__py3-none-any.whl

DiadFit/CO2_in_bubble_error.py

@@ -159,7 +159,7 @@ plot_figure=True, fig_i=0, neg_values=True):
 
         # This is the function doing the work to actually make the simulations for each variable.
         if error_vol_perc_bub is not None:
-            print('didnt get inside else loop')
+
 
             df_synthetic=propagate_CO2_in_bubble_ind(
     N_dup=N_dup,

DiadFit/H2O_fitting.py

@@ -10,7 +10,7 @@ from dataclasses import dataclass
 from typing import Tuple, Optional
 from DiadFit.importing_data_files import *
 from numpy import trapz
-from scipy.integrate import simps
+from scipy.integrate import simpson
 ##
 def extract_xstal_MI_name(*, files, char_xstal, pos_xstal, char_MI, pos_MI,
                          prefix=True, str_prefix=" ", file_ext='.txt'):
@@ -872,30 +872,30 @@ fit_sil='poly', dpi=200):
     ydat_sil=y_corr_sil
 
     xspace_sil=xdat_sil[1]-xdat_sil[0]
-    area_trap = trapz(y_corr_sil, dx=xspace_sil)
-    area_simps = simps(y_corr_sil, dx=xspace_sil)
+    area_trap = trapezoid(y_corr_sil, dx=xspace_sil)
+    area_simps = simpson(y_corr_sil, dx=xspace_sil)
     # Just the LW area
     xsil_LW=xdat_sil[(xdat_sil>LW[0]) & (xdat_sil<LW[1])]
     y_corr_sil_LW=y_corr_sil[(xdat_sil>LW[0]) & (xdat_sil<LW[1])]
     xspace_sil_LW=xsil_LW[1]-xsil_LW[0]
-    area_trap_LW=trapz(y_corr_sil_LW, dx=xspace_sil_LW)
-    area_simp_LW=simps(y_corr_sil_LW, dx=xspace_sil_LW)
+    area_trap_LW=trapezoid(y_corr_sil_LW, dx=xspace_sil_LW)
+    area_simp_LW=simpson(y_corr_sil_LW, dx=xspace_sil_LW)
 
 
     # Just the HW area
     xsil_HW=xdat_sil[(xdat_sil>HW[0]) & (xdat_sil<HW[1])]
     y_corr_sil_HW=y_corr_sil[(xdat_sil>HW[0]) & (xdat_sil<HW[1])]
     xspace_sil_HW=xsil_HW[1]-xsil_HW[0]
-    area_trap_HW=trapz(y_corr_sil_HW, dx=xspace_sil_HW)
-    area_simp_HW=simps(y_corr_sil_HW, dx=xspace_sil_HW)
+    area_trap_HW=trapezoid(y_corr_sil_HW, dx=xspace_sil_HW)
+    area_simp_HW=simpson(y_corr_sil_HW, dx=xspace_sil_HW)
 
     # MW
     if MW is not None:
         xsil_MW=xdat_sil[(xdat_sil>MW[0]) & (xdat_sil<MW[1])]
         y_corr_sil_MW=y_corr_sil[(xdat_sil>MW[0]) & (xdat_sil<MW[1])]
         xspace_sil_MW=xsil_MW[1]-xsil_MW[0]
-        area_trap_MW=trapz(y_corr_sil_MW, dx=xspace_sil_MW)
-        area_simp_MW=simps(y_corr_sil_MW, dx=xspace_sil_MW)
+        area_trap_MW=trapezoid(y_corr_sil_MW, dx=xspace_sil_MW)
+        area_simp_MW=simpson(y_corr_sil_MW, dx=xspace_sil_MW)
 
 
      # Plotting what its doing
@@ -1155,8 +1155,8 @@ def fit_area_for_water_region(*, path, filename, Spectra=None, config1: water_bc
 
 
     xspace_water=xdat_water[1]-xdat_water[0]
-    area_trap = trapz(y_corr_water, dx=xspace_water)
-    area_simps = simps(y_corr_water, dx=xspace_water)
+    area_trap = trapezoid(y_corr_water, dx=xspace_water)
+    area_simps = simpson(y_corr_water, dx=xspace_water)
 
 
      # Plotting what its doing

DiadFit/_version.py

@@ -5,4 +5,4 @@
 # 1) we don't load dependencies by storing it in __init__.py
 # 2) we can import it in setup.py for the same reason
 # 3) we can import it into your module
-__version__ = '1.0.0'
+__version__ = '1.0.1'

DiadFit/densimeters.py

@@ -944,20 +944,29 @@ def merge_fit_files(path):
 
     if os.path.exists(os.path.join(path, 'Weak_Diads.xlsx')):
         grp1 = pd.read_excel(os.path.join(path, 'Weak_Diads.xlsx'))
+        grp1['Standard']='No'
     else:
         grp1 = None
 
     if os.path.exists(os.path.join(path, 'Medium_Diads.xlsx')):
         grp2 = pd.read_excel(os.path.join(path, 'Medium_Diads.xlsx'))
+        grp2['Standard']='No'
     else:
         grp2 = None
 
     if os.path.exists(os.path.join(path, 'Strong_Diads.xlsx')):
         grp3 = pd.read_excel(os.path.join(path, 'Strong_Diads.xlsx'))
+        grp3['Standard']='No'
     else:
         grp3 = None
+        
+    if os.path.exists(os.path.join(path, 'Std_Diads.xlsx')):
+        grp4 = pd.read_excel(os.path.join(path, 'Std_Diads.xlsx'))
+        grp4['Standard']='Yes'
+    else:
+        grp4 = None
 
-    df2 = pd.concat([grp1, grp2, grp3], axis=0).reset_index(drop=True)
+    df2 = pd.concat([grp1, grp2, grp3, grp4], axis=0).reset_index(drop=True)
 
     if discard is not None:
         discard_cols=discard[discard.columns.intersection(df2.columns)]

DiadFit/diads.py

@@ -18,7 +18,7 @@ import matplotlib.patches as patches
 import warnings as w
 from tqdm import tqdm
 from numpy import trapz
-from scipy.integrate import simps
+from scipy.integrate import simpson
 from scipy.interpolate import interp1d
 
 # Allowed models
@@ -725,7 +725,7 @@ def plot_peak_params(fit_params,
 def filter_splitting_prominence(*, fit_params, data_y_all,
                                 x_cord,
                                 splitting_limits=[100, 107],
-                                lower_diad1_prom=10, exclude_str):
+                                lower_diad1_prom=10, exclude_str=None, str_filt=None):
     """ Filters Spectra based on approximate splitting, draws a plot showing spectra to discard and those to keep
 
     Parameters
@@ -746,6 +746,12 @@ def filter_splitting_prominence(*, fit_params, data_y_all,
         Only keeps spectra that meet the splitting parameter, and have an absolute
         diad1 prominence greater than this value (helps filter out other weird spectra)
 
+    exclude_str: str
+        Excludes files with this string.
+
+    str_filt: str
+        Filters just based on string in filename
+
     Returns
     --------------
     fit_params_filt: pd.DataFrame
@@ -759,21 +765,30 @@ def filter_splitting_prominence(*, fit_params, data_y_all,
 
 
     """
+    if str_filt is not None:
+        filt=fit_params['filename'].str.contains(str_filt)
 
-    reas_split=(fit_params['approx_split'].between(splitting_limits[0], splitting_limits[1]))
-    reas_heigh=fit_params['Diad1_abs_prom']>lower_diad1_prom
-    if exclude_str is not None:
-        name_in_file=~fit_params['filename'].str.contains(exclude_str)
     else:
-        name_in_file=reas_heigh
 
-    fit_params_filt=fit_params.loc[(reas_split&reas_heigh&name_in_file)].reset_index(drop=True)
-    fit_params_disc=fit_params.loc[~(reas_split&reas_heigh&name_in_file)].reset_index(drop=True)
+        reas_split=(fit_params['approx_split'].between(splitting_limits[0], splitting_limits[1]))
+        reas_heigh=fit_params['Diad1_abs_prom']>lower_diad1_prom
+
+        if exclude_str is not None:
+            name_in_file=~fit_params['filename'].str.contains(exclude_str)
+        else:
+            name_in_file=reas_heigh
+
+        filt=reas_split&reas_heigh&name_in_file
+
+    fit_params_filt=fit_params.loc[filt].reset_index(drop=True)
+    fit_params_disc=fit_params.loc[~(filt)].reset_index(drop=True)
 
     print('Keeping N='+str(len(fit_params_filt)))
     print('Discarding N='+str(len(fit_params_disc)))
 
-    filt=reas_split&reas_heigh&name_in_file
+
+
+    # Then apply to get data
     data_y_filt=data_y_all[:, (filt)]
     data_y_disc=data_y_all[:, ~(filt)]
 
@@ -804,17 +819,21 @@ def filter_splitting_prominence(*, fit_params, data_y_all,
             Diff=np.nanmax(data_y_filt[:, i])-np.nanmin(data_y_filt[:, i])
             av_prom_Keep=fit_params_filt['Diad1_abs_prom'].iloc[i]
             prom_filt=prom_filt+av_prom_Keep
+            file=fit_params_filt['filename'].iloc[i]
             ax2.plot(x_cord+i*5, (data_y_filt[:, i]-np.nanmin(data_y_filt[:, i]))/Diff+i/3, '-b', lw=0.5)
+            yplot=np.quantile((data_y_filt[:, i]-np.nanmin(data_y_filt[:, i]))/Diff+i/3, 0.65)
+            ax2.annotate(str(file), xy=(1450+i*5, yplot),
+                    xycoords="data", fontsize=8, bbox=dict(facecolor='white', edgecolor='none', pad=2))
 
 
         ax2.set_xlim([1250, 1450+i*5])
         ax2.set_xticks([])
         ax2.set_yticks([])
 
-    return fit_params_filt, data_y_filt, fit_params_disc, data_y_disc
+    return fit_params_filt.reset_index(drop=True), data_y_filt, fit_params_disc.reset_index(drop=True), data_y_disc
 
 
-def identify_diad_group(*, fit_params, data_y,  x_cord, filter_bool,y_fig_scale=0.1, grp_filter='Weak'):
+def identify_diad_group(*, fit_params, data_y,  x_cord, filter_bool,y_fig_scale=0.1, grp_filter='Weak', str_filt=None):
 
     """Sorts diads into two groups. Those meeting the 'filter_bool' criteria, and those not
      meeting this criteria. Ones meeting the criteria are shown on the left hand plot,
@@ -853,6 +872,9 @@ def identify_diad_group(*, fit_params, data_y,  x_cord, filter_bool,y_fig_scale=
 
 
     """
+    if str_filt is not None:
+        filt_name=~fit_params['filename'].str.contains(str_filt)
+        filt_bool=filt_name&filt_bool
 
     if np.shape(data_y)[1]==0:
         Group1_df=pd.DataFrame().reindex_like(fit_params)
@@ -3835,8 +3857,8 @@ path=None, filename=None, filetype=None,
 
 
                 xspace_sil=x_new[1]-x_new[0]
-                area_trap = trapz(Baseline_ysub_sil, dx=xspace_sil)
-                area_simps = simps(Baseline_ysub_sil, dx=xspace_sil)
+                area_trap = trapezoid(Baseline_ysub_sil, dx=xspace_sil)
+                area_simps = simpson(Baseline_ysub_sil, dx=xspace_sil)
 
 
 

{DiadFit-1.0.0.dist-info → DiadFit-1.0.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: DiadFit
-Version: 1.0.0
+Version: 1.0.1
 Summary: DiadFit
 Home-page: https://github.com/PennyWieser/DiadFit
 Author: Penny Wieser
@@ -9,13 +9,13 @@ License: UNKNOWN
 Platform: UNKNOWN
 Classifier: Programming Language :: Python :: 3
 Classifier: Operating System :: OS Independent
-Requires-Python: >=3.7
+Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 Requires-Dist: pandas
 Requires-Dist: numpy <2
 Requires-Dist: matplotlib
 Requires-Dist: scikit-learn
-Requires-Dist: scipy
+Requires-Dist: scipy >1.6
 Requires-Dist: lmfit >=1.1.0
 Requires-Dist: tqdm
 Requires-Dist: python-docx

{DiadFit-1.0.0.dist-info → DiadFit-1.0.1.dist-info}/RECORD

@@ -1,6 +1,6 @@
 DiadFit/CO2_EOS.py,sha256=ti9HEZynM-BoiPGYG1C7vUpO-RNZY-NqmjUl2FFnO3E,73918
-DiadFit/CO2_in_bubble_error.py,sha256=ehr1JZCjwVLD24A3tl2RW-UsdTTJV39zHs49JD9ZpLg,21271
-DiadFit/H2O_fitting.py,sha256=E3xD9VkYNDdmsQ3wZNkMOHhPkNO0F-al_Vb9Q8ZR6oI,43888
+DiadFit/CO2_in_bubble_error.py,sha256=Nq5YEf2oa2rWRreEPXl2lEA86NXOGvll0Gca2AOu_RE,21224
+DiadFit/H2O_fitting.py,sha256=ZOLWL8j7HQYKlx1-ISm1twvH34jhrGFwukU8ElRj0Dw,43920
 DiadFit/Highrho_polyfit_data.pkl,sha256=7t6uXxI-HdfsvreAWORzMa9dXxUsnXqKBSo1O3EgiBw,1213
 DiadFit/Highrho_polyfit_dataUCB_1117_1400.pkl,sha256=oBOarETLyfq2DJhYGQrJofgHjvRMLamE6G2b7EE5m-Y,1213
 DiadFit/Highrho_polyfit_dataUCB_1117_1447.pkl,sha256=OG1qip_xU1hl3xp3HC8e9_2497-KYEV3Xz3mx0gdJ4Y,1213
@@ -29,13 +29,13 @@ DiadFit/Mediumrho_polyfit_data_CCMR.pkl,sha256=U6ODSdurqS0-lynm1MG1zktg8NuhYRbrY
 DiadFit/Mediumrho_polyfit_data_CMASS.pkl,sha256=SBy1pIdqCAF9UtB9FLNTuD0-tFyD7swwJppdE2U_FsY,1557
 DiadFit/Psensor.py,sha256=C2xSlgxhUJIKIBDvUp02QaYRs5QsIqjGGRMP25ZLRZ0,10435
 DiadFit/__init__.py,sha256=F-HjhCYKL_U8PfiH8tZ9DUCkxPvo6lAslJS4fyvxkbY,1148
-DiadFit/_version.py,sha256=v3boCh9grt_Ozt8KUuSV2NTg8ZSYLiJfte4wgxefVbk,295
+DiadFit/_version.py,sha256=e_2FV9coO7Ff17u_XZSeherC0HKwQWI8n-EaEUoHEuU,295
 DiadFit/argon_lines.py,sha256=vtzsuDdEgrAmEF9xwpejpFqKV9hKPS1JUYhIl4AfXZ0,7675
 DiadFit/cosmicray_filter.py,sha256=a45x2_nmpi9Qcjc_L39UA9JOd1NMorIjtTRGnCdG3MU,23634
 DiadFit/densimeter_fitting.py,sha256=zEyCwq1zDV3z6-MIu-eZqgp3YQPUGqwZiKczN3-22LQ,8247
-DiadFit/densimeters.py,sha256=0_Kh9KxyCnGp8cep3zif3eIQIlMLpPbQj7eF9JHiGWk,54956
+DiadFit/densimeters.py,sha256=p3jY9709vKegCfWZIwoU4Rt5jFfwJJQobLb71AXUxAY,55250
 DiadFit/density_depth_crustal_profiles.py,sha256=b072IJaoGDydKpqWWKoJHeXKIkcIXxKf82whpvLAPpw,17761
-DiadFit/diads.py,sha256=zGzK_TM5iFL7_9BFnQR1Rpncr3HNmq8RBhBbLoKq_7I,176663
+DiadFit/diads.py,sha256=PmCIxMXU6O5zFQWzQmhwkqEVoRDyw89ogKTYjIAvvMg,177423
 DiadFit/error_propagation.py,sha256=STJUqjdhwo0geXi0iZ9XN0RD3t7qYqbNGHzelfLOANQ,50038
 DiadFit/importing_data_files.py,sha256=0Cx_CKJZR8efssMzQit0aPRh_rsjQFGXgLtI285FW_k,41961
 DiadFit/lookup_table.csv,sha256=Hs1tmSQ9ArTUDv3ymEXbvnLlPBxYUP0P51dz7xAKk-Q,2946857
@@ -43,7 +43,7 @@ DiadFit/lookup_table_noneg.csv,sha256=HelvewKbBy4cqT2GAqsMo-1ps1lBYqZ-8hCJZWPGfh
 DiadFit/molar_gas_proportions.py,sha256=_oEZn_vndHGDaXAjZ6UU8ycujBx_qB2KGCGqZSzotQU,3389
 DiadFit/ne_lines.py,sha256=6z9oo4lgh0iYv1mkSscgzCt_Pe4gQTnquG99pR6cJS8,63811
 DiadFit/relaxifi.py,sha256=hHzRsJPQIVohYi3liy9IQJpaomgsa2zbLQmhqkpdfrI,31549
-DiadFit-1.0.0.dist-info/METADATA,sha256=PVcVAd7UDrIeBqratoKKVBe1nCoXv_M3-1jKSrQTFc4,1169
-DiadFit-1.0.0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-DiadFit-1.0.0.dist-info/top_level.txt,sha256=yZC6OFLVznaFA5kcPlFPkvhKotcVd-YO4bKxZZw3LQE,8
-DiadFit-1.0.0.dist-info/RECORD,,
+DiadFit-1.0.1.dist-info/METADATA,sha256=JVJSzMb7lAJj2u6IRmjuZRfn5i6um5XOl-jj3ISQU58,1174
+DiadFit-1.0.1.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+DiadFit-1.0.1.dist-info/top_level.txt,sha256=yZC6OFLVznaFA5kcPlFPkvhKotcVd-YO4bKxZZw3LQE,8
+DiadFit-1.0.1.dist-info/RECORD,,