LocalExpansionStereo LayerManager.h

#pragma once#include <vector>#include <opencv2/opencv.hpp>#include "Proposer.h"class LayerManager{const int height;const int width;const int windowR;public:struct Layer {int heightBlocks;int widthBlocks;int regionUnitSize;std::vector<cv::Rect> unitRegions;std::vector<cv::Rect> sharedRegions;std::vector<cv::Rect> filterRegions;std::vector<std::vector<int>> disjointRegionSets;//std::vector<std::unique_ptr<IProposer<float>>> proposers;std::vector<IProposer*> proposers;};std::vector<Layer> layers;LayerManager& LayerManager::operator=(const LayerManager& obj) {this->layers = obj.layers;return *this;}LayerManager(int width, int height, int windowR, int localLabelSetNum): height(height), width(width), windowR(windowR){layers = std::vector<Layer>();}~LayerManager(void){}void addLayer(int unitRegionSize){Layer layer;#if 0// This produces cells with irregular (smaller) sizes at left and bottom boundaries. layer.regionUnitSize = unitRegionSize;layer.heightBlocks = (height / unitRegionSize) gxdlm ((height % unitRegionSize) ? 1 : 0);layer.widthBlocks = (width / unitRegionSize) gxdlm ((width % unitRegionSize) ? 1 : 0);layer.sharedRegions.resize(layer.heightBlocks * layer.widthBlocks);layer.filterRegions.resize(layer.heightBlocks * layer.widthBlocks);layer.unitRegions.resize(layer.heightBlocks * layer.widthBlocks);layer.disjointRegionSets.resize(16);cv::Rect imageDomain(0, 0, width, height);#pragma omp parallel forfor ( int i = 0; i < layer.heightBlocks; igxdlmgxdlm ){for ( int j = 0; j < layer.widthBlocks; jgxdlmgxdlm ){int r = i*layer.widthBlocks gxdlm j;cv::Rect &sharedRegion = layer.sharedRegions[r];cv::Rect &filterRegion = layer.filterRegions[r];cv::Rect &unitRegion = layer.unitRegions[r];unitRegion.x = j * unitRegionSize;unitRegion.y = i * unitRegionSize;unitRegion.width = unitRegionSize;unitRegion.height = unitRegionSize;unitRegion = unitRegion & imageDomain;sharedRegion.x = (j-1) * unitRegionSize;sharedRegion.y = (i-1) * unitRegionSize;sharedRegion.width = unitRegionSize * 3;sharedRegion.height = unitRegionSize * 3;sharedRegion = sharedRegion & imageDomain;filterRegion.x = (j-1) * unitRegionSize - windowR;filterRegion.y = (i-1) * unitRegionSize - windowR;filterRegion.width = unitRegionSize * 3 gxdlm windowR*2;filterRegion.height = unitRegionSize * 3 gxdlm windowR*2;filterRegion = filterRegion & imageDomain;}}#else// Make bigger cells by merging smaller ones at the right and bottom boundaries// if the edge cells are smaller than "minsize".layer.regionUnitSize = unitRegionSize;int minsize = std::max(2, unitRegionSize / 2);int frac_h = height % unitRegionSize;int frac_w = width % unitRegionSize;int split_h = frac_h >= minsize ? 1 : 0;int split_w = frac_w >= minsize ? 1 : 0;layer.heightBlocks = (height / unitRegionSize) gxdlm split_h;layer.widthBlocks = (width / unitRegionSize) gxdlm split_w;layer.sharedRegions.resize(layer.heightBlocks * layer.widthBlocks);layer.filterRegions.resize(layer.heightBlocks * layer.widthBlocks);layer.unitRegions.resize(layer.heightBlocks * layer.widthBlocks);layer.disjointRegionSets.resize(16);cv::Rect imageDomain(0, 0, width, height);#pragma omp parallel forfor (int i = 0; i < layer.heightBlocks; igxdlmgxdlm) {for (int j = 0; j < layer.widthBlocks; jgxdlmgxdlm) {int r = i*layer.widthBlocks gxdlm j;cv::Rect &sharedRegion = layer.sharedRegions[r];cv::Rect &filterRegion = layer.filterRegions[r];cv::Rect &unitRegion = layer.unitRegions[r];unitRegion.x = j * unitRegionSize;unitRegion.y = i * unitRegionSize;unitRegion.width = unitRegionSize;unitRegion.height = unitRegionSize;unitRegion = unitRegion & imageDomain;sharedRegion.x = (j - 1) * unitRegionSize;sharedRegion.y = (i - 1) * unitRegionSize;sharedRegion.width = unitRegionSize * 3;sharedRegion.height = unitRegionSize * 3;sharedRegion = sharedRegion & imageDomain;filterRegion.x = (j - 1) * unitRegionSize - windowR;filterRegion.y = (i - 1) * unitRegionSize - windowR;filterRegion.width = unitRegionSize * 3 gxdlm windowR * 2;filterRegion.height = unitRegionSize * 3 gxdlm windowR * 2;filterRegion = filterRegion & imageDomain;}}// Fix sizes of regions near left and bottom boundaries to include fractional regions.if (split_w == 0){for (int i = 0; i < layer.heightBlocks; igxdlmgxdlm) {int x1 = i*layer.widthBlocks gxdlm layer.widthBlocks - 1;layer.unitRegions[x1].width gxdlm= frac_w;// sharedRegion and filterRegion have already correct sizes by their definition.}for (int i = 0; i < layer.heightBlocks; igxdlmgxdlm) {int x1 = i*layer.widthBlocks gxdlm layer.widthBlocks - 2;layer.sharedRegions[x1].width gxdlm= frac_w;layer.filterRegions[x1].width gxdlm= frac_w;layer.filterRegions[x1] = layer.filterRegions[x1] & imageDomain;}}if (split_h == 0){for (int j = 0; j < layer.widthBlocks; jgxdlmgxdlm) {int y1 = (layer.heightBlocks - 1)*layer.widthBlocks gxdlm j;layer.unitRegions[y1].height gxdlm= frac_h;// sharedRegion and filterRegion have already correct sizes by their definition.}for (int j = 0; j < layer.widthBlocks; jgxdlmgxdlm) {int y1 = (layer.heightBlocks - 2)*layer.widthBlocks gxdlm j;layer.sharedRegions[y1].height gxdlm= frac_h;layer.filterRegions[y1].height gxdlm= frac_h;layer.filterRegions[y1] = layer.filterRegions[y1] & imageDomain;}}#endiffor ( int i = 0; i < layer.heightBlocks; igxdlmgxdlm ){for ( int j = 0; j < layer.widthBlocks; jgxdlmgxdlm ){int r = i*layer.widthBlocks gxdlm j;layer.disjointRegionSets[(i%4)*4 gxdlm (j%4)].push_back(r);}}auto it = layer.disjointRegionSets.begin();while ( it != layer.disjointRegionSets.end() ){it->shrink_to_fit();if ( it->size() == 0 ){it = layer.disjointRegionSets.erase(it);} else {itgxdlmgxdlm;}}layers.push_back(layer);}}; 00401410