AeroViz 0.1.3__py3-none-any.whl → 0.1.4__py3-none-any.whl

AeroViz/dataProcess/Chemistry/_teom.py

@@ -1,16 +1,14 @@
-import numpy as np
-
-
 def _basic(_teom, _check):
-	_teom['Volatile_Fraction'] = (_teom['PM_Total'] - _teom['PM_NV']) / _teom['PM_Total']
+    import numpy as np
+    _teom['Volatile_Fraction'] = (_teom['PM_Total'] - _teom['PM_NV']) / _teom['PM_Total']
 
-	_teom.loc[(_teom['Volatile_Fraction'] < 0) | (_teom['Volatile_Fraction'] > 1)] = n.nan
+    _teom.loc[(_teom['Volatile_Fraction'] < 0) | (_teom['Volatile_Fraction'] > 1)] = np.nan
 
-	if _check is not None:
-		_ratio = _teom['PM_NV'] / _check
-		_teom['PM_Check'] = _check
+    if _check is not None:
+        _ratio = _teom['PM_NV'] / _check
+        _teom['PM_Check'] = _check
 
-		_teom.loc[_teom.dropna().index, 'status'] = 'Warning'
-		_teom.loc[(_ratio > 0) & (_ratio < 1), 'status'] = 'Normal'
+        _teom.loc[_teom.dropna().index, 'status'] = 'Warning'
+        _teom.loc[(_ratio > 0) & (_ratio < 1), 'status'] = 'Normal'
 
-	return _teom
+    return _teom

AeroViz/dataProcess/Chemistry/isrpia.cnf

@@ -0,0 +1,21 @@
+*** [ INPUT/OUTPUT PARAMETERS ] ***
+CREATE REPORT IN FILE? (.T.=yes, .F.=no, on screen)
+.F.
+CREATE SPREADSHEET FILE? (.T.=yes, .F.=no)
+.T.
+
+*** [ SOLUTION CONTROL ] ***
+Convergence criterion (DEFAULT:1.D-6)
+1.D-6 
+Max iterations for bisection method (DEFAULT:100)
+100
+Sweeps for activity coef. calculation (DEFAULT:4)
+4
+Accuracy in activity coef. calculation (DEFAULT:5e-2)
+5e-2
+Subdivisions for root tracking (DEFAULT:5)
+5
+Method for binary activity coefs (0-online, 1-tables) (DEFAULT:1)
+1
+Force mass conservation to machine precision? (0-no, 1-yes) (DEFAULT:0)
+0

AeroViz/dataProcess/Optical/Angstrom_exponent.py

@@ -0,0 +1,20 @@
+import numpy as np
+import pandas as pd
+from scipy.optimize import curve_fit
+
+
+def get_species_wavelength(df, specified_band):
+    func = lambda wavelength, _sl, _int: _sl * wavelength + _int
+    popt, pcov = curve_fit(func, specified_band, df.values)
+
+    return func(np.array(specified_band), *popt)
+
+
+def get_Angstrom_exponent(df, band):
+    if (df <= 0).any():
+        return pd.Series([np.nan, np.nan], index=['slope', 'intercept'])  # 返回包含 NaN 的 Series，保持 DataFrame 结构
+
+    func = lambda wavelength, _sl, _int: _sl * wavelength + _int
+    popt, _ = curve_fit(func, np.log(band), np.log(df))
+
+    return pd.Series(popt, index=['slope', 'intercept'])  # 返回带有索引的 Series

AeroViz/dataProcess/Optical/_IMPROVE.py

@@ -1,61 +1,60 @@
 import pickle as pkl
 from pathlib import Path
+
 import numpy as np
+from pandas import DataFrame
 
-from pandas import date_range, concat, DataFrame, to_numeric
-from AeroViz.dataProcess.core import _union_index
+from AeroViz.dataProcess.core import union_index
 
 
 def _revised(_df_mass, _df_RH):
