/*
 * Copyright (C) 2022 HiHope Open Source Organization .
 * Licensed 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 <stdio.h>
#include <stdlib.h>
#include <memory.h>

#include "ohos_init.h"
#include "cmsis_os2.h"
#include "iot_gpio.h"
#include "hi_io.h"
#include "hi_time.h"
#include "ssd1306.h"
#include "iot_i2c.h"
#include "iot_watchdog.h"
#include "robot_control.h"
#include "iot_errno.h"

#define OLED_I2C_BAUDRATE (400*1000)
#define GPIO13 13
#define GPIO14 14
#define FUNC_SDA 6
#define FUNC_SCL 6
#define I2C0 0
#define STACKSIZE 10240
#define PRIORITY 25

void Ssd1306TestTask(void* arg)
{
    (void) arg;
    hi_io_set_func(GPIO13, FUNC_SDA);
    hi_io_set_func(GPIO14, FUNC_SCL);
    IoTI2cInit(I2C0, OLED_I2C_BAUDRATE);

    IoTWatchDogDisable();

    unsigned int time1 = 20000;
    unsigned int time2 = 100;
    unsigned int time3 = 10;
    unsigned int x1 = 0;
    unsigned int y1 = 0;
    unsigned int x2 = 10;
    unsigned int y2 = 10;
    usleep(time1);
    ssd1306_Init();
    ssd1306_Fill(Black);
    ssd1306_SetCursor(x1, y1);
    ssd1306_DrawString("Hello OpenHarmony!", Font_7x10, White);

    uint32_t start = HAL_GetTick();
    ssd1306_UpdateScreen();
    uint32_t end = HAL_GetTick();
    printf("ssd1306_UpdateScreen time cost: %ud ms.\r\n", end - start);
    osDelay(time2);

    while (1) {
        ssd1306_Fill(Black);
        ssd1306_SetCursor(x2, y2);
        unsigned char status = GetCarStatus();
        if (status == CAR_OBSTACLE_AVOIDANCE_STATUS) {
            ssd1306_DrawString("ultrasonic", Font_7x10, White);
        } else if (status == CAR_STOP_STATUS) {
            ssd1306_DrawString("Robot Car Stop", Font_7x10, White);
        } else if (status == CAR_TRACE_STATUS) {
            ssd1306_DrawString("trace", Font_7x10, White);
        }
        ssd1306_UpdateScreen();
        osDelay(time3);
    }
}

void Ssd1306TestDemo(void)
{
    osThreadAttr_t attr;

    attr.name = "Ssd1306Task";
    attr.attr_bits = 0U;
    attr.cb_mem = NULL;
    attr.cb_size = 0U;
    attr.stack_mem = NULL;
    attr.stack_size = STACKSIZE;
    attr.priority = PRIORITY;

    if (osThreadNew(Ssd1306TestTask, NULL, &attr) == NULL) {
        printf("[Ssd1306TestDemo] Failed to create Ssd1306TestTask!\n");
    }
}
APP_FEATURE_INIT(Ssd1306TestDemo);