ucs7604.h

Go to the documentation of this file.

#pragma once
 
#ifndef __INC_UCS7604_H
#define __INC_UCS7604_H
 
#include "pixel_controller.h"
#include "cpixel_ledcontroller.h"
#include "fl/chipsets/led_timing.h"
#include "fl/chipsets/encoders/pixel_iterator.h"
#include "fl/chipsets/encoders/ucs7604.h"
#include "fl/stl/bit_cast.h"
#include "fl/stl/vector.h"
#include "fl/math/ease.h"
#include "fl/stl/iterator.h"
#include "fl/stl/noexcept.h"

 
#ifndef FL_UCS7604_BRIGHTNESS
#define FL_UCS7604_BRIGHTNESS 0x0F
#endif
 
namespace fl {

namespace ucs7604 {
    using CurrentControl = UCS7604CurrentControl;

    void set_brightness(CurrentControl current) FL_NO_EXCEPT;

    inline void set_brightness(u8 r, u8 g, u8 b, u8 w) FL_NO_EXCEPT {
        set_brightness(CurrentControl(r, g, b, w));
    }

    CurrentControl brightness() FL_NO_EXCEPT;
 
}  // namespace ucs7604

template <
    int DATA_PIN,
    EOrder RGB_ORDER,  // Color order for input pixels (converted to RGB internally)
    fl::UCS7604Mode MODE,
    typename CHIPSET_TIMING,
    template<int, typename, EOrder> class CLOCKLESS_CONTROLLER
>
class UCS7604ControllerT : public CPixelLEDController<RGB_ORDER>
{
private:
    using DelegateControllerBase = CLOCKLESS_CONTROLLER<DATA_PIN, CHIPSET_TIMING, RGB>;
 
    // Helper class to access protected methods of the delegate controller
    class DelegateController : public DelegateControllerBase {
        friend class UCS7604ControllerT<DATA_PIN, RGB_ORDER, MODE, CHIPSET_TIMING, CLOCKLESS_CONTROLLER>;
        void callShowPixels(PixelController<RGB> & pixels) FL_NO_EXCEPT {
            DelegateControllerBase::showPixels(pixels);
        }
    };
 
    DelegateController mDelegate;  // Clockless controller for wire transmission (always RGB)
 
    // Reusable byte buffer (uses PSRAM on ESP32, regular heap elsewhere)
    // Cleared each frame but memory is reused (no reallocation after first use)
    float mGamma = 2.8f;
    fl::vector_psram<u8> mByteBuffer;
 
public:
    UCS7604ControllerT() FL_NO_EXCEPT {}
 
    void setGamma(float gamma) FL_NO_EXCEPT { mGamma = gamma; }
    void clearGamma() FL_NO_EXCEPT { mGamma = 2.8f; }
 
    virtual void init() FL_NO_EXCEPT override {
        mDelegate.init();
        // UCS7604 is always RGBW — override any user setting.
        this->setRgbw(RgbwDefault::value());
    }
 
    // Access delegate controller (for testing)
    const DelegateController& getDelegate() const FL_NO_EXCEPT {
        return mDelegate;
    }
 
    DelegateController& getDelegate() FL_NO_EXCEPT {
        return mDelegate;
    }
 
protected:
 
    fl::span<const u8> bytes() const FL_NO_EXCEPT {
        return mByteBuffer;
    }
 
