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

Go to the documentation of this file.

#include <memory>
#include <sstream>
#include <string>

#include <gtkmm/image.h>
#include <gtkmm/messagedialog.h>
#include <gtkmm/stock.h>
#include <gtkmm/radiobutton.h>
#include <gtkmm/label.h>
#include <gtkmm/entry.h>
#include <gtkmm/table.h>

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

namespace{
    class Config: public Gtk::Dialog {
        public:
            Config(): size("Size:"), dlabel("Divisor:"), threeb("3"), fiveb("5"), sevenb("7"), nineb("9"), threet(3,3), fivet(5,5), sevent(7, 7), ninet(9,9) {
                set_resizable(false);

                Gtk::RadioButton::Group group = threeb.get_group();
                fiveb.set_group(group);
                sevenb.set_group(group);
                nineb.set_group(group);

                threeb.set_active(true);
                threeb.signal_clicked().connect(sigc::mem_fun(*this, &Config::on_three));
                fiveb.signal_clicked().connect(sigc::mem_fun(*this, &Config::on_five));
                sevenb.signal_clicked().connect(sigc::mem_fun(*this, &Config::on_seven));
                nineb.signal_clicked().connect(sigc::mem_fun(*this, &Config::on_nine));

                get_vbox()->pack_start(size);
                get_vbox()->pack_start(threeb);
                get_vbox()->pack_start(fiveb);
                get_vbox()->pack_start(sevenb);
                get_vbox()->pack_start(nineb);

                for(unsigned char y=0; y<3; ++y)
                    for(unsigned char x=0; x<3; ++x){
                        Gtk::Entry &r=three[y][x];
                        r.set_text("0");
                        r.set_width_chars(3);
                        threet.attach(r, y, y+1, x, x+1);
                    }

                for(unsigned char y=0; y<5; ++y)
                    for(unsigned char x=0; x<5; ++x){
                        Gtk::Entry &r=five[y][x];
                        r.set_text("0");
                        r.set_width_chars(3);
                        fivet.attach(r, y, y+1, x, x+1);
                    }

                for(unsigned char y=0; y<7; ++y)
                    for(unsigned char x=0; x<7; ++x){
                        Gtk::Entry &r=seven[y][x];
                        r.set_text("0");
                        r.set_width_chars(3);
                        sevent.attach(r, y, y+1, x, x+1);
                    }

                for(unsigned char y=0; y<9; ++y)
                    for(unsigned char x=0; x<9; ++x){
                        Gtk::Entry &r=nine[y][x];
                        r.set_text("0");
                        r.set_width_chars(3);
                        ninet.attach(r, y, y+1, x, x+1);
                    }

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

                get_vbox()->pack_start(threet);
                get_vbox()->pack_start(fivet);
                get_vbox()->pack_start(sevent);
                get_vbox()->pack_start(ninet);

                get_vbox()->pack_start(dlabel);
                divent.set_text("1");
                divent.set_width_chars(3);
                get_vbox()->pack_start(divent);

                show_all_children();
                on_three();
            }

            void on_three(){
                threet.show_all();
                fivet.hide();
                sevent.hide();
                ninet.hide();
            }
            void on_five(){
                threet.hide();
                fivet.show_all();
                sevent.hide();
                ninet.hide();
            }
            void on_seven(){
                threet.hide();
                fivet.hide();
                sevent.show_all();
                ninet.hide();
            }
            void on_nine(){
                threet.hide();
                fivet.hide();
                sevent.hide();
                ninet.show_all();
            }

            unsigned char size_a() const {
                return threeb.get_active()?3:fiveb.get_active()?5:sevenb.get_active()?7:nineb.get_active()?9:0;
            }

            std::array<std::array<double, 3>, 3> get_3k() const {
                std::array<std::array<double, 3>, 3> r;
                for(unsigned char y=0; y<3; ++y)
                    for(unsigned char x=0; x<3; ++x){
                        std::stringstream ss;
                        ss << three[y][x].get_text();
                        if(!(ss >> r[y][x]))
                            throw std::runtime_error("Not a number");
                    }
                return r;
            }

            std::array<std::array<double, 5>, 5> get_5k() const {
                std::array<std::array<double, 5>, 5> r;
                for(unsigned char y=0; y<5; ++y)
                    for(unsigned char x=0; x<5; ++x){
                        std::stringstream ss;
                        ss << five[y][x].get_text();
                        if(!(ss >> r[y][x]))
                            throw std::runtime_error("Not a number");
                    }
                return r;
            }

            std::array<std::array<double, 7>, 7> get_7k() const {
                std::array<std::array<double, 7>, 7> r;
                for(unsigned char y=0; y<7; ++y)
                    for(unsigned char x=0; x<7; ++x){
                        std::stringstream ss;
                        ss << seven[y][x].get_text();
                        if(!(ss >> r[y][x]))
                            throw std::runtime_error("Not a number");
                    }
                return r;
            }

            std::array<std::array<double, 9>, 9> get_9k() const {
                std::array<std::array<double, 9>, 9> r;
                for(unsigned char y=0; y<9; ++y)
                    for(unsigned char x=0; x<9; ++x){
                        std::stringstream ss;
                        ss << nine[y][x].get_text();
                        if(!(ss >> r[y][x]))
                            throw std::runtime_error("Not a number");
                    }
                return r;
            }

            double get_d() const {
                std::stringstream ss;
                ss << divent.get_text();
                double r;
                ss >> r;
                if(!r)
                    throw std::runtime_error("Divisor is 0");
                return r;
            }

        private:
            Gtk::Label size, dlabel;
            Gtk::RadioButton threeb, fiveb, sevenb, nineb;
            Gtk::Table threet, fivet, sevent, ninet;
            std::array<std::array<Gtk::Entry, 3>, 3> three;
            std::array<std::array<Gtk::Entry, 5>, 5> five;
            std::array<std::array<Gtk::Entry, 7>, 7> seven;
            std::array<std::array<Gtk::Entry, 9>, 9> nine;
            Gtk::Entry divent;
    };
    
    Display_unit *impl(const Image_unit *in){
        Config c;
        if(c.run()!=Gtk::RESPONSE_OK)
            return nullptr;
        c.hide();

        switch(c.size_a()){
            case 3:
                return apply<1>(in, c.get_3k(), c.get_d());
            case 5:
                return apply<2>(in, c.get_5k(), c.get_d());
            case 7:
                return apply<3>(in, c.get_7k(), c.get_d());
            case 9:
                return apply<4>(in, c.get_9k(), c.get_d());
            default:
                return nullptr;
        }
    }
}

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