import mtpy.utils.array2raster as a2r
import os
model_fn = r"c:\Users\jpeacock\Documents\SaudiArabia\modem_inv\inv_n30w\med_z03_t02_c03_164.rho"
save_path = r"c:\Users\jpeacock\Documents\SaudiArabia\modem_inv\inv_n30w\med_raster"
# model_center = (39.756403, 24.342190)
model_center = (39.759, 24.338)
def check_dir(directory_path):
if os.path.isdir(directory_path) is False:
os.mkdir(directory_path)
print "Made directory {0}".format(directory_path)
check_dir(save_path)
# ll_cc = (-120.09987536766145-.008, 36.3497971057915+.405) # lv16 ll
modem_raster = a2r.ModEM_to_Raster()
modem_raster.model_fn = model_fn
ll_cc = modem_raster.get_model_lower_left_coord(model_center=model_center,
pad_east=5,
pad_north=5)
modem_raster.lower_left_corner = (ll_cc[0] + 0.375, ll_cc[1] - 0.08)
modem_raster.save_path = save_path
modem_raster.projection = "WGS84"
modem_raster.rotation_angle = 30.0
modem_raster.write_raster_files()
k1 = np.where(m.grid_z >= mbase[jj, ii])[0][0]
k2 = np.where(m.grid_z >= mbase[jj, ii] + 3000)[0][0]
c = (1.0 / m.res_model[jj, ii, k1:k2]) * abs(m.grid_z[k1:k2])
c_base[jj, ii] = np.nansum(c)
k1 = np.where(m.grid_z >= 6000)[0][0]
k2 = np.where(m.grid_z >= 10000)[0][0]
c = (1.0 / m.res_model[jj, ii, k1:k2]) * abs(m.grid_z[k1:k2])
c_8k[jj, ii] = np.nansum(c)
k1 = np.where(m.grid_z >= 10000)[0][0]
k2 = np.where(m.grid_z >= 15000)[0][0]
c = (1.0 / m.res_model[jj, ii, k1:k2]) * abs(m.grid_z[k1:k2])
c_15k[jj, ii] = np.nansum(c)
m2r = array2raster.ModEM_to_Raster()
m2r.model_obj = m
ll_cc = m2r.get_model_lower_left_coord(model_center=model_center,
pad_east=9,
pad_north=9)
m2r.lower_left_corner = ll_cc
m2r.save_path = r"c:\Users\jpeacock\OneDrive - DOI\Geothermal\Umatilla\modem_inv\inv_09\depth_slices"
m2r.projection = "NAD83"
for ii, carray in enumerate([c_top, c_crb, c_base, c_8k, c_15k]):
m2r.model_obj.grid_z = np.array([ii, 100])
m2r.model_obj.res_model = carray.reshape((mdem.shape[0], mdem.shape[1], 1))
m2r.write_raster_files()
print(ii)
