conndetect_features.h

Go to the documentation of this file.

#pragma once
#include "config.h"
#include <array>
#include <libraries/Fft/Fft.h>
#include <vector>
 
class Features {
 
  const Channel *_rawdata;
  Channel _window;
  Channel _absdelta;
  std::vector<float> _wbd;
  std::array<float, BLOCKSIZE / 2 + 1> _freq;
 
  Fft fft;
  std::array<float, BLOCKSIZE / 2 + 1> _Ssq;
  float _fc;
  us _idx_peak;
 
  float _Ssq_tot;
 
  Channel _signal_squared;
 
  float _ms = 0.0f;
 
public:
  Features(const float fs);
  void setChannel(const Channel &channel);
 
  float peak_dB();
 
  float rms_dB();
 
  float kurtosis();
 
  float peak_freq();
 
  float wamp(float wampl);
 
  float zc(float zc_l);
 
  float TM3();
  float TM5();
 
  float inverse_variance();
 
private:
  bool _ms_computed = false;
  bool _fft_computed = false;
  bool _delta_computed = false;
 
  void compute_ms();
  void compute_fft();
  void compute_delta();
};
 
#include <numeric>
#include <algorithm>
class SimpleFeatures {
  public:
  static float average(const Channel& i) { 
    return std::accumulate(i.begin(), i.end(), 0.0)/BLOCKSIZE;
  }
  static float hasNonZero(const Channel& i) {
    auto nonzero = [](us i) { return i>1e-6; };
    return (float) (std::any_of(i.begin(), i.end(), nonzero));
  }
  static float maximum(const Channel& i){
    return *std::max_element(i.begin(), i.end());
  }
 
};