AlsonMat.h

浏览该文件的文档.

//
// Created by ALSON08 on 2022/8/30.
//
 
#ifndef ALSON_CLASSIC_BASE_ENTITY_ALSON_MAT_H
#define ALSON_CLASSIC_BASE_ENTITY_ALSON_MAT_H
 
 
// std
#include <cstdint>
#include <string>
#include <memory>
 
// alson
#include <alson/communication_base/export.h>
 
 
#define ALSON_MAT_CN_SHIFT   3
#define ALSON_MAT_DEPTH_MAX  (1 << ALSON_MAT_CN_SHIFT)
 
#define ALSON_MAT_8U   0
#define ALSON_MAT_8S   1
#define ALSON_MAT_16U  2
#define ALSON_MAT_16S  3
#define ALSON_MAT_32S  4
#define ALSON_MAT_32F  5
#define ALSON_MAT_64F  6
#define ALSON_MAT_16F  7
 
#define ALSON_MAT_MAT_DEPTH_MASK       (ALSON_MAT_DEPTH_MAX - 1)
#define ALSON_MAT_MAT_DEPTH(flags)     ((flags) & ALSON_MAT_MAT_DEPTH_MASK)
 
#define ALSON_MAT_MAKETYPE(depth, cn) (ALSON_MAT_MAT_DEPTH(depth) + (((cn)-1) << ALSON_MAT_CN_SHIFT))
#define ALSON_MAT_MAKE_TYPE ALSON_MAT_MAKETYPE
 
 
#define ALSON_MAT_8UC1 ALSON_MAT_MAKETYPE(ALSON_MAT_8U,1)
#define ALSON_MAT_8UC2 ALSON_MAT_MAKETYPE(ALSON_MAT_8U,2)
#define ALSON_MAT_8UC3 ALSON_MAT_MAKETYPE(ALSON_MAT_8U,3)
#define ALSON_MAT_8UC4 ALSON_MAT_MAKETYPE(ALSON_MAT_8U,4)
#define ALSON_MAT_8UC(n) ALSON_MAT_MAKETYPE(ALSON_MAT_8U,(n))
 
#define ALSON_MAT_8SC1 ALSON_MAT_MAKETYPE(ALSON_MAT_8S,1)
#define ALSON_MAT_8SC2 ALSON_MAT_MAKETYPE(ALSON_MAT_8S,2)
#define ALSON_MAT_8SC3 ALSON_MAT_MAKETYPE(ALSON_MAT_8S,3)
#define ALSON_MAT_8SC4 ALSON_MAT_MAKETYPE(ALSON_MAT_8S,4)
#define ALSON_MAT_8SC(n) ALSON_MAT_MAKETYPE(ALSON_MAT_8S,(n))
 
#define ALSON_MAT_16UC1 ALSON_MAT_MAKETYPE(ALSON_MAT_16U,1)
#define ALSON_MAT_16UC2 ALSON_MAT_MAKETYPE(ALSON_MAT_16U,2)
#define ALSON_MAT_16UC3 ALSON_MAT_MAKETYPE(ALSON_MAT_16U,3)
#define ALSON_MAT_16UC4 ALSON_MAT_MAKETYPE(ALSON_MAT_16U,4)
#define ALSON_MAT_16UC(n) ALSON_MAT_MAKETYPE(ALSON_MAT_16U,(n))
 
#define ALSON_MAT_16SC1 ALSON_MAT_MAKETYPE(ALSON_MAT_16S,1)
#define ALSON_MAT_16SC2 ALSON_MAT_MAKETYPE(ALSON_MAT_16S,2)
#define ALSON_MAT_16SC3 ALSON_MAT_MAKETYPE(ALSON_MAT_16S,3)
#define ALSON_MAT_16SC4 ALSON_MAT_MAKETYPE(ALSON_MAT_16S,4)
#define ALSON_MAT_16SC(n) ALSON_MAT_MAKETYPE(ALSON_MAT_16S,(n))
 
