]> Creatis software - gdcm.git/blob - src/gdcmJpeg.cxx
Avoid warnings
[gdcm.git] / src / gdcmJpeg.cxx
1 /*=========================================================================
2                                                                                 
3   Program:   gdcm
4   Module:    $RCSfile: gdcmJpeg.cxx,v $
5   Language:  C++
6   Date:      $Date: 2005/11/29 08:32:44 $
7   Version:   $Revision: 1.54 $
8                                                                                 
9   Copyright (c) CREATIS (Centre de Recherche et d'Applications en Traitement de
10   l'Image). All rights reserved. See Doc/License.txt or
11   http://www.creatis.insa-lyon.fr/Public/Gdcm/License.html for details.
12                                                                                 
13      This software is distributed WITHOUT ANY WARRANTY; without even
14      the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
15      PURPOSE.  See the above copyright notices for more information.
16                                                                                 
17 =========================================================================*/
18 #include "gdcmFileHelper.h"
19 #include "gdcmJPEGFragment.h"
20 #include "gdcmDebug.h"
21
22 #if defined(__sgi) && !defined(__GNUC__)
23 // Try to get rid of the warning:
24 //cc-3505 CC: WARNING File = /usr/include/internal/setjmp_core.h, Line = 74
25 //  setjmp not marked as unknown_control_flow because it is not declared as a
26 //          function
27 //
28 //  #pragma unknown_control_flow (setjmp)
29 #  if   (_COMPILER_VERSION >= 730)
30 #  pragma set woff 3505
31 #  endif
32 #endif
33 #ifdef _MSC_VER
34 // Let us get rid of this funny warning on /W4:
35 // warning C4611: interaction between '_setjmp' and C++ object
36 // destruction is non-portable
37 #pragma warning( disable : 4611 )
38 #endif
39
40 #include <setjmp.h>
41 #include <fstream>
42 #include "jdatasrc.cxx"
43 #include "jdatadst.cxx"
44
45 namespace gdcm 
46 {
47
48  /**
49  * \brief   routine for JPEG decompression 
50  * @param fp pointer to an already open file descriptor 
51  *                      8 significant bits per pixel
52  * @param im_buf Points to array (of R,G,B-order) data to compress
53  * @param quality compression quality
54  * @param image_height Number of rows in image 
55  * @param image_width Number of columns in image
56  * @return 1 on success, 0 on error
57  */
58  
59 bool gdcm_write_JPEG_file (std::ostream *fp, void *im_buf, 
60                            int image_width, int image_height, int quality)
61 {
62
63    JSAMPLE *image_buffer = (JSAMPLE*) im_buf;
64
65   /* This struct contains the JPEG compression parameters and pointers to
66    * working space (which is allocated as needed by the JPEG library).
67    * It is possible to have several such structures, representing multiple
68    * compression/decompression processes, in existence at once.  We refer
69    * to any one struct (and its associated working data) as a "JPEG object".
70    */
71   struct jpeg_compress_struct cinfo;
72   /* This struct represents a JPEG error handler.  It is declared separately
73    * because applications often want to supply a specialized error handler
74    * (see the second half of this file for an example).  But here we just
75    * take the easy way out and use the standard error handler, which will
76    * print a message on stderr and call exit() if compression fails.
77    * Note that this struct must live as long as the main JPEG parameter
78    * struct, to avoid dangling-pointer problems.
79    */
80   struct jpeg_error_mgr jerr;
81   /* More stuff */
82   //FILE*  outfile;    /* target FILE* /
83   JSAMPROW row_pointer[1];   /* pointer to JSAMPLE row[s] */
84   int row_stride;            /* physical row width in image buffer */
85
86   /* Step 1: allocate and initialize JPEG compression object */
87
88   /* We have to set up the error handler first, in case the initialization
89    * step fails.  (Unlikely, but it could happen if you are out of memory.)
90    * This routine fills in the contents of struct jerr, and returns jerr's
91    * address which we place into the link field in cinfo.
92    */
93   cinfo.err = jpeg_std_error(&jerr);
94   /* Now we can initialize the JPEG compression object. */
95   jpeg_create_compress(&cinfo);
96
97   /* Step 2: specify data destination (eg, a file) */
98   /* Note: steps 2 and 3 can be done in either order. */
99
100   /* Here we use the library-supplied code to send compressed data to a
101    * stdio stream.  You can also write your own code to do something else.
102    * VERY IMPORTANT: use "b" option to fopen() if you are on a machine that
103    * requires it in order to write binary files.
104    */
105  // if ((outfile = fopen(filename, "wb")) == NULL) {
106  //   fprintf(stderr, "can't open %s\n", filename);
107  //   exit(1);
108  //
109  // }
110   assert( 0 );
111   (void)fp;
112   //jpeg_stdio_dest(&cinfo, fp, 0, 0, image_width, image_height, quality);
113
114   /* Step 3: set parameters for compression */
115
116   /* First we supply a description of the input image.
117    * Four fields of the cinfo struct must be filled in:
118    */
119   cinfo.image_width = image_width;/* image width and height, in pixels */
120   cinfo.image_height = image_height;
121   cinfo.input_components = 3;     /* # of color components per pixel */
122   cinfo.in_color_space = JCS_RGB; /* colorspace of input image */
123   /* Now use the library's routine to set default compression parameters.
124    * (You must set at least cinfo.in_color_space before calling this,
125    * since the defaults depend on the source color space.)
126    */
127   jpeg_set_defaults(&cinfo);
128   /* Now you can set any non-default parameters you wish to.
129    * Here we just illustrate the use of quality (quantization table) scaling:
130    */
131   jpeg_set_quality(&cinfo, quality, TRUE /* limit to baseline-JPEG values */);
132
133   /* Step 4: Start compressor */
134
135   /* TRUE ensures that we will write a complete interchange-JPEG file.
136    * Pass TRUE unless you are very sure of what you're doing.
137    */
138   jpeg_start_compress(&cinfo, TRUE);
139
140   /* Step 5: while (scan lines remain to be written) */
141   /*           jpeg_write_scanlines(...); */
142
143   /* Here we use the library's state variable cinfo.next_scanline as the
144    * loop counter, so that we don't have to keep track ourselves.
145    * To keep things simple, we pass one scanline per call; you can pass
146    * more if you wish, though.
147    */
148   row_stride = image_width * 3;/* JSAMPLEs per row in image_buffer */
149
150   while (cinfo.next_scanline < cinfo.image_height) {
151     /* jpeg_write_scanlines expects an array of pointers to scanlines.
152      * Here the array is only one element long, but you could pass
153      * more than one scanline at a time if that's more convenient.
154      */
155     row_pointer[0] = & image_buffer[cinfo.next_scanline * row_stride];
156
157     (void) jpeg_write_scanlines(&cinfo, row_pointer, 1);
158   }
159
160   /* Step 6: Finish compression */
161
162   jpeg_finish_compress(&cinfo);
163   
164   /* After finish_compress, we can close the output file. */
165   
166  // fclose(fp); --> the caller will close (multiframe treatement)
167
168   /* Step 7: release JPEG compression object */
169
170   /* This is an important step since it will release a good deal of memory. */
171   jpeg_destroy_compress(&cinfo);
172
173   /* And we're done! */
174
175   return true;
176 }
177
178 //-----------------------------------------------------------------------------
179 struct my_error_mgr {
180    struct jpeg_error_mgr pub; /* "public" fields */
181    jmp_buf setjmp_buffer;     /* for return to caller */
182 };
183 typedef struct my_error_mgr* my_error_ptr;
184 //-----------------------------------------------------------------------------
185
186 /*
187  * Here's the routine that will replace the standard error_exit method:
188  */
189 extern "C" {
190 METHODDEF(void) my_error_exit (j_common_ptr cinfo) {
191    /* cinfo->err really points to a my_error_mgr struct, so coerce pointer */
192    my_error_ptr myerr = (my_error_ptr) cinfo->err;
193
194    /* Always display the message. */
195    /* We could postpone this until after returning, if we chose. */
196    (*cinfo->err->output_message) (cinfo);
197
198    /* Return control to the setjmp point */
199    longjmp(myerr->setjmp_buffer, 1);
200 }
201
202 //METHODDEF(void) my_output_message (j_common_ptr cinfo)
203 //{
204 //   char buffer[JMSG_LENGTH_MAX];
205 // 
206 //   /* Create the message */
207 //   (*cinfo->err->format_message) (cinfo, buffer);
208 //
209 //   // Custom display message, we could be more fancy and throw an exception:
210 //   gdcmStaticErrorMacro( buffer );
211 //}
212
213 }
214 //-----------------------------------------------------------------------------
215  
216 /**
217  * \brief   routine for JPEG decompression 
218  * @param fp pointer to an already open file descriptor 
219  *                      8 significant bits per pixel
220  * @param image_buffer to receive uncompressed pixels
221  * @param statesuspension Suspension State basically it should be 3 otherwise more complex to handle
222  * @return 1 on success, 0 on error
223  */
224 void *SampBuffer; 
225 bool JPEGFragment::ReadJPEGFile (std::ifstream *fp, void *image_buffer, int &statesuspension)
226 {
227    pImage = (uint8_t*)image_buffer;
228    // This struct contains the JPEG decompression parameters and pointers to
229    // working space (which is allocated as needed by the JPEG library).
230
231    static struct jpeg_decompress_struct cinfo;
232
233    // -------------- inside, we found :
234    // JDIMENSION image_width;       // input image width 
235    // JDIMENSION image_height;      // input image height 
236    // int input_components;         // nb of color components in input image 
237    // J_COLOR_SPACE in_color_space; // colorspace of input image 
238    // double input_gamma;           // image gamma of input image 
239
240    // We use our private extension JPEG error handler.
241    // Note that this struct must live as long as the main JPEG parameter
242    // struct, to avoid dangling-pointer problems.
243
244    struct my_error_mgr jerr;
245
246    JSAMPARRAY buffer;// Output row buffer
247   
248    // rappel :
249    // ------
250    // typedef unsigned char JSAMPLE;
251    // typedef JSAMPLE FAR *JSAMPROW;/* ptr to one image row of pixel samples. */
252    // typedef JSAMPROW *JSAMPARRAY;/* ptr to some rows (a 2-D sample array) */
253    // typedef JSAMPARRAY *JSAMPIMAGE;/* a 3-D sample array: top index is color */
254
255    int row_stride;// physical row width in output buffer
256   
257   // We set up the normal JPEG error routines, then override error_exit.
258   
259   cinfo.err = jpeg_std_error(&jerr.pub);
260   // for any jpeg error call my_error_exit
261   jerr.pub.error_exit = my_error_exit;
262   // for any output message call my_output_message
263   //jerr.pub.output_message = my_output_message;
264
265   // Establish the setjmp return context for my_error_exit to use.
266   if (setjmp(jerr.setjmp_buffer))
267   {
268     // If we get here, the JPEG code has signaled an error.
269     // We need to clean up the JPEG object, close the input file, and return.
270
271     gdcmErrorMacro( "Serious Problem !" );
272     jpeg_destroy_decompress(&cinfo);
273     return 0;
274   }
275   // Now we can initialize the JPEG decompression object.
276   if ( statesuspension == 0 )
277     {
278     jpeg_create_decompress(&cinfo);
279     jpeg_stdio_src(&cinfo, fp, this, 1);
280     }
281   else
282     {
283     jpeg_stdio_src(&cinfo, fp, this, 0);
284     }
285    // Step 3: read file parameters with jpeg_read_header()
286
287    if ( statesuspension < 2 )
288    {
289       if ( jpeg_read_header(&cinfo, TRUE) == JPEG_SUSPENDED )
290       {
291       // Suspension in jpeg_read_header
292       statesuspension = 2; 
293       }
294  
295       // Step 4: set parameters for decompression
296       // prevent the library from performing any color space conversion
297       if ( cinfo.process == JPROC_LOSSLESS )
298       {
299          cinfo.jpeg_color_space = JCS_UNKNOWN;
300          cinfo.out_color_space = JCS_UNKNOWN;
301       }
302    }
303
304    // Step 5: Start decompressor
305    if (statesuspension < 3 )
306    {
307       if ( jpeg_start_decompress(&cinfo) == FALSE )
308       {
309          // Suspension: jpeg_start_decompress
310          statesuspension = 3;
311       }
312
313       // JSAMPLEs per row in output buffer
314       row_stride = cinfo.output_width * cinfo.output_components*2;
315   
316       // Make a one-row-high sample array that will go away when done with image
317       buffer = (*cinfo.mem->alloc_sarray)
318             ((j_common_ptr) &cinfo, JPOOL_IMAGE, row_stride, 1);
319
320       // Step 6: while (scan lines remain to be read)
321
322       // Save the buffer in case of suspension to be able to reuse it later:
323       SampBuffer = buffer;
324    }
325    else
326    {
327       // Suspension: re-use the buffer:
328       buffer = (JSAMPARRAY)SampBuffer;
329    }
330    int bufsize = cinfo.output_width * cinfo.output_components;
331    size_t rowsize = bufsize * sizeof(JSAMPLE);
332
333    while (cinfo.output_scanline < cinfo.output_height)
334    {
335       if ( jpeg_read_scanlines(&cinfo, buffer, 1) == 0 )
336         {
337         // Suspension in jpeg_read_scanlines
338         statesuspension = 3;
339         return true;
340         }
341 // The ijg has no notion of big endian, therefore always swap the jpeg stream
342 #if (defined(GDCM_WORDS_BIGENDIAN) || defined(GDCM_FORCE_BIGENDIAN_EMULATION)) && (CMAKE_BITS_IN_JSAMPLE != 8)
343       uint16_t *buffer16 = (uint16_t*)*buffer;
344       uint16_t *pimage16 = (uint16_t*)pImage;
345       for(unsigned int i=0;i<rowsize/2;i++)
346         pimage16[i] = (buffer16[i] >> 8) | (buffer16[i] << 8 );
347 #else
348       memcpy( pImage, *buffer,rowsize);
349 #endif //GDCM_WORDS_BIGENDIAN
350       pImage+=rowsize;
351    }
352
353    // Step 7: Finish decompression
354    if ( jpeg_finish_decompress(&cinfo) == FALSE )
355      {
356      // Suspension: jpeg_finish_decompress
357      statesuspension = 4;
358      }
359    
360    // Step 8: Release JPEG decompression object
361    jpeg_destroy_decompress(&cinfo);
362
363    // At this point you may want to check to see whether any corrupt-data
364    // warnings occurred (test whether jerr.pub.num_warnings is nonzero).
365
366    return true;
367 }
368
369 #ifdef _MSC_VER
370 // Put the warning back
371 #pragma warning( default : 4611 )
372 #endif
373
374 } // end namespace gdcm