function parameter handling

nac3core/src/codegen/expr.rs

@@ -1,12 +1,9 @@
-use std::{convert::TryInto, iter::once};
+use std::{collections::HashMap, convert::TryInto, iter::once};
 
 use crate::{
-    top_level::DefinitionId,
+    symbol_resolver::SymbolValue,
-    typecheck::typedef::{Type, TypeEnum},
+    top_level::{CodeGenContext, DefinitionId, TopLevelDef},
-};
+    typecheck::typedef::{FunSignature, Type, TypeEnum},
-use crate::{
-    top_level::{CodeGenContext, TopLevelDef},
-    typecheck::typedef::FunSignature,
 };
 use inkwell::{
     types::{BasicType, BasicTypeEnum},
@@ -65,10 +62,7 @@ impl<'ctx> CodeGenContext<'ctx> {
                     let fields = fields.borrow();
                     let fields =
                         fields_list.iter().map(|f| self.get_llvm_type(fields[&f.0])).collect_vec();
-                    self.ctx
+                    self.ctx.struct_type(&fields, false).ptr_type(AddressSpace::Generic).into()
-                        .struct_type(&fields, false)
-                        .ptr_type(AddressSpace::Generic)
-                        .into()
                 } else {
                     unreachable!()
                 };
@@ -93,16 +87,20 @@ impl<'ctx> CodeGenContext<'ctx> {
         })
     }
 
+    fn gen_symbol_val(&mut self, val: &SymbolValue) -> BasicValueEnum<'ctx> {
+        unimplemented!()
+    }
+
     fn gen_call(
         &mut self,
         obj: Option<(Type, BasicValueEnum<'ctx>)>,
         fun: (&FunSignature, DefinitionId),
-        params: &[BasicValueEnum<'ctx>],
+        params: Vec<(Option<String>, BasicValueEnum<'ctx>)>,
         ret: Type,
     ) -> Option<BasicValueEnum<'ctx>> {
         let key = self.get_subst_key(obj.map(|(a, _)| a), fun.0);
         let defs = self.top_level.definitions.read();
-        let definition = defs.get(fun.1.0).unwrap();
+        let definition = defs.get(fun.1 .0).unwrap();
         let val = if let TopLevelDef::Function { instance_to_symbol, .. } = &*definition.read() {
             let symbol = instance_to_symbol.get(&key).unwrap_or_else(|| {
                 // TODO: codegen for function that are not yet generated
@@ -117,8 +115,19 @@ impl<'ctx> CodeGenContext<'ctx> {
                 };
                 self.module.add_function(symbol, fun_ty, None)
             });
-            // TODO: deal with default parameters and reordering based on keys
+            let mut keys = fun.0.args.clone();
-            self.builder.build_call(fun_val, params, "call").try_as_basic_value().left()
+            let mut mapping = HashMap::new();
+            for (key, value) in params.into_iter() {
+                mapping.insert(key.unwrap_or_else(|| keys.remove(0).name), value);
+            }
+            // default value handling
+            for k in keys.into_iter() {
+                mapping.insert(k.name, self.gen_symbol_val(&k.default_value.unwrap()));
+            }
+            // reorder the parameters
+            let params =
+                fun.0.args.iter().map(|arg| mapping.remove(&arg.name).unwrap()).collect_vec();
+            self.builder.build_call(fun_val, &params, "call").try_as_basic_value().left()
         } else {
             unreachable!()
         };
@@ -158,7 +167,7 @@ impl<'ctx> CodeGenContext<'ctx> {
                 let ty = self.ctx.struct_type(&types, false);
                 ty.const_named_struct(&values).into()
             }
-            _ => unreachable!()
+            _ => unreachable!(),
         }
     }
 
@@ -261,10 +270,7 @@ impl<'ctx> CodeGenContext<'ctx> {
                     "tmparr",
                 );
                 let arr_ty = self.ctx.struct_type(
-                    &[
+                    &[self.ctx.i32_type().into(), ty.ptr_type(AddressSpace::Generic).into()],
-                        self.ctx.i32_type().into(),
-                        ty.ptr_type(AddressSpace::Generic).into(),
-                    ],
                     false,
                 );
                 let arr_str_ptr = self.builder.build_alloca(arr_ty, "tmparrstr");
@@ -445,12 +451,14 @@ impl<'ctx> CodeGenContext<'ctx> {
                 )
                 .fold(None, |prev, (lhs, rhs, op)| {
                     let ty = lhs.custom.unwrap();
-                    let current = if [self.primitives.int32, self.primitives.int64, self.primitives.bool]
+                    let current =
+                        if [self.primitives.int32, self.primitives.int64, self.primitives.bool]
                             .contains(&ty)
                         {
-                        let (lhs, rhs) =
+                            let (lhs, rhs) = if let (
-                            if let (BasicValueEnum::IntValue(lhs), BasicValueEnum::IntValue(rhs)) =
+                                BasicValueEnum::IntValue(lhs),
-                                (self.gen_expr(lhs), self.gen_expr(rhs))
+                                BasicValueEnum::IntValue(rhs),
+                            ) = (self.gen_expr(lhs), self.gen_expr(rhs))
                             {
                                 (lhs, rhs)
                             } else {

nac3core/src/symbol_resolver.rs

@@ -3,13 +3,15 @@ use crate::typecheck::typedef::Type;
 use crate::top_level::DefinitionId;
 use rustpython_parser::ast::Expr;
 
-pub enum SymbolValue<'a> {
+#[derive(Clone, PartialEq)]
+pub enum SymbolValue {
     I32(i32),
     I64(i64),
     Double(f64),
     Bool(bool),
-    Tuple(&'a [SymbolValue<'a>]),
+    Tuple(Vec<SymbolValue>),
-    Bytes(&'a [u8]),
+    // we should think about how to implement bytes later...
+    // Bytes(&'a [u8]),
 }
 
 pub trait SymbolResolver {

nac3core/src/typecheck/magic_methods.rs

@@ -77,7 +77,7 @@ pub fn impl_binop(unifier: &mut Unifier, _store: &PrimitiveStore, ty: Type, othe
                         vars: HashMap::new(),
                         args: vec![FuncArg {
                             ty: other,
-                            is_optional: false,
+                            default_value: None,
                             name: "other".into()
                         }]
                     }))
@@ -97,7 +97,7 @@ pub fn impl_binop(unifier: &mut Unifier, _store: &PrimitiveStore, ty: Type, othe
                         vars: HashMap::new(),
                         args: vec![FuncArg {
                             ty: other,
-                            is_optional: false,
+                            default_value: None,
                             name: "other".into()
                         }]
                     }))
@@ -132,7 +132,7 @@ pub fn impl_cmpop(unifier: &mut Unifier, store: &PrimitiveStore, ty: Type, other
                     vars: HashMap::new(),
                     args: vec![FuncArg {
                         ty: other_ty,
-                        is_optional: false,
+                        default_value: None,
                         name: "other".into()
                     }]
                 }))

