rxxopticalflow.h

来自「3D reconstruction, medical image process」· C头文件 代码 · 共 56 行

class RxVolumeData;
class RxTransform3DInfo;

class RxxOpticalFlow
{
private:
	int m_iRefVolX;
	int m_iRefVolY;
	int m_iRefVolZ;
	int m_iFloatVolX;
	int m_iFloatVolY;
	int m_iFloatVolZ;
	int m_iVolumeSizeInByte;

	short m_nIteration;
	double m_lfAlpha;
	double m_lfMaxDeform; // The assumption of two times of maximum deformation magnitude

	unsigned char* m_pIGX;
	unsigned char* m_pIGY;
	unsigned char* m_pIGZ;
	unsigned char* m_pIGT;

	unsigned short* m_pU;
	unsigned short* m_pV;
	unsigned short* m_pW;

//	float* m_pU;
//	float* m_pV;
//	float* m_pW;

	RxVolumeData* m_pRefVolumeData;	
	RxVolumeData* m_pFloatVolumeData;

	unsigned short *m_pRefVol;
	unsigned short *m_pFloatVol;

private:
	void CalculateIntensityGradient(RxTransform3DInfo* pTransform);
	void UpdateDeformationField(RxTransform3DInfo* pTransform);
	void UpdateDeformationFieldHighPrecision(RxTransform3DInfo* pTransform);
	void VisualizeDeformationField(void);
	void VisualizeDeformationFieldHighPrecision(void);
	void UpdateVolumeFromSegmentation(void);

public:
	RxxOpticalFlow();
	~RxxOpticalFlow();
	void PerformOpticalFlow(RxTransform3DInfo* pTransform);
	void GetLocalDeformation(int iX, int iY, int iZ, double* u, double* v, double* w);
};

/*
	Float狼 Ref肺狼 local deformation阑 拌魂
*/