#include <iostream>

#include <cstdlib>

#include <string>

#include <omp.h>

#include "convolution.hpp"

using namespace std;

int* convolute_image(int *img, int *dims, double **kernel, int k) {

double a;

int *filtered;

filtered = (int*) malloc(dims[0] * dims[1] * sizeof(int));

#pragma omp parallel for shared(filtered) private(a)

for(int i = 0; i < dims[0]; i++) {

for(int j = 0; j < dims[1]; j++) {

a = 0.0;

for(int ki = -(k/2); ki <= (k/2); ki++) {

for(int kj = -(k/2); kj <= (k/2); kj++) {

if((i + ki < 0) || (j + kj < 0) || (i + ki > dims[0] - 1) || (j + kj > dims[1] - 1)) {

continue;

} else {

double kval = kernel[kj + (k/2)][ki + (k/2)];

double ival = img[(i + ki) * dims[1] + (j + kj)];

a += kval * ival;

}

}

}

if(a > 255) a = 255;

if(a < 0) a = 0;

filtered[(i * dims[1]) + j] = (int) a;

}

}

return filtered;

}