From b469cfc7a25f845dc4e7861176027e16457fb1b5 Mon Sep 17 00:00:00 2001
From: Michael Kuron <mkuron@icp.uni-stuttgart.de>
Date: Wed, 19 Jul 2017 11:36:23 +0200
Subject: [PATCH] Squirmer VTK output
---
src/pe/vtk/SphereVtkOutput.cpp | 30 +++++++++++++++++++++++++++++-
1 file changed, 29 insertions(+), 1 deletion(-)
diff --git a/src/pe/vtk/SphereVtkOutput.cpp b/src/pe/vtk/SphereVtkOutput.cpp
index 71e192dc6..a0f6045fa 100644
--- a/src/pe/vtk/SphereVtkOutput.cpp
+++ b/src/pe/vtk/SphereVtkOutput.cpp
@@ -23,6 +23,7 @@
#include "core/DataTypes.h"
#include "SphereVtkOutput.h"
#include "pe/rigidbody/BodyStorage.h"
+#include "pe/rigidbody/Squirmer.h"
#include "core/selectable/IsSetSelected.h"
#include "core/uid/GlobalState.h"
@@ -54,7 +55,7 @@ void SphereVtkOutput::configure()
for( auto it = bs[0].begin(); it != bs[0].end(); ++it )
{
- if (it->getTypeID() == Sphere::getStaticTypeID())
+ if (it->getTypeID() == Sphere::getStaticTypeID() || it->getTypeID() == Squirmer::getStaticTypeID())
bodies_.push_back( static_cast<ConstSphereID> (*it) );
if (it->getTypeID() == Union<boost::tuple<Sphere> >::getStaticTypeID())
{
@@ -65,6 +66,33 @@ void SphereVtkOutput::configure()
bodies_.push_back( static_cast<ConstSphereID> (*it2) );
}
}
+ if (it->getTypeID() == Union<boost::tuple<Squirmer> >::getStaticTypeID())
+ {
+ auto un = static_cast<Union<boost::tuple<Squirmer> > const * > (*it);
+ for( auto it2 = un->begin(); it2 != un->end(); ++it2 )
+ {
+ if (it2->getTypeID() == Squirmer::getStaticTypeID())
+ bodies_.push_back( static_cast<ConstSphereID> (*it2) );
+ }
+ }
+ if (it->getTypeID() == Union<boost::tuple<Sphere,Squirmer> >::getStaticTypeID())
+ {
+ auto un = static_cast<Union<boost::tuple<Sphere,Squirmer> > const * > (*it);
+ for( auto it2 = un->begin(); it2 != un->end(); ++it2 )
+ {
+ if (it2->getTypeID() == Sphere::getStaticTypeID() || it2->getTypeID() == Squirmer::getStaticTypeID())
+ bodies_.push_back( static_cast<ConstSphereID> (*it2) );
+ }
+ }
+ if (it->getTypeID() == Union<boost::tuple<Squirmer,Sphere> >::getStaticTypeID())
+ {
+ auto un = static_cast<Union<boost::tuple<Squirmer,Sphere> > const * > (*it);
+ for( auto it2 = un->begin(); it2 != un->end(); ++it2 )
+ {
+ if (it2->getTypeID() == Sphere::getStaticTypeID() || it2->getTypeID() == Squirmer::getStaticTypeID())
+ bodies_.push_back( static_cast<ConstSphereID> (*it2) );
+ }
+ }
}
}
}
--
GitLab