WS2812B Tester

//Author: jbrad2089@gmail.com
//V3: Added Brightness mode and all white mode
//V4: Added Current meter mode

#include <Adafruit_NeoPixel.h>
#ifdef __AVR__
  #include <avr/power.h>
#endif

#include "Display.h"
#include "Button.h"

#define PIN_LED 12
#define PIN_HUNDRED 9
#define PIN_TEN 10
#define PIN_ONE 11
#define PIN_SEG_A 8
#define PIN_SEG_B 2
#define PIN_SEG_C 4
#define PIN_SEG_D 5
#define PIN_SEG_E 6
#define PIN_SEG_F 7
#define PIN_SEG_G 3
#define PIN_MODE 13
#define PIN_DOWN A0
#define PIN_UP A1
#define PIN_VOLT A2
#define PIN_AMP A3

//define modes to set time
enum ModeEnum { COUNT, BRIGHTNESS, PATTERN, CURRENT };
int currentMode;
bool modeButtonPressed = false;

// Parameter 1 = number of pixels in strip
// Parameter 2 = Arduino pin number (most are valid)
// Parameter 3 = pixel type flags, add together as needed:
//   NEO_KHZ800  800 KHz bitstream (most NeoPixel products w/WS2812 LEDs)
//   NEO_KHZ400  400 KHz (classic 'v1' (not v2) FLORA pixels, WS2811 drivers)
//   NEO_GRB     Pixels are wired for GRB bitstream (most NeoPixel products)
//   NEO_RGB     Pixels are wired for RGB bitstream (v1 FLORA pixels, not v2)
//   NEO_RGBW    Pixels are wired for RGBW bitstream (NeoPixel RGBW products)
Adafruit_NeoPixel strip = Adafruit_NeoPixel(1, PIN_LED, NEO_GRB + NEO_KHZ800);
Display display = Display();
Button mode = Button(PIN_MODE);
Button down = Button(PIN_DOWN);
Button up = Button(PIN_UP);

//Storage for current values
int pixels = 1;
int red = 128;
int green = 128;
int blue = 128;
int brightness = 128;
int pattern = 1;
int maxCurrent = 0;

// IMPORTANT: To reduce NeoPixel burnout risk, add 1000 uF capacitor across
// pixel power leads, add 300 - 500 Ohm resistor on first pixel's data input
// and minimize distance between Arduino and first pixel.  Avoid connecting
// on a live circuit...if you must, connect GND first.

void setup() 
{
  Serial.begin(115200);
  
  //Pixel Strip
  Serial.println("Adafruit_NeoPixel");
  strip.begin();
  strip.setBrightness(brightness);
  strip.show(); // Initialize all pixels to 'off'

  //Volt and Amp Meter
  pinMode(PIN_VOLT,INPUT);
  pinMode(PIN_AMP,INPUT);

  //7 Seg display
  Serial.println("Display");
  display.Initialise(PIN_SEG_A, PIN_SEG_B, PIN_SEG_C, PIN_SEG_D, PIN_SEG_E, PIN_SEG_F, PIN_SEG_G, -1, 3, PIN_ONE, PIN_TEN, PIN_HUNDRED);
  currentMode = COUNT;
  display.ShowChar(_C2, 2);
  display.ShowValue(pixels, HEX, 2);

  //Button callbacks
  mode.Background(ButtonBackground);
  down.Repeat(DownButtonPressed);
  down.Background(ButtonBackground);
  up.Repeat(UpButtonPressed);
  up.Background(ButtonBackground);

  //Set ISR for pin change on MODE pin
  Button::PinChangeSetup(PIN_MODE);
}

