#[cfg(feature = "stubs")]
use pyo3_stub_gen::derive::{gen_stub_pyclass, gen_stub_pyclass_complex_enum};
use crate::{pickleable_new, quil::Quil};
use super::QuotedString;
#[derive(Clone, Debug, PartialEq, Eq, Hash)]
#[cfg_attr(feature = "stubs", gen_stub_pyclass)]
#[cfg_attr(
feature = "python",
pyo3::pyclass(module = "quil.instructions", eq, frozen, hash, get_all, subclass)
)]
pub struct Pragma {
pub name: String,
pub arguments: Vec<PragmaArgument>,
pub data: Option<String>,
}
pickleable_new! {
impl Pragma {
pub fn new(name: String, arguments: Vec<PragmaArgument>, data: Option<String>);
}
}
impl Quil for Pragma {
fn write(
&self,
f: &mut impl std::fmt::Write,
fall_back_to_debug: bool,
) -> crate::quil::ToQuilResult<()> {
write!(f, "PRAGMA {}", self.name)?;
for arg in &self.arguments {
write!(f, " ")?;
arg.write(f, fall_back_to_debug)?;
}
if let Some(data) = &self.data {
write!(f, " {}", QuotedString(data))?;
}
Ok(())
}
}
#[derive(Clone, Debug, PartialEq, Eq, Hash)]
#[cfg_attr(feature = "stubs", gen_stub_pyclass_complex_enum)]
#[cfg_attr(
feature = "python",
pyo3::pyclass(module = "quil.instructions", eq, frozen, hash)
)]
pub enum PragmaArgument {
Identifier(String),
Integer(u64),
}
impl Quil for PragmaArgument {
fn write(
&self,
f: &mut impl std::fmt::Write,
_fall_back_to_debug: bool,
) -> crate::quil::ToQuilResult<()> {
match self {
PragmaArgument::Identifier(i) => write!(f, "{i}"),
PragmaArgument::Integer(i) => write!(f, "{i}"),
}
.map_err(Into::into)
}
}
#[derive(Clone, Debug, PartialEq, Eq, Hash)]
#[cfg_attr(feature = "stubs", gen_stub_pyclass)]
#[cfg_attr(
feature = "python",
pyo3::pyclass(module = "quil.instructions", eq, frozen, hash, get_all, subclass)
)]
pub struct Include {
pub filename: String,
}
impl Quil for Include {
fn write(
&self,
f: &mut impl std::fmt::Write,
_fall_back_to_debug: bool,
) -> crate::quil::ToQuilResult<()> {
write!(f, r#"INCLUDE {}"#, QuotedString(&self.filename)).map_err(Into::into)
}
}
pickleable_new! {
impl Include {
pub fn new(filename: String);
}
}
pub const RESERVED_PRAGMA_EXTERN: &str = "EXTERN";