/homes/esi30/DCS339/coursework/src/Processes/sobel.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:"), sX("Sobel X"), s45("Sobel 45"), sY("Sobel Y"), s135("Sobel 135") {
                set_resizable(false);

                get_vbox()->pack_start(q);
                get_vbox()->pack_start(sX);
                get_vbox()->pack_start(s45);
                get_vbox()->pack_start(sY);
                get_vbox()->pack_start(s135);

                sX.set_active();
                s45.set_active();
                sY.set_active();
                s135.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 sX.get_active() || s45.get_active() || sY.get_active() || s135.get_active();
            }

            bool do_sX() const {
                return sX.get_active();
            }

            bool do_s45() const {
                return s45.get_active();
            }

            bool do_sY() const {
                return sY.get_active();
            }

            bool do_s135() const {
                return s135.get_active();
            }

        private:
            Gtk::Label q;
            Gtk::CheckButton sX, s45, sY, s135;
    };

    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_sX()){
            std::array<std::array<double, 3>, 3> kernel={{{{-1., 0., 1.}},{{-2., 0., 2.}},{{-1., 0., 1.}}}};
            std::unique_ptr<Image_unit> tmp(apply<1>(in, kernel, 1));
            add_max(&out, tmp.get());
        }
        if(c.do_s45()){
            std::array<std::array<double, 3>, 3> kernel={{{{-2., -1., 0.}},{{-1., 0., 1.}},{{0., 1., 2.}}}};
            std::unique_ptr<Image_unit> tmp(apply<1>(in, kernel, 1));
            add_max(&out, tmp.get());
        }
        if(c.do_sY()){
            std::array<std::array<double, 3>, 3> kernel={{{{-1., -2., -1.}},{{0., 0., 0.}},{{1., 2., 1.}}}};
            std::unique_ptr<Image_unit> tmp(apply<1>(in, kernel, 1));
            add_max(&out, tmp.get());
        }
        if(c.do_s135()){
            std::array<std::array<double, 3>, 3> kernel={{{{0., 1., 2.}},{{-1., 0., 1.}},{{-2., -1., 0.}}}};
            std::unique_ptr<Image_unit> tmp(apply<1>(in, kernel, 1));
            add_max(&out, tmp.get());
        }

        return out;
    }
}

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