paddle_inference 0.4.0

use crate::call;
use crate::common::OneDimArrayCstr;
use crate::config::model::Model;
use crate::config::Config;
use crate::ctypes::{
    PD_Predictor, PD_PredictorClone, PD_PredictorDestroy, PD_PredictorGetInputHandle,
    PD_PredictorGetInputNames, PD_PredictorGetInputNum, PD_PredictorGetOutputHandle,
    PD_PredictorGetOutputNames, PD_PredictorGetOutputNum, PD_PredictorRun,
};
use crate::tensor::Tensor;
use crate::utils::to_c_str;

/// Paddle Inference 的预测器
pub struct Predictor {
    ptr: *mut PD_Predictor,
}

impl Predictor {
    pub fn builder(model: Model) -> Config {
        Config::new(model)
    }

    pub(crate) fn from_ptr(ptr: *mut PD_Predictor) -> Self {
        Self { ptr }
    }
}

impl Predictor {
    /// 获取输入 Tensor 名称
    pub fn input_names(&self) -> OneDimArrayCstr {
        let ptr = call! { PD_PredictorGetInputNames(self.ptr) };
        OneDimArrayCstr::from_ptr(ptr)
    }

    /// 获取输入 Tensor 数量
    pub fn input_num(&self) -> usize {
        call! { PD_PredictorGetInputNum(self.ptr) }
    }

    /// 根据名称获取输入 Tensor
    ///
    /// **注意:** 如果输入名称中包含字符`\0`，则只会将`\0`之前的字符作为输入
    pub fn input(&self, name: &str) -> Tensor {
        let (_n, name) = to_c_str(name);
        let ptr = call! { PD_PredictorGetInputHandle(self.ptr, name) };
        Tensor::from_ptr(ptr)
    }

    /// 获取输出 Tensor 名称
    pub fn output_names(&self) -> OneDimArrayCstr {
        let ptr = call! { PD_PredictorGetOutputNames(self.ptr) };
        OneDimArrayCstr::from_ptr(ptr)
    }

    /// 获取输出 Tensor 数量
    pub fn output_num(&self) -> usize {
        call! { PD_PredictorGetOutputNum(self.ptr) }
    }

    /// 根据名称获取输出 Tensor
    ///
    /// **注意:** 如果输入名称中包含字符`\0`，则只会将`\0`之前的字符作为输入
    pub fn output(&self, name: &str) -> Tensor {
        let (_n, name) = to_c_str(name);
        let ptr = call! { PD_PredictorGetOutputHandle(self.ptr, name) };
        Tensor::from_ptr(ptr)
    }
}

impl Predictor {
    /// 执行模型预测，**需要在设置输入Tensor数据后调用**
    pub fn run(&self) -> bool {
        call! { PD_PredictorRun(self.ptr) }
    }
}

impl Clone for Predictor {
    fn clone(&self) -> Self {
        let ptr = call! { PD_PredictorClone(self.ptr) };
        Self { ptr }
    }
}

impl Drop for Predictor {
    fn drop(&mut self) {
        call! {
            PD_PredictorDestroy(self.ptr)
        };
    }
}