#ifndef _ROAD_EXTRACT_H_
#define _ROAD_EXTRACT_H_

#include <memory>
#include "utils.h"
#include "acl/acl.h"
#include "model_process.h"

using namespace std;

namespace atlas_utils {


#ifndef DATA_TYPE_SIZE
#define DATA_TYPE_SIZE 4
#endif

class ROADExtract {
public:
    struct ConfigParams {
        float confThr = 0.0;
    };

    ROADExtract() {}
    ~ROADExtract() {
        Release();
    }

    int Init(const char* modelPath);
    int Inference(ImageData& input);
    int PostProcess(vector<float>& result);

    int GetInputWidth();
    int GetInputHeight();

    ConfigParams config;

private:
    void Release();

    vector<vector<int>> outDims_;
    vector<vector<int>> inDims_;
    ModelProcess model_;
    uint32_t modelWidth_;
    uint32_t modelHeight_;
    aclrtRunMode runMode_;

};

}

#endif