/* 
 * grove_alphanumeric_display.cpp
 * A library for Grove - grove_alphanumeric_display
 * 
 * Copyright (c) 2018 seeed technology inc.
 * Website    : www.seeed.cc
 * Author     : Jerry Yip
 * Create Time: 2018-06
 * Version    : 0.1
 * Change Log :
 * Copyright (c) 2018 seeed technology inc.
 * Website    : www.seeed.cc
 * Author     : downey 
 * Create Time: 2018-06
 * Version    : 1.1
 * Change Log :for digital tube development.
 *
 * The MIT License (MIT)
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */

#include "grove_alphanumeric_display.h"
#include "Wire.h"


uint16_t g_display_font4[]={
    // 0x0000,                 // display nothing.
    0x0080,                 // 'upper .'
    0x2000,                 // 'lower .'
    0x4478,                 // '0'
    0x0060,                 // '1'
    0x0758,                 // '2'
    0x0770,                 // '3'
    0x4360,                 // '4'
    0x4730,                 // '5'
    0x4738,                 // '6'
    0x0070,                 // '7'
    0x4778,                 // '8'
    0x4770,                 // '9'
    0x4378,                 // 'A'
    0x2d70,                 // 'B'
    0x4418,                 // 'C'
    0x2c70,                 // 'D'
    0x4718,                 // 'E'
    0x4318,                 // 'F'
    0x4538,                 // 'G'
    0x4368,                 // 'H'
    0x2c10,                 // 'I'
    0x0478,                 // 'J'
    0x2806,                 // 'K'
    0x4408,                 // 'L'
    0x40ea,                 // 'M'
    0x40ec,                 // 'N'
    0x4478,                 // 'O'
    0x4358,                 // 'P'
    0x447c,                 // 'Q'
    0x435c,                 // 'R'
    0x0494,                 // 'S'
    0x2810,                 // 'T'
    0x4468,                 // 'U'
    0x500a,                 // 'V'
    0x506c,                 // 'W'
    0x1086,                 // 'X'
    0x0882,                 // 'Y'
    0x1412,                 // 'Z'
    0x0000,                 //  '/' ilegal num
};



#define LAST_ITEM 38
uint16_t g_display_font2[]={
    // 0x0000,                 // display nothing.
    0x4000,                 // '.'
    0x0000,                 //  '/'   display nothing.
    0xa145,                 // '0'
    0x8001,                 // '1'
    0x3107,                 // '2'
    0xb007,                 // '3'
    0x9043,                 // '4'
    0xb046,                 // '5'
    0xb146,                 // '6'
    0x8005,                 // '7'
    0xb147,                 // '8'
    0xb047,                 // '9'
    0x9147,                 // 'A'
    0xb415,                 // 'B'
    0x2144,                 // 'C'
    0xa415,                 // 'D'
    0x3146,                 // 'E'
    0x1146,                 // 'F'
    0xb144,                 // 'G'
    0x9143,                 // 'H'
    0x2414,                 // 'I'
    0xA101,                 // 'J'
    0x0c18,                 // 'K'
    0x2140,                 // 'L'
    0x8169,                 // 'M'
    0x8961,                 // 'N'
    0xa145,                 // 'O'
    0x1147,                 // 'P'
    0xa945,                 // 'Q'
    0x1947,                 // 'R'
    0x2824,                 // 'S'
    0x0414,                 // 'T'
    0xa141,                 // 'U'
    0x8821,                 // 'V'
    0x8b41,                 // 'W'
    0x0a28,                 // 'X'
    0x0428,                 // 'Y'
    0x220c,                 // 'Z'
    0x0000,                 // ilegal num
};


Seeed_Digital_Tube::Seeed_Digital_Tube()
{
    
    _ms = 100;
}


void Seeed_Digital_Tube::setTubeType(TubeType_t type,uint8_t addr)
{
    _type = type;
    (_type == TYPE_2) ? (_tube_cnt = CNT_2) : (_tube_cnt = CNT_4);
    (_type == TYPE_2) ? (disp_font_p = g_display_font2) : (disp_font_p = g_display_font4); 
    HT16K33::init(addr);
}

bool Seeed_Digital_Tube::isLegalToDisplay(char byte)
{
    if( !((byte >= '.' && byte <= '9') || (byte >= 'A' && byte <= 'Z') || (byte >= 'a' && byte <= 'z') ) )
        return false;
    return true;
}




