AMR/src/rsi_calc.cpp

#include <Rcpp.h>
#include <functional>    // for std::less, etc
#include <algorithm>     // for count_if

using namespace Rcpp;

// [[Rcpp::export]]
int rsi_calc_S(DoubleVector x, bool include_I) {
  if (include_I == TRUE) {
    return count_if(x.begin(), x.end(), bind2nd(std::less_equal<double>(), 2));
  } else {
    return count_if(x.begin(), x.end(), bind2nd(std::less<double>(), 2));
  }
}

// [[Rcpp::export]]
int rsi_calc_R(DoubleVector x, bool include_I) {
  if (include_I == TRUE) {
    return count_if(x.begin(), x.end(), bind2nd(std::greater_equal<double>(), 2));
  } else {
    return count_if(x.begin(), x.end(), bind2nd(std::greater<double>(), 2));
  }
}

// [[Rcpp::export]]
int rsi_calc_total(DoubleVector x) {
  return count_if(x.begin(), x.end(), bind2nd(std::less_equal<double>(), 3));
}