solverforge-macros 0.15.0

Derive macros for SolverForge constraint solver
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

use proc_macro2::TokenStream;
use syn::{Expr, Ident, ItemFn, Stmt};

#[derive(Debug, PartialEq, Eq)]
pub(crate) enum ImpactKind {
    Penalty,
    Reward,
}

#[derive(Debug, PartialEq, Eq)]
pub(crate) enum TerminalKind {
    GroupedScore,
}

#[derive(Debug)]
pub(crate) struct StreamNode {
    pub(crate) binding: String,
    pub(crate) supports_grouped_sharing: bool,
}

#[derive(Debug)]
pub(crate) struct TerminalConstraint {
    pub(crate) source_binding: String,
    pub(crate) impact: ImpactKind,
    pub(crate) weight: TokenStream,
    pub(crate) name: TokenStream,
    pub(crate) order: usize,
    pub(crate) kind: TerminalKind,
}

#[derive(Debug)]
pub(crate) enum TailMember {
    Terminal(TerminalConstraint),
    Other { tokens: TokenStream },
}

#[derive(Debug)]
pub(crate) struct ConstraintFunction {
    pub(crate) item: ItemFn,
    pub(crate) prefix_statements: Vec<Stmt>,
    pub(crate) tail: Option<Expr>,
    pub(crate) stream_nodes: Vec<StreamNode>,
    pub(crate) tail_members: Vec<TailMember>,
}

#[derive(Debug)]
pub(crate) struct SharedGroupedProgram {
    pub(crate) item: ItemFn,
    pub(crate) prefix_statements: Vec<Stmt>,
    pub(crate) bindings: Vec<String>,
    pub(crate) tail_members: Vec<TailMember>,
}

#[derive(Debug)]
pub(crate) enum ConstraintProgram {
    Passthrough(ConstraintFunction),
    SharedGrouped(SharedGroupedProgram),
}

impl ImpactKind {
    pub(crate) fn method_ident(&self) -> Ident {
        match self {
            ImpactKind::Penalty => Ident::new("penalize", proc_macro2::Span::call_site()),
            ImpactKind::Reward => Ident::new("reward", proc_macro2::Span::call_site()),
        }
    }
}