MASoundAdvancedAnalyzer.cpp

/*
 *  This file is part of the AiBO+ project
 *
 *  Copyright (C) 2005-2016 Csaba Kertész (csaba.kertesz@gmail.com)
 *
 *  AiBO+ 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 2 of the License, or
 *  (at your option) any later version.
 *
 *  AiBO+ 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, write to the Free Software
 *  Foundation, Inc., 59 Temple Street #330, Boston, MA 02111-1307, USA.
 *
 */

#include "MASoundAdvancedAnalyzer.hpp"

#include "core/MAContainerStatistics.hpp"
#include "MASoundAnalyzerPrivate.hpp"

#include <MCSampleStatistics.hpp>

MASoundAdvancedAnalyzer::MASoundAdvancedAnalyzer(unsigned int frequency, unsigned int sliding_window_limit) :
  MASoundBasicAnalyzer(frequency)
{
  SlidingWindowLimit = sliding_window_limit;
}


MASoundAdvancedAnalyzer::~MASoundAdvancedAnalyzer()
{
}


MC::FloatList MASoundAdvancedAnalyzer::CalculateHorizontalFeatures()
{
  MC::FloatList NewVector;

  // The horizontal features have not been committed yet.
  if (unlikely(FeatureNamesReady && FeatureNames.size() < FeatureVectors[0].size()))
    FeatureNamesReady = false;

  for (unsigned int i = 0; i < SlidingWindow[0].size(); ++i)
  {
    MC::FloatList Vector = MCGetTableColumn<MC::FloatList>(SlidingWindow, i);

//    MCMergeContainers(NewVector, MAChebyshevPolyCoeffFromContainer(Vector, 5, -1, 1));
//    for (int i1 = 0; i1 < 5; ++i1)
//      MA_ANALYZER_ADD_FEATURE_NAME_STR(FeatureNames[i]+"-Chebyshev"+MCToStr(i1))
//    if (Vector.size() == 8)
//      MCMergeContainers(NewVector, MAFFTComponentsFromContainer(Vector)[0]);
//    else
//      MCMergeContainers(NewVector, MC::FloatList(5, 0));
//    for (int i1 = 0; i1 < 5; ++i1)
//      MA_ANALYZER_ADD_FEATURE_NAME_STR(FeatureNames[i]+"-Fft"+MCToStr(i1))
    float Minimum = MCCalculateColumnStatInTable(SlidingWindow, i, *new MCMinimum<float>);
    float Maximum = MCCalculateColumnStatInTable(SlidingWindow, i, *new MCMaximum<float>);
    float Mean = MCCalculateColumnStatInTable(SlidingWindow, i, *new MCArithmeticMean<float>);

    NewVector.push_back(Minimum);
    MA_ANALYZER_ADD_FEATURE_NAME_STR(FeatureNames[i]+"-Min")
    NewVector.push_back(Maximum);
    MA_ANALYZER_ADD_FEATURE_NAME_STR(FeatureNames[i]+"-Max")
    NewVector.push_back(Maximum-Minimum); // Min-max range
    MA_ANALYZER_ADD_FEATURE_NAME_STR(FeatureNames[i]+"-Range")
    NewVector.push_back(MCCalculateColumnStatInTable(SlidingWindow, i, *new MCIqr<float>));
    MA_ANALYZER_ADD_FEATURE_NAME_STR(FeatureNames[i]+"-Iqr")
    NewVector.push_back(Mean);
    MA_ANALYZER_ADD_FEATURE_NAME_STR(FeatureNames[i]+"-AritMean")
    NewVector.push_back(SlidingWindow.back()[i]-Mean);
    MA_ANALYZER_ADD_FEATURE_NAME_STR(FeatureNames[i]+"-AritMeanDiff")
    NewVector.push_back(MCCalculateColumnStatInTable(SlidingWindow, i, *new MCAbsoluteSum<float>));
    MA_ANALYZER_ADD_FEATURE_NAME_STR(FeatureNames[i]+"-AbsSum")
    NewVector.push_back(MCCalculateColumnStatInTable(SlidingWindow, i, *new MCLogarithmicSquareSum<float>));
    MA_ANALYZER_ADD_FEATURE_NAME_STR(FeatureNames[i]+"-LogSquareSum")
    NewVector.push_back(MCCalculateColumnStatInTable(SlidingWindow, i, *new MCPower<float>));
    MA_ANALYZER_ADD_FEATURE_NAME_STR(FeatureNames[i]+"-Power")
  }
  if (unlikely(!FeatureNamesReady))
  {
    FeatureNamesReady = true;
    if (FeatureNames.size() != FeatureVectors[0].size())
    {
      MC_ERROR("Feature vector size and feature name count are different (%d != %d)",
               (int)FeatureVectors[0].size(), (int)FeatureNames.size());
    }
  }
  return NewVector;
}