netcdf-mpi

test_integral_operators.py
View Options

			#!/usr/bin/env python
			# coding: utf-8
			# -----------------------------------------------------------------------------
			# @author Lucas Frérot <lucas.frerot@epfl.ch>
			#
			# @section LICENSE
			#
			# Copyright (©) 2016 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/>.
			# -----------------------------------------------------------------------------
			import tamaas as tm
			import numpy as np
			import matplotlib.pyplot as plt

			model = tm.ModelFactory.createModel(tm.model_type.volume_2d,
			[1.]*3,
			[8]*3)
			displacement = model.getDisplacement()

			engine = tm._tamaas._test_features.Kelvin(model)

			source = np.zeros_like(displacement)
			source[4, 0, 0, 2] = 1

			engine.applyVolumeForcePotential(source, displacement)

			plt.imshow(displacement[4, :, :, 2])
			plt.colorbar()
			plt.show()