]> Creatis software - gdcm.git/blob - src/gdcmJPEGFragment.cxx
ENH: Minor patch that rework code to avoid duplicate code.
[gdcm.git] / src / gdcmJPEGFragment.cxx
1 /*=========================================================================
2                                                                                 
3   Program:   gdcm
4   Module:    $RCSfile: gdcmJPEGFragment.cxx,v $
5   Language:  C++
6   Date:      $Date: 2005/01/17 03:05:55 $
7   Version:   $Revision: 1.4 $
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                                                                                 
19 #include "gdcmJPEGFragment.h"
20 #include "gdcmDebug.h"
21
22 namespace gdcm
23 {
24
25 // For JPEG 2000, body in file gdcmJpeg2000.cxx
26 bool gdcm_read_JPEG2000_file (std::ifstream* fp, void* image_buffer);
27
28 // For JPEG 8 Bits, body in file gdcmJpeg8.cxx
29 bool gdcm_read_JPEG_file8 (std::ifstream *fp, void *image_buffer);
30 bool gdcm_read_JPEG_memory8    (const JOCTET *buffer, const size_t buflen, 
31                                 void *image_buffer,
32                                 size_t *howManyRead, size_t *howManyWritten);
33 //
34 // For JPEG 12 Bits, body in file gdcmJpeg12.cxx
35 bool gdcm_read_JPEG_file12 (std::ifstream *fp, void *image_buffer);
36 bool gdcm_read_JPEG_memory12   (const JOCTET *buffer, const size_t buflen, 
37                                 void *image_buffer,
38                                 size_t *howManyRead, size_t *howManyWritten);
39
40 // For JPEG 16 Bits, body in file gdcmJpeg16.cxx
41 // Beware this is misleading there is no 16bits DCT algorithm, only
42 // jpeg lossless compression exist in 16bits.
43 bool gdcm_read_JPEG_file16 (std::ifstream *fp, void *image_buffer);
44 bool gdcm_read_JPEG_memory16   (const JOCTET *buffer, const size_t buflen, 
45                                 void* image_buffer,
46                                 size_t *howManyRead, size_t *howManyWritten);
47
48 /**
49  * \brief Default constructor.
50  */
51 JPEGFragment::JPEGFragment()
52 {
53    Offset = 0;
54    Length = 0;
55 }
56
57 /**
58  * \brief        Print self.
59  * @param os     Stream to print to.
60  * @param indent Indentation string to be prepended during printing.
61  */
62 void JPEGFragment::Print( std::ostream &os, std::string indent )
63 {
64    os << indent
65       << "JPEG fragment: offset : " <<  Offset
66       << "   length : " <<  Length
67       << std::endl;
68 }
69
70 /**
71  * \brief Decompress 8bits JPEG Fragment
72  * @param buffer     output (data decompress)
73  * @param nBits      8/12 or 16 bits jpeg
74  */
75 void JPEGFragment::DecompressJPEGFramesFromFile(std::ifstream *fp, uint8_t *buffer, int nBits)
76 {
77    // First thing need to reset file to proper position:
78    fp->seekg( Offset, std::ios::beg);
79
80    if ( nBits == 8 )
81    {
82       // JPEG Lossy : call to IJG 6b
83       if ( ! gdcm_read_JPEG_file8( fp, buffer) )
84       {
85          //return false;
86       }
87    }
88    else if ( nBits <= 12 )
89    {
90       // Reading Fragment pixels
91       if ( ! gdcm_read_JPEG_file12 ( fp, buffer) )
92       {
93          //return false;
94       }
95    }
96    else if ( nBits <= 16 )
97    {
98       // Reading Fragment pixels
99       if ( ! gdcm_read_JPEG_file16 ( fp, buffer) )
100       {
101          //return false;
102       }
103       //gdcmAssertMacro( IsJPEGLossless );
104    }
105    else
106    {
107       // other JPEG lossy not supported
108       gdcmErrorMacro( "Unknown jpeg lossy compression ");
109       //return false;
110    }
111
112 }
113
114 void JPEGFragment::DecompressJPEGSingleFrameFragmentsFromFile(JOCTET *buffer, size_t totalLength, uint8_t* raw, int nBits)
115 {
116    size_t howManyRead = 0;
117    size_t howManyWritten = 0;
118    
119    if ( nBits == 8)
120    {
121       if ( ! gdcm_read_JPEG_memory8( buffer, totalLength, raw,
122                                      &howManyRead, &howManyWritten ) ) 
123       {
124          gdcmErrorMacro( "Failed to read jpeg8 ");
125          delete [] buffer;
126          //return false;
127       }
128    }
129    else if ( nBits <= 12)
130    {
131       if ( ! gdcm_read_JPEG_memory12( buffer, totalLength, raw,
132                                       &howManyRead, &howManyWritten ) ) 
133       {
134          gdcmErrorMacro( "Failed to read jpeg12 ");
135             delete [] buffer;
136             //return false;
137       }
138    }
139    else if ( nBits <= 16)
140    {
141       
142       if ( ! gdcm_read_JPEG_memory16( buffer, totalLength, raw,
143                                       &howManyRead, &howManyWritten ) ) 
144       {
145          gdcmErrorMacro( "Failed to read jpeg16 ");
146          delete [] buffer;
147          //return false;
148       }
149    }
150    else
151    {
152       // other JPEG lossy not supported
153       gdcmErrorMacro( "Unsupported jpeg lossy compression ");
154       delete [] buffer;
155       //return false;
156    }      
157
158 }
159
160 void JPEGFragment::DecompressJPEGFragmentedFramesFromFile(JOCTET *buffer, uint8_t* raw, int nBits, size_t &howManyRead, size_t &howManyWritten, size_t totalLength)
161 {
162    if ( nBits == 8 )
163    {
164      if ( ! gdcm_read_JPEG_memory8( buffer+howManyRead, totalLength-howManyRead,
165                                   raw+howManyWritten,
166                                   &howManyRead, &howManyWritten ) ) 
167        {
168          gdcmErrorMacro( "Failed to read jpeg8");
169          //delete [] buffer;
170          //return false;
171        }
172    }
173    else if ( nBits <= 12 )
174    {
175    
176      if ( ! gdcm_read_JPEG_memory12( buffer+howManyRead, totalLength-howManyRead,
177                                    raw+howManyWritten,
178                                    &howManyRead, &howManyWritten ) ) 
179        {
180          gdcmErrorMacro( "Failed to read jpeg12");
181          //delete [] buffer;
182          //return false;
183       }
184    }
185    else if ( nBits <= 16 )
186    {
187    
188      if ( ! gdcm_read_JPEG_memory16( buffer+howManyRead, totalLength-howManyRead,
189                                    raw+howManyWritten,
190                                    &howManyRead, &howManyWritten ) ) 
191        {
192          gdcmErrorMacro( "Failed to read jpeg16 ");
193          //delete [] buffer;
194          //return false;
195        }
196    }
197    else
198    {
199       // other JPEG lossy not supported
200       gdcmErrorMacro( "Unsupported jpeg lossy compression ");
201       //delete [] buffer;
202       //return false;
203    }
204 }
205
206 } // end namespace gdcm
207