Page Menu
Home
c4science
Search
Configure Global Search
Log In
Files
F93534881
mesh_utils.cc
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, Nov 29, 13:03
Size
27 KB
Mime Type
text/x-c
Expires
Sun, Dec 1, 13:03 (1 d, 21 h)
Engine
blob
Format
Raw Data
Handle
22661259
Attached To
rAKA akantu
mesh_utils.cc
View Options
/**
* Copyright (©) 2010-2023 EPFL (Ecole Polytechnique Fédérale de Lausanne)
* Laboratory (LSMS - Laboratoire de Simulation en Mécanique des Solides)
*
* This file is part of Akantu
*
* Akantu 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.
*
* Akantu 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 Akantu. If not, see <http://www.gnu.org/licenses/>.
*/
/* -------------------------------------------------------------------------- */
#include "mesh_utils.hh"
#include "element_synchronizer.hh"
#include "fe_engine.hh"
#include "mesh_accessor.hh"
#include "mesh_iterators.hh"
/* -------------------------------------------------------------------------- */
#include <limits>
#include <numeric>
#include <queue>
#include <set>
/* -------------------------------------------------------------------------- */
namespace
akantu
{
/* -------------------------------------------------------------------------- */
void
MeshUtils
::
buildNode2Elements
(
const
Mesh
&
mesh
,
CSR
<
Element
>
&
node_to_elem
,
Int
spatial_dimension
)
{
AKANTU_DEBUG_IN
();
if
(
spatial_dimension
==
_all_dimensions
)
{
spatial_dimension
=
mesh
.
getSpatialDimension
();
}
/// count number of occurrence of each node
auto
nb_nodes
=
mesh
.
getNbNodes
();
/// array for the node-element list
node_to_elem
.
resizeRows
(
nb_nodes
);
node_to_elem
.
clearRows
();
for_each_element
(
mesh
,
[
&
](
auto
&&
element
)
{
auto
conn
=
mesh
.
getConnectivity
(
element
);
for
(
auto
&&
node
:
conn
)
{
++
node_to_elem
.
rowOffset
(
node
);
}
},
_spatial_dimension
=
spatial_dimension
,
_element_kind
=
_ek_not_defined
);
node_to_elem
.
countToCSR
();
node_to_elem
.
resizeCols
();
/// rearrange element to get the node-element list
// Element e;
node_to_elem
.
beginInsertions
();
for_each_element
(
mesh
,
[
&
](
auto
&&
element
)
{
auto
conn
=
mesh
.
getConnectivity
(
element
);
for
(
auto
&&
node
:
conn
)
{
node_to_elem
.
insertInRow
(
node
,
element
);
}
},
_spatial_dimension
=
spatial_dimension
,
_element_kind
=
_ek_not_defined
);
node_to_elem
.
endInsertions
();
AKANTU_DEBUG_OUT
();
}
/* -------------------------------------------------------------------------- */
void
MeshUtils
::
buildNode2ElementsElementTypeMap
(
const
Mesh
&
mesh
,
CSR
<
Idx
>
&
node_to_elem
,
ElementType
type
,
GhostType
ghost_type
)
{
AKANTU_DEBUG_IN
();
auto
nb_nodes
=
mesh
.
getNbNodes
();
auto
nb_nodes_per_element
=
Mesh
::
getNbNodesPerElement
(
type
);
auto
nb_elements
=
mesh
.
getConnectivity
(
type
,
ghost_type
).
size
();
auto
*
conn_val
=
mesh
.
getConnectivity
(
type
,
ghost_type
).
data
();
/// array for the node-element list
node_to_elem
.
resizeRows
(
nb_nodes
);
node_to_elem
.
clearRows
();
/// count number of occurrence of each node
for
(
Int
el
=
0
;
el
<
nb_elements
;
++
el
)
{
auto
el_offset
=
el
*
nb_nodes_per_element
;
for
(
Int
n
=
0
;
n
<
nb_nodes_per_element
;
++
n
)
{
++
node_to_elem
.
rowOffset
(
conn_val
[
el_offset
+
n
]);
}
}
/// convert the occurrence array in a csr one
node_to_elem
.
countToCSR
();
node_to_elem
.
resizeCols
();
node_to_elem
.
beginInsertions
();
/// save the element index in the node-element list
for
(
Int
el
=
0
;
el
<
nb_elements
;
++
el
)
{
auto
el_offset
=
el
*
nb_nodes_per_element
;
for
(
Int
n
=
0
;
n
<
nb_nodes_per_element
;
++
n
)
{
node_to_elem
.
insertInRow
(
conn_val
[
el_offset
+
n
],
el
);
}
}
node_to_elem
.
endInsertions
();
AKANTU_DEBUG_OUT
();
}
/* -------------------------------------------------------------------------- */
void
MeshUtils
::
buildFacets
(
Mesh
&
mesh
)
{
AKANTU_DEBUG_IN
();
auto
spatial_dimension
=
mesh
.
getSpatialDimension
();
for
(
auto
ghost_type
:
ghost_types
)
{
for
(
const
auto
&
type
:
mesh
.
elementTypes
(
spatial_dimension
-
1
,
ghost_type
))
{
mesh
.
getConnectivity
(
type
,
ghost_type
).
resize
(
0
);
// \todo inform the mesh event handler
}
}
buildFacetsDimension
(
mesh
,
mesh
,
true
,
spatial_dimension
);
AKANTU_DEBUG_OUT
();
}
/* -------------------------------------------------------------------------- */
void
MeshUtils
::
buildAllFacets
(
const
Mesh
&
mesh
,
Mesh
&
mesh_facets
,
Int
to_dimension
)
{
AKANTU_DEBUG_IN
();
auto
spatial_dimension
=
mesh
.
getSpatialDimension
();
buildAllFacets
(
mesh
,
mesh_facets
,
spatial_dimension
,
to_dimension
);
AKANTU_DEBUG_OUT
();
}
/* -------------------------------------------------------------------------- */
void
MeshUtils
::
buildAllFacets
(
const
Mesh
&
mesh
,
Mesh
&
mesh_facets
,
Int
from_dimension
,
Int
to_dimension
)
{
AKANTU_DEBUG_IN
();
to_dimension
=
std
::
max
(
to_dimension
,
Int
(
0
));
AKANTU_DEBUG_ASSERT
(
mesh_facets
.
isMeshFacets
(),
"The mesh_facets should be initialized with initMeshFacets"
);
/// generate facets
buildFacetsDimension
(
mesh
,
mesh_facets
,
false
,
from_dimension
);
/// sort facets and generate sub-facets
for
(
Int
i
=
from_dimension
-
1
;
i
>
to_dimension
;
--
i
)
{
buildFacetsDimension
(
mesh_facets
,
mesh_facets
,
false
,
i
);
}
AKANTU_DEBUG_OUT
();
}
/* -------------------------------------------------------------------------- */
void
MeshUtils
::
buildFacetsDimension
(
const
Mesh
&
mesh
,
Mesh
&
mesh_facets
,
bool
boundary_only
,
Int
dimension
)
{
AKANTU_DEBUG_IN
();
// save the current parent of mesh_facets and set it temporarly to mesh since
// mesh is the one containing the elements for which mesh_facets has the
// sub-elements
// example: if the function is called with mesh = mesh_facets
const
Mesh
*
mesh_facets_parent
=
nullptr
;
try
{
mesh_facets_parent
=
&
mesh_facets
.
getMeshParent
();
}
catch
(...)
{
}
mesh_facets
.
defineMeshParent
(
mesh
);
MeshAccessor
mesh_accessor
(
mesh_facets
);
auto
spatial_dimension
=
mesh
.
getSpatialDimension
();
const
Array
<
Real
>
&
mesh_facets_nodes
=
mesh_facets
.
getNodes
();
const
auto
mesh_facets_nodes_it
=
mesh_facets_nodes
.
begin
(
spatial_dimension
);
CSR
<
Element
>
node_to_elem
;
buildNode2Elements
(
mesh
,
node_to_elem
,
dimension
);
Array
<
Int
>
counter
;
std
::
vector
<
Element
>
connected_elements
;
NewElementsEvent
event
(
AKANTU_CURRENT_FUNCTION
);
// init the SubelementToElement data to improve performance
for
(
auto
&&
ghost_type
:
ghost_types
)
{
for
(
auto
&&
type
:
mesh
.
elementTypes
(
dimension
,
ghost_type
))
{
auto
&
subelement_to_element
=
mesh_accessor
.
getSubelementToElement
(
type
,
ghost_type
);
subelement_to_element
.
resize
(
mesh
.
getNbElement
(
type
,
ghost_type
),
ElementNull
);
auto
facet_types
=
mesh
.
getAllFacetTypes
(
type
);
for
(
auto
&&
facet_type
:
facet_types
)
{
mesh_accessor
.
getElementToSubelement
(
facet_type
,
ghost_type
);
mesh_accessor
.
getConnectivity
(
facet_type
,
ghost_type
);
}
}
}
const
ElementSynchronizer
*
synchronizer
=
nullptr
;
if
(
mesh
.
isDistributed
())
{
synchronizer
=
&
(
mesh
.
getElementSynchronizer
());
}
Element
current_element
;
for
(
auto
&&
ghost_type
:
ghost_types
)
{
GhostType
facet_ghost_type
=
ghost_type
;
current_element
.
ghost_type
=
ghost_type
;
for
(
auto
&&
type
:
mesh
.
elementTypes
(
dimension
,
ghost_type
))
{
auto
facet_types
=
mesh
.
getAllFacetTypes
(
type
);
current_element
.
type
=
type
;
for
(
auto
&&
ft
:
arange
(
facet_types
.
size
()))
{
auto
facet_type
=
facet_types
(
ft
);
auto
nb_element
=
mesh
.
getNbElement
(
type
,
ghost_type
);
auto
&&
element_to_subelement
=
&
mesh_accessor
.
getElementToSubelementNC
(
facet_type
,
ghost_type
);
auto
&&
connectivity_facets
=
&
mesh_accessor
.
getConnectivity
(
facet_type
,
ghost_type
);
auto
nb_nodes_per_facet
=
connectivity_facets
->
getNbComponent
();
// Vector<UInt> facet(nb_nodes_per_facet);
for
(
Int
el
=
0
;
el
<
nb_element
;
++
el
)
{
current_element
.
element
=
el
;
Matrix
<
Idx
>
facets
=
mesh
.
getFacetConnectivity
(
current_element
,
ft
).
transpose
();
for
(
auto
&
facet
:
facets
)
{
// facet = facets(f);
auto
first_node_nb_elements
=
node_to_elem
.
getNbCols
(
facet
(
0
));
counter
.
resize
(
first_node_nb_elements
);
counter
.
zero
();
// loop over the other nodes to search intersecting elements,
// which are the elements that share another node with the
// starting element after first_node
for
(
auto
&&
data
:
enumerate
(
node_to_elem
.
getRow
(
facet
(
0
))))
{
auto
&&
local_el
=
std
::
get
<
0
>
(
data
);
auto
&&
first_node
=
std
::
get
<
1
>
(
data
);
for
(
auto
n
:
arange
(
1
,
nb_nodes_per_facet
))
{
auto
&&
node_elements
=
node_to_elem
.
getRow
(
facet
(
n
));
counter
(
local_el
)
+=
std
::
count
(
node_elements
.
begin
(),
node_elements
.
end
(),
first_node
);
}
}
// counting the number of elements connected to the facets and
// taking the minimum element number, because the facet should
// be inserted just once
auto
nb_element_connected_to_facet
=
0
;
auto
minimum_el
=
ElementNull
;
connected_elements
.
clear
();
for
(
auto
&&
data
:
enumerate
(
node_to_elem
.
getRow
(
facet
(
0
))))
{
if
(
not
(
counter
(
std
::
get
<
0
>
(
data
))
==
nb_nodes_per_facet
-
1
))
{
continue
;
}
auto
&&
real_el
=
std
::
get
<
1
>
(
data
);
++
nb_element_connected_to_facet
;
minimum_el
=
std
::
min
(
minimum_el
,
real_el
);
connected_elements
.
push_back
(
real_el
);
}
if
(
minimum_el
!=
current_element
)
{
continue
;
}
bool
full_ghost_facet
=
false
;
Int
n
=
0
;
while
(
n
<
nb_nodes_per_facet
and
mesh
.
isPureGhostNode
(
facet
(
n
)))
{
++
n
;
}
if
(
n
==
nb_nodes_per_facet
)
{
full_ghost_facet
=
true
;
}
if
(
full_ghost_facet
)
{
continue
;
}
if
(
boundary_only
and
nb_element_connected_to_facet
!=
1
)
{
continue
;
}
std
::
vector
<
Element
>
elements
;
// build elements_on_facets: linearized_el must come first
// in order to store the facet in the correct direction
// and avoid to invert the sign in the normal computation
elements
.
reserve
(
elements
.
size
()
+
connected_elements
.
size
());
for
(
auto
&&
connected_element
:
connected_elements
)
{
elements
.
push_back
(
connected_element
);
}
if
(
nb_element_connected_to_facet
==
1
)
{
/// boundary facet
elements
.
push_back
(
ElementNull
);
}
else
if
(
nb_element_connected_to_facet
==
2
)
{
/// internal facet
/// check if facet is in between ghost and normal
/// elements: if it's the case, the facet is either
/// ghost or not ghost. The criterion to decide this
/// is arbitrary. It was chosen to check the processor
/// id (prank) of the two neighboring elements. If
/// prank of the ghost element is lower than prank of
/// the normal one, the facet is not ghost, otherwise
/// it's ghost
GhostType
gt
[
2
]
=
{
_not_ghost
,
_not_ghost
};
for
(
auto
&&
data
:
enumerate
(
connected_elements
))
{
gt
[
std
::
get
<
0
>
(
data
)]
=
std
::
get
<
1
>
(
data
).
ghost_type
;
}
if
((
gt
[
0
]
==
_not_ghost
)
xor
(
gt
[
1
]
==
_not_ghost
))
{
Int
prank
[
2
];
for
(
Int
el
=
0
;
el
<
2
;
++
el
)
{
prank
[
el
]
=
synchronizer
->
getRank
(
connected_elements
[
el
]);
}
// ugly trick from Marco detected :P
bool
ghost_one
=
(
gt
[
0
]
!=
_ghost
);
if
(
prank
[
ghost_one
]
>
prank
[
!
ghost_one
])
{
facet_ghost_type
=
_not_ghost
;
}
else
{
facet_ghost_type
=
_ghost
;
}
connectivity_facets
=
&
mesh_accessor
.
getConnectivity
(
facet_type
,
facet_ghost_type
);
element_to_subelement
=
&
mesh_accessor
.
getElementToSubelementNC
(
facet_type
,
facet_ghost_type
);
}
}
element_to_subelement
->
push_back
(
elements
);
connectivity_facets
->
push_back
(
facet
);
/// current facet index
auto
current_facet
=
connectivity_facets
->
size
()
-
1
;
Element
facet_element
{
facet_type
,
current_facet
,
facet_ghost_type
};
event
.
getList
().
push_back
(
facet_element
);
/// loop on every element connected to current facet and
/// insert current facet in the first free spot of the
/// subelement_to_element vector
for
(
auto
&
loc_el
:
elements
)
{
if
(
loc_el
==
ElementNull
)
{
continue
;
}
auto
&&
subelements
=
mesh_accessor
.
getSubelementToElement
(
loc_el
);
for
(
auto
&
el
:
subelements
)
{
if
(
el
!=
ElementNull
)
{
continue
;
}
el
=
facet_element
;
break
;
}
}
/// reset connectivity in case a facet was found in
/// between ghost and normal elements
if
(
facet_ghost_type
!=
ghost_type
)
{
facet_ghost_type
=
ghost_type
;
connectivity_facets
=
&
mesh_accessor
.
getConnectivity
(
facet_type
,
facet_ghost_type
);
element_to_subelement
=
&
mesh_accessor
.
getElementToSubelement
(
facet_type
,
facet_ghost_type
);
}
}
}
}
}
}
mesh_facets
.
sendEvent
(
event
);
// restore the parent of mesh_facet
if
(
mesh_facets_parent
!=
nullptr
)
{
mesh_facets
.
defineMeshParent
(
*
mesh_facets_parent
);
}
AKANTU_DEBUG_OUT
();
}
/* -------------------------------------------------------------------------- */
void
MeshUtils
::
renumberMeshNodes
(
Mesh
&
mesh
,
Array
<
Idx
>
&
local_connectivities
,
Int
nb_local_element
,
Int
nb_ghost_element
,
ElementType
type
,
Array
<
Idx
>
&
old_nodes_numbers
)
{
AKANTU_DEBUG_IN
();
auto
nb_nodes_per_element
=
Mesh
::
getNbNodesPerElement
(
type
);
std
::
map
<
Idx
,
Idx
>
renumbering_map
;
for
(
Int
i
=
0
;
i
<
old_nodes_numbers
.
size
();
++
i
)
{
renumbering_map
[
old_nodes_numbers
(
i
)]
=
i
;
}
/// renumber the nodes
renumberNodesInConnectivity
(
local_connectivities
,
(
nb_local_element
+
nb_ghost_element
)
*
nb_nodes_per_element
,
renumbering_map
);
old_nodes_numbers
.
resize
(
renumbering_map
.
size
());
for
(
auto
&
renumber_pair
:
renumbering_map
)
{
old_nodes_numbers
(
renumber_pair
.
second
)
=
renumber_pair
.
first
;
}
renumbering_map
.
clear
();
MeshAccessor
mesh_accessor
(
mesh
);
/// copy the renumbered connectivity to the right place
auto
&
local_conn
=
mesh_accessor
.
getConnectivity
(
type
);
local_conn
.
resize
(
nb_local_element
);
if
(
nb_local_element
>
0
)
{
std
::
copy_n
(
local_connectivities
.
data
(),
nb_local_element
*
nb_nodes_per_element
,
local_conn
.
data
());
}
auto
&
ghost_conn
=
mesh_accessor
.
getConnectivity
(
type
,
_ghost
);
ghost_conn
.
resize
(
nb_ghost_element
);
if
(
nb_ghost_element
>
0
)
{
std
::
copy_n
(
local_connectivities
.
data
()
+
nb_local_element
*
nb_nodes_per_element
,
nb_ghost_element
*
nb_nodes_per_element
,
ghost_conn
.
data
());
}
AKANTU_DEBUG_OUT
();
}
/* -------------------------------------------------------------------------- */
void
MeshUtils
::
renumberNodesInConnectivity
(
Array
<
Idx
>
&
list_nodes
,
Int
nb_nodes
,
std
::
map
<
Idx
,
Idx
>
&
renumbering_map
)
{
AKANTU_DEBUG_IN
();
auto
*
connectivity
=
list_nodes
.
data
();
auto
new_node_num
=
renumbering_map
.
size
();
for
(
Int
n
=
0
;
n
<
nb_nodes
;
++
n
,
++
connectivity
)
{
auto
&
node
=
*
connectivity
;
auto
it
=
renumbering_map
.
find
(
node
);
if
(
it
==
renumbering_map
.
end
())
{
auto
old_node
=
node
;
renumbering_map
[
old_node
]
=
new_node_num
;
node
=
new_node_num
;
++
new_node_num
;
}
else
{
node
=
it
->
second
;
}
}
AKANTU_DEBUG_OUT
();
}
/* -------------------------------------------------------------------------- */
void
MeshUtils
::
purifyMesh
(
Mesh
&
mesh
)
{
AKANTU_DEBUG_IN
();
std
::
map
<
Idx
,
Idx
>
renumbering_map
;
RemovedNodesEvent
remove_nodes
(
mesh
,
AKANTU_CURRENT_FUNCTION
);
auto
&
nodes_removed
=
remove_nodes
.
getList
();
for
(
auto
ghost_type
:
ghost_types
)
{
for
(
auto
type
:
mesh
.
elementTypes
(
_all_dimensions
,
ghost_type
,
_ek_not_defined
))
{
auto
nb_nodes_per_element
=
Mesh
::
getNbNodesPerElement
(
type
);
auto
&
connectivity
=
mesh
.
getConnectivity
(
type
,
ghost_type
);
auto
nb_element
(
connectivity
.
size
());
renumberNodesInConnectivity
(
connectivity
,
nb_element
*
nb_nodes_per_element
,
renumbering_map
);
}
}
auto
&
new_numbering
=
remove_nodes
.
getNewNumbering
();
std
::
fill
(
new_numbering
.
begin
(),
new_numbering
.
end
(),
Int
(
-
1
));
for
(
auto
&&
pair
:
renumbering_map
)
{
new_numbering
(
std
::
get
<
0
>
(
pair
))
=
std
::
get
<
1
>
(
pair
);
}
for
(
Int
i
=
0
;
i
<
new_numbering
.
size
();
++
i
)
{
if
(
new_numbering
(
i
)
==
Int
(
-
1
))
{
nodes_removed
.
push_back
(
i
);
}
}
mesh
.
sendEvent
(
remove_nodes
);
AKANTU_DEBUG_OUT
();
}
/* -------------------------------------------------------------------------- */
void
MeshUtils
::
flipFacets
(
Mesh
&
mesh_facets
,
const
ElementTypeMapArray
<
Idx
>
&
remote_global_connectivities
,
GhostType
gt_facet
)
{
AKANTU_DEBUG_IN
();
auto
spatial_dimension
=
mesh_facets
.
getSpatialDimension
();
/// get global connectivity for local mesh
ElementTypeMapArray
<
Idx
>
local_global_connectivities
(
"local_global_connectivity"
,
mesh_facets
.
getID
());
local_global_connectivities
.
initialize
(
mesh_facets
,
_spatial_dimension
=
spatial_dimension
-
1
,
_ghost_type
=
gt_facet
,
_with_nb_nodes_per_element
=
true
,
_with_nb_element
=
true
);
mesh_facets
.
getGlobalConnectivity
(
local_global_connectivities
);
MeshAccessor
mesh_accessor
(
mesh_facets
);
/// loop on every facet
for
(
auto
type_facet
:
mesh_facets
.
elementTypes
(
spatial_dimension
-
1
,
gt_facet
))
{
auto
&
connectivity
=
mesh_accessor
.
getConnectivity
(
type_facet
,
gt_facet
);
auto
&
local_global_connectivity
=
local_global_connectivities
(
type_facet
,
gt_facet
);
const
auto
&
remote_global_connectivity
=
remote_global_connectivities
(
type_facet
,
gt_facet
);
auto
&
element_per_facet
=
mesh_accessor
.
getElementToSubelementNC
(
type_facet
,
gt_facet
);
auto
&
subfacet_to_facet
=
mesh_accessor
.
getSubelementToElementNC
(
type_facet
,
gt_facet
);
auto
nb_nodes_per_facet
=
connectivity
.
getNbComponent
();
auto
nb_nodes_per_P1_facet
=
Mesh
::
getNbNodesPerElement
(
Mesh
::
getP1ElementType
(
type_facet
));
for
(
auto
&&
data
:
zip
(
make_view
(
connectivity
,
nb_nodes_per_facet
),
make_view
(
local_global_connectivity
,
nb_nodes_per_facet
),
make_view
(
remote_global_connectivity
,
nb_nodes_per_facet
),
make_view
(
subfacet_to_facet
,
subfacet_to_facet
.
getNbComponent
()),
make_view
(
element_per_facet
)))
{
auto
&
conn
=
std
::
get
<
0
>
(
data
);
auto
&
local_gconn
=
std
::
get
<
1
>
(
data
);
const
auto
&
remote_gconn
=
std
::
get
<
2
>
(
data
);
/// skip facet if connectivities are the same
if
(
local_gconn
==
remote_gconn
)
{
continue
;
}
/// re-arrange connectivity
Vector
<
Idx
>
conn_tmp
=
conn
;
auto
begin
=
local_gconn
.
begin
();
auto
end
=
local_gconn
.
end
();
AKANTU_DEBUG_ASSERT
(
std
::
is_permutation
(
begin
,
end
,
remote_gconn
.
begin
()),
"This facets are not just permutation of each other, "
<<
local_gconn
<<
" and "
<<
remote_gconn
);
for
(
auto
&&
data
:
enumerate
(
remote_gconn
))
{
auto
it
=
std
::
find
(
begin
,
end
,
std
::
get
<
1
>
(
data
));
AKANTU_DEBUG_ASSERT
(
it
!=
end
,
"Node not found"
);
auto
new_position
=
it
-
begin
;
conn
(
new_position
)
=
conn_tmp
(
std
::
get
<
0
>
(
data
));
}
// std::transform(remote_gconn.begin(), remote_gconn.end(), conn.begin(),
// [&](auto && gnode) {
// auto it = std::find(begin, end, gnode);
// AKANTU_DEBUG_ASSERT(it != end, "Node not found");
// return conn_tmp(it - begin);
// });
/// if 3D, check if facets are just rotated
if
(
spatial_dimension
==
3
)
{
auto
begin
=
remote_gconn
.
begin
();
/// find first node
auto
it
=
std
::
find
(
begin
,
remote_gconn
.
end
(),
local_gconn
(
0
));
auto
start
=
it
-
begin
;
decltype
(
start
)
n
;
/// count how many nodes in the received connectivity follow
/// the same order of those in the local connectivity
for
(
n
=
1
;
n
<
nb_nodes_per_P1_facet
&&
local_gconn
(
n
)
==
remote_gconn
((
start
+
n
)
%
nb_nodes_per_P1_facet
);
++
n
)
{
;
}
/// skip the facet inversion if facet is just rotated
if
(
n
==
nb_nodes_per_P1_facet
)
{
continue
;
}
}
/// update data to invert facet
auto
&
element_per_facet
=
std
::
get
<
4
>
(
data
);
if
(
element_per_facet
[
1
]
!=
ElementNull
)
{
// by convention the first facet
// cannot be a ElementNull
std
::
swap
(
element_per_facet
[
0
],
element_per_facet
[
1
]);
}
auto
&
subfacets_of_facet
=
std
::
get
<
3
>
(
data
);
std
::
swap
(
subfacets_of_facet
(
0
),
subfacets_of_facet
(
1
));
}
}
AKANTU_DEBUG_OUT
();
}
/* -------------------------------------------------------------------------- */
void
MeshUtils
::
fillElementToSubElementsData
(
Mesh
&
mesh
)
{
AKANTU_DEBUG_IN
();
if
(
mesh
.
getNbElement
(
mesh
.
getSpatialDimension
()
-
1
)
==
0
)
{
AKANTU_DEBUG_INFO
(
"There are not facets, add them in the mesh file or call "
"the buildFacet method."
);
return
;
}
Int
spatial_dimension
=
mesh
.
getSpatialDimension
();
ElementTypeMapArray
<
Real
>
barycenters
(
"barycenter_tmp"
,
mesh
.
getID
());
barycenters
.
initialize
(
mesh
,
_nb_component
=
spatial_dimension
,
_spatial_dimension
=
_all_dimensions
);
Element
element
;
for
(
auto
ghost_type
:
ghost_types
)
{
element
.
ghost_type
=
ghost_type
;
for
(
const
auto
&
type
:
mesh
.
elementTypes
(
_all_dimensions
,
ghost_type
))
{
element
.
type
=
type
;
auto
nb_element
=
mesh
.
getNbElement
(
type
,
ghost_type
);
auto
&
barycenters_arr
=
barycenters
(
type
,
ghost_type
);
barycenters_arr
.
resize
(
nb_element
);
auto
bary
=
barycenters_arr
.
begin
(
spatial_dimension
);
auto
bary_end
=
barycenters_arr
.
end
(
spatial_dimension
);
for
(
Int
el
=
0
;
bary
!=
bary_end
;
++
bary
,
++
el
)
{
element
.
element
=
el
;
mesh
.
getBarycenter
(
element
,
*
bary
);
}
}
}
MeshAccessor
mesh_accessor
(
mesh
);
for
(
Int
sp
(
spatial_dimension
);
sp
>=
1
;
--
sp
)
{
if
(
mesh
.
getNbElement
(
sp
)
==
0
)
{
continue
;
}
for
(
auto
ghost_type
:
ghost_types
)
{
for
(
const
auto
&
type
:
mesh
.
elementTypes
(
sp
,
ghost_type
))
{
auto
&
subelement_to_element
=
mesh_accessor
.
getSubelementToElement
(
type
,
ghost_type
);
subelement_to_element
.
resize
(
mesh
.
getNbElement
(
type
,
ghost_type
));
subelement_to_element
.
set
(
ElementNull
);
}
for
(
const
auto
&
type
:
mesh
.
elementTypes
(
sp
-
1
,
ghost_type
))
{
auto
&
element_to_subelement
=
mesh_accessor
.
getElementToSubelement
(
type
,
ghost_type
);
element_to_subelement
.
resize
(
mesh
.
getNbElement
(
type
,
ghost_type
));
element_to_subelement
.
clear
();
}
}
CSR
<
Element
>
nodes_to_elements
;
buildNode2Elements
(
mesh
,
nodes_to_elements
,
sp
);
Element
facet_element
;
for
(
auto
ghost_type
:
ghost_types
)
{
facet_element
.
ghost_type
=
ghost_type
;
for
(
const
auto
&
type
:
mesh
.
elementTypes
(
sp
-
1
,
ghost_type
))
{
facet_element
.
type
=
type
;
auto
&
element_to_subelement
=
mesh_accessor
.
getElementToSubelement
(
type
,
ghost_type
);
const
auto
&
connectivity
=
mesh
.
getConnectivity
(
type
,
ghost_type
);
for
(
auto
&&
data
:
enumerate
(
make_view
(
connectivity
,
connectivity
.
getNbComponent
())))
{
const
auto
&
facet
=
std
::
get
<
1
>
(
data
);
facet_element
.
element
=
std
::
get
<
0
>
(
data
);
std
::
map
<
Element
,
Int
>
element_seen_counter
;
auto
nb_nodes_per_facet
=
mesh
.
getNbNodesPerElement
(
Mesh
::
getP1ElementType
(
type
));
// count the number of node in common between the facet and the
// other element connected to the nodes of the facet
for
(
auto
node
:
arange
(
nb_nodes_per_facet
))
{
for
(
auto
&
elem
:
nodes_to_elements
.
getRow
(
facet
(
node
)))
{
auto
cit
=
element_seen_counter
.
find
(
elem
);
if
(
cit
!=
element_seen_counter
.
end
())
{
cit
->
second
++
;
}
else
{
element_seen_counter
[
elem
]
=
1
;
}
}
}
// check which are the connected elements
std
::
vector
<
Element
>
connected_elements
;
for
(
auto
&&
cit
:
element_seen_counter
)
{
if
(
cit
.
second
==
nb_nodes_per_facet
)
{
connected_elements
.
push_back
(
cit
.
first
);
}
}
// add the connected elements as sub-elements
for
(
auto
&
connected_element
:
connected_elements
)
{
element_to_subelement
(
facet_element
.
element
)
.
push_back
(
connected_element
);
}
// add the element as sub-element to the connected elements
for
(
auto
&
connected_element
:
connected_elements
)
{
auto
&&
subelements_to_element
=
mesh_accessor
.
getSubelementToElement
(
connected_element
);
// find the position where to insert the element
auto
it
=
std
::
find
(
subelements_to_element
.
begin
(),
subelements_to_element
.
end
(),
ElementNull
);
AKANTU_DEBUG_ASSERT
(
it
!=
subelements_to_element
.
end
(),
"The element "
<<
connected_element
<<
" seems to have too many facets!! ("
<<
(
it
-
subelements_to_element
.
begin
())
<<
" < "
<<
mesh
.
getNbFacetsPerElement
(
connected_element
.
type
)
<<
")"
);
*
it
=
facet_element
;
}
}
}
}
}
AKANTU_DEBUG_OUT
();
}
}
// namespace akantu
Event Timeline
Log In to Comment