/homes/esi30/DCS339/coursework/src/Processes/roberts.cpp

Go to the documentation of this file.

#include <array>
#include <memory>
#include <string>

#include <gtkmm/image.h>
#include <gtkmm/dialog.h>
#include <gtkmm/checkbutton.h>
#include <gtkmm/stock.h>

#include "../display_unit.hpp"
#include "../processing_page.hpp"
#include "../processing_unit.hpp"
#include "../utils.hpp"
#include "convolution.hpp"
#include "utils.hpp"

namespace{
    class Config: public Gtk::Dialog {
        public:
            Config(): q("Algorithm:"), rX("Roberts X"), rY("Roberts Y") {
                set_resizable(false);

                get_vbox()->pack_start(q);
                get_vbox()->pack_start(rX);
                get_vbox()->pack_start(rY);

                rX.set_active();
                rY.set_active();

                add_button(Gtk::Stock::OK, Gtk::RESPONSE_OK);
                add_button(Gtk::Stock::CANCEL, Gtk::RESPONSE_CANCEL);

                show_all_children();
            }

            bool is_act() const {
                return rX.get_active() || rY.get_active();
            }

            bool do_rX() const {
                return rX.get_active();
            }

            bool do_rY() const {
                return rY.get_active();
            }

        private:
            Gtk::Label q;
            Gtk::CheckButton rX, rY;
    };

    Display_unit* impl(const Image_unit *in){
        Config c;
        if(c.run()!=Gtk::RESPONSE_OK || !(c.is_act()))
            return nullptr;
        c.hide();

        Image_unit *out=nullptr;

        if(c.do_rX()){
            std::array<std::array<double, 3>, 3> kernel={{{{0., 0., 0.}},{{0., 0., -1.}},{{0., 1., 0.}}}};
            std::unique_ptr<Image_unit> tmp(apply<1>(in, kernel, 1));
            add_max(&out, tmp.get());
        }
        if(c.do_rY()){
            std::array<std::array<double, 3>, 3> kernel={{{{0., 0., 0.}},{{0., -1., 0.}},{{0., 0., 1.}}}};
            std::unique_ptr<Image_unit> tmp(apply<1>(in, kernel, 1));
            add_max(&out, tmp.get());
        }

        return out;
    }
}

namespace Processes{
    void roberts(){
        Gtk::Image icon(cmake_install_prefix+std::string("/share/egami/icons/roberts.png"));
        std::shared_ptr<Processing_unit> pu(new Processing_unit(Processing(&impl), "Roberts", icon, false));
        Processing_page::add_unit_to_page("Edge detection", pu);
    }
}