EmotionsEstimator.h

См. документацию.

#ifndef __PBIO_API__PBIO__EMOTIONS_ESTIMATOR_H_
#define __PBIO_API__PBIO__EMOTIONS_ESTIMATOR_H_

#include <sstream>
#include <vector>


#include "ComplexObject.h"
#include "RawSample.h"
#include "SmartPtr.h"
#include "Error.h"
#include "stl_wraps_impls/WrapOStreamImpl.h"


namespace pbio
{

class FacerecService;

class EmotionsEstimator : public ComplexObject
{
public:

    typedef LightSmartPtr<EmotionsEstimator>::tPtr Ptr;

    enum Emotion
    {
        EMOTION_NEUTRAL = 0,

        EMOTION_HAPPY = 1,
        
        EMOTION_ANGRY = 2,

        EMOTION_SURPRISE = 3,

        EMOTION_DISGUSTED = 4,

        EMOTION_SCARED = 5,

        EMOTION_SAD = 6,
    };

    struct EmotionConfidence
    {
        Emotion emotion;

        float confidence;
    };

    typedef std::vector<EmotionConfidence> EstimatedEmotionsVector;

    EstimatedEmotionsVector estimateEmotions(const pbio::RawSample &sample) const;

private:

    EmotionsEstimator(
        const DHPtr &dll_handle,
        void* impl);

    friend class FacerecService;
    friend class object_with_ref_counter<EmotionsEstimator>;
};

}  // pbio namespace




namespace pbio
{


inline
EmotionsEstimator::EmotionsEstimator(
    const DHPtr &dll_handle,
    void* impl):
ComplexObject(dll_handle, impl)
{
    // nothing else
}


inline
std::vector<EmotionsEstimator::EmotionConfidence> EmotionsEstimator::estimateEmotions(
    const pbio::RawSample &sample) const
{
    void* exception = NULL;

    std::ostringstream data_oss;

    stl_wraps::WrapOStreamImpl data_oss_w(data_oss);

    const int count = _dll_handle->EmotionsEstimator_estimateEmotions(
        _impl,
        (pbio::facerec::RawSampleImpl const*) sample._impl,
        &data_oss_w,
        pbio::stl_wraps::WrapOStreamImpl::write_func,
        &exception);

    checkException(exception, *_dll_handle);

    std::istringstream data_iss(data_oss.str());

    std::vector<EmotionConfidence> result(count);

    for(int i = 0; i < count; ++i)
    {
        int32_t emotion;
        float confidence;

        data_iss.read((char*)&emotion, sizeof(emotion));
        data_iss.read((char*)&confidence, sizeof(confidence));

        result[i].emotion = (Emotion) emotion;
        result[i].confidence = confidence;
    }

    return result;
}



}  // pbio namespace

#endif  // __PBIO_API__PBIO__EMOTIONS_ESTIMATOR_H_