CentroidDetection.h

#pragma once
#include <stdint.h>
#include <vector>
#include <memory>
 
class CentroidDetection
{
public:
        typedef float DATA_T;
        CentroidDetection() {};
        CentroidDetection(unsigned int numReadings, unsigned int maxNumCentroids, float sizeScale);
        CentroidDetection(const std::vector<unsigned int>& order, unsigned int maxNumCentroids, float sizeScale);
        int setup(unsigned int numReadings, unsigned int maxNumCentroids, float sizeScale);
        int setup(const std::vector<unsigned int>& order, unsigned int maxNumCentroids, float sizeScale);
        void process(const DATA_T* rawData);
        void setSizeScale(float sizeScale);
        void setMinimumTouchSize(DATA_T minSize);
        void setNoiseThreshold(DATA_T threshold);
        void setWrapAround(unsigned int n);
        void setMultiplierBits(unsigned int n);
        unsigned int getNumTouches() const;
        DATA_T touchLocation(unsigned int touch_num) const;
        DATA_T touchSize(unsigned int touch_num) const;
        DATA_T compoundTouchLocation() const;
        DATA_T compoundTouchSize() const;
private:
        typedef uint16_t WORD;
        class CalculateCentroids;
        std::vector<DATA_T> centroids;
        std::vector<DATA_T> sizes;
        std::vector<WORD> centroidBuffer;
        std::vector<WORD> sizeBuffer;
        unsigned int maxNumCentroids;
        std::vector<unsigned int> order;
        unsigned int num_sensors;
        std::vector<WORD> data;
        float sizeScale;
        float locationScale;
        std::shared_ptr<CalculateCentroids> cc;
        unsigned int num_touches;
        DATA_T noiseThreshold;
};