View_Creator_Dialog.h

Go to the documentation of this file.

#ifndef VIEW_CREATOR_DIALOG_H_
#define VIEW_CREATOR_DIALOG_H_

#include "View_Creator_Dialog_Base.h"
#include <wx/wx.h>
#include <ros/ros.h>
#include <sensor_msgs/Image.h>
#include <HalconCpp.h>
#include "View_Params_Wrapper.h"
#include <boost/thread.hpp>

class ViewCreatorDialog : public ViewCreatorDialogBase
{

private:

    ros::NodeHandle nh;

    ros::Subscriber image_model_sub;

    ros::Subscriber image_test_sub;

    HalconCpp::HImage fixedImage;

    ViewParamsWrapper *params;

    bool fixed_image_available;

    bool desc_model_available;

    bool test_running;

    int frame_counter;

    double score_sum;

    double feature_sum;

    double time_sum;

    wxTimer* update_timer;

    HalconCpp::HDescriptorModel desc_model;





    boost::thread paint_thread;

    int update_counter;

    void onUpdate(wxTimerEvent& evt);

    void onModelCameraImage(const sensor_msgs::Image::ConstPtr& msg);

    void onTestCameraImage(const sensor_msgs::Image::ConstPtr& msg, bool use_file);

    void enableGuiElements(bool enable = true);

    void enableGuiElementsTest(bool enable = true);

    void updateImageOnCrop();

    void paint_test_image(HalconCpp::HImage img);




protected:

    void OnDialogClose(wxCloseEvent &event);
    void onChoiceImageSource( wxCommandEvent& event );
    void onChoiceImage( wxCommandEvent& event );
    void onChoiceTestImageSource(wxCommandEvent &event);
    void onChoiceTestImage( wxCommandEvent& event );
    void onButtonCancelClicked( wxCommandEvent& event );
    void onButtonSaveClicked( wxCommandEvent& event );
    void onButtonStartTestClicked(wxCommandEvent &event);
    void onButtonEndTestClicked(wxCommandEvent &event);
    void onCheckUseCurrentImage(wxCommandEvent &event);

    void onEditTextUpperLeftRow(wxFocusEvent &event);
    void onEditTextUpperLeftColumn(wxFocusEvent& event);
    void onEditTextLowerRightRow(wxFocusEvent& event);
    void onEditTextLowerRightColumn(wxFocusEvent& event);

    void onSlideUpperLeftRow(wxScrollEvent &event);
    void onSlideUpperLeftColumn(wxScrollEvent &event);
    void onSlideLowerRightRow(wxScrollEvent &event);
    void onSlideLowerRightColumn(wxScrollEvent &event);

    void onEditTextUpperLeftRowEnter(wxCommandEvent &event);
    void onEditTextUpperLeftColumnEnter(wxCommandEvent &event);
    void onEditTextLowerRightRowEnter(wxCommandEvent &event);
    void onEditTextLowerRightColumnEnter(wxCommandEvent &event);

    void onEditTextOrientationX(wxFocusEvent &event);
    void onEditTextOrientationY(wxFocusEvent &event);
    void onEditTextOrientationZ(wxFocusEvent &event);
    void onEditTextScore(wxFocusEvent &event);
    void onEditTextVerticalOffset(wxFocusEvent &event);
    void onEditTextHorizontalOffset(wxFocusEvent &event);
    void onEditTextDepth(wxFocusEvent &event);
    void onEditTextFernNumber(wxFocusEvent &event);
    void onEditTextPatchSize(wxFocusEvent &event);
    void onEditTextMinScale(wxFocusEvent &event);
    void onEditTextMaxScale(wxFocusEvent &event);

    void onEditTextOrientationXEnter(wxCommandEvent &event);
    void onEditTextOrientationYEnter(wxCommandEvent &event);
    void onEditTextOrientationZEnter(wxCommandEvent &event);
    void onEditTextScoreEnter(wxCommandEvent &event);
    void onEditTextVerticalOffsetEnter(wxCommandEvent &event);
    void onEditTextHorizontalOffsetEnter(wxCommandEvent &event);
    void onEditTextDepthEnter(wxCommandEvent &event);
    void onEditTextFernNumberEnter(wxCommandEvent &event);
    void onEditTextPatchSizeEnter(wxCommandEvent &event);
    void onEditTextMinScaleEnter(wxCommandEvent &event);
    void onEditTextMaxScaleEnter(wxCommandEvent &event);

    void onEditTextAxis1X(wxFocusEvent &event);
    void onEditTextAxis1Y(wxFocusEvent &event);
    void onEditTextAxis1Z(wxFocusEvent &event);
    void onEditTextAxis1Angle(wxFocusEvent &event);
    void onEditTextAxis2X(wxFocusEvent &event);
    void onEditTextAxis2Y(wxFocusEvent &event);
    void onEditTextAxis2Z(wxFocusEvent &event);
    void onEditTextAxis2Angle(wxFocusEvent &event);

    void onEditTextAxis1XEnter(wxCommandEvent &event);
    void onEditTextAxis1YEnter(wxCommandEvent &event);
    void onEditTextAxis1ZEnter(wxCommandEvent &event);
    void onEditTextAxis1AngleEnter(wxCommandEvent &event);
    void onEditTextAxis2XEnter(wxCommandEvent &event);
    void onEditTextAxis2YEnter(wxCommandEvent &event);
    void onEditTextAxis2ZEnter(wxCommandEvent &event);
    void onEditTextAxis2AngleEnter(wxCommandEvent &event);

    void onCheckUpsideDown(wxCommandEvent &event);
    void onCheckUseColor(wxCommandEvent &event);


public:
    ViewCreatorDialog(wxWindow* parent, ViewParamsWrapper* params);
};


#endif //DESC_MODELS_DIALOG_H_