segmentfiller.cpp

/*
 * Author: Artur Dobrogowski
 * 2020-01-04
 */

#include "segmentfiller.h"
SegmentFiller::SegmentFiller(cv::Mat &vm, Segment &s) : segment(s) {
    mask_ptr = vm.ptr<unsigned char>(0);
    n_rows = vm.rows;
    n_cols = vm.cols;
    printf("%d %d\n", n_rows, n_cols);
}

    
void SegmentFiller::operator()(Coord x, Coord y) {
    // Process the pixel as segment fragment
    segment.addPoint(x, y, *(mask_ptr + n_cols*y + x));
    *(mask_ptr + n_cols*y + x) = 0;
    int neighbors = 0;
    for(int i=0; i < 4; ++i) {
        const Coord new_x = x_translation[i] + x;
        const Coord new_y = y_translation[i] + y;
        
        // Validate coordinates
        if(new_x > 0 && new_x < n_cols && new_y > 0 && new_y < n_rows) {
            if(segment.origin.ptr<unsigned char>(new_y)[new_x] > 0)
                ++neighbors;
            
            // If mask not yet processed
            if (*(mask_ptr + n_cols * new_y + new_x) > 0)
                
                // Process pixel of new coordinates as segment fragment
                this->operator()(new_x, new_y);
        }
    }
    if(neighbors < 4) {
        ++(segment.circumference);
    }
}