cl-ast: Remove Param, replace with flat Pattern

This commit is contained in:
John 2025-02-23 02:01:38 -06:00
parent e3d94d8949
commit cc6168b55e
10 changed files with 16 additions and 49 deletions

View File

@ -145,17 +145,10 @@ pub enum ModuleKind {
pub struct Function { pub struct Function {
pub name: Sym, pub name: Sym,
pub sign: TyFn, pub sign: TyFn,
pub bind: Vec<Param>, pub bind: Vec<Pattern>,
pub body: Option<Expr>, pub body: Option<Expr>,
} }
/// A single parameter for a [Function]
#[derive(Clone, Debug, PartialEq, Eq, Hash)]
pub struct Param {
pub mutability: Mutability,
pub bind: Pattern,
}
/// A user-defined product type /// A user-defined product type
#[derive(Clone, Debug, PartialEq, Eq, Hash)] #[derive(Clone, Debug, PartialEq, Eq, Hash)]
pub struct Struct { pub struct Struct {

View File

@ -192,13 +192,6 @@ mod display {
} }
} }
impl Display for Param {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
let Self { mutability, bind } = self;
write!(f, "{mutability}{bind}")
}
}
impl Display for Struct { impl Display for Struct {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
let Self { name, kind } = self; let Self { name, kind } = self;

View File

@ -106,14 +106,10 @@ pub trait Fold {
Function { Function {
name: self.fold_sym(name), name: self.fold_sym(name),
sign: self.fold_ty_fn(sign), sign: self.fold_ty_fn(sign),
bind: bind.into_iter().map(|p| self.fold_param(p)).collect(), bind: bind.into_iter().map(|p| self.fold_pattern(p)).collect(),
body: body.map(|b| self.fold_expr(b)), body: body.map(|b| self.fold_expr(b)),
} }
} }
fn fold_param(&mut self, p: Param) -> Param {
let Param { mutability, bind } = p;
Param { mutability: self.fold_mutability(mutability), bind: self.fold_pattern(bind) }
}
fn fold_struct(&mut self, s: Struct) -> Struct { fn fold_struct(&mut self, s: Struct) -> Struct {
let Struct { name, kind } = s; let Struct { name, kind } = s;
Struct { name: self.fold_sym(name), kind: self.fold_struct_kind(kind) } Struct { name: self.fold_sym(name), kind: self.fold_struct_kind(kind) }

View File

@ -83,16 +83,11 @@ pub trait Visit<'a>: Sized {
let Function { name, sign, bind, body } = f; let Function { name, sign, bind, body } = f;
self.visit_sym(name); self.visit_sym(name);
self.visit_ty_fn(sign); self.visit_ty_fn(sign);
bind.iter().for_each(|p| self.visit_param(p)); bind.iter().for_each(|p| self.visit_pattern(p));
if let Some(b) = body { if let Some(b) = body {
self.visit_expr(b) self.visit_expr(b)
} }
} }
fn visit_param(&mut self, p: &'a Param) {
let Param { mutability, bind } = p;
self.visit_mutability(mutability);
self.visit_pattern(bind);
}
fn visit_struct(&mut self, s: &'a Struct) { fn visit_struct(&mut self, s: &'a Struct) {
let Struct { name, kind } = s; let Struct { name, kind } = s;
self.visit_sym(name); self.visit_sym(name);

View File

@ -3,7 +3,7 @@
use collect_upvars::collect_upvars; use collect_upvars::collect_upvars;
use super::{pattern, Callable, ConValue, Environment, Error, IResult, Interpret}; use super::{pattern, Callable, ConValue, Environment, Error, IResult, Interpret};
use cl_ast::{Function as FnDecl, Param, Sym}; use cl_ast::{Function as FnDecl, Sym};
use std::{ use std::{
cell::{Ref, RefCell}, cell::{Ref, RefCell},
collections::HashMap, collections::HashMap,
@ -70,7 +70,7 @@ impl Callable for Function {
// TODO: completely refactor data storage // TODO: completely refactor data storage
let mut frame = env.frame("fn args"); let mut frame = env.frame("fn args");
for (Param { mutability: _, bind }, value) in bind.iter().zip(args) { for (bind, value) in bind.iter().zip(args) {
for (name, value) in pattern::substitution(bind, value.clone())? { for (name, value) in pattern::substitution(bind, value.clone())? {
frame.insert(*name, Some(value)); frame.insert(*name, Some(value));
} }

View File

@ -1,6 +1,6 @@
//! Collects the "Upvars" of a function at the point of its creation, allowing variable capture //! Collects the "Upvars" of a function at the point of its creation, allowing variable capture
use crate::{convalue::ConValue, env::Environment}; use crate::{convalue::ConValue, env::Environment};
use cl_ast::{ast_visitor::visit::*, Function, Let, Param, Path, PathPart, Pattern, Sym}; use cl_ast::{ast_visitor::visit::*, Function, Let, Path, PathPart, Pattern, Sym};
use std::collections::{HashMap, HashSet}; use std::collections::{HashMap, HashSet};
pub fn collect_upvars(f: &Function, env: &Environment) -> super::Upvars { pub fn collect_upvars(f: &Function, env: &Environment) -> super::Upvars {
@ -67,9 +67,7 @@ impl<'a> Visit<'a> for CollectUpvars<'_> {
fn visit_function(&mut self, f: &'a cl_ast::Function) { fn visit_function(&mut self, f: &'a cl_ast::Function) {
let Function { name: _, sign: _, bind, body } = f; let Function { name: _, sign: _, bind, body } = f;
// parameters can never be upvars // parameters can never be upvars
for Param { mutability: _, bind } in bind { bind.iter().for_each(|pat| self.visit_pattern(pat));
self.visit_pattern(bind);
}
if let Some(body) = body { if let Some(body) = body {
self.visit_expr(body); self.visit_expr(body);
} }

View File

@ -137,10 +137,7 @@ impl Interpret for Struct {
bind: args bind: args
.iter() .iter()
.enumerate() .enumerate()
.map(|(idx, _)| Param { .map(|(idx, _)| Pattern::Name(idx.to_string().into()))
mutability: Mutability::Not,
bind: Pattern::Name(idx.to_string().into()),
})
.collect(), .collect(),
body: None, body: None,
}; };

View File

@ -469,7 +469,7 @@ impl Parse<'_> for Function {
} }
} }
type FnSig = (Vec<Param>, Vec<TyKind>); type FnSig = (Vec<Pattern>, Vec<TyKind>);
impl Parse<'_> for FnSig { impl Parse<'_> for FnSig {
/// Parses the [parameters](Param) associated with a Function /// Parses the [parameters](Param) associated with a Function
@ -488,16 +488,17 @@ impl Parse<'_> for FnSig {
} }
} }
type TypedParam = (Param, TyKind); type TypedParam = (Pattern, TyKind);
impl Parse<'_> for TypedParam { impl Parse<'_> for TypedParam {
/// Parses a single function [parameter](Param) /// Parses a single function [parameter](Param)
fn parse(p: &mut Parser) -> PResult<(Param, TyKind)> { fn parse(p: &mut Parser) -> PResult<(Pattern, TyKind)> {
Ok(( Ok((
Param { mutability: Mutability::parse(p)?, bind: Pattern::parse(p)? }, Pattern::parse(p)?,
{ if p.match_type(TokenKind::Colon, Parsing::Param).is_ok() {
p.match_type(TokenKind::Colon, Parsing::Param)?;
TyKind::parse(p)? TyKind::parse(p)?
} else {
TyKind::Path(Path::from(Sym::from("_")))
}, },
)) ))
} }

View File

@ -641,12 +641,6 @@ pub mod yamlify {
y.value(self); y.value(self);
} }
} }
impl Yamlify for Param {
fn yaml(&self, y: &mut Yamler) {
let Self { mutability, bind } = self;
y.key("Param").yaml(mutability).pair("pat", bind);
}
}
impl Yamlify for Ty { impl Yamlify for Ty {
fn yaml(&self, y: &mut Yamler) { fn yaml(&self, y: &mut Yamler) {
let Self { extents: _, kind } = self; let Self { extents: _, kind } = self;

View File

@ -106,7 +106,7 @@ impl<'a> Visit<'a> for Populator<'_, 'a> {
self.set_name(*name); self.set_name(*name);
self.visit_ty_fn(sign); self.visit_ty_fn(sign);
bind.iter().for_each(|p| self.visit_param(p)); bind.iter().for_each(|p| self.visit_pattern(p));
if let Some(b) = body { if let Some(b) = body {
self.visit_expr(b) self.visit_expr(b)
} }