classifiedhsimage.h

#ifndef CLASSIFIEDHSIMAGE_H
#define CLASSIFIEDHSIMAGE_H

#include <map>
#include <utility>

#include <opencv2/core.hpp>
#include <opencv2/imgcodecs.hpp>
#include <opencv2/highgui.hpp>
#include <opencv2/imgproc.hpp>

//#include <boost/python.hpp>

#include <Python.h>

#include "pybind11_opencv_numpy/pybind11/pybind11.h"
#include "pybind11_opencv_numpy/pybind11/stl.h"
#include "pybind11_opencv_numpy/ndarray_converter.h"

#include "hsimage.h"
#include "colormap.h"
#include "labelfile.h"


struct Vec3bCompare
{
    bool operator() (const cv::Vec3b& lhs, const cv::Vec3b& rhs) const
    {
        return cv::norm(lhs, CV_L2) < cv::norm(rhs, CV_L2);
    }
};

typedef std::pair<std::string,cv::Vec3b> classColor;

class ClassifiedHSImage
{
public:
    ClassifiedHSImage();
    ClassifiedHSImage(HSImage hsimage, cv::Mat labels, std::vector<classColor> c_names);
    ClassifiedHSImage(std::string raw_file, std::string hdr_file, std::string label_file, std::string class_hdr_file);
    ClassifiedHSImage(std::string raw_file, std::string hdr_file, std::string lif_file);

    HSImage image;
    cv::Mat label;
    std::map<std::string, cv::Vec3b> class_names;
    std::map<cv::Vec3b,std::string, Vec3bCompare> class_keys;
    void load(HSImage hsimage, cv::Mat labels, std::vector<classColor> c_names);
    std::vector<std::vector<u_int16_t> > getClassSpectra(std::string class_label, unsigned int num_spectra = 100);

    std::vector<double> getAvgClassTF(std::string class_label);
    std::vector<std::vector<double> > getClassTF(std::string class_label, unsigned int num_spectra = 100);

    std::string getPixelClass(int row, int col);

    void setSpectraClass(int row, int col, std::string class_label);
    void setSpectraClass(std::vector<std::pair< int, int > > pixel_locs, std::string class_label);
    void setImageClass(cv::Mat class_labels, std::vector<classColor> class_list);
    cv::Mat getImageClass();
};

#endif // CLASSIFIEDHSIMAGE_H