223031012
/
DAMASK_EICMD
mirror of https://github.com/achalhp/DAMASK_EICMD.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

output file name now contains both position and mode. 

e.g. ${prefix}_ipinitialcoord(point).vtr
This commit is contained in:
Chen Zhang 2015-11-03 21:44:36 +00:00
parent 519484233c
commit fe1dbf2ea0
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
processing/post/vtk_rectilinearGrid.py
View File

@ -49,7 +49,7 @@ for name in filenames:
table.head_read() table.head_read()
errors = [] errors = []
if table.label_dimension(options.position) != 3: if table.label_dimension(options.position) != 3:
errors.append('coordinates {} are not a vector.'.format(options.position)) errors.append('coordinates {} are not a vector.'.format(options.position))
if errors != []: if errors != []:
@ -68,39 +68,40 @@ for name in filenames:
[3.0 * coords[i][-1] - coords[i][-1 - (len(coords[i]) > 1)]]) for i in xrange(3)] [3.0 * coords[i][-1] - coords[i][-1 - (len(coords[i]) > 1)]]) for i in xrange(3)]
grid = np.array(map(len,coords),'i') grid = np.array(map(len,coords),'i')
N = grid.prod() if options.mode == 'point' else (grid-1).prod() N = grid.prod() if options.mode == 'point' else (grid-1).prod()
if N != len(table.data): errors.append('data count {} does not match grid {}x{}x{}.'.format(N,*(grid - options.mode == 'cell') )) if N != len(table.data): errors.append('data count {} does not match grid {}x{}x{}.'.format(N,*(grid - options.mode == 'cell') ))
if errors != []: if errors != []:
damask.util.croak(errors) damask.util.croak(errors)
table.close(dismiss = True) table.close(dismiss = True)
continue continue
# ------------------------------------------ process data --------------------------------------- # ------------------------------------------ process data ---------------------------------------
rGrid = vtk.vtkRectilinearGrid() rGrid = vtk.vtkRectilinearGrid()
coordArray = [vtk.vtkDoubleArray(), coordArray = [vtk.vtkDoubleArray(),
vtk.vtkDoubleArray(), vtk.vtkDoubleArray(),
vtk.vtkDoubleArray(), vtk.vtkDoubleArray(),
] ]
rGrid.SetDimensions(*grid) rGrid.SetDimensions(*grid)
for i,points in enumerate(coords): for i,points in enumerate(coords):
for point in points: for point in points:
coordArray[i].InsertNextValue(point) coordArray[i].InsertNextValue(point)
rGrid.SetXCoordinates(coordArray[0]) rGrid.SetXCoordinates(coordArray[0])
rGrid.SetYCoordinates(coordArray[1]) rGrid.SetYCoordinates(coordArray[1])
rGrid.SetZCoordinates(coordArray[2]) rGrid.SetZCoordinates(coordArray[2])
# ------------------------------------------ output result --------------------------------------- # ------------------------------------------ output result ---------------------------------------
if name: if name:
writer = vtk.vtkXMLRectilinearGridWriter() writer = vtk.vtkXMLRectilinearGridWriter()
(directory,filename) = os.path.split(name) (directory,filename) = os.path.split(name)
writer.SetDataModeToBinary() writer.SetDataModeToBinary()
writer.SetCompressorTypeToZLib() writer.SetCompressorTypeToZLib()
writer.SetFileName(os.path.join(directory,os.path.splitext(filename)[0]+'_{}'.format(options.mode) \ writer.SetFileName(os.path.join(directory,os.path.splitext(filename)[0] \
+'_{}({})'.format(options.position, options.mode) \
+'.'+writer.GetDefaultFileExtension())) +'.'+writer.GetDefaultFileExtension()))
else: else:
writer = vtk.vtkDataSetWriter() writer = vtk.vtkDataSetWriter()