forked from M-Labs/nac3
nac3core/artiq: support default param of option type
This commit is contained in:
parent
a38cc04444
commit
6ab73a223c
|
@ -361,7 +361,10 @@ fn rpc_codegen_callback_fn<'ctx, 'a>(
|
||||||
}
|
}
|
||||||
// default value handling
|
// default value handling
|
||||||
for k in keys.into_iter() {
|
for k in keys.into_iter() {
|
||||||
mapping.insert(k.name, ctx.gen_symbol_val(generator, &k.default_value.unwrap()).into());
|
mapping.insert(
|
||||||
|
k.name,
|
||||||
|
ctx.gen_symbol_val(generator, &k.default_value.unwrap(), k.ty).into()
|
||||||
|
);
|
||||||
}
|
}
|
||||||
// reorder the parameters
|
// reorder the parameters
|
||||||
let mut real_params = fun
|
let mut real_params = fun
|
||||||
|
|
|
@ -922,6 +922,7 @@ impl InnerResolver {
|
||||||
py: Python,
|
py: Python,
|
||||||
obj: &PyAny,
|
obj: &PyAny,
|
||||||
) -> PyResult<Result<SymbolValue, String>> {
|
) -> PyResult<Result<SymbolValue, String>> {
|
||||||
|
let id: u64 = self.helper.id_fn.call1(py, (obj,))?.extract(py)?;
|
||||||
let ty_id: u64 =
|
let ty_id: u64 =
|
||||||
self.helper.id_fn.call1(py, (self.helper.type_fn.call1(py, (obj,))?,))?.extract(py)?;
|
self.helper.id_fn.call1(py, (self.helper.type_fn.call1(py, (obj,))?,))?.extract(py)?;
|
||||||
Ok(if ty_id == self.primitive_ids.int || ty_id == self.primitive_ids.int32 {
|
Ok(if ty_id == self.primitive_ids.int || ty_id == self.primitive_ids.int32 {
|
||||||
|
@ -940,13 +941,17 @@ impl InnerResolver {
|
||||||
let elements: &PyTuple = obj.cast_as()?;
|
let elements: &PyTuple = obj.cast_as()?;
|
||||||
let elements: Result<Result<Vec<_>, String>, _> =
|
let elements: Result<Result<Vec<_>, String>, _> =
|
||||||
elements.iter().map(|elem| self.get_default_param_obj_value(py, elem)).collect();
|
elements.iter().map(|elem| self.get_default_param_obj_value(py, elem)).collect();
|
||||||
let elements = match elements? {
|
elements?.map(SymbolValue::Tuple)
|
||||||
Ok(el) => el,
|
} else if ty_id == self.primitive_ids.option {
|
||||||
Err(err) => return Ok(Err(err)),
|
if id == self.primitive_ids.none {
|
||||||
};
|
Ok(SymbolValue::OptionNone)
|
||||||
Ok(SymbolValue::Tuple(elements))
|
|
||||||
} else {
|
} else {
|
||||||
Err("only primitives values and tuple can be default parameter value".into())
|
self
|
||||||
|
.get_default_param_obj_value(py, obj.getattr("_nac3_option").unwrap())?
|
||||||
|
.map(|v| SymbolValue::OptionSome(Box::new(v)))
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
Err("only primitives values, option and tuple can be default parameter value".into())
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -962,8 +967,9 @@ impl SymbolResolver for Resolver {
|
||||||
for (key, val) in members.iter() {
|
for (key, val) in members.iter() {
|
||||||
let key: &str = key.extract()?;
|
let key: &str = key.extract()?;
|
||||||
if key == id.to_string() {
|
if key == id.to_string() {
|
||||||
sym_value =
|
if let Ok(Ok(v)) = self.0.get_default_param_obj_value(py, val) {
|
||||||
Some(self.0.get_default_param_obj_value(py, val).unwrap().unwrap());
|
sym_value = Some(v)
|
||||||
|
}
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -971,7 +977,7 @@ impl SymbolResolver for Resolver {
|
||||||
})
|
})
|
||||||
.unwrap()
|
.unwrap()
|
||||||
}
|
}
|
||||||
_ => unimplemented!("other type of expr not supported at {}", expr.location),
|
_ => unreachable!("only for resolving names"),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -91,6 +91,7 @@ impl<'ctx, 'a> CodeGenContext<'ctx, 'a> {
|
||||||
&mut self,
|
&mut self,
|
||||||
generator: &mut dyn CodeGenerator,
|
generator: &mut dyn CodeGenerator,
|
||||||
val: &SymbolValue,
|
val: &SymbolValue,
|
||||||
|
ty: Type,
|
||||||
) -> BasicValueEnum<'ctx> {
|
) -> BasicValueEnum<'ctx> {
|
||||||
match val {
|
match val {
|
||||||
SymbolValue::I32(v) => self.ctx.i32_type().const_int(*v as u64, true).into(),
|
SymbolValue::I32(v) => self.ctx.i32_type().const_int(*v as u64, true).into(),
|
||||||
|
@ -107,7 +108,7 @@ impl<'ctx, 'a> CodeGenContext<'ctx, 'a> {
|
||||||
ty.const_named_struct(&[str_ptr, size.into()]).into()
|
ty.const_named_struct(&[str_ptr, size.into()]).into()
|
||||||
}
|
}
|
||||||
SymbolValue::Tuple(ls) => {
|
SymbolValue::Tuple(ls) => {
|
||||||
let vals = ls.iter().map(|v| self.gen_symbol_val(generator, v)).collect_vec();
|
let vals = ls.iter().map(|v| self.gen_symbol_val(generator, v, ty)).collect_vec();
|
||||||
let fields = vals.iter().map(|v| v.get_type()).collect_vec();
|
let fields = vals.iter().map(|v| v.get_type()).collect_vec();
|
||||||
let ty = self.ctx.struct_type(&fields, false);
|
let ty = self.ctx.struct_type(&fields, false);
|
||||||
let ptr = self.builder.build_alloca(ty, "tuple");
|
let ptr = self.builder.build_alloca(ty, "tuple");
|
||||||
|
@ -124,6 +125,37 @@ impl<'ctx, 'a> CodeGenContext<'ctx, 'a> {
|
||||||
}
|
}
|
||||||
self.builder.build_load(ptr, "tup_val")
|
self.builder.build_load(ptr, "tup_val")
|
||||||
}
|
}
|
||||||
|
SymbolValue::OptionSome(v) => {
|
||||||
|
let ty = match self.unifier.get_ty_immutable(ty).as_ref() {
|
||||||
|
TypeEnum::TObj { obj_id, params, .. }
|
||||||
|
if *obj_id == self.primitives.option.get_obj_id(&self.unifier) =>
|
||||||
|
{
|
||||||
|
*params.iter().next().unwrap().1
|
||||||
|
}
|
||||||
|
_ => unreachable!("must be option type"),
|
||||||
|
};
|
||||||
|
let val = self.gen_symbol_val(generator, v, ty);
|
||||||
|
let ptr = self.builder.build_alloca(val.get_type(), "default_opt_some");
|
||||||
|
self.builder.build_store(ptr, val);
|
||||||
|
ptr.into()
|
||||||
|
}
|
||||||
|
SymbolValue::OptionNone => {
|
||||||
|
let ty = match self.unifier.get_ty_immutable(ty).as_ref() {
|
||||||
|
TypeEnum::TObj { obj_id, params, .. }
|
||||||
|
if *obj_id == self.primitives.option.get_obj_id(&self.unifier) =>
|
||||||
|
{
|
||||||
|
*params.iter().next().unwrap().1
|
||||||
|
}
|
||||||
|
_ => unreachable!("must be option type"),
|
||||||
|
};
|
||||||
|
let actual_ptr_type =
|
||||||
|
self.get_llvm_type(generator, ty).ptr_type(AddressSpace::Generic);
|
||||||
|
self.builder.build_bitcast(
|
||||||
|
self.ctx.i8_type().ptr_type(AddressSpace::Generic).const_null(),
|
||||||
|
actual_ptr_type,
|
||||||
|
"default_opt_none",
|
||||||
|
)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -605,7 +637,7 @@ pub fn gen_call<'ctx, 'a, G: CodeGenerator>(
|
||||||
}
|
}
|
||||||
mapping.insert(
|
mapping.insert(
|
||||||
k.name,
|
k.name,
|
||||||
ctx.gen_symbol_val(generator, &k.default_value.unwrap()).into(),
|
ctx.gen_symbol_val(generator, &k.default_value.unwrap(), k.ty).into(),
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
// reorder the parameters
|
// reorder the parameters
|
||||||
|
|
|
@ -29,6 +29,8 @@ pub enum SymbolValue {
|
||||||
Double(f64),
|
Double(f64),
|
||||||
Bool(bool),
|
Bool(bool),
|
||||||
Tuple(Vec<SymbolValue>),
|
Tuple(Vec<SymbolValue>),
|
||||||
|
OptionSome(Box<SymbolValue>),
|
||||||
|
OptionNone,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Display for SymbolValue {
|
impl Display for SymbolValue {
|
||||||
|
@ -50,6 +52,8 @@ impl Display for SymbolValue {
|
||||||
SymbolValue::Tuple(t) => {
|
SymbolValue::Tuple(t) => {
|
||||||
write!(f, "({})", t.iter().map(|v| format!("{}", v)).collect::<Vec<_>>().join(", "))
|
write!(f, "({})", t.iter().map(|v| format!("{}", v)).collect::<Vec<_>>().join(", "))
|
||||||
}
|
}
|
||||||
|
SymbolValue::OptionSome(v) => write!(f, "Some({})", v),
|
||||||
|
SymbolValue::OptionNone => write!(f, "none"),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -480,6 +480,33 @@ impl TopLevelComposer {
|
||||||
Some("tuple".to_string())
|
Some("tuple".to_string())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
SymbolValue::OptionNone => {
|
||||||
|
if let TypeAnnotation::CustomClass { id, .. } = ty {
|
||||||
|
if *id == primitive.option.get_obj_id(unifier) {
|
||||||
|
None
|
||||||
|
} else {
|
||||||
|
Some("option".into())
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
Some("option".into())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
SymbolValue::OptionSome(v) => {
|
||||||
|
if let TypeAnnotation::CustomClass { id, params } = ty {
|
||||||
|
if *id == primitive.option.get_obj_id(unifier) {
|
||||||
|
if params.len() == 1 {
|
||||||
|
Self::check_default_param_type(v, ¶ms[0], primitive, unifier)?;
|
||||||
|
None
|
||||||
|
} else {
|
||||||
|
Some("option".into())
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
Some("option".into())
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
Some("option".into())
|
||||||
|
}
|
||||||
|
}
|
||||||
};
|
};
|
||||||
if let Some(found) = res {
|
if let Some(found) = res {
|
||||||
Err(format!(
|
Err(format!(
|
||||||
|
@ -511,6 +538,10 @@ pub fn parse_parameter_default_value(
|
||||||
Constant::Tuple(tuple) => Ok(SymbolValue::Tuple(
|
Constant::Tuple(tuple) => Ok(SymbolValue::Tuple(
|
||||||
tuple.iter().map(|x| handle_constant(x, loc)).collect::<Result<Vec<_>, _>>()?,
|
tuple.iter().map(|x| handle_constant(x, loc)).collect::<Result<Vec<_>, _>>()?,
|
||||||
)),
|
)),
|
||||||
|
Constant::None => Err(format!(
|
||||||
|
"`None` is not supported, use `none` for option type instead ({})",
|
||||||
|
loc
|
||||||
|
)),
|
||||||
_ => unimplemented!("this constant is not supported at {}", loc),
|
_ => unimplemented!("this constant is not supported at {}", loc),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -548,6 +579,11 @@ pub fn parse_parameter_default_value(
|
||||||
}
|
}
|
||||||
_ => Err(format!("only allow constant integer here at {}", default.location))
|
_ => Err(format!("only allow constant integer here at {}", default.location))
|
||||||
}
|
}
|
||||||
|
ast::ExprKind::Name { id, .. } if *id == "Some".into() => Ok(
|
||||||
|
SymbolValue::OptionSome(
|
||||||
|
Box::new(parse_parameter_default_value(&args[0], resolver)?)
|
||||||
|
)
|
||||||
|
),
|
||||||
_ => Err(format!("unsupported default parameter at {}", default.location)),
|
_ => Err(format!("unsupported default parameter at {}", default.location)),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -556,15 +592,20 @@ pub fn parse_parameter_default_value(
|
||||||
.map(|x| parse_parameter_default_value(x, resolver))
|
.map(|x| parse_parameter_default_value(x, resolver))
|
||||||
.collect::<Result<Vec<_>, _>>()?
|
.collect::<Result<Vec<_>, _>>()?
|
||||||
)),
|
)),
|
||||||
|
ast::ExprKind::Name { id, .. } if id == &"none".into() => Ok(SymbolValue::OptionNone),
|
||||||
ast::ExprKind::Name { id, .. } => {
|
ast::ExprKind::Name { id, .. } => {
|
||||||
resolver.get_default_param_value(default).ok_or_else(
|
resolver.get_default_param_value(default).ok_or_else(
|
||||||
|| format!(
|
|| format!(
|
||||||
"`{}` cannot be used as a default parameter at {} (not primitive type or tuple / not defined?)",
|
"`{}` cannot be used as a default parameter at {} \
|
||||||
|
(not primitive type, option or tuple / not defined?)",
|
||||||
id,
|
id,
|
||||||
default.location
|
default.location
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
_ => Err(format!("unsupported default parameter at {}", default.location))
|
_ => Err(format!(
|
||||||
|
"unsupported default parameter (not primitive type, option or tuple) at {}",
|
||||||
|
default.location
|
||||||
|
))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue