default to 0–1 for z range
This commit is contained in:
parent
21bcafe3e4
commit
a2a7664e8e
|
@ -42,13 +42,13 @@ parser.add_option('-p','--position',
|
||||||
|
|
||||||
parser.set_defaults(x = False,
|
parser.set_defaults(x = False,
|
||||||
y = False,
|
y = False,
|
||||||
|
z = [0.0,1.0],
|
||||||
N = 16,
|
N = 16,
|
||||||
position = 'pos',
|
position = 'pos',
|
||||||
)
|
)
|
||||||
|
|
||||||
(options,filenames) = parser.parse_args()
|
(options,filenames) = parser.parse_args()
|
||||||
|
|
||||||
if options.z == None: parser.error('please specify top and bottom z plane...')
|
|
||||||
# --- loop over output files -------------------------------------------------------------------------
|
# --- loop over output files -------------------------------------------------------------------------
|
||||||
|
|
||||||
if filenames == []: filenames = [None]
|
if filenames == []: filenames = [None]
|
||||||
|
@ -61,7 +61,6 @@ for name in filenames:
|
||||||
except: continue
|
except: continue
|
||||||
damask.util.report(scriptName,name)
|
damask.util.report(scriptName,name)
|
||||||
|
|
||||||
print os.path.splitext(name)[-1]+'_poked_{}.seeds'.format(options.N)
|
|
||||||
# --- interpret header ----------------------------------------------------------------------------
|
# --- interpret header ----------------------------------------------------------------------------
|
||||||
|
|
||||||
table.head_read()
|
table.head_read()
|
||||||
|
@ -95,13 +94,13 @@ for name in filenames:
|
||||||
Nx = int(options.N/math.sqrt(options.N*info['size'][1]/info['size'][0]))
|
Nx = int(options.N/math.sqrt(options.N*info['size'][1]/info['size'][0]))
|
||||||
Ny = int(options.N/math.sqrt(options.N*info['size'][0]/info['size'][1]))
|
Ny = int(options.N/math.sqrt(options.N*info['size'][0]/info['size'][1]))
|
||||||
Nz = int((max(options.z)-min(options.z))/info['size'][2]*info['grid'][2])
|
Nz = int((max(options.z)-min(options.z))/info['size'][2]*info['grid'][2])
|
||||||
|
offset = int(min(options.z)/info['size'][2]*info['grid'][2]) # offset due to lower z-plane
|
||||||
|
|
||||||
damask.util.croak('poking {} x {} x {}...'.format(Nx,Ny,Nz))
|
damask.util.croak('poking {} x {} x {}...'.format(Nx,Ny,Nz))
|
||||||
|
|
||||||
seeds = np.zeros((Nx*Ny*Nz,4),'d')
|
seeds = np.zeros((Nx*Ny*Nz,4),'d')
|
||||||
grid = np.zeros(3,'i')
|
grid = np.zeros(3,'i')
|
||||||
|
|
||||||
offset = min(options.z)/info['size'][2]*info['grid'][2] # offset due to lower z-plane
|
|
||||||
n = 0
|
n = 0
|
||||||
for i in xrange(Nx):
|
for i in xrange(Nx):
|
||||||
grid[0] = round((i+0.5)*info['grid'][0]/Nx-0.5)
|
grid[0] = round((i+0.5)*info['grid'][0]/Nx-0.5)
|
||||||
|
@ -110,8 +109,7 @@ for name in filenames:
|
||||||
for k in xrange(Nz):
|
for k in xrange(Nz):
|
||||||
grid[2] = offset + k
|
grid[2] = offset + k
|
||||||
grid %= info['grid']
|
grid %= info['grid']
|
||||||
coordinates = (0.5+grid)*info['size']/info['grid']
|
seeds[n,0:3] = (0.5+grid)/info['grid'] # normalize coordinates to box
|
||||||
seeds[n,0:3] = coordinates/info['size'] # normalize coordinates to box
|
|
||||||
seeds[n, 3] = microstructure[grid[0],grid[1],grid[2]]
|
seeds[n, 3] = microstructure[grid[0],grid[1],grid[2]]
|
||||||
if options.x: grid[0] += 1
|
if options.x: grid[0] += 1
|
||||||
if options.y: grid[1] += 1
|
if options.y: grid[1] += 1
|
||||||
|
|
Loading…
Reference in New Issue