/**@brief Display number,If the param-num's len less than 4(or equal to),The tubes display static number,otherwise,it displays scroll number.
 * When it displays scroll number,the param interval is scrolling interval(ms) .. 
 * @param num the number to display.
 * @param interval :the interval of scroll number.
 * */
void Seeed_Digital_Tube::displayNum(uint32_t num,uint32_t interval)
{
    char num_str[15] = {0};
    sprintf(num_str,"%ld",(uint32_t)num);
    displayString(num_str,interval);
}



void Seeed_Digital_Tube::shiftDisplay(char *origin_disp_buf,char new_item)
{
    char temp[MAX_TUBE_COUNT] = {0};

    memcpy(temp,&origin_disp_buf[1],_tube_cnt - 1);

    temp[_tube_cnt-1] = new_item;

    memcpy(origin_disp_buf,temp,_tube_cnt);

    for(int i =0;i< _tube_cnt;i++)
    {
        setTubeBuf((TubeNum)(i+1),temp[i]);
        
    }
    writeBytes(_addr, 0x00, 16, _buffer);
    delay(_ms);
}

/**@brief Display string,If the param-str's len less than 4(or equal to),The tubes display static string,otherwise,it displays scroll string.
 * When it displays scroll string,the param interval is scrolling interval(ms) .. 
 * @param str the str to display.
 * @param interval :the interval of scroll string.
 * */
void Seeed_Digital_Tube::displayString(char *str,uint32_t interval)
{
    int len = 0;
    clearBuf();
    
    len = strlen(str);
    
    if(_tube_cnt >= len)
    {
        for(int i=0;i<len;i++)
        {
            //setTubeBuf((TubeNum)(_tube_cnt-len+i+1),disp_font_p[get_char_index(str[i])]);  //Convert number to char;
            setTubeBuf((TubeNum)(_tube_cnt-len+i+1),str[i]);  //Convert number to char;
        }
        writeBytes(_addr, 0x00, 16, _buffer);
        delay(_ms);
    }
    else
    {
        if(len > 255 ) len = 255;
        char origin_disp_buf[MAX_TUBE_COUNT]={0};

        for(int i=0;i<len;i++)
        {
            shiftDisplay(origin_disp_buf,str[i]);
            delay(interval);
        }
        for(int i=0;i<_tube_cnt;i++)
        {
            shiftDisplay(origin_disp_buf,0);
            delay(interval);
        }

    }
}


void Seeed_Digital_Tube::clear()
{
    memset(_buffer, 0, sizeof(_buffer));
    writeBytes(_addr, 0x00, 16, _buffer);
    delay(_ms);
}


void Seeed_Digital_Tube::fulDisplay()
{
    uint8_t buf[16] = {0};
    memcpy(_buffer,buf,sizeof(buf));
    memset(_buffer, 0xff, sizeof(_buffer));
    writeBytes(_addr, 0x00, 16, _buffer);
    delay(_ms);
}

/** There are some differences of the hardware design of two kinds of display(two & four tubes).
 *  two digital tubes use _buffer[2,3,4,5] to control display.
 *  four digital tubes use _buffer[2,3,4,5,6,7,8,9,10,11] to control display.
 *  
 **/
void Seeed_Digital_Tube::replace_bit12(TubeNum tube_num,bool bit1,bool bit2)
{
    switch(tube_num)
    {
        case FIRST_TUBE:
        _buffer[10] &= ~(1 << 4);
        _buffer[10] &= ~(1<<3);

        _buffer[10] |= (bit1<<4);
        _buffer[10] |= (bit2<<3);
        break;
        case SECOND_TUBE:

        _buffer[10] &= ~(1 << 6);
        _buffer[11] &= ~(1 << 6);

        _buffer[10] |= (bit1<<6);
        _buffer[11] |= (bit2<<6);
        break;
        case THIRD_TUBE:
        _buffer[10] &= ~(1 << 5);
        _buffer[11] &= ~(1 << 1);

        _buffer[10] |= (bit1<<5);
        _buffer[11] |= (bit2<<1);
        break;
        case FOURTH_TUBE:
        _buffer[11] &= ~(1 << 2);
        _buffer[11] &= ~(1 << 0);

        _buffer[11] |= (bit1<<2);
        _buffer[11] |= (bit2<<0);
        break;
        default:break;
    }
}

