Mercurial > hg > ltpda
diff m-toolbox/classes/@ssm/CPSD.m @ 0:f0afece42f48
Import.
author | Daniele Nicolodi <nicolodi@science.unitn.it> |
---|---|
date | Wed, 23 Nov 2011 19:22:13 +0100 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/m-toolbox/classes/@ssm/CPSD.m Wed Nov 23 19:22:13 2011 +0100 @@ -0,0 +1,290 @@ +% CPSD computes the output theoretical CPSD shape with given inputs. +%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% +% +% DESCRIPTION: CPSD computes the output theoretical CPSD shape with given inputs. +% Unlike PSD, it returns the total contribution and takes +% input vectors/square matrices of objects +% +% CALL: [mat_out] = CPSD(sys, pl) +% +% INPUTS: +% - sys, (array of) ssm object +% +% OUTPUTS: +% _ mat_out contains specified returned aos +% +% <a href="matlab:utils.helper.displayMethodInfo('ssm', 'CPSD')">Parameters Description</a> +% +% VERSION: $Id: CPSD.m,v 1.11 2011/04/27 17:21:27 adrien Exp $ +% +%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% + +function varargout = CPSD(varargin) + + %% starting initial checks + + % use the caller is method flag + callerIsMethod = utils.helper.callerIsMethod; + + % Check if this is a call for parameters + if utils.helper.isinfocall(varargin{:}) + varargout{1} = getInfo(varargin{3}); + return + end + + utils.helper.msg(utils.const.msg.MNAME, ['running ', mfilename]); + + % Collect input variable names + in_names = cell(size(varargin)); + for ii = 1:nargin,in_names{ii} = inputname(ii);end + + % Collect all SSMs and plists + [sys, ssm_invars, rest] = utils.helper.collect_objects(varargin(:), 'ssm', in_names); + [pl, invars2, rest] = utils.helper.collect_objects(rest(:), 'plist'); + if ~isempty(rest) + pl = combine(pl, plist(rest{:})); + end + pl = combine(pl, getDefaultPlist()); + + %%% Internal call: Only one object + don't look for a plist + internal = strcmp(varargin{end}, 'internal'); + + %% begin function body + + %% retrieve system infos + + if numel(sys)~=1 + error('noisespectrum needs exactly one ssm as an input') + end + if ~sys.isnumerical + error(['error because system ',sys.name,' is not numerical']); + end + if ~sys.isStable + error('input ssm is not stable!') + end + if sys.timestep==0 + timestep = 1; + else + timestep = sys.timestep; + end + if ~internal + inhist = sys.hist; + end + + %% modifying system's ordering + if find(pl, 'reorganize') + sys = reorganize(sys, pl, 'set', 'for cpsd', 'internal', 'internal'); + end + + %% collecting functions i/o data + aos_in = find(pl, 'aos'); + PZ_in = find(pl, 'PZmodels'); + cov_in = find(pl, 'covariance'); + cpsd_in = find(pl, 'CPSD'); + noise_in = blkdiag(cov_in, cpsd_in/(timestep*2)); + [U1,S1,V1] = svd(noise_in.'); %#ok<NASGU> + noise_mat = U1*sqrt(S1); + + %% getting system's i/o sizes + inputSizes = sys.inputsizes; + outputSizes = sys.outputsizes; %#ok<NASGU> + + Naos_in = inputSizes(1); + NPZmodels = inputSizes(3); + + %% retrieving frequency vector + if isempty(Naos_in)==0 + f1 = find(pl,'f1'); + f2 = find(pl,'f2'); + NFreqs = find(pl,'nf'); + if isempty(f1) || isempty(f2)|| isempty(NFreqs) + error('### Please specify frequency vector a start and stop frequency .'); + else + freqs = 10.^linspace(log10(f1), log10(f2), NFreqs); + end + else + freqs = aos_in(1).x; + end + + %% checking frequency vector + for i=2:numel(aos_in) + if ~isequal(freqs,aos_in(i).x) + error('there exist different frequency vectors'); + end + end + + %% reshape pzmodels and aos for input cross-spectra + if numel(PZ_in)==NPZmodels + PZfull = false; + PZdata = zeros(NPZmodels,NFreqs); + for i=1:NPZmodels + a = resp(PZ_in(i), freqs); + PZdata(i,:) = reshape(a.y,[1,NFreqs]) ; + end + elseif size(PZ_in,1)==NPZmodels && size(PZ_in,2)==NPZmodels + PZfull = true; + PZdata = zeros(Npzmodels,Npzmodels,NFreqs); + for i=1:NPZmodels + for j=1:Npzmodels + a = resp(PZ_in(i,j), freqs); + PZdata(i,j,:) = reshape(a.y,[1,NFreqs]) ; + end + end + else + error('Wrong size for field PZ_in') + end + + if numel(aos_in)==Naos_in + AOfull = false; + AOdata = zeros(Naos_in,NFreqs); + for i=1:Naos_in + AOdata(i,:) = reshape(aos_in(i).y,[1,NFreqs]) ; + end + elseif size(aos_in,1)==Naos_in && size(aos_in,2)==Naos_in + AOfull = true; + AOdata = zeros(Naos_in,Naos_in,NFreqs); + for i=1:Naos_in + for j=1:Naos_in + AOdata(i,j,:) = reshape(aos_in(i,j).y,[1,NFreqs]) ; + end + end + else + error('Wrong size for field aos_in') + end + + %% SSM Transfer function + [a, b, c, d, Ts, InputName, StateName, OutputName,... + inputvarunits, ssvarunits, outputvarunits] = double(sys); %#ok<ASGLU> + resps = ssm.doBode(a, b, c, d, 2*pi*freqs, Ts); + Noutputs = numel(OutputName); + + %% power for each frequency with SVD computation + Result = zeros(Noutputs,Noutputs,NFreqs); + + for i_freq=1:NFreqs + %% contribution from aos, testing positiveness + if AOfull + PowAO = squeeze(AOdata(:,:,i_freq)); + [U1,S1,V1] = svd(PowAO.'); %#ok<NASGU> + if (sum(S1<0)>0) + error('AO covariance matrix is not positive') + end + AmpAO = U1*sqrt(S1); + else + if (sum(AOdata(:,i_freq)<0)>0) + error('non positive covariance') + end + AmpAO = diag(AOdata(:,i_freq).^0.5); + end + %% contribution from PZmodels, testing positiveness + if PZfull + PowPZ = squeeze(PZdata(:,:,i_freq)); + [U1,S1,V1] = svd(PowPZ.'); %#ok<NASGU> + if (sum(S1<0)>0) + error('PZmodels covariance matrix is not positive') + end + AmpPZ = U1*sqrt(S1); + else + if (sum(PZdata(:,i_freq)<0)>0) + error('non positive covariance') + end + AmpPZ = diag(PZdata(:,i_freq).^0.5); + end + %% summing all three contributions sources, computing CPSD + Amp = blkdiag(AmpAO, noise_mat, AmpPZ); + RespLoc = squeeze(resps(:,:,i_freq)); + noise = RespLoc * Amp * (RespLoc*Amp)'; + Result(:,:,i_freq) = real(noise) * (2*timestep); % 2 correction added here + end + + %% saving in aos + ao_out = ao.initObjectWithSize(Noutputs, Noutputs); + + for io=1:Noutputs + for jo=1:Noutputs + ao_out(io,jo).setData(fsdata(freqs, squeeze(Result(jo,io,:)))); + if io~=jo + ao_out(io,jo).setName( ['Cross PSD of ', OutputName{jo}, ' and ', OutputName{io}]); + else + ao_out(io,jo).setName( ['PSD of ' , OutputName{jo}]); + end + ao_out(io,jo).setXunits('Hz'); + ao_out(io,jo).setYunits(outputvarunits(io)*outputvarunits(jo)/unit('Hz')); + if io~=jo + ao_out(io,jo).setDescription( ['Cross PSD of ', OutputName{jo}, ' and ', OutputName{io}]); + else + ao_out(io,jo).setDescription( ['PSD of ' , OutputName{jo}]); + end + end + end + + %% construct output matrix object + out = matrix(ao_out); + if callerIsMethod + % do nothing + else + myinfo = getInfo('None'); + out.addHistory(myinfo, pl , ssm_invars(1), inhist ); + end + + %% Set output depending on nargout + if nargout == 1; + varargout = {out}; + elseif nargout == 0; + iplot(ao_out); + else + error('Wrong number of outputs') + end +end + + +%-------------------------------------------------------------------------- +% Get Info Object +%-------------------------------------------------------------------------- +function ii = getInfo(varargin) + + if nargin == 1 && strcmpi(varargin{1}, 'None') + sets = {}; + pl = []; + else + sets = {'Default'}; + pl = getDefaultPlist; + end + % Build info object + ii = minfo(mfilename, 'ssm', 'ltpda', utils.const.categories.op, '$Id: CPSD.m,v 1.11 2011/04/27 17:21:27 adrien Exp $', sets, pl); + +end + +%-------------------------------------------------------------------------- +% Get Default Plist +%-------------------------------------------------------------------------- +function pl = getDefaultPlist() + pl = ssm.getInfo('reorganize', 'for CPSD').plists; + pl.remove('set'); + + p = param({'covariance', 'The covariance of this noise between input ports for the <i>time-discrete</i> noise model.'}, []); + pl.append(p); + + p = param({'CPSD', 'The one sided cross-psd of the white noise between input ports.'}, []); + pl.append(p); + + p = param({'aos', 'An array of input AOs, The power spectrum of this noise between input ports for the <i>time-continuous</i> noise model.'}, ao.initObjectWithSize(1,0)); + pl.append(p); + + p = param({'PZmodels', 'An array of input pzmodels for noise filtering'}, paramValue.DOUBLE_VALUE(zeros(0,1))); + pl.append(p); + + p = param({'reorganize', 'When set to 0, this means the ssm does not need be modified to match the requested i/o. Faster but dangerous!'}, paramValue.TRUE_FALSE); + pl.append(p); + + p = param({'f2', 'The maximum frequency. Default is Nyquist or 1Hz.'}, paramValue.EMPTY_DOUBLE); + pl.append(p); + + p = param({'f1', 'The minimum frequency. Default is f2*1e-5.'}, paramValue.EMPTY_DOUBLE); + pl.append(p); + + p = param({'nf', 'The number of frequency bins. Frequencies are scale logarithmically'}, paramValue.DOUBLE_VALUE(200)); + pl.append(p); + +end +