Page Menu
Home
c4science
Search
Configure Global Search
Log In
Files
F85184214
Element.hpp
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
Fri, Sep 27, 08:54
Size
3 KB
Mime Type
text/x-c
Expires
Sun, Sep 29, 08:54 (2 d)
Engine
blob
Format
Raw Data
Handle
21139454
Attached To
rGOOSEFEM GooseFEM
Element.hpp
View Options
/* =================================================================================================
(c - GPLv3) T.W.J. de Geus (Tom) | tom@geus.me | www.geus.me | github.com/tdegeus/GooseFEM
================================================================================================= */
#ifndef GOOSEFEM_ELEMENT_CPP
#define GOOSEFEM_ELEMENT_CPP
// -------------------------------------------------------------------------------------------------
#include "Element.h"
// ======================================= GooseFEM::Element =======================================
namespace
GooseFEM
{
namespace
Element
{
// -------------------------------------------------------------------------------------------------
inline
ArrD
asElementVector
(
const
xt
::
xtensor
<
size_t
,
2
>
&
conn
,
const
xt
::
xtensor
<
double
,
2
>
&
nodevec
)
{
// extract dimensions
size_t
nelem
=
static_cast
<
size_t
>
(
conn
.
rows
());
size_t
nne
=
static_cast
<
size_t
>
(
conn
.
cols
());
size_t
ndim
=
static_cast
<
size_t
>
(
nodevec
.
cols
());
// allocate output: nodal vectors stored per element
ArrD
elemvec
=
ArrD
::
Zero
({
nelem
,
nne
,
ndim
});
// read from nodal vectors
#pragma omp parallel for
for
(
size_t
e
=
0
;
e
<
nelem
;
++
e
)
for
(
size_t
m
=
0
;
m
<
nne
;
++
m
)
for
(
size_t
i
=
0
;
i
<
ndim
;
++
i
)
elemvec
(
e
,
m
,
i
)
=
nodevec
(
conn
(
e
,
m
),
i
);
return
elemvec
;
}
// -------------------------------------------------------------------------------------------------
inline
xt
::
xtensor
<
double
,
2
>
assembleNodeVector
(
const
xt
::
xtensor
<
size_t
,
2
>
&
conn
,
const
ArrD
&
elemvec
)
{
// check input
assert
(
elemvec
.
rank
()
==
3
);
// nodal vector stored per element [nelem, nne, ndim]
// extract dimensions
size_t
nelem
=
static_cast
<
size_t
>
(
conn
.
rows
());
// number of elements
size_t
nne
=
static_cast
<
size_t
>
(
conn
.
cols
());
// number of nodes per element
size_t
ndim
=
elemvec
.
shape
(
2
);
// number of dimensions
size_t
nnode
=
conn
.
maxCoeff
()
+
1
;
// number of nodes
// check input
assert
(
elemvec
.
shape
(
0
)
==
nelem
);
assert
(
elemvec
.
shape
(
1
)
==
nne
);
// zero-initialize output: nodal vectors
xt
::
xtensor
<
double
,
2
>
nodevec
=
xt
::
xtensor
<
double
,
2
>::
Zero
(
nnode
,
ndim
);
// temporarily disable parallelization by Eigen
Eigen
::
setNbThreads
(
1
);
// start threads (all variables declared in this scope are local to each thread)
#pragma omp parallel
{
// zero-initialize output: nodal vectors
xt
::
xtensor
<
double
,
2
>
t_nodevec
=
xt
::
xtensor
<
double
,
2
>::
Zero
(
nnode
,
ndim
);
// assemble from nodal vectors stored per element
#pragma omp for
for
(
size_t
e
=
0
;
e
<
nelem
;
++
e
)
for
(
size_t
m
=
0
;
m
<
nne
;
++
m
)
for
(
size_t
i
=
0
;
i
<
ndim
;
++
i
)
t_nodevec
(
conn
(
e
,
m
),
i
)
+=
elemvec
(
e
,
m
,
i
);
// reduce: combine result obtained on the different threads
#pragma omp critical
nodevec
+=
t_nodevec
;
}
// reset automatic parallelization by Eigen
Eigen
::
setNbThreads
(
0
);
return
nodevec
;
}
// -------------------------------------------------------------------------------------------------
}}
// namespace ...
// =================================================================================================
#endif
Event Timeline
Log In to Comment