view general/algo/gatherer.m @ 6:0ce3c2070089

Removed duplicate code and fixed doc in timed_action.

% gatherer - API for autoexpanding arrays for accumulating data
function api=gatherer(size_in,varargin)
	opts=prefs('chunk',256,'init',512,'grow',2,'max',1e9,varargin{:});
	n=uint32(0); 
	cap=opts.init; % initial capacity of buffer
	buf=zeros(size_in(1),cap); % buffer

	api.remove=@remove;
	api.collect=@collect;
	api.length=@curlen;

	if isnan(size_in(2)),
		api.append=@append;
	elseif size_in(2)>1,
		m=uint32(size_in(2)); M=1:m;
		api.append=@append_m;
	else
		api.append=@append_1;
	end

	function l=curlen, l=n; end
	function remove(rem), n=n-rem;  end
	function x=collect, x=buf(:,1:n); buf=[]; end 
	function append_1(x)
		if n+1>cap, grow; end % need more room
		n=n+1; buf(:,n)=x;
	end

	function grow
		if n>opts.max, error('maximum capacity exceeded'); end
		cap=opts.grow*cap; 
		buf=repmat(buf,1,opts.grow); 
	end

	function append_var(x)
		chunk=size(x,2);
		if n+chunk>cap, grow; end % need more room
		buf(:,n+(1:chunk))=x;
		n=n+chunk;
	end

	function append_m(x)
		if n+m>cap, grow; end % need more room
		buf(:,n+M)=x;
		n=n+m;
	end
end