/homes/esi30/DCS339/coursework/src/Processes/prewitt.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:"), pX("Prewitt X"), p45("Prewitt 45"), pY("Prewitt Y"), p135("Prewitt 135") {
                set_resizable(false);

                get_vbox()->pack_start(q);
                get_vbox()->pack_start(pX);
                get_vbox()->pack_start(p45);
                get_vbox()->pack_start(pY);
                get_vbox()->pack_start(p135);

                pX.set_active();
                p45.set_active();
                pY.set_active();
                p135.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 pX.get_active() || p45.get_active() || pY.get_active() || p135.get_active();
            }

            bool do_pX() const {
                return pX.get_active();
            }

            bool do_p45() const {
                return p45.get_active();
            }

            bool do_pY() const {
                return pY.get_active();
            }

            bool do_p135() const {
                return p135.get_active();
            }

        private:
            Gtk::Label q;
            Gtk::CheckButton pX, p45, pY, p135;
    };

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

        Image_unit *out=nullptr;

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

        return out;
    }
}

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