AgeGenderEstimator.h

Go to the documentation of this file.

#ifndef __PBIO_API__PBIO__AGE_GENDER_ESTIMATOR_H_
#define __PBIO_API__PBIO__AGE_GENDER_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 AgeGenderEstimator : public ComplexObject
{
public:

    typedef LightSmartPtr<AgeGenderEstimator>::tPtr Ptr;

    enum Age
    {
        AGE_KID = 0, 

        AGE_YOUNG = 1,

        AGE_ADULT = 2,

        AGE_SENIOR = 3  
    };

    enum Gender
    {
        GENDER_FEMALE = 0,

        GENDER_MALE = 1
    };

    struct AgeGender
    {
        Age age;

        Gender gender;  

        float age_years;
    };

    AgeGender estimateAgeGender(const pbio::RawSample &sample) const;

private:

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

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

}  // pbio namespace




namespace pbio
{

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


inline
AgeGenderEstimator::AgeGender AgeGenderEstimator::estimateAgeGender(const pbio::RawSample &sample) const
{
    void* exception = NULL;

    int32_t age_class, gender;
    float age_years;

    _dll_handle->AgeGenderEstimator_estimateAgeGender(
        _impl,
        (pbio::facerec::RawSampleImpl const*) sample._impl,
        &age_class,
        &gender,
        &age_years,
        &exception);

    checkException(exception, *_dll_handle);

    AgeGender result;
    result.age = (Age) age_class;
    result.gender = (Gender) gender;
    result.age_years = age_years;

    return result;
}



}  // pbio namespace

#endif  // __PBIO_API__PBIO__AGE_GENDER_ESTIMATOR_H_