"""SRHarness Engine: structured symbolic models for scientific discovery."""

from __future__ import annotations

from typing import Any, Callable

from .expression import (
    Aggregate,
    Binary,
    Expression,
    Function,
    Gather,
    GroupedParameter,
    Index,
    Indexed,
    Number,
    Parameter,
    Reduction,
    RelationLift,
    Symbol,
    Variable,
    aggr,
    function,
    gather,
    grouped_param,
    param,
    reduction,
    sour,
    targ,
)
from .analysis import count_parameters, fold_constants, parameter_values, unbound_parameters
from .desugar import desugar
from .evaluation import evaluate
from .optimize import FitResult, bind_parameters, fit
from .indexed_evaluation import RelationField
from .parser import parse
from .render import render


def _unary(name: str) -> Callable[[Any], Expression]:
    def constructor(value: Any) -> Expression:
        """Create a unary symbolic function expression.

        Args:
            value: Symbolic expression or value to wrap.

        Returns:
            A symbolic unary function call.
        """
        return function(name, value)

    constructor.__name__ = name
    constructor.__qualname__ = name
    return constructor


sin = _unary("sin")
cos = _unary("cos")
tan = _unary("tan")
tanh = _unary("tanh")
sinh = _unary("sinh")
cosh = _unary("cosh")
arcsin = _unary("arcsin")
arccos = _unary("arccos")
arctan = _unary("arctan")
asin = arcsin
acos = arccos
atan = arctan
exp = _unary("exp")
log = _unary("log")
log10 = _unary("log10")
log2 = _unary("log2")
ln = log
lg = log10
sqrt = _unary("sqrt")
abs = _unary("abs")
sigmoid = _unary("sigmoid")
sign = _unary("sign")
sec = _unary("sec")
sech = _unary("sech")
csc = _unary("csc")
cot = _unary("cot")
inv = _unary("inv")


def delay(value: Any, delta: Any) -> Expression:
    """Create a delayed-value expression.

    Args:
        value: Time-dependent expression to sample from the past.
        delta: Scalar or sample-aligned delay interval.

    Returns:
        A symbolic ``delay(value, delta)`` call.
    """
    return function("delay", value, delta)


__all__ = [
    "Aggregate", "Binary", "Expression", "FitResult", "Function", "Gather",
    "GroupedParameter",
    "Index", "Indexed", "Number", "Parameter", "Reduction", "RelationField",
    "RelationLift", "Symbol",
    "Variable", "abs", "acos", "aggr", "arccos", "arcsin", "arctan", "asin", "atan",
    "cos", "cosh",
    "bind_parameters", "cot", "count_parameters", "csc", "delay", "desugar",
    "evaluate", "exp", "fit",
    "fold_constants", "function", "gather", "grouped_param", "inv", "lg", "ln", "log",
    "log10", "log2",
    "param", "parameter_values", "parse", "reduction", "render", "sec", "sech",
    "sigmoid", "sign", "sin",
    "sinh", "sour", "sqrt", "tan", "tanh", "targ", "unbound_parameters",
]