#define ALSON_MAT_32SC1 ALSON_MAT_MAKETYPE(ALSON_MAT_32S,1)
#define ALSON_MAT_32SC2 ALSON_MAT_MAKETYPE(ALSON_MAT_32S,2)
#define ALSON_MAT_32SC3 ALSON_MAT_MAKETYPE(ALSON_MAT_32S,3)
#define ALSON_MAT_32SC4 ALSON_MAT_MAKETYPE(ALSON_MAT_32S,4)
#define ALSON_MAT_32SC(n) ALSON_MAT_MAKETYPE(ALSON_MAT_32S,(n))
 
#define ALSON_MAT_32FC1 ALSON_MAT_MAKETYPE(ALSON_MAT_32F,1)
#define ALSON_MAT_32FC2 ALSON_MAT_MAKETYPE(ALSON_MAT_32F,2)
#define ALSON_MAT_32FC3 ALSON_MAT_MAKETYPE(ALSON_MAT_32F,3)
#define ALSON_MAT_32FC4 ALSON_MAT_MAKETYPE(ALSON_MAT_32F,4)
#define ALSON_MAT_32FC(n) ALSON_MAT_MAKETYPE(ALSON_MAT_32F,(n))
 
#define ALSON_MAT_64FC1 ALSON_MAT_MAKETYPE(ALSON_MAT_64F,1)
#define ALSON_MAT_64FC2 ALSON_MAT_MAKETYPE(ALSON_MAT_64F,2)
#define ALSON_MAT_64FC3 ALSON_MAT_MAKETYPE(ALSON_MAT_64F,3)
#define ALSON_MAT_64FC4 ALSON_MAT_MAKETYPE(ALSON_MAT_64F,4)
#define ALSON_MAT_64FC(n) ALSON_MAT_MAKETYPE(ALSON_MAT_64F,(n))
 
#define ALSON_MAT_16FC1 ALSON_MAT_MAKETYPE(ALSON_MAT_16F,1)
#define ALSON_MAT_16FC2 ALSON_MAT_MAKETYPE(ALSON_MAT_16F,2)
#define ALSON_MAT_16FC3 ALSON_MAT_MAKETYPE(ALSON_MAT_16F,3)
#define ALSON_MAT_16FC4 ALSON_MAT_MAKETYPE(ALSON_MAT_16F,4)
#define ALSON_MAT_16FC(n) ALSON_MAT_MAKETYPE(ALSON_MAT_16F,(n))
namespace ALSON {
 
class ALSON_API AlsonMat {
 
public:
    enum Depth {
        DEPTH_8U = 0,
        DEPTH_8S = 1,
        DEPTH_16U = 2,
        DEPTH_16S = 3,
        DEPTH_32S = 4,
        DEPTH_32F = 5,
        DEPTH_64F = 6,
        DEPTH_16F = 7
    };
 
public:
    AlsonMat();
    
    AlsonMat(int r, int c, int t);
    
    AlsonMat(int r, int c, int t, uint8_t* d);
    
    AlsonMat(const AlsonMat&);
    AlsonMat& operator=(const AlsonMat&);
    AlsonMat(AlsonMat&&) noexcept;
    AlsonMat& operator=(AlsonMat&&) noexcept;
    
    ~AlsonMat();
 
public:
    static AlsonMat create(int rows, int cols, int type);
    
    static AlsonMat create(int rows, int cols, int type, uint8_t* data);
    
    AlsonMat clone();
    
    static AlsonMat load(const std::string& filename);
    
    void save(const std::string& filename);
    
    uint8_t* data() const { return m_data; }
    
    void setData(uint8_t* data) { m_data = data; }
 
public:
    int rows;
    int cols;
    int type;
    int depth;
    int channels;
    size_t elementSize;
    size_t total;
 
private:
    uint8_t* m_data;
    std::allocator<uint8_t> m_allocator;
    int* m_refCount;
};
 
 
} // namespace ALSON
 
 
#endif // !ALSON_CLASSIC_BASE_ENTITY_ALSON_MAT_H