Page Menu
Home
c4science
Search
Configure Global Search
Log In
Files
F79213767
test_mesh_partitionate_mesh_data.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
Sun, Aug 25, 07:38
Size
4 KB
Mime Type
text/x-c
Expires
Tue, Aug 27, 07:38 (2 d)
Engine
blob
Format
Raw Data
Handle
20158278
Attached To
rAKA akantu
test_mesh_partitionate_mesh_data.cc
View Options
/**
* @file test_mesh_partitionate_mesh_data.cc
*
* @author Dana Christen <dana.christen@epfl.ch>
*
* @date creation: Wed May 08 2013
* @date last modification: Mon May 18 2015
*
* @brief test of manual partitioner
*
* @section LICENSE
*
* Copyright (©) 2014, 2015 EPFL (Ecole Polytechnique Fédérale de Lausanne)
* Laboratory (LSMS - Laboratoire de Simulation en Mécanique des Solides)
*
* 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 <cmath>
/* -------------------------------------------------------------------------- */
#include "aka_common.hh"
#include "mesh.hh"
#include "mesh_partition_mesh_data.hh"
/* -------------------------------------------------------------------------- */
#ifdef AKANTU_USE_IOHELPER
# include "dumper_paraview.hh"
#include "dumper_elemental_field.hh"
#endif
//AKANTU_USE_IOHELPER
using
namespace
akantu
;
/* -------------------------------------------------------------------------- */
/* Main */
/* -------------------------------------------------------------------------- */
int
main
(
int
argc
,
char
*
argv
[])
{
initialize
(
argc
,
argv
);
UInt
dim
=
2
;
UInt
nb_partitions
=
8
;
akantu
::
Mesh
mesh
(
dim
);
mesh
.
read
(
"quad.msh"
);
ElementTypeMapArray
<
UInt
>
partition
;
UInt
nb_component
=
1
;
GhostType
gt
=
_not_ghost
;
Mesh
::
type_iterator
tit
=
mesh
.
firstType
(
dim
,
gt
);
Mesh
::
type_iterator
tend
=
mesh
.
lastType
(
dim
,
gt
);
for
(;
tit
!=
tend
;
++
tit
)
{
UInt
nb_element
=
mesh
.
getNbElement
(
*
tit
,
gt
);
partition
.
alloc
(
nb_element
,
nb_component
,
*
tit
,
gt
);
Array
<
UInt
>
&
type_partition_reference
=
partition
(
*
tit
,
gt
);
for
(
UInt
i
(
0
);
i
<
nb_element
;
++
i
)
{
Real
barycenter
[
dim
];
mesh
.
getBarycenter
(
i
,
*
tit
,
barycenter
,
gt
);
Real
real_proc
=
barycenter
[
0
]
*
nb_partitions
;
if
(
std
::
abs
(
real_proc
-
round
(
real_proc
))
<
10
*
std
::
numeric_limits
<
Real
>::
epsilon
())
{
type_partition_reference
(
i
)
=
round
(
real_proc
);
}
else
{
std
::
cout
<<
"*"
;
type_partition_reference
(
i
)
=
floor
(
real_proc
);
}
std
::
cout
<<
"Assigned proc "
<<
type_partition_reference
(
i
)
<<
" to elem "
<<
i
<<
" (type "
<<
*
tit
<<
", barycenter x-coordinate "
<<
barycenter
[
0
]
<<
")"
<<
std
::
endl
;
}
}
akantu
::
MeshPartitionMeshData
*
partitioner
=
new
akantu
::
MeshPartitionMeshData
(
mesh
,
dim
);
partitioner
->
setPartitionMapping
(
partition
);
partitioner
->
partitionate
(
nb_partitions
);
tit
=
mesh
.
firstType
(
dim
,
gt
);
for
(;
tit
!=
tend
;
++
tit
)
{
UInt
nb_element
=
mesh
.
getNbElement
(
*
tit
,
gt
);
const
Array
<
UInt
>
&
type_partition_reference
=
partition
(
*
tit
,
gt
);
const
Array
<
UInt
>
&
type_partition
=
partitioner
->
getPartitions
()(
*
tit
,
gt
);
for
(
UInt
i
(
0
);
i
<
nb_element
;
++
i
)
{
AKANTU_DEBUG_ASSERT
(
type_partition
(
i
)
==
type_partition_reference
(
i
),
"Incorrect partitioning"
);
}
}
//#define DEBUG_TEST
#ifdef DEBUG_TEST
DumperParaview
dumper
(
"test-mesh-data-partition"
);
dumper
::
Field
*
field1
=
new
dumper
::
ElementalField
<
UInt
>
(
partitioner
->
getPartitions
(),
dim
);
dumper
::
Field
*
field2
=
new
dumper
::
ElementalField
<
UInt
>
(
partition
,
dim
);
dumper
.
registerMesh
(
mesh
,
dim
);
dumper
.
registerField
(
"partitions"
,
field1
);
dumper
.
registerField
(
"partitions_ref"
,
field2
);
dumper
.
dump
();
#endif
delete
partitioner
;
finalize
();
return
EXIT_SUCCESS
;
}
Event Timeline
Log In to Comment