vrep.m

来自「JLAB is a set of Matlab functions I have」· M 代码 · 共 54 行

function[varargout]=vrep(varargin)
%VREP  Replicates an array along a specified dimension.
%
%   Y=VREP(X,N,DIM) replicates the (1-D) array by N times along 
%   dimension DIM.  For instance:   
%                                                                         
%        VREP([1:4]',3,2)=[ [1:4]' [1:4]' [1:4]' ]                            
%                                                                         
%   This is often useful in array algebra.                                
%                                                                         
%   See also VINDEX, DIM.      
%   __________________________________________________________________
%   This is part of JLAB --- type 'help jlab' for more information
%   (C) 2001--2006 J.M. Lilly --- type 'help jlab_license' for details    

if strcmp(varargin{1}, '--t')
  vrep_test,return
end

n=varargin{end-1};
ndim=varargin{end};

for i=1:length(varargin)-2
  varargout{i}=vrep1(varargin{i},n,ndim);
end

eval(to_overwrite(nargin-2))
 
%You would think Matlab would provide a simpler way to do this.
function[y]=vrep1(x,n,dim)
  
str=['y=repmat(x,['];
ndx=ndims(x);
for i=1:max(ndx,dim)
    if i~=dim
        str=[str '1,'];
    else
	str=[str 'n,'];
    end
end
str=[str(1:end-1) ']);'];
eval(str);


function[]=vrep_test

ans1=vrep([1:4]',3,2);
ans2=[ [1:4]' [1:4]' [1:4]' ];
reporttest('VREP', aresame(ans1,ans2))

x1=[1:4]';x2=[1:4]';
vrep(x1,x2,3,2);
reporttest('VREP output redirect', aresame(x1,ans1) && aresame(x2,ans2))