camera.c 49 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592
  1. // Copyright 2015-2016 Espressif Systems (Shanghai) PTE LTD
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. #include <stdio.h>
  15. #include <stdlib.h>
  16. #include <string.h>
  17. #include "time.h"
  18. #include "sys/time.h"
  19. #include "freertos/FreeRTOS.h"
  20. #include "freertos/task.h"
  21. #include "freertos/semphr.h"
  22. #include "soc/soc.h"
  23. #include "soc/gpio_sig_map.h"
  24. #include "soc/i2s_reg.h"
  25. #include "soc/i2s_struct.h"
  26. #include "soc/io_mux_reg.h"
  27. #include "driver/gpio.h"
  28. #include "driver/rtc_io.h"
  29. #include "driver/periph_ctrl.h"
  30. #include "esp_intr_alloc.h"
  31. #include "esp_system.h"
  32. #include "nvs_flash.h"
  33. #include "nvs.h"
  34. #include "sensor.h"
  35. #include "sccb.h"
  36. #include "esp_camera.h"
  37. #include "camera_common.h"
  38. #include "xclk.h"
  39. #if CONFIG_OV2640_SUPPORT
  40. #include "ov2640.h"
  41. #endif
  42. #if CONFIG_OV7725_SUPPORT
  43. #include "ov7725.h"
  44. #endif
  45. #if CONFIG_OV3660_SUPPORT
  46. #include "ov3660.h"
  47. #endif
  48. #if CONFIG_OV5640_SUPPORT
  49. #include "ov5640.h"
  50. #endif
  51. #if CONFIG_NT99141_SUPPORT
  52. #include "nt99141.h"
  53. #endif
  54. #if CONFIG_OV7670_SUPPORT
  55. #include "ov7670.h"
  56. #endif
  57. typedef enum {
  58. CAMERA_NONE = 0,
  59. CAMERA_UNKNOWN = 1,
  60. CAMERA_OV7725 = 7725,
  61. CAMERA_OV2640 = 2640,
  62. CAMERA_OV3660 = 3660,
  63. CAMERA_OV5640 = 5640,
  64. CAMERA_OV7670 = 7670,
  65. CAMERA_NT99141 = 9141,
  66. } camera_model_t;
  67. #define REG_PID 0x0A
  68. #define REG_VER 0x0B
  69. #define REG_MIDH 0x1C
  70. #define REG_MIDL 0x1D
  71. #define REG16_CHIDH 0x300A
  72. #define REG16_CHIDL 0x300B
  73. #if defined(ARDUINO_ARCH_ESP32) && defined(CONFIG_ARDUHAL_ESP_LOG)
  74. #include "esp32-hal-log.h"
  75. #define TAG ""
  76. #else
  77. #include "esp_log.h"
  78. static const char* TAG = "camera";
  79. #endif
  80. static const char* CAMERA_SENSOR_NVS_KEY = "sensor";
  81. static const char* CAMERA_PIXFORMAT_NVS_KEY = "pixformat";
  82. typedef void (*dma_filter_t)(const dma_elem_t* src, lldesc_t* dma_desc, uint8_t* dst);
  83. typedef struct camera_fb_s {
  84. uint8_t * buf;
  85. size_t len;
  86. size_t width;
  87. size_t height;
  88. pixformat_t format;
  89. struct timeval timestamp;
  90. size_t size;
  91. uint8_t ref;
  92. uint8_t bad;
  93. struct camera_fb_s * next;
  94. } camera_fb_int_t;
  95. typedef struct fb_s {
  96. uint8_t * buf;
  97. size_t len;
  98. struct fb_s * next;
  99. } fb_item_t;
  100. typedef struct {
  101. camera_config_t config;
  102. sensor_t sensor;
  103. camera_fb_int_t *fb;
  104. size_t fb_size;
  105. size_t data_size;
  106. size_t width;
  107. size_t height;
  108. size_t in_bytes_per_pixel;
  109. size_t fb_bytes_per_pixel;
  110. size_t dma_received_count;
  111. size_t dma_filtered_count;
  112. size_t dma_per_line;
  113. size_t dma_buf_width;
  114. size_t dma_sample_count;
  115. lldesc_t *dma_desc;
  116. dma_elem_t **dma_buf;
  117. size_t dma_desc_count;
  118. size_t dma_desc_cur;
  119. i2s_sampling_mode_t sampling_mode;
  120. dma_filter_t dma_filter;
  121. intr_handle_t i2s_intr_handle;
  122. QueueHandle_t data_ready;
  123. QueueHandle_t fb_in;
  124. QueueHandle_t fb_out;
  125. SemaphoreHandle_t frame_ready;
  126. TaskHandle_t dma_filter_task;
  127. } camera_state_t;
  128. camera_state_t* s_state = NULL;
  129. static void i2s_init();
  130. static int i2s_run();
  131. static void IRAM_ATTR vsync_isr(void* arg);
  132. static void IRAM_ATTR i2s_isr(void* arg);
  133. static esp_err_t dma_desc_init();
  134. static void dma_desc_deinit();
  135. static void dma_filter_task(void *pvParameters);
  136. static void dma_filter_grayscale(const dma_elem_t* src, lldesc_t* dma_desc, uint8_t* dst);
  137. static void dma_filter_grayscale_highspeed(const dma_elem_t* src, lldesc_t* dma_desc, uint8_t* dst);
  138. static void dma_filter_yuyv(const dma_elem_t* src, lldesc_t* dma_desc, uint8_t* dst);
  139. static void dma_filter_yuyv_highspeed(const dma_elem_t* src, lldesc_t* dma_desc, uint8_t* dst);
  140. static void dma_filter_jpeg(const dma_elem_t* src, lldesc_t* dma_desc, uint8_t* dst);
  141. static void i2s_stop(bool* need_yield);
  142. static bool is_hs_mode()
  143. {
  144. return s_state->config.xclk_freq_hz > 10000000;
  145. }
  146. static size_t i2s_bytes_per_sample(i2s_sampling_mode_t mode)
  147. {
  148. switch(mode) {
  149. case SM_0A00_0B00:
  150. return 4;
  151. case SM_0A0B_0B0C:
  152. return 4;
  153. case SM_0A0B_0C0D:
  154. return 2;
  155. default:
  156. assert(0 && "invalid sampling mode");
  157. return 0;
  158. }
  159. }
  160. static int IRAM_ATTR _gpio_get_level(gpio_num_t gpio_num)
  161. {
  162. if (gpio_num < 32) {
  163. return (GPIO.in >> gpio_num) & 0x1;
  164. } else {
  165. return (GPIO.in1.data >> (gpio_num - 32)) & 0x1;
  166. }
  167. }
  168. static void IRAM_ATTR vsync_intr_disable()
  169. {
  170. gpio_set_intr_type(s_state->config.pin_vsync, GPIO_INTR_DISABLE);
  171. }
  172. static void vsync_intr_enable()
  173. {
  174. gpio_set_intr_type(s_state->config.pin_vsync, GPIO_INTR_NEGEDGE);
  175. }
  176. static int skip_frame()
  177. {
  178. if (s_state == NULL) {
  179. return -1;
  180. }
  181. int64_t st_t = esp_timer_get_time();
  182. while (_gpio_get_level(s_state->config.pin_vsync) == 0) {
  183. if((esp_timer_get_time() - st_t) > 1000000LL){
  184. goto timeout;
  185. }
  186. }
  187. while (_gpio_get_level(s_state->config.pin_vsync) != 0) {
  188. if((esp_timer_get_time() - st_t) > 1000000LL){
  189. goto timeout;
  190. }
  191. }
  192. while (_gpio_get_level(s_state->config.pin_vsync) == 0) {
  193. if((esp_timer_get_time() - st_t) > 1000000LL){
  194. goto timeout;
  195. }
  196. }
  197. return 0;
  198. timeout:
  199. ESP_LOGE(TAG, "Timeout waiting for VSYNC");
  200. return -1;
  201. }
  202. static void camera_fb_deinit()
  203. {
  204. camera_fb_int_t * _fb1 = s_state->fb, * _fb2 = NULL;
  205. while(s_state->fb) {
  206. _fb2 = s_state->fb;
  207. s_state->fb = _fb2->next;
  208. if(_fb2->next == _fb1) {
  209. s_state->fb = NULL;
  210. }
  211. free(_fb2->buf);
  212. free(_fb2);
  213. }
  214. }
  215. static esp_err_t camera_fb_init(size_t count)
  216. {
  217. if(!count) {
  218. return ESP_ERR_INVALID_ARG;
  219. }
  220. camera_fb_deinit();
  221. ESP_LOGI(TAG, "Allocating %u frame buffers (%d KB total)", count, (s_state->fb_size * count) / 1024);
  222. camera_fb_int_t * _fb = NULL, * _fb1 = NULL, * _fb2 = NULL;
  223. for(size_t i = 0; i < count; i++) {
  224. _fb2 = (camera_fb_int_t *)malloc(sizeof(camera_fb_int_t));
  225. if(!_fb2) {
  226. goto fail;
  227. }
  228. memset(_fb2, 0, sizeof(camera_fb_int_t));
  229. _fb2->size = s_state->fb_size;
  230. _fb2->buf = (uint8_t*) calloc(_fb2->size, 1);
  231. if(!_fb2->buf) {
  232. ESP_LOGI(TAG, "Allocating %d KB frame buffer in PSRAM", s_state->fb_size/1024);
  233. _fb2->buf = (uint8_t*) heap_caps_calloc(_fb2->size, 1, MALLOC_CAP_SPIRAM | MALLOC_CAP_8BIT);
  234. } else {
  235. ESP_LOGI(TAG, "Allocating %d KB frame buffer in OnBoard RAM", s_state->fb_size/1024);
  236. }
  237. if(!_fb2->buf) {
  238. free(_fb2);
  239. ESP_LOGE(TAG, "Allocating %d KB frame buffer Failed", s_state->fb_size/1024);
  240. goto fail;
  241. }
  242. memset(_fb2->buf, 0, _fb2->size);
  243. _fb2->next = _fb;
  244. _fb = _fb2;
  245. if(!i) {
  246. _fb1 = _fb2;
  247. }
  248. }
  249. if(_fb1) {
  250. _fb1->next = _fb;
  251. }
  252. s_state->fb = _fb;//load first buffer
  253. return ESP_OK;
  254. fail:
  255. while(_fb) {
  256. _fb2 = _fb;
  257. _fb = _fb->next;
  258. free(_fb2->buf);
  259. free(_fb2);
  260. }
  261. return ESP_ERR_NO_MEM;
  262. }
  263. static esp_err_t dma_desc_init()
  264. {
  265. assert(s_state->width % 4 == 0);
  266. size_t line_size = s_state->width * s_state->in_bytes_per_pixel *
  267. i2s_bytes_per_sample(s_state->sampling_mode);
  268. ESP_LOGD(TAG, "Line width (for DMA): %d bytes", line_size);
  269. size_t dma_per_line = 1;
  270. size_t buf_size = line_size;
  271. while (buf_size >= 4096) {
  272. buf_size /= 2;
  273. dma_per_line *= 2;
  274. }
  275. size_t dma_desc_count = dma_per_line * 4;
  276. s_state->dma_buf_width = line_size;
  277. s_state->dma_per_line = dma_per_line;
  278. s_state->dma_desc_count = dma_desc_count;
  279. ESP_LOGD(TAG, "DMA buffer size: %d, DMA buffers per line: %d", buf_size, dma_per_line);
  280. ESP_LOGD(TAG, "DMA buffer count: %d", dma_desc_count);
  281. ESP_LOGD(TAG, "DMA buffer total: %d bytes", buf_size * dma_desc_count);
  282. s_state->dma_buf = (dma_elem_t**) malloc(sizeof(dma_elem_t*) * dma_desc_count);
  283. if (s_state->dma_buf == NULL) {
  284. return ESP_ERR_NO_MEM;
  285. }
  286. s_state->dma_desc = (lldesc_t*) malloc(sizeof(lldesc_t) * dma_desc_count);
  287. if (s_state->dma_desc == NULL) {
  288. return ESP_ERR_NO_MEM;
  289. }
  290. size_t dma_sample_count = 0;
  291. for (int i = 0; i < dma_desc_count; ++i) {
  292. ESP_LOGD(TAG, "Allocating DMA buffer #%d, size=%d", i, buf_size);
  293. dma_elem_t* buf = (dma_elem_t*) malloc(buf_size);
  294. if (buf == NULL) {
  295. return ESP_ERR_NO_MEM;
  296. }
  297. s_state->dma_buf[i] = buf;
  298. ESP_LOGV(TAG, "dma_buf[%d]=%p", i, buf);
  299. lldesc_t* pd = &s_state->dma_desc[i];
  300. pd->length = buf_size;
  301. if (s_state->sampling_mode == SM_0A0B_0B0C &&
  302. (i + 1) % dma_per_line == 0) {
  303. pd->length -= 4;
  304. }
  305. dma_sample_count += pd->length / 4;
  306. pd->size = pd->length;
  307. pd->owner = 1;
  308. pd->sosf = 1;
  309. pd->buf = (uint8_t*) buf;
  310. pd->offset = 0;
  311. pd->empty = 0;
  312. pd->eof = 1;
  313. pd->qe.stqe_next = &s_state->dma_desc[(i + 1) % dma_desc_count];
  314. }
  315. s_state->dma_sample_count = dma_sample_count;
  316. return ESP_OK;
  317. }
  318. static void dma_desc_deinit()
  319. {
  320. if (s_state->dma_buf) {
  321. for (int i = 0; i < s_state->dma_desc_count; ++i) {
  322. free(s_state->dma_buf[i]);
  323. }
  324. }
  325. free(s_state->dma_buf);
  326. free(s_state->dma_desc);
  327. }
  328. static inline void IRAM_ATTR i2s_conf_reset()
  329. {
  330. const uint32_t lc_conf_reset_flags = I2S_IN_RST_M | I2S_AHBM_RST_M
  331. | I2S_AHBM_FIFO_RST_M;
  332. I2S0.lc_conf.val |= lc_conf_reset_flags;
  333. I2S0.lc_conf.val &= ~lc_conf_reset_flags;
  334. const uint32_t conf_reset_flags = I2S_RX_RESET_M | I2S_RX_FIFO_RESET_M
  335. | I2S_TX_RESET_M | I2S_TX_FIFO_RESET_M;
  336. I2S0.conf.val |= conf_reset_flags;
  337. I2S0.conf.val &= ~conf_reset_flags;
  338. while (I2S0.state.rx_fifo_reset_back) {
  339. ;
  340. }
  341. }
  342. static void i2s_gpio_init(const camera_config_t* config)
  343. {
  344. // Configure input GPIOs
  345. const gpio_num_t pins[] = {
  346. config->pin_d7,
  347. config->pin_d6,
  348. config->pin_d5,
  349. config->pin_d4,
  350. config->pin_d3,
  351. config->pin_d2,
  352. config->pin_d1,
  353. config->pin_d0,
  354. config->pin_vsync,
  355. config->pin_href,
  356. config->pin_pclk
  357. };
  358. gpio_config_t conf = {
  359. .mode = GPIO_MODE_INPUT,
  360. .pull_up_en = GPIO_PULLUP_ENABLE,
  361. .pull_down_en = GPIO_PULLDOWN_DISABLE,
  362. .intr_type = GPIO_INTR_DISABLE,
  363. .pin_bit_mask = 0LL
  364. };
  365. for (int i = 0; i < sizeof(pins) / sizeof(gpio_num_t); ++i) {
  366. if (rtc_gpio_is_valid_gpio(pins[i])) {
  367. rtc_gpio_deinit(pins[i]);
  368. }
  369. conf.pin_bit_mask |= 1LL << pins[i];
  370. }
  371. gpio_config(&conf);
  372. }
  373. static void i2s_init()
  374. {
  375. camera_config_t* config = &s_state->config;
  376. // Route input GPIOs to I2S peripheral using GPIO matrix
  377. gpio_matrix_in(config->pin_d0, I2S0I_DATA_IN0_IDX, false);
  378. gpio_matrix_in(config->pin_d1, I2S0I_DATA_IN1_IDX, false);
  379. gpio_matrix_in(config->pin_d2, I2S0I_DATA_IN2_IDX, false);
  380. gpio_matrix_in(config->pin_d3, I2S0I_DATA_IN3_IDX, false);
  381. gpio_matrix_in(config->pin_d4, I2S0I_DATA_IN4_IDX, false);
  382. gpio_matrix_in(config->pin_d5, I2S0I_DATA_IN5_IDX, false);
  383. gpio_matrix_in(config->pin_d6, I2S0I_DATA_IN6_IDX, false);
  384. gpio_matrix_in(config->pin_d7, I2S0I_DATA_IN7_IDX, false);
  385. gpio_matrix_in(config->pin_vsync, I2S0I_V_SYNC_IDX, false);
  386. gpio_matrix_in(0x38, I2S0I_H_SYNC_IDX, false);
  387. gpio_matrix_in(config->pin_href, I2S0I_H_ENABLE_IDX, false);
  388. gpio_matrix_in(config->pin_pclk, I2S0I_WS_IN_IDX, false);
  389. // Enable and configure I2S peripheral
  390. periph_module_enable(PERIPH_I2S0_MODULE);
  391. // Toggle some reset bits in LC_CONF register
  392. // Toggle some reset bits in CONF register
  393. i2s_conf_reset();
  394. // Enable slave mode (sampling clock is external)
  395. I2S0.conf.rx_slave_mod = 1;
  396. // Enable parallel mode
  397. I2S0.conf2.lcd_en = 1;
  398. // Use HSYNC/VSYNC/HREF to control sampling
  399. I2S0.conf2.camera_en = 1;
  400. // Configure clock divider
  401. I2S0.clkm_conf.clkm_div_a = 1;
  402. I2S0.clkm_conf.clkm_div_b = 0;
  403. I2S0.clkm_conf.clkm_div_num = 2;
  404. // FIFO will sink data to DMA
  405. I2S0.fifo_conf.dscr_en = 1;
  406. // FIFO configuration
  407. I2S0.fifo_conf.rx_fifo_mod = s_state->sampling_mode;
  408. I2S0.fifo_conf.rx_fifo_mod_force_en = 1;
  409. I2S0.conf_chan.rx_chan_mod = 1;
  410. // Clear flags which are used in I2S serial mode
  411. I2S0.sample_rate_conf.rx_bits_mod = 0;
  412. I2S0.conf.rx_right_first = 0;
  413. I2S0.conf.rx_msb_right = 0;
  414. I2S0.conf.rx_msb_shift = 0;
  415. I2S0.conf.rx_mono = 0;
  416. I2S0.conf.rx_short_sync = 0;
  417. I2S0.timing.val = 0;
  418. I2S0.timing.rx_dsync_sw = 1;
  419. // Allocate I2S interrupt, keep it disabled
  420. ESP_ERROR_CHECK(esp_intr_alloc(ETS_I2S0_INTR_SOURCE,
  421. ESP_INTR_FLAG_INTRDISABLED | ESP_INTR_FLAG_LOWMED | ESP_INTR_FLAG_IRAM,
  422. &i2s_isr, NULL, &s_state->i2s_intr_handle));
  423. }
  424. static void IRAM_ATTR i2s_start_bus()
  425. {
  426. s_state->dma_desc_cur = 0;
  427. s_state->dma_received_count = 0;
  428. //s_state->dma_filtered_count = 0;
  429. esp_intr_disable(s_state->i2s_intr_handle);
  430. i2s_conf_reset();
  431. I2S0.rx_eof_num = s_state->dma_sample_count;
  432. I2S0.in_link.addr = (uint32_t) &s_state->dma_desc[0];
  433. I2S0.in_link.start = 1;
  434. I2S0.int_clr.val = I2S0.int_raw.val;
  435. I2S0.int_ena.val = 0;
  436. I2S0.int_ena.in_done = 1;
  437. esp_intr_enable(s_state->i2s_intr_handle);
  438. I2S0.conf.rx_start = 1;
  439. if (s_state->config.pixel_format == PIXFORMAT_JPEG) {
  440. vsync_intr_enable();
  441. }
  442. }
  443. static int i2s_run()
  444. {
  445. for (int i = 0; i < s_state->dma_desc_count; ++i) {
  446. lldesc_t* d = &s_state->dma_desc[i];
  447. ESP_LOGV(TAG, "DMA desc %2d: %u %u %u %u %u %u %p %p",
  448. i, d->length, d->size, d->offset, d->eof, d->sosf, d->owner, d->buf, d->qe.stqe_next);
  449. memset(s_state->dma_buf[i], 0, d->length);
  450. }
  451. // wait for frame
  452. camera_fb_int_t * fb = s_state->fb;
  453. while(s_state->config.fb_count > 1) {
  454. while(s_state->fb->ref && s_state->fb->next != fb) {
  455. s_state->fb = s_state->fb->next;
  456. }
  457. if(s_state->fb->ref == 0) {
  458. break;
  459. }
  460. vTaskDelay(2);
  461. }
  462. //todo: wait for vsync
  463. ESP_LOGV(TAG, "Waiting for negative edge on VSYNC");
  464. int64_t st_t = esp_timer_get_time();
  465. while (_gpio_get_level(s_state->config.pin_vsync) != 0) {
  466. if((esp_timer_get_time() - st_t) > 1000000LL){
  467. ESP_LOGE(TAG, "Timeout waiting for VSYNC");
  468. return -1;
  469. }
  470. }
  471. ESP_LOGV(TAG, "Got VSYNC");
  472. i2s_start_bus();
  473. return 0;
  474. }
  475. static void IRAM_ATTR i2s_stop_bus()
  476. {
  477. esp_intr_disable(s_state->i2s_intr_handle);
  478. vsync_intr_disable();
  479. i2s_conf_reset();
  480. I2S0.conf.rx_start = 0;
  481. }
  482. static void IRAM_ATTR i2s_stop(bool* need_yield)
  483. {
  484. if(s_state->config.fb_count == 1 && !s_state->fb->bad) {
  485. i2s_stop_bus();
  486. } else {
  487. s_state->dma_received_count = 0;
  488. }
  489. size_t val = SIZE_MAX;
  490. BaseType_t higher_priority_task_woken;
  491. BaseType_t ret = xQueueSendFromISR(s_state->data_ready, &val, &higher_priority_task_woken);
  492. if(need_yield && !*need_yield) {
  493. *need_yield = (ret == pdTRUE && higher_priority_task_woken == pdTRUE);
  494. }
  495. }
  496. static void IRAM_ATTR signal_dma_buf_received(bool* need_yield)
  497. {
  498. size_t dma_desc_filled = s_state->dma_desc_cur;
  499. s_state->dma_desc_cur = (dma_desc_filled + 1) % s_state->dma_desc_count;
  500. s_state->dma_received_count++;
  501. if(!s_state->fb->ref && s_state->fb->bad){
  502. *need_yield = false;
  503. return;
  504. }
  505. BaseType_t higher_priority_task_woken;
  506. BaseType_t ret = xQueueSendFromISR(s_state->data_ready, &dma_desc_filled, &higher_priority_task_woken);
  507. if (ret != pdTRUE) {
  508. if(!s_state->fb->ref) {
  509. s_state->fb->bad = 1;
  510. }
  511. //ESP_EARLY_LOGW(TAG, "qsf:%d", s_state->dma_received_count);
  512. //ets_printf("qsf:%d\n", s_state->dma_received_count);
  513. //ets_printf("qovf\n");
  514. }
  515. *need_yield = (ret == pdTRUE && higher_priority_task_woken == pdTRUE);
  516. }
  517. static void IRAM_ATTR i2s_isr(void* arg)
  518. {
  519. I2S0.int_clr.val = I2S0.int_raw.val;
  520. bool need_yield = false;
  521. signal_dma_buf_received(&need_yield);
  522. if (s_state->config.pixel_format != PIXFORMAT_JPEG
  523. && s_state->dma_received_count == s_state->height * s_state->dma_per_line) {
  524. i2s_stop(&need_yield);
  525. }
  526. if (need_yield) {
  527. portYIELD_FROM_ISR();
  528. }
  529. }
  530. static void IRAM_ATTR vsync_isr(void* arg)
  531. {
  532. GPIO.status1_w1tc.val = GPIO.status1.val;
  533. GPIO.status_w1tc = GPIO.status;
  534. bool need_yield = false;
  535. //if vsync is low and we have received some data, frame is done
  536. if (_gpio_get_level(s_state->config.pin_vsync) == 0) {
  537. if(s_state->dma_received_count > 0) {
  538. signal_dma_buf_received(&need_yield);
  539. //ets_printf("end_vsync\n");
  540. if(s_state->dma_filtered_count > 1 || s_state->fb->bad || s_state->config.fb_count > 1) {
  541. i2s_stop(&need_yield);
  542. }
  543. //ets_printf("vs\n");
  544. }
  545. if(s_state->config.fb_count > 1 || s_state->dma_filtered_count < 2) {
  546. I2S0.conf.rx_start = 0;
  547. I2S0.in_link.start = 0;
  548. I2S0.int_clr.val = I2S0.int_raw.val;
  549. i2s_conf_reset();
  550. s_state->dma_desc_cur = (s_state->dma_desc_cur + 1) % s_state->dma_desc_count;
  551. //I2S0.rx_eof_num = s_state->dma_sample_count;
  552. I2S0.in_link.addr = (uint32_t) &s_state->dma_desc[s_state->dma_desc_cur];
  553. I2S0.in_link.start = 1;
  554. I2S0.conf.rx_start = 1;
  555. s_state->dma_received_count = 0;
  556. }
  557. }
  558. if (need_yield) {
  559. portYIELD_FROM_ISR();
  560. }
  561. }
  562. static void IRAM_ATTR camera_fb_done()
  563. {
  564. camera_fb_int_t * fb = NULL, * fb2 = NULL;
  565. BaseType_t taskAwoken = 0;
  566. if(s_state->config.fb_count == 1) {
  567. xSemaphoreGive(s_state->frame_ready);
  568. return;
  569. }
  570. fb = s_state->fb;
  571. if(!fb->ref && fb->len) {
  572. //add reference
  573. fb->ref = 1;
  574. //check if the queue is full
  575. if(xQueueIsQueueFullFromISR(s_state->fb_out) == pdTRUE) {
  576. //pop frame buffer from the queue
  577. if(xQueueReceiveFromISR(s_state->fb_out, &fb2, &taskAwoken) == pdTRUE) {
  578. //free the popped buffer
  579. fb2->ref = 0;
  580. fb2->len = 0;
  581. //push the new frame to the end of the queue
  582. xQueueSendFromISR(s_state->fb_out, &fb, &taskAwoken);
  583. } else {
  584. //queue is full and we could not pop a frame from it
  585. }
  586. } else {
  587. //push the new frame to the end of the queue
  588. xQueueSendFromISR(s_state->fb_out, &fb, &taskAwoken);
  589. }
  590. } else {
  591. //frame was referenced or empty
  592. }
  593. //return buffers to be filled
  594. while(xQueueReceiveFromISR(s_state->fb_in, &fb2, &taskAwoken) == pdTRUE) {
  595. fb2->ref = 0;
  596. fb2->len = 0;
  597. }
  598. //advance frame buffer only if the current one has data
  599. if(s_state->fb->len) {
  600. s_state->fb = s_state->fb->next;
  601. }
  602. //try to find the next free frame buffer
  603. while(s_state->fb->ref && s_state->fb->next != fb) {
  604. s_state->fb = s_state->fb->next;
  605. }
  606. //is the found frame buffer free?
  607. if(!s_state->fb->ref) {
  608. //buffer found. make sure it's empty
  609. s_state->fb->len = 0;
  610. *((uint32_t *)s_state->fb->buf) = 0;
  611. } else {
  612. //stay at the previous buffer
  613. s_state->fb = fb;
  614. }
  615. }
  616. static void IRAM_ATTR dma_finish_frame()
  617. {
  618. size_t buf_len = s_state->width * s_state->fb_bytes_per_pixel / s_state->dma_per_line;
  619. if(!s_state->fb->ref) {
  620. // is the frame bad?
  621. if(s_state->fb->bad){
  622. s_state->fb->bad = 0;
  623. s_state->fb->len = 0;
  624. *((uint32_t *)s_state->fb->buf) = 0;
  625. if(s_state->config.fb_count == 1) {
  626. i2s_start_bus();
  627. }
  628. //ets_printf("bad\n");
  629. } else {
  630. s_state->fb->len = s_state->dma_filtered_count * buf_len;
  631. if(s_state->fb->len) {
  632. //find the end marker for JPEG. Data after that can be discarded
  633. if(s_state->fb->format == PIXFORMAT_JPEG){
  634. uint8_t * dptr = &s_state->fb->buf[s_state->fb->len - 1];
  635. while(dptr > s_state->fb->buf){
  636. if(dptr[0] == 0xFF && dptr[1] == 0xD9 && dptr[2] == 0x00 && dptr[3] == 0x00){
  637. dptr += 2;
  638. s_state->fb->len = dptr - s_state->fb->buf;
  639. if((s_state->fb->len & 0x1FF) == 0){
  640. s_state->fb->len += 1;
  641. }
  642. if((s_state->fb->len % 100) == 0){
  643. s_state->fb->len += 1;
  644. }
  645. break;
  646. }
  647. dptr--;
  648. }
  649. }
  650. //send out the frame
  651. camera_fb_done();
  652. } else if(s_state->config.fb_count == 1){
  653. //frame was empty?
  654. i2s_start_bus();
  655. } else {
  656. //ets_printf("empty\n");
  657. }
  658. }
  659. } else if(s_state->fb->len) {
  660. camera_fb_done();
  661. }
  662. s_state->dma_filtered_count = 0;
  663. }
  664. static void IRAM_ATTR dma_filter_buffer(size_t buf_idx)
  665. {
  666. //no need to process the data if frame is in use or is bad
  667. if(s_state->fb->ref || s_state->fb->bad) {
  668. return;
  669. }
  670. //check if there is enough space in the frame buffer for the new data
  671. size_t buf_len = s_state->width * s_state->fb_bytes_per_pixel / s_state->dma_per_line;
  672. size_t fb_pos = s_state->dma_filtered_count * buf_len;
  673. if(fb_pos > s_state->fb_size - buf_len) {
  674. //size_t processed = s_state->dma_received_count * buf_len;
  675. //ets_printf("[%s:%u] ovf pos: %u, processed: %u\n", __FUNCTION__, __LINE__, fb_pos, processed);
  676. return;
  677. }
  678. //convert I2S DMA buffer to pixel data
  679. (*s_state->dma_filter)(s_state->dma_buf[buf_idx], &s_state->dma_desc[buf_idx], s_state->fb->buf + fb_pos);
  680. //first frame buffer
  681. if(!s_state->dma_filtered_count) {
  682. //check for correct JPEG header
  683. if(s_state->sensor.pixformat == PIXFORMAT_JPEG) {
  684. uint32_t sig = *((uint32_t *)s_state->fb->buf) & 0xFFFFFF;
  685. if(sig != 0xffd8ff) {
  686. ESP_LOGD(TAG,"unexpected JPEG signature 0x%08x\n", sig);
  687. s_state->fb->bad = 1;
  688. return;
  689. }
  690. }
  691. //set the frame properties
  692. s_state->fb->width = resolution[s_state->sensor.status.framesize].width;
  693. s_state->fb->height = resolution[s_state->sensor.status.framesize].height;
  694. s_state->fb->format = s_state->sensor.pixformat;
  695. uint64_t us = (uint64_t)esp_timer_get_time();
  696. s_state->fb->timestamp.tv_sec = us / 1000000UL;
  697. s_state->fb->timestamp.tv_usec = us % 1000000UL;
  698. }
  699. s_state->dma_filtered_count++;
  700. }
  701. static void IRAM_ATTR dma_filter_task(void *pvParameters)
  702. {
  703. s_state->dma_filtered_count = 0;
  704. while (true) {
  705. size_t buf_idx;
  706. if(xQueueReceive(s_state->data_ready, &buf_idx, portMAX_DELAY) == pdTRUE) {
  707. if (buf_idx == SIZE_MAX) {
  708. //this is the end of the frame
  709. dma_finish_frame();
  710. } else {
  711. dma_filter_buffer(buf_idx);
  712. }
  713. }
  714. }
  715. }
  716. static void IRAM_ATTR dma_filter_jpeg(const dma_elem_t* src, lldesc_t* dma_desc, uint8_t* dst)
  717. {
  718. size_t end = dma_desc->length / sizeof(dma_elem_t) / 4;
  719. // manually unrolling 4 iterations of the loop here
  720. for (size_t i = 0; i < end; ++i) {
  721. dst[0] = src[0].sample1;
  722. dst[1] = src[1].sample1;
  723. dst[2] = src[2].sample1;
  724. dst[3] = src[3].sample1;
  725. src += 4;
  726. dst += 4;
  727. }
  728. }
  729. static void IRAM_ATTR dma_filter_grayscale(const dma_elem_t* src, lldesc_t* dma_desc, uint8_t* dst)
  730. {
  731. size_t end = dma_desc->length / sizeof(dma_elem_t) / 4;
  732. for (size_t i = 0; i < end; ++i) {
  733. // manually unrolling 4 iterations of the loop here
  734. dst[0] = src[0].sample1;
  735. dst[1] = src[1].sample1;
  736. dst[2] = src[2].sample1;
  737. dst[3] = src[3].sample1;
  738. src += 4;
  739. dst += 4;
  740. }
  741. }
  742. static void IRAM_ATTR dma_filter_grayscale_highspeed(const dma_elem_t* src, lldesc_t* dma_desc, uint8_t* dst)
  743. {
  744. size_t end = dma_desc->length / sizeof(dma_elem_t) / 8;
  745. for (size_t i = 0; i < end; ++i) {
  746. // manually unrolling 4 iterations of the loop here
  747. dst[0] = src[0].sample1;
  748. dst[1] = src[2].sample1;
  749. dst[2] = src[4].sample1;
  750. dst[3] = src[6].sample1;
  751. src += 8;
  752. dst += 4;
  753. }
  754. // the final sample of a line in SM_0A0B_0B0C sampling mode needs special handling
  755. if ((dma_desc->length & 0x7) != 0) {
  756. dst[0] = src[0].sample1;
  757. dst[1] = src[2].sample1;
  758. }
  759. }
  760. static void IRAM_ATTR dma_filter_yuyv(const dma_elem_t* src, lldesc_t* dma_desc, uint8_t* dst)
  761. {
  762. size_t end = dma_desc->length / sizeof(dma_elem_t) / 4;
  763. for (size_t i = 0; i < end; ++i) {
  764. dst[0] = src[0].sample1;//y0
  765. dst[1] = src[0].sample2;//u
  766. dst[2] = src[1].sample1;//y1
  767. dst[3] = src[1].sample2;//v
  768. dst[4] = src[2].sample1;//y0
  769. dst[5] = src[2].sample2;//u
  770. dst[6] = src[3].sample1;//y1
  771. dst[7] = src[3].sample2;//v
  772. src += 4;
  773. dst += 8;
  774. }
  775. }
  776. static void IRAM_ATTR dma_filter_yuyv_highspeed(const dma_elem_t* src, lldesc_t* dma_desc, uint8_t* dst)
  777. {
  778. size_t end = dma_desc->length / sizeof(dma_elem_t) / 8;
  779. for (size_t i = 0; i < end; ++i) {
  780. dst[0] = src[0].sample1;//y0
  781. dst[1] = src[1].sample1;//u
  782. dst[2] = src[2].sample1;//y1
  783. dst[3] = src[3].sample1;//v
  784. dst[4] = src[4].sample1;//y0
  785. dst[5] = src[5].sample1;//u
  786. dst[6] = src[6].sample1;//y1
  787. dst[7] = src[7].sample1;//v
  788. src += 8;
  789. dst += 8;
  790. }
  791. if ((dma_desc->length & 0x7) != 0) {
  792. dst[0] = src[0].sample1;//y0
  793. dst[1] = src[1].sample1;//u
  794. dst[2] = src[2].sample1;//y1
  795. dst[3] = src[2].sample2;//v
  796. }
  797. }
  798. static void IRAM_ATTR dma_filter_rgb888(const dma_elem_t* src, lldesc_t* dma_desc, uint8_t* dst)
  799. {
  800. size_t end = dma_desc->length / sizeof(dma_elem_t) / 4;
  801. uint8_t lb, hb;
  802. for (size_t i = 0; i < end; ++i) {
  803. hb = src[0].sample1;
  804. lb = src[0].sample2;
  805. dst[0] = (lb & 0x1F) << 3;
  806. dst[1] = (hb & 0x07) << 5 | (lb & 0xE0) >> 3;
  807. dst[2] = hb & 0xF8;
  808. hb = src[1].sample1;
  809. lb = src[1].sample2;
  810. dst[3] = (lb & 0x1F) << 3;
  811. dst[4] = (hb & 0x07) << 5 | (lb & 0xE0) >> 3;
  812. dst[5] = hb & 0xF8;
  813. hb = src[2].sample1;
  814. lb = src[2].sample2;
  815. dst[6] = (lb & 0x1F) << 3;
  816. dst[7] = (hb & 0x07) << 5 | (lb & 0xE0) >> 3;
  817. dst[8] = hb & 0xF8;
  818. hb = src[3].sample1;
  819. lb = src[3].sample2;
  820. dst[9] = (lb & 0x1F) << 3;
  821. dst[10] = (hb & 0x07) << 5 | (lb & 0xE0) >> 3;
  822. dst[11] = hb & 0xF8;
  823. src += 4;
  824. dst += 12;
  825. }
  826. }
  827. static void IRAM_ATTR dma_filter_rgb888_highspeed(const dma_elem_t* src, lldesc_t* dma_desc, uint8_t* dst)
  828. {
  829. size_t end = dma_desc->length / sizeof(dma_elem_t) / 8;
  830. uint8_t lb, hb;
  831. for (size_t i = 0; i < end; ++i) {
  832. hb = src[0].sample1;
  833. lb = src[1].sample1;
  834. dst[0] = (lb & 0x1F) << 3;
  835. dst[1] = (hb & 0x07) << 5 | (lb & 0xE0) >> 3;
  836. dst[2] = hb & 0xF8;
  837. hb = src[2].sample1;
  838. lb = src[3].sample1;
  839. dst[3] = (lb & 0x1F) << 3;
  840. dst[4] = (hb & 0x07) << 5 | (lb & 0xE0) >> 3;
  841. dst[5] = hb & 0xF8;
  842. hb = src[4].sample1;
  843. lb = src[5].sample1;
  844. dst[6] = (lb & 0x1F) << 3;
  845. dst[7] = (hb & 0x07) << 5 | (lb & 0xE0) >> 3;
  846. dst[8] = hb & 0xF8;
  847. hb = src[6].sample1;
  848. lb = src[7].sample1;
  849. dst[9] = (lb & 0x1F) << 3;
  850. dst[10] = (hb & 0x07) << 5 | (lb & 0xE0) >> 3;
  851. dst[11] = hb & 0xF8;
  852. src += 8;
  853. dst += 12;
  854. }
  855. if ((dma_desc->length & 0x7) != 0) {
  856. hb = src[0].sample1;
  857. lb = src[1].sample1;
  858. dst[0] = (lb & 0x1F) << 3;
  859. dst[1] = (hb & 0x07) << 5 | (lb & 0xE0) >> 3;
  860. dst[2] = hb & 0xF8;
  861. hb = src[2].sample1;
  862. lb = src[2].sample2;
  863. dst[3] = (lb & 0x1F) << 3;
  864. dst[4] = (hb & 0x07) << 5 | (lb & 0xE0) >> 3;
  865. dst[5] = hb & 0xF8;
  866. }
  867. }
  868. /*
  869. * Public Methods
  870. * */
  871. esp_err_t camera_probe(const camera_config_t* config, camera_model_t* out_camera_model)
  872. {
  873. if (s_state != NULL) {
  874. return ESP_ERR_INVALID_STATE;
  875. }
  876. s_state = (camera_state_t*) calloc(sizeof(*s_state), 1);
  877. if (!s_state) {
  878. return ESP_ERR_NO_MEM;
  879. }
  880. if(config->pin_xclk >= 0) {
  881. ESP_LOGD(TAG, "Enabling XCLK output");
  882. camera_enable_out_clock(config);
  883. }
  884. if (config->pin_sscb_sda != -1) {
  885. ESP_LOGD(TAG, "Initializing SSCB");
  886. SCCB_Init(config->pin_sscb_sda, config->pin_sscb_scl);
  887. }
  888. if(config->pin_pwdn >= 0) {
  889. ESP_LOGD(TAG, "Resetting camera by power down line");
  890. gpio_config_t conf = { 0 };
  891. conf.pin_bit_mask = 1LL << config->pin_pwdn;
  892. conf.mode = GPIO_MODE_OUTPUT;
  893. gpio_config(&conf);
  894. // carefull, logic is inverted compared to reset pin
  895. gpio_set_level(config->pin_pwdn, 1);
  896. vTaskDelay(10 / portTICK_PERIOD_MS);
  897. gpio_set_level(config->pin_pwdn, 0);
  898. vTaskDelay(10 / portTICK_PERIOD_MS);
  899. }
  900. if(config->pin_reset >= 0) {
  901. ESP_LOGD(TAG, "Resetting camera");
  902. gpio_config_t conf = { 0 };
  903. conf.pin_bit_mask = 1LL << config->pin_reset;
  904. conf.mode = GPIO_MODE_OUTPUT;
  905. gpio_config(&conf);
  906. gpio_set_level(config->pin_reset, 0);
  907. vTaskDelay(10 / portTICK_PERIOD_MS);
  908. gpio_set_level(config->pin_reset, 1);
  909. vTaskDelay(10 / portTICK_PERIOD_MS);
  910. }
  911. ESP_LOGD(TAG, "Searching for camera address");
  912. vTaskDelay(10 / portTICK_PERIOD_MS);
  913. uint8_t slv_addr = SCCB_Probe();
  914. if (slv_addr == 0) {
  915. *out_camera_model = CAMERA_NONE;
  916. camera_disable_out_clock();
  917. return ESP_ERR_CAMERA_NOT_DETECTED;
  918. }
  919. //slv_addr = 0x30;
  920. ESP_LOGD(TAG, "Detected camera at address=0x%02x", slv_addr);
  921. sensor_id_t* id = &s_state->sensor.id;
  922. #if CONFIG_OV2640_SUPPORT
  923. if (slv_addr == 0x30) {
  924. ESP_LOGD(TAG, "Resetting OV2640");
  925. //camera might be OV2640. try to reset it
  926. SCCB_Write(0x30, 0xFF, 0x01);//bank sensor
  927. SCCB_Write(0x30, 0x12, 0x80);//reset
  928. vTaskDelay(10 / portTICK_PERIOD_MS);
  929. slv_addr = SCCB_Probe();
  930. }
  931. #endif
  932. #if CONFIG_NT99141_SUPPORT
  933. if (slv_addr == 0x2a)
  934. {
  935. ESP_LOGD(TAG, "Resetting NT99141");
  936. SCCB_Write16(0x2a, 0x3008, 0x01);//bank sensor
  937. }
  938. #endif
  939. s_state->sensor.slv_addr = slv_addr;
  940. s_state->sensor.xclk_freq_hz = config->xclk_freq_hz;
  941. #if (CONFIG_OV3660_SUPPORT || CONFIG_OV5640_SUPPORT || CONFIG_NT99141_SUPPORT)
  942. if(s_state->sensor.slv_addr == 0x3c){
  943. id->PID = SCCB_Read16(s_state->sensor.slv_addr, REG16_CHIDH);
  944. id->VER = SCCB_Read16(s_state->sensor.slv_addr, REG16_CHIDL);
  945. vTaskDelay(10 / portTICK_PERIOD_MS);
  946. ESP_LOGD(TAG, "Camera PID=0x%02x VER=0x%02x", id->PID, id->VER);
  947. } else if(s_state->sensor.slv_addr == 0x2a){
  948. id->PID = SCCB_Read16(s_state->sensor.slv_addr, 0x3000);
  949. id->VER = SCCB_Read16(s_state->sensor.slv_addr, 0x3001);
  950. vTaskDelay(10 / portTICK_PERIOD_MS);
  951. ESP_LOGD(TAG, "Camera PID=0x%02x VER=0x%02x", id->PID, id->VER);
  952. if(config->xclk_freq_hz > 10000000)
  953. {
  954. ESP_LOGE(TAG, "NT99141: only XCLK under 10MHz is supported, and XCLK is now set to 10M");
  955. s_state->sensor.xclk_freq_hz = 10000000;
  956. }
  957. } else {
  958. #endif
  959. id->PID = SCCB_Read(s_state->sensor.slv_addr, REG_PID);
  960. id->VER = SCCB_Read(s_state->sensor.slv_addr, REG_VER);
  961. id->MIDL = SCCB_Read(s_state->sensor.slv_addr, REG_MIDL);
  962. id->MIDH = SCCB_Read(s_state->sensor.slv_addr, REG_MIDH);
  963. vTaskDelay(10 / portTICK_PERIOD_MS);
  964. ESP_LOGD(TAG, "Camera PID=0x%02x VER=0x%02x MIDL=0x%02x MIDH=0x%02x",
  965. id->PID, id->VER, id->MIDH, id->MIDL);
  966. #if (CONFIG_OV3660_SUPPORT || CONFIG_OV5640_SUPPORT || CONFIG_NT99141_SUPPORT)
  967. }
  968. #endif
  969. switch (id->PID) {
  970. #if CONFIG_OV2640_SUPPORT
  971. case OV2640_PID:
  972. *out_camera_model = CAMERA_OV2640;
  973. ov2640_init(&s_state->sensor);
  974. break;
  975. #endif
  976. #if CONFIG_OV7725_SUPPORT
  977. case OV7725_PID:
  978. *out_camera_model = CAMERA_OV7725;
  979. ov7725_init(&s_state->sensor);
  980. break;
  981. #endif
  982. #if CONFIG_OV3660_SUPPORT
  983. case OV3660_PID:
  984. *out_camera_model = CAMERA_OV3660;
  985. ov3660_init(&s_state->sensor);
  986. break;
  987. #endif
  988. #if CONFIG_OV5640_SUPPORT
  989. case OV5640_PID:
  990. *out_camera_model = CAMERA_OV5640;
  991. ov5640_init(&s_state->sensor);
  992. break;
  993. #endif
  994. #if CONFIG_OV7670_SUPPORT
  995. case OV7670_PID:
  996. *out_camera_model = CAMERA_OV7670;
  997. ov7670_init(&s_state->sensor);
  998. break;
  999. #endif
  1000. #if CONFIG_NT99141_SUPPORT
  1001. case NT99141_PID:
  1002. *out_camera_model = CAMERA_NT99141;
  1003. NT99141_init(&s_state->sensor);
  1004. break;
  1005. #endif
  1006. default:
  1007. id->PID = 0;
  1008. *out_camera_model = CAMERA_UNKNOWN;
  1009. camera_disable_out_clock();
  1010. ESP_LOGE(TAG, "Detected camera not supported.");
  1011. return ESP_ERR_CAMERA_NOT_SUPPORTED;
  1012. }
  1013. ESP_LOGD(TAG, "Doing SW reset of sensor");
  1014. s_state->sensor.reset(&s_state->sensor);
  1015. return ESP_OK;
  1016. }
  1017. esp_err_t camera_init(const camera_config_t* config)
  1018. {
  1019. if (!s_state) {
  1020. return ESP_ERR_INVALID_STATE;
  1021. }
  1022. if (s_state->sensor.id.PID == 0) {
  1023. return ESP_ERR_CAMERA_NOT_SUPPORTED;
  1024. }
  1025. memcpy(&s_state->config, config, sizeof(*config));
  1026. esp_err_t err = ESP_OK;
  1027. framesize_t frame_size = (framesize_t) config->frame_size;
  1028. pixformat_t pix_format = (pixformat_t) config->pixel_format;
  1029. switch (s_state->sensor.id.PID) {
  1030. #if CONFIG_OV2640_SUPPORT
  1031. case OV2640_PID:
  1032. if (frame_size > FRAMESIZE_UXGA) {
  1033. frame_size = FRAMESIZE_UXGA;
  1034. }
  1035. break;
  1036. #endif
  1037. #if CONFIG_OV7725_SUPPORT
  1038. case OV7725_PID:
  1039. if (frame_size > FRAMESIZE_VGA) {
  1040. frame_size = FRAMESIZE_VGA;
  1041. }
  1042. break;
  1043. #endif
  1044. #if CONFIG_OV3660_SUPPORT
  1045. case OV3660_PID:
  1046. if (frame_size > FRAMESIZE_QXGA) {
  1047. frame_size = FRAMESIZE_QXGA;
  1048. }
  1049. break;
  1050. #endif
  1051. #if CONFIG_OV5640_SUPPORT
  1052. case OV5640_PID:
  1053. if (frame_size > FRAMESIZE_QSXGA) {
  1054. frame_size = FRAMESIZE_QSXGA;
  1055. }
  1056. break;
  1057. #endif
  1058. #if CONFIG_OV7670_SUPPORT
  1059. case OV7670_PID:
  1060. if (frame_size > FRAMESIZE_VGA) {
  1061. frame_size = FRAMESIZE_VGA;
  1062. }
  1063. break;
  1064. #endif
  1065. #if CONFIG_NT99141_SUPPORT
  1066. case NT99141_PID:
  1067. if (frame_size > FRAMESIZE_HD) {
  1068. frame_size = FRAMESIZE_HD;
  1069. }
  1070. break;
  1071. #endif
  1072. default:
  1073. return ESP_ERR_CAMERA_NOT_SUPPORTED;
  1074. }
  1075. s_state->width = resolution[frame_size].width;
  1076. s_state->height = resolution[frame_size].height;
  1077. if (pix_format == PIXFORMAT_GRAYSCALE) {
  1078. s_state->fb_size = s_state->width * s_state->height;
  1079. if (s_state->sensor.id.PID == OV3660_PID || s_state->sensor.id.PID == OV5640_PID || s_state->sensor.id.PID == NT99141_PID) {
  1080. if (is_hs_mode()) {
  1081. s_state->sampling_mode = SM_0A00_0B00;
  1082. s_state->dma_filter = &dma_filter_yuyv_highspeed;
  1083. } else {
  1084. s_state->sampling_mode = SM_0A0B_0C0D;
  1085. s_state->dma_filter = &dma_filter_yuyv;
  1086. }
  1087. s_state->in_bytes_per_pixel = 1; // camera sends Y8
  1088. } else {
  1089. if (is_hs_mode() && s_state->sensor.id.PID != OV7725_PID) {
  1090. s_state->sampling_mode = SM_0A00_0B00;
  1091. s_state->dma_filter = &dma_filter_grayscale_highspeed;
  1092. } else {
  1093. s_state->sampling_mode = SM_0A0B_0C0D;
  1094. s_state->dma_filter = &dma_filter_grayscale;
  1095. }
  1096. s_state->in_bytes_per_pixel = 2; // camera sends YU/YV
  1097. }
  1098. s_state->fb_bytes_per_pixel = 1; // frame buffer stores Y8
  1099. } else if (pix_format == PIXFORMAT_YUV422 || pix_format == PIXFORMAT_RGB565) {
  1100. s_state->fb_size = s_state->width * s_state->height * 2;
  1101. if (is_hs_mode() && s_state->sensor.id.PID != OV7725_PID) {
  1102. if(s_state->sensor.id.PID == OV7670_PID) {
  1103. s_state->sampling_mode = SM_0A0B_0B0C;
  1104. }else{
  1105. s_state->sampling_mode = SM_0A00_0B00;
  1106. }
  1107. s_state->dma_filter = &dma_filter_yuyv_highspeed;
  1108. } else {
  1109. s_state->sampling_mode = SM_0A0B_0C0D;
  1110. s_state->dma_filter = &dma_filter_yuyv;
  1111. }
  1112. s_state->in_bytes_per_pixel = 2; // camera sends YU/YV
  1113. s_state->fb_bytes_per_pixel = 2; // frame buffer stores YU/YV/RGB565
  1114. } else if (pix_format == PIXFORMAT_RGB888) {
  1115. s_state->fb_size = s_state->width * s_state->height * 3;
  1116. if (is_hs_mode()) {
  1117. if(s_state->sensor.id.PID == OV7670_PID) {
  1118. s_state->sampling_mode = SM_0A0B_0B0C;
  1119. }else{
  1120. s_state->sampling_mode = SM_0A00_0B00;
  1121. }
  1122. s_state->dma_filter = &dma_filter_rgb888_highspeed;
  1123. } else {
  1124. s_state->sampling_mode = SM_0A0B_0C0D;
  1125. s_state->dma_filter = &dma_filter_rgb888;
  1126. }
  1127. s_state->in_bytes_per_pixel = 2; // camera sends RGB565
  1128. s_state->fb_bytes_per_pixel = 3; // frame buffer stores RGB888
  1129. } else if (pix_format == PIXFORMAT_JPEG) {
  1130. if (s_state->sensor.id.PID != OV2640_PID && s_state->sensor.id.PID != OV3660_PID && s_state->sensor.id.PID != OV5640_PID && s_state->sensor.id.PID != NT99141_PID) {
  1131. ESP_LOGE(TAG, "JPEG format is only supported for ov2640, ov3660 and ov5640");
  1132. err = ESP_ERR_NOT_SUPPORTED;
  1133. goto fail;
  1134. }
  1135. int qp = config->jpeg_quality;
  1136. int compression_ratio_bound = 1;
  1137. if (qp > 10) {
  1138. compression_ratio_bound = 16;
  1139. } else if (qp > 5) {
  1140. compression_ratio_bound = 10;
  1141. } else {
  1142. compression_ratio_bound = 4;
  1143. }
  1144. (*s_state->sensor.set_quality)(&s_state->sensor, qp);
  1145. s_state->in_bytes_per_pixel = 2;
  1146. s_state->fb_bytes_per_pixel = 2;
  1147. s_state->fb_size = (s_state->width * s_state->height * s_state->fb_bytes_per_pixel) / compression_ratio_bound;
  1148. s_state->dma_filter = &dma_filter_jpeg;
  1149. s_state->sampling_mode = SM_0A00_0B00;
  1150. } else {
  1151. ESP_LOGE(TAG, "Requested format is not supported");
  1152. err = ESP_ERR_NOT_SUPPORTED;
  1153. goto fail;
  1154. }
  1155. ESP_LOGD(TAG, "in_bpp: %d, fb_bpp: %d, fb_size: %d, mode: %d, width: %d height: %d",
  1156. s_state->in_bytes_per_pixel, s_state->fb_bytes_per_pixel,
  1157. s_state->fb_size, s_state->sampling_mode,
  1158. s_state->width, s_state->height);
  1159. i2s_init();
  1160. err = dma_desc_init();
  1161. if (err != ESP_OK) {
  1162. ESP_LOGE(TAG, "Failed to initialize I2S and DMA");
  1163. goto fail;
  1164. }
  1165. //s_state->fb_size = 75 * 1024;
  1166. err = camera_fb_init(s_state->config.fb_count);
  1167. if (err != ESP_OK) {
  1168. ESP_LOGE(TAG, "Failed to allocate frame buffer");
  1169. goto fail;
  1170. }
  1171. s_state->data_ready = xQueueCreate(16, sizeof(size_t));
  1172. if (s_state->data_ready == NULL) {
  1173. ESP_LOGE(TAG, "Failed to dma queue");
  1174. err = ESP_ERR_NO_MEM;
  1175. goto fail;
  1176. }
  1177. if(s_state->config.fb_count == 1) {
  1178. s_state->frame_ready = xSemaphoreCreateBinary();
  1179. if (s_state->frame_ready == NULL) {
  1180. ESP_LOGE(TAG, "Failed to create semaphore");
  1181. err = ESP_ERR_NO_MEM;
  1182. goto fail;
  1183. }
  1184. } else {
  1185. s_state->fb_in = xQueueCreate(s_state->config.fb_count, sizeof(camera_fb_t *));
  1186. s_state->fb_out = xQueueCreate(1, sizeof(camera_fb_t *));
  1187. if (s_state->fb_in == NULL || s_state->fb_out == NULL) {
  1188. ESP_LOGE(TAG, "Failed to fb queues");
  1189. err = ESP_ERR_NO_MEM;
  1190. goto fail;
  1191. }
  1192. }
  1193. //ToDo: core affinity?
  1194. #if CONFIG_CAMERA_CORE0
  1195. if (!xTaskCreatePinnedToCore(&dma_filter_task, "dma_filter", 4096, NULL, 10, &s_state->dma_filter_task, 0))
  1196. #elif CONFIG_CAMERA_CORE1
  1197. if (!xTaskCreatePinnedToCore(&dma_filter_task, "dma_filter", 4096, NULL, 10, &s_state->dma_filter_task, 1))
  1198. #else
  1199. if (!xTaskCreate(&dma_filter_task, "dma_filter", 4096, NULL, 10, &s_state->dma_filter_task))
  1200. #endif
  1201. {
  1202. ESP_LOGE(TAG, "Failed to create DMA filter task");
  1203. err = ESP_ERR_NO_MEM;
  1204. goto fail;
  1205. }
  1206. vsync_intr_disable();
  1207. err = gpio_install_isr_service(ESP_INTR_FLAG_LOWMED | ESP_INTR_FLAG_IRAM);
  1208. if (err != ESP_OK) {
  1209. if (err != ESP_ERR_INVALID_STATE) {
  1210. ESP_LOGE(TAG, "gpio_install_isr_service failed (%x)", err);
  1211. goto fail;
  1212. }
  1213. else {
  1214. ESP_LOGW(TAG, "gpio_install_isr_service already installed");
  1215. }
  1216. }
  1217. err = gpio_isr_handler_add(s_state->config.pin_vsync, &vsync_isr, NULL);
  1218. if (err != ESP_OK) {
  1219. ESP_LOGE(TAG, "vsync_isr_handler_add failed (%x)", err);
  1220. goto fail;
  1221. }
  1222. s_state->sensor.status.framesize = frame_size;
  1223. s_state->sensor.pixformat = pix_format;
  1224. ESP_LOGD(TAG, "Setting frame size to %dx%d", s_state->width, s_state->height);
  1225. if (s_state->sensor.set_framesize(&s_state->sensor, frame_size) != 0) {
  1226. ESP_LOGE(TAG, "Failed to set frame size");
  1227. err = ESP_ERR_CAMERA_FAILED_TO_SET_FRAME_SIZE;
  1228. goto fail;
  1229. }
  1230. s_state->sensor.set_pixformat(&s_state->sensor, pix_format);
  1231. if (s_state->sensor.id.PID == OV2640_PID) {
  1232. s_state->sensor.set_gainceiling(&s_state->sensor, GAINCEILING_2X);
  1233. s_state->sensor.set_bpc(&s_state->sensor, false);
  1234. s_state->sensor.set_wpc(&s_state->sensor, true);
  1235. s_state->sensor.set_lenc(&s_state->sensor, true);
  1236. }
  1237. if (skip_frame()) {
  1238. err = ESP_ERR_CAMERA_FAILED_TO_SET_OUT_FORMAT;
  1239. goto fail;
  1240. }
  1241. //todo: for some reason the first set of the quality does not work.
  1242. if (pix_format == PIXFORMAT_JPEG) {
  1243. (*s_state->sensor.set_quality)(&s_state->sensor, config->jpeg_quality);
  1244. }
  1245. s_state->sensor.init_status(&s_state->sensor);
  1246. return ESP_OK;
  1247. fail:
  1248. esp_camera_deinit();
  1249. return err;
  1250. }
  1251. esp_err_t esp_camera_init(const camera_config_t* config)
  1252. {
  1253. camera_model_t camera_model = CAMERA_NONE;
  1254. i2s_gpio_init(config);
  1255. esp_err_t err = camera_probe(config, &camera_model);
  1256. if (err != ESP_OK) {
  1257. ESP_LOGE(TAG, "Camera probe failed with error 0x%x", err);
  1258. goto fail;
  1259. }
  1260. if (camera_model == CAMERA_OV7725) {
  1261. ESP_LOGI(TAG, "Detected OV7725 camera");
  1262. if(config->pixel_format == PIXFORMAT_JPEG) {
  1263. ESP_LOGE(TAG, "Camera does not support JPEG");
  1264. err = ESP_ERR_CAMERA_NOT_SUPPORTED;
  1265. goto fail;
  1266. }
  1267. } else if (camera_model == CAMERA_OV2640) {
  1268. ESP_LOGI(TAG, "Detected OV2640 camera");
  1269. } else if (camera_model == CAMERA_OV3660) {
  1270. ESP_LOGI(TAG, "Detected OV3660 camera");
  1271. } else if (camera_model == CAMERA_OV5640) {
  1272. ESP_LOGI(TAG, "Detected OV5640 camera");
  1273. } else if (camera_model == CAMERA_OV7670) {
  1274. ESP_LOGI(TAG, "Detected OV7670 camera");
  1275. } else if (camera_model == CAMERA_NT99141) {
  1276. ESP_LOGI(TAG, "Detected NT99141 camera");
  1277. } else {
  1278. ESP_LOGI(TAG, "Camera not supported");
  1279. err = ESP_ERR_CAMERA_NOT_SUPPORTED;
  1280. goto fail;
  1281. }
  1282. err = camera_init(config);
  1283. if (err != ESP_OK) {
  1284. ESP_LOGE(TAG, "Camera init failed with error 0x%x", err);
  1285. return err;
  1286. }
  1287. return ESP_OK;
  1288. fail:
  1289. free(s_state);
  1290. s_state = NULL;
  1291. camera_disable_out_clock();
  1292. return err;
  1293. }
  1294. esp_err_t esp_camera_deinit()
  1295. {
  1296. if (s_state == NULL) {
  1297. return ESP_ERR_INVALID_STATE;
  1298. }
  1299. if (s_state->dma_filter_task) {
  1300. vTaskDelete(s_state->dma_filter_task);
  1301. }
  1302. if (s_state->data_ready) {
  1303. vQueueDelete(s_state->data_ready);
  1304. }
  1305. if (s_state->fb_in) {
  1306. vQueueDelete(s_state->fb_in);
  1307. }
  1308. if (s_state->fb_out) {
  1309. vQueueDelete(s_state->fb_out);
  1310. }
  1311. if (s_state->frame_ready) {
  1312. vSemaphoreDelete(s_state->frame_ready);
  1313. }
  1314. gpio_isr_handler_remove(s_state->config.pin_vsync);
  1315. if (s_state->i2s_intr_handle) {
  1316. esp_intr_disable(s_state->i2s_intr_handle);
  1317. esp_intr_free(s_state->i2s_intr_handle);
  1318. }
  1319. dma_desc_deinit();
  1320. camera_fb_deinit();
  1321. if(s_state->config.pin_xclk >= 0) {
  1322. camera_disable_out_clock();
  1323. }
  1324. free(s_state);
  1325. s_state = NULL;
  1326. periph_module_disable(PERIPH_I2S0_MODULE);
  1327. return ESP_OK;
  1328. }
  1329. #define FB_GET_TIMEOUT (4000 / portTICK_PERIOD_MS)
  1330. camera_fb_t* esp_camera_fb_get()
  1331. {
  1332. if (s_state == NULL) {
  1333. return NULL;
  1334. }
  1335. if(!I2S0.conf.rx_start) {
  1336. if(s_state->config.fb_count > 1) {
  1337. ESP_LOGD(TAG, "i2s_run");
  1338. }
  1339. if (i2s_run() != 0) {
  1340. return NULL;
  1341. }
  1342. }
  1343. bool need_yield = false;
  1344. if (s_state->config.fb_count == 1) {
  1345. if (xSemaphoreTake(s_state->frame_ready, FB_GET_TIMEOUT) != pdTRUE){
  1346. i2s_stop(&need_yield);
  1347. ESP_LOGE(TAG, "Failed to get the frame on time!");
  1348. return NULL;
  1349. }
  1350. return (camera_fb_t*)s_state->fb;
  1351. }
  1352. camera_fb_int_t * fb = NULL;
  1353. if(s_state->fb_out) {
  1354. if (xQueueReceive(s_state->fb_out, &fb, FB_GET_TIMEOUT) != pdTRUE) {
  1355. i2s_stop(&need_yield);
  1356. ESP_LOGE(TAG, "Failed to get the frame on time!");
  1357. return NULL;
  1358. }
  1359. }
  1360. return (camera_fb_t*)fb;
  1361. }
  1362. void esp_camera_fb_return(camera_fb_t * fb)
  1363. {
  1364. if(fb == NULL || s_state == NULL || s_state->config.fb_count == 1 || s_state->fb_in == NULL) {
  1365. return;
  1366. }
  1367. xQueueSend(s_state->fb_in, &fb, portMAX_DELAY);
  1368. }
  1369. sensor_t * esp_camera_sensor_get()
  1370. {
  1371. if (s_state == NULL) {
  1372. return NULL;
  1373. }
  1374. return &s_state->sensor;
  1375. }
  1376. esp_err_t esp_camera_save_to_nvs(const char *key)
  1377. {
  1378. #if ESP_IDF_VERSION_MAJOR > 3
  1379. nvs_handle_t handle;
  1380. #else
  1381. nvs_handle handle;
  1382. #endif
  1383. esp_err_t ret = nvs_open(key,NVS_READWRITE,&handle);
  1384. if (ret == ESP_OK) {
  1385. sensor_t *s = esp_camera_sensor_get();
  1386. if (s != NULL) {
  1387. ret = nvs_set_blob(handle,CAMERA_SENSOR_NVS_KEY,&s->status,sizeof(camera_status_t));
  1388. if (ret == ESP_OK) {
  1389. uint8_t pf = s->pixformat;
  1390. ret = nvs_set_u8(handle,CAMERA_PIXFORMAT_NVS_KEY,pf);
  1391. }
  1392. return ret;
  1393. } else {
  1394. return ESP_ERR_CAMERA_NOT_DETECTED;
  1395. }
  1396. nvs_close(handle);
  1397. return ret;
  1398. } else {
  1399. return ret;
  1400. }
  1401. }
  1402. esp_err_t esp_camera_load_from_nvs(const char *key)
  1403. {
  1404. #if ESP_IDF_VERSION_MAJOR > 3
  1405. nvs_handle_t handle;
  1406. #else
  1407. nvs_handle handle;
  1408. #endif
  1409. uint8_t pf;
  1410. esp_err_t ret = nvs_open(key,NVS_READWRITE,&handle);
  1411. if (ret == ESP_OK) {
  1412. sensor_t *s = esp_camera_sensor_get();
  1413. camera_status_t st;
  1414. if (s != NULL) {
  1415. size_t size = sizeof(camera_status_t);
  1416. ret = nvs_get_blob(handle,CAMERA_SENSOR_NVS_KEY,&st,&size);
  1417. if (ret == ESP_OK) {
  1418. s->set_ae_level(s,st.ae_level);
  1419. s->set_aec2(s,st.aec2);
  1420. s->set_aec_value(s,st.aec_value);
  1421. s->set_agc_gain(s,st.agc_gain);
  1422. s->set_awb_gain(s,st.awb_gain);
  1423. s->set_bpc(s,st.bpc);
  1424. s->set_brightness(s,st.brightness);
  1425. s->set_colorbar(s,st.colorbar);
  1426. s->set_contrast(s,st.contrast);
  1427. s->set_dcw(s,st.dcw);
  1428. s->set_denoise(s,st.denoise);
  1429. s->set_exposure_ctrl(s,st.aec);
  1430. s->set_framesize(s,st.framesize);
  1431. s->set_gain_ctrl(s,st.agc);
  1432. s->set_gainceiling(s,st.gainceiling);
  1433. s->set_hmirror(s,st.hmirror);
  1434. s->set_lenc(s,st.lenc);
  1435. s->set_quality(s,st.quality);
  1436. s->set_raw_gma(s,st.raw_gma);
  1437. s->set_saturation(s,st.saturation);
  1438. s->set_sharpness(s,st.sharpness);
  1439. s->set_special_effect(s,st.special_effect);
  1440. s->set_vflip(s,st.vflip);
  1441. s->set_wb_mode(s,st.wb_mode);
  1442. s->set_whitebal(s,st.awb);
  1443. s->set_wpc(s,st.wpc);
  1444. }
  1445. ret = nvs_get_u8(handle,CAMERA_PIXFORMAT_NVS_KEY,&pf);
  1446. if (ret == ESP_OK) {
  1447. s->set_pixformat(s,pf);
  1448. }
  1449. } else {
  1450. return ESP_ERR_CAMERA_NOT_DETECTED;
  1451. }
  1452. nvs_close(handle);
  1453. return ret;
  1454. } else {
  1455. ESP_LOGW(TAG,"Error (%d) opening nvs key \"%s\"",ret,key);
  1456. return ret;
  1457. }
  1458. }