_helper.py
fdb103752c5eci-fix-static-checks

_helper.py
View Options

			from .. import ModelDumper

			import numpy as np

			__all__ = ["makePeriodic", "makeTractionPeriodic"]


			def makePeriodic(data):
			data = np.append(data, np.expand_dims(data[:, 0, ...], axis=1), axis=1)
			data = np.append(data, np.expand_dims(data[:, :, 0, ...], axis=2), axis=2)
			return data


			def makeTractionPeriodic(data):
			data = np.append(data, np.expand_dims(data[0, ...], axis=0), axis=0)
			data = np.append(data, np.expand_dims(data[:, 0, ...], axis=1), axis=1)
			return data


			class FieldDumper(ModelDumper):
			"""Abstract dumper for python classes using fields"""
			def __init__(self, basename, default_fields, *fields):
			super().__init__()
			self.basename = basename

			if len(fields):
			self.fields = fields
			else:
			self.fields = default_fields

			self.make_periodic = {field: makePeriodic
			for field in self.fields if field != 'traction'}
			self.make_periodic['traction'] = makeTractionPeriodic