			#!/usr/bin/env python3
			# -- mode: python; coding: utf-8 --
			# vim: set ft=python:
			"""
			Read Numpy data from standard input, plot contact tractions and displacements.
			"""

			import sys
			import io

			import matplotlib.pyplot as plt
			import numpy as np

			__author__ = "Lucas Frérot"
			__copyright__ = (
			"Copyright (©) 2019-2022, 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)


			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()

tamaas/examples/pipe_tools/plot
96c60334eadbci-fix-static-checks

plot
View Options