SegmentationParameters.h

Go to the documentation of this file.

#ifndef SEGMENTATIONPARAMETERS_H
#define SEGMENTATIONPARAMETERS_H

#include <vector>
#include <unordered_map>
#include <unordered_set>
#include <algorithm>

namespace agroeye {
namespace operations {
        
enum class StatisticsTypes { 
    MEAN,   
    STDDEV  
};

using BandsList = std::vector<size_t>;  
using StatisticsList = std::vector<StatisticsTypes>;    
using BandsToStatsMap = std::unordered_map<size_t, StatisticsList>; 
using StatsToBandsMap = std::unordered_map<StatisticsTypes, BandsList>; 
class Parameters {
public:
    Parameters() {};
    
    Parameters(BandsList bands);
    Parameters(const Parameters& other) = default;
    Parameters& operator=(const Parameters& other) = default;
    Parameters(Parameters&& other) = default;
    Parameters& operator=(Parameters&& other) = default;
    virtual ~Parameters() {};
    
    void setSquareSide(size_t side);
    size_t getSquareSide() const;   
    void setThreshold(double value) { threshold = value; };
    double getThreshold() const { return threshold; }; //*< Obtains stored threshold value */
    
    BandsList getReadBands() const;         
    BandsList getSegmentationBands() const; 
    BandsList getStatisticsBands() const;   
    StatisticsList getStatsTypes() const;   
    StatisticsList getStatsToBand(size_t band) const;
    
    BandsList getBandsToStat(StatisticsTypes type) const;

    static std::string statisticEnumToName(const StatisticsTypes type);
    
private:
    // Bands participating in creation of statistics database
    BandsToStatsMap bandsToStats;
    
    // Statistic type with bands pariticpating
    StatsToBandsMap statsToBands;
        
    // Bands participating in segmentation
    BandsList segmentationBands;
    
    // Bands participating in read from raster
    BandsList readBands;
    
    StatisticsList statsTypes;
    // Square Side used in chessboard segmentation
    size_t squareSide {static_cast<size_t>(-1)};
    
    // Threshold in Quadtree segmentation
    double threshold {-1.0};   
};
    
    
} // namespace operations
} // namespace agroye


#endif