front_features.h

Go to the documentation of this file.

#ifndef VR_FRONT_FEATURES_H
#define VR_FRONT_FEATURES_H

namespace vr
{

class CoordinateTransformation;

class FrontFeatures : public FeatureContainer
{
    void checkIrisResult(const LateralSide side, int& result,
        CircleImageData& irisData);
    void ffaMarkChanged(const LateralSide side);
    bool locateMainMarks(const cv::Mat& image, const bool faceFound);
    void locateFfaMarks(const cv::Mat& image);
    void locateIrises(const cv::Mat& image);
    void locateFace(const cv::Mat& image);
    void locateEye(const cv::Mat& image, const LateralSide side);
    cv::Rect getScaledPeakMarkRoi() const;
    cv::Rect getScaledEyeRoi(const LateralSide side) const;
    cv::Rect getScaledIrisRoi(const LateralSide side) const;
    cv::Rect getPupilVicinity(const LateralSide side, const float margin) const;
    cv::Point3f getGuessedPupil(const LateralSide side) const;
    inline bool isFrontCameraImage() const {
        switch (arrangement_) {
            case iPad_4_front:
            case iPad_Air_front:
            case iPad_mini_3_front:
                return true;
            default:
                return false;
        }
    }
    // Zeiger, damit Feature-Toolbox CoordinateTransformation nicht veröffentlichen muß.
    CoordinateTransformation* scaledTransformation_;
    CoordinateTransformation* transformation_;
    cv::CascadeClassifier& faceCascade_;
    cv::CascadeClassifier& eyeCascade_;
    // Rotations- und Translationsvektor aus cv::solvePnP();
    // Argumente für cv::projectPoints().
    const ExposureArrangement arrangement_;
    float pixelsPerMm_;
    FeatureStatus status_;
    cv::Rect mainMarkRoi_;
    cv::Rect peakMarkRoi_;
    cv::Rect eyeRoi[LATERAL_SIDE];
    cv::Rect irisRoi[LATERAL_SIDE];
    cv::Rect ffaRoi[LATERAL_SIDE];
    cv::Vec3d ffaMark3d[LATERAL_SIDE];
protected:
    void centerChanged(Feature& feature);
    void computeTrafo();
public:
    Mark mainMark[LATERAL_SIDE];
    Mark peakMark;
    Mark faceFormAngleMark[LATERAL_SIDE];
    CircularFeature iris[LATERAL_SIDE];
    AreaFeature face;
    AreaFeature eye[LATERAL_SIDE];

    FrontFeatures(const ExposureArrangement arrangement,
        cv::CascadeClassifier& faceCascade,
        cv::CascadeClassifier& eyeCascade);
    FrontFeatures(const FrontFeatures& src);
    FrontFeatures(FrontFeatures&& src) noexcept;
    ~FrontFeatures() noexcept;
    void setContainer();
    inline const CoordinateTransformation& getTrafo() const { return *transformation_; };
    inline FeatureStatus getStatus() const { return status_; };
    inline void setStatus(const FeatureStatus status) { status_ = status; };
    bool isInfantConfiguration(FeatureStatus& status);
    bool locate(const cv::Mat& image);
    inline float getScaledPixelsPerMm() const { return pixelsPerMm_; }
    inline float getPixelsPerMm() const { return pixelsPerMm_ * scaleFactor_; }
    void reset();
    double getFaceFormAngle(LateralSide side) const;
// ab hier zu Debugzwecken
    cv::Rect getEyeRoi(LateralSide side) { return scaledRect(eyeRoi[side], scaleFactor_); };
    cv::Rect getMainMarksRoi() { return scaledRect(mainMarkRoi_, scaleFactor_); };
    cv::Rect getPeakMarkRoi() { return scaledRect(peakMarkRoi_, scaleFactor_); };
    cv::Rect getIrisRoi(LateralSide side) { return scaledRect(irisRoi[side], scaleFactor_); };
    cv::Rect getFfaRoi(LateralSide side) { return scaledRect(ffaRoi[side], scaleFactor_); };
    const cv::Vec3d& getFfaMark3d(LateralSide side) const { return ffaMark3d[side]; };
};

typedef std::shared_ptr<FrontFeatures> FrontFeaturesSPtr;

}

#endif // VR_FRONT_FEATURES_H