idamnjanovic@8: function solver = SMALL_solve(Problem, solver)
idamnjanovic@8: %%% SMALL sparse solver
idamnjanovic@24: %
idamnjanovic@24: %   Centre for Digital Music, Queen Mary, University of London.
idamnjanovic@24: %   This file copyright 2009 Ivan Damnjanovic.
idamnjanovic@24: %
idamnjanovic@24: %   This program is free software; you can redistribute it and/or
idamnjanovic@24: %   modify it under the terms of the GNU General Public License as
idamnjanovic@24: %   published by the Free Software Foundation; either version 2 of the
idamnjanovic@24: %   License, or (at your option) any later version.  See the file
idamnjanovic@24: %   COPYING included with this distribution for more information.
idamnjanovic@24: %   
idamnjanovic@8: %   Function gets as input SMALL structure that contains SPARCO problem to
idamnjanovic@8: %   be solved, name of the toolbox and solver, and parameters file for
idamnjanovic@8: %   particular solver.
idamnjanovic@8: %
idamnjanovic@8: %   Outputs are solution, reconstructed signal and time spent
idamnjanovic@8: %%
idamnjanovic@8: 
idamnjanovic@8: if isa(Problem.A,'float')
idamnjanovic@8:     A = Problem.A;
idamnjanovic@8:     SparseLab_A=Problem.A;
idamnjanovic@8:     m = size(Problem.A,1);      % m is the no. of rows.
idamnjanovic@8:     n = size(Problem.A,2);      % n is the no. of columns.
idamnjanovic@8: else
idamnjanovic@8:     A  = @(x) Problem.A(x,1);  % The operator
idamnjanovic@8:     AT = @(y) Problem.A(y,2);  % and its transpose.
idamnjanovic@8:     SparseLab_A =@(mode, m, n, x, I, dim) SL_A(Problem.A, mode, m, n, x, I, dim);
idamnjanovic@8:     m = Problem.sizeA(1);      % m is the no. of rows.
idamnjanovic@8:     n = Problem.sizeA(2);      % n is the no. of columns.
idamnjanovic@8:     
idamnjanovic@1: end
idamnjanovic@37: % if signal that needs to be represented is different then training set for
idamnjanovic@37: % dictionary learning it should be stored in Problem.b1 matix
idamnjanovic@37: if isfield(Problem, 'b1')
idamnjanovic@37:     b = Problem.b1;
idamnjanovic@37: else
idamnjanovic@37:     b = Problem.b;            % The right-hand-side vector.
idamnjanovic@37: end
idamnjanovic@8: %%
idamnjanovic@8: fprintf('\nStarting solver %s... \n', solver.name);
idamnjanovic@8: start=cputime;
idamnjanovic@37: tStart=tic;
idamnjanovic@8: if strcmpi(solver.toolbox,'sparselab')
idamnjanovic@8:     y = eval([solver.name,'(SparseLab_A, b, n,',solver.param,');']);
idamnjanovic@8: elseif strcmpi(solver.toolbox,'sparsify')
idamnjanovic@8:     y = eval([solver.name,'(b, A, n, ''P_trans'', AT,',solver.param,');']);
idamnjanovic@8: elseif (strcmpi(solver.toolbox,'spgl1')||strcmpi(solver.toolbox,'gpsr'))
idamnjanovic@8:     y = eval([solver.name,'(b, A,',solver.param,');']);
idamnjanovic@8: elseif (strcmpi(solver.toolbox,'SPAMS'))
idamnjanovic@8:     y = eval([solver.name,'(b, A, solver.param);']);
idamnjanovic@8: elseif (strcmpi(solver.toolbox,'SMALL'))
idamnjanovic@8:     if isa(Problem.A,'float')
idamnjanovic@8:         y = eval([solver.name,'(A, b, n,',solver.param,');']);
idamnjanovic@8:     else
idamnjanovic@8:         y = eval([solver.name,'(A, b, n,',solver.param,',AT);']);
idamnjanovic@8:     end
idamnjanovic@37: elseif (strcmpi(solver.toolbox, 'ompbox'))
idamnjanovic@37:     G=A'*A;
idamnjanovic@37:     epsilon=solver.param.epsilon;
idamnjanovic@37:     maxatoms=solver.param.maxatoms;
idamnjanovic@37:     y = eval([solver.name,'(A, b, G,epsilon,''maxatoms'',maxatoms,''checkdict'',''off'');']);
idamnjanovic@37: elseif (strcmpi(solver.toolbox, 'ompsbox'))
idamnjanovic@37:     basedict = Problem.basedict;
idamnjanovic@37:      if issparse(Problem.A)
idamnjanovic@37:         A = Problem.A;
idamnjanovic@37:       else
idamnjanovic@37:         A = sparse(Problem.A);
idamnjanovic@37:       end
idamnjanovic@37:     G = dicttsep(basedict,A,dictsep(basedict,A,speye(size(A,2))));
idamnjanovic@37:     epsilon=solver.param.epsilon;
idamnjanovic@37:     maxatoms=solver.param.maxatoms;
idamnjanovic@37:     y = eval([solver.name,'(basedict, A, b, G,epsilon,''maxatoms'',maxatoms,''checkdict'',''off'');']);
idamnjanovic@37:     Problem.sparse=1;
idamnjanovic@8: %   To introduce new sparse representation algorithm put the files in
idamnjanovic@8: %   your Matlab path. Next, unique name <TolboxID> for your toolbox and
idamnjanovic@8: %   prefferd API <Preffered_API> needs to be defined.
idamnjanovic@8: %
idamnjanovic@8: % elseif strcmpi(solver.toolbox,'<ToolboxID>')
idamnjanovic@8: %
idamnjanovic@8: %     % - Evaluate the function (solver.name - defined in the main) with
idamnjanovic@8: %     %   parameters given above
idamnjanovic@8: %
idamnjanovic@8: %     y = eval([solver.name,'(<Preffered_API>);']);
idamnjanovic@8: 
idamnjanovic@8: else
idamnjanovic@8:     printf('\nToolbox has not been registered. Please change SMALL_solver file.\n');
idamnjanovic@8:     return
idamnjanovic@8: end
idamnjanovic@8: 
idamnjanovic@8: %%
idamnjanovic@8: %   Sparse representation time
idamnjanovic@37: tElapsed=toc(tStart);
idamnjanovic@8: solver.time = cputime - start;
idamnjanovic@8: fprintf('Solver %s finished task in %2f seconds. \n', solver.name, solver.time);
idamnjanovic@37: fprintf('Solver %s finished task in %2f seconds. \n', solver.name, tElapsed);
idamnjanovic@37: solver.time=tElapsed;
idamnjanovic@8: % geting around out of memory problem when converting big matrix from
idamnjanovic@8: % sparse to full...
idamnjanovic@8: 
idamnjanovic@18: if isfield(Problem, 'sparse')&&(Problem.sparse==1)
idamnjanovic@8:     solver.solution = y;
idamnjanovic@8: else
idamnjanovic@8:     solver.solution = full(y);
idamnjanovic@8: end
idamnjanovic@37: if isfield(Problem,'reconstruct')
idamnjanovic@8: %   Reconstruct the signal from the solution
idamnjanovic@8: solver.reconstructed  = Problem.reconstruct(solver.solution);
idamnjanovic@8: end
idamnjanovic@37: end