disparity.h

来自「一个基于MFC的SIFT图像配准算法代码。好用」· C头文件 代码 · 共 44 行

// Disparity.h: interface for the CDisparity class.
//
//////////////////////////////////////////////////////////////////////

#if !defined(AFX_DISPARITY_H__0DDEC43D_0AF4_496B_B3EF_EA6A37DFED23__INCLUDED_)
#define AFX_DISPARITY_H__0DDEC43D_0AF4_496B_B3EF_EA6A37DFED23__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000


#include "inc/cv.h"
#include "inc/highgui.h"

#include "keypoint.h"
#include "Img.h"
#include "heap.h"
#include "color.h"

#define	MAP(x,y)	(map[(y)][(x)])
#define	MAPR(x,y)	(map_r[(y)][(x)])

class CDisparity  
{
	public:
	int		width, height;
	CvPoint **map;				//视差图
	bool	**map_r;
	CHeap	heap;				//储存种子的堆结构
	CColor	color;				//随机颜色发生器(好看而已)
public:
	CDisparity();
	virtual ~CDisparity();
    void Initialization(int w_l, int h_l, int w_r, int h_r);	//初始化视差图
	void SetSeed(IplImage *l_img, CKeypoint *l_kp, int l_n_kp, IplImage *r_img, CKeypoint *r_kp, int r_n_kp);
	void DrawMatchedKp(IplImage *l_img, IplImage *r_img);
	void Propagation(IplImage *l_img, IplImage *r_img);
	void ShowMatch(IplImage *show, int w, int h);
	void ShowDisp(IplImage *show, int w, int h);
};

#endif // !defined(AFX_DISPARITY_H__0DDEC43D_0AF4_496B_B3EF_EA6A37DFED23__INCLUDED_)