void loop() 
{
  delay(5);
  display.Multiplex();

  //Handle polling of up and down repeating buttons
  up.Pressed();
  down.Pressed();

  if (mode.Pressed())
  {
    modeButtonPressed = false;
    switch (currentMode)
    {
      case CURRENT:
        currentMode = COUNT;
        clear();
        display.ShowChar(_C2, 2);
        display.ShowValue(pixels, HEX, 2);
        break;
      
      case COUNT:
        currentMode = BRIGHTNESS;
        strip.updateLength(pixels);
        display.ShowChar(_B, 2);
        display.ShowValue(brightness, HEX, 2);
        break;
      
      case BRIGHTNESS:
        strip.setBrightness(brightness);
        currentMode = PATTERN;
        display.ShowChar(_P, 2);
        display.ShowValue(pattern, HEX, 2);
        break;
      
      case PATTERN:
        maxCurrent = 0;
        display.ShowChar(_HYPHEN, 2);
        display.ShowChar(_HYPHEN, 1);
        display.ShowChar(_HYPHEN, 0);
        display.MultiplexAll();
        switch (pattern)
        {
          case 1: colorWipe(strip.Color(255, 0, 0), 50); break; // Red
          case 2: colorWipe(strip.Color(0, 255, 0), 50); break; // Green
          case 3: colorWipe(strip.Color(0, 0, 255), 50); break; // Blue
          case 4: colorWipe(strip.Color(255, 255, 255), 50); break; // White
          case 5: theaterChase(strip.Color(127, 127, 127), 50); break; // White
          case 6: theaterChase(strip.Color(0, 127, 0), 50); break; // Green                                                  
          case 7: rainbow(20); break;
          case 8: rainbowCycle(20); break;
          case 9: theaterChaseRainbow(50); break;
        }
        if (modeButtonPressed && mode.State() == LOW)
        {
          mode.Pressed();
        }

        //Switch to current mode
        currentMode = CURRENT;
        if (maxCurrent > 9999)
        {
          maxCurrent = 9999;
        }
        display.ShowValue(maxCurrent);
        break;
    }
      
  }
}

//Mode button interrupt to break out of loops etc
//PCINT1 handles pin changes for pins for D8 to D13
ISR (PCINT0_vect)
{
  modeButtonPressed = modeButtonPressed | (mode.State() == LOW);
}

void ButtonBackground(void)
{
  display.MultiplexAll();
}

//Invoked when DOWN button is pressed
void DownButtonPressed(void)
{
    switch (currentMode)
    {
      case COUNT:
        pixels = max(pixels - 1, 1); 
        display.ShowValue(pixels, HEX, 2); 
        break;
        
      case BRIGHTNESS:
        brightness = max(brightness - 1, 1); 
        display.ShowValue(brightness, HEX, 2); 
        break;
        
      case PATTERN:
        pattern = max(pattern - 1, 1); 
        display.ShowValue(pattern, HEX, 2); 
        break;
    }
}

void UpButtonPressed(void)
{
    switch (currentMode)
    {
      case COUNT: 
        pixels = min(pixels + 1, 255); 
        display.ShowValue(pixels, HEX, 2); 
        break;
      
      case BRIGHTNESS: 
        brightness = min(brightness + 1, 255); 
        display.ShowValue(brightness, HEX, 2); 
        break;
      
      case PATTERN:
        pattern = min(pattern + 1, 9); 
        display.ShowValue(pattern, HEX, 2); 
        break;
    }
}

// Fill the dots one after the other with a color
void clear() 
{
  for(uint16_t i=0; i < strip.numPixels(); i++) 
  {
    strip.setPixelColor(i, 0);
  }
  strip.show();
}

// Fill the dots one after the other with a color
void colorWipe(uint32_t c, uint8_t wait) 
{
  for(uint16_t i=0; i < strip.numPixels() && !modeButtonPressed; i++) 
  {
    strip.setPixelColor(i, c);
    strip.show();
    delay(wait);
  }
  ReadCurrent();
}

void rainbow(uint8_t wait) 
{
  for(uint16_t j=0; j < 256 && !modeButtonPressed; j++) 
  {
    for(uint16_t i=0; i < strip.numPixels() && !modeButtonPressed; i++) 
	  {
      strip.setPixelColor(i, Wheel((i+j) & 255));
    }
    strip.show();
    delay(wait);
  }
  ReadCurrent();
}

