void UserApp::onUser( nite::UserTrackerFrameRef frame, const OpenNI::DeviceOptions& deviceOptions )
{
mChannel = OpenNI::toChannel16u( frame.getDepthFrame() );
mUsers = OpenNI::toVector( frame.getUsers() );
for ( vector<nite::UserData>::iterator iter = mUsers.begin(); iter != mUsers.end(); ++iter ) {
if ( iter->isNew() ) {
mDevice->getUserTracker().startSkeletonTracking( iter->getId() );
} else if ( iter->isLost() ) {
mDevice->getUserTracker().stopSkeletonTracking( iter->getId() );
}
}
}
void drawSkeleton( nite::UserTrackerFrameRef& userFrame,
nite::UserTracker& userTracker,
cv::Mat& image)
{
const nite::Array<nite::UserData>& users = userFrame.getUsers();
for ( int i = 0; i < users.getSize(); ++i ) {
const nite::UserData& user = users[i];
if ( user.isNew() ) {
userTracker.startSkeletonTracking( user.getId() );
userTracker.startPoseDetection( user.getId(), nite::POSE_PSI);
userTracker.startPoseDetection( user.getId(), nite::POSE_CROSSED_HANDS);
}
else if ( !user.isLost() ) {
// skeletonの表示
const auto skeleton = user.getSkeleton();
if ( skeleton.getState() == nite::SkeletonState::SKELETON_TRACKED ) {
for ( int j = 0; j < 15; j++ ) {
const auto joint = skeleton.getJoint((nite::JointType)j);
if ( joint.getPositionConfidence() >= 0.7f ) {
const auto position = joint.getPosition();
float x = 0, y = 0;
userTracker.convertJointCoordinatesToDepth(
position.x, position.y, position.z, &x, &y );
cv::circle( image, cvPoint( (int)x, (int)y ),
3, cv::Scalar( 0, 0, 255 ), 5 );
}
}
}
// poseの表示
const auto pose_psi = user.getPose(nite::POSE_PSI);
if( pose_psi.isHeld() || pose_psi.isEntered() )
{
auto center = user.getCenterOfMass();
float x = 0, y = 0;
userTracker.convertJointCoordinatesToDepth(center.x, center.y, center.z, &x, &y);
cv::putText(image, "PSI", cv::Point2f(x,y), cv::FONT_HERSHEY_SIMPLEX, 2, cv::Scalar(0xFF,0xFF,0xFF));
}
const auto pose_cross = user.getPose(nite::POSE_CROSSED_HANDS);
if( pose_cross.isHeld() || pose_cross.isEntered() ){
auto center = user.getCenterOfMass();
float x = 0, y = 0;
userTracker.convertJointCoordinatesToDepth(center.x, center.y, center.z, &x, &y);
cv::putText(image, "Cross", cv::Point2f(x,y), cv::FONT_HERSHEY_COMPLEX, 2, cv::Scalar(0xFF,0xFF,0xFF));
}
}
}
}
