use super::PassManager;
use crate::{pass::Pass, string_ref::StringRef};
use mlir_sys::{
mlirOpPassManagerAddOwnedPass, mlirOpPassManagerGetNestedUnder, mlirPrintPassPipeline,
MlirOpPassManager, MlirStringRef,
};
use std::{
ffi::c_void,
fmt::{self, Display, Formatter},
marker::PhantomData,
};
#[derive(Clone, Copy, Debug)]
pub struct OperationPassManager<'c, 'a> {
raw: MlirOpPassManager,
_parent: PhantomData<&'a PassManager<'c>>,
}
impl<'c, 'a> OperationPassManager<'c, 'a> {
pub fn nested_under(&self, name: &str) -> Self {
let name = StringRef::new(name);
unsafe { Self::from_raw(mlirOpPassManagerGetNestedUnder(self.raw, name.to_raw())) }
}
pub fn add_pass(&self, pass: Pass) {
unsafe { mlirOpPassManagerAddOwnedPass(self.raw, pass.to_raw()) }
}
pub const fn to_raw(self) -> MlirOpPassManager {
self.raw
}
pub unsafe fn from_raw(raw: MlirOpPassManager) -> Self {
Self {
raw,
_parent: Default::default(),
}
}
}
impl<'c, 'a> Display for OperationPassManager<'c, 'a> {
fn fmt(&self, formatter: &mut Formatter) -> fmt::Result {
let mut data = (formatter, Ok(()));
unsafe extern "C" fn callback(string: MlirStringRef, data: *mut c_void) {
let data = &mut *(data as *mut (&mut Formatter, fmt::Result));
let result = (|| -> fmt::Result {
write!(
data.0,
"{}",
StringRef::from_raw(string)
.as_str()
.map_err(|_| fmt::Error)?
)
})();
if data.1.is_ok() {
data.1 = result;
}
}
unsafe {
mlirPrintPassPipeline(self.raw, Some(callback), &mut data as *mut _ as *mut c_void);
}
data.1
}
}