/*****************************************************************
;Project:   Light
;MCU:       
;Date:     
;File:      
;Function:  
******************************************************************/
#include "ws2812.h"
#include "delay.h"
#include "gpio.h"
/*---------------------------------------------------------------*/
//变量定义
unsigned char gLedBuffer[LED_NUM][GRB] = {0}; //亮灯数据缓存
/*---------------------------------------------------------------*/
//函数声明
static void WS_Port_Init(void);
static void WS_DIN_Clear(void);
static void WS_Write_0(void);
static void WS_Write_1(void);
/*---------------------------------------------------------------*/
//WS2812初始化
void WS_Init(void)
{
	WS_Port_Init();
	WS_CloseAll();
}
/*---------------------------------------------------------------*/
//WS供电使能
void WS_Power(FunctionalState NewState)
{
	 GPIO_Set_Output(U32BIT(WS_EN_PIN)); 
	
   if(NewState != DISABLE)
   GPIO_Pin_Set(U32BIT(WS_EN_PIN));
   else
   GPIO_Pin_Clear(U32BIT(WS_EN_PIN)); 
}
/*---------------------------------------------------------------*/
//关闭所有的led
void WS_CloseAll(void)
{
	unsigned char led_i, color_i;
	
	for(led_i=0;led_i<LED_NUM;led_i++)
	{
		for(color_i=0;color_i<GRB;color_i++)
		{
			gLedBuffer[led_i][color_i] = 0;
		}
	}
	
	WS_Show(); //点灯
	
 //WS_Power(DISABLE); //关闭WS的供电	
}
/*---------------------------------------------------------------*/
//--清除所有ledbuff 缓存
void LedBuff_CleanAll(void)
{
    unsigned char led_i,color_i;  //用于for循坏
    
    for(led_i=0;led_i<LED_NUM;led_i++)
        for(color_i=0;color_i<GRB;color_i++)
             gLedBuffer[led_i][color_i] = 0;
}
/*---------------------------------------------------------------*/
//点亮所有灯
void WS_LightAll(void)
{
  WS_WriteNum(LED_NUM,0xff0000);
}
/*---------------------------------------------------------------*/
//指定写某个灯写入color数据，不show灯
void LedBuff_WriteOrder(unsigned char ledOrder, unsigned int color)
{
    unsigned char color_i;
     
//    ledOrder = ledOrder-1;  
    //先把要改变的颜色填入到led的buffer中
    for(color_i=0;color_i<GRB;color_i++)
    {
        if(color&0x800000)
        {
           gLedBuffer[ledOrder][color_i] = 1;
        }
        else
        {
           gLedBuffer[ledOrder][color_i] = 0;
        }
        color <<= 1;
    }
}
/*---------------------------------------------------------------*/
//指定写某个灯的color并显示颜色
void WS_WriteOrder(unsigned char ledOrder, unsigned int color)
{
    unsigned char  color_i;
     
//    ledOrder = ledOrder-1;  
    //先把要改变的颜色填入到led的buffer中
    for(color_i=0;color_i<GRB;color_i++)
    {
        if(color&0x800000)
        {
           gLedBuffer[ledOrder][color_i] = 1;
        }
        else
        {
           gLedBuffer[ledOrder][color_i] = 0;
        }
        color <<= 1;
    }
    WS_Show();  //点灯  
}
/*---------------------------------------------------------------*/
//-给前num个led写入color,但不show
void LedBuff_WriteNum(unsigned char numLed,unsigned int color)
{
    unsigned char led_i,color_i;  //用于for循坏
    unsigned int color_temp;
    
    for(led_i=0;led_i<numLed;led_i++)
    {
        color_temp = color;
        for(color_i=0;color_i<GRB;color_i++)
        {
            if(color_temp&0x800000)
            {
               gLedBuffer[led_i][color_i] = 1;
            }
            else
            {
               gLedBuffer[led_i][color_i] = 0;
            }
            color_temp <<= 1;
        }
    } 
    color_temp <<= 1;
}
/*---------------------------------------------------------------*/
//--给前num个led写入color 并显示
void WS_WriteNum(unsigned char numLed, unsigned int color)
{
    unsigned char led_i,color_i;
    unsigned int color_temp;

    //先把要改变的颜色填入到led的buffer中
    for(led_i=0;led_i<numLed;led_i++)
    {
        color_temp = color;
        for(color_i=0;color_i<GRB;color_i++)
        {
            if(color_temp&0x800000)
            {
               gLedBuffer[led_i][color_i] = 1;
            }
            else
            {
               gLedBuffer[led_i][color_i] = 0;
            }
            color_temp <<= 1;
        }
    }
    WS_Show();  //点灯  
}
/*---------------------------------------------------------------*/
//WS驱动IO口初始化
static void WS_Port_Init(void)
{
	PIN_Set_GPIO(U32BIT(WS_DIN_PIN) | U32BIT(WS_EN_PIN),PIN_SEL_GPIO); //引脚设置成普通IO口
	GPIO_Set_Output(U32BIT(WS_DIN_PIN) | U32BIT(WS_EN_PIN)); //设置引脚为输出模式
	GPIO_Pin_Clear(U32BIT(WS_DIN_PIN)|U32BIT(WS_EN_PIN)); //WS数据IO和使能IO都清零
}
/*---------------------------------------------------------------*/
//灯带显示函数
void WS_Show(void)
{
	unsigned led_i, color_i;
	
	WS_Power(ENABLE);
		
	__disable_irq();    //关闭全局中断（防止打断）
	//依次写10个灯
	for(led_i=0; led_i<LED_NUM; led_i++)
	{
			//依次写入每个灯的24bit时序
			for(color_i=0;color_i<GRB;color_i++)
			{
					if(gLedBuffer[led_i][color_i] != 0)  //写高
					{
							WS_Write_1();
					}
					else                                //写低
					{
							WS_Write_0();
					}
			}
	}
	WS_DIN_Clear();
	delay_us(300);   //实践证明大于20us就行 
	__enable_irq();  //打开全局中断
	
}

/*---------------------------------------------------------------*/
//灯带显示函数但是最后不打开总中断
void WS_Show_Disable_irq(void)
{
	unsigned led_i, color_i;
	
	WS_Power(ENABLE);
	
	__disable_irq();    //关闭全局中断（防止打断）
	//依次写10个灯
	for(led_i=0; led_i<LED_NUM; led_i++)
	{
			//依次写入每个灯的24bit时序
			for(color_i=0;color_i<GRB;color_i++)
			{
					if(gLedBuffer[led_i][color_i] != 0)  //写高
					{
							WS_Write_1();
					}
					else                                //写低
					{
							WS_Write_0();
					}
			}
	}
	WS_DIN_Clear();
	delay_us(300);   //实践证明大于20us就行 	
}
/*---------------------------------------------------------------*/
//WS的数据清零
static void WS_DIN_Clear(void)
{
	GPIO_Set_Output(U32BIT(WS_DIN_PIN));
	GPIO_Pin_Clear(U32BIT(WS_DIN_PIN));
}
/*---------------------------------------------------------------*/
//写bit_0
static void WS_Write_0(void)
{
	WS_DIN_HIGH;
  __NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();
	WS_DIN_LOW;
	__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();
	__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();
	__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();
}
/*---------------------------------------------------------------*/
//写bit_1
static void WS_Write_1(void)
{
	WS_DIN_HIGH;
	__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();
	__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();
	__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();
	WS_DIN_LOW;
	__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();__NOP();
}
/*---------------------------------------------------------------*/