conndetect.h

Go to the documentation of this file.

 
#pragma once
#include "config.h"
#include "buffer.h"
#include <Bela.h>
#include <cmath>
 
class Features;
class Classifier;
class TimeDelay;
 
template <us ncols> class LogFile;
class Gui;
 
 
enum class FeatureType: int {
  HP_8k_PEAKdB_MICT,
  HP_8k_PEAKdB_MICW,
  LP_500_PEAKdB_MICT,
  LP_500_PEAKdB_MICW,
  HP_2k_PEAKdB_VPU,
  LP_500_PEAKdB_VPU,
 
  HP_2k_PEAKFREQ_MICT,
 
  KURTOSIS_MICT,
  INVERSE_FREQ_VARIANCE_MICT,
  WAMP_MICT,
  TM3_MICT,
  ZC_MICT,
 
  KURTOSIS_MICW,
  INVERSE_FREQ_VARIANCE_MICW,
  WAMP_MICW,
  TM3_MICW,
  ZC_MICW,
 
  KURTOSIS_VPU,
  WAMP_VPU,
  TM3_VPU,
  ZC_VPU,
 
  TIMEDELAY_MICW_MICT,
 
  AVG_IMU_ABS_ROT, // Average value of the IMU rotation speed vector length
  MAX_IMU_ABS_HP_ACC, // Maximum value of the high-pass filtered
  // accelerometer vector length
 
  PEDAL,
  CLASS,
  NumberOfFeatures
};
const char* const logFileHeader = "HP_8k_PEAKdB_MICT, HP_8k_PEAKdB_MICW, LP_500_PEAKdB_MICT, LP_500_PEAKdB_MICW, HP_2k_PEAKdB_VPU, LP_500_PEAKdB_VPU, HP_2k_PEAKFREQ_MICT, KURTOSIS_MICT, INVERSE_FREQ_VARIANCE_MICT, WAMP_MICT, TM3_MICT, ZC_MICT, KURTOSIS_MICW, INVERSE_FREQ_VARIANCE_MICW, WAMP_MICW, TM3_MICW, ZC_MICW, KURTOSIS_VPU, WAMP_VPU, TM3_VPU, ZC_VPU, TIMEDELAY_MICW_MICT, AVG_IMU_ABS_ROT, MAX_IMU_ABS_HP_ACC, PEDAL, CLASS";
 
const us NUMBER_FEATURES = static_cast<int>(FeatureType::NumberOfFeatures);
 
typedef std::array<float, (us) NUMBER_FEATURES> ComputedFeatures;
 
class ConnDetect {
 
    LogFile<NUMBER_FEATURES> *_logfile = nullptr;
    FilteredDataBuffer *_bufs[2];
    bool _connDetected = false;
    AuxiliaryTask _detect_task;
    Features* _features;
    Classifier* _classifier;
    TimeDelay* _timedelay;
 
    friend void raw_conndetect_auxtask_callback(void *);
 
  public:
    bool connectionDetected();
 
    ConnDetect(BelaContext *context);
    ~ConnDetect();
    void addFilteredFrame(const FilteredFrame &filteredframe);
 
  private:
    void auxTaskCallback();
    void processFullBuffer(const FilteredData& data);
 
};