lster.h

/*
 * This file is part of Copiste.
 *
 * Copiste 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.
 *
 *  Copiste 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 Copiste.  If not, see <http://www.gnu.org/licenses/>.
 */


#ifndef INCLUDED_LSTERH
#define INCLUDED_LSTERH

class LSTERExtr;

#include "features/extractor.h"
#include "features/ste.h"

const float DEFAULT_STE_BOUND = 0.5;
const int DEFAULT_LSTER_CHUNK_SIZE = 40;

class LSTERExtr : public FeatureExtractor
{
    public:
        LSTERExtr(int chunkSize = 0);

        ~LSTERExtr();

        STEExtr* setSTEExtractor(STEExtr* extr);

        bool extract(std::deque<uint16_t> data, int size);

        float value(int index = 0) { return mLSTER; };

        int size() { return 1; }
        float min() { return 0; }
        float max() { return 1; }

        void setFloat(string key, float value);
        void setInt(string key, int value);
        float getFloat(string key);
        int getInt(string key);

        void clear();
    private:
        STEExtr *mSteExtr;
        float* mHistory;
        int mCurrentFrame;

        float mLSTER;
        float mBound;
        int mChunkSize;
};

#endif