#pragma once
#include "IPreproc.h"
#include <map>
#include "commondef.h"

class CSizedetect : public IPreproc
{
public:
    explicit CSizedetect(int papaertype);
    virtual ~CSizedetect(void);
    virtual int  preprocess(cv::Mat& mat,void* unused);
    void SetPapertype(int papertype);
private:
    void rotate(float& x, float& y, float angle);
    void minAeaRect(const float* x, const float* y, int size, float& width, float& height);
    void fastMeasureSize(const unsigned char* data, int src_width, int src_height, int channels, float& dst_width, float& dst_height, int threshold = 40, int indent = 20, int step = 5);
private:
    int m_papertype;
    int m_horThre;
    int m_verThre;
    std::map<PaperSize,HGSize> m_supportPaper={
        {PaperSize::G400_A3,HGSize{2338,3307}},
        {PaperSize::G400_A4,HGSize{1653,2338}},
        {PaperSize::G400_A4R,HGSize{2338,1653}},
        {PaperSize::G400_A5,HGSize{1165,1653}},
        {PaperSize::G400_A5R,HGSize{1653,1165}},
        {PaperSize::G400_A6,HGSize{826,1165}},
        {PaperSize::G400_A6R,HGSize{1165,826}},
        {PaperSize::G400_B4,HGSize{1969,2780}},
        {PaperSize::G400_B5,HGSize{1385,1968}},
        {PaperSize::G400_B5R,HGSize{1968,1385}},
        {PaperSize::G400_B6R,HGSize{1433,1007}},
        {PaperSize::G400_B6,HGSize{1007,1433}},
        {PaperSize::G400_DOUBLELETTER,HGSize{2200,3400}},
        {PaperSize::G400_LEGAL,HGSize{1700,2800}},
        {PaperSize::G400_LETTER,HGSize{1700,2198}},
        {PaperSize::G400_LETTERR,HGSize{2198,1700}},
        {PaperSize::G400_LONGLETTER,HGSize{2040,2640}},
        {PaperSize::G400_MAXSIZE,HGSize{2338,8189}}
    };
};