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