# 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