#ws_mesh = ws.WSMesh()
#ws_mesh.read_initial_file(r"/home/jpeacock/Documents/wsinv3d/LV/WSInitialModel_small")
#ws_mesh.station_fn = r"/home/jpeacock/Documents/wsinv3d/LV/WS_Station_Locations_small.txt"
#ws_stations = ws.WSStation()
#ws_stations.read_station_file(ws_mesh.station_fn)
#ws_mesh.station_locations = ws_stations.station_locations.copy()
#ws_mesh.plot_mesh()
#--> make data files
folder = 'sm_modem_inv1'
ws_data = ws.WSData(edi_list=edi_list,
period_list=inv_periods,
station_fn=ws_mesh.station_fn)
ws_data.n_z = 8
ws_data.z_err = .07
ws_data.z_err_map = [10, 1, 1, 10]
ws_data.z_err_floor = .07
sv_path = os.path.join(save_path, '{0}'.format(folder))
if not os.path.isdir(sv_path):
os.mkdir(sv_path)
ws_data.write_data_file(save_path=sv_path,
data_basename='WS_data_{0}_8_small.dat'.format(folder))
ws_data_fn = os.path.join(save_path, 'Inv1_{0}'.format(folder),
'WS_data_{0}_8_small.dat'.format(folder))
ws_startup = ws.WSStartup(data_fn=os.path.basename(ws_data_fn),
initial_fn=os.path.basename(ws_mesh.initial_fn))
ws_startup.output_stem = 'lv_{0}'.format(folder)
ws_startup.save_path = sv_path
ws_startup.write_startup_file()
save_path = r"c:\Users\jrpeacock\Documents\kates_data\Inv_01" # 2) make a grid from the stations themselves with 200m cell spacing wsmesh = ws.WSMesh(edi_list=edi_list, cell_size_east=12500, cell_size_north=12500) wsmesh.save_path = save_path wsmesh.make_mesh() # check to see if the mesh is what you think it should be # wsmesh.plot_mesh() # all is good write the mesh file wsmesh.res_list = np.logspace(-1, 3, 8) wsmesh.write_initial_file(save_path=save_path) # note this will write a file with relative station locations # change the starting model to be different than a halfspace mm = ws.WSModelManipulator(initial_fn=wsmesh.initial_fn) # an interactive gui will pop up to change the resistivity model # once finished write a new initial file mm.rewrite_initial_file() # 3) write data file wsdata = ws.WSData(edi_list=edi_list, period_list=p_list, station_fn=wsmesh.station_fn) wsdata.save_path = save_path wsdata.write_data_file() # 4) plot mt response to make sure everything looks ok # rp = ws.PlotResponse(data_fn=wsdata.data_fn) # 5) make startup file sws = ws.WSStartup(data_fn=wsdata.data_fn, initial_fn=mm.initial_fn) sws.save_path = save_path sws.error_tol = 5 sws.write_startup_file()