idamnjanovic@60: function y = sampgrid(x,blocksize,varargin)
idamnjanovic@60: %SAMPGRID Sample a multi-dimensional matrix on a regular grid.
idamnjanovic@60: %  Y = SAMPGRID(X,BLOCKSIZE,I1,I2,...,Ip) extracts block samples of size
idamnjanovic@60: %  BLOCKSIZE from the p-dimensional matrix X, arranging the samples as the
idamnjanovic@60: %  column vectors of the matrix Y. The locations of the (1,1,..,1)-th
idamnjanovic@60: %  elements of each block are given in the index vectors I1,I2,..Ip. The
idamnjanovic@60: %  total number of samples taken is length(I1)xlength(I2)x...xlength(Ip).
idamnjanovic@60: %  BLOCKSIZE should either be a p-element vector of the form [N1,N2,...Np],
idamnjanovic@60: %  or a scalar N which is shorthand for the square block size [N N ... N].
idamnjanovic@60: %
idamnjanovic@60: %  Example: Sample a set of blocks uniformly from a 2D image.
idamnjanovic@60: %
idamnjanovic@60: %    n = 512; blocknum = 20000; blocksize = [8 8];
idamnjanovic@60: %    im = rand(n,n);
idamnjanovic@60: %    [i1,i2] = reggrid(size(im)-blocksize+1, blocknum);
idamnjanovic@60: %    blocks = sampgrid(im, blocksize, i1, i2);
idamnjanovic@60: %
idamnjanovic@60: %  See also REGGRID.
idamnjanovic@60: 
idamnjanovic@60: %  Ron Rubinstein
idamnjanovic@60: %  Computer Science Department
idamnjanovic@60: %  Technion, Haifa 32000 Israel
idamnjanovic@60: %  ronrubin@cs
idamnjanovic@60: %
idamnjanovic@60: %  November 2007
idamnjanovic@60: 
idamnjanovic@60: 
idamnjanovic@60: p = ndims(x);
idamnjanovic@60: if (p==2 && any(size(x)==1) && length(blocksize)==1)
idamnjanovic@60:   p = 1;
idamnjanovic@60: end
idamnjanovic@60: 
idamnjanovic@60: if (numel(blocksize)==1)
idamnjanovic@60:   blocksize = ones(1,p)*blocksize;
idamnjanovic@60: end
idamnjanovic@60: 
idamnjanovic@60: n = zeros(1,p);
idamnjanovic@60: for i = 1:p
idamnjanovic@60:   n(i) = length(varargin{i});
idamnjanovic@60: end
idamnjanovic@60: 
idamnjanovic@60: nsamps = prod(n);
idamnjanovic@60: 
idamnjanovic@60: % create y of the same class as x
idamnjanovic@60: y = zeros(prod(blocksize),nsamps,class(x));
idamnjanovic@60: 
idamnjanovic@60: % ids() contains the index of the current block in I1..Ip
idamnjanovic@60: ids = ones(p,1);
idamnjanovic@60: 
idamnjanovic@60: % block_ids contains the indices of the current block in X
idamnjanovic@60: block_ids = cell(p,1);
idamnjanovic@60: for j = 1:p
idamnjanovic@60:   block_ids{j} = varargin{j}(1) : varargin{j}(1)+blocksize(j)-1;
idamnjanovic@60: end
idamnjanovic@60: 
idamnjanovic@60: for k = 1:nsamps
idamnjanovic@60:   block = x(block_ids{:});
idamnjanovic@60:   y(:,k) = block(:);
idamnjanovic@60:   
idamnjanovic@60:   % increment ids() and block_ids{}
idamnjanovic@60:   if (k<nsamps)
idamnjanovic@60:     j = 1;
idamnjanovic@60:     while (ids(j) == n(j))
idamnjanovic@60:       ids(j) = 1;
idamnjanovic@60:       block_ids{j} = varargin{j}(1) : varargin{j}(1)+blocksize(j)-1;
idamnjanovic@60:       j = j+1;
idamnjanovic@60:     end
idamnjanovic@60:     ids(j) = ids(j)+1;
idamnjanovic@60:     block_ids{j} = varargin{j}(ids(j)) : varargin{j}(ids(j))+blocksize(j)-1;
idamnjanovic@60:   end
idamnjanovic@60: end
idamnjanovic@60: