/*****************************************************************************
Биты регистра состояния
******************************************************************************/

/* Bit 7 = "1" (RDY) indicates the device is ready.
Bit 7 = "0" indicates the write cycle is in progress. */
#define AT45_SREG_RDY  7



/* Bit 6 = "0" data in the main memory page matches the data in the buffer.
Bit 6 = "1" at least one bit of the data in the main memory page not match
the data in the buffer. */
#define AT45_SREG_COMP  6



/* маска идентификатора чипа, сдвинутая в право на два бита */
#define AT45_DEVICE_ID_MASK (0x0F << 2)



/*****************************************************************************
Инициализация SPI драйвера и тестирование МС памяти
Возвращает TRUE, если нет ошибок памяти, иначе FALSE
******************************************************************************/
BOOL SPIMEM_Init(void)
{
   SPI_Init(); /* инициализируем порт SPI */

    

/* ждем установки памяти в рабочее состояние */
ULONG start = g_SysTIME_ms;
    while(AT45_STARTUP_TIME > (g_SysTIME_ms - start));

    

/* читаем регистр статуса МС */
    UCHAR status = SPIMEM_GetStatus();
    if(AT45_DEVICE_ID != (status  & AT45_DEVICE_ID_MASK))
       return FALSE; /* ошибка! не совпал ID установленной МС */

    

/* тестируем внутренние буферы МС */
UCHAR test = 0x55; /* тест-байт */
    for(UCHAR j = 0; j < 2; j++)
    {
     for(UINT i = 0; i < AT45_PAGE_SIZE; i++)
     {
           /* пишем тест-байт в оба буфера */
           SPIMEM_BufWrite(AT45_BUFFER1_WRITE, i, 1, &test);
         SPIMEM_BufWrite(AT45_BUFFER2_WRITE, i, 1, &test);
      
            /* проверяем прочитанный байт на равенство тест-байту */
        SPIMEM_BufRead(AT45_BUFFER1_READ, i, 1, &status);
         if(test != status)
                return FALSE;
            SPIMEM_BufRead(AT45_BUFFER2_READ, i, 1, &status);
         if(test != status)
              return FALSE;
        }
        test = 0xAA; /* инвертируем тест-байт */
    }

    /* OK */
    return TRUE;
}



/*****************************************************************************
Чтение регистра статуса
AT45_STATUS_READ
******************************************************************************/
UCHAR SPIMEM_GetStatus(void)
{
   SPI_SlaveSelect(); /* выбираем чип */
    SPI_Write(AT45_STATUS_READ); /* посылаем команду */
    UCHAR status = SPI_Read(); /* читаем статус */
    SPI_SlaveDeselect(); /* освобождаем чип */
    return status;
}


/*****************************************************************************
Чтение порта SPI
******************************************************************************/
UCHAR SPI_Read(void)
{
   SPDR = 0xff; /* загружаем SPI данными */
    while(!(SPSR & (1<<SPIF))); /* ждем окончания передачи */
    return SPDR; /* возвращаем принятые данные */
}

/*****************************************************************************
Запись в порт SPI
******************************************************************************/
void SPI_Write(UCHAR data)
{
   SPDR = data; /* загружаем SPI данными */
    while(!(SPSR & (1<<SPIF))); /* ждем окончания передачи */
}



/*****************************************************************************
Чтение буфера
AT45_BUFFER1(2)_READ = 0x54(0x56)
Вход:
command - команда чтения буфера №1 или №2
buf_pos - адрес начала читаемого блока в буфере
size  - размер блока
trg  - буфер для записи
******************************************************************************/
void SPIMEM_BufRead(UCHAR command, UINT buf_pos, UINT size, UCHAR *trg)
{
   /* сылаем команду */
   SPIMEM_Tx5(command, 0, buf_pos);

    /* читаем массив */
    while(size--)
     *trg++ = SPI_Read();

    /* освобождаем чип */
    SPI_SlaveDeselect();
}

