diff --git a/src/pe/vtk/SphereVtkOutput.cpp b/src/pe/vtk/SphereVtkOutput.cpp index f02130a721c21063b8091de3b3f29c91ae47a310..ade96605a4ac0eea93f49ad7a95b3402eb70ac52 100644 --- a/src/pe/vtk/SphereVtkOutput.cpp +++ b/src/pe/vtk/SphereVtkOutput.cpp @@ -42,6 +42,7 @@ std::vector< SphereVtkOutput::Attributes > SphereVtkOutput::getAttributes() cons attributes.emplace_back( vtk::typeToString< int >(), "rank", uint_c(1) ); attributes.emplace_back( vtk::typeToString< id_t >(), "id", uint_c(1) ); attributes.emplace_back( vtk::typeToString< id_t >(), "uid", uint_c(1) ); + attributes.emplace_back( vtk::typeToString< float >(), "angular_velocity", uint_c(3) ); return attributes; } diff --git a/src/pe/vtk/SphereVtkOutput.h b/src/pe/vtk/SphereVtkOutput.h index 56ea0146798c5c03b708b0c92c3d796f3ee2629b..655faec083d49e8adbb9397dcb9d9b043b84694c 100755 --- a/src/pe/vtk/SphereVtkOutput.h +++ b/src/pe/vtk/SphereVtkOutput.h @@ -98,6 +98,9 @@ void SphereVtkOutput::push( std::ostream& os, const uint_t data, const uint_t po case 6: vtk::toStream( os, numeric_cast< id_t >(bodies_.at( point )->getTopSuperBody()->getID()) ); break; + case 7: + vtk::toStream( os, numeric_cast< float >(bodies_.at( point )->getAngularVel()[component]) ); + break; } } @@ -139,6 +142,9 @@ void SphereVtkOutput::push( vtk::Base64Writer& b64, const uint_t data, const uin case 6: b64 << numeric_cast< id_t >(bodies_.at( point )->getTopSuperBody()->getID()); break; + case 7: + b64 << numeric_cast< float >(bodies_.at( point )->getAngularVel()[component]); + break; } }