def image_edi(datadir):
""" image """
edi_file = datadir + r"edifiles2\15125A.edi"
save_file = datadir + 'penetration_depth1d.png'
edi_path = datadir + 'edifiles'
savepath = datadir
pd1d.plot_edi_file(edi_file, savefile=save_file, fig_dpi=400)
# Choose indices of periods to plot
period_index_list = [0, 1, 10, 20, 30, 40, 50, 59]
# Plot profiles for different modes ('det','zxy' or 'zyx')
pen2d.plot2Dprofile(edi_path,
period_index_list,
'det',
marker='o',
tick_params={'rotation': 'vertical'})
# Create plot for period index number 10 for determinant
pen3d.plot_latlon_depth_profile(edi_path,
10,
'det',
showfig=True,
savefig=True,
savepath=savepath,
fontsize=11,
fig_dpi=400,
file_format='png')
def test_plot2Dprofile_no_period_index_list(self):
"""
testing plot2Dprofile without period index list
exception should be raised
:return:
"""
try:
plot2Dprofile(self._edifiles)
except Exception:
pass
# selected_periods: the periods in seconds to plot depth for across each
# station.
selected_periods = [10., 100., 500., 600.]
# ptol: tolerance to use when finding nearest period to each selected
# period. If abs(selected period - nearest period) is greater than
# selected period * ptol, then the period is discarded and will appear
# as a gap in the plot.
ptol = 0.20
# zcomponent: component to plot. Valid parameters are 'det, 'zxy' and
# 'zyx'
zcomponent = 'det' # 'zxy', 'zyx' also options
pen2d.plot2Dprofile(edi_dir=edidir,
selected_periods=selected_periods,
ptol=ptol,
zcomponent=zcomponent,
save=True,
savepath='/tmp/Depth2D.png')
# selected_period_indices: indices of periods to plot.
# 'ptol' ins't required if using indices.
selected_period_indices = [0, 10, 20]
# p_index: needs to be set to True if using indices.
period_by_index = True
pen2d.plot2Dprofile(edi_dir=edidir,
selected_periods=selected_period_indices,
period_by_index=period_by_index,
zcomponent=zcomponent,
save=True,
def test_plot2Dprofile_wrong_rho(self):
try:
plot2Dprofile(self._edifiles, self._period_index_list, 'dat')
except Exception:
pass
def test_plot2Dprofile_zyx(self):
plot2Dprofile(self._edifiles, self._period_index_list, 'zyx')
def test_plot2Dprofile_det(self):
plot2Dprofile(self._edifiles, self._period_index_list, 'det')
# change the variable below according to your edi files folder !!!
# edidir = r'C:/mtpywin/mtpy/data/edifiles' # / is Unix and Win-Dos compatible
# or get this variable from the cmdline: edidir = sys.argv[1]
edidir = os.path.join(mtpy_path, 'data', 'edifiles2')
# savepath = r'C:\tmp'
temp_dir = tempfile.gettempdir()
print('Using temporary directory ' + temp_dir)
savepath = temp_dir
if not os.path.isdir(edidir):
print("Error: please provide the path to edi folder")
sys.exit(1)
period_index_list = [0, 1, 10, 20, 30, 40, 50, 59] # user to customise
# show three different kind of calculated pen-depth
pen2d.plot2Dprofile(edidir, period_index_list, 'det')
pen2d.plot2Dprofile(edidir, period_index_list, 'zxy')
pen2d.plot2Dprofile(edidir, period_index_list, 'zyx')
plt.savefig(
os.path.join(
savepath,
'penetration_depth_profile.png'), # change to your preffered filename
dpi=400) # change to your preferred file resolution
def test_plot2Dprofile_wrong_rho(self):
try:
plot2Dprofile(self._edifiles, self._period_list, zcomponent='dat')
except Exception:
pass
def test_plot2Dprofile_zyx_indices(self):
plot2Dprofile(self._edifiles,
self._period_indices,
zcomponent='zyx',
period_by_index=True)
def test_plot2Dprofile_zyx(self):
plot2Dprofile(self._edifiles, self._period_list, zcomponent='zyx')
def test_plot2Dprofile_det(self):
plot2Dprofile(self._edifiles, self._period_list, zcomponent='det')