Page Menu
Home
c4science
Search
Configure Global Search
Log In
Files
F67759920
test_integral_operators.py
No One
Temporary
Actions
Download File
Edit File
Delete File
View Transforms
Subscribe
Mute Notifications
Award Token
Subscribers
None
File Metadata
Details
File Info
Storage
Attached
Created
Mon, Jun 24, 04:50
Size
2 KB
Mime Type
text/x-python
Expires
Wed, Jun 26, 04:50 (2 d)
Engine
blob
Format
Raw Data
Handle
18449734
Attached To
rTAMAAS tamaas
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
from
pyevtk.hl
import
gridToVTK
def
dumpGrid
(
name
,
coords
,
field
,
func
=
lambda
x
:
x
):
field_vtk
=
[
np
.
zeros
([
coords
[
i
]
.
shape
[
0
]
for
i
in
range
(
3
)])
for
j
in
range
(
3
)]
for
i
in
range
(
3
):
field_vtk
[
i
][:,
:,
:]
=
func
(
field
[:,
:,
:,
i
])
gridToVTK
(
name
,
*
coords
,
pointData
=
{
'X'
:
field_vtk
[
0
],
'Y'
:
field_vtk
[
1
],
'Z'
:
field_vtk
[
2
],
})
N
=
27
domain
=
np
.
array
([
1.
]
*
3
)
omega
=
2
*
np
.
pi
*
np
.
array
([
1
,
1
])
/
domain
[:
2
]
discretization
=
[
N
]
*
3
coords
=
[
np
.
linspace
(
0
,
domain
[
i
],
discretization
[
i
])
for
i
in
range
(
3
)]
x
=
coords
[
1
]
y
=
coords
[
2
]
x
,
y
=
np
.
meshgrid
(
x
,
y
)
model
=
tm
.
ModelFactory
.
createModel
(
tm
.
model_type
.
volume_2d
,
domain
,
discretization
)
displacement
=
model
.
getDisplacement
()
engine
=
tm
.
_tamaas
.
_test_features
.
Kelvin
(
model
)
source
=
np
.
zeros_like
(
displacement
)
source
[
N
//
2
,
:,
:,
2
]
=
np
.
sin
(
omega
[
0
]
*
x
)
*
np
.
sin
(
omega
[
1
]
*
y
)
print
(
"Starting computation"
)
engine
.
applyVolumeForcePotential
(
source
,
displacement
)
print
(
"End computation"
)
dumpGrid
(
'kelvin'
,
coords
,
displacement
,
lambda
x
:
np
.
einsum
(
'ijk->jki'
,
x
))
dumpGrid
(
'source'
,
coords
,
source
,
lambda
x
:
np
.
einsum
(
'ijk->jki'
,
x
))
plt
.
imshow
(
displacement
[:,
4
,
:,
2
])
plt
.
colorbar
()
#plt.show()
Event Timeline
Log In to Comment