#include "h264_encoder.h"
#include "esp_camera.h"

// Camera config
camera_config_t config;
config.ledc_channel = LEDC_CHANNEL_0;
config.ledc_timer = LEDC_TIMER_0;
config.pin_d0 = Y2_GPIO_NUM;
config.pin_d1 = Y3_GPIO_NUM;
config.pin_d2 = Y4_GPIO_NUM;
config.pin_d3 = Y5_GPIO_NUM;
config.pin_d4 = Y6_GPIO_NUM;
config.pin_d5 = Y7_GPIO_NUM;
config.pin_d6 = Y8_GPIO_NUM;
config.pin_d7 = Y9_GPIO_NUM;
config.pin_xclk = XCLK_GPIO_NUM;
config.pin_pclk = PCLK_GPIO_NUM;
config.pin_vsync = VSYNC_GPIO_NUM;
config.pin_href = HREF_GPIO_NUM;
config.pin_sscb_sda = SIOD_GPIO_NUM;
config.pin_sscb_scl = SIOC_GPIO_NUM;
config.pin_pwdn = PWDN_GPIO_NUM;
config.pin_reset = RESET_GPIO_NUM;
config.xclk_freq_hz = 20000000;
config.pixel_format = PIXFORMAT_JPEG;
config.frame_size = FRAMESIZE_QVGA; //320x240
config.jpeg_quality = 12;
config.fb_count = 1;

void setup() {
  Serial.begin(115200);
  
  // Initialize camera
  if (esp_camera_init(&config) != ESP_OK) {
    Serial.println("Camera init failed");
    return;
  }

  // Initialize H.264 encoder
  h264_encoder_init(320, 240);

  Serial.println("Camera and H.264 encoder initialized");
}

void loop() {
  // Capture frame
  camera_fb_t *fb = esp_camera_fb_get();
  
  if (!fb) {
    Serial.println("Camera capture failed");
    return;
  }
  
  // Encode to H.264
  uint8_t *encoded_data;
  size_t encoded_size;
  if (h264_encoder_encode_frame(fb->buf, fb->len, &encoded_data, &encoded_size)) {
    Serial.println("Frame encoded successfully");
    // Handle encoded data, e.g., save or stream
  }

  esp_camera_fb_return(fb);
}