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

//! # Solutions
//! Data types that are used to create solutions to predicates.

use serde::{Deserialize, Serialize};

use crate::{Key, PredicateAddress, Value};

#[cfg(feature = "schema")]
use schemars::JsonSchema;

/// Index into the solution data.
pub type SolutionDataIndex = u16;

#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash, Serialize, Deserialize)]
#[cfg_attr(feature = "schema", derive(JsonSchema))]
/// A solution to predicates.
pub struct Solution {
    /// The input data for each predicate.
    pub data: Vec<SolutionData>,
}

#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash, Serialize, Deserialize)]
#[cfg_attr(feature = "schema", derive(JsonSchema))]
/// The data the solver is required to provide to solve a predicate.
pub struct SolutionData {
    /// Which predicate this input data is for.
    pub predicate_to_solve: PredicateAddress,
    /// The decision variables for the predicate.
    pub decision_variables: Vec<Value>,
    /// The transient data being proposed.
    pub transient_data: Vec<Mutation>,
    /// The state mutations being proposed.
    pub state_mutations: Vec<Mutation>,
}

#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash, Serialize, Deserialize)]
#[cfg_attr(feature = "schema", derive(JsonSchema))]
/// A mutation to a single key in state or transient data.
pub struct Mutation {
    /// Key to data.
    pub key: Key,
    /// Value to contract the key to.
    /// Empty value means the value is being deleted.
    pub value: Value,
}

impl Solution {
    /// Get the length of all the state mutations in the solution.
    pub fn state_mutations_len(&self) -> usize {
        self.data.iter().map(|d| d.state_mutations.len()).sum()
    }

    /// Get the length of all the transient data in the solution.
    pub fn transient_data_len(&self) -> usize {
        self.data.iter().map(|d| d.transient_data.len()).sum()
    }
}