matrix2d.h

小波图像编码中SPIHT算法的C++实现源码

// Matrix2D.h: interface for the CMatrix2D class.
//
//////////////////////////////////////////////////////////////////////

#if !defined(AFX_MATRIX2D_H__861EEFD7_9B24_11D5_B119_5254AB32C4D0__INCLUDED_)
#define AFX_MATRIX2D_H__861EEFD7_9B24_11D5_B119_5254AB32C4D0__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000
#pragma optimize("", off)

//////////////////////
/*#include <stdio.h>
#include <stdlib.h>*/
//////////////////////
#include <math.h>
#include <memory.h>
typedef int ElementType;

#define MinElementType -32768
#define MaxElementType  32767

template <class T>
class CMatrix2D  
{
public:
	void LoadData(T *example, int r,int c);
	void Destroy();
	void Clear();
	void Create(int r, int c);
	T MaxMagnitude();
	T **m; 
	int row;
	int col;
	CMatrix2D();
	virtual ~CMatrix2D();
private:
	T* buffer;
};

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////
template <class T>
CMatrix2D<T>::CMatrix2D():col(0),row(0),m(NULL),buffer(NULL)
{
}

template <class T>
CMatrix2D<T>::~CMatrix2D()
{
}

template <class T>
void CMatrix2D<T>::Create(int r, int c)
{
    row = r;
    col = c;
	buffer = new T[row*col];
	ASSERT(buffer);
	m = new T*[r];
	ASSERT(m);
    for (int i=0; i<r; i++)
		m[i] = buffer+col*i; 
}

template <class T>
void CMatrix2D<T>::Clear()
{
    memset(buffer,0,col*row*sizeof(T));
}

/*
 * Returns max(abs(matrix)), the absolute maximum value of a matrix.
 */ 

template <class T>
T CMatrix2D<T>::MaxMagnitude()
{
	T max = 0;
	for(int i=0;i<col*row;i++)
		if(buffer[i]>max || buffer[i]<-max)
			max = buffer[i]>=0?buffer[i]:-buffer[i];
	return max;
}

template <class T>
void CMatrix2D<T>::Destroy()
{

  delete []buffer;
  delete []m;
  buffer = NULL;
  m = NULL;
}

template <class T>
void CMatrix2D<T>::LoadData(T *example, int r, int c)
{
	memcpy(buffer,example,r*c*sizeof(T));
}
#pragma optimize("", on)

#endif // !defined(AFX_MATRIX2D_H__861EEFD7_9B24_11D5_B119_5254AB32C4D0__INCLUDED_)