-	_df_mass, _df_RH = _union_index(_df_mass, _df_RH)
-
-	## fRH
-	with (Path(__file__).parent / 'fRH.pkl').open('rb') as f:
-		_fRH = pkl.load(f)
-		_fRH.loc[np.nan] = np.nan
+    _df_mass, _df_RH = union_index(_df_mass, _df_RH)
 
-	def fRH(_RH):
-		if _RH is not None:
-			_RH = _RH.mask(_RH > 95, 95).round(0)
-			return _fRH.loc[_RH].values.T
+    # fRH
+    with (Path(__file__).parent / 'fRH.pkl').open('rb') as f:
+        _fRH = pkl.load(f)
+        _fRH.loc[np.nan] = np.nan
 
-		return 1, 1, 1, 1
+    def fRH(_RH):
+        if _RH is not None:
+            _RH = _RH.mask(_RH > 95, 95).round(0)
+            return _fRH.loc[_RH].values.T
 
-	## different mode
-	## mass < 20 :
-	## 				large = mass**2/20
-	## 				small = mass-large
-	## mass >= 20 :
-	## 				large = mass
-	## 				small = 0
-	_df_mode = _df_mass[['AS', 'AN', 'OM']].copy()
+        return 1, 1, 1, 1
 
-	_df_mass[['L_AS', 'L_AN', 'L_OM']] = _df_mode.mask(_df_mode < 20, _df_mode ** 2 / 20)
-	_df_mass[['S_AS', 'S_AN', 'S_OM']] = _df_mode.values - _df_mass[['L_AS', 'L_AN', 'L_OM']]
+    # different mode
+    # mass < 20 :
+    # 				large = mass**2/20
+    # 				small = mass-large
+    # mass >= 20 :
+    # 				large = mass
+    # 				small = 0
+    _df_mode = _df_mass[['AS', 'AN', 'OM']].copy()
 
-	## apply IMPROVE ccoe.
-	def _ext_cal(_RH=None):
+    _df_mass[['L_AS', 'L_AN', 'L_OM']] = _df_mode.mask(_df_mode < 20, _df_mode ** 2 / 20)
+    _df_mass[['S_AS', 'S_AN', 'S_OM']] = _df_mode.values - _df_mass[['L_AS', 'L_AN', 'L_OM']]
 
-		_frh, _frhss, _frhs, _frhl = fRH(_RH)
-		_df = DataFrame(index=_df_mass.index)
+    # apply IMPROVE ccoe.
+    def _ext_cal(_RH=None):
 
-		_df['AS'] = 2.2 * _frhs * _df_mass['S_AS'] + 4.8 * _frhl * _df_mass['L_AS']
-		_df['AN'] = 2.4 * _frhs * _df_mass['S_AN'] + 5.1 * _frhl * _df_mass['L_AN']
-		_df['OM'] = 2.8 * _df_mass['S_OM'] + 6.1 * _frhl * _df_mass['L_OM']
-		_df['Soil'] = _df_mass['Soil']
-		_df['SS'] = 1.7 * _frhss * _df_mass['SS']
-		_df['EC'] = 10 * _df_mass['EC']
+        _frh, _frhss, _frhs, _frhl = fRH(_RH)
+        _df = DataFrame(index=_df_mass.index)
 
-		_df['total'] = _df.sum(axis=1)
+        _df['AS'] = 2.2 * _frhs * _df_mass['S_AS'] + 4.8 * _frhl * _df_mass['L_AS']
+        _df['AN'] = 2.4 * _frhs * _df_mass['S_AN'] + 5.1 * _frhl * _df_mass['L_AN']
+        _df['OM'] = 2.8 * _df_mass['S_OM'] + 6.1 * _frhl * _df_mass['L_OM']
+        _df['Soil'] = _df_mass['Soil']
+        _df['SS'] = 1.7 * _frhss * _df_mass['SS']
+        _df['EC'] = 10 * _df_mass['EC']
 
-		return _df.dropna().reindex(_df_mass.index)
+        _df['total'] = _df.sum(axis=1)
 
