LocalBinaryPattern.cpp

#include "LocalBinaryPattern.h"
#include "writeCsv.h"

#ifdef _OPENMP

#include <omp.h>

#endif

Mat localBinaryPattern(Mat &imgIn, int numThreads) {
    const int weights[8] = {1, 2, 4, 8, 16, 32, 64, 128};
    int histogram[256] = {0};

    Mat imgOut = Mat::zeros(imgIn.rows, imgIn.cols, CV_8UC1);
    Mat workingImg;
    copyMakeBorder(imgIn, workingImg, 1, 1, 1, 1, BORDER_CONSTANT, 0);

    #ifdef _OPENMP
    //std::cout << "_OPENMP defined" << std::endl;
    //std::cout << "Num processors (Phys+HT): " << omp_get_num_procs() << std::endl;
    omp_set_num_threads(numThreads);
    #endif

    #pragma omp parallel default(none) reduction(+: histogram) shared(workingImg, imgIn, weights, imgOut)
    {
        #pragma omp for //collapse(2)
        for (int i = 1; i < imgIn.rows; i++) {
            for (int j = 1; j < imgIn.cols; j++) {
                int neighbors[8];
                neighbors[0] = workingImg.at<uchar>(i - 1, j - 1);
                neighbors[1] = workingImg.at<uchar>(i - 1, j);
                neighbors[2] = workingImg.at<uchar>(i - 1, j + 1);
                neighbors[3] = workingImg.at<uchar>(i, j + 1);
                neighbors[4] = workingImg.at<uchar>(i + 1, j + 1);
                neighbors[5] = workingImg.at<uchar>(i + 1, j);
                neighbors[6] = workingImg.at<uchar>(i + 1, j - 1);
                neighbors[7] = workingImg.at<uchar>(i, j - 1);

                int oldVal = workingImg.at<uchar>(i, j);

                int newVal = 0;
                for (int k = 0; k < 8; k++) {
                    if (neighbors[k] >= oldVal)
                        newVal += weights[k];
                }
                imgOut.at<uchar>(i - 1, j - 1) = newVal;
                histogram[newVal]++;
            }
        }
    }

    writeCsv(histogram);
    return imgOut;
}