/**
|
* Licensed to the Apache Software Foundation (ASF) under one
|
* or more contributor license agreements. See the NOTICE file
|
* distributed with this work for additional information
|
* regarding copyright ownership. The ASF licenses this file
|
* to you under the Apache License, Version 2.0 (the
|
* "License"); you may not use this file except in compliance
|
* with the License. You may obtain a copy of the License at
|
*
|
* http://www.apache.org/licenses/LICENSE-2.0
|
*
|
* Unless required by applicable law or agreed to in writing,
|
* software distributed under the License is distributed on an
|
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
* KIND, either express or implied. See the License for the
|
* specific language governing permissions and limitations
|
* under the License.
|
*/
|
|
#include "nimble_syscfg.h"
|
|
#if (MYNEWT_VAL(USE_NEOPIXEL))
|
|
#include <assert.h>
|
#include <string.h>
|
#include <stdio.h>
|
#include "sysinit/sysinit.h"
|
#include "os/os.h"
|
#include "bsp/bsp.h"
|
#include "pwm/pwm.h"
|
#include "nrfx.h"
|
#include "nrfx_pwm.h"
|
#include "ws2812.h"
|
|
#define BITS_PER_SEQ (24)
|
#define BIT0 (0x8000 | 6)
|
#define BIT1 (0x8000 | 11)
|
|
static const nrfx_pwm_t pwm = NRFX_PWM_INSTANCE(WS2812_PWM);
|
|
static const nrfx_pwm_config_t pwm_config = {
|
.output_pins = { WS2812_GPIO, NRFX_PWM_PIN_NOT_USED, NRFX_PWM_PIN_NOT_USED, NRFX_PWM_PIN_NOT_USED },
|
.irq_priority = 3,
|
.base_clock = NRF_PWM_CLK_16MHz,
|
.count_mode = NRF_PWM_MODE_UP,
|
.top_value = 20,
|
.load_mode = NRF_PWM_LOAD_COMMON,
|
.step_mode = NRF_PWM_STEP_AUTO,
|
};
|
|
static uint16_t pwm_seq_values[2][BITS_PER_SEQ];
|
|
static const nrf_pwm_sequence_t pwm_seq[2] = {
|
{
|
.values.p_raw = pwm_seq_values[0],
|
.length = BITS_PER_SEQ,
|
.repeats = 0,
|
.end_delay = 0,
|
}, {
|
.values.p_raw = pwm_seq_values[1],
|
.length = BITS_PER_SEQ,
|
.repeats = 0,
|
.end_delay = 0,
|
},
|
};
|
|
static uint32_t led_color[WS2812_NUM_LED];
|
static int led_idx;
|
|
static void
|
load_pixel(void)
|
{
|
uint16_t *seq_values;
|
uint32_t grb;
|
int i;
|
|
seq_values = pwm_seq_values[led_idx & 1];
|
grb = led_color[led_idx];
|
|
for (i = 0; i < BITS_PER_SEQ; i++) {
|
*seq_values = grb & 0x800000 ? BIT1 : BIT0;
|
grb <<= 1;
|
seq_values++;
|
}
|
|
led_idx++;
|
}
|
|
static void
|
pwm_handler_func(nrfx_pwm_evt_type_t event_type)
|
{
|
switch (event_type) {
|
case NRFX_PWM_EVT_END_SEQ0:
|
case NRFX_PWM_EVT_END_SEQ1:
|
load_pixel();
|
break;
|
default:
|
break;
|
}
|
}
|
|
int
|
ws2812_init(void)
|
{
|
nrfx_err_t err;
|
|
err = nrfx_pwm_init(&pwm, &pwm_config, pwm_handler_func);
|
|
return err != NRFX_SUCCESS;
|
}
|
|
int
|
ws2812_write(const uint32_t *rgb)
|
{
|
uint32_t grb;
|
int i;
|
|
for (i = 0; i < WS2812_NUM_LED; i++) {
|
grb = 0;
|
grb |= (rgb[i] & 0x00FF00) << 8;
|
grb |= (rgb[i] & 0xFF0000) >> 8;
|
grb |= (rgb[i] & 0x0000FF);
|
|
led_color[i] = grb;
|
}
|
|
led_idx = 0;
|
|
load_pixel();
|
load_pixel();
|
nrfx_pwm_complex_playback(&pwm, &pwm_seq[0], &pwm_seq[1], WS2812_NUM_LED,
|
NRFX_PWM_FLAG_SIGNAL_END_SEQ0 |
|
NRFX_PWM_FLAG_SIGNAL_END_SEQ1 |
|
NRFX_PWM_FLAG_STOP);
|
|
return 0;
|
}
|
#endif
|