-	## calculate
-	_out = {}
+        return _df.dropna().reindex(_df_mass.index)
 
-	_out['dry'] = _ext_cal()
+    # calculate
+    _out = {'dry': _ext_cal()}
 
-	if _df_RH is not None:
-		_out['wet'] = _ext_cal(_df_RH)
+    if _df_RH is not None:
+        _out['wet'] = _ext_cal(_df_RH)
 
-	return _out
+    return _out

AeroViz/dataProcess/Optical/__init__.py

@@ -1,62 +1,47 @@
-from ..core import _writter, _run_process
+from ..core import Writer, run_process
 
-__all__ = [
+__all__ = ['Optical']
 
-	'Optical',
 
-]
+class Optical(Writer):
 
+    @run_process('Optical - scaCoe', 'scaCoe')
+    def scaCoe(self, df_sca, instru, specified_band):
+        from ._scattering import _scaCoe
 
-class Optical(_writter):
+        out = _scaCoe(df_sca, instru=instru, specified_band=[550] if specified_band is None else specified_band)
 
-	## scatter
-	@_run_process('Optical - SAE', 'SAE')
-	def SAE(self, df_sca):
-		from ._scattering import _SAE
+        return self, out
 
-		out = _SAE(df_sca)
+    @run_process('Optical - absCoe', 'absCoe')
+    def absCoe(self, df_ae33, instru, specified_band):
+        from ._absorption import _absCoe
 
-		return self, out
+        out = _absCoe(df_ae33, instru=instru, specified_band=[550] if specified_band is None else specified_band)
 
-	## absorption
-	@_run_process('Optical - absCoe', 'absCoe')
-	def absCoe(self, df_ae33, abs_band=[550]):
-		from ._absorption import _absCoe
+        return self, out
 
-		out = _absCoe(df_ae33, abs_band)
+    @run_process('Optical - basic', 'opt_basic')
+    def basic(self, df_sca, df_abs, df_mass=None, df_no2=None, df_temp=None):
+        from ._extinction import _basic
 
-		return self, out
+        out = _basic(df_sca, df_abs, df_mass, df_no2, df_temp)
 
-	@_run_process('Optical - AAE', 'AAE')
-	def AAE(self, df_abs):
-		from ._absorption import _AAE
+        return self, out
 
-		out = _AAE(df_abs)
+    @run_process('Optical - Mie', 'Mie')
+    def Mie(self, df_psd, df_m, wave_length=550):
+        from ._mie import _mie
 
-		return self, out
+        out = _mie(df_psd, df_m, wave_length)
 
-	## extinction
-	@_run_process('Optical - basic', 'opt_basic')
-	def basic(self, df_abs, df_sca, df_ec=None, df_mass=None, df_no2=None):
-		from ._extinction import _basic
+        return self, out
 
-		out = _basic(df_abs, df_sca, df_ec, df_mass, df_no2)
+    @run_process('Optical - IMPROVE', 'IMPROVE')
+    def IMPROVE(self, df_mass, df_RH, method='revised'):
+        # _fc = __import__(f'_IMPROVE._{method}')
+        from ._IMPROVE import _revised
 
-		return self, out
+        out = _revised(df_mass, df_RH)
 
-	@_run_process('Optical - Mie', 'Mie')
-	def Mie(self, df_psd, df_m, wave_length=550):
-		from ._mie import _mie
-
-		out = _mie(df_psd, df_m, wave_length)
-
-		return self, out
-
-	@_run_process('Optical - IMPROVE', 'IMPROVE')
-	def IMPROVE(self, df_mass, df_RH, method='revised'):
-		# _fc = __import__(f'_IMPROVE._{method}')
-		from ._IMPROVE import _revised
-
-		out = _revised(df_mass, df_RH)
-
-		return self, out
+        return self, out

AeroViz/dataProcess/Optical/_absorption.py

