ufodecode.c 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720
  1. #include <stdio.h>
  2. #include <stdlib.h>
  3. #include <string.h>
  4. #include <errno.h>
  5. #include "ufodecode.h"
  6. #include "ufodecode-private.h"
  7. #include "config.h"
  8. #ifdef HAVE_SSE
  9. #include <xmmintrin.h>
  10. #endif
  11. #define CHECKS
  12. #define IPECAMERA_NUM_ROWS 1088
  13. #define IPECAMERA_NUM_CHANNELS 16 /**< Number of channels per row */
  14. #define IPECAMERA_PIXELS_PER_CHANNEL 128 /**< Number of pixels per channel */
  15. #define IPECAMERA_WIDTH (IPECAMERA_NUM_CHANNELS * IPECAMERA_PIXELS_PER_CHANNEL) /**< Total pixel width of row */
  16. #define IPECAMERA_MODE_16_CHAN_IO 0
  17. #define IPECAMERA_MODE_4_CHAN_IO 2
  18. #define IPECAMERA_MODE_12_BIT_ADC 2
  19. #define IPECAMERA_MODE_11_BIT_ADC 1
  20. #define IPECAMERA_MODE_10_BIT_ADC 0
  21. typedef struct {
  22. unsigned int pixel_number : 8;
  23. unsigned int row_number : 12;
  24. unsigned int pixel_size : 4;
  25. unsigned int magic : 8;
  26. } payload_header_v5;
  27. /**
  28. * Check if value matches expected input.
  29. */
  30. #ifdef DEBUG
  31. # define CHECK_VALUE(value, expected) \
  32. if (value != expected) { \
  33. fprintf(stderr, "<%s:%i> 0x%x != 0x%x\n", __FILE__, __LINE__, value, expected); \
  34. err = 1; \
  35. }
  36. #else
  37. # define CHECK_VALUE(value, expected) \
  38. if (value != expected) { \
  39. err = 1; \
  40. }
  41. #endif
  42. /**
  43. * Check that flag evaluates to non-zero.
  44. */
  45. #ifdef DEBUG
  46. # define CHECK_FLAG(flag, check, ...) \
  47. if (!(check)) { \
  48. fprintf(stderr, "<%s:%i> Unexpected value 0x%x of " flag "\n", __FILE__, __LINE__, __VA_ARGS__); \
  49. err = 1; \
  50. }
  51. #else
  52. # define CHECK_FLAG(flag, check, ...) \
  53. if (!(check)) { \
  54. err = 1; \
  55. }
  56. #endif
  57. /**
  58. * \brief Setup a new decoder instance
  59. *
  60. * \param height Number of rows that are expected in the data stream. Set this
  61. * smaller 0 to let the decoder figure out the number of rows.
  62. * \param raw The data stream from the camera or NULL if set later with
  63. * ufo_decoder_set_raw_data.
  64. * \param num_bytes Size of the data stream buffer in bytes
  65. *
  66. * \return A new decoder instance that can be used to iterate over the frames
  67. * using ufo_decoder_get_next_frame.
  68. */
  69. UfoDecoder *
  70. ufo_decoder_new (int32_t height, uint32_t width, uint32_t *raw, size_t num_bytes)
  71. {
  72. if (width % IPECAMERA_PIXELS_PER_CHANNEL)
  73. return NULL;
  74. UfoDecoder *decoder = malloc(sizeof(UfoDecoder));
  75. if (decoder == NULL)
  76. return NULL;
  77. decoder->width = width;
  78. decoder->height = height;
  79. ufo_decoder_set_raw_data(decoder, raw, num_bytes);
  80. return decoder;
  81. }
  82. /**
  83. * \brief Release decoder instance
  84. *
  85. * \param decoder An UfoDecoder instance
  86. */
  87. void
  88. ufo_decoder_free(UfoDecoder *decoder)
  89. {
  90. free(decoder);
  91. }
  92. /**
  93. * \brief Set raw data stream
  94. *
  95. * \param decoder An UfoDecoder instance
  96. * \param raw Raw data stream
  97. * \param num_bytes Size of data stream buffer in bytes
  98. */
  99. void
  100. ufo_decoder_set_raw_data(UfoDecoder *decoder, uint32_t *raw, size_t num_bytes)
  101. {
  102. decoder->raw = raw;
  103. decoder->num_bytes = num_bytes;
  104. decoder->current_pos = 0;
  105. }
  106. static int
  107. ufo_decode_frame_channels_v0(UfoDecoder *decoder,
  108. uint16_t *pixel_buffer,
  109. uint32_t *raw,
  110. size_t num_words,
  111. size_t *offset)
  112. {
  113. static int channel_order[IPECAMERA_NUM_CHANNELS] = { 15, 13, 14, 12, 10, 8, 11, 7, 9, 6, 5, 2, 4, 3, 0, 1 };
  114. static int channel_size = (2 + IPECAMERA_PIXELS_PER_CHANNEL / 3);
  115. const int bytes = channel_size - 1;
  116. const int num_rows = decoder->height;
  117. const size_t cpl = (decoder->width / IPECAMERA_PIXELS_PER_CHANNEL);
  118. const size_t cpi = num_rows * cpl;
  119. int pos = 0;
  120. uint32_t data;
  121. #if defined(HAVE_SSE) && !defined(DEBUG)
  122. __m128i mask = _mm_set_epi32(0x3FF, 0x3FF, 0x3FF, 0x3FF);
  123. __m128i packed;
  124. __m128i tmp1, tmp2;
  125. uint32_t result[4] __attribute__ ((aligned (16))) = {0};
  126. #endif
  127. if (cpi * channel_size > num_words) {
  128. #ifdef DEBUG
  129. fprintf(stderr, "Not enough data to decode frame, expected %lu bytes, but received %lu\n", cpi * channel_size * sizeof(uint32_t), num_words * sizeof(uint32_t));
  130. #endif
  131. return EILSEQ;
  132. }
  133. for (size_t c = 0; c < cpi; c++) {
  134. const int info = raw[0];
  135. int row = (info >> 4) & 0x7FF;
  136. int channel = info & 0x0F;
  137. int pixels = (info >> 20) & 0xFF;
  138. #ifdef CHECKS
  139. int header = (info >> 30) & 0x03;
  140. const int bpp = (info >> 16) & 0x0F;
  141. int err;
  142. CHECK_FLAG("raw header magick", header == 2, header);
  143. CHECK_FLAG("row number, only %i rows requested", row < num_rows, row, num_rows);
  144. CHECK_FLAG("pixel size, only 10 bits are supported", bpp == 10, bpp);
  145. CHECK_FLAG("channel, limited by %zu output channels", channel < cpl, channel, cpl);
  146. #endif
  147. if ((row > num_rows) || (channel > cpl) || (pixels > IPECAMERA_PIXELS_PER_CHANNEL))
  148. return EILSEQ;
  149. channel = channel_order[channel];
  150. int base = row * IPECAMERA_WIDTH + channel * IPECAMERA_PIXELS_PER_CHANNEL;
  151. /* "Correct" missing pixel */
  152. if ((row < 2) && (pixels == (IPECAMERA_PIXELS_PER_CHANNEL - 1))) {
  153. pixel_buffer[base] = 0;
  154. /* base++; */
  155. }
  156. #ifdef CHECKS
  157. else
  158. CHECK_FLAG("number of pixels, %i is expected", pixels == IPECAMERA_PIXELS_PER_CHANNEL, pixels, IPECAMERA_PIXELS_PER_CHANNEL);
  159. #endif
  160. #if defined(HAVE_SSE) && !defined(DEBUG)
  161. for (int i = 1 ; i < bytes-4; i += 4, base += 12) {
  162. packed = _mm_set_epi32(raw[i], raw[i+1], raw[i+2], raw[i+3]);
  163. tmp1 = _mm_srli_epi32(packed, 20);
  164. tmp2 = _mm_and_si128(tmp1, mask);
  165. _mm_storeu_si128((__m128i*) result, tmp2);
  166. pixel_buffer[base+0] = result[3];
  167. pixel_buffer[base+3] = result[2];
  168. pixel_buffer[base+6] = result[1];
  169. pixel_buffer[base+9] = result[0];
  170. tmp1 = _mm_srli_epi32(packed, 10);
  171. tmp2 = _mm_and_si128(tmp1, mask);
  172. _mm_storeu_si128((__m128i*) result, tmp2);
  173. pixel_buffer[base+1] = result[3];
  174. pixel_buffer[base+4] = result[2];
  175. pixel_buffer[base+7] = result[1];
  176. pixel_buffer[base+10] = result[0];
  177. tmp1 = _mm_and_si128(packed, mask);
  178. _mm_storeu_si128((__m128i*) result, tmp1);
  179. pixel_buffer[base+2] = result[3];
  180. pixel_buffer[base+5] = result[2];
  181. pixel_buffer[base+8] = result[1];
  182. pixel_buffer[base+11] = result[0];
  183. }
  184. /* Compute last pixels by hand */
  185. data = raw[41];
  186. pixel_buffer[base++] = (data >> 20) & 0x3FF;
  187. pixel_buffer[base++] = (data >> 10) & 0x3FF;
  188. pixel_buffer[base++] = data & 0x3FF;
  189. data = raw[42];
  190. pixel_buffer[base++] = (data >> 20) & 0x3FF;
  191. pixel_buffer[base++] = (data >> 10) & 0x3FF;
  192. pixel_buffer[base++] = data & 0x3FF;
  193. #else
  194. for (int i = 1 ; i < bytes; i++) {
  195. data = raw[i];
  196. #ifdef DEBUG
  197. header = (data >> 30) & 0x03;
  198. CHECK_FLAG("raw data magic", header == 3, header);
  199. if (err)
  200. return err;
  201. #endif
  202. pixel_buffer[base++] = (data >> 20) & 0x3FF;
  203. pixel_buffer[base++] = (data >> 10) & 0x3FF;
  204. pixel_buffer[base++] = data & 0x3FF;
  205. }
  206. #endif
  207. data = raw[bytes];
  208. #ifdef DEBUG
  209. header = (data >> 30) & 0x03;
  210. CHECK_FLAG("raw data magic", header == 3, header);
  211. CHECK_FLAG("raw footer magic", (data & 0x3FF) == 0x55, (data & 0x3FF));
  212. if (err)
  213. return err;
  214. #endif
  215. int ppw = pixels >> 6;
  216. for (int j = 0; j < ppw; j++)
  217. pixel_buffer[base++] = (data >> (10 * (ppw - j))) & 0x3FF;
  218. pos += channel_size;
  219. raw += channel_size;
  220. }
  221. *offset = pos;
  222. return 0;
  223. }
  224. static int
  225. ufo_decode_frame_channels_v4(UfoDecoder *decoder,
  226. uint16_t *pixel_buffer,
  227. uint32_t *raw,
  228. size_t num_words,
  229. size_t num_rows,
  230. size_t *offset)
  231. {
  232. static const int channel_order[IPECAMERA_NUM_CHANNELS] = { 15, 13, 14, 12, 10, 8, 11, 7, 9, 6, 5, 2, 4, 3, 0, 1 };
  233. static const int channel_size = (2 + IPECAMERA_PIXELS_PER_CHANNEL / 3);
  234. const int bytes = channel_size - 1;
  235. const size_t channels_per_row = (decoder->width / IPECAMERA_PIXELS_PER_CHANNEL);
  236. const size_t cpi = num_rows * channels_per_row;
  237. int pos = 0;
  238. uint32_t data;
  239. #if defined(HAVE_SSE) && !defined(DEBUG)
  240. __m128i mask = _mm_set_epi32(0x3FF, 0x3FF, 0x3FF, 0x3FF);
  241. __m128i packed;
  242. __m128i tmp1, tmp2;
  243. uint32_t result[4] __attribute__ ((aligned (16))) = {0};
  244. #endif
  245. if (cpi * channel_size > num_words) {
  246. #ifdef DEBUG
  247. fprintf(stderr, "Not enough data to decode frame, expected %lu bytes, but received %lu\n", cpi * channel_size * sizeof(uint32_t), num_words * sizeof(uint32_t));
  248. #endif
  249. return EILSEQ;
  250. }
  251. for (size_t c = 0; c < cpi; c++) {
  252. const int info = raw[0];
  253. int row = (info >> 4) & 0x7FF;
  254. int channel = info & 0x0F;
  255. int pixels = (info >> 20) & 0xFF;
  256. #ifdef DEBUG
  257. int err = 0;
  258. int header = (info >> 30) & 0x03;
  259. const int bpp = (info >> 16) & 0x0F;
  260. CHECK_FLAG("raw header magick", header == 2, header);
  261. CHECK_FLAG("pixel size, only 10 bits are supported", bpp == 10, bpp);
  262. CHECK_FLAG("channel, limited by %zu output channels", channel < channels_per_row, channel, channels_per_row);
  263. #endif
  264. if ((channel > channels_per_row) || (pixels > IPECAMERA_PIXELS_PER_CHANNEL))
  265. return EILSEQ;
  266. channel = channel_order[channel];
  267. int base = row * IPECAMERA_WIDTH + channel * IPECAMERA_PIXELS_PER_CHANNEL;
  268. /* "Correct" missing pixel */
  269. if ((row < 2) && (pixels == (IPECAMERA_PIXELS_PER_CHANNEL - 1))) {
  270. pixel_buffer[base] = 0;
  271. /* base++; */
  272. }
  273. #if defined(HAVE_SSE) && !defined(DEBUG)
  274. for (int i = 1 ; i < bytes-4; i += 4, base += 12) {
  275. packed = _mm_set_epi32(raw[i], raw[i+1], raw[i+2], raw[i+3]);
  276. tmp1 = _mm_srli_epi32(packed, 20);
  277. tmp2 = _mm_and_si128(tmp1, mask);
  278. _mm_storeu_si128((__m128i*) result, tmp2);
  279. pixel_buffer[base+0] = result[3];
  280. pixel_buffer[base+3] = result[2];
  281. pixel_buffer[base+6] = result[1];
  282. pixel_buffer[base+9] = result[0];
  283. tmp1 = _mm_srli_epi32(packed, 10);
  284. tmp2 = _mm_and_si128(tmp1, mask);
  285. _mm_storeu_si128((__m128i*) result, tmp2);
  286. pixel_buffer[base+1] = result[3];
  287. pixel_buffer[base+4] = result[2];
  288. pixel_buffer[base+7] = result[1];
  289. pixel_buffer[base+10] = result[0];
  290. tmp1 = _mm_and_si128(packed, mask);
  291. _mm_storeu_si128((__m128i*) result, tmp1);
  292. pixel_buffer[base+2] = result[3];
  293. pixel_buffer[base+5] = result[2];
  294. pixel_buffer[base+8] = result[1];
  295. pixel_buffer[base+11] = result[0];
  296. }
  297. /* Compute last pixels by hand */
  298. data = raw[41];
  299. pixel_buffer[base++] = (data >> 20) & 0x3FF;
  300. pixel_buffer[base++] = (data >> 10) & 0x3FF;
  301. pixel_buffer[base++] = data & 0x3FF;
  302. data = raw[42];
  303. pixel_buffer[base++] = (data >> 20) & 0x3FF;
  304. pixel_buffer[base++] = (data >> 10) & 0x3FF;
  305. pixel_buffer[base++] = data & 0x3FF;
  306. #else
  307. for (int i = 1 ; i < bytes; i++) {
  308. data = raw[i];
  309. #ifdef DEBUG
  310. header = (data >> 30) & 0x03;
  311. CHECK_FLAG("raw data magic", header == 3, header);
  312. if (err)
  313. return err;
  314. #endif
  315. pixel_buffer[base++] = (data >> 20) & 0x3FF;
  316. pixel_buffer[base++] = (data >> 10) & 0x3FF;
  317. pixel_buffer[base++] = data & 0x3FF;
  318. }
  319. #endif
  320. data = raw[bytes];
  321. #ifdef DEBUG
  322. header = (data >> 30) & 0x03;
  323. CHECK_FLAG("raw data magic", header == 3, header);
  324. CHECK_FLAG("raw footer magic", (data & 0x3FF) == 0x55, (data & 0x3FF));
  325. if (err)
  326. return err;
  327. #endif
  328. int ppw = pixels >> 6;
  329. for (int j = 0; j < ppw; j++)
  330. pixel_buffer[base++] = (data >> (10 * (ppw - j))) & 0x3FF;
  331. pos += channel_size;
  332. raw += channel_size;
  333. }
  334. *offset = pos;
  335. return 0;
  336. }
  337. static int
  338. ufo_decode_frame_channels_v5(UfoDecoder *decoder,
  339. uint16_t *pixel_buffer,
  340. uint32_t *raw,
  341. size_t num_rows,
  342. size_t *offset,
  343. uint8_t output_mode)
  344. {
  345. payload_header_v5 *header;
  346. size_t base = 0, index = 0;
  347. header = (payload_header_v5 *) &raw[base];
  348. if (output_mode == IPECAMERA_MODE_4_CHAN_IO) {
  349. size_t off = 0;
  350. while (raw[base] != 0xAAAAAAA) {
  351. header = (payload_header_v5 *) &raw[base];
  352. index = header->row_number * IPECAMERA_WIDTH + header->pixel_number;
  353. /* Skip header + one zero-filled words */
  354. base += 2;
  355. if ((header->magic != 0xe0) && (header->magic != 0xc0)) {
  356. pixel_buffer[index + (0+off)*IPECAMERA_PIXELS_PER_CHANNEL] = 0xfff & (raw[base+5] >> 12);
  357. pixel_buffer[index + (4+off)*IPECAMERA_PIXELS_PER_CHANNEL] = 0xfff & (raw[base+4] >> 4);
  358. pixel_buffer[index + (8+off)*IPECAMERA_PIXELS_PER_CHANNEL] = ((0xf & raw[base+1]) << 8) | (raw[base+2] >> 24);
  359. pixel_buffer[index + (12+off)*IPECAMERA_PIXELS_PER_CHANNEL] = 0xfff & (raw[base+1] >> 16);
  360. }
  361. else {
  362. off++;
  363. if (header->magic == 0xc0)
  364. off = 0;
  365. }
  366. base += 6;
  367. }
  368. }
  369. else {
  370. while (raw[base] != 0xAAAAAAA) {
  371. header = (payload_header_v5 *) &raw[base];
  372. index = header->row_number * IPECAMERA_WIDTH + header->pixel_number;
  373. /* Skip header + two zero-filled words */
  374. base += 2;
  375. if (header->magic != 0xc0) {
  376. pixel_buffer[index + 15*IPECAMERA_PIXELS_PER_CHANNEL] = 0x3ff & (raw[base] >> 20);
  377. pixel_buffer[index + 13*IPECAMERA_PIXELS_PER_CHANNEL] = 0x3ff & (raw[base] >> 8);
  378. pixel_buffer[index + 14*IPECAMERA_PIXELS_PER_CHANNEL] = 0x3ff & (((0xff & raw[base]) << 4) | (raw[base+1] >> 28));
  379. pixel_buffer[index + 12*IPECAMERA_PIXELS_PER_CHANNEL] = 0x3ff & (raw[base+1] >> 16);
  380. pixel_buffer[index + 10*IPECAMERA_PIXELS_PER_CHANNEL] = 0x3ff & (raw[base+1] >> 4);
  381. pixel_buffer[index + 8*IPECAMERA_PIXELS_PER_CHANNEL] = ((0x3 & raw[base+1]) << 8) | (raw[base+2] >> 24);
  382. pixel_buffer[index + 11*IPECAMERA_PIXELS_PER_CHANNEL] = 0x3ff & (raw[base+2] >> 12);
  383. pixel_buffer[index + 7*IPECAMERA_PIXELS_PER_CHANNEL] = 0x3ff & raw[base+2];
  384. pixel_buffer[index + 9*IPECAMERA_PIXELS_PER_CHANNEL] = 0x3ff & (raw[base+3] >> 20);
  385. pixel_buffer[index + 6*IPECAMERA_PIXELS_PER_CHANNEL] = 0x3ff & (raw[base+3] >> 8);
  386. pixel_buffer[index + 5*IPECAMERA_PIXELS_PER_CHANNEL] = 0x3ff & (((0xff & raw[base+3]) << 4) | (raw[base+4] >> 28));
  387. pixel_buffer[index + 2*IPECAMERA_PIXELS_PER_CHANNEL] = 0x3ff & (raw[base+4] >> 16);
  388. pixel_buffer[index + 4*IPECAMERA_PIXELS_PER_CHANNEL] = 0x3ff & (raw[base+4] >> 4);
  389. pixel_buffer[index + 3*IPECAMERA_PIXELS_PER_CHANNEL] = ((0x3 & raw[base+4]) << 8) | (raw[base+5] >> 24);
  390. pixel_buffer[index + 0*IPECAMERA_PIXELS_PER_CHANNEL] = 0x3ff & (raw[base+5] >> 12);
  391. pixel_buffer[index + 1*IPECAMERA_PIXELS_PER_CHANNEL] = 0x3ff & raw[base+5];
  392. }
  393. base += 6;
  394. }
  395. }
  396. *offset = base;
  397. return 0;
  398. }
  399. /**
  400. * \brief Deinterlace by interpolating between two rows
  401. *
  402. * \param in Input frame
  403. * \param out Destination of interpolated frame
  404. * \param width Width of frame in pixels
  405. * \param height Height of frame in pixels
  406. */
  407. void ufo_deinterlace_interpolate(const uint16_t *in, uint16_t *out, int width, int height)
  408. {
  409. const size_t row_size_bytes = width * sizeof(uint16_t);
  410. for (int row = 0; row < height; row++) {
  411. /* Copy one line */
  412. memcpy(out, in + row*width, row_size_bytes);
  413. out += width;
  414. /* Interpolate between source row and row+1 */
  415. for (int x = 0; x < width; x++)
  416. out[x] = (int) (0.5 * in[row*width + x] + 0.5 * in[(row+1)*width + x]);
  417. out += width;
  418. }
  419. /* Copy last row */
  420. memcpy(out, in + width * (height - 1), row_size_bytes);
  421. }
  422. /**
  423. * \brief Deinterlace by "weaving" the rows of two frames
  424. *
  425. * \param in1 First frame
  426. * \param in2 Second frame
  427. * \param out Destination of weaved frame
  428. * \param width Width of frame in pixels
  429. * \param height Height of frame in pixels
  430. */
  431. void ufo_deinterlace_weave(const uint16_t *in1, const uint16_t *in2, uint16_t *out, int width, int height)
  432. {
  433. const size_t row_size_bytes = width * sizeof(uint16_t);
  434. for (int row = 0; row < height; row++) {
  435. memcpy(out, in1 + row*width, row_size_bytes);
  436. out += width;
  437. memcpy(out, in2 + row*width, row_size_bytes);
  438. out += width;
  439. }
  440. }
  441. /**
  442. * \brief Decodes frame
  443. *
  444. * This function tries to decode the supplied data
  445. *
  446. * \param decoder An UfoDecoder instance
  447. * \param raw Raw data stream
  448. * \param num_bytes Size of data stream buffer in bytes
  449. * \param pixels If pointer with NULL content is passed, a new buffer is
  450. * allocated otherwise, this user-supplied buffer is used.
  451. * \param frame_number Frame number as reported in the header
  452. * \param time_stamp Time stamp of the frame as reported in the header
  453. *
  454. * \return number of decoded bytes or 0 in case of error
  455. */
  456. size_t ufo_decoder_decode_frame(UfoDecoder *decoder,
  457. uint32_t *raw,
  458. size_t num_bytes,
  459. uint16_t *pixels,
  460. UfoDecoderMeta *meta)
  461. {
  462. int err = 0;
  463. size_t pos = 0;
  464. size_t advance = 0;
  465. const size_t num_words = num_bytes / 4;
  466. if ((pixels == NULL) || (num_words < 16))
  467. return 0;
  468. size_t rows_per_frame = decoder->height;
  469. const int version = (raw[pos+6] >> 24) & 0xF;
  470. #ifdef DEBUG
  471. CHECK_VALUE(raw[pos++], 0x51111111);
  472. CHECK_VALUE(raw[pos++], 0x52222222);
  473. CHECK_VALUE(raw[pos++], 0x53333333);
  474. CHECK_VALUE(raw[pos++], 0x54444444);
  475. CHECK_VALUE(raw[pos++], 0x55555555);
  476. switch (version) {
  477. case 0:
  478. CHECK_VALUE(raw[pos++], 0x56666666);
  479. CHECK_VALUE(raw[pos] >> 28, 0x5);
  480. meta->frame_number = raw[pos++] & 0xFFFFFFF;
  481. CHECK_VALUE(raw[pos] >> 28, 0x5);
  482. meta->time_stamp = raw[pos++] & 0xFFFFFFF;
  483. break;
  484. case 4:
  485. case 5:
  486. CHECK_VALUE(raw[pos] >> 28, 0x5);
  487. meta->cmosis_start_address = (raw[pos] >> 21) & 0x1FF;
  488. meta->n_skipped_rows = (raw[pos] >> 15) & 0x3F;
  489. meta->n_rows = rows_per_frame = raw[pos] & 0x7FF;
  490. pos++;
  491. meta->frame_number = raw[pos++] & 0x1FFFFFF;
  492. CHECK_VALUE(raw[pos] >> 28, 0x5);
  493. meta->time_stamp = raw[pos] & 0xFFFFFF;
  494. meta->output_mode = (raw[pos] >> 24) & 0x3;
  495. meta->adc_resolution = (raw[pos] >> 26) & 0x3;
  496. pos++;
  497. if ((meta->output_mode != IPECAMERA_MODE_4_CHAN_IO) && (meta->output_mode != IPECAMERA_MODE_16_CHAN_IO)) {
  498. #ifdef DEBUG
  499. fprintf(stderr, "Output mode 0x%x is not supported\n", meta->output_mode);
  500. #endif
  501. return EILSEQ;
  502. }
  503. break;
  504. default:
  505. fprintf(stderr, "Unsupported data format version %i detected\n", version);
  506. return 0;
  507. }
  508. if (err)
  509. return 0;
  510. #else
  511. switch (version) {
  512. case 0:
  513. meta->frame_number = raw[pos + 6] & 0xFFFFFFF;
  514. meta->time_stamp = raw[pos + 7] & 0xFFFFFFF;
  515. break;
  516. case 4:
  517. case 5:
  518. meta->frame_number = raw[pos + 6] & 0x1FFFFFF;
  519. meta->time_stamp = raw[pos + 7] & 0xFFFFFF;
  520. meta->output_mode = (raw[pos + 7] >> 24) & 0x3;
  521. meta->adc_resolution = (raw[pos + 7] >> 26) & 0x3;
  522. break;
  523. default:
  524. fprintf(stderr, "Unsupported data format detected\n");
  525. return 0;
  526. }
  527. pos += 8;
  528. #endif
  529. switch (version) {
  530. case 0:
  531. err = ufo_decode_frame_channels_v0(decoder, pixels, raw + pos, num_words - pos - 8, &advance);
  532. break;
  533. case 4:
  534. err = ufo_decode_frame_channels_v4(decoder, pixels, raw + pos, num_words - pos - 8, rows_per_frame, &advance);
  535. break;
  536. case 5:
  537. err = ufo_decode_frame_channels_v5(decoder, pixels, raw + pos, rows_per_frame, &advance, meta->output_mode);
  538. break;
  539. default:
  540. break;
  541. }
  542. if (err)
  543. return 0;
  544. pos += advance;
  545. #ifdef CHECKS
  546. CHECK_VALUE(raw[pos++], 0x0AAAAAAA);
  547. meta->status1.bits = raw[pos++];
  548. meta->status2.bits = raw[pos++];
  549. meta->status3.bits = raw[pos++];
  550. pos++;
  551. pos++;
  552. CHECK_VALUE(raw[pos++], 0x00000000);
  553. CHECK_VALUE(raw[pos++], 0x01111111);
  554. if (err)
  555. return 0;
  556. #else
  557. pos += 8;
  558. #endif
  559. return pos;
  560. }
  561. /**
  562. * \brief Iterate and decode next frame
  563. *
  564. * This function tries to decode the next frame in the currently set raw data
  565. * stream.
  566. *
  567. * \param decoder An UfoDecoder instance
  568. * \param pixels If pointer with NULL content is passed, a new buffer is
  569. * allocated otherwise, this user-supplied buffer is used.
  570. * \param num_rows Number of actual decoded rows
  571. * \param frame_number Frame number as reported in the header
  572. * \param time_stamp Time stamp of the frame as reported in the header
  573. *
  574. * \return 0 in case of no error, EIO if end of stream was reached, ENOMEM if
  575. * NULL was passed but no memory could be allocated, EILSEQ if data stream is
  576. * corrupt and EFAULT if pixels is a NULL-pointer.
  577. */
  578. int ufo_decoder_get_next_frame(UfoDecoder *decoder,
  579. uint16_t **pixels,
  580. UfoDecoderMeta *meta)
  581. {
  582. uint32_t *raw = decoder->raw;
  583. size_t pos = decoder->current_pos;
  584. size_t advance;
  585. const size_t num_words = decoder->num_bytes / 4;
  586. if (pixels == NULL)
  587. return 0;
  588. if (pos >= num_words)
  589. return EIO;
  590. if (num_words < 16)
  591. return EILSEQ;
  592. if (*pixels == NULL) {
  593. *pixels = (uint16_t *) malloc(IPECAMERA_WIDTH * decoder->height * sizeof(uint16_t));
  594. if (*pixels == NULL)
  595. return ENOMEM;
  596. }
  597. while ((pos < num_words) && (raw[pos] != 0x51111111))
  598. pos++;
  599. advance = ufo_decoder_decode_frame(decoder, raw + pos, decoder->num_bytes - pos, *pixels, meta);
  600. /*
  601. * On error, advance is 0 but we have to advance at least a bit to net get
  602. * caught in an infinite loop when trying to decode subsequent frames.
  603. */
  604. pos += advance == 0 ? 1 : advance;
  605. /* if bytes left and we see fill bytes, skip them */
  606. if (((pos + 2) < num_words) && ((raw[pos] == 0x0) && (raw[pos+1] == 0x1111111))) {
  607. pos += 2;
  608. while ((pos < num_words) && ((raw[pos] == 0x89abcdef) || (raw[pos] == 0x1234567)))
  609. pos++;
  610. }
  611. decoder->current_pos = pos;
  612. if (!advance)
  613. return EILSEQ;
  614. return 0;
  615. }