main.cpp

/* 
    Author: Thibaut Temkeng
    Email: Temkengthibaut@gmail.com
    Created: 2022-01-21 02:41
 */
#include "images.h"
#include <iostream>

#define debug(x) DEBUG(x, "MAIN|> ")
using namespace std;


int main() {
    std::cout << "IMAGE PROCESSING!" << std::endl;
    string test("La vie est belle !!!");
    debug(test)
    clock_t t = clock();
    srand(time(nullptr));
//    Image img("data/Alida.jpg");
   Image img("data/orig.png");
//    Image img("data/pp.jpg");
//    Image img("data/screenshot.png");
    // Image img("data/profile.jpg");
//    Image img("data/g-star-shoe.jpg");
//    Image img(500, 500, 3);
    debug(img)
    // Image flip = img.flip_v();
    // flip.write("data/flip_vertical.png");
    // Image flip_h = img.flip_h();
    // flip_h.write("data/flip_horizontal.png");
    // Image rot_90 = img.rot90();;
    // rot_90.write("data/rot_90.png");
    // Image concat_v = flip.concat_v(flip_h);
    // concat_v.write("data/concat_vertical.png");
    // Image concat_h = flip.concat_h(flip_h);
    // concat_h.write("data/concat_horizontal.png");
//    Image seam = img.shrink_n();
//    seam.write("data/seam.png");
//    Tree tmp2 = Tree(positions);

    Image gray = img.brightness();
//    gray.getHistogram().plot();
     gray.write("data/gray.png");
    int step = 20;
    Image sobel = img.sobel();
    Image binary = sobel.threshold(10);
    Histogram histo = sobel.getHistogram();
    int row_step = ceil(1.*binary.rows_() / step);
    int col_step = ceil(1.*binary.cols_() / step);
    debug(row_step)
    debug(col_step)

    int counter[row_step*col_step];
    DEBUG(counter, "MAIN")
    // histo.plot();
    // histo.info();
    sobel.write("data/sobel.png");
     binary.write("data/threshold.png");
     img.sobel().write("data/sobel.png");
     gray.gauss(3).write("data/gaussian_blur.png");
     img.luminance(1, 1, 1.2).write("data/luminance.png");

    //  Image zp = img.add_padding(10, 10);
    //  zp.write("data/padding_add.png");
    //  zp.remove_padding(10, 10).write("data/padding_remove.png");


//    float kernel[] = {0.075, 0.125, 0.075, 0.125, 0.200, 0.125, 0.075, 0.125, 0.075};
//    float kernel[] = {-1, -2, 0, -2, 0, 2, 0, 2, 1};
    float kernel_[] = {1, 0, -1};

    std::cout << binary << std::endl;

    auto values = static_cast<float *>(init("float", binary.getChannels()));
    int count, index;
    float pixel_value;
    for (int i = 0; i < binary.rows_(); i +=step) {
        for (int j = 0; j < binary.cols_(); j++) {
            count =0;
            for (int k = 0; k < binary.getChannels(); k++) {
                pixel_value = binary.at(i, j, k);
                if (pixel_value != 0){
                    count++;
                }
            }
            index =  (col_step * (i - i % step) + (j - j % step))/step;
            // printf("Index: %i i: %i j: %i  ", index, i, j);
            counter[index] += count;
        }
    }
//    print(counter, col_step, row_step);
    for (int i = 0; i < binary.rows_(); i +=step) {
        for (int j = 0; j < binary.cols_(); j++) {
            for (int k = 0; k < binary.getChannels(); k++) {
                // values[j] = rand() % 256;
                values[k]  = 255;
            }
            binary.setPixel(i, j, values);
            // img.set(i, img.cols_() - i, values);
        }
    }

    for (int j = 0; j < binary.cols_(); j +=step) {
        for (int i = 0; i < binary.rows_(); i++) {
            for (int k = 0; k < binary.getChannels(); k++) {
                // values[j] = rand() % 256;
                values[k]  = 255;
            }
            binary.setPixel(i, j, values);
            // img.set(i, img.cols_() - i, values);
        }
    }

    binary.write("data/threshold.png");    

//    img.write("data/zeros.png");

     img.autoContrast().write("data/autocontrast.png");
     img.equalize().write("data/equalize.png");

//    Image cpy = img.copy();
//    for (int i = 0; i < img.rows_() / 2; i++) {
//        for (int k = 0; k < 2; ++k) {
//            for (int j = 0; j < img.getChannels(); j++) {
//                values[j] = (rand() % 256);
//                //            values[j]  = 255;
//            }
//            int index = rand() % cpy.cols_();
//            cpy.set(i, index, values);
//            cpy.set(i, cpy.cols_() - index - 1, values);
//            cpy.set(img.rows_() - i - 1, index, values);
//            cpy.set(img.rows_() - i - 1, cpy.cols_() - index - 1, values);
//        }
//    }
//    cpy.write("data/copy.png");
//
    Image blur = gray.blur();
     blur.write("data/blur.png");
     Image convolve1 = blur.convolve(kernel_, 3, 1);
     convolve1.write("data/convolve_y.png");
     Image convolve2 = blur.convolve(kernel_, 1, 3);
     convolve2.write("data/convolve_x.png");
     //    Image add_test = Image::add(convolve2, convolve1);
     Image add_test = Image::add(convolve2, convolve1);
     add_test.write("data/convolve_add.png");

     Image convolve = convolve1 * convolve2;
     convolve.norm();
     convolve.write("data/corner.png");
     img.invert().write("data/invert.png");
     img.contrast(0.5).write("data/contrast.png");
     img.threshold(20).write("data/threshold.png");
     img.clamping(20, 180).write("data/clamping.png");

    //  img.show_seam();
    //  img.write("data/show_seam.png");
    double time_taken = ((double) (clock() - t)) / CLOCKS_PER_SEC; // Calculate the elapsed time
    printf("\nExecution time %.6f second\n", time_taken);

//
    printf("Finish\n");

    return 0;
}