function sD = som_modify_dataset(sD,action,varargin)
%SOM_MODIFY_DATASET Add or remove components or samples to/from a data struct.
%
% sD = som_modify_dataset(sD, 'addcomp', D, [indsto], [cnames})
% sD = som_modify_dataset(sD, 'addsamp', D, [indsto], ['norm'])
% sD = som_modify_dataset(sD, 'removecomp', [inds])
% sD = som_modify_dataset(sD, 'removesamp', inds)
% sD = som_modify_dataset(sD, 'extractcomp', [inds])
% sD = som_modify_dataset(sD, 'extractsamp', inds)
% sD = som_modify_dataset(sD, 'movecomp', inds, indsto)
% sD = som_modify_dataset(sD, 'movesamp', inds, indsto)
%
% Input and output arguments ([]'s are optional)
% sD (struct) data struct
% action (string) 'addcomp', 'addsamp', 'removecomp', 'removesamp',
% 'extractcomp', 'extractsamp', 'movecomp', or 'movesamp'
%
% other input arguments depend on the action
%
% 'addcomp':
% D (matrix) data matrix, size [dlen x d]
% (struct) data struct, size of .data field [dlen x d]
% [indsto] (vector) new indeces of the components, length=d
% [cnames] (cellstr) of size d x 1, the component names
%
% 'addsamp':
% D (matrix) data matrix, size [n x dim]
% [indsto] (vector) new indeces of the samples, length=n
% ['norm'] (string) specified if the normalization procedure
% should be applied to the new samples
%
% 'removecomp', 'extractcomp':
% [inds] (vector) indeces of the components to be removed/extracted.
% If not given, a prompt will appear from which the
% user can select the appropriate components.
%
% 'removesamp', 'extractsamp':
% inds (vector) indeces of the samples to be removed/extracted
%
% 'movecomp', 'movesamp':
% inds (vector) indeces of the components/samples to be moved
% indsto (vector) new indeces of the components/samples
%
% See also SOM_DATA_STRUCT.
% Copyright (c) 2000 by Juha Vesanto
% Contributed to SOM Toolbox on June 16th, 2000 by Juha Vesanto
% http://www.cis.hut.fi/projects/somtoolbox/
% Version 2.0beta juuso 200400 160600 280800
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%% input arguments
[dlen dim] = size(sD.data);
switch action,
case 'addcomp',
D = varargin{1};
if isstruct(D), [n d] = size(D.data); else [n d] = size(D); end
if n ~= dlen, error('The number of samples in the data struct and new data should match.'); end
indsto = [];
cnames = [];
for i=2:length(varargin),
if isnumeric(varargin{i}),
indsto = varargin{i};
if length(indsto) ~= d,
error('The number of indeces should match the number of new components');
end
else
if ischar(varargin{i}), cnames = cellstr(varargin{i});
elseif iscellstr(varargin{i}), cnames = varargin{i};
else
error(['[som_modify_dataset] Unrecognized argument #' num2str(i+1)]);
end
if length(cnames) ~= d,
error('The number of component names should match the number of new components');
end
end
end
case 'addsamp',
D = varargin{1};
if isstruct(D),
lab = D.labels;
if isfield(D,'data'), D = D.data; else D = D.codebook; end
else lab = [];
end
[n d] = size(D);
if d ~= dim,
error(['The dimensions of the old and new data sets should match.']);
end
norm = 0;
indsto = [];
for i=2:length(varargin),
if ischar(varargin{i}) & strcmp(varargin{i},'norm'), norm = 1;
elseif isnumeric(varargin{i}),
indsto = varargin{i};
if length(indsto) ~= n,
error(['The number of new indeces should match the number of new' ...
' samples']);
end
else
warning(['[som_modify_dataset] Ignoring unrecognized argument #', ...
num2str(i+2)]);
end
end
case 'removecomp',
if length(varargin)>0,
inds = varargin{1};
else
[inds, ok] = listdlg('ListString',sD.comp_names, 'PromptString', ...
'Components', 'Name', 'Remove components', 'uh', 25);
if ~ok, return; end
end
if min(inds)<1 | max(inds)>dim,
error('The component indeces must be within [1,dim]');
end
case 'removesamp',
inds = varargin{1};
if min(inds)<1 | max(inds)>dlen,
error('The sample indeces must be within [1,dlen]');
end
case 'extractcomp',
if length(varargin)>0,
inds = varargin{1};
else
[inds, ok] = listdlg('ListString',sD.comp_names, 'PromptString',...
'Components', 'Name', 'Extract components', 'uh', 25);
if ~ok, return; end
end
if min(inds)<1 | max(inds)>dim,
error('The component indeces must be within [1,dim]');
end
case 'extractsamp',
inds = varargin{1};
if min(inds)<1 | max(inds)>dlen,
error('The sample indeces must be within [1,dlen]');
end
case 'movecomp',
inds = varargin{1};
indsto = varargin{2};
if min(inds)<1 | max(inds)>dim | min(indsto)<1 | max(indsto)>dim,
error('The component indeces must be within [1,dim]');
end
case 'movesamp',
inds = varargin{1};
indsto = varargin{2};
if min(inds)<1 | max(inds)>dlen | min(indsto)<1 | max(indsto)>dlen,
error('The sample indeces must be within [1,dlen]');
end
otherwise,
error('Unrecognized action mode');
end
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%% action
switch action,
case 'addcomp',
if isstruct(D),
sD.data = [sD.data, D.data];
sD.comp_names(dim+[1:d]) = D.comp_names;
sD.comp_norm(dim+[1:d]) = D.comp_norm;
sD = som_label(sD,'add',1:dlen,D.labels);
else
sD.data = [sD.data, D];
if isempty(cnames),
for i=1:d, sD.comp_names(dim+i) = {sprintf('Variable%d',i+dim)}; end
else
sD.comp_names(dim+[1:d]) = cnames;
end
for i=1:d, sD.comp_norm(dim+i) = {[]}; end
end
if ~isempty(indsto),
sD = som_modify_dataset(sD,'movecomp',dim+[1:d],indsto);
end
case 'addsamp',
if norm, D = som_normalize(D,sD); end
sD.data = [sD.data; D];
nl = size(sD.labels,2);
sD.labels(dlen+[1:n],1:nl) = {''};
if ~isempty(lab),
nl2 = size(lab,2);
if nl2>nl, sD.labels(1:dlen,nl+[1:(nl2-nl)]) = {''}; end
sD.labels(dlen+[1:n],1:nl2) = lab;
end
if ~isempty(indsto),
sD = som_modify_dataset(sD,'movesamp',dlen+[1:n],indsto);
end
case 'removecomp',
includeinds = 1:dim;
includeinds(inds) = 0;
sD = som_modify_dataset(sD,'extractcomp',find(includeinds));
case 'removesamp',
includeinds = 1:dlen;
includeinds(inds) = 0;
sD = som_modify_dataset(sD,'extractsamp',find(includeinds));
case 'extractcomp',
sD.data = sD.data(:,inds);
sD.comp_names = sD.comp_names(inds);
sD.comp_norm = sD.comp_norm(inds);
case 'extractsamp',
sD.data = sD.data(inds,:);
sD.labels = sD.labels(inds,:);
case 'movecomp',
[indsto,order] = sort(indsto);
inds = inds(order);
oldinds = 1:dim;
oldinds(inds) = 0;
newinds = oldinds(oldinds>0);
for i=1:length(indsto),
ifrom = inds(i); ito = indsto(i);
if ito==1, newinds = [ifrom, newinds];
else newinds = [newinds(1:ito-1), ifrom, newinds(ito:end)];
end
end
sD.data = sD.data(:,newinds);
sD.comp_names = sD.comp_names(:,newinds);
sD.comp_norm = sD.comp_norm(:,newinds);
case 'movesamp',
[indsto,order] = sort(indsto);
inds = inds(order);
oldinds = 1:dim;
oldinds(inds) = 0;
newinds = oldinds(oldinds>0);
for i=1:length(indsto),
ifrom = inds(i); ito = indsto(i);
if ito==1, newinds = [ifrom, newinds];
else newinds = [newinds(1:ito-1), ifrom, newinds(ito:end)];
end
end
sD.data = sD.data(newinds,:);
sD.labels = sD.labels(newinds,:);
end
%som_set(sD);