idamnjanovic@8
|
1 function solver = SMALL_solve(Problem, solver)
|
idamnjanovic@8
|
2 %%% SMALL sparse solver
|
idamnjanovic@24
|
3 %
|
ivan@85
|
4 % Function gets as input SMALL structure that contains SPARCO problem to
|
ivan@85
|
5 % be solved, name of the toolbox and solver, and parameters file for
|
ivan@85
|
6 % particular solver.
|
ivan@85
|
7 %
|
ivan@85
|
8 % Outputs are solution, reconstructed signal and time spent
|
ivan@85
|
9
|
idamnjanovic@24
|
10 % Centre for Digital Music, Queen Mary, University of London.
|
idamnjanovic@24
|
11 % This file copyright 2009 Ivan Damnjanovic.
|
idamnjanovic@24
|
12 %
|
idamnjanovic@24
|
13 % This program is free software; you can redistribute it and/or
|
idamnjanovic@24
|
14 % modify it under the terms of the GNU General Public License as
|
idamnjanovic@24
|
15 % published by the Free Software Foundation; either version 2 of the
|
idamnjanovic@24
|
16 % License, or (at your option) any later version. See the file
|
idamnjanovic@24
|
17 % COPYING included with this distribution for more information.
|
idamnjanovic@24
|
18 %
|
idamnjanovic@8
|
19 %%
|
idamnjanovic@8
|
20
|
idamnjanovic@8
|
21 if isa(Problem.A,'float')
|
idamnjanovic@8
|
22 A = Problem.A;
|
idamnjanovic@8
|
23 SparseLab_A=Problem.A;
|
idamnjanovic@8
|
24 m = size(Problem.A,1); % m is the no. of rows.
|
idamnjanovic@8
|
25 n = size(Problem.A,2); % n is the no. of columns.
|
idamnjanovic@8
|
26 else
|
idamnjanovic@8
|
27 A = @(x) Problem.A(x,1); % The operator
|
idamnjanovic@8
|
28 AT = @(y) Problem.A(y,2); % and its transpose.
|
idamnjanovic@8
|
29 SparseLab_A =@(mode, m, n, x, I, dim) SL_A(Problem.A, mode, m, n, x, I, dim);
|
idamnjanovic@8
|
30 m = Problem.sizeA(1); % m is the no. of rows.
|
idamnjanovic@8
|
31 n = Problem.sizeA(2); % n is the no. of columns.
|
idamnjanovic@8
|
32
|
idamnjanovic@1
|
33 end
|
idamnjanovic@37
|
34 % if signal that needs to be represented is different then training set for
|
idamnjanovic@37
|
35 % dictionary learning it should be stored in Problem.b1 matix
|
idamnjanovic@37
|
36 if isfield(Problem, 'b1')
|
idamnjanovic@37
|
37 b = Problem.b1;
|
idamnjanovic@37
|
38 else
|
idamnjanovic@37
|
39 b = Problem.b; % The right-hand-side vector.
|
idamnjanovic@37
|
40 end
|
idamnjanovic@8
|
41 %%
|
idamnjanovic@8
|
42 fprintf('\nStarting solver %s... \n', solver.name);
|
idamnjanovic@8
|
43 start=cputime;
|
idamnjanovic@37
|
44 tStart=tic;
|
idamnjanovic@8
|
45 if strcmpi(solver.toolbox,'sparselab')
|
idamnjanovic@8
|
46 y = eval([solver.name,'(SparseLab_A, b, n,',solver.param,');']);
|
idamnjanovic@8
|
47 elseif strcmpi(solver.toolbox,'sparsify')
|
idamnjanovic@8
|
48 y = eval([solver.name,'(b, A, n, ''P_trans'', AT,',solver.param,');']);
|
idamnjanovic@8
|
49 elseif (strcmpi(solver.toolbox,'spgl1')||strcmpi(solver.toolbox,'gpsr'))
|
idamnjanovic@8
|
50 y = eval([solver.name,'(b, A,',solver.param,');']);
|
idamnjanovic@8
|
51 elseif (strcmpi(solver.toolbox,'SPAMS'))
|
idamnjanovic@8
|
52 y = eval([solver.name,'(b, A, solver.param);']);
|
idamnjanovic@8
|
53 elseif (strcmpi(solver.toolbox,'SMALL'))
|
idamnjanovic@8
|
54 if isa(Problem.A,'float')
|
idamnjanovic@8
|
55 y = eval([solver.name,'(A, b, n,',solver.param,');']);
|
idamnjanovic@8
|
56 else
|
idamnjanovic@8
|
57 y = eval([solver.name,'(A, b, n,',solver.param,',AT);']);
|
idamnjanovic@8
|
58 end
|
idamnjanovic@37
|
59 elseif (strcmpi(solver.toolbox, 'ompbox'))
|
idamnjanovic@37
|
60 G=A'*A;
|
idamnjanovic@37
|
61 epsilon=solver.param.epsilon;
|
idamnjanovic@37
|
62 maxatoms=solver.param.maxatoms;
|
idamnjanovic@37
|
63 y = eval([solver.name,'(A, b, G,epsilon,''maxatoms'',maxatoms,''checkdict'',''off'');']);
|
idamnjanovic@37
|
64 elseif (strcmpi(solver.toolbox, 'ompsbox'))
|
idamnjanovic@37
|
65 basedict = Problem.basedict;
|
idamnjanovic@37
|
66 if issparse(Problem.A)
|
idamnjanovic@37
|
67 A = Problem.A;
|
idamnjanovic@37
|
68 else
|
idamnjanovic@37
|
69 A = sparse(Problem.A);
|
idamnjanovic@37
|
70 end
|
idamnjanovic@37
|
71 G = dicttsep(basedict,A,dictsep(basedict,A,speye(size(A,2))));
|
idamnjanovic@37
|
72 epsilon=solver.param.epsilon;
|
idamnjanovic@37
|
73 maxatoms=solver.param.maxatoms;
|
idamnjanovic@37
|
74 y = eval([solver.name,'(basedict, A, b, G,epsilon,''maxatoms'',maxatoms,''checkdict'',''off'');']);
|
idamnjanovic@37
|
75 Problem.sparse=1;
|
idamnjanovic@8
|
76 % To introduce new sparse representation algorithm put the files in
|
idamnjanovic@8
|
77 % your Matlab path. Next, unique name <TolboxID> for your toolbox and
|
idamnjanovic@8
|
78 % prefferd API <Preffered_API> needs to be defined.
|
idamnjanovic@8
|
79 %
|
idamnjanovic@8
|
80 % elseif strcmpi(solver.toolbox,'<ToolboxID>')
|
idamnjanovic@8
|
81 %
|
idamnjanovic@8
|
82 % % - Evaluate the function (solver.name - defined in the main) with
|
idamnjanovic@8
|
83 % % parameters given above
|
idamnjanovic@8
|
84 %
|
idamnjanovic@8
|
85 % y = eval([solver.name,'(<Preffered_API>);']);
|
idamnjanovic@8
|
86
|
idamnjanovic@8
|
87 else
|
idamnjanovic@8
|
88 printf('\nToolbox has not been registered. Please change SMALL_solver file.\n');
|
idamnjanovic@8
|
89 return
|
idamnjanovic@8
|
90 end
|
idamnjanovic@8
|
91
|
idamnjanovic@8
|
92 %%
|
idamnjanovic@8
|
93 % Sparse representation time
|
idamnjanovic@37
|
94 tElapsed=toc(tStart);
|
idamnjanovic@8
|
95 solver.time = cputime - start;
|
idamnjanovic@8
|
96 fprintf('Solver %s finished task in %2f seconds. \n', solver.name, solver.time);
|
idamnjanovic@37
|
97 fprintf('Solver %s finished task in %2f seconds. \n', solver.name, tElapsed);
|
idamnjanovic@37
|
98 solver.time=tElapsed;
|
idamnjanovic@8
|
99 % geting around out of memory problem when converting big matrix from
|
idamnjanovic@8
|
100 % sparse to full...
|
idamnjanovic@8
|
101
|
idamnjanovic@18
|
102 if isfield(Problem, 'sparse')&&(Problem.sparse==1)
|
idamnjanovic@8
|
103 solver.solution = y;
|
idamnjanovic@8
|
104 else
|
idamnjanovic@8
|
105 solver.solution = full(y);
|
idamnjanovic@8
|
106 end
|
idamnjanovic@37
|
107 if isfield(Problem,'reconstruct')
|
idamnjanovic@8
|
108 % Reconstruct the signal from the solution
|
idamnjanovic@8
|
109 solver.reconstructed = Problem.reconstruct(solver.solution);
|
idamnjanovic@8
|
110 end
|
idamnjanovic@37
|
111 end
|