zenith-float-macro 1.0.2

//! `expr!` procedural macro for zenith-float.
//!
//! Depend on the `zenith-float` crate and use `zenith_float::expr`. This crate is not a direct dependency for applications.

#![allow(missing_docs)]
#![deny(unused)]
#![deny(clippy::suspicious)]

mod cplx;
mod util;

use proc_macro2::TokenStream;
use quote::quote;
use syn::{
    parse::Parse, spanned::Spanned, BinOp, Error, Expr, ExprBinary, ExprCall, ExprGroup, ExprLit,
    ExprParen, ExprPath, ExprUnary, Lit, Token, UnOp,
};
use util::{check_arg_num, str_to_exact_num_expr, str_to_exact_num_literal};
use zenith_float_num::{Consts, EXPONENT_BIT_SIZE};

// Speculative error estimation.
// This error is added upfront, before actual error is known.
// It helps to avoid additional recalculations due to changing error estimation.
pub(crate) const SPEC_ADD_ERR: usize = 32;

pub(crate) struct MacroInput {
    pub(crate) expr: Expr,
    pub(crate) ctx: Expr,
}

impl Parse for MacroInput {
    fn parse(input: syn::parse::ParseStream) -> syn::Result<Self> {
        let expr = input.parse()?;
        input.parse::<Token![,]>()?;

        let ctx = input.parse()?;

        Ok(MacroInput { expr, ctx })
    }
}

