void KinectHDFaceGrabber::updateFaceModelStatusOfFaceModelBuilder(IFaceModelBuilder** faceModelBuilder, IFaceModel* faceModel)
{
FaceModelBuilderCollectionStatus status;
HRESULT hr = faceModelBuilder[0]->get_CollectionStatus(&status);
if (SUCCEEDED(hr)){
std::wstring statusString = getCaptureStatusText(status);
statusChanged(statusString, true);
if (status == FaceModelBuilderCollectionStatus::FaceModelBuilderCollectionStatus_Complete){
std::cout << "Status : Complete" << std::endl;
IFaceModelData* pFaceModelData = nullptr;
hr = faceModelBuilder[0]->GetFaceData(&pFaceModelData);
if (SUCCEEDED(hr) && pFaceModelData != nullptr){
hr = pFaceModelData->ProduceFaceModel(&faceModel);
}
faceModelBuilder[0]->Release();
SafeRelease(pFaceModelData);
faceModelBuilder[0] = nullptr;
}
}
}
void Kin2::buildHDFaceModels(int &collectionStatus, int &captureStatus)
{
collectionStatus = -1;
captureStatus = -1;
if (!(m_flags & k2::HD_FACE))
{
mexPrintf("ERROR: NO HD-FACE FUNCTIONALITY SELECTED!\n");
return;
}
HRESULT hr;
// iterate through each HD face reader
for (int iFace = 0; iFace < BODY_COUNT; ++iFace)
{
// retrieve the latest face frame from this reader
IHighDefinitionFaceFrame *pHDFaceFrame = nullptr;
hr = m_pHDFaceFrameReaders[iFace]->AcquireLatestFrame(&pHDFaceFrame);
BOOLEAN bFaceTracked = false;
if (SUCCEEDED(hr) && nullptr != pHDFaceFrame)
{
// check if a valid face is tracked in this face frame
hr = pHDFaceFrame->get_IsTrackingIdValid(&bFaceTracked);
}
// If face tracked, try to align it
if (SUCCEEDED(hr) && bFaceTracked)
{
IFaceModel *pFaceModel = nullptr;
hr = pHDFaceFrame->GetAndRefreshFaceAlignmentResult(m_pFaceAlignment[iFace]);
// If face aligned, continue building the model
if (SUCCEEDED(hr) && m_pFaceAlignment[iFace] != nullptr)
{
// If face model not ready
if (!m_faceModelReady[iFace])
{
FaceModelBuilderCollectionStatus collection;
hr = m_pFaceModelBuilder[iFace]->get_CollectionStatus(&collection);
collectionStatus = (int)collection;
// If model completed
if (collection == FaceModelBuilderCollectionStatus::FaceModelBuilderCollectionStatus_Complete)
{
mexPrintf("Face Model Completed!\n");
IFaceModelData* pFaceModelData = nullptr;
hr = m_pFaceModelBuilder[iFace]->GetFaceData(&pFaceModelData);
// Produce the model
if (SUCCEEDED(hr) && pFaceModelData != nullptr)
{
mexPrintf("Producing model...\n");
hr = pFaceModelData->ProduceFaceModel(&m_pFaceModel[iFace]);
mexPrintf("Model Ready!\n");
// Set the model ready flag
if (SUCCEEDED(hr) && m_pFaceModel[iFace] != nullptr)
{
m_faceModelReady[iFace] = true;
}
}
SafeRelease(pFaceModelData);
// Get the shape units (SU) i.e. the deformations wrt the base face model
/*
if (SUCCEEDED(hr))
{
float deformations[FaceShapeDeformations_Count];
hr = m_pFaceModel[iFace]->GetFaceShapeDeformations(FaceShapeDeformations_Count, deformations);
}
*/
}
// if model not completed yet
else
{
// Display Collection Status
/*
if (collection >= FaceModelBuilderCollectionStatus::FaceModelBuilderCollectionStatus_TiltedUpViewsNeeded)
{
mexPrintf("Need : Tilted Up Views\n");
}
else if (collection >= FaceModelBuilderCollectionStatus::FaceModelBuilderCollectionStatus_RightViewsNeeded)
{
mexPrintf("Need : Right Views\n");
}
else if (collection >= FaceModelBuilderCollectionStatus::FaceModelBuilderCollectionStatus_LeftViewsNeeded)
{
mexPrintf("Need : Left Views\n");
}
else if (collection >= FaceModelBuilderCollectionStatus::FaceModelBuilderCollectionStatus_FrontViewFramesNeeded)
{
mexPrintf("Need : Front ViewFrames\n");
}
*/
// Display Capture Status
FaceModelBuilderCaptureStatus capture;
hr = m_pFaceModelBuilder[iFace]->get_CaptureStatus(&capture);
captureStatus = (int)capture;
/*
switch (capture)
{
case FaceModelBuilderCaptureStatus::FaceModelBuilderCaptureStatus_OtherViewsNeeded:
std::cout << "Other views needed" << std::endl;
break;
case FaceModelBuilderCaptureStatus::FaceModelBuilderCaptureStatus_FaceTooFar:
std::cout << "Face Too Far from Camera" << std::endl;
break;
case FaceModelBuilderCaptureStatus::FaceModelBuilderCaptureStatus_FaceTooNear:
std::cout << "Face Too Near to Camera" << std::endl;
break;
case FaceModelBuilderCaptureStatus_MovingTooFast:
std::cout << "Moving Too Fast" << std::endl;
break;
case FaceModelBuilderCaptureStatus::FaceModelBuilderCaptureStatus_LostFaceTrack:
std::cout << "Lost Face Track" << std::endl;
break;
case FaceModelBuilderCaptureStatus::FaceModelBuilderCaptureStatus_SystemError:
std::cout << "ERROR: System Error" << std::endl;
break;
default:
break;
}
*/
} // collection not complete
} // If face model not ready
} // If face aligned
} // If face tracked
else
{
// face tracking is not valid - attempt to fix the issue
// a valid body is required to perform this step
if (m_bHaveBodyData)
{
// check if the corresponding body is tracked
// if this is true then update the face frame source to track this body
IBody* pBody = m_ppBodies[iFace];
if (pBody != nullptr)
{
BOOLEAN bTracked = false;
hr = pBody->get_IsTracked(&bTracked);
UINT64 bodyTId;
if (SUCCEEDED(hr) && bTracked)
{
// get the tracking ID of this body
hr = pBody->get_TrackingId(&bodyTId);
if (SUCCEEDED(hr))
{
// update the face frame source with the tracking ID
m_pHDFaceFrameSources[iFace]->put_TrackingId(bodyTId);
}
}
} // if (pBody != nullptr)
} // if (m_bHaveBodyData)
} // if face tracked
}// for each face reader
} // end buildHDFaceModels
