forked from StudioInfinity/dyna3
b117c00992
Before, a `ConstraintProblem` only specified the indices of the frozen entries. During realization, the frozen entries kept whatever values they had in the initial guess. This commit adds the values of the frozen entries to the `frozen` field of `ConstraintProblem`. The frozen entries of the guess are set to the desired values at the beginning of realization. This commit also improves the `PartialMatrix` structure, which is used to specify the indices and values of the frozen entries.
31 lines
No EOL
1.2 KiB
Rust
31 lines
No EOL
1.2 KiB
Rust
use dyna3::engine::{Q, point, realize_gram, sphere, ConstraintProblem};
|
|
|
|
fn main() {
|
|
let mut problem = ConstraintProblem::from_guess(&[
|
|
point(0.0, 0.0, 2.0),
|
|
sphere(0.0, 0.0, 0.0, 1.0)
|
|
]);
|
|
for j in 0..2 {
|
|
for k in j..2 {
|
|
problem.gram.push_sym(j, k, if (j, k) == (1, 1) { 1.0 } else { 0.0 });
|
|
}
|
|
}
|
|
problem.frozen.push(3, 0, problem.guess[(3, 0)]);
|
|
println!();
|
|
let (config, _, success, history) = realize_gram(
|
|
&problem, 1.0e-12, 0.5, 0.9, 1.1, 200, 110
|
|
);
|
|
print!("\nCompleted Gram matrix:{}", config.tr_mul(&*Q) * &config);
|
|
print!("Configuration:{}", config);
|
|
if success {
|
|
println!("Target accuracy achieved!");
|
|
} else {
|
|
println!("Failed to reach target accuracy");
|
|
}
|
|
println!("Steps: {}", history.scaled_loss.len() - 1);
|
|
println!("Loss: {}", history.scaled_loss.last().unwrap());
|
|
println!("\nStep │ Loss\n─────┼────────────────────────────────");
|
|
for (step, scaled_loss) in history.scaled_loss.into_iter().enumerate() {
|
|
println!("{:<4} │ {}", step, scaled_loss);
|
|
}
|
|
} |