LedStrip.cpp

#include <string.h>
#include "LedStrip.h"


uint16_t LedStrip::stripTotalLen;
uint16_t LedStrip::stripLen[NUMBER_LEDSTRIP];
uint16_t LedStrip::stripStartOffset[NUMBER_LEDSTRIP];
uint8_t LedStrip::stripIndex[NUMBER_LEDSTRIP*MaxLedsPerStrip];

CRGB leds[NUMBER_LEDSTRIP][MaxLedsPerStrip];

LedStrip::LedStrip(uint32_t numPerStrip)
{
  stripTotalLen = numPerStrip;
  int offset=0;
  for (int j=0;j<NUMBER_LEDSTRIP;j++) {
    stripStartOffset[j]=offset;
    for (int i=0;i<stripTotalLen;i++) stripIndex[offset++]=j;
  } 
}

void LedStrip::begin(void) {
  int offset=0;
  for (int j=0;j<NUMBER_LEDSTRIP;j++) {
    stripLen[j]=MaxLedsPerStrip;
    stripStartOffset[j]=offset;
    for (int i=0;i<stripTotalLen;i++) stripIndex[offset++]=j;
  } 
  //6 led strips, hardcoded
  FastLED.addLeds<LED_TYPE, DATA_PIN1, COLOR_ORDER>(leds[0], MaxLedsPerStrip).setCorrection(CRGB(BRIGHTNESS_PIN1,BRIGHTNESS_PIN1,BRIGHTNESS_PIN1) );
  FastLED.addLeds<LED_TYPE, DATA_PIN2, COLOR_ORDER>(leds[1], MaxLedsPerStrip).setCorrection(CRGB(BRIGHTNESS_PIN2,BRIGHTNESS_PIN2,BRIGHTNESS_PIN2) );
  FastLED.addLeds<LED_TYPE, DATA_PIN3, COLOR_ORDER>(leds[2], MaxLedsPerStrip).setCorrection(CRGB(BRIGHTNESS_PIN3,BRIGHTNESS_PIN3,BRIGHTNESS_PIN3) );
  FastLED.addLeds<LED_TYPE, DATA_PIN4, COLOR_ORDER>(leds[3], MaxLedsPerStrip).setCorrection(CRGB(BRIGHTNESS_PIN4,BRIGHTNESS_PIN4,BRIGHTNESS_PIN4) );
  FastLED.addLeds<LED_TYPE, DATA_PIN5, COLOR_ORDER>(leds[4], MaxLedsPerStrip).setCorrection(CRGB(BRIGHTNESS_PIN5,BRIGHTNESS_PIN5,BRIGHTNESS_PIN5) );
  FastLED.addLeds<LED_TYPE, DATA_PIN6, COLOR_ORDER>(leds[5], MaxLedsPerStrip).setCorrection(CRGB(BRIGHTNESS_PIN6,BRIGHTNESS_PIN6,BRIGHTNESS_PIN6) );
  FastLED.addLeds<LED_TYPE, DATA_PIN7, COLOR_ORDER>(leds[6], MaxLedsPerStrip).setCorrection(CRGB(BRIGHTNESS_PIN7,BRIGHTNESS_PIN7,BRIGHTNESS_PIN7) );
#if (NUMBER_LEDSTRIP == 8)
  FastLED.addLeds<LED_TYPE, DATA_PIN8, COLOR_ORDER>(leds[7], MaxLedsPerStrip).setCorrection(CRGB(BRIGHTNESS_PIN8,BRIGHTNESS_PIN8,BRIGHTNESS_PIN8) );
#endif

  //FastLED.setBrightness( BRIGHTNESS );
  FastLED.setDither( 0 );
}

void LedStrip::addNewStrip(uint8_t index,uint16_t length) {
  if ((index>=0)&&(index<NUMBER_LEDSTRIP)) {
    if (length>=MaxLedsPerStrip) length=MaxLedsPerStrip-1;
    if (!length) length=1; //to confirm if really required or not
    stripLen[index]=length;
    FastLED[index].setLeds(leds[index],length);
  }
}

void LedStrip::setStripBrightness(uint8_t index, uint8_t brightness){
  if ((index>=0)&&(index<NUMBER_LEDSTRIP)) {
    FastLED[index].setCorrection(CRGB(brightness,brightness,brightness));
  }
}

void LedStrip::show(void)
{
  FastLED.show();
}

void LedStrip::setStripLength(uint16_t length)
{
  stripTotalLen = length;
  int offset=0;
  for (int j=0;j<NUMBER_LEDSTRIP;j++) {
    stripStartOffset[j]=offset;
    for (int i=0;i<stripTotalLen;i++) stripIndex[offset++]=j;
  }
}

void LedStrip::clearAll() {
  FastLED.clear();
}

void LedStrip::reset() {
  for(int i = 0; i < NUMBER_LEDSTRIP; i++){
    FastLED[i].setLeds(leds[i],0);
  }
}

void LedStrip::setPixels(uint32_t start_num,uint16_t len,int color)
{
  uint32_t strip, offset;
  char r=((color >> 16) & 0xFF);
  char g=((color >> 8) & 0xFF);
  char b=(color & 0xFF);

  strip = stripIndex[start_num];//start_num / stripLen;
  offset = start_num-stripStartOffset[strip];// % stripLen;

  if (offset+len>stripLen[strip]) return;
  
  for (uint16_t i=len;i;i--) {
    leds[strip][offset].b = b;
    leds[strip][offset].g = g;
    leds[strip][offset].r = r;
    offset++;
    /*if (offset==stripLen) { //should never happen
      offset=0;
      strip++;
    }*/
  }
}

void LedStrip::setPixel(uint32_t num, int color)
{
  uint32_t strip, offset;

  strip = stripIndex[num];//start_num / stripLen;
  offset = num-stripStartOffset[strip];// % stripLen;

  if (offset>=stripLen[strip]) return;

  leds[strip][offset].b = (color & 0xFF); // Take just the lowest 8 bits.
  leds[strip][offset].g = ((color >> 8) & 0xFF); // Shift the integer right 8 bits.
  leds[strip][offset].r = ((color >> 16) & 0xFF); // Shift the integer right 8 bits.
}

CRGB blackpixel(0);

CRGB& LedStrip::getPixel(uint32_t num)
{
  uint32_t strip, offset;

  strip = stripIndex[num];//start_num / stripLen;
  offset = num-stripStartOffset[strip];// % stripLen;

  if (offset>=stripLen[strip]) return blackpixel;
  
  return leds[strip][offset];
}