# movingrms.R
# Copyright (C) 2020 Geert van Boxtel <gjmvanboxtel@gmail.com>
# Original Matlab/Octave version:
# Copyright (C) 2012 Juan Pablo Carbajal <carbajal@ifi.uzh.ch>
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 3
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
# Version history
# 20200322 GvB setup for gsignal v0.1.0
# 20210517 GvB adapted examples
#------------------------------------------------------------------------------
#' Moving Root Mean Square
#'
#' Compute the moving root mean square (RMS) of the input signal.
#'
#' The signal is convoluted against a sigmoid window of width \code{w} and
#' risetime \code{rc}. The units of these parameters are relative to the value
#' of the sampling frequency given in \code{fs}.
#'
#' @param x Input signal, specified as a numeric vector or matrix. In case of a
#' matrix, the function operates along the columns
#' @param width width of the sigmoid window, in units relative to \code{fs}.
#' Default: 0.1
#' @param rc Rise time (time constant) of the sigmoid window, in units relative
#' to \code{fs}. Default: 1e-3
#' @param fs Sampling frequency. Default: 1
#'
#' @return A \code{\link{list}} containing 2 variables:
#' \describe{
#' \item{rmsx}{Output signal with the same dimensions as \code{x}}
#' \item{w}{Window, returned as a vector}
#' }
#'
#' @examples
#'
#' N <- 128
#' fs <- 5
#' t <- seq(0, 1, length.out = N)
#' x <- sin(2 * pi * fs * t) + runif(N)
#' y <- movingrms(x, 5)
#'
#' @seealso \code{\link{sigmoid_train}}
#'
#' @author Juan Pablo Carbajal, \email{carbajal@@ifi.uzh.ch}.\cr
#' Conversion to R by Geert van Boxtel, \email{G.J.M.vanBoxtel@@gmail.com}.
#'
#' @export
movingrms <- function(x, width = 0.1, rc = 1e-3, fs = 1) {
if (is.vector(x)) {
n <- length(x)
} else if (is.matrix(x)) {
n <- nrow(x)
} else {
stop("x must be a vector or a matrix")
}
if (!isPosscal(width)) {
stop("width must be a positive scalar")
}
if (!isPosscal(rc)) {
stop("rc must be a positive scalar")
}
if (!isPosscal(fs)) {
stop("fs must be a positive scalar")
}
if (width * fs > n / 2) {
idx <- c(1, n)
w <- rep(1L, n)
} else {
idx <- round((n + width * fs * c(-1, 1)) / 2)
w <- sigmoid_train((1:n), idx, rc * fs)$y
}
rmsx_single <- function(x, w, n, idx) {
fx <- stats::fft(as.vector(x)^2)
fw <- stats::fft(as.vector(w)^2)
out <- ifft(fx * fw) / (n - 1)
out[out < .Machine$double.eps * max(out)] <- 0
out <- pracma::circshift(sqrt(out), round(mean(idx)))
}
if (is.vector(x)) {
rmsx <- rmsx_single(x, w, n, idx)
} else {
rmsx <- apply(x, 2, rmsx_single, w, n, idx)
}
list(rmsx = rmsx, w = w)
}
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.