summaryrefslogtreecommitdiff
path: root/macros/findpeaks.sci
diff options
context:
space:
mode:
Diffstat (limited to 'macros/findpeaks.sci')
-rw-r--r--macros/findpeaks.sci380
1 files changed, 326 insertions, 54 deletions
diff --git a/macros/findpeaks.sci b/macros/findpeaks.sci
index 3e7ce47..a96dff7 100644
--- a/macros/findpeaks.sci
+++ b/macros/findpeaks.sci
@@ -1,58 +1,330 @@
-function [PKS, LOC, EXTRA] = findpeaks(DATA, varargin)
-//This function find peaks on DATA.
-//Calling Sequence
-//[PKS, LOC, EXTRA] = findpeaks(DATA)
-//[PKS, LOC, EXTRA] = findpeaks(..., PROPERTY, VALUE)
-//[PKS, LOC, EXTRA] = findpeaks(..., "DoubleSided")
-//Description
-//Peaks of a positive array of data are defined as local maxima. For double-sided data, they are maxima of the positive part and minima of the negative part. DATA is expected to be a single column vector.
-//
-//The function returns the value of DATA at the peaks in PKS. The index indicating their position is returned in LOC.
-//
-//The third output argument is a structure with additional information:
-//
-//"parabol":
-// A structure containing the parabola fitted to each returned peak. The structure has two fields, "x" and "pp". The field "pp" contains the coefficients of the 2nd degree polynomial and "x" the extrema of the intercal here it was fitted.
-//
-//"height":
-// The estimated height of the returned peaks (in units of DATA).
-//
-//"baseline":
-// The height at which the roots of the returned peaks were calculated (in units of DATA).
-//
-//"roots":
-// The abscissa values (in index units) at which the parabola fitted to each of the returned peaks crosses the "baseline" value. The width of the peak is calculated by 'diff(roots)'.
-//
-//This function accepts property-value pair given in the list below:
-//
-//"MinPeakHeight":
-// Minimum peak height (positive scalar). Only peaks that exceed this value will be returned. For data taking positive and negative values use the option "DoubleSided". Default value '2*std (abs (detrend (data,0)))'.
-//
-//"MinPeakDistance":
-// Minimum separation between (positive integer). Peaks separated by less than this distance are considered a single peak. This distance is also used to fit a second order polynomial to the peaks to estimate their width, therefore it acts as a smoothing parameter. Default value 4.
-//
-//"MinPeakWidth":
-// Minimum width of peaks (positive integer). The width of the peaks is estimated using a parabola fitted to the neighborhood of each peak. The neighborhood size is equal to the value of "MinPeakDistance". The width is evaluated at the half height of the peak with baseline at "MinPeakHeight". Default value 2.
-//
-//"DoubleSided":
-// Tells the function that data takes positive and negative values. The base-line for the peaks is taken as the mean value of the function. This is equivalent as passing the absolute value of the data after removing the mean.
-funcprot(0);
-rhs=argn(2);
-lhs=argn(1)
-if(rhs<1 | rhs>2) then
- error("Wrong number of input arguments.");
-end
-if(lhs<3 | lhs>3) then
- error("Wrong number of output arguments.");
-end
+// Copyright (C) 2018 - IIT Bombay - FOSSEE
+// This file must be used under the terms of the CeCILL.
+// This source file is licensed as described in the file COPYING, which
+// you should have received as part of this distribution. The terms
+// are also available at
+// http://www.cecill.info/licences/Licence_CeCILL_V2-en.txt
+// Original Source : https://octave.sourceforge.io/signal/
+// Modifieded by: Abinash Singh Under FOSSEE Internship
+// Last Modified on : 3 Feb 2024
+// Organization: FOSSEE, IIT Bombay
+// Email: toolbox@scilab.in
+function [pks ,idx, varargout] = findpeaks (data, varargin)
+/*
+
+ [pks, loc, extra] = findpeaks (data)
+ … = findpeaks (…, property, value)
+ … = findpeaks (…, "DoubleSided")
+ Finds peaks on data.
+
+ Peaks of a positive array of data are defined as local maxima. For double-sided data, they are maxima of the positive part and minima of the negative part. data is expected to be a single column vector.
+
+ The function returns the value of data at the peaks in pks. The index indicating their position is returned in loc.
+
+ The third output argument is a structure with additional information:
+
+ "parabol"
+ A structure containing the parabola fitted to each returned peak. The structure has two fields, "x" and "pp". The field "pp" contains the coefficients of the 2nd degree polynomial and "x" the extrema of the interval where it was fitted.
+
+ "height"
+ The estimated height of the returned peaks (in units of data).
+
+ "baseline"
+ The height at which the roots of the returned peaks were calculated (in units of data).
+
+ "roots"
+ The abscissa values (in index units) at which the parabola fitted to each of the returned peaks realizes its width as defined below.
+
+ This function accepts property-value pair given in the list below:
+
+ "MinPeakHeight"
+ Minimum peak height (non-negative scalar). Only peaks that exceed this value will be returned. For data taking positive and negative values use the option "DoubleSided". Default value eps.
+
+ "MinPeakDistance"
+ Minimum separation between (positive integer). Peaks separated by less than this distance are considered a single peak. This distance is also used to fit a second order polynomial to the peaks to estimate their width, therefore it acts as a smoothing parameter. The neighborhood size is equal to the value of "MinPeakDistance". Default value 1.
+
+ "MinPeakWidth"
+ Minimum width of peaks (positive integer). The width of the peaks is estimated using a parabola fitted to the neighborhood of each peak. The width is calculated with the formula
+
+ a * (width - x0)^2 = 1
+ where a is the concavity of the parabola and x0 its vertex. Default value 1.
+
+ "MaxPeakWidth"
+ Maximum width of peaks (positive integer). Default value Inf.
+
+ "DoubleSided"
+ Tells the function that data takes positive and negative values. The base-line for the peaks is taken as the mean value of the function. This is equivalent as passing the absolute value of the data after removing the mean.
+*/
+
+ if (nargin < 1)
+ error("findpeaks:InsufficientInputArguments\nfindpeaks: DATA must be given");
+ end
+
+ if (~(isvector (data) && length (data) >= 3))
+ error ("findpeaks:InvalidArgument\nfindpeaks: DATA must be a vector of at least 3 elements");
+ end
+ transpose = (size(data,1) == 1);
+ if (transpose)
+ data = data.';
+ end
+ __data__ = abs (detrend (data, 'c'));
+ // --- Parse arguments --- //
+[ dSided, minH, minD, minW, maxW ] = parser ( varargin(:) );
+
+ if (dSided)
+ temp = __data__
+ __data__ = data
+ data = temp
+ temp = [] // free temp
+ elseif (min (data) < 0)
+ error ("findpeaks:InvalidArgument\nData contains negative values. You may want to DoubleSided option");
+ end
+ // Rough estimates of first and second derivative
+ df1 = diff (data, 1)
+ df1=df1([1; [1:length(df1)].']);
+ df2 = diff (data, 2)
+ df2=df2([1; 1; [1:length(df2)].']);
+
+ // check for changes of sign of 1st derivative and negativity of 2nd
+ // derivative.
+ // <= in 1st derivative includes the case of oversampled signals.
+ idx = find (df1.*[df1(2:$); 0] <= 0 & [df2(2:$); 0] < 0);
+
+ // Get peaks that are beyond given height
+ tf = data(idx) > minH;
+ idx = idx(tf);
+
+ // sort according to magnitude
+ [_, tmp] = gsort (data(idx));
+ idx_s = idx(tmp);
+
+ // Treat peaks separated less than minD as one
+ D = abs (bsminuseq (idx_s));
+ D = D + diag(ones(1,size(D,1))*%nan); // eliminate diagonal cpmparison
+
+ if (or(D(:) < minD)) // FIXME : this branch is not tested
+ i = 1;
+ peak = cell ();
+ node2visit = 1:size(D,1);
+ visited = [];
+ idx_pruned = idx_s;
+
+ // debug
+// h = plot(1:length(data),data,"-",idx_s,data(idx_s),'.r',idx_s,data(idx_s),'.g');
+// set(h(3),"visible","off");
+
+ while (~isempty (node2visit))
+
+ d = D(node2visit(1),:);
+
+ visited = [visited node2visit(1)];
+ node2visit(1) = [];
+
+ neighs = setdiff (find (d < minD), visited);
+ if ( ~isempty (neighs))
+ // debug
+// set(h(3),"xdata",idx_s(neighs),"ydata",data(idx_s(neighs)),"visible","on")
+// pause(0.2)
+// set(h(3),"visible","off");
+
+ idx_pruned = setdiff (idx_pruned, idx_s(neighs));
+
+ visited = [visited neighs];
+ node2visit = setdiff (node2visit, visited);
+
+ // debug
+// set(h(2),"xdata",idx_pruned,"ydata",data(idx_pruned))
+// pause
+ end
+
+ end
+ idx = idx_pruned;
+ end
+
+ extra = struct ("parabol", [], "height", [], "baseline", [], "roots", []);
+
+ // Estimate widths of peaks and filter for:
+ // width smaller than given.
+ // wrong concavity.
+ // not high enough
+ // data at peak is lower than parabola by 1%
+ // position of extrema minus center is bigger equal than minD/2
+ // debug
+// h = plot(1:length(data),data,"-",idx,data(idx),'.r',...
+// idx,data(idx),'og',idx,data(idx),'-m');
+// set(h(4),"linewidth",2)
+// set(h(3:4),"visible","off");
+
+ idx_pruned = idx;
+ n = length (idx);
+ np = length (data);
+ struct_count = 0;
+
+ for i=1:n
+ ind = (floor (max(idx(i)-minD/2,1)) : ...
+ ceil (min(idx(i)+minD/2,np))).';
+ pp = zeros (1,3);
+ // If current peak is not local maxima, then fit parabola to neighbor
+ if or(data(idx(i)-1) == data(idx(i)))
+ // sample on left same as peak
+ xm = 0;
+ pp = ones (1,3);
+ elseif or(data(ind) > data(idx(i)))
+ pp = polyfit (ind, data(ind), 2);
+ xm = -pp(2)^2 / (2*pp(1)); // position of extrema
+ H = polyval (pp, xm); // value at extrema
+ else // use it as vertex of parabola
+ H = data(idx(i));
+ xm = idx(i);
+ pp = zeros (1,3);
+ pp(1) = pinv((ind-xm).^2 ) * (data(ind)-H);
+ pp(2) = - 2 * pp(1) * xm;
+ pp(3) = H + pp(1) * xm^2;
+ end
+
+ // debug
+// x = linspace(ind(1)-1,ind(end)+1,10);
+// set(h(4),"xdata",x,"ydata",polyval(pp,x),"visible","on")
+// set(h(3),"xdata",ind,"ydata",data(ind),"visible","on")
+// pause(0.2)
+// set(h(3:4),"visible","off");
+
+// thrsh = min (data(ind([1 end])));
+// rz = roots ([pp(1:2) pp(3)-thrsh]);
+// width = abs (diff (rz));
+ width = sqrt (abs(1 / pp(1)));
+
+ if ( (width > maxW || width < minW) || pp(1) > 0 || H < minH || data(idx(i)) < 0.99*H ||abs (idx(i) - xm) > minD/2) then
+ idx_pruned = setdiff (idx_pruned, idx(i));
+ elseif (nargout > 2)
+ struct_count=struct_count+1;
+ extra.parabol(struct_count).x = ind([1 $]);
+ extra.parabol(struct_count).pp = pp;
+
+ extra.roots(struct_count,1:2)= xm + [-width width]/2;
+ extra.height(struct_count) = H;
+ extra.baseline(struct_count) = mean ([H minH]);
+ end
+
+ // debug
+// set(h(2),"xdata",idx_pruned,"ydata",data(idx_pruned))
+// pause(0.2)
-select(rhs)
-case 1 then
- [PKS, LOC, EXTRA] = callOctave("findpeaks", DATA);
-case 2 then
- [PKS, LOC, EXTRA] = callOctave("findpeaks", DATA, varargin(1));
-case 3 then
- [PKS, LOC, EXTRA] = callOctave("findpeaks", DATA, varargin(1), varargin(2));
end
+
+
+ idx = idx_pruned;
+ if (dSided)
+ pks = __data__(idx);
+ else
+ pks = data(idx);
+ end
+
+ if (transpose)
+ pks = pks.';
+ idx = idx.';
+ end
+ idx = idx.';
+ if (nargout() > 2)
+ varargout(1) = extra;
+ end
+
endfunction
+/*
+demo
+pi = %pi ;
+ t = 2*pi*linspace(0,1,1024)';
+ y = sin(3.14*t) + 0.5*cos(6.09*t) + 0.1*sin(10.11*t+1/6) + 0.1*sin(15.3*t+1/3);
+ data1 = abs(y);
+ [pks idx] = findpeaks(data1);
+ data2 = y;
+ [pks2 idx2] = findpeaks(data2,"DoubleSided");
+ [pks3 idx3] = findpeaks(data2,"DoubleSided","MinPeakHeight",0.5);
+
+ subplot(1,2,1)
+ plot(t,data1,t(idx),data1(idx),'xm')
+ subplot(1,2,2)
+ plot(t,data2,t(idx2),data2(idx2),"xm",t(idx3),data2(idx3),"or")
+ legend("Location","NorthOutside","Orientation","horizontal");
+/////////////////////////////// octave version /////////////////////////////
+ t = 2*pi*linspace(0,1,1024)';
+ y = sin(3.14*t) + 0.5*cos(6.09*t) + 0.1*sin(10.11*t+1/6) + 0.1*sin(15.3*t+1/3);
+ data1 = abs(y);
+ [pks idx] = findpeaks(data1);
+ data2 = y;
+ [pks2 idx2] = findpeaks(data2,"DoubleSided");
+ [pks3 idx3] = findpeaks(data2,"DoubleSided","MinPeakHeight",0.5);
+ subplot(1,2,1)
+ plot(t,data1,t(idx),data1(idx),'xm')
+ axis tight
+ subplot(1,2,2)
+ plot(t,data2,t(idx2),data2(idx2),"xm",t(idx3),data2(idx3),"or")
+ axis tight
+ legend("Location","NorthOutside","Orientation","horizontal");
+////////////////////octvae version end ////////////////////////////////////////////
+assert_checkequal(size(pks),[11 1]);
+assert_checkequal(size(idx),[11 1]);
+assert_checkequal(size(pks2),[11 1]);
+assert_checkequal(size(idx2),[11 1]);
+assert_checkequal(size(pks3),[8 1]);
+assert_checkequal(size(idx3),[8 1]);
+ //----------------------------------------------------------------------------
+
+ Finding the peaks of smooth data is not a big deal!
+// Not as accurate as Octave
+demo
+ t = 2*pi*linspace(0,1,1024)';
+ y = sin(3.14*t) + 0.5*cos(6.09*t) + 0.1*sin(10.11*t+1/6) + 0.1*sin(15.3*t+1/3);
+ data = abs(y + 0.1*rand(length(y),1,'normal'));
+ [pks idx] = findpeaks(data,"MinPeakHeight",1);
+ dt = t(2)-t(1);
+ [pks2 idx2] = findpeaks(data,"MinPeakHeight",1,"MinPeakDistance",round(0.5/dt));
+ subplot(1,2,1)
+ plot(t,data,t(idx),data(idx),'or')
+ subplot(1,2,2)
+ plot(t,data,t(idx2),data(idx2),'or')
+
+ ///octave version///////////////////////////////////////////////////////
+ t = 2*pi*linspace(0,1,1024)';
+ y = sin(3.14*t) + 0.5*cos(6.09*t) + 0.1*sin(10.11*t+1/6) + 0.1*sin(15.3*t+1/3);
+ data = abs(y + 0.1*randn(length(y),1));
+ [pks idx] = findpeaks(data,"MinPeakHeight",1);
+ dt = t(2)-t(1);
+ [pks2 idx2] = findpeaks(data,"MinPeakHeight",1,"MinPeakDistance",round(0.5/dt));
+ subplot(1,2,1)
+ plot(t,data,t(idx),data(idx),'or')
+ subplot(1,2,2)
+ plot(t,data,t(idx2),data(idx2),'or')
+ ///////////////////////////////////DO not run enclosed on scilab//////////////////////////////////////////////////////
+ assert_checkequal(size(pks),[86 1]);
+ assert_checkequal(size(pks2),[7 1]);
+ assert_checkequal(size(idx),[86 1]);
+ assert_checkequal(size(idx2),[7 1]);
+
+ //----------------------------------------------------------------------------
+ // Noisy data may need tuning of the parameters. In the 2nd example,
+ // MinPeakDistance is used as a smoother of the peaks.
+
+assert_checkequal (findpeaks ([1, 1, 1]),[])
+assert_checkequal (findpeaks ([1; 1; 1]),[])
+
+
+test
+ // Test input vector is an oversampled sinusoid with clipped peaks
+ x = min (3, cos (2*pi*[0:8000] ./ 600) + 2.01);
+ assert_checkequal (~isempty (findpeaks (x)),%t)
+
+
+test
+ x = [1 10 2 2 1 9 1];
+ [pks, loc] = findpeaks(x);
+ assert_checkequal (loc, [2 6])
+ assert_checkequal (pks, [10 9])
+
+// Test input validation
+error findpeaks ()
+error findpeaks (1)
+error findpeaks ([1, 2])
+
+*/