mtspecgramc.m 4.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102
  1. function [S,t,f,Serr]=mtspecgramc(data,movingwin,params)
  2. % Multi-taper time-frequency spectrum - continuous process
  3. %
  4. % Usage:
  5. % [S,t,f,Serr]=mtspecgramc(data,movingwin,params)
  6. % Input:
  7. % Note units have to be consistent. Thus, if movingwin is in seconds, Fs
  8. % has to be in Hz. see chronux.m for more information.
  9. % data (in form samples x channels/trials) -- required
  10. % movingwin (in the form [window winstep] i.e length of moving
  11. % window and step size)
  12. % Note that units here have
  13. % to be consistent with
  14. % units of Fs - required
  15. % params: structure with fields tapers, pad, Fs, fpass, err, trialave
  16. % - optional
  17. % tapers : precalculated tapers from dpss or in the one of the following
  18. % forms:
  19. % (1) A numeric vector [TW K] where TW is the
  20. % time-bandwidth product and K is the number of
  21. % tapers to be used (less than or equal to
  22. % 2TW-1).
  23. % (2) A numeric vector [W T p] where W is the
  24. % bandwidth, T is the duration of the data and p
  25. % is an integer such that 2TW-p tapers are used. In
  26. % this form there is no default i.e. to specify
  27. % the bandwidth, you have to specify T and p as
  28. % well. Note that the units of W and T have to be
  29. % consistent: if W is in Hz, T must be in seconds
  30. % and vice versa. Note that these units must also
  31. % be consistent with the units of params.Fs: W can
  32. % be in Hz if and only if params.Fs is in Hz.
  33. % The default is to use form 1 with TW=3 and K=5
  34. % Note that T has to be equal to movingwin(1).
  35. %
  36. % pad (padding factor for the FFT) - optional (can take values -1,0,1,2...).
  37. % -1 corresponds to no padding, 0 corresponds to padding
  38. % to the next highest power of 2 etc.
  39. % e.g. For N = 500, if PAD = -1, we do not pad; if PAD = 0, we pad the FFT
  40. % to 512 points, if pad=1, we pad to 1024 points etc.
  41. % Defaults to 0.
  42. % Fs (sampling frequency) - optional. Default 1.
  43. % fpass (frequency band to be used in the calculation in the form
  44. % [fmin fmax])- optional.
  45. % Default all frequencies between 0 and Fs/2
  46. % err (error calculation [1 p] - Theoretical error bars; [2 p] - Jackknife error bars
  47. % [0 p] or 0 - no error bars) - optional. Default 0.
  48. % trialave (average over trials/channels when 1, don't average when 0) - optional. Default 0
  49. % Output:
  50. % S (spectrum in form time x frequency x channels/trials if trialave=0;
  51. % in the form time x frequency if trialave=1)
  52. % t (times)
  53. % f (frequencies)
  54. % Serr (error bars) only for err(1)>=1
  55. if nargin < 2; error('Need data and window parameters'); end;
  56. if nargin < 3; params=[]; end;
  57. [tapers,pad,Fs,fpass,err,trialave,params]=getparams(params);
  58. if length(params.tapers)==3 & movingwin(1)~=params.tapers(2);
  59. error('Duration of data in params.tapers is inconsistent with movingwin(1), modify params.tapers(2) to proceed')
  60. end
  61. if nargout > 3 && err(1)==0;
  62. % Cannot compute error bars with err(1)=0. change params and run again.
  63. error('When Serr is desired, err(1) has to be non-zero.');
  64. end;
  65. data=change_row_to_column(data);
  66. [N,Ch]=size(data);
  67. Nwin=round(Fs*movingwin(1)); % number of samples in window
  68. Nstep=round(movingwin(2)*Fs); % number of samples to step through
  69. nfft=max(2^(nextpow2(Nwin)+pad),Nwin);
  70. f=getfgrid(Fs,nfft,fpass); Nf=length(f);
  71. params.tapers=dpsschk(tapers,Nwin,Fs); % check tapers
  72. winstart=1:Nstep:N-Nwin+1;
  73. nw=length(winstart);
  74. if trialave
  75. S = zeros(nw,Nf);
  76. if nargout==4; Serr=zeros(2,nw,Nf); end;
  77. else
  78. S = zeros(nw,Nf,Ch);
  79. if nargout==4; Serr=zeros(2,nw,Nf,Ch); end;
  80. end
  81. for n=1:nw;
  82. indx=winstart(n):winstart(n)+Nwin-1;
  83. datawin=data(indx,:);
  84. if nargout==4
  85. [s,f,serr]=mtspectrumc(datawin,params);
  86. Serr(1,n,:,:)=squeeze(serr(1,:,:));
  87. Serr(2,n,:,:)=squeeze(serr(2,:,:));
  88. else
  89. [s,f]=mtspectrumc(datawin,params);
  90. end
  91. S(n,:,:)=s;
  92. end;
  93. S=squeeze(S);
  94. if nargout==4;Serr=squeeze(Serr);end;
  95. winmid=winstart+round(Nwin/2);
  96. t=winmid/Fs;