qiezi的学习园地

AS/C/C++/D/Java/JS/Python/Ruby

  C++博客 :: 首页 :: 新随笔 ::  ::  :: 管理 ::

矩阵就不用再解释了,写成泛型主要是为了几个方便:
1、方便在栈上分配空间。由于维度在编译期已知,所以可以做到在栈上分配空间。当然如果这个对象是new出来的,自然是在堆上分配,这里说的是在栈上分配这个对象时,矩阵元素所占用的空间也在栈上分配。
2、方便在编译期检查非法的矩阵运算。C++模板的强大推导能力可以在编译期推导出结果矩阵的维度。
3、泛型类在方法内联上具有优势。

这个矩阵类为了能够直接从数组赋值,使用了一个ArrayPorxy类(可参考《Imperfect C++》)。

代码如下:

template  < class  T,  int  D1,  int  D2 = 1 >
class  ArrayProxy
{
    T
*  data;
public :
    ArrayProxy(T (
& value)[D1][D2])
        : data(
& value[ 0 ][ 0 ])
    {
    }

    ArrayProxy(T (
& value)[D1 * D2])
        : data(value)
    {
    }

    T
*  getData()  const
    {
        
return  data;
    }
};

这个只是简单的实现。

因为我基本上不使用这个矩阵类,所以只完成几个简单功能:
1、从数组赋值:
int a[][3] = {{1,2,3}, {4,5,6}};
Matrix<int, 2, 3> m1(a);

int a[] = {1,2,3, 4,5,6};
Matrix<int, 2, 3> m1(a);
Matrix<int, 3, 2> m2(a);
Matrix<int, 6, 1> m3(a);
Matrix<int, 1, 6> m4(a);

2、矩阵乘法:
Matrix<int, 2, 3> m1;
Matrix<int, 2, 4> m2;
// m1 * m2  <== 编译错误,维度不匹配
Matrix<int, 3, 5> m3;
Matrix<int, 2, 5> m4 = m1 * m3; // <== 合法
// m3 * m1; // <== 编译错误,维度不匹配

源码如下:

template <class T, int R, int C>
class Matrix
{
    T matrix[R][C];

public:
    
// Big three
    Matrix(void)
    {
        memset(matrix, 
0sizeof(matrix));
    }

    Matrix(
const Matrix& rhs)
    {
        memcpy(matrix, rhs.matrix, 
sizeof(matrix));
    }

    Matrix
& operator =(const Matrix& rhs)
    {
        memcpy(matrix, rhs.matrix, 
sizeof(matrix));
        
return *this;
    }

public:
    Matrix(
const ArrayProxy<T,R,C>& arr)
    {
        memcpy(matrix, arr.getData(), 
sizeof(matrix));
    }

    
~Matrix(void)
    {
    }

public:
    T 
get(int r, int c) const
    {
        assert(c 
< C && c >= 0 && r < R && r >= 0);
        
return matrix[r][c];
    }

    
void set(int r, int c, T v)
    {
        assert(c 
< C && c >= 0 && r < R && r >= 0);
        matrix[r][c] 
= v;
    }

    
int getCols () const
    {
        
return C;
    }

    
int getRows () const
    {
        
return R;
    }

    
bool operator == (const Matrix& rhs) const
    {
        
return memcmp(matrix, rhs.matrix, sizeof(matrix)) == 0;
    }

    
bool operator != (const Matrix& rhs) const
    {
        
return !(*this == rhs);
    }
};

template 
<class T, int R, int C, int C1>
Matrix
<T,R,C1> operator* (const Matrix<T,R,C>& lhs, const Matrix<T,C,C1>& rhs)
{
    Matrix
<T,R,C1> result;
    
for (int r=0; r<R; ++r)
    {
        
for (int c=0; c<C1; ++c)
        {
            
int value = 0;
            
for (int i=0; i<C; ++i)
            {
                value 
+= lhs.get(r,i) * rhs.get(i,c);
            }
            result.
set(r,c,value);
        }
    }
    
return result;
}

测试代码:

int main()
{
    {
        
// 测试初始化
        Matrix<int34> m1;
        Matrix
<int34> m2(m1);
        Matrix
<int34> m3 = m1;
        Matrix
<int34> m4;
        m4 
= m1;

        
for (int i=0; i<3; i++)
            
for (int j=0; j<4; j++)
            {
                assert (m1.
get(i, j) == 0);
                assert (m2.
get(i, j) == 0);
                assert (m3.
get(i, j) == 0);
                assert (m4.
get(i, j) == 0);
            }

        
int a[] = {1,2,3,45,6,7,89,10,11,12};
        Matrix
<int34> m5(a);

        
int b[3][4= { {1,2,3,4},
                        {
5,6,7,8},
                        {
9,10,11,12}};

        Matrix
<int34> m6(b);

        Matrix
<int34> m7(m5);
        Matrix
<int34> m8 = m5;
        Matrix
<int34> m9;
        m9 
= m5;

        
for (int i=0; i<3; i++)
            
for (int j=0; j<4; j++)
            {
                assert (m5.
get(i, j) == i*4+j+1);
                assert (m6.
get(i, j) == i*4+j+1);
                assert (m7.
get(i, j) == i*4+j+1);
                assert (m8.
get(i, j) == i*4+j+1);
                assert (m9.
get(i, j) == i*4+j+1);
            }

        
// 维数不匹配,编译错误
        
// Matrix<int, 4, 5> m10 = m9;
        int c[][2= {{1,2}, {2,3}};
        
// 数组大小不匹配,编译错误
        
//Matrix<int, 3, 4> m10(c);
        int d[] = {1,2};
        
// 数组大小不匹配,编译错误
        
//Matrix<int, 3, 4> m11(d);

        
// 乘法维数不合适,无法相乘
        
//m1 * m2;

        Matrix
<int,4,3> m12;
        
// 匹配,可以相乘
        Matrix<int33> m13 = m1 * m12;

        Matrix
<int83> m14;
        
// 无法相乘
        
//Matrix<int, 3, 3> m15 = m1 * m14;
        
// 可以相乘
        Matrix<int84> m15 = m14 * m1;
    }

    {
        
// 检查点乘
        int a[2][5= {{1,2,3,4,5}, {6,7,8,9,10}};
        Matrix
<int25> m1(a);

        
int b[5][3= {{1,2,3}, {4,5,6}, {7,8,9}, {10,11,12}, {13,14,15}};
        Matrix
<int53> m2(b);

        
int c[2][3= {{135,150,165}, {310,350,390}};
        Matrix
<int23> m3(c);

        Matrix
<int23> m4 = m1 * m2;
        assert(m4 
== m3);

        cout 
<< m4.get(0,0<< endl;
    }

    
return 0;
}

补充:
1、加法、减法只需要2个矩阵维度相同即可。
template <class T, class R, class C>
Matrix
<T,R,C> operator+ (const Matrix<T,R,C>& lhs, const Matrix<T,R,C>& rhs)
{
   
// 
}

2、由于1x1的矩阵可以看成一个标量,矩阵与标量运算结果维数与原矩阵相同,可以重载来实现。
template <class T, class R, class C>
Matrix
<T,R,C> operator* (const Matrix<T,R,C>& lhs, const Matrix<T,1,1>& rhs)
{
    
// 
}

3、由于类型泛化,可能某些合理的运算无法进行,比如float型矩阵,与一个int型标量运算等。这些最好是借助类型萃取等手段,推导出运算以后的类型。(c++0x中包含自动获取运算结果类型的关键字typeof,等几年就可以用了:)。GCC编译器中已有实现,不过似乎有BUG)。

4、其它。泛型实现可能会有一些考虑不周的地方,强类型有强类型的好处,不过必须要有完整的泛型算法支撑,否则难以使用。也可以把泛型矩阵类从一个普通矩阵类派生,这样更容易写出通用算法,不过在实现上可能要借助于运行期多态,对于矩阵类来说并不合适。

5、其它。。前面说C++的模板相当强大,D语言模板到目前为止似乎已经完全实现了C++模板的功能,还增加了一些比如字符串值参模板等特性,比C++模板功能更多。在代码编写上,由于可以编写静态判断语句(编译期)以及静态断言,编写模板比C++更容易。有时间可以试试用它写个矩阵类,纯粹是兴趣,这些东西真的很难用到,现成的库也挺多。

6、其它。。。c++0x要提供“template typedef”,也就是可以这样定义:
template <int R, int C> typedef Matrix<int, R, C> MatrixInt;  // 定义类型,维度不定
template <class T> typedef Matrix<T, 4, 4> Matrix4x4; // 定义维度,类型不定
由此可以出定义行向量、列向量、标量等,当然实际使用起来可能没那么舒服了。
posted on 2006-04-13 13:52 qiezi 阅读(835) 评论(1)  编辑 收藏 引用 所属分类: C++