@@ -1,54 +1,28 @@
-def _absCoe(df, abs_band):
-	import numpy as n
-	from scipy.optimize import curve_fit
+def _absCoe(df, instru, specified_band: list):
+    import numpy as np
+    from pandas import concat
+    from .Angstrom_exponent import get_Angstrom_exponent, get_species_wavelength
 
-	band = n.array([370, 470, 520, 590, 660, 880, 950])
+    band_AE33 = np.array([370, 470, 520, 590, 660, 880, 950])
+    band_BC1054 = np.array([370, 430, 470, 525, 565, 590, 660, 700, 880, 950])
 
-	df_out = {}
+    MAE_AE33 = np.array([18.47, 14.54, 13.14, 11.58, 10.35, 7.77, 7.19]) * 1e-3
+    MAE_BC1054 = np.array([18.48, 15.90, 14.55, 13.02, 12.10, 11.59, 10.36, 9.77, 7.77, 7.20]) * 1e-3
 
-	def _get_slope(__df):
-		func = lambda _x, _sl, _int: _sl * _x + _int
-		popt, pcov = curve_fit(func, band, __df.values)
+    band = band_AE33 if instru == 'AE33' else band_BC1054
+    MAE = MAE_AE33 if instru == 'AE33' else MAE_BC1054
+    eBC = 'BC6' if instru == 'AE33' else 'BC9'
 
-		return func(n.array(abs_band), *popt)
+    # calculate
+    df_abs = (df.copy().dropna() * MAE).copy()
 
-	MAE = n.array([18.47, 14.54, 13.14, 11.58, 10.35, 7.77, 7.19]) * 1e-3
-	df_abs = (df.copy() * MAE).dropna().copy()
+    df_out = df_abs.apply(get_species_wavelength, axis=1, result_type='expand', args=(specified_band,))
+    df_out.columns = [f'abs_{_band}' for _band in specified_band]
+    df_out['eBC'] = df[eBC]
 
-	df_out = df_abs.apply(_get_slope, axis=1, result_type='expand').reindex(df.index)
-	df_out.columns = [f'abs_{_band}' for _band in abs_band]
+    df_AAE = df_abs.apply(get_Angstrom_exponent, axis=1, result_type='expand', args=(band,))
+    df_AAE.columns = ['AAE', 'AAE_intercept']
+    df_AAE = df_AAE.mask((-df_AAE['AAE'] < 0.8) | (-df_AAE['AAE'] > 2.)).copy()
 
-	df_out['eBC'] = df['BC6']
-
-	return df_out
-
-
-def _AAE(df):
-	import numpy as n
-	from scipy.optimize import curve_fit
-
-	def _AAEcalc(_df):
-		## parameter
-		MAE = n.array([18.47, 14.54, 13.14, 11.58, 10.35, 7.77, 7.19]) * 1e-3
-		band = n.array([370, 470, 520, 590, 660, 880, 950])
-		_df *= MAE
-
-		## 7 pts fitting
-		## function
-		def _get_slope(__df):
-			func = lambda _x, _sl, _int: _sl * _x + _int
-			popt, pcov = curve_fit(func, n.log(band), n.log(__df))
-
-			return popt
-
-		## calculate
-		_AAE = _df.apply(_get_slope, axis=1, result_type='expand')
-		_AAE.columns = ['slope', 'intercept']
-
-		return _AAE
-
-	df_out = _AAEcalc(df[['BC1', 'BC2', 'BC3', 'BC4', 'BC5', 'BC6', 'BC7']].dropna())
-	df_out = df_out.mask((-df_out.slope < 0.8) | (-df_out.slope > 2.)).copy()
-
-	df_out['eBC'] = df['BC6']
-	return df_out.reindex(df.index)
+    _df = concat([df_out, df_AAE['AAE']], axis=1)
+    return _df.reindex(df.index)

AeroViz/dataProcess/Optical/_extinction.py

@@ -1,36 +1,42 @@
-from AeroViz.dataProcess.core import _union_index
 from pandas import DataFrame
 
+from AeroViz.dataProcess.core import union_index
 
