00001 
00002 
00003 
00004 
00005 
00006 
00007 
00008 
00009 
00010 
00011 #define JPEG_INTERNALS
00012 #include "jinclude.h"
00013 #include "jpeglib.h"
00014 
00015 
00016 typedef enum {                  
00017   M_SOF0  = 0xc0,
00018   M_SOF1  = 0xc1,
00019   M_SOF2  = 0xc2,
00020   M_SOF3  = 0xc3,
00021   
00022   M_SOF5  = 0xc5,
00023   M_SOF6  = 0xc6,
00024   M_SOF7  = 0xc7,
00025   
00026   M_JPG   = 0xc8,
00027   M_SOF9  = 0xc9,
00028   M_SOF10 = 0xca,
00029   M_SOF11 = 0xcb,
00030   
00031   M_SOF13 = 0xcd,
00032   M_SOF14 = 0xce,
00033   M_SOF15 = 0xcf,
00034   
00035   M_DHT   = 0xc4,
00036   
00037   M_DAC   = 0xcc,
00038   
00039   M_RST0  = 0xd0,
00040   M_RST1  = 0xd1,
00041   M_RST2  = 0xd2,
00042   M_RST3  = 0xd3,
00043   M_RST4  = 0xd4,
00044   M_RST5  = 0xd5,
00045   M_RST6  = 0xd6,
00046   M_RST7  = 0xd7,
00047   
00048   M_SOI   = 0xd8,
00049   M_EOI   = 0xd9,
00050   M_SOS   = 0xda,
00051   M_DQT   = 0xdb,
00052   M_DNL   = 0xdc,
00053   M_DRI   = 0xdd,
00054   M_DHP   = 0xde,
00055   M_EXP   = 0xdf,
00056   
00057   M_APP0  = 0xe0,
00058   M_APP1  = 0xe1,
00059   M_APP2  = 0xe2,
00060   M_APP3  = 0xe3,
00061   M_APP4  = 0xe4,
00062   M_APP5  = 0xe5,
00063   M_APP6  = 0xe6,
00064   M_APP7  = 0xe7,
00065   M_APP8  = 0xe8,
00066   M_APP9  = 0xe9,
00067   M_APP10 = 0xea,
00068   M_APP11 = 0xeb,
00069   M_APP12 = 0xec,
00070   M_APP13 = 0xed,
00071   M_APP14 = 0xee,
00072   M_APP15 = 0xef,
00073   
00074   M_JPG0  = 0xf0,
00075   M_JPG13 = 0xfd,
00076   M_COM   = 0xfe,
00077   
00078   M_TEM   = 0x01,
00079   
00080   M_ERROR = 0x100
00081 } JPEG_MARKER;
00082 
00083 
00084 
00085 
00086 typedef struct {
00087   struct jpeg_marker_writer pub; 
00088 
00089   unsigned int last_restart_interval; 
00090 } my_marker_writer;
00091 
00092 typedef my_marker_writer * my_marker_ptr;
00093 
00094 
00095 
00096 
00097 
00098 
00099 
00100 
00101 
00102 
00103 
00104 
00105 
00106 
00107 LOCAL(void)
00108 emit_byte (j_compress_ptr cinfo, int val)
00109 
00110 {
00111   struct jpeg_destination_mgr * dest = cinfo->dest;
00112 
00113   *(dest->next_output_byte)++ = (JOCTET) val;
00114   if (--dest->free_in_buffer == 0) {
00115     if (! (*dest->empty_output_buffer) (cinfo))
00116       ERREXIT(cinfo, JERR_CANT_SUSPEND);
00117   }
00118 }
00119 
00120 
00121 LOCAL(void)
00122 emit_marker (j_compress_ptr cinfo, JPEG_MARKER mark)
00123 
00124 {
00125   emit_byte(cinfo, 0xFF);
00126   emit_byte(cinfo, (int) mark);
00127 }
00128 
00129 
00130 LOCAL(void)
00131 emit_2bytes (j_compress_ptr cinfo, int value)
00132 
00133 {
00134   emit_byte(cinfo, (value >> 8) & 0xFF);
00135   emit_byte(cinfo, value & 0xFF);
00136 }
00137 
00138 
00139 
00140 
00141 
00142 
00143 LOCAL(int)
00144 emit_dqt (j_compress_ptr cinfo, int index)
00145 
00146 
00147 {
00148   JQUANT_TBL * qtbl = cinfo->quant_tbl_ptrs[index];
00149   int prec;
00150   int i;
00151 
00152   if (qtbl == NULL)
00153     ERREXIT1(cinfo, JERR_NO_QUANT_TABLE, index);
00154 
00155   prec = 0;
00156   for (i = 0; i < DCTSIZE2; i++) {
00157     if (qtbl->quantval[i] > 255)
00158       prec = 1;
00159   }
00160 
00161   if (! qtbl->sent_table) {
00162     emit_marker(cinfo, M_DQT);
00163 
00164     emit_2bytes(cinfo, prec ? DCTSIZE2*2 + 1 + 2 : DCTSIZE2 + 1 + 2);
00165 
00166     emit_byte(cinfo, index + (prec<<4));
00167 
00168     for (i = 0; i < DCTSIZE2; i++) {
00169       
00170       unsigned int qval = qtbl->quantval[jpeg_natural_order[i]];
00171       if (prec)
00172         emit_byte(cinfo, (int) (qval >> 8));
00173       emit_byte(cinfo, (int) (qval & 0xFF));
00174     }
00175 
00176     qtbl->sent_table = TRUE;
00177   }
00178 
00179   return prec;
00180 }
00181 
00182 
00183 LOCAL(void)
00184 emit_dht (j_compress_ptr cinfo, int index, boolean is_ac)
00185 
00186 {
00187   JHUFF_TBL * htbl;
00188   int length, i;
00189   
00190   if (is_ac) {
00191     htbl = cinfo->ac_huff_tbl_ptrs[index];
00192     index += 0x10;              
00193   } else {
00194     htbl = cinfo->dc_huff_tbl_ptrs[index];
00195   }
00196 
00197   if (htbl == NULL)
00198     ERREXIT1(cinfo, JERR_NO_HUFF_TABLE, index);
00199   
00200   if (! htbl->sent_table) {
00201     emit_marker(cinfo, M_DHT);
00202     
00203     length = 0;
00204     for (i = 1; i <= 16; i++)
00205       length += htbl->bits[i];
00206     
00207     emit_2bytes(cinfo, length + 2 + 1 + 16);
00208     emit_byte(cinfo, index);
00209     
00210     for (i = 1; i <= 16; i++)
00211       emit_byte(cinfo, htbl->bits[i]);
00212     
00213     for (i = 0; i < length; i++)
00214       emit_byte(cinfo, htbl->huffval[i]);
00215     
00216     htbl->sent_table = TRUE;
00217   }
00218 }
00219 
00220 
00221 LOCAL(void)
00222 emit_dac (j_compress_ptr cinfo)
00223 
00224 
00225 
00226 {
00227 #ifdef C_ARITH_CODING_SUPPORTED
00228   char dc_in_use[NUM_ARITH_TBLS];
00229   char ac_in_use[NUM_ARITH_TBLS];
00230   int length, i;
00231   jpeg_component_info *compptr;
00232   
00233   for (i = 0; i < NUM_ARITH_TBLS; i++)
00234     dc_in_use[i] = ac_in_use[i] = 0;
00235   
00236   for (i = 0; i < cinfo->comps_in_scan; i++) {
00237     compptr = cinfo->cur_comp_info[i];
00238     dc_in_use[compptr->dc_tbl_no] = 1;
00239     ac_in_use[compptr->ac_tbl_no] = 1;
00240   }
00241   
00242   length = 0;
00243   for (i = 0; i < NUM_ARITH_TBLS; i++)
00244     length += dc_in_use[i] + ac_in_use[i];
00245   
00246   emit_marker(cinfo, M_DAC);
00247   
00248   emit_2bytes(cinfo, length*2 + 2);
00249   
00250   for (i = 0; i < NUM_ARITH_TBLS; i++) {
00251     if (dc_in_use[i]) {
00252       emit_byte(cinfo, i);
00253       emit_byte(cinfo, cinfo->arith_dc_L[i] + (cinfo->arith_dc_U[i]<<4));
00254     }
00255     if (ac_in_use[i]) {
00256       emit_byte(cinfo, i + 0x10);
00257       emit_byte(cinfo, cinfo->arith_ac_K[i]);
00258     }
00259   }
00260 #endif 
00261 }
00262 
00263 
00264 LOCAL(void)
00265 emit_dri (j_compress_ptr cinfo)
00266 
00267 {
00268   emit_marker(cinfo, M_DRI);
00269   
00270   emit_2bytes(cinfo, 4);        
00271 
00272   emit_2bytes(cinfo, (int) cinfo->restart_interval);
00273 }
00274 
00275 
00276 LOCAL(void)
00277 emit_sof (j_compress_ptr cinfo, JPEG_MARKER code)
00278 
00279 {
00280   int ci;
00281   jpeg_component_info *compptr;
00282   
00283   emit_marker(cinfo, code);
00284   
00285   emit_2bytes(cinfo, 3 * cinfo->num_components + 2 + 5 + 1); 
00286 
00287   
00288   if ((long) cinfo->image_height > 65535L ||
00289       (long) cinfo->image_width > 65535L)
00290     ERREXIT1(cinfo, JERR_IMAGE_TOO_BIG, (unsigned int) 65535);
00291 
00292   emit_byte(cinfo, cinfo->data_precision);
00293   emit_2bytes(cinfo, (int) cinfo->image_height);
00294   emit_2bytes(cinfo, (int) cinfo->image_width);
00295 
00296   emit_byte(cinfo, cinfo->num_components);
00297 
00298   for (ci = 0, compptr = cinfo->comp_info; ci < cinfo->num_components;
00299        ci++, compptr++) {
00300     emit_byte(cinfo, compptr->component_id);
00301     emit_byte(cinfo, (compptr->h_samp_factor << 4) + compptr->v_samp_factor);
00302     emit_byte(cinfo, compptr->quant_tbl_no);
00303   }
00304 }
00305 
00306 
00307 LOCAL(void)
00308 emit_sos (j_compress_ptr cinfo)
00309 
00310 {
00311   int i, td, ta;
00312   jpeg_component_info *compptr;
00313   
00314   emit_marker(cinfo, M_SOS);
00315   
00316   emit_2bytes(cinfo, 2 * cinfo->comps_in_scan + 2 + 1 + 3); 
00317   
00318   emit_byte(cinfo, cinfo->comps_in_scan);
00319   
00320   for (i = 0; i < cinfo->comps_in_scan; i++) {
00321     compptr = cinfo->cur_comp_info[i];
00322     emit_byte(cinfo, compptr->component_id);
00323     td = compptr->dc_tbl_no;
00324     ta = compptr->ac_tbl_no;
00325     if (cinfo->progressive_mode) {
00326       
00327 
00328 
00329 
00330 
00331       if (cinfo->Ss == 0) {
00332         ta = 0;                 
00333         if (cinfo->Ah != 0 && !cinfo->arith_code)
00334           td = 0;               
00335       } else {
00336         td = 0;                 
00337       }
00338     }
00339     emit_byte(cinfo, (td << 4) + ta);
00340   }
00341 
00342   emit_byte(cinfo, cinfo->Ss);
00343   emit_byte(cinfo, cinfo->Se);
00344   emit_byte(cinfo, (cinfo->Ah << 4) + cinfo->Al);
00345 }
00346 
00347 
00348 LOCAL(void)
00349 emit_jfif_app0 (j_compress_ptr cinfo)
00350 
00351 {
00352   
00353 
00354 
00355 
00356 
00357 
00358 
00359 
00360 
00361 
00362 
00363   
00364   emit_marker(cinfo, M_APP0);
00365   
00366   emit_2bytes(cinfo, 2 + 4 + 1 + 2 + 1 + 2 + 2 + 1 + 1); 
00367 
00368   emit_byte(cinfo, 0x4A);       
00369   emit_byte(cinfo, 0x46);
00370   emit_byte(cinfo, 0x49);
00371   emit_byte(cinfo, 0x46);
00372   emit_byte(cinfo, 0);
00373   emit_byte(cinfo, cinfo->JFIF_major_version); 
00374   emit_byte(cinfo, cinfo->JFIF_minor_version);
00375   emit_byte(cinfo, cinfo->density_unit); 
00376   emit_2bytes(cinfo, (int) cinfo->X_density);
00377   emit_2bytes(cinfo, (int) cinfo->Y_density);
00378   emit_byte(cinfo, 0);          
00379   emit_byte(cinfo, 0);
00380 }
00381 
00382 
00383 LOCAL(void)
00384 emit_adobe_app14 (j_compress_ptr cinfo)
00385 
00386 {
00387   
00388 
00389 
00390 
00391 
00392 
00393 
00394 
00395 
00396 
00397 
00398 
00399 
00400 
00401 
00402   
00403   emit_marker(cinfo, M_APP14);
00404   
00405   emit_2bytes(cinfo, 2 + 5 + 2 + 2 + 2 + 1); 
00406 
00407   emit_byte(cinfo, 0x41);       
00408   emit_byte(cinfo, 0x64);
00409   emit_byte(cinfo, 0x6F);
00410   emit_byte(cinfo, 0x62);
00411   emit_byte(cinfo, 0x65);
00412   emit_2bytes(cinfo, 100);      
00413   emit_2bytes(cinfo, 0);        
00414   emit_2bytes(cinfo, 0);        
00415   switch (cinfo->jpeg_color_space) {
00416   case JCS_YCbCr:
00417     emit_byte(cinfo, 1);        
00418     break;
00419   case JCS_YCCK:
00420     emit_byte(cinfo, 2);        
00421     break;
00422   default:
00423     emit_byte(cinfo, 0);        
00424     break;
00425   }
00426 }
00427 
00428 
00429 
00430 
00431 
00432 
00433 
00434 
00435 
00436 
00437 METHODDEF(void)
00438 write_marker_header (j_compress_ptr cinfo, int marker, unsigned int datalen)
00439 
00440 {
00441   if (datalen > (unsigned int) 65533)           
00442     ERREXIT(cinfo, JERR_BAD_LENGTH);
00443 
00444   emit_marker(cinfo, (JPEG_MARKER) marker);
00445 
00446   emit_2bytes(cinfo, (int) (datalen + 2));      
00447 }
00448 
00449 METHODDEF(void)
00450 write_marker_byte (j_compress_ptr cinfo, int val)
00451 
00452 {
00453   emit_byte(cinfo, val);
00454 }
00455 
00456 
00457 
00458 
00459 
00460 
00461 
00462 
00463 
00464 
00465 
00466 
00467 
00468 METHODDEF(void)
00469 write_file_header (j_compress_ptr cinfo)
00470 {
00471   my_marker_ptr marker = (my_marker_ptr) cinfo->marker;
00472 
00473   emit_marker(cinfo, M_SOI);    
00474 
00475   
00476   marker->last_restart_interval = 0;
00477 
00478   if (cinfo->write_JFIF_header) 
00479     emit_jfif_app0(cinfo);
00480   if (cinfo->write_Adobe_marker) 
00481     emit_adobe_app14(cinfo);
00482 }
00483 
00484 
00485 
00486 
00487 
00488 
00489 
00490 
00491 
00492 
00493 METHODDEF(void)
00494 write_frame_header (j_compress_ptr cinfo)
00495 {
00496   int ci, prec;
00497   boolean is_baseline;
00498   jpeg_component_info *compptr;
00499   
00500   
00501 
00502 
00503   prec = 0;
00504   for (ci = 0, compptr = cinfo->comp_info; ci < cinfo->num_components;
00505        ci++, compptr++) {
00506     prec += emit_dqt(cinfo, compptr->quant_tbl_no);
00507   }
00508   
00509 
00510   
00511 
00512 
00513   if (cinfo->arith_code || cinfo->progressive_mode ||
00514       cinfo->data_precision != 8) {
00515     is_baseline = FALSE;
00516   } else {
00517     is_baseline = TRUE;
00518     for (ci = 0, compptr = cinfo->comp_info; ci < cinfo->num_components;
00519          ci++, compptr++) {
00520       if (compptr->dc_tbl_no > 1 || compptr->ac_tbl_no > 1)
00521         is_baseline = FALSE;
00522     }
00523     if (prec && is_baseline) {
00524       is_baseline = FALSE;
00525       
00526       TRACEMS(cinfo, 0, JTRC_16BIT_TABLES);
00527     }
00528   }
00529 
00530   
00531   if (cinfo->arith_code) {
00532     emit_sof(cinfo, M_SOF9);    
00533   } else {
00534     if (cinfo->progressive_mode)
00535       emit_sof(cinfo, M_SOF2);  
00536     else if (is_baseline)
00537       emit_sof(cinfo, M_SOF0);  
00538     else
00539       emit_sof(cinfo, M_SOF1);  
00540   }
00541 }
00542 
00543 
00544 
00545 
00546 
00547 
00548 
00549 
00550 METHODDEF(void)
00551 write_scan_header (j_compress_ptr cinfo)
00552 {
00553   my_marker_ptr marker = (my_marker_ptr) cinfo->marker;
00554   int i;
00555   jpeg_component_info *compptr;
00556 
00557   if (cinfo->arith_code) {
00558     
00559 
00560 
00561 
00562     emit_dac(cinfo);
00563   } else {
00564     
00565 
00566 
00567     for (i = 0; i < cinfo->comps_in_scan; i++) {
00568       compptr = cinfo->cur_comp_info[i];
00569       if (cinfo->progressive_mode) {
00570         
00571         if (cinfo->Ss == 0) {
00572           if (cinfo->Ah == 0)   
00573             emit_dht(cinfo, compptr->dc_tbl_no, FALSE);
00574         } else {
00575           emit_dht(cinfo, compptr->ac_tbl_no, TRUE);
00576         }
00577       } else {
00578         
00579         emit_dht(cinfo, compptr->dc_tbl_no, FALSE);
00580         emit_dht(cinfo, compptr->ac_tbl_no, TRUE);
00581       }
00582     }
00583   }
00584 
00585   
00586 
00587 
00588   if (cinfo->restart_interval != marker->last_restart_interval) {
00589     emit_dri(cinfo);
00590     marker->last_restart_interval = cinfo->restart_interval;
00591   }
00592 
00593   emit_sos(cinfo);
00594 }
00595 
00596 
00597 
00598 
00599 
00600 
00601 METHODDEF(void)
00602 write_file_trailer (j_compress_ptr cinfo)
00603 {
00604   emit_marker(cinfo, M_EOI);
00605 }
00606 
00607 
00608 
00609 
00610 
00611 
00612 
00613 
00614 
00615 METHODDEF(void)
00616 write_tables_only (j_compress_ptr cinfo)
00617 {
00618   int i;
00619 
00620   emit_marker(cinfo, M_SOI);
00621 
00622   for (i = 0; i < NUM_QUANT_TBLS; i++) {
00623     if (cinfo->quant_tbl_ptrs[i] != NULL)
00624       (void) emit_dqt(cinfo, i);
00625   }
00626 
00627   if (! cinfo->arith_code) {
00628     for (i = 0; i < NUM_HUFF_TBLS; i++) {
00629       if (cinfo->dc_huff_tbl_ptrs[i] != NULL)
00630         emit_dht(cinfo, i, FALSE);
00631       if (cinfo->ac_huff_tbl_ptrs[i] != NULL)
00632         emit_dht(cinfo, i, TRUE);
00633     }
00634   }
00635 
00636   emit_marker(cinfo, M_EOI);
00637 }
00638 
00639 
00640 
00641 
00642 
00643 
00644 GLOBAL(void)
00645 jinit_marker_writer (j_compress_ptr cinfo)
00646 {
00647   my_marker_ptr marker;
00648 
00649   
00650   marker = (my_marker_ptr)
00651     (*cinfo->mem->alloc_small) ((j_common_ptr) cinfo, JPOOL_IMAGE,
00652                                 SIZEOF(my_marker_writer));
00653   cinfo->marker = (struct jpeg_marker_writer *) marker;
00654   
00655   marker->pub.write_file_header = write_file_header;
00656   marker->pub.write_frame_header = write_frame_header;
00657   marker->pub.write_scan_header = write_scan_header;
00658   marker->pub.write_file_trailer = write_file_trailer;
00659   marker->pub.write_tables_only = write_tables_only;
00660   marker->pub.write_marker_header = write_marker_header;
00661   marker->pub.write_marker_byte = write_marker_byte;
00662   
00663   marker->last_restart_interval = 0;
00664 }