momba-explore 0.1.1

State space exploration engine for PTAs and MDPs augmented with variables.
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110

//! Convenience functions for simulating random runs.

use rand::seq::IteratorRandom;
use rand::Rng;

use super::*;

/// Represents an *oracle* used to choose a transition.
///
/// When faced with a non-deterministic chose between multiple transitions, the simulator
/// uses an oracle to determine which transition to choose.
pub trait Oracle<T: time::TimeType> {
    /// Chooses a transition from a vector of transitions.
    fn choose<'e, 't>(
        &self,
        state: &State<T::Valuations>,
        transitions: &'t Vec<Transition<'e, T>>,
    ) -> &'t Transition<'e, T>;
}

/// An [Oracle] choosing a transition uniformly at random.
pub struct UniformOracle {}

impl UniformOracle {
    /// Constructs a new uniform oracle.
    pub fn new() -> Self {
        UniformOracle {}
    }
}

impl<T: time::TimeType> Oracle<T> for UniformOracle {
    fn choose<'e, 't>(
        &self,
        _state: &State<T::Valuations>,
        transitions: &'t Vec<Transition<'e, T>>,
    ) -> &'t Transition<'e, T> {
        let mut rng = rand::thread_rng();
        transitions.iter().choose(&mut rng).unwrap()
    }
}

/// Injects certain transitions after a specified delay.
///
/// It is often useful to be able to enforce specific transitions based on
/// their action after a certain amount of time.
/// This oracle allows injecting specific transitions after a certain
/// amount of transitions.
pub struct InjectionOracle {}

/// A simulator used to simulate random runs.
pub struct Simulator<O: Oracle<T>, T: time::TimeType> {
    pub(crate) oracle: O,

    _phontom_time_type: std::marker::PhantomData<T>,
}

impl<O: Oracle<T>, T: time::TimeType> Simulator<O, T> {
    /// Creates a new simulator with the given oracle.
    pub fn new(oracle: O) -> Self {
        Simulator {
            oracle,
            _phontom_time_type: std::marker::PhantomData,
        }
    }

    /// Returns the oracle used by the simulator.
    pub fn oracle(&self) -> &O {
        &self.oracle
    }

    /// Simulates a random run.
    pub fn simulate(&self, explorer: &Explorer<T>, steps: usize) {
        let mut rng = rand::thread_rng();
        let mut state = explorer
            .initial_states()
            .into_iter()
            .choose(&mut rng)
            .unwrap();

        for _ in 0..steps {
            let transition = explorer
                .transitions(&state)
                .into_iter()
                .choose(&mut rng)
                .unwrap();

            match transition.result_action() {
                Action::Silent => println!("τ"),
                Action::Labeled(labeled) => println!(
                    "{} {:?}",
                    labeled.label(&explorer.network).unwrap(),
                    labeled.arguments()
                ),
            }

            let destinations = explorer.destinations(&state, &transition);

            let threshold: f64 = rng.gen();
            let mut accumulated = 0.0;

            for destination in destinations {
                accumulated += destination.probability();
                if accumulated >= threshold {
                    state = explorer.successor(&state, &transition, &destination);
                    break;
                }
            }
        }
    }
}