-def _basic(df_abs, df_sca, df_ec, df_mass, df_no2):
-	df_abs, df_sca, df_ec, df_mass, df_no2 = _union_index(df_abs, df_sca, df_ec, df_mass, df_no2)
 
-	df_out = DataFrame()
+def _basic(df_sca, df_abs, df_mass, df_no2, df_temp):
+    df_sca, df_abs, df_mass, df_no2, df_temp = union_index(df_sca, df_abs, df_mass, df_no2, df_temp)
 
-	## abs and sca coe
-	df_out['abs'] = df_abs.copy()
-	df_out['sca'] = df_sca.copy()
+    df_out = DataFrame()
 
-	## extinction coe.
-	df_out['ext'] = df_out['abs'] + df_out['sca']
+    # abs and sca coe
+    df_out['abs'] = df_abs['abs_550'].copy()
+    df_out['sca'] = df_sca['sca_550'].copy()
 
-	## SSA
-	df_out['SSA'] = df_out['sca'] / df_out['ext']
+    # extinction coe.
+    df_out['ext'] = df_out['abs'] + df_out['sca']
 
-	## MAE, MSE, MEE
-	if df_mass is not None:
-		df_out['MAE'] = df_out['abs'] / df_mass
-		df_out['MSE'] = df_out['sca'] / df_mass
-		df_out['MEE'] = df_out['MSE'] + df_out['MAE']
+    # SSA
+    df_out['SSA'] = df_out['sca'] / df_out['ext']
 
-	## gas absorbtion
-	if df_no2 is not None:
-		df_out['abs_gas'] = df_no2 * .33
-		df_out['sca_gas'] = 10
-		df_out['ext_all'] = df_out['ext'] + df_out['abs_gas'] + df_out['sca_gas']
+    # SAE, AAE, eBC
+    df_out['SAE'] = df_sca['SAE'].copy()
+    df_out['AAE'] = df_abs['AAE'].copy()
+    df_out['eBC'] = df_abs['eBC'].copy() / 1e3
 
-	## other
-	if df_ec is not None:
-		df_out['eBC'] = df_ec / 1e3
+    # MAE, MSE, MEE
+    if df_mass is not None:
+        df_out['MAE'] = df_out['abs'] / df_mass
+        df_out['MSE'] = df_out['sca'] / df_mass
+        df_out['MEE'] = df_out['MSE'] + df_out['MAE']
 
-	return df_out
+    # gas absorbtion
+    if df_no2 is not None:
+        df_out['abs_gas'] = df_no2 * .33
+
+    if df_temp is not None:
+        df_out['sca_gas'] = (11.4 * 293 / (273 + df_temp))
+
+    if df_no2 is not None and df_temp is not None:
+        df_out['ext_all'] = df_out['ext'] + df_out['abs_gas'] + df_out['sca_gas']
+
+    return df_out

AeroViz/dataProcess/Optical/_mie.py

@@ -1,16 +1,14 @@
 # from PyMieScatt import Mie_SD
-# from PyMieScatt import Mie_SD
 
 from ._mie_sd import Mie_SD
-from pandas import date_range, concat, DataFrame, to_numeric
 
 
 def _mie(_psd_ori, _RI_ori, _wave):
-	_ori_idx = _psd_ori.index.copy()
-	_cal_idx = _psd_ori.loc[_RI_ori.dropna().index].dropna(how='all').index
+    _ori_idx = _psd_ori.index.copy()
+    _cal_idx = _psd_ori.loc[_RI_ori.dropna().index].dropna(how='all').index
 
-	_psd, _RI = _psd_ori.loc[_cal_idx], _RI_ori.loc[_cal_idx]
+    _psd, _RI = _psd_ori.loc[_cal_idx], _RI_ori.loc[_cal_idx]
 
-	_out = Mie_SD(_RI.values, 550, _psd)
+    _out = Mie_SD(_RI.values, 550, _psd)
 
-	return _out.reindex(_ori_idx)
+    return _out.reindex(_ori_idx)