zig2matrix.m

来自「matlab中的Zig_Zag扫描程序」· M 代码 · 共 36 行

function Matrix = Zig2Matrix(v_zig)
%将zigzag排列恢复成矩阵Matrix
zigLen = length(v_zig);
zigWidth = sqrt(zigLen);
zigHeight = sqrt(zigLen);                       %计算矩阵Matrix的宽度和高度
i=1;
j=1;
h=2;
Matrix(1,1) = v_zig(1);
while(h <= zigLen)
        if ( ( (mod(i-1,2)==0) && (mod(j-1,2)==0) ) || ( (mod(i-1,2)~=0) && (mod(j-1,2)~=0) ) )
			if( j==(zigWidth) )
                i=i+1;
            end                                 %如果是最后一列，则往下走
			if( i==1 )
                j=j+1; 
            end                                 %如果是第一行，则往右走
			if( (i~=1) && (j~=zigWidth) )
                i = i-1; j = j+1;
            end                                 %既不是第一行，也不是最后一列，则往右上方走
			Matrix(i,j) = v_zig(h);
			h = h+1;
		else
			if( i==zigHeight )
                j=j+1;
            end                                 %如果是最后一行，则往右走
			if( j==1 )
                i=i+1;
            end                                 %如果是第一列，则往下走
			if( (i~=zigHeight) && (j~=1) )
                i = i+1; j = j-1;
            end                                 %既不是第一列，也不是最后一行，则往左下方走
			Matrix(i,j) = v_zig(h);
			h = h+1;
        end
end