This commit is contained in:
2018-10-24 11:13:06 +02:00
parent be74c1eac7
commit 61f8a9ccce
3 changed files with 25 additions and 11 deletions

View File

@@ -58,7 +58,7 @@ impl Variable {
self.value = value;
}
pub fn get_value(&self) -> Gaussian {
pub fn get_value(&self) -> Gaussian {
self.value
}
@@ -101,7 +101,10 @@ impl PriorFactor {
}
pub fn start(&self, variable_arena: &mut VariableArena) {
variable_arena.get_mut(self.variable).unwrap().update_value(self.id, self.gaussian);
variable_arena
.get_mut(self.variable)
.unwrap()
.update_value(self.id, self.gaussian);
}
}
@@ -146,7 +149,10 @@ impl LikelihoodFactor {
let gaussian = Gaussian::from_pi_tau(a * (x.pi() - fx.pi()), a * (x.tau() - fx.tau()));
variable_arena.get_mut(self.mean).unwrap().update_message(self.id, gaussian);
variable_arena
.get_mut(self.mean)
.unwrap()
.update_message(self.id, gaussian);
}
pub fn update_value(&self, variable_arena: &mut VariableArena) {
@@ -164,7 +170,10 @@ impl LikelihoodFactor {
let gaussian = Gaussian::from_pi_tau(a * (y.pi() - fy.pi()), a * (y.tau() - fy.tau()));
variable_arena.get_mut(self.value).unwrap().update_message(self.id, gaussian);
variable_arena
.get_mut(self.value)
.unwrap()
.update_message(self.id, gaussian);
}
}
@@ -224,7 +233,10 @@ impl SumFactor {
let gaussian = Gaussian::from_pi_tau(new_pi, new_tau);
variable_arena.get_mut(variable).unwrap().update_message(self.id, gaussian);
variable_arena
.get_mut(variable)
.unwrap()
.update_message(self.id, gaussian);
}
pub fn update_sum(&self, variable_arena: &mut VariableArena) {
@@ -372,6 +384,9 @@ impl TruncateFactor {
let gaussian = Gaussian::from_pi_tau(c / m_w, (d + sqrt_c * v) / m_w);
variable_arena.get_mut(self.variable).unwrap().update_value(self.id, gaussian);
variable_arena
.get_mut(self.variable)
.unwrap()
.update_value(self.id, gaussian);
}
}