# Number of padding cells to remove from plotting padc = 15 # Plotting parameters xmin, xmax = 10500, 13000 zmin, zmax = -600, 600 vmin, vmax = 0, 75 z = np.linspace(zmin, zmax, 4) #%% load obs file 3D dobs = DC.readUBC_DC3Dobs(home_dir + dsep + obs_file, rtype='IP') DCsurvey = dobs['DCsurvey'] # Assign line ID to the survey lineID = DC.xy_2_lineID(DCsurvey) uniqueID = np.unique(lineID) # Convert 3D locations to 2D survey dobs2D = DC.convertObs_DC3D_to_2D(DCsurvey, lineID,'Xloc') srcMat = DC.getSrc_locs(DCsurvey) #DCdata[src0, src0.rxList[0]] # Find 2D data correspondance dataID = np.zeros(dobs2D.nD) count = 0 for ii in range(dobs2D.nSrc): nD = dobs2D.srcList[ii].rxList[0].nD dataID[count:count+nD] = ii count += nD
# Assign Z-value from topo for ii in range(IPsurvey.nSrc): IPsurvey.srcList[ii].loc[0][2] = Ftopo(IPsurvey.srcList[ii].loc[0][0:2]) IPsurvey.srcList[ii].loc[1][2] = Ftopo(IPsurvey.srcList[ii].loc[1][0:2]) rx_x = IPsurvey.srcList[ii].rxList[0].locs[0][:, 0] rx_y = IPsurvey.srcList[ii].rxList[0].locs[0][:, 1] IPsurvey.srcList[ii].rxList[0].locs[0][:, 2] = Ftopo(rx_x, rx_y) rx_x = IPsurvey.srcList[ii].rxList[0].locs[1][:, 0] rx_y = IPsurvey.srcList[ii].rxList[0].locs[1][:, 1] IPsurvey.srcList[ii].rxList[0].locs[1][:, 2] = Ftopo(rx_x, rx_y) # Assign line ID to the survey lineID = DC.xy_2_lineID(DCsurvey) uniqueID = np.unique(lineID) IPlineID = DC.xy_2_lineID(IPsurvey) # Convert 3D locations to 2D survey DCdobs2D = DC.convertObs_DC3D_to_2D(DCsurvey, lineID, 'Xloc') IPdobs2D = DC.convertObs_DC3D_to_2D(IPsurvey, IPlineID, 'Xloc') srcMat = DC.getSrc_locs(IPsurvey) #DCdata[src0, src0.rxList[0]] # Find 2D data correspondance dataID = np.zeros(DCdobs2D.nD) count = 0