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

use crate::semirings::SerializableSemiring;
use crate::{Label, StateId};

/// Structure representing a transition from a state to another state in a FST.
#[derive(Debug, Clone, PartialEq, PartialOrd, Eq, Hash)]
pub struct Tr<W> {
    /// Input label.
    pub ilabel: Label,
    /// Output label.
    pub olabel: Label,
    /// Weight.
    pub weight: W,
    /// ID of the next state.
    pub nextstate: StateId,
}

impl<W> Tr<W> {
    /// Creates a new Tr.
    ///
    /// # Example
    ///
    /// ```
    /// # use rustfst::Tr;
    /// # use rustfst::semirings::{TropicalWeight, Semiring};
    /// let transition = Tr::<TropicalWeight>::new(0, 1, 1.3, 2);
    ///
    /// assert_eq!(transition.ilabel, 0);
    /// assert_eq!(transition.olabel, 1);
    /// assert_eq!(transition.weight, TropicalWeight::new(1.3));
    /// assert_eq!(transition.nextstate, 2);
    ///
    /// ```
    pub fn new<S: Into<W>>(ilabel: Label, olabel: Label, weight: S, nextstate: StateId) -> Self {
        Tr {
            ilabel,
            olabel,
            weight: weight.into(),
            nextstate,
        }
    }

    /// Updates the values of the attributes of the Tr from another Tr.
    ///
    /// # Example
    ///
    /// ```
    /// # use rustfst::Tr;
    /// # use rustfst::semirings::{Semiring, TropicalWeight};
    /// let mut tr_1 = Tr::<TropicalWeight>::new(0, 1, 1.3, 2);
    /// let tr_2 = Tr::new(1, 2, 1.2, 3);
    ///
    /// tr_1.set_value(&tr_2);
    ///
    /// assert_eq!(tr_1, tr_2);
    /// ```
    #[inline]
    pub fn set_value(&mut self, tr: &Tr<W>)
    where
        W: std::clone::Clone,
    {
        self.ilabel = tr.ilabel;
        self.olabel = tr.olabel;
        self.weight = tr.weight.clone();
        self.nextstate = tr.nextstate;
    }
}

impl<W: SerializableSemiring> Tr<W> {
    pub fn tr_type() -> String {
        let weight_type = W::weight_type();
        if weight_type.as_str() == "tropical" {
            "standard".to_string()
        } else {
            weight_type
        }
    }
}