mirror of
https://github.com/wpilibsuite/allwpilib
synced 2026-06-22 01:11:42 +00:00
[wpimath] Add TimeInterpolatableBuffer (#2695)
These classes are useful for storing previous robot positions to use in conjunction with the upcoming pose estimators. Co-authored-by: Prateek Machiraju <prateek.machiraju@gmail.com> Co-authored-by: Tyler Veness <calcmogul@gmail.com> Co-authored-by: cttew <cttewari@gmail.com>
This commit is contained in:
@@ -0,0 +1,43 @@
|
||||
// Copyright (c) FIRST and other WPILib contributors.
|
||||
// Open Source Software; you can modify and/or share it under the terms of
|
||||
// the WPILib BSD license file in the root directory of this project.
|
||||
|
||||
package edu.wpi.first.math.interpolation;
|
||||
|
||||
import static org.junit.jupiter.api.Assertions.assertEquals;
|
||||
|
||||
import edu.wpi.first.math.geometry.Pose2d;
|
||||
import edu.wpi.first.math.geometry.Rotation2d;
|
||||
import org.junit.jupiter.api.Test;
|
||||
|
||||
public class TimeInterpolatableBufferTest {
|
||||
@Test
|
||||
public void testInterpolation() {
|
||||
TimeInterpolatableBuffer<Rotation2d> buffer = TimeInterpolatableBuffer.createBuffer(10);
|
||||
|
||||
buffer.addSample(0, new Rotation2d());
|
||||
assertEquals(0, buffer.getSample(0).getRadians(), 0.001);
|
||||
buffer.addSample(1, new Rotation2d(1));
|
||||
assertEquals(0.5, buffer.getSample(0.5).getRadians(), 0.001);
|
||||
assertEquals(1.0, buffer.getSample(1.0).getRadians(), 0.001);
|
||||
buffer.addSample(3, new Rotation2d(2));
|
||||
assertEquals(1.5, buffer.getSample(2).getRadians(), 0.001);
|
||||
|
||||
buffer.addSample(10.5, new Rotation2d(2));
|
||||
assertEquals(new Rotation2d(1), buffer.getSample(0));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testPose2d() {
|
||||
TimeInterpolatableBuffer<Pose2d> buffer = TimeInterpolatableBuffer.createBuffer(10);
|
||||
|
||||
// We expect to be at (1 - 1/Math.sqrt(2), 1/Math.sqrt(2), 45deg) at t=0.5
|
||||
buffer.addSample(0, new Pose2d(0, 0, Rotation2d.fromDegrees(90)));
|
||||
buffer.addSample(1, new Pose2d(1, 1, Rotation2d.fromDegrees(0)));
|
||||
Pose2d sample = buffer.getSample(0.5);
|
||||
|
||||
assertEquals(1 - 1 / Math.sqrt(2), sample.getTranslation().getX(), 0.01);
|
||||
assertEquals(1 / Math.sqrt(2), sample.getTranslation().getY(), 0.01);
|
||||
assertEquals(45, sample.getRotation().getDegrees(), 0.01);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user