falkTX
/
FFmpeg
mirror of https://github.com/falkTX/FFmpeg.git
1
0
Fork 0
Code Issues 0 Releases 338 Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
2617 Commits
32 Branches
341MB
Tree: 8470540344
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '8470540344'
${ noResults }
FFmpeg/libavcodec/mjpeg.c

2262 lines
66KB
Raw Blame History 

  1. /*

  2.  * MJPEG encoder and decoder

  3.  * Copyright (c) 2000, 2001 Fabrice Bellard.

  4.  *

  5.  * This library is free software; you can redistribute it and/or

  6.  * modify it under the terms of the GNU Lesser General Public

  7.  * License as published by the Free Software Foundation; either

  8.  * version 2 of the License, or (at your option) any later version.

  9.  *

  10.  * This library is distributed in the hope that it will be useful,

  11.  * but WITHOUT ANY WARRANTY; without even the implied warranty of

  12.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

  13.  * Lesser General Public License for more details.

  14.  *

  15.  * You should have received a copy of the GNU Lesser General Public

  16.  * License along with this library; if not, write to the Free Software

  17.  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

  18.  *

  19.  * Support for external huffman table, various fixes (AVID workaround),

  20.  * aspecting, new decode_frame mechanism and apple mjpeg-b support

  21.  *                                  by Alex Beregszaszi <alex@naxine.org>

  22.  */

  23. 

  24. /**

  25.  * @file mjpeg.c

  26.  * MJPEG encoder and decoder.

  27.  */

  28.  

  29. //#define DEBUG

  30. #include <assert.h>

  31. 

  32. #include "avcodec.h"

  33. #include "dsputil.h"

  34. #include "mpegvideo.h"

  35. 

  36. /* use two quantizer tables (one for luminance and one for chrominance) */

  37. /* not yet working */

  38. #undef TWOMATRIXES

  39. 

  40. typedef struct MJpegContext {

  41.     uint8_t huff_size_dc_luminance[12]; //FIXME use array [3] instead of lumi / chrom, for easier addressing 

  42.     uint16_t huff_code_dc_luminance[12];

  43.     uint8_t huff_size_dc_chrominance[12];

  44.     uint16_t huff_code_dc_chrominance[12];

  45. 

  46.     uint8_t huff_size_ac_luminance[256];

  47.     uint16_t huff_code_ac_luminance[256];

  48.     uint8_t huff_size_ac_chrominance[256];

  49.     uint16_t huff_code_ac_chrominance[256];

  50. } MJpegContext;

  51. 

  52. /* JPEG marker codes */

  53. typedef enum {

  54.     /* start of frame */

  55.     SOF0  = 0xc0,	/* baseline */

  56.     SOF1  = 0xc1,	/* extended sequential, huffman */

  57.     SOF2  = 0xc2,	/* progressive, huffman */

  58.     SOF3  = 0xc3,	/* lossless, huffman */

  59. 

  60.     SOF5  = 0xc5,	/* differential sequential, huffman */

  61.     SOF6  = 0xc6,	/* differential progressive, huffman */

  62.     SOF7  = 0xc7,	/* differential lossless, huffman */

  63.     JPG   = 0xc8,	/* reserved for JPEG extension */

  64.     SOF9  = 0xc9,	/* extended sequential, arithmetic */

  65.     SOF10 = 0xca,	/* progressive, arithmetic */

  66.     SOF11 = 0xcb,	/* lossless, arithmetic */

  67. 

  68.     SOF13 = 0xcd,	/* differential sequential, arithmetic */

  69.     SOF14 = 0xce,	/* differential progressive, arithmetic */

  70.     SOF15 = 0xcf,	/* differential lossless, arithmetic */

  71. 

  72.     DHT   = 0xc4,	/* define huffman tables */

  73. 

  74.     DAC   = 0xcc,	/* define arithmetic-coding conditioning */

  75. 

  76.     /* restart with modulo 8 count "m" */

  77.     RST0  = 0xd0,

  78.     RST1  = 0xd1,

  79.     RST2  = 0xd2,

  80.     RST3  = 0xd3,

  81.     RST4  = 0xd4,

  82.     RST5  = 0xd5,

  83.     RST6  = 0xd6,

  84.     RST7  = 0xd7,

  85. 

  86.     SOI   = 0xd8,	/* start of image */

  87.     EOI   = 0xd9,	/* end of image */

  88.     SOS   = 0xda,	/* start of scan */

  89.     DQT   = 0xdb,	/* define quantization tables */

  90.     DNL   = 0xdc,	/* define number of lines */

  91.     DRI   = 0xdd,	/* define restart interval */

  92.     DHP   = 0xde,	/* define hierarchical progression */

  93.     EXP   = 0xdf,	/* expand reference components */

  94. 

  95.     APP0  = 0xe0,

  96.     APP1  = 0xe1,

  97.     APP2  = 0xe2,

  98.     APP3  = 0xe3,

  99.     APP4  = 0xe4,

  100.     APP5  = 0xe5,

  101.     APP6  = 0xe6,

  102.     APP7  = 0xe7,

  103.     APP8  = 0xe8,

  104.     APP9  = 0xe9,

  105.     APP10 = 0xea,

  106.     APP11 = 0xeb,

  107.     APP12 = 0xec,

  108.     APP13 = 0xed,

  109.     APP14 = 0xee,

  110.     APP15 = 0xef,

  111. 

  112.     JPG0  = 0xf0,

  113.     JPG1  = 0xf1,

  114.     JPG2  = 0xf2,

  115.     JPG3  = 0xf3,

  116.     JPG4  = 0xf4,

  117.     JPG5  = 0xf5,

  118.     JPG6  = 0xf6,

  119.     JPG7  = 0xf7,

  120.     JPG8  = 0xf8,

  121.     JPG9  = 0xf9,

  122.     JPG10 = 0xfa,

  123.     JPG11 = 0xfb,

  124.     JPG12 = 0xfc,

  125.     JPG13 = 0xfd,

  126. 

  127.     COM   = 0xfe,	/* comment */

  128. 

  129.     TEM   = 0x01,	/* temporary private use for arithmetic coding */

  130. 

  131.     /* 0x02 -> 0xbf reserved */

  132. } JPEG_MARKER;

  133. 

  134. #if 0

  135. /* These are the sample quantization tables given in JPEG spec section K.1.

  136.  * The spec says that the values given produce "good" quality, and

  137.  * when divided by 2, "very good" quality.

  138.  */

  139. static const unsigned char std_luminance_quant_tbl[64] = {

  140.     16,  11,  10,  16,  24,  40,  51,  61,

  141.     12,  12,  14,  19,  26,  58,  60,  55,

  142.     14,  13,  16,  24,  40,  57,  69,  56,

  143.     14,  17,  22,  29,  51,  87,  80,  62,

  144.     18,  22,  37,  56,  68, 109, 103,  77,

  145.     24,  35,  55,  64,  81, 104, 113,  92,

  146.     49,  64,  78,  87, 103, 121, 120, 101,

  147.     72,  92,  95,  98, 112, 100, 103,  99

  148. };

  149. static const unsigned char std_chrominance_quant_tbl[64] = {

  150.     17,  18,  24,  47,  99,  99,  99,  99,

  151.     18,  21,  26,  66,  99,  99,  99,  99,

  152.     24,  26,  56,  99,  99,  99,  99,  99,

  153.     47,  66,  99,  99,  99,  99,  99,  99,

  154.     99,  99,  99,  99,  99,  99,  99,  99,

  155.     99,  99,  99,  99,  99,  99,  99,  99,

  156.     99,  99,  99,  99,  99,  99,  99,  99,

  157.     99,  99,  99,  99,  99,  99,  99,  99

  158. };

  159. #endif

  160. 

  161. /* Set up the standard Huffman tables (cf. JPEG standard section K.3) */

  162. /* IMPORTANT: these are only valid for 8-bit data precision! */

  163. static const uint8_t bits_dc_luminance[17] =

  164. { /* 0-base */ 0, 0, 1, 5, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0 };

  165. static const uint8_t val_dc_luminance[] =

  166. { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11 };

  167. 

  168. static const uint8_t bits_dc_chrominance[17] =

  169. { /* 0-base */ 0, 0, 3, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0 };

  170. static const uint8_t val_dc_chrominance[] =

  171. { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11 };

  172. 

  173. static const uint8_t bits_ac_luminance[17] =

  174. { /* 0-base */ 0, 0, 2, 1, 3, 3, 2, 4, 3, 5, 5, 4, 4, 0, 0, 1, 0x7d };

  175. static const uint8_t val_ac_luminance[] =

  176. { 0x01, 0x02, 0x03, 0x00, 0x04, 0x11, 0x05, 0x12,

  177.   0x21, 0x31, 0x41, 0x06, 0x13, 0x51, 0x61, 0x07,

  178.   0x22, 0x71, 0x14, 0x32, 0x81, 0x91, 0xa1, 0x08,

  179.   0x23, 0x42, 0xb1, 0xc1, 0x15, 0x52, 0xd1, 0xf0,

  180.   0x24, 0x33, 0x62, 0x72, 0x82, 0x09, 0x0a, 0x16,

  181.   0x17, 0x18, 0x19, 0x1a, 0x25, 0x26, 0x27, 0x28,

  182.   0x29, 0x2a, 0x34, 0x35, 0x36, 0x37, 0x38, 0x39,

  183.   0x3a, 0x43, 0x44, 0x45, 0x46, 0x47, 0x48, 0x49,

  184.   0x4a, 0x53, 0x54, 0x55, 0x56, 0x57, 0x58, 0x59,

  185.   0x5a, 0x63, 0x64, 0x65, 0x66, 0x67, 0x68, 0x69,

  186.   0x6a, 0x73, 0x74, 0x75, 0x76, 0x77, 0x78, 0x79,

  187.   0x7a, 0x83, 0x84, 0x85, 0x86, 0x87, 0x88, 0x89,

  188.   0x8a, 0x92, 0x93, 0x94, 0x95, 0x96, 0x97, 0x98,

  189.   0x99, 0x9a, 0xa2, 0xa3, 0xa4, 0xa5, 0xa6, 0xa7,

  190.   0xa8, 0xa9, 0xaa, 0xb2, 0xb3, 0xb4, 0xb5, 0xb6,

  191.   0xb7, 0xb8, 0xb9, 0xba, 0xc2, 0xc3, 0xc4, 0xc5,

  192.   0xc6, 0xc7, 0xc8, 0xc9, 0xca, 0xd2, 0xd3, 0xd4,

  193.   0xd5, 0xd6, 0xd7, 0xd8, 0xd9, 0xda, 0xe1, 0xe2,

  194.   0xe3, 0xe4, 0xe5, 0xe6, 0xe7, 0xe8, 0xe9, 0xea,

  195.   0xf1, 0xf2, 0xf3, 0xf4, 0xf5, 0xf6, 0xf7, 0xf8,

  196.   0xf9, 0xfa 

  197. };

  198. 

  199. static const uint8_t bits_ac_chrominance[17] =

  200. { /* 0-base */ 0, 0, 2, 1, 2, 4, 4, 3, 4, 7, 5, 4, 4, 0, 1, 2, 0x77 };

  201. 

  202. static const uint8_t val_ac_chrominance[] =

  203. { 0x00, 0x01, 0x02, 0x03, 0x11, 0x04, 0x05, 0x21,

  204.   0x31, 0x06, 0x12, 0x41, 0x51, 0x07, 0x61, 0x71,

  205.   0x13, 0x22, 0x32, 0x81, 0x08, 0x14, 0x42, 0x91,

  206.   0xa1, 0xb1, 0xc1, 0x09, 0x23, 0x33, 0x52, 0xf0,

  207.   0x15, 0x62, 0x72, 0xd1, 0x0a, 0x16, 0x24, 0x34,

  208.   0xe1, 0x25, 0xf1, 0x17, 0x18, 0x19, 0x1a, 0x26,

  209.   0x27, 0x28, 0x29, 0x2a, 0x35, 0x36, 0x37, 0x38,

  210.   0x39, 0x3a, 0x43, 0x44, 0x45, 0x46, 0x47, 0x48,

  211.   0x49, 0x4a, 0x53, 0x54, 0x55, 0x56, 0x57, 0x58,

  212.   0x59, 0x5a, 0x63, 0x64, 0x65, 0x66, 0x67, 0x68,

  213.   0x69, 0x6a, 0x73, 0x74, 0x75, 0x76, 0x77, 0x78,

  214.   0x79, 0x7a, 0x82, 0x83, 0x84, 0x85, 0x86, 0x87,

  215.   0x88, 0x89, 0x8a, 0x92, 0x93, 0x94, 0x95, 0x96,

  216.   0x97, 0x98, 0x99, 0x9a, 0xa2, 0xa3, 0xa4, 0xa5,

  217.   0xa6, 0xa7, 0xa8, 0xa9, 0xaa, 0xb2, 0xb3, 0xb4,

  218.   0xb5, 0xb6, 0xb7, 0xb8, 0xb9, 0xba, 0xc2, 0xc3,

  219.   0xc4, 0xc5, 0xc6, 0xc7, 0xc8, 0xc9, 0xca, 0xd2,

  220.   0xd3, 0xd4, 0xd5, 0xd6, 0xd7, 0xd8, 0xd9, 0xda,

  221.   0xe2, 0xe3, 0xe4, 0xe5, 0xe6, 0xe7, 0xe8, 0xe9,

  222.   0xea, 0xf2, 0xf3, 0xf4, 0xf5, 0xf6, 0xf7, 0xf8,

  223.   0xf9, 0xfa 

  224. };

  225. 

  226. /* isn't this function nicer than the one in the libjpeg ? */

  227. static void build_huffman_codes(uint8_t *huff_size, uint16_t *huff_code,

  228.                                 const uint8_t *bits_table, const uint8_t *val_table)

  229. {

  230.     int i, j, k,nb, code, sym;

  231. 

  232.     code = 0;

  233.     k = 0;

  234.     for(i=1;i<=16;i++) {

  235.         nb = bits_table[i];

  236.         for(j=0;j<nb;j++) {

  237.             sym = val_table[k++];

  238.             huff_size[sym] = i;

  239.             huff_code[sym] = code;

  240.             code++;

  241.         }

  242.         code <<= 1;

  243.     }

  244. }

  245. 

  246. int mjpeg_init(MpegEncContext *s)

  247. {

  248.     MJpegContext *m;

  249.     

  250.     m = av_malloc(sizeof(MJpegContext));

  251.     if (!m)

  252.         return -1;

  253.     

  254.     s->min_qcoeff=-1023;

  255.     s->max_qcoeff= 1023;

  256. 

  257.     /* build all the huffman tables */

  258.     build_huffman_codes(m->huff_size_dc_luminance,

  259.                         m->huff_code_dc_luminance,

  260.                         bits_dc_luminance,

  261.                         val_dc_luminance);

  262.     build_huffman_codes(m->huff_size_dc_chrominance,

  263.                         m->huff_code_dc_chrominance,

  264.                         bits_dc_chrominance,

  265.                         val_dc_chrominance);

  266.     build_huffman_codes(m->huff_size_ac_luminance,

  267.                         m->huff_code_ac_luminance,

  268.                         bits_ac_luminance,

  269.                         val_ac_luminance);

  270.     build_huffman_codes(m->huff_size_ac_chrominance,

  271.                         m->huff_code_ac_chrominance,

  272.                         bits_ac_chrominance,

  273.                         val_ac_chrominance);

  274.     

  275.     s->mjpeg_ctx = m;

  276.     return 0;

  277. }

  278. 

  279. void mjpeg_close(MpegEncContext *s)

  280. {

  281.     av_free(s->mjpeg_ctx);

  282. }

  283. 

  284. #define PREDICT(ret, topleft, top, left, predictor)\

  285.     switch(predictor){\

  286.         case 1: ret= left; break;\

  287.         case 2: ret= top; break;\

  288.         case 3: ret= topleft; break;\

  289.         case 4: ret= left   +   top - topleft; break;\

  290.         case 5: ret= left   + ((top - topleft)>>1); break;\

  291.         case 6: ret= top + ((left   - topleft)>>1); break;\

  292.         default:\

  293.         case 7: ret= (left + top)>>1; break;\

  294.     }

  295. 

  296. #ifdef CONFIG_ENCODERS

  297. static inline void put_marker(PutBitContext *p, int code)

  298. {

  299.     put_bits(p, 8, 0xff);

  300.     put_bits(p, 8, code);

  301. }

  302. 

  303. /* table_class: 0 = DC coef, 1 = AC coefs */

  304. static int put_huffman_table(MpegEncContext *s, int table_class, int table_id,

  305.                              const uint8_t *bits_table, const uint8_t *value_table)

  306. {

  307.     PutBitContext *p = &s->pb;

  308.     int n, i;

  309. 

  310.     put_bits(p, 4, table_class);

  311.     put_bits(p, 4, table_id);

  312. 

  313.     n = 0;

  314.     for(i=1;i<=16;i++) {

  315.         n += bits_table[i];

  316.         put_bits(p, 8, bits_table[i]);

  317.     }

  318. 

  319.     for(i=0;i<n;i++)

  320.         put_bits(p, 8, value_table[i]);

  321. 

  322.     return n + 17;

  323. }

  324. 

  325. static void jpeg_table_header(MpegEncContext *s)

  326. {

  327.     PutBitContext *p = &s->pb;

  328.     int i, j, size;

  329.     uint8_t *ptr;

  330. 

  331.     /* quant matrixes */

  332.     put_marker(p, DQT);

  333. #ifdef TWOMATRIXES

  334.     put_bits(p, 16, 2 + 2 * (1 + 64));

  335. #else

  336.     put_bits(p, 16, 2 + 1 * (1 + 64));

  337. #endif

  338.     put_bits(p, 4, 0); /* 8 bit precision */

  339.     put_bits(p, 4, 0); /* table 0 */

  340.     for(i=0;i<64;i++) {

  341.         j = s->intra_scantable.permutated[i];

  342.         put_bits(p, 8, s->intra_matrix[j]);

  343.     }

  344. #ifdef TWOMATRIXES

  345.     put_bits(p, 4, 0); /* 8 bit precision */

  346.     put_bits(p, 4, 1); /* table 1 */

  347.     for(i=0;i<64;i++) {

  348.         j = s->intra_scantable.permutated[i];

  349.         put_bits(p, 8, s->chroma_intra_matrix[j]);

  350.     }

  351. #endif

  352. 

  353.     /* huffman table */

  354.     put_marker(p, DHT);

  355.     flush_put_bits(p);

  356.     ptr = pbBufPtr(p);

  357.     put_bits(p, 16, 0); /* patched later */

  358.     size = 2;

  359.     size += put_huffman_table(s, 0, 0, bits_dc_luminance, val_dc_luminance);

  360.     size += put_huffman_table(s, 0, 1, bits_dc_chrominance, val_dc_chrominance);

  361.     

  362.     size += put_huffman_table(s, 1, 0, bits_ac_luminance, val_ac_luminance);

  363.     size += put_huffman_table(s, 1, 1, bits_ac_chrominance, val_ac_chrominance);

  364.     ptr[0] = size >> 8;

  365.     ptr[1] = size;

  366. }

  367. 

  368. static void jpeg_put_comments(MpegEncContext *s)

  369. {

  370.     PutBitContext *p = &s->pb;

  371.     int size;

  372.     uint8_t *ptr;

  373. 

  374.     if (s->aspect_ratio_info /* && !lossless */)

  375.     {

  376.     /* JFIF header */

  377.     put_marker(p, APP0);

  378.     put_bits(p, 16, 16);

  379.     put_string(p, "JFIF"); /* this puts the trailing zero-byte too */

  380.     put_bits(p, 16, 0x0201); /* v 1.02 */

  381.     put_bits(p, 8, 0); /* units type: 0 - aspect ratio */

  382.     switch(s->aspect_ratio_info)

  383.     {

  384. 	case FF_ASPECT_4_3_625:

  385. 	case FF_ASPECT_4_3_525:

  386. 	    put_bits(p, 16, 4); 

  387. 	    put_bits(p, 16, 3);

  388. 	    break;

  389. 	case FF_ASPECT_16_9_625:

  390. 	case FF_ASPECT_16_9_525:

  391. 	    put_bits(p, 16, 16); 

  392. 	    put_bits(p, 16, 9);

  393. 	    break;

  394. 	case FF_ASPECT_EXTENDED:

  395. 	    put_bits(p, 16, s->aspected_width);

  396. 	    put_bits(p, 16, s->aspected_height);

  397. 	    break;

  398. 	case FF_ASPECT_SQUARE:

  399. 	default:

  400. 	    put_bits(p, 16, 1); /* aspect: 1:1 */

  401. 	    put_bits(p, 16, 1);

  402. 	    break;

  403.     }

  404.     put_bits(p, 8, 0); /* thumbnail width */

  405.     put_bits(p, 8, 0); /* thumbnail height */

  406.     }

  407. 

  408.     /* comment */

  409.     if(!(s->flags & CODEC_FLAG_BITEXACT)){

  410.         put_marker(p, COM);

  411.         flush_put_bits(p);

  412.         ptr = pbBufPtr(p);

  413.         put_bits(p, 16, 0); /* patched later */

  414.         put_string(p, LIBAVCODEC_IDENT);

  415.         size = strlen(LIBAVCODEC_IDENT)+3;

  416.         ptr[0] = size >> 8;

  417.         ptr[1] = size;

  418.     }

  419. }

  420. 

  421. void mjpeg_picture_header(MpegEncContext *s)

  422. {

  423.     const int lossless= s->avctx->codec_id == CODEC_ID_LJPEG;

  424. 

  425.     put_marker(&s->pb, SOI);

  426. 

  427.     if (!s->mjpeg_data_only_frames)

  428.     {

  429.     jpeg_put_comments(s);    

  430. 

  431.     if (s->mjpeg_write_tables) jpeg_table_header(s);

  432. 

  433.     put_marker(&s->pb, lossless ? SOF3 : SOF0);

  434. 

  435.     put_bits(&s->pb, 16, 17);

  436.     if(lossless && s->avctx->pix_fmt == PIX_FMT_RGBA32)

  437.         put_bits(&s->pb, 8, 9); /* 9 bits/component RCT */

  438.     else

  439.         put_bits(&s->pb, 8, 8); /* 8 bits/component */

  440.     put_bits(&s->pb, 16, s->height);

  441.     put_bits(&s->pb, 16, s->width);

  442.     put_bits(&s->pb, 8, 3); /* 3 components */

  443.     

  444.     /* Y component */

  445.     put_bits(&s->pb, 8, 1); /* component number */

  446.     put_bits(&s->pb, 4, s->mjpeg_hsample[0]); /* H factor */

  447.     put_bits(&s->pb, 4, s->mjpeg_vsample[0]); /* V factor */

  448.     put_bits(&s->pb, 8, 0); /* select matrix */

  449.     

  450.     /* Cb component */

  451.     put_bits(&s->pb, 8, 2); /* component number */

  452.     put_bits(&s->pb, 4, s->mjpeg_hsample[1]); /* H factor */

  453.     put_bits(&s->pb, 4, s->mjpeg_vsample[1]); /* V factor */

  454. #ifdef TWOMATRIXES

  455.     put_bits(&s->pb, 8, lossless ? 0 : 1); /* select matrix */

  456. #else

  457.     put_bits(&s->pb, 8, 0); /* select matrix */

  458. #endif

  459. 

  460.     /* Cr component */

  461.     put_bits(&s->pb, 8, 3); /* component number */

  462.     put_bits(&s->pb, 4, s->mjpeg_hsample[2]); /* H factor */

  463.     put_bits(&s->pb, 4, s->mjpeg_vsample[2]); /* V factor */

  464. #ifdef TWOMATRIXES

  465.     put_bits(&s->pb, 8, lossless ? 0 : 1); /* select matrix */

  466. #else

  467.     put_bits(&s->pb, 8, 0); /* select matrix */

  468. #endif

  469.     }

  470. 

  471.     /* scan header */

  472.     put_marker(&s->pb, SOS);

  473.     put_bits(&s->pb, 16, 12); /* length */

  474.     put_bits(&s->pb, 8, 3); /* 3 components */

  475.     

  476.     /* Y component */

  477.     put_bits(&s->pb, 8, 1); /* index */

  478.     put_bits(&s->pb, 4, 0); /* DC huffman table index */

  479.     put_bits(&s->pb, 4, 0); /* AC huffman table index */

  480.     

  481.     /* Cb component */

  482.     put_bits(&s->pb, 8, 2); /* index */

  483.     put_bits(&s->pb, 4, 1); /* DC huffman table index */

  484.     put_bits(&s->pb, 4, lossless ? 0 : 1); /* AC huffman table index */

  485.     

  486.     /* Cr component */

  487.     put_bits(&s->pb, 8, 3); /* index */

  488.     put_bits(&s->pb, 4, 1); /* DC huffman table index */

  489.     put_bits(&s->pb, 4, lossless ? 0 : 1); /* AC huffman table index */

  490. 

  491.     put_bits(&s->pb, 8, lossless ? s->avctx->prediction_method+1 : 0); /* Ss (not used) */

  492.     put_bits(&s->pb, 8, lossless ? 0 : 63); /* Se (not used) */

  493.     put_bits(&s->pb, 8, 0); /* Ah/Al (not used) */

  494. }

  495. 

  496. static void escape_FF(MpegEncContext *s, int start)

  497. {

  498.     int size= get_bit_count(&s->pb) - start*8;

  499.     int i, ff_count;

  500.     uint8_t *buf= s->pb.buf + start;

  501.     int align= (-(size_t)(buf))&3;

  502.     

  503.     assert((size&7) == 0);

  504.     size >>= 3;

  505.     

  506.     ff_count=0;

  507.     for(i=0; i<size && i<align; i++){

  508.         if(buf[i]==0xFF) ff_count++;

  509.     }

  510.     for(; i<size-15; i+=16){

  511.         int acc, v;

  512. 

  513.         v= *(uint32_t*)(&buf[i]);

  514.         acc= (((v & (v>>4))&0x0F0F0F0F)+0x01010101)&0x10101010;

  515.         v= *(uint32_t*)(&buf[i+4]);

  516.         acc+=(((v & (v>>4))&0x0F0F0F0F)+0x01010101)&0x10101010;

  517.         v= *(uint32_t*)(&buf[i+8]);

  518.         acc+=(((v & (v>>4))&0x0F0F0F0F)+0x01010101)&0x10101010;

  519.         v= *(uint32_t*)(&buf[i+12]);

  520.         acc+=(((v & (v>>4))&0x0F0F0F0F)+0x01010101)&0x10101010;

  521. 

  522.         acc>>=4;

  523.         acc+= (acc>>16);

  524.         acc+= (acc>>8);

  525.         ff_count+= acc&0xFF;

  526.     }

  527.     for(; i<size; i++){

  528.         if(buf[i]==0xFF) ff_count++;

  529.     }

  530. 

  531.     if(ff_count==0) return;

  532.     

  533.     /* skip put bits */

  534.     for(i=0; i<ff_count-3; i+=4)

  535.         put_bits(&s->pb, 32, 0);

  536.     put_bits(&s->pb, (ff_count-i)*8, 0);

  537.     flush_put_bits(&s->pb); 

  538. 

  539.     for(i=size-1; ff_count; i--){

  540.         int v= buf[i];

  541. 

  542.         if(v==0xFF){

  543. //printf("%d %d\n", i, ff_count);

  544.             buf[i+ff_count]= 0;

  545.             ff_count--;

  546.         }

  547. 

  548.         buf[i+ff_count]= v;

  549.     }

  550. }

  551. 

  552. void mjpeg_picture_trailer(MpegEncContext *s)

  553. {

  554.     int pad= (-get_bit_count(&s->pb))&7;

  555.     

  556.     put_bits(&s->pb, pad,0xFF>>(8-pad));

  557.     flush_put_bits(&s->pb);

  558. 

  559.     assert((s->header_bits&7)==0);

  560.     

  561.     escape_FF(s, s->header_bits>>3);

  562. 

  563.     put_marker(&s->pb, EOI);

  564. }

  565. 

  566. static inline void mjpeg_encode_dc(MpegEncContext *s, int val,

  567. 				   uint8_t *huff_size, uint16_t *huff_code)

  568. {

  569.     int mant, nbits;

  570. 

  571.     if (val == 0) {

  572.         put_bits(&s->pb, huff_size[0], huff_code[0]);

  573.     } else {

  574.         mant = val;

  575.         if (val < 0) {

  576.             val = -val;

  577.             mant--;

  578.         }

  579.         

  580.         nbits= av_log2_16bit(val) + 1;

  581.             

  582.         put_bits(&s->pb, huff_size[nbits], huff_code[nbits]);

  583.         

  584.         put_bits(&s->pb, nbits, mant & ((1 << nbits) - 1));

  585.     }

  586. }

  587. 

  588. static void encode_block(MpegEncContext *s, DCTELEM *block, int n)

  589. {

  590.     int mant, nbits, code, i, j;

  591.     int component, dc, run, last_index, val;

  592.     MJpegContext *m = s->mjpeg_ctx;

  593.     uint8_t *huff_size_ac;

  594.     uint16_t *huff_code_ac;

  595.     

  596.     /* DC coef */

  597.     component = (n <= 3 ? 0 : n - 4 + 1);

  598.     dc = block[0]; /* overflow is impossible */

  599.     val = dc - s->last_dc[component];

  600.     if (n < 4) {

  601.         mjpeg_encode_dc(s, val, m->huff_size_dc_luminance, m->huff_code_dc_luminance);

  602.         huff_size_ac = m->huff_size_ac_luminance;

  603.         huff_code_ac = m->huff_code_ac_luminance;

  604.     } else {

  605.         mjpeg_encode_dc(s, val, m->huff_size_dc_chrominance, m->huff_code_dc_chrominance);

  606.         huff_size_ac = m->huff_size_ac_chrominance;

  607.         huff_code_ac = m->huff_code_ac_chrominance;

  608.     }

  609.     s->last_dc[component] = dc;

  610.     

  611.     /* AC coefs */

  612.     

  613.     run = 0;

  614.     last_index = s->block_last_index[n];

  615.     for(i=1;i<=last_index;i++) {

  616.         j = s->intra_scantable.permutated[i];

  617.         val = block[j];

  618.         if (val == 0) {

  619.             run++;

  620.         } else {

  621.             while (run >= 16) {

  622.                 put_bits(&s->pb, huff_size_ac[0xf0], huff_code_ac[0xf0]);

  623.                 run -= 16;

  624.             }

  625.             mant = val;

  626.             if (val < 0) {

  627.                 val = -val;

  628.                 mant--;

  629.             }

  630.             

  631.             nbits= av_log2(val) + 1;

  632.             code = (run << 4) | nbits;

  633. 

  634.             put_bits(&s->pb, huff_size_ac[code], huff_code_ac[code]);

  635.         

  636.             put_bits(&s->pb, nbits, mant & ((1 << nbits) - 1));

  637.             run = 0;

  638.         }

  639.     }

  640. 

  641.     /* output EOB only if not already 64 values */

  642.     if (last_index < 63 || run != 0)

  643.         put_bits(&s->pb, huff_size_ac[0], huff_code_ac[0]);

  644. }

  645. 

  646. void mjpeg_encode_mb(MpegEncContext *s, 

  647.                      DCTELEM block[6][64])

  648. {

  649.     int i;

  650.     for(i=0;i<6;i++) {

  651.         encode_block(s, block[i], i);

  652.     }

  653. }

  654. 

  655. static int encode_picture_lossless(AVCodecContext *avctx, unsigned char *buf, int buf_size, void *data){

  656.     MpegEncContext * const s = avctx->priv_data;

  657.     MJpegContext * const m = s->mjpeg_ctx;

  658.     AVFrame *pict = data;

  659.     const int width= s->width;

  660.     const int height= s->height;

  661.     AVFrame * const p= (AVFrame*)&s->current_picture;

  662.     const int predictor= avctx->prediction_method+1;

  663. 

  664.     init_put_bits(&s->pb, buf, buf_size);

  665. 

  666.     *p = *pict;

  667.     p->pict_type= FF_I_TYPE;

  668.     p->key_frame= 1;

  669.     

  670.     mjpeg_picture_header(s);

  671. 

  672.     s->header_bits= get_bit_count(&s->pb);

  673. 

  674.     if(avctx->pix_fmt == PIX_FMT_RGBA32){

  675.         int x, y, i;

  676.         const int linesize= p->linesize[0];

  677.         uint16_t buffer[2048][4];

  678.         int left[3], top[3], topleft[3];

  679. 

  680.         for(i=0; i<3; i++){

  681.             buffer[0][i]= 1 << (9 - 1);

  682.         }

  683. 

  684.         for(y = 0; y < height; y++) {

  685.             const int modified_predictor= y ? predictor : 1;

  686.             uint8_t *ptr = p->data[0] + (linesize * y);

  687. 

  688.             for(i=0; i<3; i++){

  689.                 top[i]= left[i]= topleft[i]= buffer[0][i];

  690.             }

  691.             for(x = 0; x < width; x++) {

  692.                 buffer[x][1] = ptr[4*x+0] - ptr[4*x+1] + 0x100;

  693.                 buffer[x][2] = ptr[4*x+2] - ptr[4*x+1] + 0x100;

  694.                 buffer[x][0] = (ptr[4*x+0] + 2*ptr[4*x+1] + ptr[4*x+2])>>2;

  695. 

  696.                 for(i=0;i<3;i++) {

  697.                     int pred, diff;

  698. 

  699.                     PREDICT(pred, topleft[i], top[i], left[i], modified_predictor);

  700.                         

  701.                     topleft[i]= top[i];

  702.                     top[i]= buffer[x+1][i];

  703.                     

  704.                     left[i]= buffer[x][i];

  705. 

  706.                     diff= ((left[i] - pred + 0x100)&0x1FF) - 0x100;

  707.                     

  708.                     if(i==0)

  709.                         mjpeg_encode_dc(s, diff, m->huff_size_dc_luminance, m->huff_code_dc_luminance); //FIXME ugly

  710.                     else

  711.                         mjpeg_encode_dc(s, diff, m->huff_size_dc_chrominance, m->huff_code_dc_chrominance);

  712.                 }

  713.             }

  714.         }

  715.     }else{

  716.         int mb_x, mb_y, i;

  717.         const int mb_width  = (width  + s->mjpeg_hsample[0] - 1) / s->mjpeg_hsample[0];

  718.         const int mb_height = (height + s->mjpeg_vsample[0] - 1) / s->mjpeg_vsample[0];

  719.         

  720.         for(mb_y = 0; mb_y < mb_height; mb_y++) {

  721.             for(mb_x = 0; mb_x < mb_width; mb_x++) {

  722.                 if(mb_x==0 || mb_y==0){

  723.                     for(i=0;i<3;i++) {

  724.                         uint8_t *ptr;

  725.                         int x, y, h, v, linesize;

  726.                         h = s->mjpeg_hsample[i];

  727.                         v = s->mjpeg_vsample[i];

  728.                         linesize= p->linesize[i];

  729. 

  730.                         for(y=0; y<v; y++){

  731.                             for(x=0; x<h; x++){

  732.                                 int pred;

  733. 

  734.                                 ptr = p->data[i] + (linesize * (v * mb_y + y)) + (h * mb_x + x); //FIXME optimize this crap

  735.                                 if(y==0 && mb_y==0){

  736.                                     if(x==0 && mb_x==0){

  737.                                         pred= 128;

  738.                                     }else{

  739.                                         pred= ptr[-1];

  740.                                     }

  741.                                 }else{

  742.                                     if(x==0 && mb_x==0){

  743.                                         pred= ptr[-linesize];

  744.                                     }else{

  745.                                         PREDICT(pred, ptr[-linesize-1], ptr[-linesize], ptr[-1], predictor);

  746.                                     }

  747.                                 }

  748.                                 

  749.                                 if(i==0)

  750.                                     mjpeg_encode_dc(s, (int8_t)(*ptr - pred), m->huff_size_dc_luminance, m->huff_code_dc_luminance); //FIXME ugly

  751.                                 else

  752.                                     mjpeg_encode_dc(s, (int8_t)(*ptr - pred), m->huff_size_dc_chrominance, m->huff_code_dc_chrominance);

  753.                             }

  754.                         }

  755.                     }

  756.                 }else{

  757.                     for(i=0;i<3;i++) {

  758.                         uint8_t *ptr;

  759.                         int x, y, h, v, linesize;

  760.                         h = s->mjpeg_hsample[i];

  761.                         v = s->mjpeg_vsample[i];

  762.                         linesize= p->linesize[i];

  763.                              

  764.                         for(y=0; y<v; y++){

  765.                             for(x=0; x<h; x++){

  766.                                 int pred;

  767. 

  768.                                 ptr = p->data[i] + (linesize * (v * mb_y + y)) + (h * mb_x + x); //FIXME optimize this crap

  769. //printf("%d %d %d %d %8X\n", mb_x, mb_y, x, y, ptr); 

  770.                                 PREDICT(pred, ptr[-linesize-1], ptr[-linesize], ptr[-1], predictor);

  771. 

  772.                                 if(i==0)

  773.                                     mjpeg_encode_dc(s, (int8_t)(*ptr - pred), m->huff_size_dc_luminance, m->huff_code_dc_luminance); //FIXME ugly

  774.                                 else

  775.                                     mjpeg_encode_dc(s, (int8_t)(*ptr - pred), m->huff_size_dc_chrominance, m->huff_code_dc_chrominance);

  776.                             }

  777.                         }

  778.                     }

  779.                 }

  780.             }

  781.         }

  782.     }

  783. 

  784.     emms_c();

  785.     

  786.     mjpeg_picture_trailer(s);

  787.     s->picture_number++;

  788. 

  789.     flush_put_bits(&s->pb);

  790.     return pbBufPtr(&s->pb) - s->pb.buf;

  791. //    return (get_bit_count(&f->pb)+7)/8;

  792. }

  793. 

  794. #endif //CONFIG_ENCODERS

  795. 

  796. /******************************************/

  797. /* decoding */

  798. 

  799. #define MAX_COMPONENTS 4

  800. 

  801. typedef struct MJpegDecodeContext {

  802.     AVCodecContext *avctx;

  803.     GetBitContext gb;

  804.     int mpeg_enc_ctx_allocated; /* true if decoding context allocated */

  805. 

  806.     int start_code; /* current start code */

  807.     int buffer_size;

  808.     uint8_t *buffer;

  809. 

  810.     int16_t quant_matrixes[4][64];

  811.     VLC vlcs[2][4];

  812.     int qscale[4];      ///< quantizer scale calculated from quant_matrixes

  813. 

  814.     int org_width, org_height;  /* size given at codec init */

  815.     int first_picture;    /* true if decoding first picture */

  816.     int interlaced;     /* true if interlaced */

  817.     int bottom_field;   /* true if bottom field */

  818.     int lossless;

  819.     int rgb;

  820.     int rct;            /* standard rct */  

  821.     int pegasus_rct;    /* pegasus reversible colorspace transform */  

  822.     int bits;           /* bits per component */

  823. 

  824.     int width, height;

  825.     int mb_width, mb_height;

  826.     int nb_components;

  827.     int component_id[MAX_COMPONENTS];

  828.     int h_count[MAX_COMPONENTS]; /* horizontal and vertical count for each component */

  829.     int v_count[MAX_COMPONENTS];

  830.     int comp_index[MAX_COMPONENTS];

  831.     int dc_index[MAX_COMPONENTS];

  832.     int ac_index[MAX_COMPONENTS];

  833.     int nb_blocks[MAX_COMPONENTS];

  834.     int h_scount[MAX_COMPONENTS];

  835.     int v_scount[MAX_COMPONENTS];

  836.     int h_max, v_max; /* maximum h and v counts */

  837.     int quant_index[4];   /* quant table index for each component */

  838.     int last_dc[MAX_COMPONENTS]; /* last DEQUANTIZED dc (XXX: am I right to do that ?) */

  839.     AVFrame picture; /* picture structure */

  840.     int linesize[MAX_COMPONENTS];                   ///< linesize << interlaced

  841.     uint8_t *qscale_table;

  842.     DCTELEM block[64] __align8;

  843.     ScanTable scantable;

  844.     void (*idct_put)(uint8_t *dest/*align 8*/, int line_size, DCTELEM *block/*align 16*/);

  845. 

  846.     int restart_interval;

  847.     int restart_count;

  848. 

  849.     int buggy_avid;

  850.     int interlace_polarity;

  851. } MJpegDecodeContext;

  852. 

  853. static int mjpeg_decode_dht(MJpegDecodeContext *s);

  854. 

  855. static int build_vlc(VLC *vlc, const uint8_t *bits_table, const uint8_t *val_table, 

  856.                       int nb_codes)

  857. {

  858.     uint8_t huff_size[256];

  859.     uint16_t huff_code[256];

  860. 

  861.     memset(huff_size, 0, sizeof(huff_size));

  862.     build_huffman_codes(huff_size, huff_code, bits_table, val_table);

  863.     

  864.     return init_vlc(vlc, 9, nb_codes, huff_size, 1, 1, huff_code, 2, 2);

  865. }

  866. 

  867. static int mjpeg_decode_init(AVCodecContext *avctx)

  868. {

  869.     MJpegDecodeContext *s = avctx->priv_data;

  870.     MpegEncContext s2;

  871. 

  872.     s->avctx = avctx;

  873. 

  874.     /* ugly way to get the idct & scantable FIXME */

  875.     memset(&s2, 0, sizeof(MpegEncContext));

  876.     s2.flags= avctx->flags;

  877.     s2.avctx= avctx;

  878. //    s2->out_format = FMT_MJPEG;

  879.     s2.width = 8;

  880.     s2.height = 8;

  881.     if (MPV_common_init(&s2) < 0)

  882.        return -1;

  883.     s->scantable= s2.intra_scantable;

  884.     s->idct_put= s2.dsp.idct_put;

  885.     MPV_common_end(&s2);

  886. 

  887.     s->mpeg_enc_ctx_allocated = 0;

  888.     s->buffer_size = 102400; /* smaller buffer should be enough,

  889. 				but photojpg files could ahive bigger sizes */

  890.     s->buffer = av_malloc(s->buffer_size);

  891.     if (!s->buffer)

  892. 	return -1;

  893.     s->start_code = -1;

  894.     s->first_picture = 1;

  895.     s->org_width = avctx->width;

  896.     s->org_height = avctx->height;

  897.     

  898.     build_vlc(&s->vlcs[0][0], bits_dc_luminance, val_dc_luminance, 12);

  899.     build_vlc(&s->vlcs[0][1], bits_dc_chrominance, val_dc_chrominance, 12);

  900.     build_vlc(&s->vlcs[1][0], bits_ac_luminance, val_ac_luminance, 251);

  901.     build_vlc(&s->vlcs[1][1], bits_ac_chrominance, val_ac_chrominance, 251);

  902. 

  903.     if (avctx->flags & CODEC_FLAG_EXTERN_HUFF)

  904.     {

  905. 	printf("mjpeg: using external huffman table\n");

  906. 	init_get_bits(&s->gb, avctx->extradata, avctx->extradata_size*8);

  907. 	mjpeg_decode_dht(s);

  908. 	/* should check for error - but dunno */

  909.     }

  910. 

  911.     return 0;

  912. }

  913. 

  914. /* quantize tables */

  915. static int mjpeg_decode_dqt(MJpegDecodeContext *s)

  916. {

  917.     int len, index, i, j;

  918.     

  919.     len = get_bits(&s->gb, 16) - 2;

  920. 

  921.     while (len >= 65) {

  922.         /* only 8 bit precision handled */

  923.         if (get_bits(&s->gb, 4) != 0)

  924. 	{

  925. 	    dprintf("dqt: 16bit precision\n");

  926.             return -1;

  927. 	}

  928.         index = get_bits(&s->gb, 4);

  929.         if (index >= 4)

  930.             return -1;

  931.         dprintf("index=%d\n", index);

  932.         /* read quant table */

  933.         for(i=0;i<64;i++) {

  934.             j = s->scantable.permutated[i];

  935. 	    s->quant_matrixes[index][j] = get_bits(&s->gb, 8);

  936.         }

  937. 

  938.         //XXX FIXME finetune, and perhaps add dc too

  939.         s->qscale[index]= FFMAX(

  940.             s->quant_matrixes[index][s->scantable.permutated[1]],

  941.             s->quant_matrixes[index][s->scantable.permutated[8]]) >> 1;

  942. 	dprintf("qscale[%d]: %d\n", index, s->qscale[index]);

  943.         len -= 65;

  944.     }

  945.     

  946.     return 0;

  947. }

  948. 

  949. /* decode huffman tables and build VLC decoders */

  950. static int mjpeg_decode_dht(MJpegDecodeContext *s)

  951. {

  952.     int len, index, i, class, n, v, code_max;

  953.     uint8_t bits_table[17];

  954.     uint8_t val_table[256];

  955.     

  956.     len = get_bits(&s->gb, 16) - 2;

  957. 

  958.     while (len > 0) {

  959.         if (len < 17)

  960.             return -1;

  961.         class = get_bits(&s->gb, 4);

  962.         if (class >= 2)

  963.             return -1;

  964.         index = get_bits(&s->gb, 4);

  965.         if (index >= 4)

  966.             return -1;

  967.         n = 0;

  968.         for(i=1;i<=16;i++) {

  969.             bits_table[i] = get_bits(&s->gb, 8);

  970.             n += bits_table[i];

  971.         }

  972.         len -= 17;

  973.         if (len < n || n > 256)

  974.             return -1;

  975. 

  976.         code_max = 0;

  977.         for(i=0;i<n;i++) {

  978.             v = get_bits(&s->gb, 8);

  979.             if (v > code_max)

  980.                 code_max = v;

  981.             val_table[i] = v;

  982.         }

  983.         len -= n;

  984. 

  985.         /* build VLC and flush previous vlc if present */

  986.         free_vlc(&s->vlcs[class][index]);

  987.         dprintf("class=%d index=%d nb_codes=%d\n",

  988.                class, index, code_max + 1);

  989.         if(build_vlc(&s->vlcs[class][index], bits_table, val_table, code_max + 1) < 0){

  990.             return -1;

  991.         }

  992.     }

  993.     return 0;

  994. }

  995. 

  996. static int mjpeg_decode_sof(MJpegDecodeContext *s)

  997. {

  998.     int len, nb_components, i, width, height;

  999. 

  1000.     /* XXX: verify len field validity */

  1001.     len = get_bits(&s->gb, 16);

  1002.     s->bits= get_bits(&s->gb, 8);

  1003.     

  1004.     if(s->pegasus_rct) s->bits=9;  

  1005.     if(s->bits==9 && !s->pegasus_rct) s->rct=1;    //FIXME ugly

  1006. 

  1007.     if (s->bits != 8 && !s->lossless){

  1008.         printf("only 8 bits/component accepted\n");

  1009.         return -1;

  1010.     }

  1011.     height = get_bits(&s->gb, 16);

  1012.     width = get_bits(&s->gb, 16);

  1013.     dprintf("sof0: picture: %dx%d\n", width, height);

  1014. 

  1015.     nb_components = get_bits(&s->gb, 8);

  1016.     if (nb_components <= 0 ||

  1017.         nb_components > MAX_COMPONENTS)

  1018.         return -1;

  1019.     s->nb_components = nb_components;

  1020.     s->h_max = 1;

  1021.     s->v_max = 1;

  1022.     for(i=0;i<nb_components;i++) {

  1023.         /* component id */

  1024.         s->component_id[i] = get_bits(&s->gb, 8) - 1;

  1025.         s->h_count[i] = get_bits(&s->gb, 4);

  1026.         s->v_count[i] = get_bits(&s->gb, 4);

  1027.         /* compute hmax and vmax (only used in interleaved case) */

  1028.         if (s->h_count[i] > s->h_max)

  1029.             s->h_max = s->h_count[i];

  1030.         if (s->v_count[i] > s->v_max)

  1031.             s->v_max = s->v_count[i];

  1032.         s->quant_index[i] = get_bits(&s->gb, 8);

  1033.         if (s->quant_index[i] >= 4)

  1034.             return -1;

  1035.         dprintf("component %d %d:%d id: %d quant:%d\n", i, s->h_count[i],

  1036. 	    s->v_count[i], s->component_id[i], s->quant_index[i]);

  1037.     }

  1038.     

  1039.     if(s->v_max==1 && s->h_max==1 && s->lossless==1) s->rgb=1;

  1040. 

  1041.     /* if different size, realloc/alloc picture */

  1042.     /* XXX: also check h_count and v_count */

  1043.     if (width != s->width || height != s->height) {

  1044.         av_freep(&s->qscale_table);

  1045.             

  1046.         s->width = width;

  1047.         s->height = height;

  1048.         /* test interlaced mode */

  1049.         if (s->first_picture &&

  1050.             s->org_height != 0 &&

  1051.             s->height < ((s->org_height * 3) / 4)) {

  1052.             s->interlaced = 1;

  1053. //	    s->bottom_field = (s->interlace_polarity) ? 1 : 0;

  1054. 	    s->bottom_field = 0;

  1055.         }

  1056. 

  1057.         s->qscale_table= av_mallocz((s->width+15)/16);

  1058. 

  1059.         s->first_picture = 0;

  1060.     }

  1061.     

  1062.     if(s->interlaced && s->bottom_field)

  1063.         return 0;

  1064.  

  1065.     /* XXX: not complete test ! */

  1066.     switch((s->h_count[0] << 4) | s->v_count[0]) {

  1067.     case 0x11:

  1068.         if(s->rgb){

  1069.             s->avctx->pix_fmt = PIX_FMT_RGBA32;

  1070.         }else

  1071.             s->avctx->pix_fmt = PIX_FMT_YUV444P;

  1072.         break;

  1073.     case 0x21:

  1074.         s->avctx->pix_fmt = PIX_FMT_YUV422P;

  1075.         break;

  1076.     default:

  1077.     case 0x22:

  1078.         s->avctx->pix_fmt = PIX_FMT_YUV420P;

  1079.         break;

  1080.     }

  1081. 

  1082.     if(s->picture.data[0])

  1083.         s->avctx->release_buffer(s->avctx, &s->picture);

  1084. 

  1085.     s->picture.reference= 0;

  1086.     if(s->avctx->get_buffer(s->avctx, &s->picture) < 0){

  1087.         fprintf(stderr, "get_buffer() failed\n");

  1088.         return -1;

  1089.     }

  1090.     s->picture.pict_type= I_TYPE;

  1091.     s->picture.key_frame= 1;

  1092.     

  1093.     for(i=0; i<3; i++){

  1094.         s->linesize[i]= s->picture.linesize[i] << s->interlaced;

  1095.     }

  1096. 

  1097. //    printf("%d %d %d %d %d %d\n", s->width, s->height, s->linesize[0], s->linesize[1], s->interlaced, s->avctx->height);

  1098.     

  1099.     if (len != (8+(3*nb_components)))

  1100.     {

  1101. 	dprintf("decode_sof0: error, len(%d) mismatch\n", len);

  1102.     }

  1103.     

  1104.     return 0;

  1105. }

  1106. 

  1107. static inline int mjpeg_decode_dc(MJpegDecodeContext *s, int dc_index)

  1108. {

  1109.     int code;

  1110.     code = get_vlc2(&s->gb, s->vlcs[0][dc_index].table, 9, 2);

  1111.     if (code < 0)

  1112.     {

  1113. 	dprintf("mjpeg_decode_dc: bad vlc: %d:%d (%p)\n", 0, dc_index,

  1114.                 &s->vlcs[0][dc_index]);

  1115.         return 0xffff;

  1116.     }

  1117. 

  1118.     if(code)

  1119.         return get_xbits(&s->gb, code);

  1120.     else

  1121.         return 0;

  1122. }

  1123. 

  1124. /* decode block and dequantize */

  1125. static int decode_block(MJpegDecodeContext *s, DCTELEM *block, 

  1126.                         int component, int dc_index, int ac_index, int quant_index)

  1127. {

  1128.     int code, i, j, level, val;

  1129.     VLC *ac_vlc;

  1130.     int16_t *quant_matrix;

  1131. 

  1132.     /* DC coef */

  1133.     val = mjpeg_decode_dc(s, dc_index);

  1134.     if (val == 0xffff) {

  1135.         dprintf("error dc\n");

  1136.         return -1;

  1137.     }

  1138.     quant_matrix = s->quant_matrixes[quant_index];

  1139.     val = val * quant_matrix[0] + s->last_dc[component];

  1140.     s->last_dc[component] = val;

  1141.     block[0] = val;

  1142.     /* AC coefs */

  1143.     ac_vlc = &s->vlcs[1][ac_index];

  1144.     i = 1;

  1145.     for(;;) {

  1146. 	code = get_vlc2(&s->gb, s->vlcs[1][ac_index].table, 9, 2);

  1147. 

  1148.         if (code < 0) {

  1149.             dprintf("error ac\n");

  1150.             return -1;

  1151.         }

  1152.         /* EOB */

  1153.         if (code == 0)

  1154.             break;

  1155.         if (code == 0xf0) {

  1156.             i += 16;

  1157.         } else {

  1158.             level = get_xbits(&s->gb, code & 0xf);

  1159.             i += code >> 4;

  1160.             if (i >= 64) {

  1161.                 dprintf("error count: %d\n", i);

  1162.                 return -1;

  1163.             }

  1164.             j = s->scantable.permutated[i];

  1165.             block[j] = level * quant_matrix[j];

  1166.             i++;

  1167.             if (i >= 64)

  1168.                 break;

  1169.         }

  1170.     }

  1171.     return 0;

  1172. }

  1173. 

  1174. static int ljpeg_decode_rgb_scan(MJpegDecodeContext *s, int predictor, int point_transform){

  1175.     int i, mb_x, mb_y;

  1176.     uint16_t buffer[2048][4];

  1177.     int left[3], top[3], topleft[3];

  1178.     const int linesize= s->linesize[0];

  1179.     const int mask= (1<<s->bits)-1;

  1180.     

  1181.     for(i=0; i<3; i++){

  1182.         buffer[0][i]= 1 << (s->bits + point_transform - 1);

  1183.     }

  1184.     for(mb_y = 0; mb_y < s->mb_height; mb_y++) {

  1185.         const int modified_predictor= mb_y ? predictor : 1;

  1186.         uint8_t *ptr = s->picture.data[0] + (linesize * mb_y);

  1187. 

  1188.         if (s->interlaced && s->bottom_field)

  1189.             ptr += linesize >> 1;

  1190. 

  1191.         for(i=0; i<3; i++){

  1192.             top[i]= left[i]= topleft[i]= buffer[0][i];

  1193.         }

  1194.         for(mb_x = 0; mb_x < s->mb_width; mb_x++) {

  1195.             if (s->restart_interval && !s->restart_count)

  1196.                 s->restart_count = s->restart_interval;

  1197. 

  1198.             for(i=0;i<3;i++) {

  1199.                 int pred;

  1200. 

  1201.                 topleft[i]= top[i];

  1202.                 top[i]= buffer[mb_x][i];

  1203. 

  1204.                 PREDICT(pred, topleft[i], top[i], left[i], modified_predictor);

  1205.                 

  1206.                 left[i]= 

  1207.                 buffer[mb_x][i]= mask & (pred + (mjpeg_decode_dc(s, s->dc_index[i]) << point_transform));

  1208.             }

  1209. 

  1210.             if (s->restart_interval && !--s->restart_count) {

  1211.                 align_get_bits(&s->gb);

  1212.                 skip_bits(&s->gb, 16); /* skip RSTn */

  1213.             }

  1214.         }

  1215. 

  1216.         if(s->rct){

  1217.             for(mb_x = 0; mb_x < s->mb_width; mb_x++) {

  1218.                 ptr[4*mb_x+1] = buffer[mb_x][0] - ((buffer[mb_x][1] + buffer[mb_x][2] - 0x200)>>2);

  1219.                 ptr[4*mb_x+0] = buffer[mb_x][1] + ptr[4*mb_x+1];

  1220.                 ptr[4*mb_x+2] = buffer[mb_x][2] + ptr[4*mb_x+1];

  1221.             }

  1222.         }else if(s->pegasus_rct){

  1223.             for(mb_x = 0; mb_x < s->mb_width; mb_x++) {

  1224.                 ptr[4*mb_x+1] = buffer[mb_x][0] - ((buffer[mb_x][1] + buffer[mb_x][2])>>2);

  1225.                 ptr[4*mb_x+0] = buffer[mb_x][1] + ptr[4*mb_x+1];

  1226.                 ptr[4*mb_x+2] = buffer[mb_x][2] + ptr[4*mb_x+1];

  1227.             }

  1228.         }else{

  1229.             for(mb_x = 0; mb_x < s->mb_width; mb_x++) {

  1230.                 ptr[4*mb_x+0] = buffer[mb_x][0];

  1231.                 ptr[4*mb_x+1] = buffer[mb_x][1];

  1232.                 ptr[4*mb_x+2] = buffer[mb_x][2];

  1233.             }

  1234.         }

  1235.     }

  1236.     return 0;

  1237. }

  1238. 

  1239. static int ljpeg_decode_yuv_scan(MJpegDecodeContext *s, int predictor, int point_transform){

  1240.     int i, mb_x, mb_y;

  1241.     const int nb_components=3;

  1242. 

  1243.     for(mb_y = 0; mb_y < s->mb_height; mb_y++) {

  1244.         for(mb_x = 0; mb_x < s->mb_width; mb_x++) {

  1245.             if (s->restart_interval && !s->restart_count)

  1246.                 s->restart_count = s->restart_interval;

  1247. 

  1248.             if(mb_x==0 || mb_y==0 || s->interlaced){

  1249.                 for(i=0;i<nb_components;i++) {

  1250.                     uint8_t *ptr;

  1251.                     int n, h, v, x, y, c, j, linesize;

  1252.                     n = s->nb_blocks[i];

  1253.                     c = s->comp_index[i];

  1254.                     h = s->h_scount[i];

  1255.                     v = s->v_scount[i];

  1256.                     x = 0;

  1257.                     y = 0;

  1258.                     linesize= s->linesize[c];

  1259.                     

  1260.                     for(j=0; j<n; j++) {

  1261.                         int pred;

  1262. 

  1263.                         ptr = s->picture.data[c] + (linesize * (v * mb_y + y)) + (h * mb_x + x); //FIXME optimize this crap

  1264.                         if(y==0 && mb_y==0){

  1265.                             if(x==0 && mb_x==0){

  1266.                                 pred= 128 << point_transform;

  1267.                             }else{

  1268.                                 pred= ptr[-1];

  1269.                             }

  1270.                         }else{

  1271.                             if(x==0 && mb_x==0){

  1272.                                 pred= ptr[-linesize];

  1273.                             }else{

  1274.                                 PREDICT(pred, ptr[-linesize-1], ptr[-linesize], ptr[-1], predictor);

  1275.                             }

  1276.                         }

  1277.                         

  1278.                         if (s->interlaced && s->bottom_field)

  1279.                             ptr += linesize >> 1;

  1280.                         *ptr= pred + (mjpeg_decode_dc(s, s->dc_index[i]) << point_transform);

  1281. 

  1282.                         if (++x == h) {

  1283.                             x = 0;

  1284.                             y++;

  1285.                         }

  1286.                     }

  1287.                 }

  1288.             }else{

  1289.                 for(i=0;i<nb_components;i++) {

  1290.                     uint8_t *ptr;

  1291.                     int n, h, v, x, y, c, j, linesize;

  1292.                     n = s->nb_blocks[i];

  1293.                     c = s->comp_index[i];

  1294.                     h = s->h_scount[i];

  1295.                     v = s->v_scount[i];

  1296.                     x = 0;

  1297.                     y = 0;

  1298.                     linesize= s->linesize[c];

  1299.                     

  1300.                     for(j=0; j<n; j++) {

  1301.                         int pred;

  1302. 

  1303.                         ptr = s->picture.data[c] + (linesize * (v * mb_y + y)) + (h * mb_x + x); //FIXME optimize this crap

  1304.                         PREDICT(pred, ptr[-linesize-1], ptr[-linesize], ptr[-1], predictor);

  1305.                         *ptr= pred + (mjpeg_decode_dc(s, s->dc_index[i]) << point_transform);

  1306.                         if (++x == h) {

  1307.                             x = 0;

  1308.                             y++;

  1309.                         }

  1310.                     }

  1311.                 }

  1312.             }

  1313.             if (s->restart_interval && !--s->restart_count) {

  1314.                 align_get_bits(&s->gb);

  1315.                 skip_bits(&s->gb, 16); /* skip RSTn */

  1316.             }

  1317.         }

  1318.     }

  1319.     return 0;

  1320. }

  1321. 

  1322. static int mjpeg_decode_scan(MJpegDecodeContext *s){

  1323.     int i, mb_x, mb_y;

  1324.     const int nb_components=3;

  1325. 

  1326.     for(mb_y = 0; mb_y < s->mb_height; mb_y++) {

  1327.         for(mb_x = 0; mb_x < s->mb_width; mb_x++) {

  1328.             if (s->restart_interval && !s->restart_count)

  1329.                 s->restart_count = s->restart_interval;

  1330. 

  1331.             for(i=0;i<nb_components;i++) {

  1332.                 uint8_t *ptr;

  1333.                 int n, h, v, x, y, c, j;

  1334.                 n = s->nb_blocks[i];

  1335.                 c = s->comp_index[i];

  1336.                 h = s->h_scount[i];

  1337.                 v = s->v_scount[i];

  1338.                 x = 0;

  1339.                 y = 0;

  1340.                 for(j=0;j<n;j++) {

  1341.                     memset(s->block, 0, sizeof(s->block));

  1342.                     if (decode_block(s, s->block, i, 

  1343.                                      s->dc_index[i], s->ac_index[i], 

  1344.                                      s->quant_index[c]) < 0) {

  1345.                         dprintf("error y=%d x=%d\n", mb_y, mb_x);

  1346.                         return -1;

  1347.                     }

  1348. //		    dprintf("mb: %d %d processed\n", mb_y, mb_x);

  1349.                     ptr = s->picture.data[c] + 

  1350.                         (s->linesize[c] * (v * mb_y + y) * 8) + 

  1351.                         (h * mb_x + x) * 8;

  1352.                     if (s->interlaced && s->bottom_field)

  1353.                         ptr += s->linesize[c] >> 1;

  1354. //printf("%d %d %d %d %d %d %d %d \n", mb_x, mb_y, x, y, c, s->bottom_field, (v * mb_y + y) * 8, (h * mb_x + x) * 8);

  1355.                     s->idct_put(ptr, s->linesize[c], s->block);

  1356.                     if (++x == h) {

  1357.                         x = 0;

  1358.                         y++;

  1359.                     }

  1360.                 }

  1361.             }

  1362.             /* (< 1350) buggy workaround for Spectralfan.mov, should be fixed */

  1363.             if (s->restart_interval && (s->restart_interval < 1350) &&

  1364.                 !--s->restart_count) {

  1365.                 align_get_bits(&s->gb);

  1366.                 skip_bits(&s->gb, 16); /* skip RSTn */

  1367.                 for (i=0; i<nb_components; i++) /* reset dc */

  1368.                     s->last_dc[i] = 1024;

  1369.             }

  1370.         }

  1371.     }

  1372.     return 0;

  1373. }

  1374. 

  1375. static int mjpeg_decode_sos(MJpegDecodeContext *s)

  1376. {

  1377.     int len, nb_components, i, h, v, predictor, point_transform;

  1378.     int vmax, hmax, index, id;

  1379.     const int block_size= s->lossless ? 1 : 8;

  1380. 

  1381.     /* XXX: verify len field validity */

  1382.     len = get_bits(&s->gb, 16);

  1383.     nb_components = get_bits(&s->gb, 8);

  1384.     if (len != 6+2*nb_components)

  1385.     {

  1386. 	dprintf("decode_sos: invalid len (%d)\n", len);

  1387. 	return -1;

  1388.     }

  1389.     /* XXX: only interleaved scan accepted */

  1390.     if (nb_components != 3)

  1391.     {

  1392. 	dprintf("decode_sos: components(%d) mismatch\n", nb_components);

  1393.         return -1;

  1394.     }

  1395.     vmax = 0;

  1396.     hmax = 0;

  1397.     for(i=0;i<nb_components;i++) {

  1398.         id = get_bits(&s->gb, 8) - 1;

  1399. 	dprintf("component: %d\n", id);

  1400.         /* find component index */

  1401.         for(index=0;index<s->nb_components;index++)

  1402.             if (id == s->component_id[index])

  1403.                 break;

  1404.         if (index == s->nb_components)

  1405. 	{

  1406. 	    dprintf("decode_sos: index(%d) out of components\n", index);

  1407.             return -1;

  1408. 	}

  1409. 

  1410.         s->comp_index[i] = index;

  1411. 

  1412.         s->nb_blocks[i] = s->h_count[index] * s->v_count[index];

  1413.         s->h_scount[i] = s->h_count[index];

  1414.         s->v_scount[i] = s->v_count[index];

  1415. 

  1416.         s->dc_index[i] = get_bits(&s->gb, 4);

  1417.         s->ac_index[i] = get_bits(&s->gb, 4);

  1418. 

  1419. 	if (s->dc_index[i] <  0 || s->ac_index[i] < 0 ||

  1420. 	    s->dc_index[i] >= 4 || s->ac_index[i] >= 4)

  1421. 	    goto out_of_range;

  1422. #if 0 //buggy

  1423. 	switch(s->start_code)

  1424. 	{

  1425. 	    case SOF0:

  1426. 		if (dc_index[i] > 1 || ac_index[i] > 1)

  1427. 		    goto out_of_range;

  1428. 		break;

  1429. 	    case SOF1:

  1430. 	    case SOF2:

  1431. 		if (dc_index[i] > 3 || ac_index[i] > 3)

  1432. 		    goto out_of_range;

  1433. 		break;

  1434. 	    case SOF3:

  1435. 		if (dc_index[i] > 3 || ac_index[i] != 0)

  1436. 		    goto out_of_range;

  1437. 		break;	

  1438. 	}

  1439. #endif

  1440.     }

  1441. 

  1442.     predictor= get_bits(&s->gb, 8); /* lossless predictor or start of spectral (Ss) */

  1443.     skip_bits(&s->gb, 8); /* Se */

  1444.     skip_bits(&s->gb, 4); /* Ah */

  1445.     point_transform= get_bits(&s->gb, 4); /* Al */

  1446. 

  1447.     for(i=0;i<nb_components;i++) 

  1448.         s->last_dc[i] = 1024;

  1449. 

  1450.     if (nb_components > 1) {

  1451.         /* interleaved stream */

  1452.         s->mb_width  = (s->width  + s->h_max * block_size - 1) / (s->h_max * block_size);

  1453.         s->mb_height = (s->height + s->v_max * block_size - 1) / (s->v_max * block_size);

  1454.     } else {

  1455.         h = s->h_max / s->h_scount[s->comp_index[0]];

  1456.         v = s->v_max / s->v_scount[s->comp_index[0]];

  1457.         s->mb_width  = (s->width  + h * block_size - 1) / (h * block_size);

  1458.         s->mb_height = (s->height + v * block_size - 1) / (v * block_size);

  1459.         s->nb_blocks[0] = 1;

  1460.         s->h_scount[0] = 1;

  1461.         s->v_scount[0] = 1;

  1462.     }

  1463. 

  1464.     if(s->avctx->debug & FF_DEBUG_PICT_INFO)

  1465.         printf("%s %s p:%d >>:%d\n", s->lossless ? "lossless" : "sequencial DCT", s->rgb ? "RGB" : "", predictor, point_transform);

  1466.     

  1467.     if(s->lossless){

  1468.             if(s->rgb){

  1469.                 if(ljpeg_decode_rgb_scan(s, predictor, point_transform) < 0)

  1470.                     return -1;

  1471.             }else{

  1472.                 if(ljpeg_decode_yuv_scan(s, predictor, point_transform) < 0)

  1473.                     return -1;

  1474.             }

  1475.     }else{

  1476.         if(mjpeg_decode_scan(s) < 0)

  1477.             return -1;

  1478.     }

  1479.     emms_c();

  1480.     return 0;

  1481.  out_of_range:

  1482.     dprintf("decode_sos: ac/dc index out of range\n");

  1483.     return -1;

  1484. }

  1485. 

  1486. static int mjpeg_decode_dri(MJpegDecodeContext *s)

  1487. {

  1488.     if (get_bits(&s->gb, 16) != 4)

  1489. 	return -1;

  1490.     s->restart_interval = get_bits(&s->gb, 16);

  1491.     dprintf("restart interval: %d\n", s->restart_interval);

  1492. 

  1493.     return 0;

  1494. }

  1495. 

  1496. static int mjpeg_decode_app(MJpegDecodeContext *s)

  1497. {

  1498.     int len, id;

  1499. 

  1500.     /* XXX: verify len field validity */

  1501.     len = get_bits(&s->gb, 16);

  1502.     if (len < 5)

  1503. 	return -1;

  1504. 

  1505.     id = (get_bits(&s->gb, 16) << 16) | get_bits(&s->gb, 16);

  1506.     id = be2me_32(id);

  1507.     len -= 6;

  1508. 

  1509.     if(s->avctx->debug & FF_DEBUG_STARTCODE){

  1510.         printf("APPx %8X\n", id); 

  1511.     }

  1512.     

  1513.     /* buggy AVID, it puts EOI only at every 10th frame */

  1514.     /* also this fourcc is used by non-avid files too, it holds some

  1515.        informations, but it's always present in AVID creates files */

  1516.     if (id == ff_get_fourcc("AVI1"))

  1517.     {

  1518. 	/* structure:

  1519. 	    4bytes	AVI1

  1520. 	    1bytes	polarity

  1521. 	    1bytes	always zero

  1522. 	    4bytes	field_size

  1523. 	    4bytes	field_size_less_padding

  1524. 	*/

  1525.     	s->buggy_avid = 1;

  1526. //	if (s->first_picture)

  1527. //	    printf("mjpeg: workarounding buggy AVID\n");

  1528. 	s->interlace_polarity = get_bits(&s->gb, 8);

  1529. #if 0

  1530. 	skip_bits(&s->gb, 8);

  1531. 	skip_bits(&s->gb, 32);

  1532. 	skip_bits(&s->gb, 32);

  1533. 	len -= 10;

  1534. #endif

  1535. //	if (s->interlace_polarity)

  1536. //	    printf("mjpeg: interlace polarity: %d\n", s->interlace_polarity);

  1537. 	goto out;

  1538.     }

  1539.     

  1540. //    len -= 2;

  1541.     

  1542.     if (id == ff_get_fourcc("JFIF"))

  1543.     {

  1544. 	int t_w, t_h;

  1545. 	skip_bits(&s->gb, 8); /* the trailing zero-byte */

  1546. 	printf("mjpeg: JFIF header found (version: %x.%x)\n",

  1547. 	    get_bits(&s->gb, 8), get_bits(&s->gb, 8));

  1548. 	if (get_bits(&s->gb, 8) == 0)

  1549. 	{

  1550. 	    int x_density, y_density; 

  1551. 	    x_density = get_bits(&s->gb, 16);

  1552. 	    y_density = get_bits(&s->gb, 16);

  1553. 

  1554. 	    dprintf("x/y density: %d (%f), %d (%f)\n", x_density,

  1555. 		(float)x_density, y_density, (float)y_density);

  1556. #if 0

  1557.             //MN: needs to be checked

  1558.             if(x_density)

  1559. //                s->avctx->aspect_ratio= s->width*y_density/((float)s->height*x_density);

  1560. 		s->avctx->aspect_ratio = (float)x_density/y_density;

  1561. 		/* it's better, but every JFIF I have seen stores 1:1 */

  1562.             else

  1563.                 s->avctx->aspect_ratio= 0.0;

  1564. #endif

  1565. 	}

  1566. 	else

  1567. 	{

  1568. 	    skip_bits(&s->gb, 16);

  1569. 	    skip_bits(&s->gb, 16);

  1570. 	}

  1571. 

  1572. 	t_w = get_bits(&s->gb, 8);

  1573. 	t_h = get_bits(&s->gb, 8);

  1574. 	if (t_w && t_h)

  1575. 	{

  1576. 	    /* skip thumbnail */

  1577. 	    if (len-10-(t_w*t_h*3) > 0)

  1578. 		len -= t_w*t_h*3;

  1579. 	}

  1580. 	len -= 10;

  1581. 	goto out;

  1582.     }

  1583.     

  1584.     if (id == ff_get_fourcc("Adob") && (get_bits(&s->gb, 8) == 'e'))

  1585.     {

  1586. 	printf("mjpeg: Adobe header found\n");

  1587. 	skip_bits(&s->gb, 16); /* version */

  1588. 	skip_bits(&s->gb, 16); /* flags0 */

  1589. 	skip_bits(&s->gb, 16); /* flags1 */

  1590. 	skip_bits(&s->gb, 8); /* transform */

  1591. 	len -= 7;

  1592. 	goto out;

  1593.     }

  1594. 

  1595.     if (id == ff_get_fourcc("LJIF")){

  1596.         printf("Pegasus lossless jpeg header found\n");

  1597. 	skip_bits(&s->gb, 16); /* version ? */

  1598. 	skip_bits(&s->gb, 16); /* unknwon always 0? */

  1599. 	skip_bits(&s->gb, 16); /* unknwon always 0? */

  1600. 	skip_bits(&s->gb, 16); /* unknwon always 0? */

  1601.         switch( get_bits(&s->gb, 8)){

  1602.         case 1:

  1603.             s->rgb= 1;

  1604.             s->pegasus_rct=0;

  1605.             break;

  1606.         case 2:

  1607.             s->rgb= 1;

  1608.             s->pegasus_rct=1;

  1609.             break;

  1610.         default:

  1611.             printf("unknown colorspace\n");

  1612.         }

  1613.         len -= 9;

  1614.         goto out;

  1615.     }

  1616.     

  1617.     /* Apple MJPEG-A */

  1618.     if ((s->start_code == APP1) && (len > (0x28 - 8)))

  1619.     {

  1620. 	id = (get_bits(&s->gb, 16) << 16) | get_bits(&s->gb, 16);

  1621. 	id = be2me_32(id);

  1622. 	len -= 4;

  1623. 	if (id == ff_get_fourcc("mjpg")) /* Apple MJPEG-A */

  1624. 	{

  1625. #if 0

  1626. 	    skip_bits(&s->gb, 32); /* field size */

  1627. 	    skip_bits(&s->gb, 32); /* pad field size */

  1628. 	    skip_bits(&s->gb, 32); /* next off */

  1629. 	    skip_bits(&s->gb, 32); /* quant off */

  1630. 	    skip_bits(&s->gb, 32); /* huff off */

  1631. 	    skip_bits(&s->gb, 32); /* image off */

  1632. 	    skip_bits(&s->gb, 32); /* scan off */

  1633. 	    skip_bits(&s->gb, 32); /* data off */

  1634. #endif

  1635. 	    if (s->first_picture)

  1636. 		printf("mjpeg: Apple MJPEG-A header found\n");

  1637. 	}

  1638.     }

  1639. 

  1640. out:

  1641.     /* slow but needed for extreme adobe jpegs */

  1642.     if (len < 0)

  1643. 	printf("mjpeg: error, decode_app parser read over the end\n");

  1644.     while(--len > 0)

  1645. 	skip_bits(&s->gb, 8);

  1646. 

  1647.     return 0;

  1648. }

  1649. 

  1650. static int mjpeg_decode_com(MJpegDecodeContext *s)

  1651. {

  1652.     /* XXX: verify len field validity */

  1653.     int len = get_bits(&s->gb, 16);

  1654.     if (len >= 2 && len < 32768) {

  1655. 	/* XXX: any better upper bound */

  1656. 	uint8_t *cbuf = av_malloc(len - 1);

  1657. 	if (cbuf) {

  1658. 	    int i;

  1659. 	    for (i = 0; i < len - 2; i++)

  1660. 		cbuf[i] = get_bits(&s->gb, 8);

  1661. 	    if (i > 0 && cbuf[i-1] == '\n')

  1662. 		cbuf[i-1] = 0;

  1663. 	    else

  1664. 		cbuf[i] = 0;

  1665. 

  1666. 	    printf("mjpeg comment: '%s'\n", cbuf);

  1667. 

  1668. 	    /* buggy avid, it puts EOI only at every 10th frame */

  1669. 	    if (!strcmp(cbuf, "AVID"))

  1670. 	    {

  1671. 		s->buggy_avid = 1;

  1672. 		//	if (s->first_picture)

  1673. 		//	    printf("mjpeg: workarounding buggy AVID\n");

  1674. 	    }

  1675. 

  1676. 	    av_free(cbuf);

  1677. 	}

  1678.     }

  1679. 

  1680.     return 0;

  1681. }

  1682. 

  1683. #if 0

  1684. static int valid_marker_list[] =

  1685. {

  1686.         /* 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, a, b, c, d, e, f */

  1687. /* 0 */    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,

  1688. /* 1 */    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,

  1689. /* 2 */    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,

  1690. /* 3 */    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,

  1691. /* 4 */    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,

  1692. /* 5 */    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,

  1693. /* 6 */    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,

  1694. /* 7 */    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,

  1695. /* 8 */    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,

  1696. /* 9 */    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,

  1697. /* a */    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,

  1698. /* b */    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,

  1699. /* c */    1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,

  1700. /* d */    1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,

  1701. /* e */    1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,

  1702. /* f */    1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0,

  1703. }

  1704. #endif

  1705. 

  1706. /* return the 8 bit start code value and update the search

  1707.    state. Return -1 if no start code found */

  1708. static int find_marker(uint8_t **pbuf_ptr, uint8_t *buf_end)

  1709. {

  1710.     uint8_t *buf_ptr;

  1711.     unsigned int v, v2;

  1712.     int val;

  1713. #ifdef DEBUG

  1714.     int skipped=0;

  1715. #endif

  1716. 

  1717.     buf_ptr = *pbuf_ptr;

  1718.     while (buf_ptr < buf_end) {

  1719.         v = *buf_ptr++;

  1720. 	v2 = *buf_ptr;

  1721.         if ((v == 0xff) && (v2 >= 0xc0) && (v2 <= 0xfe)) {

  1722. 	    val = *buf_ptr++;

  1723. 	    goto found;

  1724.         }

  1725. #ifdef DEBUG

  1726. 	skipped++;

  1727. #endif

  1728.     }

  1729.     val = -1;

  1730. found:

  1731. #ifdef DEBUG

  1732.     dprintf("find_marker skipped %d bytes\n", skipped);

  1733. #endif

  1734.     *pbuf_ptr = buf_ptr;

  1735.     return val;

  1736. }

  1737. 

  1738. static int mjpeg_decode_frame(AVCodecContext *avctx, 

  1739.                               void *data, int *data_size,

  1740.                               uint8_t *buf, int buf_size)

  1741. {

  1742.     MJpegDecodeContext *s = avctx->priv_data;

  1743.     uint8_t *buf_end, *buf_ptr;

  1744.     int start_code;

  1745.     AVFrame *picture = data;

  1746. 

  1747.     *data_size = 0;

  1748. 

  1749.     /* no supplementary picture */

  1750.     if (buf_size == 0)

  1751.         return 0;

  1752. 

  1753.     buf_ptr = buf;

  1754.     buf_end = buf + buf_size;

  1755.     while (buf_ptr < buf_end) {

  1756.         /* find start next marker */

  1757.         start_code = find_marker(&buf_ptr, buf_end);

  1758. 	{

  1759. 	    /* EOF */

  1760.             if (start_code < 0) {

  1761. 		goto the_end;

  1762.             } else {

  1763.                 dprintf("marker=%x avail_size_in_buf=%d\n", start_code, buf_end - buf_ptr);

  1764. 		

  1765. 		if ((buf_end - buf_ptr) > s->buffer_size)

  1766. 		{

  1767. 		    av_free(s->buffer);

  1768. 		    s->buffer_size = buf_end-buf_ptr;

  1769. 		    s->buffer = av_malloc(s->buffer_size);

  1770. 		    dprintf("buffer too small, expanding to %d bytes\n",

  1771. 			s->buffer_size);

  1772. 		}

  1773. 		

  1774. 		/* unescape buffer of SOS */

  1775. 		if (start_code == SOS)

  1776. 		{

  1777. 		    uint8_t *src = buf_ptr;

  1778. 		    uint8_t *dst = s->buffer;

  1779. 

  1780. 		    while (src<buf_end)

  1781. 		    {

  1782. 			uint8_t x = *(src++);

  1783. 

  1784. 			*(dst++) = x;

  1785. 			if (x == 0xff)

  1786. 			{

  1787. 			    while(*src == 0xff) src++;

  1788. 

  1789. 			    x = *(src++);

  1790. 			    if (x >= 0xd0 && x <= 0xd7)

  1791. 				*(dst++) = x;

  1792. 			    else if (x)

  1793. 				break;

  1794. 			}

  1795. 		    }

  1796. 		    init_get_bits(&s->gb, s->buffer, (dst - s->buffer)*8);

  1797. 		    

  1798. 		    dprintf("escaping removed %d bytes\n",

  1799. 			(buf_end - buf_ptr) - (dst - s->buffer));

  1800. 		}

  1801. 		else

  1802. 		    init_get_bits(&s->gb, buf_ptr, (buf_end - buf_ptr)*8);

  1803. 		

  1804. 		s->start_code = start_code;

  1805.                 if(s->avctx->debug & FF_DEBUG_STARTCODE){

  1806.                     printf("startcode: %X\n", start_code);

  1807.                 }

  1808. 

  1809. 		/* process markers */

  1810. 		if (start_code >= 0xd0 && start_code <= 0xd7) {

  1811. 		    dprintf("restart marker: %d\n", start_code&0x0f);

  1812. 		} else if (s->first_picture) {

  1813. 		    /* APP fields */

  1814. 		    if (start_code >= 0xe0 && start_code <= 0xef)

  1815. 			mjpeg_decode_app(s);

  1816. 		    /* Comment */

  1817. 		    else if (start_code == COM)

  1818. 			mjpeg_decode_com(s);

  1819. 		}

  1820. 

  1821.                 switch(start_code) {

  1822.                 case SOI:

  1823. 		    s->restart_interval = 0;

  1824.                     /* nothing to do on SOI */

  1825.                     break;

  1826.                 case DQT:

  1827.                     mjpeg_decode_dqt(s);

  1828.                     break;

  1829.                 case DHT:

  1830.                     if(mjpeg_decode_dht(s) < 0){

  1831.                         fprintf(stderr, "huffman table decode error\n");

  1832.                         return -1;

  1833.                     }

  1834.                     break;

  1835.                 case SOF0:

  1836.                     s->lossless=0;

  1837.                     if (mjpeg_decode_sof(s) < 0) 

  1838. 			return -1;

  1839.                     break;

  1840.                 case SOF3:

  1841.                     s->lossless=1;

  1842.                     if (mjpeg_decode_sof(s) < 0) 

  1843. 			return -1;

  1844.                     break;

  1845. 		case EOI:

  1846. 		    if ((s->buggy_avid && !s->interlaced) || s->restart_interval) 

  1847.                         break;

  1848. eoi_parser:

  1849. 		    {

  1850.                         if (s->interlaced) {

  1851.                             s->bottom_field ^= 1;

  1852.                             /* if not bottom field, do not output image yet */

  1853.                             if (s->bottom_field)

  1854.                                 goto not_the_end;

  1855.                         }

  1856.                         *picture = s->picture;

  1857.                         *data_size = sizeof(AVFrame);

  1858.                         avctx->height = s->height;

  1859.                         if (s->interlaced)

  1860.                             avctx->height *= 2;

  1861.                         avctx->width = s->width;

  1862. 

  1863.                         if(!s->lossless){

  1864.                             picture->quality= FFMAX(FFMAX(s->qscale[0], s->qscale[1]), s->qscale[2]); 

  1865.                             picture->qstride= 0;

  1866.                             picture->qscale_table= s->qscale_table;

  1867.                             memset(picture->qscale_table, picture->quality, (s->width+15)/16);

  1868.                             if(avctx->debug & FF_DEBUG_QP)

  1869.                                 printf("QP: %d\n", picture->quality);

  1870.                             picture->quality*= FF_QP2LAMBDA;

  1871.                         }

  1872.                         

  1873.                         goto the_end;

  1874.                     }

  1875. 		    break;

  1876.                 case SOS:

  1877.                     mjpeg_decode_sos(s);

  1878. 		    /* buggy avid puts EOI every 10-20th frame */

  1879. 		    /* if restart period is over process EOI */

  1880. 		    if ((s->buggy_avid && !s->interlaced) || s->restart_interval)

  1881. 			goto eoi_parser;

  1882.                     break;

  1883. 		case DRI:

  1884. 		    mjpeg_decode_dri(s);

  1885. 		    break;

  1886. 		case SOF1:

  1887. 		case SOF2:

  1888. 		case SOF5:

  1889. 		case SOF6:

  1890. 		case SOF7:

  1891. 		case SOF9:

  1892. 		case SOF10:

  1893. 		case SOF11:

  1894. 		case SOF13:

  1895. 		case SOF14:

  1896. 		case SOF15:

  1897. 		case JPG:

  1898. 		    printf("mjpeg: unsupported coding type (%x)\n", start_code);

  1899. 		    break;

  1900. //		default:

  1901. //		    printf("mjpeg: unsupported marker (%x)\n", start_code);

  1902. //		    break;

  1903.                 }

  1904. 

  1905. not_the_end:

  1906. 		/* eof process start code */

  1907. 		buf_ptr += (get_bits_count(&s->gb)+7)/8;

  1908. 		dprintf("marker parser used %d bytes (%d bits)\n",

  1909. 		    (get_bits_count(&s->gb)+7)/8, get_bits_count(&s->gb));

  1910.             }

  1911.         }

  1912.     }

  1913. the_end:

  1914.     dprintf("mjpeg decode frame unused %d bytes\n", buf_end - buf_ptr);

  1915. //    return buf_end - buf_ptr;

  1916.     return buf_ptr - buf;

  1917. }

  1918. 

  1919. static int mjpegb_decode_frame(AVCodecContext *avctx, 

  1920.                               void *data, int *data_size,

  1921.                               uint8_t *buf, int buf_size)

  1922. {

  1923.     MJpegDecodeContext *s = avctx->priv_data;

  1924.     uint8_t *buf_end, *buf_ptr;

  1925.     AVFrame *picture = data;

  1926.     GetBitContext hgb; /* for the header */

  1927.     uint32_t dqt_offs, dht_offs, sof_offs, sos_offs, second_field_offs;

  1928.     uint32_t field_size;

  1929. 

  1930.     *data_size = 0;

  1931. 

  1932.     /* no supplementary picture */

  1933.     if (buf_size == 0)

  1934.         return 0;

  1935. 

  1936.     buf_ptr = buf;

  1937.     buf_end = buf + buf_size;

  1938.     

  1939. read_header:

  1940.     /* reset on every SOI */

  1941.     s->restart_interval = 0;

  1942. 

  1943.     init_get_bits(&hgb, buf_ptr, /*buf_size*/(buf_end - buf_ptr)*8);

  1944. 

  1945.     skip_bits(&hgb, 32); /* reserved zeros */

  1946.     

  1947.     if (get_bits(&hgb, 32) != be2me_32(ff_get_fourcc("mjpg")))

  1948.     {

  1949. 	dprintf("not mjpeg-b (bad fourcc)\n");

  1950. 	return 0;

  1951.     }

  1952. 

  1953.     field_size = get_bits(&hgb, 32); /* field size */

  1954.     dprintf("field size: 0x%x\n", field_size);

  1955.     skip_bits(&hgb, 32); /* padded field size */

  1956.     second_field_offs = get_bits(&hgb, 32);

  1957.     dprintf("second field offs: 0x%x\n", second_field_offs);

  1958.     if (second_field_offs)

  1959. 	s->interlaced = 1;

  1960. 

  1961.     dqt_offs = get_bits(&hgb, 32);

  1962.     dprintf("dqt offs: 0x%x\n", dqt_offs);

  1963.     if (dqt_offs)

  1964.     {

  1965. 	init_get_bits(&s->gb, buf+dqt_offs, (buf_end - (buf+dqt_offs))*8);

  1966. 	s->start_code = DQT;

  1967. 	mjpeg_decode_dqt(s);

  1968.     }

  1969.     

  1970.     dht_offs = get_bits(&hgb, 32);

  1971.     dprintf("dht offs: 0x%x\n", dht_offs);

  1972.     if (dht_offs)

  1973.     {

  1974. 	init_get_bits(&s->gb, buf+dht_offs, (buf_end - (buf+dht_offs))*8);

  1975. 	s->start_code = DHT;

  1976. 	mjpeg_decode_dht(s);

  1977.     }

  1978. 

  1979.     sof_offs = get_bits(&hgb, 32);

  1980.     dprintf("sof offs: 0x%x\n", sof_offs);

  1981.     if (sof_offs)

  1982.     {

  1983. 	init_get_bits(&s->gb, buf+sof_offs, (buf_end - (buf+sof_offs))*8);

  1984. 	s->start_code = SOF0;

  1985. 	if (mjpeg_decode_sof(s) < 0)

  1986. 	    return -1;

  1987.     }

  1988. 

  1989.     sos_offs = get_bits(&hgb, 32);

  1990.     dprintf("sos offs: 0x%x\n", sos_offs);

  1991.     if (sos_offs)

  1992.     {

  1993. //	init_get_bits(&s->gb, buf+sos_offs, (buf_end - (buf+sos_offs))*8);

  1994. 	init_get_bits(&s->gb, buf+sos_offs, field_size*8);

  1995. 	s->start_code = SOS;

  1996. 	mjpeg_decode_sos(s);

  1997.     }

  1998. 

  1999.     skip_bits(&hgb, 32); /* start of data offset */

  2000. 

  2001.     if (s->interlaced) {

  2002.         s->bottom_field ^= 1;

  2003.         /* if not bottom field, do not output image yet */

  2004.         if (s->bottom_field && second_field_offs)

  2005. 	{

  2006. 	    buf_ptr = buf + second_field_offs;

  2007. 	    second_field_offs = 0;

  2008. 	    goto read_header;

  2009.     	}

  2010.     }

  2011. 

  2012.     //XXX FIXME factorize, this looks very similar to the EOI code

  2013. 

  2014.     *picture= s->picture;

  2015.     *data_size = sizeof(AVFrame);

  2016.     avctx->height = s->height;

  2017.     if (s->interlaced)

  2018.         avctx->height *= 2;

  2019.     avctx->width = s->width;

  2020.     

  2021.     if(!s->lossless){

  2022.         picture->quality= FFMAX(FFMAX(s->qscale[0], s->qscale[1]), s->qscale[2]); 

  2023.         picture->qstride= 0;

  2024.         picture->qscale_table= s->qscale_table;

  2025.         memset(picture->qscale_table, picture->quality, (s->width+15)/16);

  2026.         if(avctx->debug & FF_DEBUG_QP)

  2027.             printf("QP: %d\n", picture->quality);

  2028.         picture->quality*= FF_QP2LAMBDA;

  2029.     }

  2030. 

  2031.     return buf_ptr - buf;

  2032. }

  2033. 

  2034. #include "sp5x.h"

  2035. 

  2036. static int sp5x_decode_frame(AVCodecContext *avctx, 

  2037.                               void *data, int *data_size,

  2038.                               uint8_t *buf, int buf_size)

  2039. {

  2040. #if 0

  2041.     MJpegDecodeContext *s = avctx->priv_data;

  2042. #endif

  2043.     const int qscale = 5;

  2044.     uint8_t *buf_ptr, *buf_end, *recoded;

  2045.     int i = 0, j = 0;

  2046. 

  2047.     *data_size = 0;

  2048. 

  2049.     /* no supplementary picture */

  2050.     if (buf_size == 0)

  2051.         return 0;

  2052. 

  2053.     if (!avctx->width || !avctx->height)

  2054. 	return -1;

  2055. 

  2056.     buf_ptr = buf;

  2057.     buf_end = buf + buf_size;

  2058. 

  2059. #if 1

  2060.     recoded = av_mallocz(buf_size + 1024);

  2061.     if (!recoded)

  2062. 	return -1;

  2063. 

  2064.     /* SOI */

  2065.     recoded[j++] = 0xFF;

  2066.     recoded[j++] = 0xD8;

  2067. 

  2068.     memcpy(recoded+j, &sp5x_data_dqt[0], sizeof(sp5x_data_dqt));

  2069.     memcpy(recoded+j+5, &sp5x_quant_table[qscale * 2], 64);

  2070.     memcpy(recoded+j+70, &sp5x_quant_table[(qscale * 2) + 1], 64);

  2071.     j += sizeof(sp5x_data_dqt);

  2072. 

  2073.     memcpy(recoded+j, &sp5x_data_dht[0], sizeof(sp5x_data_dht));

  2074.     j += sizeof(sp5x_data_dht);

  2075. 

  2076.     memcpy(recoded+j, &sp5x_data_sof[0], sizeof(sp5x_data_sof));

  2077.     recoded[j+5] = (avctx->height >> 8) & 0xFF;

  2078.     recoded[j+6] = avctx->height & 0xFF;

  2079.     recoded[j+7] = (avctx->width >> 8) & 0xFF;

  2080.     recoded[j+8] = avctx->width & 0xFF;

  2081.     j += sizeof(sp5x_data_sof);

  2082. 

  2083.     memcpy(recoded+j, &sp5x_data_sos[0], sizeof(sp5x_data_sos));

  2084.     j += sizeof(sp5x_data_sos);

  2085. 

  2086.     for (i = 14; i < buf_size, j < buf_size+1024-2; i++)

  2087.     {

  2088. 	recoded[j++] = buf[i];

  2089. 	if (buf[i] == 0xff)

  2090. 	    recoded[j++] = 0;

  2091.     }

  2092. 

  2093.     /* EOI */

  2094.     recoded[j++] = 0xFF;

  2095.     recoded[j++] = 0xD9;

  2096. 

  2097.     i = mjpeg_decode_frame(avctx, data, data_size, recoded, j);

  2098. 

  2099.     av_free(recoded);

  2100. 

  2101. #else

  2102.     /* SOF */

  2103.     s->bits = 8;

  2104.     s->width = avctx->width;

  2105.     s->height = avctx->height;

  2106.     s->nb_components = 3;

  2107.     s->component_id[0] = 0;

  2108.     s->h_count[0] = 2;

  2109.     s->v_count[0] = 2;

  2110.     s->quant_index[0] = 0;

  2111.     s->component_id[1] = 1;

  2112.     s->h_count[1] = 1;

  2113.     s->v_count[1] = 1;

  2114.     s->quant_index[1] = 1;

  2115.     s->component_id[2] = 2;

  2116.     s->h_count[2] = 1;

  2117.     s->v_count[2] = 1;

  2118.     s->quant_index[2] = 1;

  2119.     s->h_max = 2;

  2120.     s->v_max = 2;

  2121.     

  2122.     s->qscale_table = av_mallocz((s->width+15)/16);

  2123.     avctx->pix_fmt = PIX_FMT_YUV420P;

  2124.     s->interlaced = 0;

  2125.     

  2126.     s->picture.reference = 0;

  2127.     if (avctx->get_buffer(avctx, &s->picture) < 0)

  2128.     {

  2129. 	fprintf(stderr, "get_buffer() failed\n");

  2130. 	return -1;

  2131.     }

  2132. 

  2133.     s->picture.pict_type = I_TYPE;

  2134.     s->picture.key_frame = 1;

  2135. 

  2136.     for (i = 0; i < 3; i++)

  2137. 	s->linesize[i] = s->picture.linesize[i] << s->interlaced;

  2138. 

  2139.     /* DQT */

  2140.     for (i = 0; i < 64; i++)

  2141.     {

  2142. 	j = s->scantable.permutated[i];

  2143. 	s->quant_matrixes[0][j] = sp5x_quant_table[(qscale * 2) + i];

  2144.     }

  2145.     s->qscale[0] = FFMAX(

  2146. 	s->quant_matrixes[0][s->scantable.permutated[1]],

  2147. 	s->quant_matrixes[0][s->scantable.permutated[8]]) >> 1;

  2148. 

  2149.     for (i = 0; i < 64; i++)

  2150.     {

  2151. 	j = s->scantable.permutated[i];

  2152. 	s->quant_matrixes[1][j] = sp5x_quant_table[(qscale * 2) + 1 + i];

  2153.     }

  2154.     s->qscale[1] = FFMAX(

  2155. 	s->quant_matrixes[1][s->scantable.permutated[1]],

  2156. 	s->quant_matrixes[1][s->scantable.permutated[8]]) >> 1;

  2157. 

  2158.     /* DHT */

  2159. 

  2160.     /* SOS */

  2161.     s->comp_index[0] = 0;

  2162.     s->nb_blocks[0] = s->h_count[0] * s->v_count[0];

  2163.     s->h_scount[0] = s->h_count[0];

  2164.     s->v_scount[0] = s->v_count[0];

  2165.     s->dc_index[0] = 0;

  2166.     s->ac_index[0] = 0;

  2167. 

  2168.     s->comp_index[1] = 1;

  2169.     s->nb_blocks[1] = s->h_count[1] * s->v_count[1];

  2170.     s->h_scount[1] = s->h_count[1];

  2171.     s->v_scount[1] = s->v_count[1];

  2172.     s->dc_index[1] = 1;

  2173.     s->ac_index[1] = 1;

  2174. 

  2175.     s->comp_index[2] = 2;

  2176.     s->nb_blocks[2] = s->h_count[2] * s->v_count[2];

  2177.     s->h_scount[2] = s->h_count[2];

  2178.     s->v_scount[2] = s->v_count[2];

  2179.     s->dc_index[2] = 1;

  2180.     s->ac_index[2] = 1;

  2181.     

  2182.     for (i = 0; i < 3; i++)

  2183. 	s->last_dc[i] = 1024;

  2184. 

  2185.     s->mb_width = (s->width * s->h_max * 8 -1) / (s->h_max * 8);

  2186.     s->mb_height = (s->height * s->v_max * 8 -1) / (s->v_max * 8);

  2187. 

  2188.     init_get_bits(&s->gb, buf, buf_size*8);

  2189.     

  2190.     return mjpeg_decode_scan(s);

  2191. #endif

  2192. 

  2193.     return i;

  2194. }

  2195. 

  2196. static int mjpeg_decode_end(AVCodecContext *avctx)

  2197. {

  2198.     MJpegDecodeContext *s = avctx->priv_data;

  2199.     int i, j;

  2200. 

  2201.     av_free(s->buffer);

  2202.     av_free(s->qscale_table);

  2203.     avcodec_default_free_buffers(avctx);

  2204.     

  2205.     for(i=0;i<2;i++) {

  2206.         for(j=0;j<4;j++)

  2207.             free_vlc(&s->vlcs[i][j]);

  2208.     }

  2209.     return 0;

  2210. }

  2211. 

  2212. AVCodec mjpeg_decoder = {

  2213.     "mjpeg",

  2214.     CODEC_TYPE_VIDEO,

  2215.     CODEC_ID_MJPEG,

  2216.     sizeof(MJpegDecodeContext),

  2217.     mjpeg_decode_init,

  2218.     NULL,

  2219.     mjpeg_decode_end,

  2220.     mjpeg_decode_frame,

  2221.     CODEC_CAP_DR1,

  2222.     NULL

  2223. };

  2224. 

  2225. AVCodec mjpegb_decoder = {

  2226.     "mjpegb",

  2227.     CODEC_TYPE_VIDEO,

  2228.     CODEC_ID_MJPEGB,

  2229.     sizeof(MJpegDecodeContext),

  2230.     mjpeg_decode_init,

  2231.     NULL,

  2232.     mjpeg_decode_end,

  2233.     mjpegb_decode_frame,

  2234.     CODEC_CAP_DR1,

  2235.     NULL

  2236. };

  2237. 

  2238. AVCodec sp5x_decoder = {

  2239.     "sp5x",

  2240.     CODEC_TYPE_VIDEO,

  2241.     CODEC_ID_SP5X,

  2242.     sizeof(MJpegDecodeContext),

  2243.     mjpeg_decode_init,

  2244.     NULL,

  2245.     mjpeg_decode_end,

  2246.     sp5x_decode_frame,

  2247.     CODEC_CAP_DR1,

  2248.     NULL

  2249. };

  2250. 

  2251. #ifdef CONFIG_ENCODERS

  2252. AVCodec ljpeg_encoder = { //FIXME avoid MPV_* lossless jpeg shouldnt need them

  2253.     "ljpeg",

  2254.     CODEC_TYPE_VIDEO,

  2255.     CODEC_ID_LJPEG,

  2256.     sizeof(MpegEncContext),

  2257.     MPV_encode_init,

  2258.     encode_picture_lossless,

  2259.     MPV_encode_end,

  2260. };

  2261. #endif