// Slightly different, this makes the rainbow equally distributed throughout
void rainbowCycle(uint8_t wait) 
{
  for(uint16_t j=0; j < 256*5 && !modeButtonPressed; j++) 
  { // 5 cycles of all colors on wheel
    for(uint16_t i=0; i < strip.numPixels() && !modeButtonPressed; i++) 
	  {
      strip.setPixelColor(i, Wheel(((i * 256 / strip.numPixels()) + j) & 255));
    }
    strip.show();
    delay(wait);
  }
  ReadCurrent();
}

//Theatre-style crawling lights.
void theaterChase(uint32_t c, uint8_t wait) 
{
  for (int j=0; j < 10 && !modeButtonPressed; j++) 
  {  //do 10 cycles of chasing
    for (int q=0; q < 3 && !modeButtonPressed; q++) 
	{
      for (uint16_t i=0; i < strip.numPixels() && !modeButtonPressed; i=i+3) 
	  {
        strip.setPixelColor(i+q, c);    //turn every third pixel on
      }
      strip.show();
      ReadCurrent();
      delay(wait);

      for (uint16_t i=0; i < strip.numPixels() && !modeButtonPressed; i=i+3) 
	    {
        strip.setPixelColor(i+q, 0);        //turn every third pixel off
      }
    }
  }
}

//Theatre-style crawling lights with rainbow effect
void theaterChaseRainbow(uint8_t wait) 
{
  for (int j=0; j < 256 && !modeButtonPressed; j++) 
  {     // cycle all 256 colors in the wheel
    for (int q=0; q < 3 && !modeButtonPressed; q++) 
    {
      for (uint16_t i=0; i < strip.numPixels() && !modeButtonPressed; i=i+3) 
	    {
        strip.setPixelColor(i+q, Wheel( (i+j) % 255));    //turn every third pixel on
      }
      strip.show();
      ReadCurrent();
      delay(wait);

      for (uint16_t i=0; i < strip.numPixels() && !modeButtonPressed; i=i+3) 
	    {
        strip.setPixelColor(i+q, 0);        //turn every third pixel off
      }
    }
  }
}

// Input a value 0 to 255 to get a color value.
// The colours are a transition r - g - b - back to r.
uint32_t Wheel(byte WheelPos) 
{
  WheelPos = 255 - WheelPos;
  if (WheelPos < 85) 
  {
    return strip.Color(255 - WheelPos * 3, 0, WheelPos * 3);
  }
  if(WheelPos < 170) 
  {
    WheelPos -= 85;
    return strip.Color(0, WheelPos * 3, 255 - WheelPos * 3);
  }
  WheelPos -= 170;
  return strip.Color(WheelPos * 3, 255 - WheelPos * 3, 0);
}

float ReadCurrent()
{
  // taking the average for stable readings
  float current = 0;
//float voltage = 0;
//float power = 0;

  for (int i=0; i<20; i++) 
  {
    current = current + analogRead(PIN_AMP);
    //voltage = voltage + analogRead(PIN_VOLT); 
  }
  current = (current / 20);
  Serial.print("Raw =  " + String(current));
  //RAW => MEASURED => FACTOR
  //20.5 => 152.4mA => 7.43
  //11.8 => 86.7mA =>  7.35
  //33.85 => 252mA =>  7.44
  //11.95 => 87.1 => 7.28
  //7.15 => 53.8 => 7.52
  //7.65 => 55.3 => 7.22
  //49.1 => 360 => 7.33
  current = current * 7.36; // calibration value,to be changed according to components used
  Serial.print(", Calc =  " + String(current));
  
  //voltage = (voltage / 20); 
  //voltage = voltage * 0.0508 * 5.0; // calibration value,to be changed according to components used
  //power = voltage * current;

  maxCurrent = max(maxCurrent, round(current));
  Serial.println(", Max = " + String(maxCurrent));
  return current;
}