semi_global_matching.h

#ifndef __SEMI_GLOBAL_MATCHING_H__
#define __SEMI_GLOBAL_MATCHING_H__

#include <opencv2/core.hpp>

class SemiGlobalMatching
{
public:

	static const int DISP_SHIFT = 4;
	static const int DISP_SCALE = (1 << DISP_SHIFT);
	static const int DISP_INV = -DISP_SCALE;

	enum CensusType
	{
		CENSUS_9x7,
		SYMMETRIC_CENSUS_9x7,
	};

	enum PathType
	{
		SCAN_4PATH,
		SCAN_8PATH,
	};

	struct Parameters
	{
		int P1;
		int P2;
		int numDisparities;
		float uniquenessRatio;
		int max12Diff;
		int medianKernelSize;
		CensusType censusType;
		PathType pathType;

		// default settings
		Parameters()
		{
			P1 = 10;
			P2 = 120;
			numDisparities = 64;
			uniquenessRatio = 0.95f;
			max12Diff = 5;
			medianKernelSize = 3;
			censusType = SYMMETRIC_CENSUS_9x7;
			pathType = SCAN_8PATH;
		}
	};

	SemiGlobalMatching(const Parameters& param = Parameters());
	void compute(const cv::Mat& I1, const cv::Mat& I2, cv::Mat& D1, cv::Mat& D2);

private:

	cv::Mat census[2];
	std::vector<cv::Mat1b> L;
	cv::Mat1w S;

	Parameters param_;
};

#endif // !__SEMI_GLOBAL_MATCHING_H__