53 lines
1.1 KiB
Python
53 lines
1.1 KiB
Python
#!/usr/bin/env python
|
|
|
|
import numpy as np
|
|
import matplotlib.pylab as plt
|
|
#from glob import glob
|
|
import sys
|
|
|
|
# Check if there is a proper command line argument
|
|
#if len(sys.argv) !=2:
|
|
# sys.stderr.write('Usage: ' + sys.argv[0] + ' Domain.in\n')
|
|
# sys.exit()
|
|
## end if
|
|
|
|
# Read 'Domain.in' file
|
|
f = open('Domain.in','r') # read-only
|
|
lines = f.readlines()
|
|
nprocx, nprocy, nprocz = np.fromstring(lines[0].splitlines()[0],dtype=np.int32,sep=' ')
|
|
nx, ny, nz = np.fromstring(lines[1].splitlines()[0],dtype=np.int32,sep=' ')
|
|
Lx, Ly, Lz = np.fromstring(lines[3].splitlines()[0],dtype=np.float32,sep=' ')
|
|
f.close()
|
|
|
|
|
|
lx = 3
|
|
ly = 500
|
|
lz = 500
|
|
|
|
# Load the data
|
|
SignDist = np.fromfile('SignDist.00000',dtype = np.float64)
|
|
SignDist.shape = (lz+2,ly+2,lx+2)
|
|
ID = np.fromfile('ID.00000',dtype = np.uint8)
|
|
ID.shape = (lz+2,ly+2,lx+2)
|
|
|
|
# Plot
|
|
#plt.figure(1)
|
|
#plt.title('SignDist Map')
|
|
#plt.pcolormesh(SignDist[:,:,2])
|
|
#plt.grid(True)
|
|
#plt.axis('equal')
|
|
#plt.colorbar()
|
|
#
|
|
#
|
|
#plt.figure(2)
|
|
#plt.title('ID.xxxxx')
|
|
#plt.pcolormesh(ID[:,:,2],cmap='hot')
|
|
#plt.grid(True)
|
|
#plt.axis('equal')
|
|
#plt.show()
|
|
|
|
|
|
|
|
|
|
|