first commit

libraries/FastLED/tests/test_screenmap.cpp

@@ -0,0 +1,159 @@
+
+// g++ --std=c++11 test.cpp
+
+#include "test.h"
+
+#include "test.h"
+#include "fl/screenmap.h"
+
+#include "fl/namespace.h"
+FASTLED_USING_NAMESPACE
+
+using fl::Str;
+
+TEST_CASE("ScreenMap basic functionality") {
+    // Create a screen map for 3 LEDs
+    ScreenMap map(3);
+    
+    // Set some x,y coordinates
+    map.set(0, {1.0f, 2.0f});
+    map.set(1, {3.0f, 4.0f});
+    map.set(2, {5.0f, 6.0f});
+    
+    // Test coordinate retrieval
+    CHECK(map[0].x == 1.0f);
+    CHECK(map[0].y == 2.0f);
+    CHECK(map[1].x == 3.0f);
+    CHECK(map[1].y == 4.0f);
+    CHECK(map[2].x == 5.0f);
+    CHECK(map[2].y == 6.0f);
+    
+    // Test length
+    CHECK(map.getLength() == 3);
+    
+    // Test diameter (default should be -1.0)
+    CHECK(map.getDiameter() == -1.0f);
+    
+    // Test mapToIndex (should give same results as operator[])
+    auto coords = map.mapToIndex(1);
+    CHECK(coords.x == 3.0f);
+    CHECK(coords.y == 4.0f);
+}
+
+TEST_CASE("ScreenMap JSON parsing") {
+    const char* json = R"({
+        "map": {
+            "strip1": {
+                "x": [10.5, 30.5, 50.5],
+                "y": [20.5, 40.5, 60.5],
+                "diameter": 2.5
+            },
+            "strip2": {
+                "x": [15.0, 35.0],
+                "y": [25.0, 45.0],
+                "diameter": 1.5
+            }
+            
+        }
+    })";
+
+    fl::FixedMap<Str, ScreenMap, 16> segmentMaps;
+    ScreenMap::ParseJson(json, &segmentMaps);
+
+    ScreenMap& strip1 = segmentMaps["strip1"];
+    ScreenMap& strip2 = segmentMaps["strip2"];
+
+    // Check first strip
+
+    CHECK(strip1.getLength() == 3);
+    CHECK(strip1.getDiameter() == 2.5f);
+    CHECK(strip1[0].x == 10.5f);
+    CHECK(strip1[0].y == 20.5f);
+    CHECK(strip1[1].x == 30.5f);
+    CHECK(strip1[1].y == 40.5f);
+
+    // Check second strip
+    CHECK(strip2.getLength() == 2);
+    CHECK(strip2.getDiameter() == 1.5f);
+    CHECK(strip2[0].x == 15.0f);
+    CHECK(strip2[0].y == 25.0f);
+    CHECK(strip2[1].x == 35.0f);
+    CHECK(strip2[1].y == 45.0f);
+}
+
+TEST_CASE("ScreenMap multiple strips JSON serialization") {
+    // Create a map with multiple strips
+    fl::FixedMap<Str, ScreenMap, 16> originalMaps;
+    
+    // First strip
+    ScreenMap strip1(2, 2.0f);
+    strip1.set(0, {1.0f, 2.0f});
+    strip1.set(1, {3.0f, 4.0f});
+    originalMaps.insert("strip1", strip1);
+    
+    // Second strip
+    ScreenMap strip2(3, 1.5f);
+    strip2.set(0, {10.0f, 20.0f});
+    strip2.set(1, {30.0f, 40.0f});
+    strip2.set(2, {50.0f, 60.0f});
+    originalMaps.insert("strip2", strip2);
+
+    // Serialize to JSON string
+    Str jsonStr;
+    ScreenMap::toJsonStr(originalMaps, &jsonStr);
+
+    // Deserialize back to a new map
+    fl::FixedMap<Str, ScreenMap, 16> deserializedMaps;
+    ScreenMap::ParseJson(jsonStr.c_str(), &deserializedMaps);
+
+    // Verify first strip
+    ScreenMap& deserializedStrip1 = deserializedMaps["strip1"];
+    CHECK(deserializedStrip1.getLength() == 2);
+    CHECK(deserializedStrip1.getDiameter() == 2.0f);
+    CHECK(deserializedStrip1[0].x == 1.0f);
+    CHECK(deserializedStrip1[0].y == 2.0f);
+    CHECK(deserializedStrip1[1].x == 3.0f);
+    CHECK(deserializedStrip1[1].y == 4.0f);
+
+    // Verify second strip
+    ScreenMap& deserializedStrip2 = deserializedMaps["strip2"];
+    CHECK(deserializedStrip2.getLength() == 3);
+    CHECK(deserializedStrip2.getDiameter() == 1.5f);
+    CHECK(deserializedStrip2[0].x == 10.0f);
+    CHECK(deserializedStrip2[0].y == 20.0f);
+    CHECK(deserializedStrip2[1].x == 30.0f);
+    CHECK(deserializedStrip2[1].y == 40.0f);
+    CHECK(deserializedStrip2[2].x == 50.0f);
+    CHECK(deserializedStrip2[2].y == 60.0f);
+}
+
+TEST_CASE("ScreenMap getBounds functionality") {
+    // Create a screen map with points at different coordinates
+    ScreenMap map(4);
+    map.set(0, {1.0f, 2.0f});
+    map.set(1, {-3.0f, 4.0f});
+    map.set(2, {5.0f, -6.0f});
+    map.set(3, {-2.0f, -1.0f});
+    
+    // Get the bounds
+    vec2f bounds = map.getBounds();
+    
+    // The bounds should be the difference between max and min values
+    // Max X: 5.0, Min X: -3.0 => Width = 8.0
+    // Max Y: 4.0, Min Y: -6.0 => Height = 10.0
+    CHECK(bounds.x == 8.0f);
+    CHECK(bounds.y == 10.0f);
+    
+    // Test with a single point
+    ScreenMap singlePoint(1);
+    singlePoint.set(0, {3.5f, 4.5f});
+    vec2f singleBounds = singlePoint.getBounds();
+    CHECK(singleBounds.x == 0.0f);
+    CHECK(singleBounds.y == 0.0f);
+    
+    // Test with an empty map
+    ScreenMap emptyMap(0);
+    vec2f emptyBounds = emptyMap.getBounds();
+    CHECK(emptyBounds.x == 0.0f);
+    CHECK(emptyBounds.y == 0.0f);
+}