/*****************************************************************************
Запись в буфер
AT45_BUFFER1(2)_WRITE = 0x84(0x87)
Вход:
command - команда записи в буфер №1 или №2
buf_pos - адрес начала читаемого блока в буфере
size  - размер блока
src  - буфер для чтения
******************************************************************************/
void SPIMEM_BufWrite(UCHAR command, UINT buf_pos, UINT size, UCHAR *src)
{
   /* посылаем команду */
   SPIMEM_Tx4(command, 0, buf_pos);

    /* пишем массив */
    while(size--)
     SPI_Write(*src++);

    /* освобождаем чип */
    SPI_SlaveDeselect();
}



/*****************************************************************************
4-х байтная посылка:
Вход:
command  - команда
page_addr - адрес страницы
buf_pos  - позиция в буфере

!!! Кроме этого производит выборку кристалла МС !!!
******************************************************************************/
void SPIMEM_Tx4(UCHAR command, UINT page_addr, UINT buf_pos)
{
   SPI_SlaveSelect();
   page_addr <<= 1;
    SPI_Write(command);
   SPI_Write(HIBYTE(page_addr));
    SPI_Write(LOBYTE(page_addr) | HIBYTE(buf_pos));
    SPI_Write(LOBYTE(buf_pos));
}

/*****************************************************************************
5-и байтная посылка:
Установка адреса дополняется посылкой 1-го пустого байта

!!! Кроме этого производит выборку кристалла МС !!!
******************************************************************************/
void SPIMEM_Tx5(UCHAR command, UINT page_addr, UINT buf_pos)
{
   SPIMEM_Tx4(command, page_addr, buf_pos);
    SPI_Write(0);
}

uint8_t spi_t::exchange(uint8_t out)
{
    uint8_t Data = out;
    for(uint8_t i = 0; i < 8; i++)
    {
        if(Data & (1 << 7))
            on(SPI_MOSI);
        else
            off(SPI_MOSI);

        on(SPI_SCLK);

        Data <<= 1;

        if(signal(SPI_MISO))
            Data |= (1<<0);
        off(SPI_SCLK);
    }
    return Data;
}
inline void spi_t::wait_done()
{
}
#define STATUS_READ                     0xD7

uint8_t at45_t::ReadStatus( void )
{
    Enable();
    spi_t::exchange(STATUS_READ);
    spi_t::wait_done();
    status_t Result = (status_t)spi_t::exchange();              //send dummy data and receive Status
    spi_t::wait_done();
    Disable();
    return Result;
}

#if     defined( AT45DB041 )
    #define FLASH_ID_MASK           0x38
    #define FLASH_ID                0x18
#elif   defined( AT45DB161 )
    #define FLASH_ID_MASK           0x3C
    #define FLASH_ID                0x2C
#elif   defined( AT45DB321 )
    #define FLASH_ID_MASK           0x3C
    #define FLASH_ID                0x34
#else
    #error  "No External flash type specified"
#endif
#define COMPARE_MISMATCH_BIT            (1<<6)
#define FLASH_READY_BIT                 (1<<7)

uint8_t at45_t::WaitReady(void)
{
    uint8_t Result;
    do
        Result = ReadStatus();
    while( (Result & (FLASH_ID_MASK | FLASH_READY_BIT)) != (FLASH_ID | FLASH_READY_BIT));
    return Result;
}

#define DF_CMD_STATUS_READ_SPI03            0xD7
unsigned char bLastStatus;

// *
// * additionally deactivates Chip Select signal before actual waiting,
// * places last status register value in bLastStatus
// *
void DF_Wait_Busy( void )
{
    DF_ChipSelect_Stop();
    bLastStatus = 0;
    DF_ChipSelect_Start();
    SPI_WriteByte( DF_CMD_STATUS_READ_SPI03 );
    do {
        bLastStatus = SPI_ReadByte();
    } while( (bLastStatus & 0x80) == 0 );
    DF_ChipSelect_Stop();
}

void DF_Wait_Busy( void )
{
    DF_ChipSelect_Stop();
    bLastStatus = 0;
    do {
        DF_ChipSelect_Start();
        SPI_WriteByte( DF_CMD_STATUS_READ_SPI03 );
        bLastStatus = SPI_ReadByte();
        DF_ChipSelect_Stop();
    } while( (bLastStatus & 0x80) == 0 );
}