barcode.hpp

// This file is part of OpenCV project.
// It is subject to the license terms in the LICENSE file found in the top-level directory
// of this distribution and at http://opencv.org/license.html.
// Copyright (c) 2020-2021 darkliang wangberlinT Certseeds
 
#ifndef __OPENCV_BARCODE_HPP__
#define __OPENCV_BARCODE_HPP__
 
#include <opencv2/core.hpp>
#include <ostream>
 
namespace cv {
namespace barcode {
 
 
enum BarcodeType
{
    NONE, EAN_8, EAN_13, UPC_A, UPC_E, UPC_EAN_EXTENSION
};
 
static inline std::ostream &operator<<(std::ostream &out, const BarcodeType &barcode_type)
{
    switch (barcode_type)
    {
        case BarcodeType::EAN_8:
            out << "EAN_8";
            break;
        case BarcodeType::EAN_13:
            out << "EAN_13";
            break;
        case BarcodeType::UPC_E:
            out << "UPC_E";
            break;
        case BarcodeType::UPC_A:
            out << "UPC_A";
            break;
        case BarcodeType::UPC_EAN_EXTENSION:
            out << "UPC_EAN_EXTENSION";
            break;
        default:
            out << "NONE";
    }
    return out;
}
 
class CV_EXPORTS_W BarcodeDetector
{
public:
    CV_WRAP BarcodeDetector(const std::string &prototxt_path = "", const std::string &model_path = "");
 
    ~BarcodeDetector();
 
    CV_WRAP bool detect(InputArray img, OutputArray points) const;
 
    CV_WRAP bool decode(InputArray img, InputArray points, CV_OUT std::vector<std::string> &decoded_info, CV_OUT
                        std::vector<BarcodeType> &decoded_type) const;
 
    CV_WRAP bool detectAndDecode(InputArray img, CV_OUT std::vector<std::string> &decoded_info, CV_OUT
                                 std::vector<BarcodeType> &decoded_type, OutputArray points = noArray()) const;
 
protected:
    struct Impl;
    Ptr<Impl> p;
};
}
} // cv::barcode::
#endif //__OPENCV_BARCODE_HPP__