Diffusion tamaas (features/materials) Browse

materials

main.py
View Options

			#!/usr/bin/env python3
			# -- mode: python; coding: utf-8 --
			# vim: set ft=python:
			#
			# Copyright (©) 2016-2021 EPFL (École Polytechnique Fédérale de Lausanne),
			# Laboratory (LSMS - Laboratoire de Simulation en Mécanique des Solides)
			#
			# This program is free software: you can redistribute it and/or modify
			# it under the terms of the GNU Affero General Public License as published
			# by the Free Software Foundation, either version 3 of the License, or
			# (at your option) any later version.
			#
			# This program 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 Affero General Public License for more details.
			#
			# You should have received a copy of the GNU Affero General Public License
			# along with this program. If not, see <https://www.gnu.org/licenses/>.

			import sys
			import io
			import time
			import argparse
			import tamaas as tm
			import numpy as np


			__author__ = "Lucas Frérot"
			__copyright__ = (
			"Copyright (©) 2019-2021, EPFL (École Polytechnique Fédérale de Lausanne),"
			"\nLaboratory (LSMS - Laboratoire de Simulation en Mécanique des Solides)"
			)
			__license__ = "SPDX-License-Identifier: AGPL-3.0-or-later"


			def load_stream(stream):
			"""
			Load numpy from binary stream (allows piping)

			Code from
			https://gist.github.com/CMCDragonkai/3c99fd4aabc8278b9e17f50494fcc30a
			"""
			np_magic = stream.read(6)
			# use the sys.stdin.buffer to read binary data
			np_data = stream.read()
			# read it all into an io.BytesIO object
			return io.BytesIO(np_magic + np_data)


			def print_version():
			print(
			f"""Tamaas {tm.__version__}
			{tm.__copyright__}
			Authors: {', '.join(tm.__author__)}
			This is free software; see the source for copying conditions. There is NO
			warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE."""
			)


			def surface(args):
			if args.generator == 'random_phase':
			generator = tm.SurfaceGeneratorRandomPhase2D(args.sizes)
			elif args.generator == 'filter':
			generator = tm.SurfaceGeneratorFilter2D(args.sizes)
			else:
			raise ValueError('Unknown generator method {}'.format(args.generator))

			generator.spectrum = tm.Isopowerlaw2D()
			generator.spectrum.q0 = args.cutoffs[0]
			generator.spectrum.q1 = args.cutoffs[1]
			generator.spectrum.q2 = args.cutoffs[2]
			generator.spectrum.hurst = args.hurst
			generator.random_seed = args.seed

			surface = generator.buildSurface() / generator.spectrum.rmsSlopes() \
			* args.rms

			output = args.output if args.output is not None else sys.stdout

			params = {
			'q0': generator.spectrum.q0,
			'q1': generator.spectrum.q1,
			'q2': generator.spectrum.q2,
			'hurst': generator.spectrum.hurst,
			'random_seed': generator.random_seed,
			'rms_heights': args.rms,
			'generator': args.generator,
			}

			try:
			np.savetxt(output, surface, header=str(params))
			except BrokenPipeError:
			pass


			def contact(args):
			from tamaas.dumpers import NumpyDumper

			tm.set_log_level(tm.LogLevel.error)

			if not args.input:
			input = sys.stdin
			else:
			input = args.input

			surface = np.loadtxt(input)

			discretization = surface.shape
			system_size = [1., 1.]

			model = tm.ModelFactory.createModel(tm.model_type.basic_2d,
			system_size, discretization)
			solver = tm.PolonskyKeerRey(model, surface, args.tol)

			solver.solve(args.load)

			dumper = NumpyDumper('numpy', 'traction', 'displacement')
			dumper.dump_to_file(sys.stdout.buffer, model)


			def plot(args):
			import matplotlib.pyplot as plt

			fig, (ax_traction, ax_displacement) = plt.subplots(1, 2)

			ax_traction.set_title('Traction')
			ax_displacement.set_title('Displacement')

			with load_stream(sys.stdin.buffer) as f_np:
			data = np.load(f_np)
			ax_traction.imshow(data['traction'])
			ax_displacement.imshow(data['displacement'])

			fig.set_size_inches(10, 6)
			fig.tight_layout()

			plt.show()


			def main():
			parser = argparse.ArgumentParser(
			prog='tamaas',
			description=("The tamaas command is a simple utility for surface"
			" generation, contact computation and"
			" plotting of contact solutions"),
			)

			parser.add_argument('--version', action='store_true',
			help='print version info')

			subs = parser.add_subparsers(title='commands',
			description='utility commands')

			# Arguments for surface command
			parser_surface = subs.add_parser(
			'surface', description='Generate a self-affine rough surface')
			parser_surface.add_argument("--cutoffs", "-K",
			nargs=3,
			type=int,
			help="Long, rolloff, short wavelength cutoffs",
			metavar=('k_l', 'k_r', 'k_s'),
			required=True)
			parser_surface.add_argument("--sizes",
			nargs=2,
			type=int,
			help="Number of points",
			metavar=('nx', 'ny'),
			required=True)
			parser_surface.add_argument("--hurst", "-H",
			type=float,
			help="Hurst exponent",
			required=True)
			parser_surface.add_argument("--rms",
			type=float,
			help="Root-mean-square of slopes",
			default=1.)
			parser_surface.add_argument("--seed",
			type=int,
			help="Random seed",
			default=int(time.time()))
			parser_surface.add_argument("--generator",
			help="Generation method",
			choices=('random_phase', 'filter'),
			default='random_phase')
			parser_surface.add_argument("--output", "-o",
			help="Output file name (compressed if .gz)")
			parser_surface.set_defaults(func=surface)

			# Arguments for contact command
			parser_contact = subs.add_parser(
			'contact',
			description="Compute the elastic contact solution with a given surface")

			parser_contact.add_argument("--input", "-i",
			help="Rough surface file (default stdin)")
			parser_contact.add_argument("--tol",
			type=float,
			default=1e-12,
			help="Solver tolerance")
			parser_contact.add_argument("load",
			type=float,
			help="Applied average pressure")
			parser_contact.set_defaults(func=contact)

			# Arguments for plot command
			parser_plot = subs.add_parser(
			'plot', description='Plot contact solution')
			parser_plot.set_defaults(func=plot)

			args = parser.parse_args()

			if args.version:
			print_version()
			return

			try:
			args.func(args)
			except AttributeError:
			parser.print_usage()


			if __name__ == '__main__':
			main()

tamaas/python/tamaas/__main__.py26424b111ba5features/materials

__main__.pyView Options

tamaas/python/tamaas/main.py
26424b111ba5features/materials

main.py
View Options