OilfieldToolsNet/LBPM
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

finish the function for segmenting the floating phase field data. The segmented phase field has 0: solid; 1: non-wetting phase; and 2: wetting phase.

Browse Source
This commit is contained in:
Rex Zhe Li 2017-09-26 17:29:49 +10:00
parent d50b965d81
commit 1ab68aa60c
1 changed files with 38 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
example/Python/lbpm_segment_phase_field.py
View File

@ -1,7 +1,7 @@
import numpy as np
#import matplotlib.pyplot as plt
from netCDF4 import Dataset
#from glob import glob
from glob import glob
def read_NetCDF_file_py(file_name,data_name):
@ -36,17 +36,46 @@ def write_NetCDF_file_py(file_name,data_name,data_IN,data_IN_dtype,lx,ly,lz):
print '**Info: The *.nc file is written successfully !'
#end def
input_nc_file_name = 'OutputData_vis200000.nc'
#NOTE: To do a 3-phase segmentation, you need the original segmented image file,
# which contains 0->solid and 1->pore.
print "**Info: Need to read the original input segmetned image, "
print " which contains 0 -> solid and 1 -> pore"
img_name = "domain1_64_reduced_lbpm.bin"
print "**Info: Read the image: "+img_name
img = np.fromfile(img_name,dtype = np.int8)
input_nc_file_group = glob('OutputData_vis*.nc')
input_nc_file_var_name = 'Phase'
output_nc_file_name = input_nc_file_name[:-3]+'_PhaseSeg.nc'
output_nc_file_var_name = 'Phase_seg'
phase=read_NetCDF_file_py(input_nc_file_name,input_nc_file_var_name)
(lz,ly,lx) = phase.shape
phase[phase>0.0]=1
phase[phase<0.0]=0
phase = phase.astype(np.int32)
write_NetCDF_file_py(output_nc_file_name,output_nc_file_var_name,phase,phase.dtype.char,lx,ly,lz)
if not input_nc_file_group:
print "**Error: No input data files: "+input_nc_file_group+" have been found !"
else:
for ii in range(len(input_nc_file_group)):
input_nc_file_name = input_nc_file_group[ii]
output_nc_file_name = input_nc_file_name[:-3]+'_PhaseSeg.nc'
print "**Info: Read data file: "+input_nc_file_name
phase=read_NetCDF_file_py(input_nc_file_name,input_nc_file_var_name)
(lz,ly,lx) = phase.shape
img.shape = phase.shape
# Double check the phase field has the same size as the segmented image
if phase.size != img.size:
print "**Error: The size of the loaded segmented image does not match the size of the phase field data !"
else:
phase[phase>0.0]=1
phase[phase<0.0]=2
phase[img==0]=0
phase = phase.astype(np.int32)
write_NetCDF_file_py(output_nc_file_name,output_nc_file_var_name,phase,phase.dtype.char,lx,ly,lz)
print "**Info: The segmented phase field from the floating phase field has been written."
#end if
#end for
#end if