blink_cpp.cpp

Go to the documentation of this file.

/*
 * Copyright (C) 2021  Patrick Pedersen
 
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 * 
 */
 
// #define WS2812_TARGET_PLATFORM_AVR
 
#include <avr/io.h>
#include <util/delay.h>
 
#include <ws2812_cpp.h>
 
// Parameters - ALTER THESE TO CORRESPOND WITH YOUR OWN SETUP!
#define N_LEDS 8                
#define DATA_PINS_PORT PORTB    
#define DATA_PINS_DDR DDRB      
#define DATA_PINS {PB0, PB1}    
#define RESET_TIME 50           
#define COLOR_ORDER grb         
 
int main()
{
        uint8_t pins[] = DATA_PINS;       // Data pins
        ws2812_rgb leds[N_LEDS];          // RGB array which represents the LEDs
        ws2812_cfg cfg;                   // Device configurationn
 
        // Configure the WS2812 device struct
        cfg.port = &DATA_PINS_PORT;
        cfg.ddr = &DATA_PINS_DDR;
        cfg.pins = pins;
        cfg.n_dev = sizeof(pins);         // Number of devices driven by this struct
        cfg.rst_time_us = RESET_TIME;
        cfg.order = grb;
        
        uint8_t ret;
        ws2812_cpp ws2812_dev(&cfg, &ret); // Initialize device struct
 
        if (ret != 0) {
                // HANDLE ERROR...
                void;
        };
 
        // Blink device
        while (1) {
                // Program all LEDs to white
                for (unsigned int i = 0; i < N_LEDS; i++) {
                        leds[i].r = 255;
                        leds[i].g = 255;
                        leds[i].b = 255;              
                }
 
                ws2812_dev.prep_tx();         // Prepare to transmit data
                ws2812_dev.tx(leds, N_LEDS);  // Transmit array of rgb values to the device
                ws2812_dev.close_tx();        // Close transmission
 
                // Wait 500ms
                _delay_ms(500);
 
                // Program all LEDs to black (off)
                for (unsigned int i = 0; i < N_LEDS; i++) {
                        leds[i].r = 0;
                        leds[i].g = 0;
                        leds[i].b = 0;
                }
 
                ws2812_dev.prep_tx();        // Prepare to transmit data
                ws2812_dev.tx(leds, N_LEDS); // Transmit array of rgb values to the device
                ws2812_dev.close_tx();       // Close transmission
 
                // Wait 500ms
                _delay_ms(500);
        }
 
        return 0;
}