Homec4science
Diffusion tamaas (ci-fix-static-checks) Browse

tamaas/python/tamaas/dumpers/uvw_dumper.py
a7407197cf9bci-fix-static-checks

Last Change
Raw File

uvw_dumper.py
View Options

"""
@author Lucas Frérot <lucas.frerot@epfl.ch>
@section LICENSE
Copyright (©) 2018 EPFL (Ecole Polytechnique Fédérale de
Lausanne) Laboratory (LSMS - Laboratoire de Simulation en Mécanique des
Solides)
Tamaas is free software: you can redistribute it and/or modify it under the
terms of the GNU Lesser General Public License as published by the Free
Software Foundation, either version 3 of the License, or (at your option) any
later version.
Tamaas is distributed in the hope that it will be useful, but WITHOUT ANY
WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
details.
You should have received a copy of the GNU Lesser General Public License
along with Tamaas. If not, see <http://www.gnu.org/licenses/>.
"""
from ._helper import FieldDumper, step_dump, directory_dump
import os
import uvw
import numpy as np
__all__ = ["UVWDumper"]
@directory_dump('paraview')
@step_dump
class UVWDumper(FieldDumper):
"""Dumper for elasto-plastic calculations"""
def __init__(self, basename, *fields):
default_fields = ['displacement', 'stress', 'plastic_strain']
super().__init__(basename, default_fields, *fields)
def dump(self, model):
"""Dump displacements, plastic deformations and stresses"""
discretization = model.getDiscretization().copy()
discretization[1] += 1
discretization[2] += 1
coordinates = [np.linspace(0, L, N)
for L, N in zip(model.getSystemSize(),
discretization)]
# Correct order of coordinate dimensions
dimension_indices = [1, 2, 0]
# Creating rectilinear grid with correct order for components
grid = uvw.RectilinearGrid(self.file_path(),
(coordinates[i] for i in dimension_indices))
dump_fields = {
field:
uvw.DataArray(self.make_periodic[field](model.getField(field)),
dimension_indices, field)
for field in self.fields
}
for array in dump_fields.values():
grid.addPointData(array)
grid.write()
def file_path(self):
return os.path.join(self.dirname(), "{}{}.vtr".format(self.basename,
self.postfix()))
c4science · Help