/*
 * This file is part of the MicroPython project, http://micropython.org/
 *
 * The MIT License (MIT)
 *
 * Copyright (c) 2022 Jeff Epler for Adafruit Industries
 *
 * 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 "bindings/esp32_camera/Camera.h"
#include "bindings/espidf/__init__.h"
#include "common-hal/esp32_camera/Camera.h"
#include "shared-bindings/microcontroller/Pin.h"
#include "common-hal/microcontroller/Pin.h"

static void maybe_claim_pin(const mcu_pin_obj_t *pin) {
    if (pin) {
        claim_pin(pin);
    }
}

void common_hal_esp32_camera_camera_construct(
    esp32_camera_camera_obj_t *self,
    uint8_t data_pins[8],
    const mcu_pin_obj_t *external_clock_pin,
    const mcu_pin_obj_t *pixel_clock_pin,
    const mcu_pin_obj_t *vsync_pin,
    const mcu_pin_obj_t *href_pin,
    const mcu_pin_obj_t *powerdown_pin,
    const mcu_pin_obj_t *reset_pin,
    const busio_i2c_obj_t *i2c,
    mp_int_t external_clock_frequency,
    pixformat_t pixel_format,
    framesize_t frame_size,
    mp_int_t jpeg_quality,
    mp_int_t framebuffer_count,
    camera_grab_mode_t grab_mode) {

    for (int i = 0; i < 8; i++) {
        claim_pin_number(data_pins[i]);
    }
    claim_pin(external_clock_pin);
    claim_pin(pixel_clock_pin);
    claim_pin(vsync_pin);
    claim_pin(href_pin);
    maybe_claim_pin(powerdown_pin);
    maybe_claim_pin(reset_pin);

    common_hal_pwmio_pwmout_construct(&self->pwm, external_clock_pin, 1, external_clock_frequency, true);

    self->camera_config.pin_pwdn = common_hal_mcu_pin_number(powerdown_pin);
    self->camera_config.pin_reset = common_hal_mcu_pin_number(reset_pin);
    self->camera_config.pin_xclk = common_hal_mcu_pin_number(external_clock_pin);

    self->camera_config.pin_sccb_sda = NO_PIN;
    self->camera_config.pin_sccb_scl = NO_PIN;
    /* sccb i2c port set below */

    self->camera_config.pin_d7 = data_pins[7];
    self->camera_config.pin_d6 = data_pins[6];
    self->camera_config.pin_d5 = data_pins[5];
    self->camera_config.pin_d4 = data_pins[4];
    self->camera_config.pin_d3 = data_pins[3];
    self->camera_config.pin_d2 = data_pins[2];
    self->camera_config.pin_d1 = data_pins[1];
    self->camera_config.pin_d0 = data_pins[0];

    self->camera_config.pin_vsync = common_hal_mcu_pin_number(vsync_pin);
    self->camera_config.pin_href = common_hal_mcu_pin_number(href_pin);
    self->camera_config.pin_pclk = common_hal_mcu_pin_number(pixel_clock_pin);

    self->camera_config.xclk_freq_hz = external_clock_frequency;

    self->camera_config.ledc_timer = self->pwm.tim_handle.timer_num;
    self->camera_config.ledc_channel = self->pwm.chan_handle.channel;

    self->camera_config.pixel_format = pixel_format;
    self->camera_config.frame_size = frame_size;
    self->camera_config.jpeg_quality = jpeg_quality;
    self->camera_config.fb_count = framebuffer_count;
    self->camera_config.grab_mode = grab_mode;

    self->camera_config.sccb_i2c_port = i2c->i2c_num;

    CHECK_ESP_RESULT(esp_camera_init(&self->camera_config));
}

extern void common_hal_esp32_camera_camera_deinit(esp32_camera_camera_obj_t *self) {
    if (common_hal_esp32_camera_camera_deinited(self)) {
        return;
    }

    common_hal_pwmio_pwmout_deinit(&self->pwm);

    reset_pin_number(self->camera_config.pin_pwdn);
    reset_pin_number(self->camera_config.pin_reset);
    reset_pin_number(self->camera_config.pin_xclk);

    reset_pin_number(self->camera_config.pin_d7);
    reset_pin_number(self->camera_config.pin_d6);
    reset_pin_number(self->camera_config.pin_d5);
    reset_pin_number(self->camera_config.pin_d4);
    reset_pin_number(self->camera_config.pin_d3);
    reset_pin_number(self->camera_config.pin_d2);
    reset_pin_number(self->camera_config.pin_d1);
    reset_pin_number(self->camera_config.pin_d0);

    esp_camera_deinit();

    self->camera_config.xclk_freq_hz = 0;
}

bool common_hal_esp32_camera_camera_deinited(esp32_camera_camera_obj_t *self) {
    return !self->camera_config.xclk_freq_hz;
}

bool common_hal_esp32_camera_camera_available(esp32_camera_camera_obj_t *self) {
    return esp_camera_fb_available();
}

camera_fb_t *common_hal_esp32_camera_camera_take(esp32_camera_camera_obj_t *self, int timeout_ms) {
    if (self->buffer_to_return) {
        esp_camera_fb_return(self->buffer_to_return);
        self->buffer_to_return = NULL;
    }
    return esp_camera_fb_get_timeout(timeout_ms);
}