diff m-toolbox/classes/@ao/bicohere.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/@ao/bicohere.m	Wed Nov 23 19:22:13 2011 +0100
@@ -0,0 +1,161 @@
+% BICOHERE computes the bicoherence of two input time-series
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+%
+% DESCRIPTION: BICOHERE computes the bicoherence of two input time-series.
+%
+% CALL:        bs = bicohere(a1,a2,pl)
+%
+% INPUTS:      aN    - input analysis objects
+%              a1,a2 - input analysis objects array
+%              pl    - input parameter list
+%
+% OUTPUTS:     bs   - xyz data analysis object
+%
+% <a href="matlab:utils.helper.displayMethodInfo('bicohere', 'psd')">Parameters Description</a>
+%
+% VERSION:    $Id: psd.m,v 1.59 2010/12/17 17:45:08 ingo Exp $
+%
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+
+
+function varargout = bicohere(varargin)
+  
+  % Check if this is a call for parameters
+  if utils.helper.isinfocall(varargin{:})
+    varargout{1} = getInfo(varargin{3});
+    return
+  end
+  
+  import utils.const.*
+  utils.helper.msg(msg.PROC3, 'running %s/%s', mfilename('class'), mfilename);
+  
+  % Collect input variable names
+  in_names = cell(size(varargin));
+  for ii = 1:nargin,in_names{ii} = inputname(ii);end
+  
+  % Collect all AOs and plists
+  [as, ao_invars] = utils.helper.collect_objects(varargin(:), 'ao', in_names);
+  pl              = utils.helper.collect_objects(varargin(:), 'plist', in_names);
+  
+  if numel(as) ~= 2
+    error('bicohere only works with 2 time-series at the moment.');
+  end
+  
+  % Get data
+  a  = as(1);
+  b  = as(2);
+  
+  % same fs?
+  if a.data.fs ~= b.data.fs
+    error('### Two time-series have different sample rates.');
+  end
+  
+  % Same length vectors?
+  if a.len ~= b.len
+    error('### Two time-series must have same length');
+  end
+  
+  % Combine plists
+  pl = combine(pl, getDefaultPlist); 
+  usepl = utils.helper.process_spectral_options(pl, 'lin', a.len, a.fs);
+  
+  win     = find(usepl, 'Win');
+  nfft    = find(usepl, 'Nfft');
+  olap    = find(usepl, 'Olap')/100;
+  Xolap   = round(olap*nfft);
+  order   = find(usepl, 'order');
+    
+  fs = a.data.fs;
+  x  = {a.data.y; b.data.y};
+    
+  [x,M,isreal_x,y,Ly,win,winName,winParam,noverlap,k,L,options] = ...
+    ao.welchparse(x,'',win.win, Xolap, nfft, fs);
+  
+  select = 1:(nfft+1)/2;
+
+  % loop over segments
+  LminusOverlap = L-noverlap;
+  xStart = 1:LminusOverlap:k*LminusOverlap;
+  xEnd   = xStart+L-1;
+  
+  m = zeros(length(select), length(select));
+
+  for i = 1:k
+    if order < 0
+      Xseg = x(xStart(i):xEnd(i));
+      Yseg = x(xStart(i):xEnd(i));
+    else
+      [Xseg,coeffs] = ltpda_polyreg(x(xStart(i):xEnd(i)), order);
+      [Yseg,coeffs] = ltpda_polyreg(y(xStart(i):xEnd(i)), order);
+    end
+     
+    % window
+    xw = Xseg.*win.';
+    yw = Yseg.*win.';
+
+    % FFT
+    xx2s = fft(xw);    
+    xx   = xx2s(select);
+    yy2s = fft(yw);
+    yy   = yy2s(select);
+    
+    scalex = abs(xx);
+    scaley = abs(yy);
+    sc = scalex * scaley';
+    m = m + (xx * yy')./sc;
+        
+  end
+    
+  m = m./k;
+  
+  f = psdfreqvec('npts',nfft,'Fs',fs);
+  f = f(select);
+  do = xyzdata(f, f, m);
+  do.setXunits('Hz');
+  do.setYunits('Hz');
+  do.setZunits(a.yunits*b.yunits);
+  
+  ma = ao(do);
+  ma.setName(sprintf('%s, %s', a.name, b.name));
+  ma.addHistory(getInfo('None'), usepl, ao_invars, [a.hist b.hist]);
+
+  
+  % Set output
+  varargout{1} = ma;
+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, 'ao', 'ltpda', utils.const.categories.sigproc, '$Id: psd.m,v 1.59 2010/12/17 17:45:08 ingo Exp $', sets, pl);
+end
+
+%--------------------------------------------------------------------------
+% Get Default Plist
+%--------------------------------------------------------------------------
+function plout = getDefaultPlist()
+  persistent pl;  
+  if exist('pl', 'var')==0 || isempty(pl)
+    pl = buildplist();
+  end
+  plout = pl;  
+end
+
+function pl = buildplist()
+  
+  % General plist for Welch-based, linearly spaced spectral estimators
+  pl = plist.WELCH_PLIST;
+  
+  
+end
+% END
+