    virtual void showPixels(PixelController<RGB_ORDER> &pixels) FL_NO_EXCEPT override {
        if (pixels.size() == 0) {
            return;
        }
 
        // Get current control values (recalculated each frame)
        fl::ucs7604::CurrentControl current = fl::ucs7604::brightness();
 
        // Reorder RGB current values to match the color order (RGB_ORDER template parameter)
        // The current control values are semantic (current.r controls RED LEDs, etc.)
        // but need to be sent in wire order matching the pixel data reordering
        u8 rgb_currents[3] = {current.r, current.g, current.b};
 
        // Extract channel positions from RGB_ORDER (octal encoding: 0=R, 1=G, 2=B)
        // Octal digits are read right-to-left in bit positions but left-to-right semantically
        // RGB (012 octal): wire position 0=R(0), 1=G(1), 2=B(2)
        // GRB (102 octal): wire position 0=G(1), 1=R(0), 2=B(2)
        u8 pos0 = (static_cast<int>(RGB_ORDER) >> 6) & 0x3;  // Wire position 0 (leftmost octal digit)
        u8 pos1 = (static_cast<int>(RGB_ORDER) >> 3) & 0x3;  // Wire position 1 (middle octal digit)
        u8 pos2 = (static_cast<int>(RGB_ORDER) >> 0) & 0x3;  // Wire position 2 (rightmost octal digit)
 
        // Reorder: wire R gets current for channel at pos0, etc.
        u8 r_current = rgb_currents[pos0];  // Wire R (position 0)
        u8 g_current = rgb_currents[pos1];  // Wire G (position 1)
        u8 b_current = rgb_currents[pos2];  // Wire B (position 2)
        u8 w_current = current.w;  // W always in position 3
 
        // Create current control struct with reordered values
        fl::UCS7604CurrentControl wire_current(r_current, g_current, b_current, w_current);
 
        // Get gamma LUT: per-controller override or default 2.8
        float gamma = mGamma;
        fl::shared_ptr<const Gamma8> gamma8 = Gamma8::getOrCreate(gamma);
 
        // UCS7604 is always RGBW — override any user setting.
        fl::Rgbw rgbw = RgbwDefault::value();
        fl::PixelIterator pixel_iter = pixels.as_iterator(rgbw);
 
        // Clear buffer and use encoder to fill it
        mByteBuffer.clear();
        fl::encodeUCS7604(pixel_iter, pixels.size(), fl::back_inserter(mByteBuffer),
                          MODE, wire_current, pixel_iter.get_rgbw().active(), gamma8.get());
 
        // Reinterpret byte buffer as CRGB pixels (encoder ensures divisible by 3)
        size_t num_pixels = mByteBuffer.size() / 3;
        CRGB* fake_pixels = fl::bit_cast<CRGB*>(mByteBuffer.data());
 
        // Construct PixelController and send to delegate controller
        PixelController<RGB> pixel_data(fake_pixels, num_pixels, ColorAdjustment::noAdjustment(), DISABLE_DITHER);
        mDelegate.callShowPixels(pixel_data);
    }
};
 
// now typedef the controllers
template <int DATA_PIN, EOrder RGB_ORDER, template<int, typename, EOrder> class CLOCKLESS_CONTROLLER>
using UCS7604Controller8bitT = UCS7604ControllerT<DATA_PIN, RGB_ORDER, fl::UCS7604Mode::UCS7604_MODE_8BIT_800KHZ, fl::TIMING_UCS7604_800KHZ, CLOCKLESS_CONTROLLER>;
 
template <int DATA_PIN, EOrder RGB_ORDER, template<int, typename, EOrder> class CLOCKLESS_CONTROLLER>
using UCS7604Controller16bitT = UCS7604ControllerT<DATA_PIN, RGB_ORDER, fl::UCS7604Mode::UCS7604_MODE_16BIT_800KHZ, fl::TIMING_UCS7604_800KHZ, CLOCKLESS_CONTROLLER>;
 
template <int DATA_PIN, EOrder RGB_ORDER, template<int, typename, EOrder> class CLOCKLESS_CONTROLLER>
using UCS7604Controller16bit1600T = UCS7604ControllerT<DATA_PIN, RGB_ORDER, fl::UCS7604Mode::UCS7604_MODE_16BIT_1600KHZ, fl::TIMING_UCS7604_1600KHZ, CLOCKLESS_CONTROLLER>;
 
}  // namespace fl
 
#endif // __INC_UCS7604_H