void Seeed_Digital_Tube::setPoint(bool first_dot,bool second_dot)
{
    if( TYPE_4 == _type )
    {
        //upper dot control
        if(first_dot){
        _buffer[10] |= disp_font_p[0];
        _buffer[11] |= disp_font_p[0]>>8;
        }
        //lower dot control
        if(second_dot){
            _buffer[10] |= disp_font_p[1];
            _buffer[11] |= disp_font_p[1]>>8;
        }
    }
    else if (TYPE_2 == _type)
    {
        if(first_dot)
            _buffer[SECOND_TUBE*2] |= disp_font_p[0] >> 8;

        if(second_dot)
            
            _buffer[FIRST_TUBE*2] |= disp_font_p[0] >> 8;
    }
    else{}
}


void Seeed_Digital_Tube::setTubeBuf(TubeNum tube_num,char byte)
{
    uint16_t value = disp_font_p[get_char_index(byte)];
    
    if( TYPE_4 == _type )
    {
        

        _buffer[tube_num*2] = value;
        _buffer[tube_num*2+1] = value >> 8;
        replace_bit12(tube_num,value&0x02,value&0x04);
    }
    else if (TYPE_2 == _type)
    {
        if(false == isLegalToDisplay(byte))
        {
            byte = '/';
        }
        //if first,set to 2,if second ,set to 1. To compatible with the hardware.
        tube_num = (TubeNum) (((int)tube_num+1) & (~(int)tube_num));
        
        if(byte >= 'a' && byte <= 'z')
        {
            byte -= 32;
        }

        if((byte >='.' && byte <= '9'))
        {
            _buffer[tube_num*2] = disp_font_p[byte-'0'+2] >> 8;
            _buffer[tube_num*2+1] = disp_font_p[byte-'0'+2];
        }
        else
        {
            _buffer[tube_num*2] = disp_font_p[byte-'A'+10+2] >> 8;
            _buffer[tube_num*2+1] = disp_font_p[byte-'A'+10+2];
        }
    }
    
}



/**@brief Specify the display char of a digital tube. 
 * @param tube_num The number of tube ,total 4.
 * @param c ,The char to display
 * */
void Seeed_Digital_Tube::setTubeSingleChar(TubeNum tube_num,char c)
{
    if(!( ((c >= 'A') && (c <= 'Z')) || ((c >= 'a') && (c <= 'z')) || (c=='.') ) )
    {
        return ;
    }
    if( TYPE_4 == _type )
    {
        setTubeBuf(tube_num , c);
    }
    else if (TYPE_2 == _type)
    {
        if(tube_num > SECOND_TUBE)
            return;

        setTubeBuf(tube_num ,c);
    }
    else
    {
    }
}

/**@brief Specify the display number of a digital tube. 
 * @param tube_num The number of tube ,total 4.
 * @param num ,The number to display
 * */
void Seeed_Digital_Tube::setTubeSingleNum(TubeNum tube_num,char num)
{
    if( !((num >= 0) && (num <= 9)) )
        return ;
    if( TYPE_4 == _type )
    {
        setTubeBuf(tube_num , num + 0x30);
    }
    else if (TYPE_2 == _type)
    {
        if(tube_num > SECOND_TUBE)
            return;

        setTubeBuf(tube_num ,num + 0x30);
    }
    else
    {
    }
    
}


void Seeed_Digital_Tube::display_one_tube(TubeNum tube_num,uint16_t value)
{
    _buffer[tube_num*2] = value;
    _buffer[tube_num*2+1] = value >> 8;
    replace_bit12(tube_num,value&0x02,value&0x04);
    writeBytes(_addr, 0x00, 16, _buffer);
    delay(_ms);
}


void Seeed_Digital_Tube::clearBuf()
{
    memset(_buffer,0,sizeof(_buffer));
}

int Seeed_Digital_Tube::get_char_index(char c)
{
    if((c >= '0') && (c <= '9'))
    {
        return c-0x30+2;
    }
    else if((c >= 'A') && (c <= 'Z'))
    {
        return c-0x37+2;
    }
    else if((c >= 'a') && (c <= 'z'))
    {
        return c-0x57+2;
    }
    else
    {
        return LAST_ITEM;
    }
}


void Seeed_Digital_Tube::display()
{
    writeBytes(_addr, 0x00, 16, _buffer);
    delay(_ms);
}