diff --git a/src/pe/vtk/SphereVtkOutput.cpp b/src/pe/vtk/SphereVtkOutput.cpp index 71e192dc63c9125396448436afb22b8ebe4f6971..a0f6045fa5d3ae9affb171d3f308e3ae7754babe 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) ); + } + } } } }