-
-#include <iostream>
-#include <sstream>
-
-#include <stdio.h>
-extern "C" {
-#include "gdcmjpeg/8/jconfig.h"
-#include "gdcmjpeg/8/jpeglib.h"
-#include "gdcmjpeg/8/jinclude.h"
-#include "gdcmjpeg/8/jerror.h"
-}
-
-#include "gdcmJPEGFragment.h"
-#include <setjmp.h>
-#include <fstream>
-
-#include "jdatasrc.cxx"
-#include "jdatadst.cxx"
-
-// PS 3.5, page 66
-void EncodeWithoutBasicOffsetTable(std::ostream *fp, int numFrag, uint32_t length)
-{
- assert( numFrag == 1);
-
- // Item tag:
- uint16_t group = 0xfffe;
- uint16_t elem = 0xe000;
- gdcm::binary_write(*fp, group);
- gdcm::binary_write(*fp, elem);
- // Item Length
- uint32_t item_length = 0x0000;
- gdcm::binary_write(*fp, item_length);
-
- // back again...First fragment
- // Item tag:
- gdcm::binary_write(*fp, group);
- gdcm::binary_write(*fp, elem);
- // Item Length
- gdcm::binary_write(*fp, length);
-
-}
-
-void CloseJpeg(std::ostream *fp)
-{
- // sequence terminator
- uint16_t group = 0xfffe;
- uint16_t elem = 0xe000;
- gdcm::binary_write(*fp, group);
- gdcm::binary_write(*fp, elem);
-
- uint32_t length = 0x0;
- gdcm::binary_write(*fp, length);
-}
-
-// PS 3.5, page 67
-void EncodeWithBasicOffsetTable(std::ostream *fp, int numFrag)
-{
- (void)fp; (void)numFrag;
-}
-
-bool InitializeJpeg(std::ostream *fp, int fragment_size, int image_width, int image_height,
- int sample_pixel, int quality, struct jpeg_compress_struct &cinfo, int &row_stride)
-{
-
- /* This struct contains the JPEG compression parameters and pointers to
- * working space (which is allocated as needed by the JPEG library).
- * It is possible to have several such structures, representing multiple
- * compression/decompression processes, in existence at once. We refer
- * to any one struct (and its associated working data) as a "JPEG object".
- */
- //struct jpeg_compress_struct cinfo;
- /* This struct represents a JPEG error handler. It is declared separately
- * because applications often want to supply a specialized error handler
- * (see the second half of this file for an example). But here we just
- * take the easy way out and use the standard error handler, which will
- * print a message on stderr and call exit() if compression fails.
- * Note that this struct must live as long as the main JPEG parameter
- * struct, to avoid dangling-pointer problems.
- */
- struct jpeg_error_mgr jerr;
- /* More stuff */
-
- /* Step 1: allocate and initialize JPEG compression object */
-
- /* We have to set up the error handler first, in case the initialization
- * step fails. (Unlikely, but it could happen if you are out of memory.)
- * This routine fills in the contents of struct jerr, and returns jerr's
- * address which we place into the link field in cinfo.
- */
- cinfo.err = jpeg_std_error(&jerr);
- /* Now we can initialize the JPEG compression object. */
- jpeg_create_compress(&cinfo);
-
- /* Step 2: specify data destination (eg, a file) */
- /* Note: steps 2 and 3 can be done in either order. */
-
- jpeg_stdio_dest(&cinfo, fp, fragment_size, 1);
-
- /* Step 3: set parameters for compression */
-
- /* First we supply a description of the input image.
- * Four fields of the cinfo struct must be filled in:
- */
- cinfo.image_width = image_width;/* image width and height, in pixels */
- cinfo.image_height = image_height;
- if ( sample_pixel == 3 )
- {
- cinfo.input_components = 3; /* # of color components per pixel */
- cinfo.in_color_space = JCS_RGB; /* colorspace of input image */
- }
- else
- {
- cinfo.input_components = 1; /* # of color components per pixel */
- cinfo.in_color_space = JCS_GRAYSCALE; /* colorspace of input image */
- }
- /* Now use the library's routine to set default compression parameters.
- * (You must set at least cinfo.in_color_space before calling this,
- * since the defaults depend on the source color space.)
- */
- jpeg_set_defaults(&cinfo);
- /* Now you can set any non-default parameters you wish to.
- * Here we just illustrate the use of quality (quantization table) scaling:
- */
- jpeg_set_quality(&cinfo, quality, TRUE /* limit to baseline-JPEG values */);
-
- /* Step 4: Start compressor */
-
- /* TRUE ensures that we will write a complete interchange-JPEG file.
- * Pass TRUE unless you are very sure of what you're doing.
- */
- jpeg_start_compress(&cinfo, TRUE);
-
- /* Step 5: while (scan lines remain to be written) */
- /* jpeg_write_scanlines(...); */
-
- /* Here we use the library's state variable cinfo.next_scanline as the
- * loop counter, so that we don't have to keep track ourselves.
- * To keep things simple, we pass one scanline per call; you can pass
- * more if you wish, though.
- */
- if (sample_pixel == 3)
- {
- row_stride = image_width * 3;/* JSAMPLEs per row in image_buffer */
- }
- else
- {
- row_stride = image_width * 1;/* JSAMPLEs per row in image_buffer */
- }
-
- /* everything was ok */
- return true;
-}
-
-bool FinalizeJpeg(struct jpeg_compress_struct &cinfo)
-{
- /* Step 6: Finish compression */
-
- jpeg_finish_compress(&cinfo);
-
- /* Step 7: release JPEG compression object */
-
- /* This is an important step since it will release a good deal of memory. */
- jpeg_destroy_compress(&cinfo);
-
- /* And we're done! */
- return true;
-}
-
-// If false then suspension return
-bool WriteScanlines(struct jpeg_compress_struct &cinfo, void *input_buffer, int row_stride)
-{
- JSAMPLE *image_buffer = (JSAMPLE*) input_buffer;
- JSAMPROW row_pointer[1]; /* pointer to JSAMPLE row[s] */
- row_pointer[0] = image_buffer;
-
- while (cinfo.next_scanline < cinfo.image_height) {
- /* jpeg_write_scanlines expects an array of pointers to scanlines.
- * Here the array is only one element long, but you could pass
- * more than one scanline at a time if that's more convenient.
- */
- //row_pointer[0] = & image_buffer[cinfo.next_scanline * row_stride];
-
- if( jpeg_write_scanlines(&cinfo, row_pointer, 1) != 1)
- {
- //entering suspension mode, basically we wrote the whole jpeg fragment
- //suspend = 1;
- return false;
- }
- row_pointer[0] += row_stride;
- }
-
- // Well looks like we are done writting the scanlines
- return true;
-}
-
-// input_buffer is ONE image
-// fragment_size is the size of this image (fragment)
-bool CreateOneFrame (std::ostream *fp, void *input_buffer, int fragment_size,
- int image_width, int image_height, int sample_pixel, int quality,
- int &suspend)
-{
- struct jpeg_compress_struct cinfo;
- int row_stride; /* physical row width in image buffer */
- bool r = InitializeJpeg(fp, fragment_size, image_width, image_height,
- sample_pixel, quality, cinfo, row_stride);
- assert( r );
-
- r = WriteScanlines(cinfo, input_buffer, row_stride);
- assert( r );
-
- r = FinalizeJpeg(cinfo);
- assert( r );
-
- return true;
-}