#ifndef COUTLINEDETECTOR_H
#define COUTLINEDETECTOR_H
//轮廓线识别
#include <opencv2\opencv.hpp>
#include <QImage>
#include <vector>

#include "..\..\SSBase\VoronoiMap\PolygonTreeInterface.h"
#include "ImageUtils.h"
#include "ContourUtils.h"

class COutlineDetector
{
public:
    COutlineDetector();
    void clear(void);

    void setSurface(GSurface* pSurf, float zlower = 0, float zupper = 0, unsigned int nIters = 1);
    //膨胀迭代次数
    void setIterations(unsigned int nIterations);
    //设置目标区域z值范围
    void setTargetZRange(float zlower, float zupper);
    //设置最小轮廓线面积
    void setContourMinArea(float area);
    //设置输出轮廓线平滑次数
    void setContourSmoothTimes(unsigned int times);

    //执行图像处理，生成初始轮廓线,并进行光滑
    bool ProcessImage(void);

	//生成沉积相轮廓线,局部合并内部包含孔洞的轮廓线
	int CreateFaciesContours(void);

    cv::Mat& getResultMat(); //返回结果图片

	//图片中轮廓线转换为曲面坐标 dstContour, smoothTimes转换后为平滑次数
	void multiContourToRealCoords(const Contours& inputContour, Contours& dstContour, int smoothTimes);

	//获取初始轮廓线
	Contours& GetOriginalContours(void);

	//获取沉积相轮廓线
	CPolygonTreeInterface* GetFaciesContours(void);
	//获取沉积相轮廓线
	shared_ptr<Contours> GetRealContours(void);
private:

    void morphologySolve(void);
    //由matMorp生成轮廓线
    void createContours(void);
    //面积过滤输出等值线，不符合条件的剔除
    void FilterContours(float minArea, Contours& ReservedContours);

    //五点三次平滑
    void SmoothContours();
  
    //图片中轮廓线转换为曲面坐标 dstContour, smoothTimes转换后为平滑次数
    void contourToRealCoords(const Contour& inputContour, Contour& dstContour, int smoothTimes);

    //生成边界轮廓线
    void TraceBorder(void);

    GSurface* m_pSurface;
    cv::Mat m_matSrc;  //原始图片（黑白图)
    cv::Mat m_matMorp; // 形态学处理后的图片
    cv::Mat m_matBorder; //用来追踪边界轮廓的黑白图

    int m_nIterations;
    float m_zlower;
    float m_zupper;
    double m_minArea;
    int m_smoothTimes;

	//图像处理后 生成的初始轮廓线
    Contours m_originalContours;
    std::vector<cv::Vec4i> m_hierarchy;
	CPolygonTreeInterface m_pgnTree;
    Contour m_border;
};

#endif // COUTLINEDETECTOR_H