#pragma once
#include "IDetection.h"

class NumberDectect :IDetection
{

public:
    IDetection::DectectResult GetStateResult(cv::Mat img, cv::Rect rec);
    IDetection::DectectResult GetDigitResult(cv::Mat img, cv::Rect rec);
    vector<IDetection::DectectResult> GetDigitResults();

    bool Detect(cv::Mat& SrcImg);
    bool Init(bool isCuda);
    cv::Mat DoPerspectiveTransform(cv::Mat& SrcImg);
public:
    vector<IDetection::DectectResult> resultValues;
private:
    IDetection::DectectResult resultValue;
    
    vector<Output> output;
    cv::dnn::Net net;

    const std::vector<std::string> className = {
         "0",
        "1",
        "2",
        "3",
        "4",
        "5",
        "6",
        "7",
        "8",
        "9",
      "point",
        "A",
        "V",
        "degree_celsius",
        "kV",
        "kA",
        "kW",
        "p",
    };
    const std::vector<std::string> classTypeName = {
       "A",
       "V"
       "degree_celsius",
       "kV",
       "kA",
       "kW",
       "p",
    };

};