wassname/simpeg
1
0
Fork 0
mirror of https://github.com/wassname/simpeg.git synced 2026-07-29 11:27:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

Implement active cell from topo and topocheck function

Browse Source
This commit is contained in:
D Fournier
2015-12-29 16:11:16 -08:00
parent 1a7656c34d
commit 201fe6bf46
3 changed files with 172 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files
simpegPF/Dev/Intgrl_MAG_Fwr_Driver.py
+23 -13
View File
@@ -1,14 +1,17 @@
import os
home_dir = 'C:\Users\dominiquef.MIRAGEOSCIENCE\Documents\GIT\SimPEG\simpegpf\simpegPF\Dev'
home_dir = 'C:\Users\dominiquef.MIRAGEOSCIENCE\ownCloud\Research\Modelling\Synthetic\Block_Gaussian_topo'
inpfile = 'MAG3Cfwr.inp'
inpfile = 'PYMAG3C_fwr.inp'
dsep = '\\'
os.chdir(home_dir)
#%%
from SimPEG import np, Utils
from SimPEG import np, sp, Utils, mkvc, Maps
import simpegPF as PF
import pylab as plt
## New scripts to be added to basecode
#from fwr_MAG_data import fwr_MAG_data
@@ -23,27 +26,34 @@ mesh = Utils.meshutils.readUBCTensorMesh(mshfile)
# Load model file
model = Utils.meshutils.readUBCTensorModel(modfile,mesh)
# Load in topofile or create flat surface
if topofile == 'null':
actv = np.ones(mesh.nC)
else:
topo = np.genfromtxt(topofile,skip_header=1)
actv = PF.Magnetics.getActiveTopo(mesh,topo,'N')
Utils.writeUBCTensorModel('nullcell.dat',mesh,actv)
# Load in observation file
[B,M,dobs] = PF.BaseMag.readUBCmagObs(obsfile)
rxLoc = dobs[:,0:3]
#rxLoc[:,2] += 5 # Temporary change for test
ndata = rxLoc.shape[0]
#%% Run forward modeling
# Compute forward model using integral equation
d = PF.Magnetics.Intgrl_Fwr_Data(mesh,B,M,rxLoc,model,'tmi')
d = PF.Magnetics.Intgrl_Fwr_Data(mesh,B,M,rxLoc,model,actv,'tmi')
# Form data object with coordinates and write to file
data = np.c_[rxLoc , d , np.zeros((ndata,1))]
wd = np.zeros((ndata,1))
# Save forward data to file
with file('FWR_data.dat','w') as fid:
fid.write('%6.2f %6.2f %6.2f\n' %(B[0], B[1], B[2]) )
fid.write('%6.2f %6.2f %6.2f\n' %(M[0], M[1], 1) )
fid.write('%i\n' %(ndata) )
np.savetxt(fid, data, fmt='%e',delimiter=' ',newline='\n')
print "Observation file saved to " + home_dir + '\FWR_data.dat'
PF.Magnetics.writeUBCobs(home_dir + dsep + 'FWR_data.dat',B,M,rxLoc,d,wd)
simpegPF/Dev/Intgrl_MAG_Inv_Driver.py
+22 -11
View File
@@ -1,8 +1,9 @@
import os
home_dir = 'C:\Users\dominiquef.MIRAGEOSCIENCE\Documents\GIT\SimPEG\simpegpf\simpegPF\Dev'
#home_dir = 'C:\Users\dominiquef.MIRAGEOSCIENCE\Documents\GIT\SimPEG\simpegpf\simpegPF\Dev'
home_dir = 'C:\\Users\\dominiquef.MIRAGEOSCIENCE\\ownCloud\\Research\\Modelling\\Synthetic\\Block_Gaussian_topo'
inpfile = 'MAG3D_inv.inp'
inpfile = 'PYMAG3D_inv.inp'
dsep = '\\'
os.chdir(home_dir)
@@ -27,18 +28,23 @@ mesh = Utils.meshutils.readUBCTensorMesh(mshfile)
[B,M,dobs] = PF.BaseMag.readUBCmagObs(obsfile)
rxLoc = dobs[:,0:3]
d = dobs[:,3]
wd = dobs[:,4]
ndata = rxLoc.shape[0]
# Load in topofile or create flat surface
#==============================================================================
# if topofile == 'null':
#
# Nx,Ny = np.meshgrid(mesh.vectorNx,mesh.vectorNy)
# Nz = np.ones(Nx.shape) * mesh.vectorNz[-1]
#
# topo = np.c_[mkvc(Nx),mkvc(Ny),mkvc(Nz)]
#==============================================================================
if topofile == 'null':
Nx,Ny = np.meshgrid(mesh.vectorNx,mesh.vectorNy)
Nz = np.ones(Nx.shape) * mesh.vectorNz[-1]
topo = np.c_[mkvc(Nx),mkvc(Ny),mkvc(Nz)]
else:
topofile = np.genfromtxt(topofile,delimiter=' \n',dtype=np.str,skip_header=0)
# Work with flat topogrphy for now
nullcell = np.ones(mesh.nC)
@@ -63,4 +69,9 @@ F = PF.Magnetics.Intrgl_Fwr_Op(mesh,B,M_xyz,rxLoc,'tmi')
# Get distance weighting function
wr = PF.Magnetics.get_dist_wgt(mesh,rxLoc,3.,np.min(mesh.hx)/4)
Utils.writeUBCTensorModel(home_dir+dsep+'wr.dat',mesh,wr)
Utils.writeUBCTensorModel(home_dir+dsep+'wr.dat',mesh,wr)
# Write out the predicted
pred = F.dot(mstart)
PF.Magnetics.writeUBCobs(home_dir + dsep + 'Pred.dat',B,M,rxLoc,pred,wd)