nac3core/src/typecheck/type_inferencer/mod.rs

@@ -240,7 +240,7 @@ impl<'a> Inferencer<'a> {
         let fun = FunSignature {
             args: fn_args
                 .iter()
-                .map(|(k, ty)| FuncArg { name: k.clone(), ty: *ty, is_optional: false })
+                .map(|(k, ty)| FuncArg { name: k.clone(), ty: *ty, default_value: None })
                 .collect(),
             ret,
             vars: Default::default(),
@@ -513,7 +513,13 @@ impl<'a> Inferencer<'a> {
         for (a, b, c) in izip!(once(left).chain(comparators), comparators, ops) {
             let method =
                 comparison_name(c).ok_or_else(|| "unsupported comparator".to_string())?.to_string();
-            self.build_method_call(a.location, method, a.custom.unwrap(), vec![b.custom.unwrap()], boolean)?;
+            self.build_method_call(
+                a.location,
+                method,
+                a.custom.unwrap(),
+                vec![b.custom.unwrap()],
+                boolean,
+            )?;
         }
         Ok(boolean)
     }

nac3core/src/typecheck/typedef/mod.rs

@@ -6,8 +6,9 @@ use std::iter::once;
 use std::rc::Rc;
 use std::sync::{Arc, Mutex};
 
-use crate::top_level::DefinitionId;
 use super::unification_table::{UnificationKey, UnificationTable};
+use crate::symbol_resolver::SymbolValue;
+use crate::top_level::DefinitionId;
 
 #[cfg(test)]
 mod test;
@@ -30,9 +31,7 @@ pub struct Call {
 pub struct FuncArg {
     pub name: String,
     pub ty: Type,
-    // TODO: change this to an optional value
+    pub default_value: Option<SymbolValue>,
-    // for primitive types
-    pub is_optional: bool,
 }
 
 #[derive(Clone)]
@@ -457,7 +456,7 @@ impl Unifier {
                 let required: Vec<String> = signature
                     .args
                     .iter()
-                    .filter(|v| !v.is_optional)
+                    .filter(|v| v.default_value.is_none())
                     .map(|v| v.name.clone())
                     .rev()
                     .collect();
@@ -516,7 +515,7 @@ impl Unifier {
                     if x.name != y.name {
                         return Err("Functions differ in parameter names.".to_string());
                     }
-                    if x.is_optional != y.is_optional {
+                    if x.default_value != y.default_value {
                         return Err("Functions differ in optional parameters.".to_string());
                     }
                     self.unify(x.ty, y.ty)?;