AeroViz 0.1.0__py3-none-any.whl

AeroViz/dataProcess/SizeDistr/_merge_v2.py

@@ -0,0 +1,243 @@
+from AeroViz.dataProcess.core import _union_index
+
+from datetime import datetime as dtm
+from pandas import DataFrame, to_datetime
+# from scipy.interpolate import interp1d
+from scipy.interpolate import UnivariateSpline as unvpline, interp1d
+import numpy as np
+
+__all__ = ['_merge_SMPS_APS']
+
+
+def __test_plot(smpsx, smps, apsx, aps, mergex, merge, mergeox, mergeo, _sh):
+	from matplotlib.pyplot import subplots, close, show, rcParams
+
+	## parameter
+	# '''
+	## plot
+	fig, ax = subplots()
+
+	ax.plot(smpsx, smps, c='#ff794c', label='smps', marker='o', lw=2)
+	ax.plot(apsx, aps, c='#4c79ff', label='aps', marker='o', lw=2)
+	ax.plot(mergex, merge, c='#79796a', label='merge')
+	# ax.plot(mergeox,mergeo,c='#111111',label='mergeo',marker='o',lw=.75)
+
+	ax.set(xscale='log', yscale='log', )
+
+	ax.legend(framealpha=0, )
+	ax.set_title((_sh ** 2)[0], fontsize=13)
+
+	show()
+	close()
+
+
+# '''
+
+
+## Overlap fitting
+## Create a fitting func. by smps data
+## return : shift factor
+def _overlap_fitting(_smps_ori, _aps_ori, _smps_lb, _aps_hb):
+	print(f"\t\t{dtm.now().strftime('%m/%d %X')} : \033[92moverlap range fitting\033[0m")
+
+	## overlap fitting
+	## parmeter
+	_dt_indx = _smps_ori.index
+
+	## overlap diameter data
+	_aps = _aps_ori[_aps_ori.keys()[_aps_ori.keys() < _aps_hb]].copy()
+	_smps = _smps_ori[_smps_ori.keys()[_smps_ori.keys() > _smps_lb]].copy()
+
+	## use SMPS data apply power law fitting
+	## y = Ax^B, A = e**coefa, B = coefb, x = logx, y = logy
+	## ref : http://mathworld.wolfram.com/LeastSquaresFittingPowerLaw.html
+	## power law fit to SMPS num conc at upper bins to log curve
+
+	## coefficient A, B
+	_smps_qc_cond = ((_smps != 0) & np.isfinite(_smps))
+	_smps_qc = _smps.where(_smps_qc_cond)
+
+	_size = _smps_qc_cond.sum(axis=1)
+	_size = _size.where(_size != 0.).copy()
+
+	_logx, _logy = np.log(_smps_qc.keys()._data.astype(float)), np.log(_smps_qc)
+	_x, _y, _xy, _xx = _logx.sum(), _logy.sum(axis=1), (_logx * _logy).sum(axis=1), (_logx ** 2).sum()
+
+	_coeB = ((_size * _xy - _x * _y) / (_size * _xx - _x ** 2.))
+	_coeA = np.exp((_y - _coeB * _x) / _size).values.reshape(-1, 1)
+	_coeB = _coeB.values.reshape(-1, 1)
+
+	## rebuild shift smps data by coe. A, B
+	## x_shift = (y_ori/A)**(1/B)
+	_aps_shift_x = (_aps / _coeA) ** (1 / _coeB)
+	_aps_shift_x = _aps_shift_x.where(np.isfinite(_aps_shift_x))
+
+	## the least squares of diameter
+	## the shift factor which the cklosest to 1
+	_shift_factor = (_aps_shift_x.keys()._data.astype(float) / _aps_shift_x)
+	_shift_factor.columns = range(len(_aps_shift_x.keys()))
+
+	_dropna_idx = _shift_factor.dropna(how='all').index.copy()
+
+	## use the target function to get the similar aps and smps bin
+	## S2 = sum( (smps_fit_line(dia) - aps(dia*shift_factor) )**2 )
+	## assumption : the same diameter between smps and aps should get the same conc.
+
+	## be sure they art in log value
+	_S2 = DataFrame(index=_aps_shift_x.index)
+	_dia_table = DataFrame(np.full(_aps_shift_x.shape, _aps_shift_x.keys()),
+						   columns=_aps_shift_x.keys(), index=_aps_shift_x.index)
+	for _idx, _factor in _shift_factor.items():
+		_smps_fit_df = _coeA * (_dia_table / _factor.to_frame().values) ** _coeB
+		_S2[_idx] = ((_smps_fit_df - _aps) ** 2).sum(axis=1)
+
+	_least_squ_idx = _S2.idxmin(axis=1).loc[_dropna_idx]
+
+	_shift_factor_out = DataFrame(_shift_factor.loc[_dropna_idx].values[range(len(_dropna_idx)), _least_squ_idx.values],
+								  index=_dropna_idx).reindex(_dt_indx)
+
+	return _shift_factor_out, (DataFrame(_coeA, index=_dt_indx), DataFrame(_coeB, index=_dt_indx))
+
+
+## Remove big shift data ()
+## Return : aps, smps, shift (without big shift data)
+def _shift_data_process(_shift):
+	print(f"\t\t{dtm.now().strftime('%m/%d %X')} : \033[92mshift-data quality control\033[0m")
+
+	_rho = _shift ** 2
+	_shift = _shift.mask((~np.isfinite(_shift)) | (_rho > 2.6) | (_rho < 0.6))
+
+	# _qc_index = _shift.mask((_rho<0.6) | (_shift.isna())).dropna().index
+
+	# return _qc_index, _shift
+	return _shift
+
+
+# return _smps.loc[~_big_shift], _aps.loc[~_big_shift], _shift[~_big_shift].reshape(-1,1)
+
+
+## Create merge data
+##  shift all smps bin and remove the aps bin which smaller than the latest old smps bin
+## Return : merge bins, merge data, density
+def _merge_data(_smps_ori, _aps_ori, _shift_ori, _smps_lb, _aps_hb, _coe, _shift_mode):
+	print(f"\t\t{dtm.now().strftime('%m/%d %X')} : \033[92mcreate merge data : {_shift_mode}\033[0m")
+
+	_ori_idx = _smps_ori.index
+	_merge_idx = _smps_ori.loc[_aps_ori.dropna(how='all').index].dropna(how='all').index
+
+	_corr_aps_cond = _aps_ori.keys() < 700
+	_corr_aps_ky = _aps_ori.keys()[_corr_aps_cond]
+
+	_uni_idx, _count = np.unique(np.hstack((_smps_ori.dropna(how='all').index, _aps_ori.dropna(how='all').index,
+											_shift_ori.dropna(how='all').index)), return_counts=True)
+
+	_merge_idx = to_datetime(np.unique(_uni_idx[_count == 3]))
+
+	_smps, _aps, _shift = _smps_ori.loc[_merge_idx], _aps_ori.loc[_merge_idx], _shift_ori.loc[_merge_idx].values
+
+	## parameter
+	_coeA, _coeB = _coe[0].loc[_merge_idx], _coe[1].loc[_merge_idx]
+	_smps_key, _aps_key = _smps.keys()._data.astype(float), _aps.keys()._data.astype(float)
+
+	_cntr = 1000
+	_bin_lb = _smps_key[-1]
+
+	## make shift bins
+	_smps_bin = np.full(_smps.shape, _smps_key)
+	_aps_bin = np.full(_aps.shape, _aps_key)
+
+	_std_bin = np.geomspace(_smps_key[0], _aps_key[-1], 230)
+	_std_bin_merge = _std_bin[(_std_bin < _cntr) & (_std_bin > _bin_lb)]
+	_std_bin_inte1 = _std_bin[_std_bin <= _bin_lb]
+	_std_bin_inte2 = _std_bin[_std_bin >= _cntr]
+
+	if _shift_mode == 'mobility':
+		_aps_bin /= _shift
+
+	elif _shift_mode == 'aerodynamic':
+		_smps_bin *= _shift
+
+	## merge
+	_merge_lst, _corr_lst = [], []
+	for _bin_smps, _bin_aps, _dt_smps, _dt_aps, _sh in zip(_smps_bin, _aps_bin, _smps.values, _aps.values, _shift):
+		## keep complete smps bins and data
+		## remove the aps bin data lower than smps bin
+		_condi = _bin_aps >= _bin_smps[-1]
+
+		_merge_bin = np.hstack((_bin_smps, _bin_aps[_condi]))
+		_merge_dt = np.hstack((_dt_smps, _dt_aps[_condi]))
+
+		_merge_fit_loc = (_merge_bin < 1500) & (_merge_bin > _smps_lb)
+
+		## coeA and coeB
+		_unvpl_fc = unvpline(np.log(_merge_bin[_merge_fit_loc]), np.log(_merge_dt[_merge_fit_loc]), s=50)
+		_inte_fc = interp1d(_merge_bin, _merge_dt, kind='linear', fill_value='extrapolate')
+
+		_merge_dt_fit = np.hstack((_inte_fc(_std_bin_inte1), np.exp(_unvpl_fc(np.log(_std_bin_merge))),
+								   _inte_fc(_std_bin_inte2)))
+
+		_merge_lst.append(_merge_dt_fit)
+		_corr_lst.append(interp1d(_std_bin, _merge_dt_fit)(_bin_aps[_corr_aps_cond]))
+
+	_df_merge = DataFrame(_merge_lst, columns=_std_bin, index=_merge_idx)
+	_df_merge = _df_merge.mask(_df_merge < 0)
+
+	_df_corr = DataFrame(_corr_lst, columns=_corr_aps_ky, index=_merge_idx) / _aps_ori.loc[_merge_idx, _corr_aps_ky]
+
+	## process output df
+	## average, align with index
+	def _out_df(*_df_arg, **_df_kwarg):
+		_df = DataFrame(*_df_arg, **_df_kwarg).reindex(_ori_idx)
+		_df.index.name = 'time'
+		return _df
+
+	return _out_df(_df_merge), _out_df(_shift_ori ** 2), _out_df(_df_corr)
+
+
+def merge_SMPS_APS(df_smps, df_aps, aps_unit='um', smps_overlap_lowbound=500, aps_fit_highbound=1000):
+	df_smps, df_aps = _union_index(df_smps, df_aps)
+
+	## set to the same units
+	smps, aps_ori = df_smps.copy(), df_aps.copy()
+	smps.columns = smps.keys().to_numpy(float)
+	aps_ori.columns = aps_ori.keys().to_numpy(float)
+
+	if aps_unit == 'um':
+		aps_ori.columns = aps_ori.keys() * 1e3
+
+	den_lst, mer_lst = [], []
+	aps_input = aps_ori.loc[:, aps_ori.keys() > 700].copy()
+
+	for _count in range(2):
+
+		## shift infomation, calculate by powerlaw fitting
+		shift, coe = _overlap_fitting(smps, aps_input, smps_overlap_lowbound, aps_fit_highbound)
+
+		## process data by shift infomation, and average data
+		shift = _shift_data_process(shift)
+
+		## merge aps and smps
+		merge_arg = (smps, aps_ori, shift, smps_overlap_lowbound, aps_fit_highbound, coe)
+		merge_data_mob, density, _corr = _merge_data(*merge_arg, 'mobility')
+		merge_data_aer, density, _ = _merge_data(*merge_arg, 'aerodynamic')
+		density.columns = ['density']
+
+		if _count == 0:
+			corr = _corr.resample('1d').mean().reindex(smps.index).ffill()
+			corr = corr.mask(corr < 1, 1)
+			aps_ori.loc[:, corr.keys()] *= corr
+
+			aps_input = aps_ori.copy()
+
+	## out
+	out_dic = {
+		'data_all': merge_data_mob,
+		'data_all_aer': merge_data_aer,
+		'density_all': density,
+	}
+
+	## process data
+	for _nam, _df in out_dic.items():
+		out_dic[_nam] = _df.reindex(smps.index).copy()
+
+	return out_dic