From 4be3540c8568af74168a8d6a3e86902886a5798b Mon Sep 17 00:00:00 2001
From: Tobias Leemann <tobias.leemann@fau.de>
Date: Sun, 25 Nov 2018 14:49:29 +0100
Subject: [PATCH] Added Marshalling tests for meshes and Unions, Fixed
implementation of marshalling union
---
src/pe/communication/rigidbody/Union.h | 6 +-
tests/mesh/CMakeLists.txt | 3 +
tests/mesh/MeshMarshalling.cpp | 170 +++++++++++++++++++++++++
3 files changed, 176 insertions(+), 3 deletions(-)
create mode 100644 tests/mesh/MeshMarshalling.cpp
diff --git a/src/pe/communication/rigidbody/Union.h b/src/pe/communication/rigidbody/Union.h
index 544c131ef..67ba4db6d 100644
--- a/src/pe/communication/rigidbody/Union.h
+++ b/src/pe/communication/rigidbody/Union.h
@@ -123,8 +123,7 @@ inline std::unique_ptr<Union<BodyTypeTuple>> instantiate( mpi::RecvBuffer& buffe
false,
subobjparam.communicating_,
subobjparam.infiniteMass_ );
- un->setLinearVel( subobjparam.v_ );
- un->setAngularVel( subobjparam.w_ );
+
un->MPITrait.setOwner( subobjparam.mpiTrait_.owner_ );
// Decoding the contained primitives
@@ -136,7 +135,8 @@ inline std::unique_ptr<Union<BodyTypeTuple>> instantiate( mpi::RecvBuffer& buffe
obj->setRemote( true );
un->add(std::move(obj));
}
-
+ un->setLinearVel( subobjparam.v_ );
+ un->setAngularVel( subobjparam.w_ );
newBody = un.get();
return un;
}
diff --git a/tests/mesh/CMakeLists.txt b/tests/mesh/CMakeLists.txt
index 932e96ec6..a8ba46bf9 100644
--- a/tests/mesh/CMakeLists.txt
+++ b/tests/mesh/CMakeLists.txt
@@ -94,4 +94,7 @@ if ( WALBERLA_BUILD_WITH_OPENMESH )
waLBerla_execute_test( NAME NumericIntegrationTestSphere COMMAND $<TARGET_FILE:NumericIntegrationTest> sphere.obj )
waLBerla_execute_test( NAME NumericIntegrationTestBunny COMMAND $<TARGET_FILE:NumericIntegrationTest> bunny.obj CONFIGURATIONS Release RelWithDbgInfo )
+ waLBerla_compile_test( FILES MeshMarshalling.cpp DEPENDS core mesh )
+ waLBerla_execute_test( NAME MeshMarshalling )
+
endif()
diff --git a/tests/mesh/MeshMarshalling.cpp b/tests/mesh/MeshMarshalling.cpp
new file mode 100644
index 000000000..1ba1fcc21
--- /dev/null
+++ b/tests/mesh/MeshMarshalling.cpp
@@ -0,0 +1,170 @@
+//======================================================================================================================
+//
+// This file is part of waLBerla. waLBerla is free software: you can
+// redistribute it and/or modify it under the terms of the GNU General Public
+// License as published by the Free Software Foundation, either version 3 of
+// the License, or (at your option) any later version.
+//
+// waLBerla 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 General Public License
+// for more details.
+//
+// You should have received a copy of the GNU General Public License along
+// with waLBerla (see COPYING.txt). If not, see <http://www.gnu.org/licenses/>.
+//
+//! \file Marshalling.cpp
+//! \author Sebastian Eibl <sebastian.eibl@fau.de>
+//
+//======================================================================================================================
+
+#include "core/debug/TestSubsystem.h"
+
+#include "mesh/pe/rigid_body/ConvexPolyhedron.h"
+#include "mesh/pe/rigid_body/ConvexPolyhedronFactory.h"
+#include "mesh/TriangleMeshes.h"
+#include "mesh/QHull.h"
+#include "mesh/pe/Types.h"
+#include "pe/rigidbody/Squirmer.h"
+#include "pe/rigidbody/UnionFactory.h"
+#include "pe/rigidbody/Union.h"
+#include "pe/communication/rigidbody/Squirmer.h"
+#include "pe/communication/DynamicMarshalling.h"
+#include "mesh/pe/communication/ConvexPolyhedron.h"
+#include "pe/rigidbody/SetBodyTypeIDs.h"
+#include "pe/Materials.h"
+
+#include <boost/tuple/tuple.hpp>
+#include <memory>
+namespace walberla {
+using namespace walberla::pe;
+using namespace walberla::pe::communication;
+
+using UnionTypeTuple = boost::tuple<mesh::pe::ConvexPolyhedron>;
+using UnionT = Union<UnionTypeTuple>;
+using UnionID = UnionT *;
+using UnionPtr = std::unique_ptr<UnionT>;
+
+typedef boost::tuple<mesh::pe::ConvexPolyhedron, UnionT> BodyTuple ;
+
+std::vector<Vector3<real_t>> generateOctahedron( const real_t radius)
+{
+
+ std::vector<Vector3<real_t>> okta( 6 );
+ for(size_t i = 0; i < 6; i++){
+ auto &p = okta[i];
+ p[i%3]=(i<3) ? radius: -radius;
+ }
+ return okta;
+}
+
+// Checks two mesh::TriangleMesh for pseudo-equality
+void checkMeshEquals(const mesh::TriangleMesh &m1, const mesh::TriangleMesh &m2){
+ // Very basic checks
+ WALBERLA_CHECK_FLOAT_EQUAL(mesh::computeVolume(m1), mesh::computeVolume(m2));
+ WALBERLA_CHECK_EQUAL(mesh::computeCentroid(m1), mesh::computeCentroid(m2));
+ WALBERLA_CHECK_EQUAL(mesh::computeInertiaTensor(m1), mesh::computeInertiaTensor(m2));
+}
+
+// Checks two convexPolyhedrons for equality
+void checkConvexPolyhedronEquals(const mesh::pe::ConvexPolyhedron &b1, const mesh::pe::ConvexPolyhedron &b2){
+ WALBERLA_CHECK_FLOAT_EQUAL(b1.getPosition(), b2.getPosition());
+ WALBERLA_CHECK_FLOAT_EQUAL(b1.getLinearVel(), b2.getLinearVel());
+ WALBERLA_CHECK_FLOAT_EQUAL(b1.getAngularVel(), b2.getAngularVel());
+ WALBERLA_CHECK_EQUAL(b1.getInertia(), b2.getInertia());
+ WALBERLA_CHECK_EQUAL(b1.getMaterial(), b2.getMaterial());
+ // Check equality of the meshes
+ checkMeshEquals(b1.getMesh(), b2.getMesh());
+ WALBERLA_CHECK_EQUAL(b1.getID(), b2.getID());
+ WALBERLA_CHECK_EQUAL(b1.getSystemID(), b2.getSystemID());
+}
+
+void testConvexPolyhedron()
+{
+ WALBERLA_LOG_INFO_ON_ROOT("*** testConvexPolyhedron ***");
+
+ // Generate mesh
+ shared_ptr< mesh::TriangleMesh > octamesh = make_shared<mesh::TriangleMesh>();
+ mesh::QHull< mesh::TriangleMesh > qhull( generateOctahedron(real_t(1.0)), octamesh );
+ qhull.run();
+
+ MaterialID iron = Material::find("iron");
+
+ mesh::pe::ConvexPolyhedron b1(759846, 1234794, Vec3(real_c(1), real_c(2), real_c(3)), Vec3(0,0,0), Quat(), *octamesh, iron, false, true, false);
+ b1.setLinearVel(Vec3(real_c(5.2), real_c(6.3), real_c(7.4)));
+ b1.setAngularVel(Vec3(real_c(1.2), real_c(2.3), real_c(3.4)));
+
+ mpi::SendBuffer sb;
+ MarshalDynamically<BodyTuple>::execute(sb, b1);
+ mpi::RecvBuffer rb(sb);
+
+ auto bPtr = UnmarshalDynamically<BodyTuple>::execute(rb, mesh::pe::ConvexPolyhedron::getStaticTypeID(), math::AABB(Vec3(-100,-100,-100), Vec3(100,100,100)), math::AABB(Vec3(-100,-100,-100), Vec3(100,100,100)));
+ mesh::pe::ConvexPolyhedronID b2 = static_cast<mesh::pe::ConvexPolyhedronID>(bPtr.get());
+ checkConvexPolyhedronEquals(b1, *b2);
+
+}
+
+void testUnion()
+{
+
+ WALBERLA_LOG_INFO_ON_ROOT("*** testUnion ***");
+ // Generate mesh
+ shared_ptr< mesh::TriangleMesh > octamesh = make_shared<mesh::TriangleMesh>();
+ mesh::QHull< mesh::TriangleMesh > qhull( generateOctahedron(real_t(1.0)), octamesh );
+ qhull.run();
+
+ MaterialID iron = Material::find("iron");
+
+ UnionT u1(159, 423, Vec3(real_c(1), real_c(2), real_c(3)), Vec3(0,0,0), Quat(), false, false, false);
+ u1.add(std::make_unique<mesh::pe::ConvexPolyhedron>(753326, 1267824, Vec3(real_c(2), real_c(2), real_c(3)), Vec3(0,0,0), Quat(), *octamesh, iron, false, true, false));
+ u1.add(std::make_unique<mesh::pe::ConvexPolyhedron>(753246, 1233424, Vec3(real_c(-1), real_c(4), real_c(-2)), Vec3(0,0,0), Quat(), *octamesh, iron, false, true, false));
+
+ u1.setLinearVel(Vec3(real_c(5.2), real_c(6.3), real_c(7.4)));
+ u1.setAngularVel(Vec3(real_c(1.2), real_c(2.3), real_c(3.4)));
+
+ mpi::SendBuffer sb;
+ MarshalDynamically<BodyTuple>::execute(sb, u1);
+ mpi::RecvBuffer rb(sb);
+
+ auto uPtr = UnmarshalDynamically<BodyTuple>::execute(rb, UnionT::getStaticTypeID(), math::AABB(Vec3(-100,-100,-100), Vec3(100,100,100)), math::AABB(Vec3(-100,-100,-100), Vec3(100,100,100)));
+ UnionID u2 = static_cast<UnionID>(uPtr.get());
+ WALBERLA_CHECK_NOT_NULLPTR( u2 );
+
+ WALBERLA_CHECK_EQUAL(u1.size(), 2);
+ WALBERLA_CHECK_EQUAL(u1.size(), u2->size());
+ WALBERLA_CHECK_EQUAL(u1.getInertia(), u2->getInertia());
+ WALBERLA_CHECK_EQUAL(u1.getPosition(), u2->getPosition());
+ WALBERLA_CHECK_FLOAT_EQUAL(u1.getLinearVel(), u2->getLinearVel());
+ WALBERLA_CHECK_FLOAT_EQUAL(u1.getAngularVel(), u2->getAngularVel());
+
+ //getting polyhedrons of first union
+ mesh::pe::ConvexPolyhedronID p11 = static_cast<mesh::pe::ConvexPolyhedronID > (u1.begin().getBodyID());
+ mesh::pe::ConvexPolyhedronID p21 = static_cast<mesh::pe::ConvexPolyhedronID > ((++(u1.begin())).getBodyID());
+
+ //getting polyhedrons of second union
+ mesh::pe::ConvexPolyhedronID p12 = static_cast<mesh::pe::ConvexPolyhedronID > (u2->begin().getBodyID());
+ mesh::pe::ConvexPolyhedronID p22 = static_cast<mesh::pe::ConvexPolyhedronID > ((++(u2->begin())).getBodyID());
+
+ checkConvexPolyhedronEquals(*p11, *p12);
+ checkConvexPolyhedronEquals(*p21, *p22);
+
+}
+
+int main( int argc, char** argv )
+{
+ walberla::debug::enterTestMode();
+
+ walberla::MPIManager::instance()->initializeMPI( &argc, &argv );
+
+ SetBodyTypeIDs<BodyTuple>::execute();
+ testConvexPolyhedron();
+ testUnion();
+
+ return EXIT_SUCCESS;
+}
+} // namespace walberla
+
+int main( int argc, char* argv[] )
+{
+ return walberla::main( argc, argv );
+}
--
GitLab