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

Go to the documentation of this file.

#include <memory>
#include <string>

#include <gtkmm/image.h>
#include <gtkmm/enums.h>
#include <gtkmm/dialog.h>
#include <gtkmm/table.h>
#include <gtkmm/checkbutton.h>
#include <gtkmm/stock.h>
#include <gtkmm/label.h>
#include <gtkmm/scale.h>

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

namespace{
    class Config: public Gtk::Dialog {
        public:
            Config(): randomisation_adjustment(Gtk::Adjustment::create(0, 0, 1, 0.01, 0.1)), color_red_adjustment(Gtk::Adjustment::create(0, 0, 1, 0.01, 0.1)),
                      color_green_adjustment(Gtk::Adjustment::create(0, 0, 1, 0.01, 0.1)), color_blue_adjustment(Gtk::Adjustment::create(0, 0, 1, 0.01, 0.1)),
                      randomisation_widget(randomisation_adjustment), color_red_widget(color_red_adjustment),
                      color_green_widget(color_green_adjustment), color_blue_widget(color_blue_adjustment),
                      randomisation_label("Randomisation"), color_red_label("Red"),
                      color_green_label("Green"), color_blue_label("Blue"),
                      color_box(3, 2), color_widget("Color") {

                set_resizable(false);

                color_widget.set_active();
                color_widget.signal_clicked().connect(sigc::mem_fun(*this, &Config::on_set_color));
                get_vbox()->pack_start(color_widget);

                randomisation_widget.set_digits(2);
                randomisation_widget.set_value_pos(Gtk::POS_RIGHT);
                randomisation_widget.set_draw_value();
                randomisation_widget.set_size_request(100, 30);
                randomisation_box.pack_start(randomisation_label);
                randomisation_box.pack_end(randomisation_widget);
                get_vbox()->pack_start(randomisation_box);
                
                color_red_widget.set_digits(2);
                color_red_widget.set_value_pos(Gtk::POS_RIGHT);
                color_red_widget.set_draw_value();
                color_red_widget.set_size_request(100, 30);
                color_box.attach(color_red_label, 0, 1, 0, 1);
                color_box.attach(color_red_widget, 1, 2, 0, 1);

                color_green_widget.set_digits(2);
                color_green_widget.set_value_pos(Gtk::POS_RIGHT);
                color_green_widget.set_draw_value();
                color_green_widget.set_size_request(100, 30);
                color_box.attach(color_green_label, 0, 1, 1, 2);
                color_box.attach(color_green_widget, 1, 2, 1, 2);

                color_blue_widget.set_digits(2);
                color_blue_widget.set_value_pos(Gtk::POS_RIGHT);
                color_blue_widget.set_draw_value();
                color_blue_widget.set_size_request(100, 30);
                color_box.attach(color_blue_label, 0, 1, 2, 3);
                color_box.attach(color_blue_widget, 1, 2, 2, 3);

                get_vbox()->pack_start(color_box);

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

                show_all_children();

                on_set_color();
            }

            void on_set_color(){
                if(color_widget.get_active()){
                    randomisation_box.hide();
                    color_box.show();
                    resize(1, 1);
                } else {
                    randomisation_box.show();
                    color_box.hide();
                    resize(1, 1);
                }
            }

            double randomisation() const { return randomisation_widget.get_value(); }
            bool colors() const { return color_widget.get_active(); }
            double red() const { return color_red_widget.get_value(); }
            double green() const { return color_green_widget.get_value(); }
            double blue() const { return color_blue_widget.get_value(); }

        private:
            Glib::RefPtr<Gtk::Adjustment> randomisation_adjustment, color_red_adjustment, color_green_adjustment, color_blue_adjustment;
            Gtk::HScale randomisation_widget, color_red_widget, color_green_widget, color_blue_widget;
            Gtk::Label randomisation_label, color_red_label, color_green_label, color_blue_label;
            Gtk::HBox randomisation_box;
            Gtk::Table color_box;

            Gtk::CheckButton color_widget;
    };

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

        Image_unit *out=new Image_unit(*in);

        for(Image_unit::iterator it=out->begin(); it!=out->end(); ++it)
            if(c.colors()){
                if(std::rand()<c.red()*RAND_MAX)
                    it->r=std::rand()%255;
                if(std::rand()<c.green()*RAND_MAX)
                    it->g=std::rand()%255;
                if(std::rand()<c.blue()*RAND_MAX)
                    it->b=std::rand()%255;
            } else {
                if(std::rand()<c.randomisation()*RAND_MAX)
                    *it = { std::rand()%255 };
            }

        return out;
    }
}

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