# dset=raytomo.RayTomoDataSet('/scratch/summit/life9360/ray_tomo_WUS.h5')
# dset.set_input_parameters(minlon=235., maxlon=255., minlat=31., maxlat=50., data_pfx='raytomo_in_', smoothpfx='N_INIT_', qcpfx='QC_')
# dset.run_smooth(datadir='../ray_tomo_data', outdir='../ray_tomo_working_dir', reshape=False)
# dset.run_qc(outdir='../ray_tomo_working_dir', isotropic=False, anipara=1, alphaAni4=1000)
# dset.run_qc(outdir='./ray_tomo_working_dir', isotropic=True, anipara=1, alphaAni4=1000)
#
# dset.get_data4plot(dataid='qc_run_0', period=12.)
# dset.plot_vel_iso(vmin=2.9, vmax=3.5, fastaxis=False, projection='global')
# # dset.plot_vel_iso(vmin=3.5, vmax=4.0)
# dset.plot_fast_axis()
# dset.generate_corrected_map(dataid='qc_run_0', glbdir='./MAPS', outdir='./REG_MAPS')
# dset.plot_global_map(period=50., inglbpfx='./MAPS/smpkolya_phv_R')
#
# #################################################################################
import eikonaltomo
dset = eikonaltomo.EikonalTomoDataSet('/work2/leon/eikonal_tomo_WUS.h5')
# dset.compare_raytomo('/scratch/summit/life9360/ray_tomo_WUS.h5', 1, 2, 0, 12.)
# dset.set_input_parameters(minlon=235., maxlon=255., minlat=31., maxlat=50., pers=np.array([10.]))
# # # dset.set_input_parameters(minlon=235., maxlon=255., minlat=31., maxlat=50.)
# dset.xcorr_eikonal_mp(inasdffname='/work2/leon/COR_WUS_for_shane.h5',\
# workingdir='/work2/leon/eikonal_working_WUS', fieldtype='Tph', channel='ZZ', data_type='FieldDISPpmf2interp', nprocess=10)
# dset.xcorr_eikonal(inasdffname='../COR_WUS.h5', workingdir='./eikonal_working', fieldtype='Tph', channel='ZZ', data_type='FieldDISPpmf2interp')
# #
# # t1=timeit.default_timer()
# dset.eikonal_stack(anisotropic=False)
# # t2=timeit.default_timer()
# # print t2-t1
# # dset.eikonal_stack()
# # dset._get_lon_lat_arr('Eikonal_run_0')
# dset.get_data4plot(period=28.)
# dset.np2ma()
# # dset.xcorr_eikonal_mp(inasdffname='../COR_WUS.h5', workingdir='./eikonal_working_mp', fieldtype='Tph', channel='ZZ', data_type='FieldDISPpmf2interp', nprocess=10)
# # #
# # # t1=timeit.default_timer()
# # dset.eikonal_stack()
# # # t2=timeit.default_timer()
# # # print t2-t1
# # dset.eikonal_stack()
# # # dset._get_lon_lat_arr('Eikonal_run_0')
# dset.get_data4plot(period=60.)
# dset.np2ma()
# dset.plot_vel_iso(vmin=3.7, vmax=4.2)
# # dset.plot_vel_iso(vmin=2.9, vmax=3.5)
import eikonaltomo
# #
dset = eikonaltomo.EikonalTomoDataSet('../eikonal_tomo_WUS.h5')
# dset.set_input_parameters(minlon=235., maxlon=255., minlat=31., maxlat=50., pers=np.array([12.]))
# dset.set_input_parameters(minlon=235., maxlon=255., minlat=31., maxlat=50.)
# dset.xcorr_eikonal_mp(inasdffname='../COR_WUS.h5', workingdir='./eikonal_working', fieldtype='Tph', channel='ZZ', data_type='FieldDISPpmf2interp', nprocess=10)
# dset.xcorr_eikonal(inasdffname='../COR_WUS.h5', workingdir='./eikonal_working', fieldtype='Tph', channel='ZZ', data_type='FieldDISPpmf2interp')
# #
# # t1=timeit.default_timer()
# dset.eikonal_stack()
# # t2=timeit.default_timer()
# # print t2-t1
# # dset.eikonal_stack()
# # dset._get_lon_lat_arr('Eikonal_run_0')
# dset.get_data4plot(period=28.)
# dset.np2ma()
# dset.plot_vel_iso(vmin=3.4, vmax=4.0)
#
# # dset.read_surf_waveforms_DMT(datadir='/scratch/summit/life9360/ALASKA_work/surf_19950101_20170831', verbose=False)
# # dset.get_surf_waveforms(startdate='2017-09-01', verbose=False)
#
# # dset.quake_prephp(outdir='/work1/leon/ALASKA_work/quake_working_dir/pre_disp')
# # inftan = pyaftan.InputFtanParam()
# # inftan.tmax = 100.
# # inftan.tmin = 5.
# # dset.quake_aftan(prephdir='/work1/leon/ALASKA_work/quake_working_dir/pre_disp_R', inftan=inftan)
# # dset.interp_disp(verbose=True)
# dset.quake_get_field()
#
import eikonaltomo
#
dset = eikonaltomo.EikonalTomoDataSet(
'/work1/leon/ALASKA_work/hdf5_files/helm_20190809.h5')
# dset = eikonaltomo.EikonalTomoDataSet('/work1/leon/ALASKA_work/hdf5_files/eikonal_quake_20181022_helm_40sec.h5')
# dset = eikonaltomo.EikonalTomoDataSet('/work1/leon/ALASKA_work/hdf5_files/eikonal_quake_20181022_helm_40sec_noqc.h5')
# pers = np.array([40.])
# pers = np.append( np.arange(11.)*2.+20., np.arange(10.)*5.+45.)
# dset.set_input_parameters(minlon=188, maxlon=238, minlat=52, maxlat=72, pers=pers)
# # # #
# # # # # dset = eikonaltomo.EikonalTomoDataSet('/work1/leon/ALASKA_work/hdf5_files/eikonal_quake_20181015.h5')
# # # # # # #
pers = np.array([60.])
dset.set_input_parameters(minlon=188,
maxlon=238,
minlat=52,
maxlat=72,
pers=pers)
import quakedbase
import numpy as np
import timeit
import matplotlib.pyplot as plt
import pyaftan
import eikonaltomo
# #
dset = eikonaltomo.EikonalTomoDataSet(
'/work1/leon/ALASKA_work/hdf5_files/eikonal_quake_Love_20190314.h5')
# pers = np.append( np.arange(11.)*2.+20., np.arange(10.)*5.+45.)
# dset.set_input_parameters(minlon=188, maxlon=238, minlat=52, maxlat=72, pers=pers)
# # dset.set_input_parameters(minlon=188, maxlon=238, minlat=52, maxlat=72, pers=np.array([50.]))
# # # # #
# dset.quake_eikonal_mp_lowmem(inasdffname='/work1/leon/ALASKA_work/ASDF_data/surf_Alaska_Love.h5', \
# workingdir='/work1/leon/Alaska_quake_eikonal_Love_working_mp_20190314', fieldtype='Tph', channel='T', \
# data_type='FieldDISPpmf2interp', amplplc=False, cdist=250., nprocess=30, btime_qc='2006-01-01', deletetxt=True)
# # #
# dset.eikonal_stack(runid=0)
# dset.helm_stack(runid=0, anisotropic=False, dv_thresh=0.2)
#
# dset.compare_eiktomo(ineiktomofname='/work1/leon/ALASKA_work/hdf5_files/eikonal_xcorr_tomo_Alaska_TA_AK_20190218_250km.h5', \
# inrunid=0, runid=0, period=40., Nmeasure=50, helm=False)
#
# dset = eikonaltomo.EikonalTomoDataSet('/work1/leon/ALASKA_work/hdf5_files/eikonal_quake_20181008_iso.h5')
# dset2 = eikonaltomo.EikonalTomoDataSet('/work1/leon/ALASKA_work/hdf5_files/eikonal_xcorr_tomo_Alaska_TA_AK_20180814_250km_ani.h5')
#
# data1 = dset.plot_azimuthal_single_point(inlat=65., inlon=-145.+360., runid=1, period=50., fitdata=True, getdata=True)
#
# data2 = dset2.plot_azimuthal_single_point(inlat=65., inlon=-145.+360., runid=0, period=50., fitdata=True, getdata=True)
#
# import matplotlib.pyplot as plt
# plt.errorbar(data1[0]+180., data1[1], yerr=data1[2], fmt='o')
# plt.plot(data1[0]+180., data1[3]*np.ones(data1[0].size))
# plt.errorbar(data2[0]+180., data2[1], yerr=data2[2], fmt='o')
# plt.plot(data2[0]+180., data2[3]*np.ones(data2[0].size))
# plt.show()
dset = eikonaltomo.EikonalTomoDataSet(
'/work1/leon/ALASKA_work/hdf5_files/eikonal_quake_debug_001.h5')
# #
pers = np.array([80.])
dset.set_input_parameters(minlon=188,
maxlon=238,
minlat=52,
maxlat=72,
pers=pers)
dset.quake_eikonal(inasdffname='/work1/leon/ALASKA_work/ASDF_data/surf_Alaska_for_mp.h5', \
workingdir='/work1/leon/Alaska_quake_eikonal_working', fieldtype='Tph', channel='Z', \
data_type='FieldDISPpmf2interp', amplplc=True, cdist=None, btime_qc='2006-01-01', deletetxt=False)
# dset.eikonal_stack(runid=0, anisotropic=False)
# dset.helm_stack(runid=4, anisotropic=False)
# # Downloading data
# t1=timeit.default_timer()
# dset.read_surf_waveforms_DMT(datadir='/scratch/summit/life9360/ALASKA_work/surf_19950101_20170831', verbose=False)
# dset.quake_prephp(outdir='/scratch/summit/life9360/WUS_quake_working_dir/pre_disp')
# inftan = pyaftan.InputFtanParam()
# inftan.tmax = 100.
# inftan.tmin = 5.
# dset.quake_aftan(prephdir='/scratch/summit/life9360/WUS_quake_working_dir/pre_disp_R', inftan=inftan)
# dset.interp_disp(verbose=True)
# dset.quake_get_field()
import eikonaltomo
# # #
dset = eikonaltomo.EikonalTomoDataSet(
'/scratch/summit/life9360/eikonal_quake.h5')
# dset2=eikonaltomo.EikonalTomoDataSet('../eikonal_tomo_quake_mp.h5')
dset.set_input_parameters(minlon=235.,
maxlon=255.,
minlat=31.,
maxlat=50.,
pers=np.array([60.]))
# dset2.set_input_parameters(minlon=235., maxlon=255., minlat=31., maxlat=50., pers=np.array([60.]))
# dset.set_input_parameters(minlon=235., maxlon=255., minlat=31., maxlat=50.)
# dset.xcorr_eikonal_mp(inasdffname='../COR_WUS.h5', workingdir='./eikonal_working', fieldtype='Tph', channel='ZZ', data_type='FieldDISPpmf2interp', nprocess=10)
field = dset.quake_eikonal(inasdffname='/scratch/summit/life9360/WUS_quake.h5',
workingdir='./eikonal_working',
fieldtype='Tph',
channel='Z',
data_type='FieldDISPpmf2interp',
amplplc=True)
import noisedbase
import numpy as np
import timeit
import matplotlib.pyplot as plt
import eikonaltomo
import raytomo
#-----------------------
#initialization
#-----------------------
# dset = eikonaltomo.EikonalTomoDataSet('/work1/leon/ALASKA_work/hdf5_files/eikonal_xcorr_tomo_Alaska_all_20190318_250km_snr_10.h5')
dset = eikonaltomo.EikonalTomoDataSet('/work1/leon/ALASKA_work/hdf5_files/eikonal_xcorr_tomo_Alaska_all_20190318_250km_azi.h5')
# cmap = raytomo.discrete_cmap(6, 'jet')
# dset.set_input_parameters(minlon=188, maxlon=238, minlat=52, maxlat=72)
dset.set_input_parameters(minlon=188, maxlon=238, minlat=52, maxlat=72, pers=np.array([20.]))
# # #
# # # #------------------------------
# # # # perform eikonal tomography
# # # #------------------------------
# dset.xcorr_eikonal_mp_lowmem(inasdffname='/work1/leon/ALASKA_work/ASDF_data/xcorr_Alaska_RTZ_ray_20190314_all.h5', \
# workingdir='/work1/leon/ALASKA_work/eikonal_working_all_20190318', \
# fieldtype='Tph', channel='ZZ', data_type='FieldDISPpmf2interp', nprocess=30, subsize=1000, mindp=10., cdist=250.)
# # #
# # # # dset.xcorr_eikonal(inasdffname='/scratch/summit/life9360/ALASKA_work/ASDF_data/xcorr_Alaska.h5', \
# # # # workingdir='/scratch/summit/life9360/ALASKA_work/eikonal_working_debug', \
# # # # fieldtype='Tph', channel='ZZ', data_type='FieldDISPpmf2interp', mindp=10., deletetxt=False)
# #
# # #------------------------------
# # # perform eikonal stacking
# # #------------------------------
# # # # # # # t1=timeit.default_timer()
dset.eikonal_stack_new(anisotropic=True)
import noisedbase
import numpy as np
import timeit
import matplotlib.pyplot as plt
import eikonaltomo
import raytomo
#-----------------------
#initialization
#-----------------------
dset = eikonaltomo.EikonalTomoDataSet(
'/work1/leon/ALASKA_work/hdf5_files/eikonal_xcorr_tomo_Alaska_TA_AK_20190317_250km.h5'
)
# dset = eikonaltomo.EikonalTomoDataSet('/work1/leon/ALASKA_work/hdf5_files/eikonal_xcorr_tomo_Alaska_all_20190318_250km_azi.h5')
# cmap = raytomo.discrete_cmap(6, 'jet')
# dset.set_input_parameters(minlon=188, maxlon=238, minlat=52, maxlat=72)
# dset.set_input_parameters(minlon=188, maxlon=238, minlat=52, maxlat=72, pers=np.array([20.]))
# # # #
# # # # #------------------------------
# # # # # perform eikonal tomography
# # # # #------------------------------
# # dset.xcorr_eikonal_mp_lowmem(inasdffname='/work1/leon/ALASKA_work/ASDF_data/xcorr_Alaska_RTZ_ray_20190314_all.h5', \
# # workingdir='/work1/leon/ALASKA_work/eikonal_working_all_20190318', \
# # fieldtype='Tph', channel='ZZ', data_type='FieldDISPpmf2interp', nprocess=30, subsize=1000, mindp=10., cdist=250.)
# # # #
# # # # # dset.xcorr_eikonal(inasdffname='/scratch/summit/life9360/ALASKA_work/ASDF_data/xcorr_Alaska.h5', \
# # # # # workingdir='/scratch/summit/life9360/ALASKA_work/eikonal_working_debug', \
# # # # # fieldtype='Tph', channel='ZZ', data_type='FieldDISPpmf2interp', mindp=10., deletetxt=False)
# # #
# # # #------------------------------
# # # # perform eikonal stacking
# # # #------------------------------
# dset = eikonaltomo.EikonalTomoDataSet('/work1/leon/ALASKA_work/hdf5_files/debug2_xcorr_90.h5')
# dset = eikonaltomo.EikonalTomoDataSet('/work1/leon/ALASKA_work/hdf5_files/comp_xcorr_2deg.h5')
# dset.set_input_parameters(minlon=188, maxlon=238, minlat=52, maxlat=72, pers=np.array([28., 30., 32.]))
# dset.eikonal_stack_mp(anisotropic=True, spacing_ani=2., N_bin=20, azi_amp_tresh=0.1, workingdir='./comp_xcorr_2deg', enhanced=False)
# dset.compute_azi_aniso()
# dset = eikonaltomo.EikonalTomoDataSet('/work1/leon/ALASKA_work/hdf5_files/comp_xcorr_2deg_0.05.h5')
# dset.set_input_parameters(minlon=188, maxlon=238, minlat=52, maxlat=72, pers=np.array([30.]))
# dset.set_input_parameters(minlon=188, maxlon=238, minlat=52, maxlat=72, pers=np.array([28., 30., 32.]))
# dset.eikonal_stack_mp(anisotropic=True, spacing_ani=2., N_bin=20, azi_amp_tresh=0.05, workingdir='./comp_xcorr_2deg_0.05', enhanced=False)
# dset.compute_azi_aniso()
# dset = eikonaltomo.EikonalTomoDataSet('/work1/leon/ALASKA_work/hdf5_files/comp_xcorr_2deg_0.05_untest.h5')
# dset.compute_azi_aniso_enhanced()
dset = eikonaltomo.EikonalTomoDataSet(
'/work1/leon/ALASKA_work/hdf5_files/azi_xcorr_all_2deg_0.05.h5')
# dset.set_input_parameters(minlon=188, maxlon=238, minlat=52, maxlat=72, pers=np.array([20., 22., 24., 26., 28., 30., 32., 34., 36., 38., 40.]))
# dset.eikonal_stack_mp(anisotropic=True, spacing_ani=2., N_bin=20, azi_amp_tresh=0.05, workingdir='./azi_xcorr_all_2deg_0.05')
# dset.compute_azi_aniso()
# dset.plot_fast_axis(runid=0, period=30., factor=5, ampref = 1.)
# dset.diff_fast_axis(inh5fname='/work1/leon/ALASKA_work/hdf5_files/azi_quake_2deg_0.05.h5', vmin=0., vmax=90., period=30., runid=0)
# dset.combine_amp(inh5fname='/work1/leon/ALASKA_work/hdf5_files/azi_quake_2deg_0.05.h5', period=30., runid=0)
# dset.combine_fast_axis(inh5fname='/work1/leon/ALASKA_work/hdf5_files/azi_quake_2deg_0.05.h5', period=30., runid=0)
# dset.plot_fast_axis(0, 30., datatype='vel_iso', vmin=3.6, vmax=3.9, ampref=1., normv=1.5, scaled=True, factor=5)
# dset.diff_amp(inh5fname='/work1/leon/ALASKA_work/hdf5_files/azi_quake_2deg_0.05.h5', period=30., runid=0)