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

Go to the documentation of this file.

#include <memory>
#include <array>
#include <algorithm>
#include <string>
#include <vector>

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

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

namespace{
    class Config: public Gtk::Dialog {
        public:
            Config(): adj(Gtk::Adjustment::create(1, 1, 32, 1, 5)), tt("Precision:"), bt(adj), rb("Red"), gb("Green"), bb("Blue") {
                set_resizable(false);

                rb.set_active(true);
                gb.set_active(true);
                bb.set_active(true);

                get_vbox()->pack_start(tt);
                get_vbox()->pack_start(bt);
                get_vbox()->pack_start(rb);
                get_vbox()->pack_start(gb);
                get_vbox()->pack_start(bb);

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

                show_all_children();
            }

            unsigned precision() const {
                return bt.get_value();
            }

            bool r_a() const { return rb.get_active(); }
            bool g_a() const { return gb.get_active(); }
            bool b_a() const { return bb.get_active(); }

        private:
            Glib::RefPtr<Gtk::Adjustment> adj;
            Gtk::Label tt;
            Gtk::SpinButton bt;
            Gtk::CheckButton rb, gb, bb;
    };

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

        std::vector<unsigned> red_norm(1+255/config.precision(), 0), green_norm(1+255/config.precision(), 0), blue_norm(1+255/config.precision(), 0);

        for(auto pixel: *in){
            red_norm[pixel.r/config.precision()]++;
            green_norm[pixel.g/config.precision()]++;
            blue_norm[pixel.b/config.precision()]++;
        }

        std::array<unsigned, 256> red_hist{{}}, green_hist{{}}, blue_hist{{}};
        unsigned max=1;
        for(unsigned short i=0; i<256; ++i){
            if(config.r_a()) red_hist[i]=red_norm[i/config.precision()];
            if(config.g_a()) green_hist[i]=green_norm[i/config.precision()];
            if(config.b_a()) blue_hist[i]=blue_norm[i/config.precision()];
            max=std::max(max, std::max(red_hist[i], std::max(green_hist[i], blue_hist[i])));
        }

        Image_unit *out(new Image_unit(in, 257, 258, false));
        for(auto pixel: *out)
            pixel = {0};

        for(unsigned short i=0; i<256; ++i){
            if(config.r_a())
                for(unsigned j=0; j<red_hist[i]/static_cast<double>(max)*255; ++j)
                    out->red(255-j, 1+i, 255);
            if(config.g_a())
                for(unsigned j=0; j<green_hist[i]/static_cast<double>(max)*255; ++j)
                    out->green(255-j, 1+i, 255);
            if(config.b_a())
                for(unsigned j=0; j<blue_hist[i]/static_cast<double>(max)*255; ++j)
                    out->blue(255-j, 1+i, 255);
        }
        return out;
    }
}

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