View_Params_Wrapper.h

Go to the documentation of this file.

#ifndef VIEW_PARAMS_WRAPPER_H_
#define VIEW_PARAMS_WRAPPER_H_

#include <Eigen/Dense>
#include <HalconCpp.h>
#include <vector>

class ViewParamsWrapper
{

private:

    bool is_valid;

    std::string rotation_type;

    HalconCpp::HImage image;

    HalconCpp::HImage originalImage;

    int row1;

    int column1;

    int row2;

    int column2;

    Eigen::Vector3d axis1;

    Eigen::Vector3d axis2;

    Eigen::Vector3d orientation;

    double axis1_angle;

    double axis2_angle;

    bool is_invertible;

    double score_2D;

    bool use_color;

    int vertical_offset;

    int horizontal_offset;

    int depth;

    int number_ferns;

    int patch_size;

    double min_scale;

    double max_scale;

    bool has_bounding_box;

    std::vector<Eigen::Vector2i> box_corners;


public:

    ViewParamsWrapper(std::string rotation_type);

    void setImage(HalconCpp::HImage image);
    void setOriginalImage(HalconCpp::HImage original_image);
    void setImageBounds(int row1, int column1, int row2, int column2);
    void setRotationType(std::string rotation_type);
    void setAxis1(Eigen::Vector3d axis1);
    void setAxis2(Eigen::Vector3d axis2);
    void setOrientation(Eigen::Vector3d orientation);
    void setAxis1Angle(double axis1_angle);
    void setAxis2Angle(double axis2_angle);
    void setIsInvertible(bool is_invertible);
    void setScore2D(double score_2D);
    void setUserColor(bool use_color);
    void setVerticalOffset(int vertical_offset);
    void setHorizontalOffset(int horizontal_offset);
    void setDepth(int depth);
    void setNumberFerns(int number_ferns);
    void setPatchSize(int patch_size);
    void setMinScale(double min_scale);
    void setMaxScale(double max_scale);
    void setIsValid(bool is_valid);
    void setHasBoundingBox(bool has_bounding_box = true);

    void addBoxCorner(Eigen::Vector2i corner);

    void clearBoundingBox();


    HalconCpp::HImage getImage();
    HalconCpp::HImage getOriginalImage();
    int getRow1();
    int getColumn1();
    int getRow2();
    int getColumn2();
    std::string getRotationType();
    Eigen::Vector3d getAxis1();
    Eigen::Vector3d getAxis2();
    Eigen::Vector3d getOrientation();
    double getAxis1Angle();
    double getAxis2Angle();
    bool getIsInvertible();
    double getScore2D();
    bool getUseColor();
    int getVerticalOffset();
    int getHorizontalOffset();
    int getDepth();
    int getNumberFerns();
    int getPatchSize();
    double getMinScale();
    double getMaxScale();
    bool getIsValid();
    bool getHasBoundingBox();
    std::vector<Eigen::Vector2i> getBoxCorners();

};


#endif //VIEW_PARAMS_WRAPPER_H_