cled_controller.h

Go to the documentation of this file.

#pragma once
 

 
#include "FastLED.h"
#include "led_sysdefs.h"
#include "pixeltypes.h"
#include "color.h"
 
#include "fl/force_inline.h"
#include "fl/unused.h"
#include "pixel_controller.h"
#include "dither_mode.h"
#include "pixel_iterator.h"
#include "fl/engine_events.h"
#include "fl/screenmap.h"
#include "fl/virtual_if_not_avr.h"
#include "fl/int.h"
#include "fl/bit_cast.h"
 
FASTLED_NAMESPACE_BEGIN
 

//
// LED Controller interface definition
//

class CLEDController {
protected:
    friend class CFastLED;
    CRGB *m_Data;              
    CLEDController *m_pNext;   
    CRGB m_ColorCorrection;    
    CRGB m_ColorTemperature;   
    EDitherMode m_DitherMode;  
    bool m_enabled = true;
    int m_nLeds;               
    static CLEDController *m_pHead;  
    static CLEDController *m_pTail;  
 
public:

    virtual void showColor(const CRGB & data, int nLeds, fl::u8 brightness) = 0;

    virtual void show(const struct CRGB *data, int nLeds, fl::u8 brightness) = 0;
 
 
    Rgbw mRgbMode = RgbwInvalid::value();
    CLEDController& setRgbw(const Rgbw& arg = RgbwDefault::value()) {
        // Note that at this time (Sept 13th, 2024) this is only implemented in the ESP32 driver
        // directly. For an emulated version please see RGBWEmulatedController in chipsets.h
        mRgbMode = arg;
        return *this;  // builder pattern.
    }
 
    void setEnabled(bool enabled) { m_enabled = enabled; }
    bool getEnabled() { return m_enabled; }
 
    CLEDController();
    // If we added virtual to the AVR boards then we are going to add 600 bytes of memory to the binary
    // flash size. This is because the virtual destructor pulls in malloc and free, which are the largest
    // Testing shows that this virtual destructor adds a 600 bytes to the binary on
    // attiny85 and about 1k for the teensy 4.X series.
    // Attiny85:
    //   With CLEDController destructor virtual: 11018 bytes to binary.
    //   Without CLEDController destructor virtual: 10666 bytes to binary.
    VIRTUAL_IF_NOT_AVR ~CLEDController();
 
    Rgbw getRgbw() const { return mRgbMode; }

    virtual void init() = 0;

    VIRTUAL_IF_NOT_AVR void clearLeds(int nLeds = -1) {
        clearLedDataInternal(nLeds);
        showLeds(0);
    }
 
    // Compatibility with the 3.8.x codebase.
    VIRTUAL_IF_NOT_AVR void showLeds(fl::u8 brightness) {
        void* data = beginShowLeds(m_nLeds);
        showLedsInternal(brightness);
        endShowLeds(data);
    }
 
    ColorAdjustment getAdjustmentData(fl::u8 brightness);

    void showInternal(const struct CRGB *data, int nLeds, fl::u8 brightness) {
        if (m_enabled) {
           show(data, nLeds,brightness);
        }
    }

    void showColorInternal(const struct CRGB &data, int nLeds, fl::u8 brightness) {
        if (m_enabled) {
            showColor(data, nLeds, brightness);
        }
    }

    void showLedsInternal(fl::u8 brightness) {
        if (m_enabled) {
            show(m_Data, m_nLeds, brightness);
        }
    }

    void showColorInternal(const struct CRGB & data, fl::u8 brightness) {
        if (m_enabled) {
            showColor(data, m_nLeds, brightness);
        }
    }

    static CLEDController *head() { return m_pHead; }

    CLEDController *next() { return m_pNext; }

    CLEDController & setLeds(CRGB *data, int nLeds) {
        m_Data = data;
        m_nLeds = nLeds;
        return *this;
    }

    void clearLedDataInternal(int nLeds = -1);

    virtual int size() { return m_nLeds; }

    virtual int lanes() { return 1; }

    CRGB* leds() { return m_Data; }

    CRGB &operator[](int x) { return m_Data[x]; }

    inline CLEDController & setDither(fl::u8 ditherMode = BINARY_DITHER) { m_DitherMode = ditherMode; return *this; }
 
    CLEDController& setScreenMap(const fl::XYMap& map, float diameter = -1.f) {
        // EngineEvents::onCanvasUiSet(this, map);
        fl::ScreenMap screenmap = map.toScreenMap();
        if (diameter <= 0.0f) {
            // screen map was not set.
            if (map.getTotal() <= (64*64)) {
                screenmap.setDiameter(.1f); // Assume small matrix is being used.
            }
        }
        fl::EngineEvents::onCanvasUiSet(this, screenmap);
        return *this;
    }
 
    CLEDController& setScreenMap(const fl::ScreenMap& map) {
        fl::EngineEvents::onCanvasUiSet(this, map);
        return *this;
    }
 
    CLEDController& setScreenMap(fl::u16 width, fl::u16 height, float diameter = -1.f) {
        fl::XYMap xymap = fl::XYMap::constructRectangularGrid(width, height);
        return setScreenMap(xymap, diameter);
    }

    inline fl::u8 getDither() { return m_DitherMode; }
 
    virtual void* beginShowLeds(int size) {
        FASTLED_UNUSED(size);
        // By default, emit an integer. This integer will, by default, be passed back.
        // If you override beginShowLeds() then
        // you should also override endShowLeds() to match the return state.
        //
        // For async led controllers this should be used as a sync point to block
        // the caller until the leds from the last draw frame have completed drawing.
        // for each controller:
        //   beginShowLeds();
        // for each controller:
        //   showLeds();
        // for each controller:
        //   endShowLeds();
        uintptr_t d = getDither();
        void* out = fl::int_to_ptr<void>(d);
        return out;
    }
 
    virtual void endShowLeds(void* data) {
        // By default recieves the integer that beginShowLeds() emitted.
        //For async controllers this should be used to signal the controller
        // to begin transmitting the current frame to the leds.
        uintptr_t d = fl::ptr_to_int(data);
        setDither(static_cast<fl::u8>(d));
    }

    CLEDController & setCorrection(CRGB correction) { m_ColorCorrection = correction; return *this; }

    CLEDController & setCorrection(LEDColorCorrection correction) { m_ColorCorrection = correction; return *this; }

    CRGB getCorrection() { return m_ColorCorrection; }

    CLEDController & setTemperature(CRGB temperature) { m_ColorTemperature = temperature; return *this; }

    CLEDController & setTemperature(ColorTemperature temperature) { m_ColorTemperature = temperature; return *this; }

    CRGB getTemperature() { return m_ColorTemperature; }

    CRGB getAdjustment(fl::u8 scale) {
        return CRGB::computeAdjustment(scale, m_ColorCorrection, m_ColorTemperature);
    }

    virtual fl::u16 getMaxRefreshRate() const { return 0; }
};
 
FASTLED_NAMESPACE_END