QualityEstimator.h

Go to the documentation of this file.

#ifndef __PBIO_API__PBIO__QUALITY_ESTIMATOR_H_
#define __PBIO_API__PBIO__QUALITY_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 QualityEstimator : public ComplexObject
{
public:

    typedef LightSmartPtr<QualityEstimator>::tPtr Ptr;

    struct Quality
    {
        int total;

        int lighting;

        int noise;

        int sharpness;

        int flare;
    };


    Quality estimateQuality(const pbio::RawSample&) const;

private:

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

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

}  // pbio namespace




namespace pbio
{


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


inline
QualityEstimator::Quality QualityEstimator::estimateQuality(const pbio::RawSample &sample) const
{
    void* exception = NULL;

    Quality result;

    _dll_handle->QualityEstimator_estimateQuality(
        _impl,
        (pbio::facerec::RawSampleImpl const*) sample._impl,
        &result.lighting,
        &result.noise,
        &result.sharpness,
        &result.flare,
        &exception);

    // total is the worst quality, i.e. the max value

    result.total = 1;
    result.total = std::max<int>(result.total, result.lighting);
    result.total = std::max<int>(result.total, result.noise);
    result.total = std::max<int>(result.total, result.sharpness);
    result.total = std::max<int>(result.total, result.flare);

    checkException(exception, *_dll_handle);

    return result;
}


}  // pbio namespace

#endif  // __PBIO_API__PBIO__QUALITY_ESTIMATOR_H_