/homes/esi30/DCS339/coursework/src/Processes/line.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:"), l0("Line 0"), l45("Line 45"), l90("Line 90"), l135("Line 135") {
                set_resizable(false);

                get_vbox()->pack_start(q);
                get_vbox()->pack_start(l0);
                get_vbox()->pack_start(l45);
                get_vbox()->pack_start(l90);
                get_vbox()->pack_start(l135);

                l0.set_active();
                l45.set_active();
                l90.set_active();
                l135.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 l0.get_active() || l45.get_active() || l90.get_active() || l135.get_active();
            }

            bool do_l0() const {
                return l0.get_active();
            }

            bool do_l45() const {
                return l45.get_active();
            }

            bool do_l90() const {
                return l90.get_active();
            }

            bool do_l135() const {
                return l135.get_active();
            }

        private:
            Gtk::Label q;
            Gtk::CheckButton l0, l45, l90, l135;
    };

    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_l0()){
            std::array<std::array<double, 3>, 3> kernel={{{{-1., -1., -1.}},{{2., 2., 2.}},{{-1., -1., -1.}}}};
            std::unique_ptr<Image_unit> tmp(apply<1>(in, kernel, 1));
            add_max(&out, tmp.get());
        }
        if(c.do_l45()){
            std::array<std::array<double, 3>, 3> kernel={{{{-1., -1., 2.}},{{-1., 2., -1.}},{{2., -1., -1.}}}};
            std::unique_ptr<Image_unit> tmp(apply<1>(in, kernel, 1));
            add_max(&out, tmp.get());
        }
        if(c.do_l90()){
            std::array<std::array<double, 3>, 3> kernel={{{{-1., 2., -1.}},{{-1., 2., -1.}},{{-1., 2., -1.}}}};
            std::unique_ptr<Image_unit> tmp(apply<1>(in, kernel, 1));
            add_max(&out, tmp.get());
        }
        if(c.do_l135()){
            std::array<std::array<double, 3>, 3> kernel={{{{2., -1., -1.}},{{-1., 2., -1.}},{{-1., -1., 2.}}}};
            std::unique_ptr<Image_unit> tmp(apply<1>(in, kernel, 1));
            add_max(&out, tmp.get());
        }

        return out;
    }
}

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