#include "camera.h"

#include "esp_camera.h"
#include "esp_log.h"
#include "esp_psram.h"

enum {
    XIAO_CAM_PIN_PWDN = -1,
    XIAO_CAM_PIN_RESET = -1,
    XIAO_CAM_PIN_XCLK = 10,
    XIAO_CAM_PIN_SIOD = 40,
    XIAO_CAM_PIN_SIOC = 39,
    XIAO_CAM_PIN_Y9 = 48,
    XIAO_CAM_PIN_Y8 = 11,
    XIAO_CAM_PIN_Y7 = 12,
    XIAO_CAM_PIN_Y6 = 14,
    XIAO_CAM_PIN_Y5 = 16,
    XIAO_CAM_PIN_Y4 = 18,
    XIAO_CAM_PIN_Y3 = 17,
    XIAO_CAM_PIN_Y2 = 15,
    XIAO_CAM_PIN_VSYNC = 38,
    XIAO_CAM_PIN_HREF = 47,
    XIAO_CAM_PIN_PCLK = 13,
};

static const char *TAG = "camera";

esp_err_t camera_init(void)
{
    camera_config_t config = {
        .pin_pwdn = XIAO_CAM_PIN_PWDN,
        .pin_reset = XIAO_CAM_PIN_RESET,
        .pin_xclk = XIAO_CAM_PIN_XCLK,
        .pin_sccb_sda = XIAO_CAM_PIN_SIOD,
        .pin_sccb_scl = XIAO_CAM_PIN_SIOC,
        .pin_d7 = XIAO_CAM_PIN_Y9,
        .pin_d6 = XIAO_CAM_PIN_Y8,
        .pin_d5 = XIAO_CAM_PIN_Y7,
        .pin_d4 = XIAO_CAM_PIN_Y6,
        .pin_d3 = XIAO_CAM_PIN_Y5,
        .pin_d2 = XIAO_CAM_PIN_Y4,
        .pin_d1 = XIAO_CAM_PIN_Y3,
        .pin_d0 = XIAO_CAM_PIN_Y2,
        .pin_vsync = XIAO_CAM_PIN_VSYNC,
        .pin_href = XIAO_CAM_PIN_HREF,
        .pin_pclk = XIAO_CAM_PIN_PCLK,
        .xclk_freq_hz = 20000000,
        .ledc_timer = LEDC_TIMER_0,
        .ledc_channel = LEDC_CHANNEL_0,
        .pixel_format = PIXFORMAT_JPEG,
        .frame_size = FRAMESIZE_QVGA,
        .jpeg_quality = 12,
        .fb_count = 2,
        .grab_mode = CAMERA_GRAB_LATEST,
        .fb_location = CAMERA_FB_IN_PSRAM,
    };

#if defined(CONFIG_SPIRAM)
    if (!esp_psram_is_initialized()) {
        ESP_LOGW(TAG, "PSRAM is not initialized; reducing framebuffer settings");
        config.fb_count = 1;
        config.fb_location = CAMERA_FB_IN_DRAM;
        config.grab_mode = CAMERA_GRAB_WHEN_EMPTY;
    }
#else
    ESP_LOGW(TAG, "PSRAM support is disabled in sdkconfig; using DRAM framebuffer");
    config.fb_count = 1;
    config.fb_location = CAMERA_FB_IN_DRAM;
    config.grab_mode = CAMERA_GRAB_WHEN_EMPTY;
#endif

    ESP_LOGI(TAG, "initializing camera for XIAO ESP32-S3 Sense");
    esp_err_t err = esp_camera_init(&config);
    if (err != ESP_OK) {
        return err;
    }

    sensor_t *sensor = esp_camera_sensor_get();
    if (sensor != NULL) {
        ESP_LOGI(TAG, "tuning sensor PID=0x%04x for face capture", sensor->id.PID);
        sensor->set_vflip(sensor, 1);
        sensor->set_brightness(sensor, 1);
        sensor->set_contrast(sensor, 1);
        sensor->set_saturation(sensor, 0);
        sensor->set_sharpness(sensor, 1);
        sensor->set_denoise(sensor, 1);
        sensor->set_whitebal(sensor, 1);
        sensor->set_gain_ctrl(sensor, 1);
        sensor->set_exposure_ctrl(sensor, 1);
        sensor->set_aec2(sensor, 1);
        sensor->set_ae_level(sensor, 1);
        sensor->set_quality(sensor, 10);
    }

    return ESP_OK;
}