fn traverse_binary(
    expr: &ExprBinary,
    err: &mut Vec<usize>,
    cc: &mut Consts,
) -> Result<TokenStream, Error> {
    let left_expr = traverse_expr(&expr.left, err, cc)?;
    let right_expr = traverse_expr(&expr.right, err, cc)?;

    let errs_id = err.len();

    let ts = match expr.op {
        BinOp::Add(_) => {
            err.push(2);
            quote!({
                let arg1 = #left_expr;
                let arg2 = #right_expr;
                let ret = zenith_float::ExactNum::add(&arg1, &arg2, p_wrk, zenith_float::RoundingMode::None);
                if arg1.inexact() || arg2.inexact() {
                    if let (Some(e1), Some(e2), Some(e3)) = (arg1.exponent(), arg2.exponent(), ret.exponent()) {
                        if (e1 as isize - e2 as isize).abs() <= 1 && arg1.sign() != arg2.sign() {
                            let newerr = (e1.max(e2) as isize - e3 as isize).unsigned_abs() + 1;
                            if errs[#errs_id] < newerr {
                                errs[#errs_id] = newerr;
                                continue;
                            }
                        }
                    }
                }
                ret
            })
        }
        BinOp::Sub(_) => {
            err.push(2);
            quote!({
                let arg1 = #left_expr;
                let arg2 = #right_expr;
                let ret = zenith_float::ExactNum::sub(&arg1, &arg2, p_wrk, zenith_float::RoundingMode::None);
                if arg1.inexact() || arg2.inexact() {
                    if let (Some(e1), Some(e2), Some(e3)) = (arg1.exponent(), arg2.exponent(), ret.exponent()) {
                        if (e1 as isize - e2 as isize).abs() <= 1 && arg1.sign() == arg2.sign() {
                            let newerr = (e1.max(e2) as isize - e3 as isize).unsigned_abs() + 1;
                            if errs[#errs_id] < newerr {
                                errs[#errs_id] = newerr;
                                continue;
                            }
                        }
                    }
                }
                ret
            })
        }
        BinOp::Mul(_) => {
            err.push(3);
            quote!(
                zenith_float::ExactNum::mul(&(#left_expr), &(#right_expr), p_wrk, zenith_float::RoundingMode::None))
        }
        BinOp::Div(_) => {
            err.push(3);
            quote!(zenith_float::ExactNum::div(&(#left_expr), &(#right_expr), p_wrk, zenith_float::RoundingMode::None))
        }
        BinOp::Rem(_) => {
            quote!(zenith_float::ExactNum::rem(&(#left_expr), &(#right_expr)))
        }
        _ => return Err(Error::new(
            expr.span(),
            "unexpected binary operator. Only \"+\", \"-\", \"*\", \"/\", and \"%\" are allowed.",
        )),
    };

    Ok(ts)
}

fn one_arg_fun(
    fun: TokenStream,
    expr: &ExprCall,
    initial_err: usize,
    err: &mut Vec<usize>,
    cc: &mut Consts,
    use_cc: bool,
) -> Result<TokenStream, Error> {
    check_arg_num(1, expr)?;

    let arg = traverse_expr(&expr.args[0], err, cc)?;
    err.push(initial_err);

    let ret = if use_cc {
        quote!(#fun(&(#arg), p_wrk, zenith_float::RoundingMode::None, cc))
    } else {
        quote!(#fun(&(#arg), p_wrk, zenith_float::RoundingMode::None))
    };

    Ok(ret)
}

fn two_arg_fun(
    fun: TokenStream,
    expr: &ExprCall,
    initial_err: usize,
    err: &mut Vec<usize>,
    cc: &mut Consts,
    use_cc: bool,
) -> Result<TokenStream, Error> {
    check_arg_num(2, expr)?;

    let arg1 = traverse_expr(&expr.args[0], err, cc)?;
    let arg2 = traverse_expr(&expr.args[1], err, cc)?;
    err.push(initial_err);

    let ret = if use_cc {
        quote!(#fun(&(#arg1), &(#arg2), p_wrk, zenith_float::RoundingMode::None, cc))
    } else {
        quote!(#fun(&(#arg1), &(#arg2), p_wrk, zenith_float::RoundingMode::None))
    };

    Ok(ret)
}

fn three_arg_fun(
    fun: TokenStream,
    expr: &ExprCall,
    initial_err: usize,
    err: &mut Vec<usize>,
    cc: &mut Consts,
    use_cc: bool,
) -> Result<TokenStream, Error> {
    check_arg_num(3, expr)?;

    let arg1 = traverse_expr(&expr.args[0], err, cc)?;
    let arg2 = traverse_expr(&expr.args[1], err, cc)?;
    let arg3 = traverse_expr(&expr.args[2], err, cc)?;
    err.push(initial_err);

    let ret = if use_cc {
        quote!(#fun(&(#arg1), &(#arg2), &(#arg3), p_wrk, zenith_float::RoundingMode::None, cc))
    } else {
        quote!(#fun(&(#arg1), &(#arg2), &(#arg3), p_wrk, zenith_float::RoundingMode::None))
    };
    Ok(ret)
}

fn root_fun(
    expr: &ExprCall,
    initial_err: usize,
    err: &mut Vec<usize>,
    cc: &mut Consts,
) -> Result<TokenStream, Error> {
    check_arg_num(2, expr)?;

    let arg = traverse_expr(&expr.args[0], err, cc)?;
    let n = &expr.args[1];
    err.push(initial_err);

    Ok(quote!(zenith_float::ExactNum::nth_root(
        &(#arg),
        #n as usize,
        p_wrk,
        zenith_float::RoundingMode::None
    )))
}

fn ldexp_fun(
    expr: &ExprCall,
    initial_err: usize,
    err: &mut Vec<usize>,
    cc: &mut Consts,
    scalb: bool,
) -> Result<TokenStream, Error> {
    check_arg_num(2, expr)?;

    let arg = traverse_expr(&expr.args[0], err, cc)?;
    let n = &expr.args[1];
    err.push(initial_err);

    let fun = if scalb {
        quote!(zenith_float::ExactNum::scalb)
    } else {
        quote!(zenith_float::ExactNum::ldexp)
    };

    Ok(quote!(#fun(
        &(#arg),
        #n as zenith_float::Exponent,
        p_wrk,
        zenith_float::RoundingMode::None
    )))
}

fn bessel_j_fun(
    expr: &ExprCall,
    initial_err: usize,
    err: &mut Vec<usize>,
    cc: &mut Consts,
) -> Result<TokenStream, Error> {
    check_arg_num(2, expr)?;

    let arg = traverse_expr(&expr.args[0], err, cc)?;
    let n = &expr.args[1];
    err.push(initial_err);

    Ok(quote!(zenith_float::ExactNum::bessel_j(
        &(#arg),
        #n as usize,
        p_wrk,
        zenith_float::RoundingMode::None,
        cc
    )))
}

fn four_arg_fun(
    fun: TokenStream,
    expr: &ExprCall,
    initial_err: usize,
    err: &mut Vec<usize>,
    cc: &mut Consts,
) -> Result<TokenStream, Error> {
    check_arg_num(4, expr)?;
    let arg1 = traverse_expr(&expr.args[0], err, cc)?;
    let arg2 = traverse_expr(&expr.args[1], err, cc)?;
    let arg3 = traverse_expr(&expr.args[2], err, cc)?;
    let arg4 = traverse_expr(&expr.args[3], err, cc)?;
    err.push(initial_err);
    Ok(quote!(#fun(
        &(#arg1),
        &(#arg2),
        &(#arg3),
        &(#arg4),
        p_wrk,
        zenith_float::RoundingMode::None,
        cc
    )))
}

fn legendre_p_fun(
    expr: &ExprCall,
    initial_err: usize,
    err: &mut Vec<usize>,
    cc: &mut Consts,
) -> Result<TokenStream, Error> {
    check_arg_num(2, expr)?;
    let arg = traverse_expr(&expr.args[0], err, cc)?;
    let n = &expr.args[1];
    err.push(initial_err);
    Ok(quote!(zenith_float::ExactNum::legendre_p(
        &(#arg),
        #n as u32,
        p_wrk,
        zenith_float::RoundingMode::None
    )))
}

fn legendre_p_assoc_fun(
    expr: &ExprCall,
    initial_err: usize,
    err: &mut Vec<usize>,
    cc: &mut Consts,
) -> Result<TokenStream, Error> {
    check_arg_num(3, expr)?;
    let arg = traverse_expr(&expr.args[0], err, cc)?;
    let n = &expr.args[1];
    let m = &expr.args[2];
    err.push(initial_err);
    Ok(quote!(zenith_float::ExactNum::assoc_legendre_p(
        &(#arg),
        #n as u32,
        #m as i32,
        p_wrk,
        zenith_float::RoundingMode::None
    )))
}

fn one_arg_fun_errcheck(
    fun: TokenStream,
    expr: &ExprCall,
    initial_err: usize,
    err: &mut Vec<usize>,
    errcheck: TokenStream,
    cc: &mut Consts,
) -> Result<TokenStream, Error> {
    check_arg_num(1, expr)?;

    let arg = traverse_expr(&expr.args[0], err, cc)?;
    let errs_id = err.len();
    err.push(initial_err);

    Ok(quote!({
        let arg = #arg;

        let newerr = zenith_float::macro_util::compute_added_err(#errcheck);
        if errs[#errs_id] < newerr {
            errs[#errs_id] = newerr;
            continue;
        }

        #fun(&arg, p_wrk, zenith_float::RoundingMode::None, cc)
    }))
}

fn trig_fun(
    fun: TokenStream,
    expr: &ExprCall,
    initial_err: usize,
    err: &mut Vec<usize>,
    errfun: TokenStream,
    cc: &mut Consts,
) -> Result<TokenStream, Error> {
    check_arg_num(1, expr)?;

    let arg = traverse_expr(&expr.args[0], err, cc)?;
    let errs_id = err.len();
    err.push(initial_err);

    Ok(quote!({
        let arg = zenith_float::macro_util::check_exponent_range(#arg, emin, emax);

        let newerr = zenith_float::macro_util::compute_added_err(zenith_float::macro_util::ErrAlgo::Trig(&arg, p_wrk, #errfun, cc, emin));
        if errs[#errs_id] < newerr {
            errs[#errs_id] = newerr;
            continue;
        }

        #fun(&arg, p_wrk, zenith_float::RoundingMode::None, cc)
    }))
}

fn two_arg_fun_errcheck(
    fun: TokenStream,
    expr: &ExprCall,
    initial_err: usize,
    err: &mut Vec<usize>,
    errcheck: TokenStream,
    cc: &mut Consts,
) -> Result<TokenStream, Error> {
    check_arg_num(2, expr)?;

    let arg1 = traverse_expr(&expr.args[0], err, cc)?;
    let arg2 = traverse_expr(&expr.args[1], err, cc)?;

    let errs_id = err.len();

    err.push(initial_err);

    Ok(quote!({
        let arg1 = #arg1;
        let arg2 = #arg2;

        let newerr = zenith_float::macro_util::compute_added_err(#errcheck);
        if errs[#errs_id] < newerr {
            errs[#errs_id] = newerr;
            continue;
        }

        #fun(&arg1, &arg2, p_wrk, zenith_float::RoundingMode::None, cc)
    }))
}

fn traverse_call(
    expr: &ExprCall,
    err: &mut Vec<usize>,
    cc: &mut Consts,
) -> Result<TokenStream, Error> {
    let errmes = "unexpected function name. Only \"recip\", \"sqrt\", \"cbrt\", \"root\", \"ln\", \"log2\", \"log10\", \"log\", \"log1p\", \"exp\", \"exp2\", \"exp10\", \"expm1\", \"pow\", \"rem_pi\", \"sin\", \"cos\", \"tan\", \"asin\", \"acos\", \"atan\", \"atan2\", \"hypot\", \"fma\", \"mul_add\", \"sinh\", \"cosh\", \"tanh\", \"asinh\", \"acosh\", \"atanh\", \"erf\", \"erfc\", \"gamma\", \"ln_gamma\", \"digamma\", \"gammainc\", \"gammainc_upper\", \"ei\", \"si\", \"ci\", \"li\", \"fresnel_s\", \"fresnel_c\", \"ai\", \"bi\", \"bessel_j\", \"bessel_j_nu\", \"bessel_y\", \"bessel_i\", \"bessel_k\", \"elliptic_k\", \"elliptic_e\", \"elliptic_e_inc\", \"elliptic_f\", \"elliptic_pi\", \"elliptic_pi_inc\", \"jacobi_am\", \"jacobi_sn\", \"jacobi_cn\", \"jacobi_dn\", \"jacobi_cd\", \"jacobi_ns\", \"jacobi_nc\", \"jacobi_nd\", \"jacobi_sc\", \"jacobi_sd\", \"jacobi_cs\", \"jacobi_ds\", \"jacobi_dc\", \"legendre_p\", \"legendre_p_assoc\", \"hypergeom_2f1\", \"betainc\", \"normal_pdf\", \"normal_cdf\", \"gamma_pdf\", \"beta_pdf\", \"poisson_pmf\", \"binomial_pmf\", \"chi_squared_cdf\", \"student_t_pdf\", \"ldexp\", \"scalb\", \"logb\" are allowed.";

    if let Expr::Path(fun) = expr.func.as_ref() {
        if let Some(fname) = fun.path.get_ident() {
            let ts = match fname.to_string().as_str() {
                "recip" => one_arg_fun(
                    quote!(zenith_float::ExactNum::reciprocal),
                    expr,
                    2,
                    err,
                    cc,
                    false,
                ),
                "sqrt" => one_arg_fun(
                    quote!(zenith_float::ExactNum::sqrt),
                    expr,
                    1,
                    err,
                    cc,
                    false,
                ),
                "cbrt" => one_arg_fun(
                    quote!(zenith_float::ExactNum::cbrt),
                    expr,
                    1,
                    err,
                    cc,
                    false,
                ),
                "root" => root_fun(expr, 1, err, cc),
                "ln" => one_arg_fun_errcheck(
                    quote!(zenith_float::ExactNum::ln),
                    expr,
                    SPEC_ADD_ERR,
                    err,
                    quote!(zenith_float::macro_util::ErrAlgo::Log(&arg, 2, emin)),
                    cc,
                ),
                "log2" => one_arg_fun_errcheck(
                    quote!(zenith_float::ExactNum::log2),
                    expr,
                    SPEC_ADD_ERR,
                    err,
                    quote!(zenith_float::macro_util::ErrAlgo::Log(&arg, 3, emin)),
                    cc,
                ),
                "log10" => one_arg_fun_errcheck(
                    quote!(zenith_float::ExactNum::log10),
                    expr,
                    SPEC_ADD_ERR,
                    err,
                    quote!(zenith_float::macro_util::ErrAlgo::Log(&arg, 6, emin)),
                    cc,
                ),
                "log" => two_arg_fun_errcheck(
                    quote!(zenith_float::ExactNum::log),
                    expr,
                    SPEC_ADD_ERR,
                    err,
                    quote!(zenith_float::macro_util::ErrAlgo::Log2(&arg2, &arg1, emin)),
                    cc,
                ),
                "log1p" => one_arg_fun(
                    quote!(zenith_float::ExactNum::log1p),
                    expr,
                    SPEC_ADD_ERR,
                    err,
                    cc,
                    true,
                ),
                "exp" => one_arg_fun(
                    quote!(zenith_float::ExactNum::exp),
                    expr,
                    EXPONENT_BIT_SIZE + 1,
                    err,
                    cc,
                    true,
                ),
                "exp2" => one_arg_fun(
                    quote!(zenith_float::ExactNum::exp2),
                    expr,
                    EXPONENT_BIT_SIZE + 1,
                    err,
                    cc,
                    true,
                ),
                "exp10" => one_arg_fun(
                    quote!(zenith_float::ExactNum::exp10),
                    expr,
                    EXPONENT_BIT_SIZE + 1,
                    err,
                    cc,
                    true,
                ),
                "expm1" => one_arg_fun(
                    quote!(zenith_float::ExactNum::expm1),
                    expr,
                    EXPONENT_BIT_SIZE + 1,
                    err,
                    cc,
                    true,
                ),
                "pow" => two_arg_fun_errcheck(
                    quote!(zenith_float::ExactNum::pow),
                    expr,
                    EXPONENT_BIT_SIZE + SPEC_ADD_ERR,
                    err,
                    quote!(zenith_float::macro_util::ErrAlgo::Pow(&arg1, &arg2, emin)),
                    cc,
                ),
                "rem_pi" => one_arg_fun(
                    quote!(zenith_float::ExactNum::rem_pi),
                    expr,
                    SPEC_ADD_ERR,
                    err,
                    cc,
                    true,
                ),
                "sin" => trig_fun(
                    quote!(zenith_float::ExactNum::sin),
                    expr,
                    SPEC_ADD_ERR,
                    err,
                    quote!(zenith_float::macro_util::TrigFun::Sin),
                    cc,
                ),
                "cos" => trig_fun(
                    quote!(zenith_float::ExactNum::cos),
                    expr,
                    SPEC_ADD_ERR,
                    err,
                    quote!(zenith_float::macro_util::TrigFun::Cos),
                    cc,
                ),
                "tan" => trig_fun(
                    quote!(zenith_float::ExactNum::tan),
                    expr,
                    SPEC_ADD_ERR,
                    err,
                    quote!(zenith_float::macro_util::TrigFun::Tan),
                    cc,
                ),
                "asin" => one_arg_fun_errcheck(
                    quote!(zenith_float::ExactNum::asin),
                    expr,
                    SPEC_ADD_ERR / 2,
                    err,
                    quote!(zenith_float::macro_util::ErrAlgo::Asin(&arg, emin)),
                    cc,
                ),
                "acos" => one_arg_fun_errcheck(
                    quote!(zenith_float::ExactNum::acos),
                    expr,
                    SPEC_ADD_ERR / 2,
                    err,
                    quote!(zenith_float::macro_util::ErrAlgo::Acos(&arg, emin)),
                    cc,
                ),
                "atan" => one_arg_fun(quote!(zenith_float::ExactNum::atan), expr, 2, err, cc, true),
                "atan2" => two_arg_fun(
                    quote!(zenith_float::ExactNum::atan2),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "hypot" => two_arg_fun(
                    quote!(zenith_float::ExactNum::hypot),
                    expr,
                    2,
                    err,
                    cc,
                    false,
                ),
                "fma" => {
                    three_arg_fun(quote!(zenith_float::ExactNum::fma), expr, 2, err, cc, false)
                }
                "mul_add" => three_arg_fun(
                    quote!(zenith_float::ExactNum::mul_add),
                    expr,
                    2,
                    err,
                    cc,
                    false,
                ),
                "sinh" => one_arg_fun(
                    quote!(zenith_float::ExactNum::sinh),
                    expr,
                    EXPONENT_BIT_SIZE + 1,
                    err,
                    cc,
                    true,
                ),
                "cosh" => one_arg_fun(
                    quote!(zenith_float::ExactNum::cosh),
                    expr,
                    EXPONENT_BIT_SIZE + 1,
                    err,
                    cc,
                    true,
                ),
                "tanh" => one_arg_fun(quote!(zenith_float::ExactNum::tanh), expr, 2, err, cc, true),
                "asinh" => one_arg_fun(
                    quote!(zenith_float::ExactNum::asinh),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "acosh" => one_arg_fun_errcheck(
                    quote!(zenith_float::ExactNum::acosh),
                    expr,
                    SPEC_ADD_ERR,
                    err,
                    quote!(zenith_float::macro_util::ErrAlgo::Acosh(&arg, emin)),
                    cc,
                ),
                "atanh" => one_arg_fun_errcheck(
                    quote!(zenith_float::ExactNum::atanh),
                    expr,
                    SPEC_ADD_ERR,
                    err,
                    quote!(zenith_float::macro_util::ErrAlgo::Atanh(&arg, emin)),
                    cc,
                ),
                "erf" => one_arg_fun(quote!(zenith_float::ExactNum::erf), expr, 2, err, cc, true),
                "erfc" => one_arg_fun(quote!(zenith_float::ExactNum::erfc), expr, 2, err, cc, true),
                "gamma" => one_arg_fun(
                    quote!(zenith_float::ExactNum::gamma),
                    expr,
                    EXPONENT_BIT_SIZE + 1,
                    err,
                    cc,
                    true,
                ),
                "ln_gamma" => one_arg_fun(
                    quote!(zenith_float::ExactNum::ln_gamma),
                    expr,
                    EXPONENT_BIT_SIZE + 1,
                    err,
                    cc,
                    true,
                ),
                "digamma" => one_arg_fun(
                    quote!(zenith_float::ExactNum::digamma),
                    expr,
                    EXPONENT_BIT_SIZE + 1,
                    err,
                    cc,
                    true,
                ),
                "gammainc" => two_arg_fun(
                    quote!(zenith_float::ExactNum::gammainc),
                    expr,
                    EXPONENT_BIT_SIZE + 1,
                    err,
                    cc,
                    true,
                ),
                "gammainc_upper" => two_arg_fun(
                    quote!(zenith_float::ExactNum::gammainc_upper),
                    expr,
                    EXPONENT_BIT_SIZE + 1,
                    err,
                    cc,
                    true,
                ),
                "ei" => one_arg_fun(quote!(zenith_float::ExactNum::ei), expr, 2, err, cc, true),
                "si" => one_arg_fun(quote!(zenith_float::ExactNum::si), expr, 2, err, cc, true),
                "ci" => one_arg_fun(quote!(zenith_float::ExactNum::ci), expr, 2, err, cc, true),
                "li" => one_arg_fun(quote!(zenith_float::ExactNum::li), expr, 2, err, cc, true),
                "fresnel_s" => one_arg_fun(
                    quote!(zenith_float::ExactNum::fresnel_s),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "fresnel_c" => one_arg_fun(
                    quote!(zenith_float::ExactNum::fresnel_c),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "ai" => one_arg_fun(quote!(zenith_float::ExactNum::ai), expr, 2, err, cc, true),
                "bi" => one_arg_fun(quote!(zenith_float::ExactNum::bi), expr, 2, err, cc, true),
                "bessel_j" => bessel_j_fun(expr, 2, err, cc),
                "bessel_j_nu" => two_arg_fun(
                    quote!(zenith_float::ExactNum::bessel_j_nu),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "bessel_y" => two_arg_fun(
                    quote!(zenith_float::ExactNum::bessel_y),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "bessel_i" => two_arg_fun(
                    quote!(zenith_float::ExactNum::bessel_i),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "bessel_k" => two_arg_fun(
                    quote!(zenith_float::ExactNum::bessel_k),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "elliptic_k" => one_arg_fun(
                    quote!(zenith_float::ExactNum::elliptic_k),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "elliptic_e" => one_arg_fun(
                    quote!(zenith_float::ExactNum::elliptic_e_complete),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "elliptic_e_inc" => two_arg_fun(
                    quote!(zenith_float::ExactNum::elliptic_e),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "elliptic_f" => two_arg_fun(
                    quote!(zenith_float::ExactNum::elliptic_f),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "jacobi_am" => two_arg_fun(
                    quote!(zenith_float::ExactNum::jacobi_am),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "jacobi_sn" => two_arg_fun(
                    quote!(zenith_float::ExactNum::jacobi_sn),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "jacobi_cn" => two_arg_fun(
                    quote!(zenith_float::ExactNum::jacobi_cn),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "jacobi_dn" => two_arg_fun(
                    quote!(zenith_float::ExactNum::jacobi_dn),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "jacobi_cd" => two_arg_fun(
                    quote!(zenith_float::ExactNum::jacobi_cd),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "jacobi_ns" => two_arg_fun(
                    quote!(zenith_float::ExactNum::jacobi_ns),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "jacobi_nc" => two_arg_fun(
                    quote!(zenith_float::ExactNum::jacobi_nc),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "jacobi_nd" => two_arg_fun(
                    quote!(zenith_float::ExactNum::jacobi_nd),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "jacobi_sc" => two_arg_fun(
                    quote!(zenith_float::ExactNum::jacobi_sc),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "jacobi_sd" => two_arg_fun(
                    quote!(zenith_float::ExactNum::jacobi_sd),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "jacobi_cs" => two_arg_fun(
                    quote!(zenith_float::ExactNum::jacobi_cs),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "jacobi_ds" => two_arg_fun(
                    quote!(zenith_float::ExactNum::jacobi_ds),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "jacobi_dc" => two_arg_fun(
                    quote!(zenith_float::ExactNum::jacobi_dc),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "elliptic_pi" => two_arg_fun(
                    quote!(zenith_float::ExactNum::elliptic_pi_complete),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "elliptic_pi_inc" => three_arg_fun(
                    quote!(zenith_float::ExactNum::elliptic_pi),
                    expr,
                    2,
                    err,
                    cc,
                    true,
                ),
                "legendre_p" => legendre_p_fun(expr, 2, err, cc),
                "legendre_p_assoc" => legendre_p_assoc_fun(expr, 2, err, cc),
                "hypergeom_2f1" => four_arg_fun(
                    quote!(zenith_float::ExactNum::hypergeom_2f1),
                    expr,
                    EXPONENT_BIT_SIZE + 1,
                    err,
                    cc,
                ),
                "betainc" => three_arg_fun(
                    quote!(zenith_float::ExactNum::betainc),
                    expr,
                    EXPONENT_BIT_SIZE + 1,
                    err,
                    cc,
                    true,
                ),
                "normal_pdf" => three_arg_fun(
                    quote!(zenith_float::ExactNum::normal_pdf),
                    expr,
                    EXPONENT_BIT_SIZE + 1,
                    err,
                    cc,
                    true,
                ),
                "normal_cdf" => three_arg_fun(
                    quote!(zenith_float::ExactNum::normal_cdf),
                    expr,
                    EXPONENT_BIT_SIZE + 1,
                    err,
                    cc,
                    true,
                ),
                "gamma_pdf" => three_arg_fun(
                    quote!(zenith_float::ExactNum::gamma_pdf),
                    expr,
                    EXPONENT_BIT_SIZE + 1,
                    err,
                    cc,
                    true,
                ),
                "beta_pdf" => three_arg_fun(
                    quote!(zenith_float::ExactNum::beta_pdf),
                    expr,
                    EXPONENT_BIT_SIZE + 1,
                    err,
                    cc,
                    true,
                ),
                "poisson_pmf" => two_arg_fun(
                    quote!(zenith_float::ExactNum::poisson_pmf),
                    expr,
                    EXPONENT_BIT_SIZE + 1,
                    err,
                    cc,
                    true,
                ),
                "binomial_pmf" => three_arg_fun(
                    quote!(zenith_float::ExactNum::binomial_pmf),
                    expr,
                    EXPONENT_BIT_SIZE + 1,
                    err,
                    cc,
                    true,
                ),
                "chi_squared_cdf" => two_arg_fun(
                    quote!(zenith_float::ExactNum::chi_squared_cdf),
                    expr,
                    EXPONENT_BIT_SIZE + 1,
                    err,
                    cc,
                    true,
                ),
                "student_t_pdf" => two_arg_fun(
                    quote!(zenith_float::ExactNum::student_t_pdf),
                    expr,
                    EXPONENT_BIT_SIZE + 1,
                    err,
                    cc,
                    true,
                ),
                "ldexp" => ldexp_fun(expr, 2, err, cc, false),
                "scalb" => ldexp_fun(expr, 2, err, cc, true),
                "logb" => one_arg_fun(
                    quote!(zenith_float::ExactNum::logb),
                    expr,
                    1,
                    err,
                    cc,
                    false,
                ),
                _ => return Err(Error::new(expr.span(), errmes)),
            }?;

            return Ok(ts);
        }
    }
    Err(Error::new(expr.span(), errmes))
}

fn traverse_group(
    expr: &ExprGroup,
    err: &mut Vec<usize>,
    cc: &mut Consts,
) -> Result<TokenStream, Error> {
    traverse_expr(&expr.expr, err, cc)
}

fn traverse_lit(expr: &ExprLit, cc: &mut Consts) -> Result<TokenStream, Error> {
    let span = expr.span();

    match &expr.lit {
        Lit::Str(v) => str_to_exact_num_expr(&v.value(), span, cc),
        Lit::Int(v) => str_to_exact_num_expr(v.base10_digits(), span, cc),
        Lit::Float(v) => str_to_exact_num_expr(v.base10_digits(), span, cc),
        _ => Err(Error::new(
            expr.span(),
            "unexpected literal. Only string, integer, or floating point literals are supported.",
        )),
    }
}

fn traverse_paren(
    expr: &ExprParen,
    err: &mut Vec<usize>,
    cc: &mut Consts,
) -> Result<TokenStream, Error> {
    traverse_expr(&expr.expr, err, cc)
}

fn traverse_path(expr: &ExprPath) -> Result<TokenStream, Error> {
    Ok(if expr.path.is_ident("pi") {
        quote!({ cc.pi(p_wrk, zenith_float::RoundingMode::None) })
    } else if expr.path.is_ident("e") {
        quote!({ cc.e(p_wrk, zenith_float::RoundingMode::None) })
    } else if expr.path.is_ident("ln_2") {
        quote!({ cc.ln_2(p_wrk, zenith_float::RoundingMode::None) })
    } else if expr.path.is_ident("ln_10") {
        quote!({ cc.ln_10(p_wrk, zenith_float::RoundingMode::None) })
    } else if expr.path.is_ident("sqrt2") {
        quote!({ cc.sqrt2(p_wrk, zenith_float::RoundingMode::None) })
    } else if expr.path.is_ident("phi") {
        quote!({ cc.phi(p_wrk, zenith_float::RoundingMode::None) })
    } else if expr.path.is_ident("euler_gamma") {
        quote!({ cc.euler_gamma(p_wrk, zenith_float::RoundingMode::None) })
    } else {
        quote!({
            let mut arg = zenith_float::ExactNum::from_ext((#expr).clone(), p_wrk, zenith_float::RoundingMode::ToEven, cc);
            arg.set_inexact(false);
            arg = zenith_float::macro_util::check_exponent_range(arg, emin, emax);
            arg
        })
    })
}

fn traverse_unary(
    expr: &ExprUnary,
    err: &mut Vec<usize>,
    cc: &mut Consts,
) -> Result<TokenStream, Error> {
    let op_expr = traverse_expr(&expr.expr, err, cc)?;

    match expr.op {
        UnOp::Neg(_) => Ok(quote!(zenith_float::ExactNum::neg(&(#op_expr)))),
        _ => Err(Error::new(
            expr.span(),
            "unexpected unary operator. Only \"-\" is allowed.",
        )),
    }
}

fn traverse_expr(expr: &Expr, err: &mut Vec<usize>, cc: &mut Consts) -> Result<TokenStream, Error> {
    match expr {
        Expr::Binary(e) => traverse_binary(e, err,cc),
        Expr::Call(e) => traverse_call(e, err,cc),
        Expr::Group(e) => traverse_group(e, err,cc),
        Expr::Lit(e) => traverse_lit(e, cc),
        Expr::Paren(e) => traverse_paren(e, err, cc),
        Expr::Path(e) => traverse_path(e),
        Expr::Unary(e) => traverse_unary(e, err, cc),
        _ => Err(Error::new(expr.span(), "unexpected expression. Only operators \"+\", \"-\", \"*\", \"/\", \"%\", functions \"recip\", \"sqrt\", \"cbrt\", \"ln\", \"log2\", \"log10\", \"log\", \"exp\", \"pow\", \"sin\", \"cos\", \"tan\", \"asin\", \"acos\", \"atan\", \"sinh\", \"cosh\", \"tanh\", \"asinh\", \"acosh\", \"atanh\", literals and variables, and grouping with parentheses are supported.")),
    }
}

// Docs for the macro are in the zenith-float crate.

#[proc_macro]
#[allow(missing_docs)]
pub fn expr(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    let pmi = syn::parse_macro_input!(input as MacroInput);

    let MacroInput { expr, ctx } = pmi;

    let mut err = Vec::new();

    let mut cc = Consts::new().expect("Failed to initialize constant cache.");

    let expr = traverse_expr(&expr, &mut err, &mut cc).unwrap_or_else(|e| e.to_compile_error());

    let err_sz = err.len();

    let ret = quote!({
        use zenith_float::FromExt;
        use zenith_float::ctx::Contextable;

        let mut ctx = &mut (#ctx);
        let p: usize = ctx.precision();
        let rm = ctx.rounding_mode();
        let emin = ctx.emin();
        let emax = ctx.emax();
        let cc = ctx.consts();

        let mut p_rnd = p + zenith_float::WORD_BIT_SIZE;
        let mut errs: [usize; #err_sz] = [#(#err, )*];

        loop {
            let p_wrk = p_rnd.saturating_add(errs.iter().sum());

            let mut ret: zenith_float::ExactNum = (#expr).into();

            if let Err(err) = ret.set_precision(p, rm) {
                ret = zenith_float::ExactNum::nan(Some(err));
            }

            break zenith_float::macro_util::check_exponent_range(ret, emin, emax);
        }
    });

    ret.into()
}

/// Complex `expr!`: same context and working-precision loop, `ExactComplex` leaves, cancellation on both parts.
#[proc_macro]
#[allow(missing_docs)]
pub fn cexpr(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    cplx::cexpr(input)
}

/// Compile-time decimal float literal.
///
/// Parses a string literal at compile time and expands to an exact `ExactNum`.
/// Use via the `zenith-float` crate: `use zenith_float::exact`.
#[proc_macro]
pub fn exact(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    let lit = syn::parse_macro_input!(input as syn::LitStr);
    str_to_exact_num_literal(&lit.value(), lit.span())
        .unwrap_or_else(|e| e.to_compile_error())
        .into()
}

/// Alias for [`exact`], matching dashu-float naming.
#[proc_macro]
pub fn fbig(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    exact(input)
}