/*
 * Author: Yevgeniy Kiveisha <yevgeniy.kiveisha@intel.com>
 * Copyright (c) 2014 Intel Corporation.
 *
 * 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 <string>
#include <unistd.h>

#include "ssd1308.h"

using namespace upm;

SSD1308::SSD1308(int bus_in, int addr_in) : I2CLcd(bus_in, addr_in)
{
    mraa_i2c_write_byte_data(m_i2c_lcd_control, DISPLAY_CMD_OFF, LCD_CMD); // display off
    usleep(4500);
    mraa_i2c_write_byte_data(m_i2c_lcd_control, DISPLAY_CMD_ON, LCD_CMD); // display on
    usleep(4500);
    setNormalDisplay(); // set to normal display '1' is ON

    clear();
    setAddressingMode(PAGE);
}

SSD1308::~SSD1308()
{
}

mraa_result_t
SSD1308::draw(uint8_t* data, int bytes)
{
    mraa_result_t error = MRAA_SUCCESS;

    setAddressingMode(HORIZONTAL);
    for (int idx = 0; idx < bytes; idx++) {
        mraa_i2c_write_byte_data(m_i2c_lcd_control, data[idx], LCD_DATA);
    }

    return error;
}

/*
 * **************
 *  virtual area
 * **************
 */
mraa_result_t
SSD1308::write(std::string msg)
{
    mraa_result_t error = MRAA_SUCCESS;
    uint8_t data[2] = { 0x40, 0 };

    setAddressingMode(PAGE);
    for (std::string::size_type i = 0; i < msg.size(); ++i) {
        writeChar(m_i2c_lcd_control, msg[i]);
    }

    return error;
}

mraa_result_t
SSD1308::setCursor(int row, int column)
{
    mraa_result_t error = MRAA_SUCCESS;

    error = mraa_i2c_write_byte_data(m_i2c_lcd_control,
                                     BASE_PAGE_START_ADDR + row,
                                     LCD_CMD); // set page address
    error = mraa_i2c_write_byte_data(m_i2c_lcd_control,
                                     BASE_LOW_COLUMN_ADDR + (8 * column & 0x0F),
                                     LCD_CMD); // set column lower address
    error = mraa_i2c_write_byte_data(m_i2c_lcd_control,
                                     BASE_HIGH_COLUMN_ADDR + ((8 * column >> 4) & 0x0F),
                                     LCD_CMD); // set column higher address

    return error;
}

mraa_result_t
SSD1308::clear()
{
    mraa_result_t error = MRAA_SUCCESS;
    uint8_t columnIdx, rowIdx;

    mraa_i2c_write_byte_data(m_i2c_lcd_control, DISPLAY_CMD_OFF, LCD_CMD); // display off
    for (rowIdx = 0; rowIdx < 8; rowIdx++) {
        setCursor(rowIdx, 0);

        // clear all columns
        for (columnIdx = 0; columnIdx < 16; columnIdx++) {
            writeChar(m_i2c_lcd_control, ' ');
        }
    }
    mraa_i2c_write_byte_data(m_i2c_lcd_control, DISPLAY_CMD_ON, LCD_CMD); // display on
    home();

    return MRAA_SUCCESS;
}

mraa_result_t
SSD1308::home()
{
    return setCursor(0, 0);
}

/*
 * **************
 *  private area
 * **************
 */
mraa_result_t
SSD1308::writeChar(mraa_i2c_context ctx, uint8_t value)
{
    if (value < 0x20 || value > 0x7F) {
        value = 0x20; // space
    }

    for (uint8_t idx = 0; idx < 8; idx++) {
        mraa_i2c_write_byte_data(m_i2c_lcd_control, BasicFont[value - 32][idx], LCD_DATA);
    }
}

mraa_result_t
SSD1308::setNormalDisplay()
{
    return mraa_i2c_write_byte_data(m_i2c_lcd_control,
                                    DISPLAY_CMD_SET_NORMAL_1308,
                                    LCD_CMD); // set to normal display '1' is ON
}

mraa_result_t
SSD1308::setAddressingMode(displayAddressingMode mode)
{
    mraa_i2c_write_byte_data(m_i2c_lcd_control,
                             DISPLAY_CMD_MEM_ADDR_MODE,
                             LCD_CMD);                          // set addressing mode
    mraa_i2c_write_byte_data(m_i2c_lcd_control, mode, LCD_CMD); // set page addressing mode
}