Skip to content

Commit eb84c3e

Browse files
authored
Merge pull request #137 from aeoa/fix/continuous-rotation-sampling
2 parents 42affd9 + f351631 commit eb84c3e

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

src/sample/uniform_sampler.rs

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@ use jagua_rs::geometry::geo_enums::RotationRange;
44
use jagua_rs::geometry::geo_traits::TransformableFrom;
55
use jagua_rs::geometry::primitives::Rect;
66
use jagua_rs::geometry::{normalize_rotation, DTransformation, Transformation};
7-
use ndarray::Array;
87
use ordered_float::OrderedFloat;
98
use rand::prelude::IndexedRandom;
109
use rand::{Rng, RngExt};
@@ -34,7 +33,10 @@ impl UniformBBoxSampler {
3433
RotationRange::Discrete(r) => r,
3534
RotationRange::Continuous => {
3635
// for continuous rotation, we sample a set of rotations spaced evenly
37-
&Array::linspace(0.0, 2.0 * PI, ROT_N_SAMPLES).to_vec()
36+
let step = (2.0 * PI) / ROT_N_SAMPLES as f32;
37+
&(0..ROT_N_SAMPLES)
38+
.map(|i| i as f32 * step)
39+
.collect_vec()
3840
}
3941
};
4042

0 commit comments

Comments
 (0)