cc_image.cpp

#include <iostream>
#include <algorithm>
#include <fstream>
#include <vector>


#include "Base/color.h"
#include "Base/pixel.h"
#include "Base/image.h"

#include "binary_image.h"

#include "cc_image.h"

cc_image::cc_image(Image i): binary_image(i) {
    connectize();
}

void cc_image::connectize() {
    int label = 1;

    for(int i = 0; i < _h; i++) {
        for(int j = 0; j < _w; j++) {
            if(_bin_arr[i][j] == -1) {
                _DFS_Util(i, j, label);
                label += 1;
            }
        }
    }
    _ncc = label - 1;
}

void cc_image::_DFS_Util(int i, int j, int label) {
    _bin_arr[i][j] = label;
    std::vector<std::pair<int,int> > neighbors = _get_neighbors(i, j);

    for(auto & n : neighbors) {
        if(_bin_arr[n.first][n.second] == -1) {
        _DFS_Util(n.first, n.second, label);
        }
    }
    neighbors.clear();
}

std::vector<std::pair<int, int> > cc_image::_get_neighbors(int i, int j) {
    std::vector<std::pair<int,int> > neighbors;
    // int w = get_w(), h = get_h();

    for(int m = -1; m < 2; m++) {
        if(i + m >= 0 && i + m < _h) {
            for(int n = -1; n < 2; n++) {
                if(j + n >= 0 && j + n < _w) {
                    if(m != 0 && n != 0)
                        neighbors.push_back(std::make_pair(i+m, j+n));
                }
            }
        }
    }
    return neighbors;    
}

int cc_image::get_ncc() const {return _ncc;}
/*void cc_image::print() {
    int** bin_arr = get_bin_arr();
    for(int i = 0; i < get_h(); i++) {
        for(int j = 0; j < get_w(); j++) {
            std::cout << bin_arr[i][j] << " ";
        }
        std::cout << std::endl;
    }
}*/

std::ostream& operator<<(std::ostream& os, const cc_image& cc_i) {
    os << "P3" << std::endl;
    os << cc_i.get_w() << " " << cc_i.get_h() << std::endl;
    os << 255 << std::endl;

    Color colors[cc_i._ncc + 1];
    colors[0] = Color(0, 0, 0);

    for(int i = 1; i < cc_i._ncc + 1; i++) {colors[i] = Color(rand()%256, rand()%256, rand()%256);}
    int** bin_arr = cc_i.get_bin_arr();
    for(int i = 0; i < cc_i.get_h(); i++) {
        for(int j = 0; j < cc_i.get_w(); j++) {
            os << colors[bin_arr[i][j]] << " ";
        }
        os << std::endl;
    }
}