120 lines
4.5 KiB
C++
120 lines
4.5 KiB
C++
// g++ --std=c++11 test.cpp
|
|
|
|
#include "fl/math_macros.h"
|
|
#include "test.h"
|
|
#include "fl/algorithm.h"
|
|
|
|
#include "fl/sstream.h"
|
|
|
|
#include "fl/corkscrew.h"
|
|
#include "fl/grid.h"
|
|
#include "fl/tile2x2.h" // Ensure this header is included for Tile2x2_u8
|
|
|
|
#define NUM_LEDS 288
|
|
|
|
#define TWO_PI (PI * 2.0)
|
|
|
|
using namespace fl;
|
|
|
|
TEST_CASE("Corkscrew Circle10 test") {
|
|
Corkscrew::Input input;
|
|
input.totalLength = 10.0f; // Total length of the corkscrew in centimeters
|
|
input.totalHeight = 0.0f;
|
|
input.totalTurns = 1.0f;
|
|
input.offsetCircumference = 0.0f; // No offset
|
|
input.numLeds = 10; // Default to dense 144 LEDs times two strips
|
|
Corkscrew::State output = Corkscrew::generateState(input);
|
|
fl::vector<vec2f> expected_values;
|
|
expected_values.push_back(vec2f(0.0f, 0.0f)); // First LED at the bottom
|
|
expected_values.push_back(vec2f(1.0f, 0.0f)); // Second LED in the middle
|
|
expected_values.push_back(vec2f(2.0f, 0.0f)); // Third LED at the top
|
|
expected_values.push_back(vec2f(3.0f, 0.0f)); // Fourth LED at the top
|
|
expected_values.push_back(vec2f(4.0f, 0.0f)); // Fifth LED at the top
|
|
expected_values.push_back(vec2f(5.0f, 0.0f)); // Sixth LED at the top
|
|
expected_values.push_back(vec2f(6.0f, 0.0f)); // Seventh LED at the top
|
|
expected_values.push_back(vec2f(7.0f, 0.0f)); // Eighth LED at the top
|
|
expected_values.push_back(vec2f(8.0f, 0.0f)); // Ninth LED at the top
|
|
expected_values.push_back(vec2f(9.0f, 0.0f)); // Tenth LED at the top
|
|
|
|
REQUIRE_EQ(output.width, 10);
|
|
REQUIRE_EQ(output.height, 1);
|
|
|
|
}
|
|
|
|
TEST_CASE("Tile2x2_u8_wrap wrap-around test with width and height") {
|
|
// Initialize a Tile2x2_u8 with known values and set origin beyond boundaries
|
|
Tile2x2_u8 originalTile;
|
|
originalTile.setOrigin(3, 3); // Set the origin beyond the width and height
|
|
originalTile.at(0, 0) = 1;
|
|
originalTile.at(0, 1) = 2;
|
|
originalTile.at(1, 0) = 3;
|
|
originalTile.at(1, 1) = 4;
|
|
|
|
// Convert to Tile2x2_u8_wrap with given width and height
|
|
uint16_t width = 2;
|
|
uint16_t height = 2;
|
|
Tile2x2_u8_wrap cycTile(originalTile, width, height);
|
|
|
|
// Verify that the conversion wraps around correctly
|
|
REQUIRE_EQ(cycTile.at(0, 0).first.x, 1); // Wraps around to (1, 1)
|
|
REQUIRE_EQ(cycTile.at(0, 0).first.y, 1);
|
|
REQUIRE_EQ(cycTile.at(0, 1).first.x, 1); // Wraps around to (1, 0)
|
|
REQUIRE_EQ(cycTile.at(0, 1).first.y, 0);
|
|
REQUIRE_EQ(cycTile.at(1, 0).first.x, 0); // Wraps around to (0, 1)
|
|
REQUIRE_EQ(cycTile.at(1, 0).first.y, 1);
|
|
REQUIRE_EQ(cycTile.at(1, 1).first.x, 0); // Wraps around to (0, 0)
|
|
REQUIRE_EQ(cycTile.at(1, 1).first.y, 0);
|
|
|
|
// Verify that the values are correct
|
|
REQUIRE_EQ(cycTile.at(0, 0).second, 1);
|
|
REQUIRE_EQ(cycTile.at(0, 1).second, 2);
|
|
REQUIRE_EQ(cycTile.at(1, 0).second, 3);
|
|
REQUIRE_EQ(cycTile.at(1, 1).second, 4);
|
|
}
|
|
|
|
TEST_CASE("Tile2x2_u8_wrap conversion with width and height") {
|
|
// Initialize a Tile2x2_u8 with known values
|
|
Tile2x2_u8 originalTile;
|
|
originalTile.setOrigin(0, 0); // Set the origin to (0, 0)
|
|
originalTile.at(0, 0) = 1;
|
|
originalTile.at(0, 1) = 2;
|
|
originalTile.at(1, 0) = 3;
|
|
originalTile.at(1, 1) = 4;
|
|
|
|
// Convert to Tile2x2_u8_wrap with given width and height
|
|
uint16_t width = 2;
|
|
uint16_t height = 2;
|
|
Tile2x2_u8_wrap cycTile(originalTile, width, height);
|
|
|
|
// Verify that the conversion is correct
|
|
REQUIRE_EQ(cycTile.at(0, 0).second, 1);
|
|
REQUIRE_EQ(cycTile.at(0, 1).second, 2);
|
|
REQUIRE_EQ(cycTile.at(1, 0).second, 3);
|
|
REQUIRE_EQ(cycTile.at(1, 1).second, 4);
|
|
}
|
|
|
|
TEST_CASE("Tile2x2_u8_wrap conversion test") {
|
|
// Initialize a Tile2x2_u8 with known values and a specific origin
|
|
Tile2x2_u8 originalTile;
|
|
originalTile.setOrigin(50, 50); // Set the origin to (50, 50)
|
|
originalTile.at(0, 0) = 1; // Initialize the missing element
|
|
originalTile.at(0, 1) = 2;
|
|
originalTile.at(1, 0) = 3;
|
|
originalTile.at(1, 1) = 4;
|
|
|
|
// Convert to Tile2x2_u8_wrap with a given width
|
|
uint16_t width = 10;
|
|
Tile2x2_u8_wrap cycTile(originalTile, width);
|
|
|
|
// Verify that the conversion is correct
|
|
REQUIRE_EQ(cycTile.at(0, 0).second, 1);
|
|
REQUIRE_EQ(cycTile.at(0, 1).second, 2);
|
|
REQUIRE_EQ(cycTile.at(1, 0).second, 3);
|
|
REQUIRE_EQ(cycTile.at(1, 1).second, 4);
|
|
|
|
// Verify wrap-around behavior on the x-axis
|
|
REQUIRE_EQ(cycTile.at(2, 2).second, 1); // Wraps around to (0, 0)
|
|
REQUIRE_EQ(cycTile.at(2, 3).second, 2); // Wraps around to (0, 1)
|
|
REQUIRE_EQ(cycTile.at(3, 2).second, 3); // Wraps around to (1, 0)
|
|
REQUIRE_EQ(cycTile.at(3, 3).second, 4); // Wraps around to (1, 1)
|
|
}
|