use std::fmt;
use erg_common::dict::Dict;
#[allow(unused_imports)]
use erg_common::log;
use erg_common::Str;
use erg_parser::ast::{Block, ConstBlock, Params};
use super::constructors::subr_t;
use super::value::{EvalValueResult, ValueObj};
use super::{ParamTy, Predicate, TyParam, Type};
use crate::context::Context;
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
pub struct UserConstSubr {
pub name: Str,
pub(crate) params: Params,
pub(crate) block: ConstBlock,
pub(crate) sig_t: Type,
}
impl UserConstSubr {
pub const fn new(name: Str, params: Params, block: ConstBlock, sig_t: Type) -> Self {
Self {
name,
params,
block,
sig_t,
}
}
pub fn block(self) -> Block {
self.block.downgrade()
}
}
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
pub struct ValueArgs {
pub pos_args: Vec<ValueObj>,
pub kw_args: Dict<Str, ValueObj>,
}
impl fmt::Display for ValueArgs {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
let mut args = Vec::new();
for arg in &self.pos_args {
args.push(arg.to_string());
}
for (key, arg) in self.kw_args.iter() {
args.push(format!("{key} := {arg}"));
}
write!(f, "({})", args.join(", "))
}
}
impl From<ValueArgs> for Vec<TyParam> {
fn from(args: ValueArgs) -> Self {
args.pos_args.into_iter().map(TyParam::Value).collect()
}
}
impl ValueArgs {
pub const fn new(pos_args: Vec<ValueObj>, kw_args: Dict<Str, ValueObj>) -> Self {
ValueArgs { pos_args, kw_args }
}
pub fn empty() -> Self {
ValueArgs {
pos_args: Vec::new(),
kw_args: Dict::new(),
}
}
pub fn pos_only(pos_args: Vec<ValueObj>) -> Self {
ValueArgs {
pos_args,
kw_args: Dict::new(),
}
}
pub fn remove_left_or_key(&mut self, key: &str) -> Option<ValueObj> {
if !self.pos_args.is_empty() {
Some(self.pos_args.remove(0))
} else {
self.kw_args.remove(key)
}
}
}
#[derive(Clone)]
pub struct BuiltinConstSubr {
name: Str,
subr: fn(ValueArgs, &Context) -> EvalValueResult<TyParam>,
pub(crate) sig_t: Type,
as_type: Option<Type>,
}
impl std::fmt::Debug for BuiltinConstSubr {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
f.debug_struct("BuiltinConstSubr")
.field("name", &self.name)
.field("sig_t", &self.sig_t)
.field("as_type", &self.as_type)
.finish()
}
}
impl PartialEq for BuiltinConstSubr {
fn eq(&self, other: &Self) -> bool {
self.name == other.name
}
}
impl Eq for BuiltinConstSubr {}
impl std::hash::Hash for BuiltinConstSubr {
fn hash<H: std::hash::Hasher>(&self, state: &mut H) {
self.name.hash(state);
}
}
impl fmt::Display for BuiltinConstSubr {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "<built-in const subroutine '{}'>", self.name)
}
}
impl BuiltinConstSubr {
pub fn new<S: Into<Str>>(
name: S,
subr: fn(ValueArgs, &Context) -> EvalValueResult<TyParam>,
sig_t: Type,
as_type: Option<Type>,
) -> Self {
Self {
name: name.into(),
subr,
sig_t,
as_type,
}
}
pub fn call(&self, args: ValueArgs, ctx: &Context) -> EvalValueResult<TyParam> {
(self.subr)(args, ctx)
}
}
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
pub struct ClosureData {
pub(crate) nd_params: Vec<ParamTy>,
pub(crate) d_params: Vec<ParamTy>,
pub(crate) qual_name: Str,
}
impl ClosureData {
pub const fn new(nd_params: Vec<ParamTy>, d_params: Vec<ParamTy>, qual_name: Str) -> Self {
Self {
nd_params,
d_params,
qual_name,
}
}
}
#[allow(clippy::type_complexity)]
#[derive(Clone)]
pub struct GenConstSubr {
name: Str,
data: ClosureData,
subr: fn(ClosureData, ValueArgs, &Context) -> EvalValueResult<TyParam>,
pub(crate) sig_t: Type,
as_type: Option<Type>,
}
impl std::fmt::Debug for GenConstSubr {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
f.debug_struct("BuiltinConstSubr")
.field("name", &self.name)
.field("sig_t", &self.sig_t)
.field("as_type", &self.as_type)
.finish()
}
}
impl PartialEq for GenConstSubr {
fn eq(&self, other: &Self) -> bool {
self.name == other.name
}
}
impl Eq for GenConstSubr {}
impl std::hash::Hash for GenConstSubr {
fn hash<H: std::hash::Hasher>(&self, state: &mut H) {
self.name.hash(state);
}
}
impl fmt::Display for GenConstSubr {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "<const subroutine '{}'>", self.name)
}
}
impl GenConstSubr {
pub fn new<S: Into<Str>>(
name: S,
data: ClosureData,
subr: fn(ClosureData, ValueArgs, &Context) -> EvalValueResult<TyParam>,
sig_t: Type,
as_type: Option<Type>,
) -> Self {
Self {
name: name.into(),
data,
subr,
sig_t,
as_type,
}
}
pub fn call(&self, args: ValueArgs, ctx: &Context) -> EvalValueResult<TyParam> {
(self.subr)(self.data.clone(), args, ctx)
}
}
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
pub enum ConstSubr {
User(UserConstSubr),
Builtin(BuiltinConstSubr),
Gen(GenConstSubr),
}
impl fmt::Display for ConstSubr {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
match self {
ConstSubr::User(subr) => {
write!(f, "<user-defined const subroutine '{}'>", subr.name)
}
ConstSubr::Builtin(subr) => write!(f, "{subr}"),
ConstSubr::Gen(subr) => write!(f, "{subr}"),
}
}
}
impl ConstSubr {
pub fn sig_t(&self) -> &Type {
match self {
ConstSubr::User(user) => &user.sig_t,
ConstSubr::Builtin(builtin) => &builtin.sig_t,
ConstSubr::Gen(gen) => &gen.sig_t,
}
}
pub fn as_type(&self, ctx: &Context) -> Option<Type> {
match self {
ConstSubr::User(user) => {
let subr = match &user.sig_t {
Type::Subr(subr) => subr,
Type::Quantified(subr) => {
if let Type::Subr(subr) = subr.as_ref() {
subr
} else {
return None;
}
}
_ => {
return None;
}
};
if let Type::Refinement(refine) = subr.return_t.as_ref() {
if let Predicate::Equal { rhs, .. } = refine.pred.as_ref() {
let return_t = ctx.convert_tp_into_type(rhs.clone()).ok()?;
let var_params = subr.var_params.as_ref().map(|t| t.as_ref());
let kw_var_params = subr.kw_var_params.as_ref().map(|t| t.as_ref());
let subr_t = subr_t(
subr.kind,
subr.non_default_params.clone(),
var_params.cloned(),
subr.default_params.clone(),
kw_var_params.cloned(),
return_t,
);
let subr_t = if subr_t.has_qvar() {
subr_t.quantify()
} else {
subr_t
};
return Some(subr_t);
}
}
None
}
ConstSubr::Builtin(builtin) => builtin.as_type.clone(),
ConstSubr::Gen(gen) => gen.as_type.clone(),
}
}
}