Page Menu
Home
c4science
Search
Configure Global Search
Log In
Files
F79553337
truss_fe.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
Tue, Aug 27, 02:09
Size
3 KB
Mime Type
text/x-python
Expires
Thu, Aug 29, 02:09 (2 d)
Engine
blob
Format
Raw Data
Handle
20140075
Attached To
rAKA akantu
truss_fe.py
View Options
# ------------------------------------------------------------------------------
__author__
=
"Nicolas Richart"
__copyright__
=
"Copyright (C) 2016-2018, EPFL (Ecole Polytechnique Fédérale"
\
" de Lausanne) Laboratory (LSMS - Laboratoire de Simulation"
\
" en Mécanique des Solides)"
__credits__
=
[
"Nicolas Richart"
]
__license__
=
"L-GPLv3"
__maintainer__
=
"Nicolas Richart"
__email__
=
"nicolas.richart@epfl.ch"
# ------------------------------------------------------------------------------
import
numpy
as
np
import
copy
import
scipy.sparse
as
sp
from
.
import
export
from
.
import
fe
@export
class
TrussFE
(
fe
.
Model
):
def
__init__
(
self
,
**
kwargs
):
opt
=
copy
.
copy
(
kwargs
)
self
.
_A
=
opt
.
pop
(
"A"
,
1.
)
self
.
_E
=
opt
.
pop
(
"E"
,
1.
)
self
.
_L
=
opt
.
pop
(
"L"
,
1.
)
self
.
_rho
=
opt
.
pop
(
"rho"
,
1.
)
self
.
_F
=
opt
.
pop
(
"F"
,
{
-
10
:
[
-
1
]})
self
.
_blocked
=
opt
.
pop
(
"blocked"
,
([
0
],
[
0
]))
self
.
_Ne
=
opt
.
pop
(
"Ne"
,
1
)
self
.
_Le
=
self
.
_L
/
self
.
_Ne
self
.
_F_ext
=
np
.
zeros
((
self
.
_Ne
+
1
))
self
.
_F_int
=
np
.
zeros
((
self
.
_Ne
+
1
))
self
.
_u
=
np
.
zeros
((
self
.
_Ne
+
1
))
self
.
_v
=
np
.
zeros
((
self
.
_Ne
+
1
))
self
.
_a
=
np
.
zeros
((
self
.
_Ne
+
1
))
self
.
assembleMass
()
def
assembleMass
(
self
):
Me
=
np
.
array
([[
2.
,
1.
],
[
1.
,
2.
]])
Me
*=
self
.
_rho
*
self
.
_A
*
self
.
_Le
/
6
self
.
_M
=
sp
.
lil_matrix
((
self
.
_Ne
+
1
,
self
.
_Ne
+
1
))
for
i
in
range
(
self
.
_Ne
):
self
.
_M
[
i
:
i
+
2
,
i
:
i
+
2
]
+=
Me
self
.
_M
=
self
.
_M
.
tocsr
()
def
assembleStiffness
(
self
):
Ke
=
np
.
array
([[
1.
,
-
1.
],
[
-
1.
,
1.
]])
Ke
*=
self
.
_E
*
self
.
_A
/
self
.
_Le
self
.
_K
=
sp
.
lil_matrix
((
self
.
_Ne
+
1
,
self
.
_Ne
+
1
))
for
i
in
range
(
self
.
_Ne
):
self
.
_K
[
i
:
i
+
2
,
i
:
i
+
2
]
+=
Ke
self
.
_K
=
self
.
_K
.
tocsr
()
def
applyNeumannBC
(
self
):
for
force
,
nodes
in
self
.
_F
.
items
():
self
.
_F_ext
[
nodes
]
=
force
def
applyDirichletBC
(
self
):
for
i
,
b
in
enumerate
(
self
.
_blocked
[
0
]):
self
.
_u
[
b
]
=
self
.
_blocked
[
1
][
i
]
def
computeInternalForces
(
self
):
self
.
_F_int
=
self
.
_K
*
self
.
_u
@property
def
K
(
self
):
self
.
assembleStiffness
()
return
self
.
_K
@property
def
M
(
self
):
return
self
.
_M
@property
def
C
(
self
):
return
None
@property
def
u
(
self
):
self
.
applyDirichletBC
()
return
self
.
_u
@u.setter
def
u
(
self
,
x
):
self
.
_u
=
x
self
.
applyDirichletBC
()
@property
def
v
(
self
):
return
self
.
_v
@v.setter
def
v
(
self
,
v_
):
self
.
_v
=
v_
@property
def
a
(
self
):
return
self
.
_a
@a.setter
def
a
(
self
,
a_
):
self
.
_a
=
a_
@property
def
f_int
(
self
):
self
.
computeInternalForces
()
return
self
.
_F_int
@property
def
f_ext
(
self
):
self
.
applyNeumannBC
()
return
self
.
_F_ext
@f_ext.setter
def
f_ext
(
self
,
f
):
self
.
_F_ext
=
f
@property
def
blocked
(
self
):
return
self
.
_blocked
[
0
]
@property
def
Le
(
self
):
return
self
.
_Le
Event Timeline
Log In to Comment