blob: a0dc1aff178de211b85b583b567c3999d7d7dff5 [file] [log] [blame]
cristy3ed852e2009-09-05 21:47:34 +00001/*
2%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
3% %
4% %
5% %
6% Y Y U U V V %
7% Y Y U U V V %
8% Y U U V V %
9% Y U U V V %
10% Y UUU V %
11% %
12% %
13% Read/Write Raw CCIR 601 4:1:1 or 4:2:2 Image Format %
14% %
15% Software Design %
16% John Cristy %
17% July 1992 %
18% %
19% %
cristy7e41fe82010-12-04 23:12:08 +000020% Copyright 1999-2011 ImageMagick Studio LLC, a non-profit organization %
cristy3ed852e2009-09-05 21:47:34 +000021% dedicated to making software imaging solutions freely available. %
22% %
23% You may not use this file except in compliance with the License. You may %
24% obtain a copy of the License at %
25% %
26% http://www.imagemagick.org/script/license.php %
27% %
28% Unless required by applicable law or agreed to in writing, software %
29% distributed under the License is distributed on an "AS IS" BASIS, %
30% WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. %
31% See the License for the specific language governing permissions and %
32% limitations under the License. %
33% %
34%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
35%
36%
37*/
38
39/*
40 Include declarations.
41*/
cristy4c08aed2011-07-01 19:47:50 +000042#include "MagickCore/studio.h"
43#include "MagickCore/blob.h"
44#include "MagickCore/blob-private.h"
45#include "MagickCore/cache.h"
46#include "MagickCore/colorspace.h"
47#include "MagickCore/constitute.h"
48#include "MagickCore/exception.h"
49#include "MagickCore/exception-private.h"
50#include "MagickCore/geometry.h"
51#include "MagickCore/image.h"
52#include "MagickCore/image-private.h"
53#include "MagickCore/list.h"
54#include "MagickCore/magick.h"
55#include "MagickCore/memory_.h"
56#include "MagickCore/monitor.h"
57#include "MagickCore/monitor-private.h"
58#include "MagickCore/pixel-accessor.h"
59#include "MagickCore/resize.h"
60#include "MagickCore/quantum-private.h"
61#include "MagickCore/static.h"
62#include "MagickCore/string_.h"
63#include "MagickCore/module.h"
64#include "MagickCore/utility.h"
cristy3ed852e2009-09-05 21:47:34 +000065
66/*
67 Forward declarations.
68*/
69static MagickBooleanType
cristy3a37efd2011-08-28 20:31:03 +000070 WriteYUVImage(const ImageInfo *,Image *,ExceptionInfo *);
cristy3ed852e2009-09-05 21:47:34 +000071
72/*
73%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
74% %
75% %
76% %
77% R e a d Y U V I m a g e %
78% %
79% %
80% %
81%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
82%
83% ReadYUVImage() reads an image with digital YUV (CCIR 601 4:1:1, plane
84% or partition interlaced, or 4:2:2 plane, partition interlaced or
85% noninterlaced) bytes and returns it. It allocates the memory necessary
86% for the new Image structure and returns a pointer to the new image.
87%
88% The format of the ReadYUVImage method is:
89%
90% Image *ReadYUVImage(const ImageInfo *image_info,ExceptionInfo *exception)
91%
92% A description of each parameter follows:
93%
94% o image_info: the image info.
95%
96% o exception: return any errors or warnings in this structure.
97%
98*/
99static Image *ReadYUVImage(const ImageInfo *image_info,ExceptionInfo *exception)
100{
101 Image
102 *chroma_image,
103 *image,
104 *resize_image;
105
106 InterlaceType
107 interlace;
108
cristy3ed852e2009-09-05 21:47:34 +0000109 MagickBooleanType
110 status;
111
cristy4c08aed2011-07-01 19:47:50 +0000112 register const Quantum
cristy3ed852e2009-09-05 21:47:34 +0000113 *chroma_pixels;
114
cristybb503372010-05-27 20:51:26 +0000115 register ssize_t
cristy3ed852e2009-09-05 21:47:34 +0000116 x;
117
cristy4c08aed2011-07-01 19:47:50 +0000118 register Quantum
cristy3ed852e2009-09-05 21:47:34 +0000119 *q;
120
cristy3ed852e2009-09-05 21:47:34 +0000121 register unsigned char
122 *p;
123
124 ssize_t
cristyc6da28e2011-04-28 01:41:35 +0000125 count,
126 horizontal_factor,
127 quantum,
128 vertical_factor,
129 y;
cristy3ed852e2009-09-05 21:47:34 +0000130
131 unsigned char
132 *scanline;
133
134 /*
135 Allocate image structure.
136 */
137 assert(image_info != (const ImageInfo *) NULL);
138 assert(image_info->signature == MagickSignature);
139 if (image_info->debug != MagickFalse)
140 (void) LogMagickEvent(TraceEvent,GetMagickModule(),"%s",
141 image_info->filename);
142 assert(exception != (ExceptionInfo *) NULL);
143 assert(exception->signature == MagickSignature);
144 image=AcquireImage(image_info);
145 if ((image->columns == 0) || (image->rows == 0))
146 ThrowReaderException(OptionError,"MustSpecifyImageSize");
cristy292ebf12011-02-08 20:21:31 +0000147 quantum=image->depth <= 8 ? 1 : 2;
cristy3ed852e2009-09-05 21:47:34 +0000148 interlace=image_info->interlace;
149 horizontal_factor=2;
150 vertical_factor=2;
151 if (image_info->sampling_factor != (char *) NULL)
152 {
153 GeometryInfo
154 geometry_info;
155
156 MagickStatusType
157 flags;
158
159 flags=ParseGeometry(image_info->sampling_factor,&geometry_info);
cristybb503372010-05-27 20:51:26 +0000160 horizontal_factor=(ssize_t) geometry_info.rho;
161 vertical_factor=(ssize_t) geometry_info.sigma;
cristy3ed852e2009-09-05 21:47:34 +0000162 if ((flags & SigmaValue) == 0)
163 vertical_factor=horizontal_factor;
164 if ((horizontal_factor != 1) && (horizontal_factor != 2) &&
165 (vertical_factor != 1) && (vertical_factor != 2))
166 ThrowReaderException(CorruptImageError,"UnexpectedSamplingFactor");
167 }
168 if ((interlace == UndefinedInterlace) ||
169 ((interlace == NoInterlace) && (vertical_factor == 2)))
170 {
171 interlace=NoInterlace; /* CCIR 4:2:2 */
172 if (vertical_factor == 2)
173 interlace=PlaneInterlace; /* CCIR 4:1:1 */
174 }
175 if (interlace != PartitionInterlace)
176 {
177 /*
178 Open image file.
179 */
180 status=OpenBlob(image_info,image,ReadBinaryBlobMode,exception);
181 if (status == MagickFalse)
182 {
183 image=DestroyImageList(image);
184 return((Image *) NULL);
185 }
cristyd4297022010-09-16 22:59:09 +0000186 if (DiscardBlobBytes(image,image->offset) == MagickFalse)
187 ThrowFileException(exception,CorruptImageError,"UnexpectedEndOfFile",
188 image->filename);
cristy3ed852e2009-09-05 21:47:34 +0000189 }
190 /*
191 Allocate memory for a scanline.
192 */
193 if (interlace == NoInterlace)
194 scanline=(unsigned char *) AcquireQuantumMemory((size_t) 2UL*
cristy292ebf12011-02-08 20:21:31 +0000195 image->columns+2UL,quantum*sizeof(*scanline));
cristy3ed852e2009-09-05 21:47:34 +0000196 else
197 scanline=(unsigned char *) AcquireQuantumMemory((size_t) image->columns,
cristy292ebf12011-02-08 20:21:31 +0000198 quantum*sizeof(*scanline));
cristy3ed852e2009-09-05 21:47:34 +0000199 if (scanline == (unsigned char *) NULL)
200 ThrowReaderException(ResourceLimitError,"MemoryAllocationFailed");
201 do
202 {
203 chroma_image=CloneImage(image,image->columns/horizontal_factor,
204 image->rows/vertical_factor,MagickTrue,exception);
205 if (chroma_image == (Image *) NULL)
206 ThrowReaderException(ResourceLimitError,"MemoryAllocationFailed");
207 /*
208 Convert raster image to pixel packets.
209 */
210 if ((image_info->ping != MagickFalse) && (image_info->number_scenes != 0))
211 if (image->scene >= (image_info->scene+image_info->number_scenes-1))
212 break;
213 if (interlace == PartitionInterlace)
214 {
215 AppendImageFormat("Y",image->filename);
216 status=OpenBlob(image_info,image,ReadBinaryBlobMode,exception);
217 if (status == MagickFalse)
218 {
219 image=DestroyImageList(image);
220 return((Image *) NULL);
221 }
222 }
cristybb503372010-05-27 20:51:26 +0000223 for (y=0; y < (ssize_t) image->rows; y++)
cristy3ed852e2009-09-05 21:47:34 +0000224 {
cristy4c08aed2011-07-01 19:47:50 +0000225 register Quantum
cristy3ed852e2009-09-05 21:47:34 +0000226 *chroma_pixels;
227
228 if (interlace == NoInterlace)
229 {
230 if ((y > 0) || (GetPreviousImageInList(image) == (Image *) NULL))
cristy292ebf12011-02-08 20:21:31 +0000231 count=ReadBlob(image,(size_t) (2*quantum*image->columns),scanline);
cristy3ed852e2009-09-05 21:47:34 +0000232 p=scanline;
233 q=QueueAuthenticPixels(image,0,y,image->columns,1,exception);
cristyacd2ed22011-08-30 01:44:23 +0000234 if (q == (Quantum *) NULL)
cristy3ed852e2009-09-05 21:47:34 +0000235 break;
236 chroma_pixels=QueueAuthenticPixels(chroma_image,0,y,
237 chroma_image->columns,1,exception);
cristy4c08aed2011-07-01 19:47:50 +0000238 if (chroma_pixels == (Quantum *) NULL)
cristy3ed852e2009-09-05 21:47:34 +0000239 break;
cristybb503372010-05-27 20:51:26 +0000240 for (x=0; x < (ssize_t) image->columns; x+=2)
cristy3ed852e2009-09-05 21:47:34 +0000241 {
cristy4c08aed2011-07-01 19:47:50 +0000242 SetPixelRed(image,0,chroma_pixels);
cristy292ebf12011-02-08 20:21:31 +0000243 if (quantum == 1)
cristy4c08aed2011-07-01 19:47:50 +0000244 SetPixelGreen(image,ScaleCharToQuantum(*p++),chroma_pixels);
cristy292ebf12011-02-08 20:21:31 +0000245 else
246 {
cristy4c08aed2011-07-01 19:47:50 +0000247 SetPixelGreen(image,ScaleShortToQuantum(((*p) << 8) | *(p+1)),
248 chroma_pixels);
cristy292ebf12011-02-08 20:21:31 +0000249 p+=2;
250 }
251 if (quantum == 1)
cristy4c08aed2011-07-01 19:47:50 +0000252 SetPixelRed(image,ScaleCharToQuantum(*p++),q);
cristy292ebf12011-02-08 20:21:31 +0000253 else
254 {
cristy4c08aed2011-07-01 19:47:50 +0000255 SetPixelRed(image,ScaleShortToQuantum(((*p) << 8) | *(p+1)),q);
cristy292ebf12011-02-08 20:21:31 +0000256 p+=2;
257 }
cristy4c08aed2011-07-01 19:47:50 +0000258 SetPixelGreen(image,0,q);
259 SetPixelBlue(image,0,q);
cristyed231572011-07-14 02:18:59 +0000260 q+=GetPixelChannels(image);
cristy4c08aed2011-07-01 19:47:50 +0000261 SetPixelGreen(image,0,q);
262 SetPixelBlue(image,0,q);
cristy292ebf12011-02-08 20:21:31 +0000263 if (quantum == 1)
cristy4c08aed2011-07-01 19:47:50 +0000264 SetPixelBlue(image,ScaleCharToQuantum(*p++),chroma_pixels);
cristy292ebf12011-02-08 20:21:31 +0000265 else
266 {
cristy4c08aed2011-07-01 19:47:50 +0000267 SetPixelBlue(image,ScaleShortToQuantum(((*p) << 8) | *(p+1)),
268 chroma_pixels);
cristy292ebf12011-02-08 20:21:31 +0000269 p+=2;
270 }
271 if (quantum == 1)
cristy4c08aed2011-07-01 19:47:50 +0000272 SetPixelRed(image,ScaleCharToQuantum(*p++),q);
cristy292ebf12011-02-08 20:21:31 +0000273 else
274 {
cristy4c08aed2011-07-01 19:47:50 +0000275 SetPixelRed(image,ScaleShortToQuantum(((*p) << 8) | *(p+1)),q);
cristy292ebf12011-02-08 20:21:31 +0000276 p+=2;
277 }
cristy3ed852e2009-09-05 21:47:34 +0000278 chroma_pixels++;
cristyed231572011-07-14 02:18:59 +0000279 q+=GetPixelChannels(image);
cristy3ed852e2009-09-05 21:47:34 +0000280 }
281 }
282 else
283 {
284 if ((y > 0) || (GetPreviousImageInList(image) == (Image *) NULL))
cristy292ebf12011-02-08 20:21:31 +0000285 count=ReadBlob(image,(size_t) quantum*image->columns,scanline);
cristy3ed852e2009-09-05 21:47:34 +0000286 p=scanline;
287 q=QueueAuthenticPixels(image,0,y,image->columns,1,exception);
cristyacd2ed22011-08-30 01:44:23 +0000288 if (q == (Quantum *) NULL)
cristy3ed852e2009-09-05 21:47:34 +0000289 break;
cristybb503372010-05-27 20:51:26 +0000290 for (x=0; x < (ssize_t) image->columns; x++)
cristy3ed852e2009-09-05 21:47:34 +0000291 {
cristy292ebf12011-02-08 20:21:31 +0000292 if (quantum == 1)
cristy4c08aed2011-07-01 19:47:50 +0000293 SetPixelRed(image,ScaleCharToQuantum(*p++),q);
cristy292ebf12011-02-08 20:21:31 +0000294 else
295 {
cristy4c08aed2011-07-01 19:47:50 +0000296 SetPixelRed(image,ScaleShortToQuantum(((*p) << 8) | *(p+1)),q);
cristy292ebf12011-02-08 20:21:31 +0000297 p+=2;
298 }
cristy4c08aed2011-07-01 19:47:50 +0000299 SetPixelGreen(image,0,q);
300 SetPixelBlue(image,0,q);
cristyed231572011-07-14 02:18:59 +0000301 q+=GetPixelChannels(image);
cristy3ed852e2009-09-05 21:47:34 +0000302 }
303 }
304 if (SyncAuthenticPixels(image,exception) == MagickFalse)
305 break;
306 if (interlace == NoInterlace)
307 if (SyncAuthenticPixels(chroma_image,exception) == MagickFalse)
308 break;
309 if (image->previous == (Image *) NULL)
310 {
cristycee97112010-05-28 00:44:52 +0000311 status=SetImageProgress(image,LoadImageTag,(MagickOffsetType) y,
cristy292ebf12011-02-08 20:21:31 +0000312 image->rows);
cristy3ed852e2009-09-05 21:47:34 +0000313 if (status == MagickFalse)
314 break;
315 }
316 }
317 if (interlace == PartitionInterlace)
318 {
319 (void) CloseBlob(image);
320 AppendImageFormat("U",image->filename);
321 status=OpenBlob(image_info,image,ReadBinaryBlobMode,exception);
322 if (status == MagickFalse)
323 {
324 image=DestroyImageList(image);
325 return((Image *) NULL);
326 }
327 }
328 if (interlace != NoInterlace)
329 {
cristybb503372010-05-27 20:51:26 +0000330 for (y=0; y < (ssize_t) chroma_image->rows; y++)
cristy3ed852e2009-09-05 21:47:34 +0000331 {
cristy292ebf12011-02-08 20:21:31 +0000332 count=ReadBlob(image,(size_t) quantum*chroma_image->columns,scanline);
cristy3ed852e2009-09-05 21:47:34 +0000333 p=scanline;
334 q=QueueAuthenticPixels(chroma_image,0,y,chroma_image->columns,1,
335 exception);
cristyacd2ed22011-08-30 01:44:23 +0000336 if (q == (Quantum *) NULL)
cristy3ed852e2009-09-05 21:47:34 +0000337 break;
cristybb503372010-05-27 20:51:26 +0000338 for (x=0; x < (ssize_t) chroma_image->columns; x++)
cristy3ed852e2009-09-05 21:47:34 +0000339 {
cristy4c08aed2011-07-01 19:47:50 +0000340 SetPixelRed(chroma_image,0,q);
cristy292ebf12011-02-08 20:21:31 +0000341 if (quantum == 1)
cristy4c08aed2011-07-01 19:47:50 +0000342 SetPixelGreen(chroma_image,ScaleCharToQuantum(*p++),q);
cristy292ebf12011-02-08 20:21:31 +0000343 else
344 {
cristy4c08aed2011-07-01 19:47:50 +0000345 SetPixelGreen(chroma_image,ScaleShortToQuantum(((*p) << 8) |
346 *(p+1)),q);
cristy292ebf12011-02-08 20:21:31 +0000347 p+=2;
348 }
cristy4c08aed2011-07-01 19:47:50 +0000349 SetPixelBlue(chroma_image,0,q);
cristyed231572011-07-14 02:18:59 +0000350 q+=GetPixelChannels(chroma_image);
cristy3ed852e2009-09-05 21:47:34 +0000351 }
352 if (SyncAuthenticPixels(chroma_image,exception) == MagickFalse)
353 break;
354 }
355 if (interlace == PartitionInterlace)
356 {
357 (void) CloseBlob(image);
358 AppendImageFormat("V",image->filename);
359 status=OpenBlob(image_info,image,ReadBinaryBlobMode,exception);
360 if (status == MagickFalse)
361 {
362 image=DestroyImageList(image);
363 return((Image *) NULL);
364 }
365 }
cristybb503372010-05-27 20:51:26 +0000366 for (y=0; y < (ssize_t) chroma_image->rows; y++)
cristy3ed852e2009-09-05 21:47:34 +0000367 {
cristy292ebf12011-02-08 20:21:31 +0000368 count=ReadBlob(image,(size_t) quantum*chroma_image->columns,scanline);
cristy3ed852e2009-09-05 21:47:34 +0000369 p=scanline;
370 q=GetAuthenticPixels(chroma_image,0,y,chroma_image->columns,1,
371 exception);
cristyacd2ed22011-08-30 01:44:23 +0000372 if (q == (Quantum *) NULL)
cristy3ed852e2009-09-05 21:47:34 +0000373 break;
cristybb503372010-05-27 20:51:26 +0000374 for (x=0; x < (ssize_t) chroma_image->columns; x++)
cristy3ed852e2009-09-05 21:47:34 +0000375 {
cristy292ebf12011-02-08 20:21:31 +0000376 if (quantum == 1)
cristy4c08aed2011-07-01 19:47:50 +0000377 SetPixelBlue(chroma_image,ScaleCharToQuantum(*p++),q);
cristy292ebf12011-02-08 20:21:31 +0000378 else
379 {
cristy4c08aed2011-07-01 19:47:50 +0000380 SetPixelBlue(chroma_image,ScaleShortToQuantum(((*p) << 8) |
381 *(p+1)),q);
cristy292ebf12011-02-08 20:21:31 +0000382 p+=2;
383 }
cristyed231572011-07-14 02:18:59 +0000384 q+=GetPixelChannels(chroma_image);
cristy3ed852e2009-09-05 21:47:34 +0000385 }
386 if (SyncAuthenticPixels(chroma_image,exception) == MagickFalse)
387 break;
388 }
389 }
390 /*
391 Scale image.
392 */
393 resize_image=ResizeImage(chroma_image,image->columns,image->rows,
394 TriangleFilter,1.0,exception);
395 chroma_image=DestroyImage(chroma_image);
396 if (resize_image == (Image *) NULL)
397 ThrowReaderException(ResourceLimitError,"MemoryAllocationFailed");
cristybb503372010-05-27 20:51:26 +0000398 for (y=0; y < (ssize_t) image->rows; y++)
cristy3ed852e2009-09-05 21:47:34 +0000399 {
400 q=GetAuthenticPixels(image,0,y,image->columns,1,exception);
401 chroma_pixels=GetVirtualPixels(resize_image,0,y,resize_image->columns,1,
402 &resize_image->exception);
cristyacd2ed22011-08-30 01:44:23 +0000403 if ((q == (Quantum *) NULL) ||
cristy4c08aed2011-07-01 19:47:50 +0000404 (chroma_pixels == (const Quantum *) NULL))
cristy3ed852e2009-09-05 21:47:34 +0000405 break;
cristybb503372010-05-27 20:51:26 +0000406 for (x=0; x < (ssize_t) image->columns; x++)
cristy3ed852e2009-09-05 21:47:34 +0000407 {
cristy4c08aed2011-07-01 19:47:50 +0000408 SetPixelGreen(image,GetPixelGreen(image,chroma_pixels),q);
409 SetPixelBlue(image,GetPixelBlue(image,chroma_pixels),q);
cristy3ed852e2009-09-05 21:47:34 +0000410 chroma_pixels++;
cristyed231572011-07-14 02:18:59 +0000411 q+=GetPixelChannels(image);
cristy3ed852e2009-09-05 21:47:34 +0000412 }
413 if (SyncAuthenticPixels(image,exception) == MagickFalse)
414 break;
415 }
416 resize_image=DestroyImage(resize_image);
417 image->colorspace=YCbCrColorspace;
418 if (interlace == PartitionInterlace)
419 (void) CopyMagickString(image->filename,image_info->filename,
420 MaxTextExtent);
421 if (EOFBlob(image) != MagickFalse)
422 {
423 ThrowFileException(exception,CorruptImageError,"UnexpectedEndOfFile",
424 image->filename);
425 break;
426 }
427 /*
428 Proceed to next image.
429 */
430 if (image_info->number_scenes != 0)
431 if (image->scene >= (image_info->scene+image_info->number_scenes-1))
432 break;
433 if (interlace == NoInterlace)
cristy292ebf12011-02-08 20:21:31 +0000434 count=ReadBlob(image,(size_t) (2*quantum*image->columns),scanline);
cristy3ed852e2009-09-05 21:47:34 +0000435 else
cristy292ebf12011-02-08 20:21:31 +0000436 count=ReadBlob(image,(size_t) quantum*image->columns,scanline);
cristy3ed852e2009-09-05 21:47:34 +0000437 if (count != 0)
438 {
439 /*
440 Allocate next image structure.
441 */
442 AcquireNextImage(image_info,image);
443 if (GetNextImageInList(image) == (Image *) NULL)
444 {
445 image=DestroyImageList(image);
446 return((Image *) NULL);
447 }
448 image=SyncNextImageInList(image);
449 status=SetImageProgress(image,LoadImagesTag,TellBlob(image),
450 GetBlobSize(image));
451 if (status == MagickFalse)
452 break;
453 }
454 } while (count != 0);
455 scanline=(unsigned char *) RelinquishMagickMemory(scanline);
456 (void) CloseBlob(image);
457 return(GetFirstImageInList(image));
458}
459
460/*
461%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
462% %
463% %
464% %
465% R e g i s t e r Y U V I m a g e %
466% %
467% %
468% %
469%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
470%
471% RegisterYUVImage() adds attributes for the YUV image format to
472% the list of supported formats. The attributes include the image format
473% tag, a method to read and/or write the format, whether the format
474% supports the saving of more than one frame to the same file or blob,
475% whether the format supports native in-memory I/O, and a brief
476% description of the format.
477%
478% The format of the RegisterYUVImage method is:
479%
cristybb503372010-05-27 20:51:26 +0000480% size_t RegisterYUVImage(void)
cristy3ed852e2009-09-05 21:47:34 +0000481%
482*/
cristybb503372010-05-27 20:51:26 +0000483ModuleExport size_t RegisterYUVImage(void)
cristy3ed852e2009-09-05 21:47:34 +0000484{
485 MagickInfo
486 *entry;
487
488 entry=SetMagickInfo("YUV");
489 entry->decoder=(DecodeImageHandler *) ReadYUVImage;
490 entry->encoder=(EncodeImageHandler *) WriteYUVImage;
491 entry->adjoin=MagickFalse;
492 entry->raw=MagickTrue;
493 entry->description=ConstantString("CCIR 601 4:1:1 or 4:2:2");
494 entry->module=ConstantString("YUV");
495 (void) RegisterMagickInfo(entry);
496 return(MagickImageCoderSignature);
497}
498
499/*
500%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
501% %
502% %
503% %
504% U n r e g i s t e r Y U V I m a g e %
505% %
506% %
507% %
508%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
509%
510% UnregisterYUVImage() removes format registrations made by the
511% YUV module from the list of supported formats.
512%
513% The format of the UnregisterYUVImage method is:
514%
515% UnregisterYUVImage(void)
516%
517*/
518ModuleExport void UnregisterYUVImage(void)
519{
520 (void) UnregisterMagickInfo("YUV");
521}
522
523/*
524%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
525% %
526% %
527% %
528% W r i t e Y U V I m a g e %
529% %
530% %
531% %
532%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
533%
534% WriteYUVImage() writes an image to a file in the digital YUV
535% (CCIR 601 4:1:1, plane or partition interlaced, or 4:2:2 plane, partition
536% interlaced or noninterlaced) bytes and returns it.
537%
538% The format of the WriteYUVImage method is:
539%
cristy3a37efd2011-08-28 20:31:03 +0000540% MagickBooleanType WriteYUVImage(const ImageInfo *image_info,
541% Image *image,ExceptionInfo *exception)
cristy3ed852e2009-09-05 21:47:34 +0000542%
543% A description of each parameter follows.
544%
545% o image_info: the image info.
546%
547% o image: The image.
548%
cristy3a37efd2011-08-28 20:31:03 +0000549% o exception: return any errors or warnings in this structure.
550%
cristy3ed852e2009-09-05 21:47:34 +0000551*/
cristy3a37efd2011-08-28 20:31:03 +0000552static MagickBooleanType WriteYUVImage(const ImageInfo *image_info,Image *image,
553 ExceptionInfo *exception)
cristy3ed852e2009-09-05 21:47:34 +0000554{
555 Image
556 *chroma_image,
557 *yuv_image;
558
559 InterlaceType
560 interlace;
561
cristy3ed852e2009-09-05 21:47:34 +0000562 MagickBooleanType
563 status;
564
565 MagickOffsetType
566 scene;
567
cristy4c08aed2011-07-01 19:47:50 +0000568 register const Quantum
cristy3ed852e2009-09-05 21:47:34 +0000569 *p,
570 *s;
571
cristybb503372010-05-27 20:51:26 +0000572 register ssize_t
cristy3ed852e2009-09-05 21:47:34 +0000573 x;
574
cristybb503372010-05-27 20:51:26 +0000575 size_t
cristy3ed852e2009-09-05 21:47:34 +0000576 height,
cristy292ebf12011-02-08 20:21:31 +0000577 quantum,
cristy3ed852e2009-09-05 21:47:34 +0000578 width;
579
cristyc6da28e2011-04-28 01:41:35 +0000580 ssize_t
581 horizontal_factor,
582 vertical_factor,
583 y;
584
cristy3ed852e2009-09-05 21:47:34 +0000585 assert(image_info != (const ImageInfo *) NULL);
586 assert(image_info->signature == MagickSignature);
587 assert(image != (Image *) NULL);
588 assert(image->signature == MagickSignature);
589 if (image->debug != MagickFalse)
590 (void) LogMagickEvent(TraceEvent,GetMagickModule(),"%s",image->filename);
cristy292ebf12011-02-08 20:21:31 +0000591 quantum=image->depth <= 8 ? 1 : 2;
cristy3ed852e2009-09-05 21:47:34 +0000592 interlace=image->interlace;
593 horizontal_factor=2;
594 vertical_factor=2;
595 if (image_info->sampling_factor != (char *) NULL)
596 {
597 GeometryInfo
598 geometry_info;
599
600 MagickStatusType
601 flags;
602
603 flags=ParseGeometry(image_info->sampling_factor,&geometry_info);
cristybb503372010-05-27 20:51:26 +0000604 horizontal_factor=(ssize_t) geometry_info.rho;
605 vertical_factor=(ssize_t) geometry_info.sigma;
cristy3ed852e2009-09-05 21:47:34 +0000606 if ((flags & SigmaValue) == 0)
607 vertical_factor=horizontal_factor;
608 if ((horizontal_factor != 1) && (horizontal_factor != 2) &&
609 (vertical_factor != 1) && (vertical_factor != 2))
610 ThrowWriterException(CorruptImageError,"UnexpectedSamplingFactor");
611 }
612 if ((interlace == UndefinedInterlace) ||
613 ((interlace == NoInterlace) && (vertical_factor == 2)))
614 {
615 interlace=NoInterlace; /* CCIR 4:2:2 */
616 if (vertical_factor == 2)
617 interlace=PlaneInterlace; /* CCIR 4:1:1 */
618 }
619 if (interlace != PartitionInterlace)
620 {
621 /*
622 Open output image file.
623 */
cristyedf03fa2011-08-30 12:44:39 +0000624 status=OpenBlob(image_info,image,WriteBinaryBlobMode,exception);
cristy3ed852e2009-09-05 21:47:34 +0000625 if (status == MagickFalse)
626 return(status);
627 }
628 else
629 {
630 AppendImageFormat("Y",image->filename);
cristyedf03fa2011-08-30 12:44:39 +0000631 status=OpenBlob(image_info,image,WriteBinaryBlobMode,exception);
cristy3ed852e2009-09-05 21:47:34 +0000632 if (status == MagickFalse)
633 return(status);
634 }
635 scene=0;
636 do
637 {
638 /*
639 Sample image to an even width and height, if necessary.
640 */
cristy292ebf12011-02-08 20:21:31 +0000641 image->depth=quantum == 1 ? 8 : 16;
cristy3ed852e2009-09-05 21:47:34 +0000642 width=image->columns+(image->columns & (horizontal_factor-1));
643 height=image->rows+(image->rows & (vertical_factor-1));
cristy3a37efd2011-08-28 20:31:03 +0000644 yuv_image=ResizeImage(image,width,height,TriangleFilter,1.0,exception);
cristy3ed852e2009-09-05 21:47:34 +0000645 if (yuv_image == (Image *) NULL)
cristy3a37efd2011-08-28 20:31:03 +0000646 {
647 (void) CloseBlob(image);
648 return(MagickFalse);
649 }
cristy3ed852e2009-09-05 21:47:34 +0000650 (void) TransformImageColorspace(yuv_image,YCbCrColorspace);
651 /*
652 Downsample image.
653 */
654 chroma_image=ResizeImage(image,width/horizontal_factor,
cristy3a37efd2011-08-28 20:31:03 +0000655 height/vertical_factor,TriangleFilter,1.0,exception);
cristy3ed852e2009-09-05 21:47:34 +0000656 if (chroma_image == (Image *) NULL)
cristy3a37efd2011-08-28 20:31:03 +0000657 {
658 (void) CloseBlob(image);
659 return(MagickFalse);
660 }
cristy3ed852e2009-09-05 21:47:34 +0000661 (void) TransformImageColorspace(chroma_image,YCbCrColorspace);
662 if (interlace == NoInterlace)
663 {
664 /*
665 Write noninterlaced YUV.
666 */
cristybb503372010-05-27 20:51:26 +0000667 for (y=0; y < (ssize_t) yuv_image->rows; y++)
cristy3ed852e2009-09-05 21:47:34 +0000668 {
cristy3a37efd2011-08-28 20:31:03 +0000669 p=GetVirtualPixels(yuv_image,0,y,yuv_image->columns,1,exception);
cristy4c08aed2011-07-01 19:47:50 +0000670 if (p == (const Quantum *) NULL)
cristy3ed852e2009-09-05 21:47:34 +0000671 break;
672 s=GetVirtualPixels(chroma_image,0,y,chroma_image->columns,1,
cristy3a37efd2011-08-28 20:31:03 +0000673 exception);
cristy4c08aed2011-07-01 19:47:50 +0000674 if (s == (const Quantum *) NULL)
cristy3ed852e2009-09-05 21:47:34 +0000675 break;
cristybb503372010-05-27 20:51:26 +0000676 for (x=0; x < (ssize_t) yuv_image->columns; x++)
cristy3ed852e2009-09-05 21:47:34 +0000677 {
cristy292ebf12011-02-08 20:21:31 +0000678 if (quantum == 1)
679 {
cristyc6da28e2011-04-28 01:41:35 +0000680 (void) WriteBlobByte(image,ScaleQuantumToChar(
cristy4c08aed2011-07-01 19:47:50 +0000681 GetPixelGreen(yuv_image,s)));
cristy292ebf12011-02-08 20:21:31 +0000682 (void) WriteBlobByte(image,ScaleQuantumToChar(
cristy4c08aed2011-07-01 19:47:50 +0000683 GetPixelRed(yuv_image,p)));
cristyed231572011-07-14 02:18:59 +0000684 p+=GetPixelChannels(yuv_image);
cristyc6da28e2011-04-28 01:41:35 +0000685 (void) WriteBlobByte(image,ScaleQuantumToChar(
cristy4c08aed2011-07-01 19:47:50 +0000686 GetPixelBlue(yuv_image,s)));
cristy292ebf12011-02-08 20:21:31 +0000687 (void) WriteBlobByte(image,ScaleQuantumToChar(
cristy4c08aed2011-07-01 19:47:50 +0000688 GetPixelRed(yuv_image,p)));
cristy292ebf12011-02-08 20:21:31 +0000689 }
690 else
691 {
cristyc6da28e2011-04-28 01:41:35 +0000692 (void) WriteBlobByte(image,ScaleQuantumToChar(
cristy4c08aed2011-07-01 19:47:50 +0000693 GetPixelGreen(yuv_image,s)));
cristy292ebf12011-02-08 20:21:31 +0000694 (void) WriteBlobShort(image,ScaleQuantumToShort(
cristy4c08aed2011-07-01 19:47:50 +0000695 GetPixelRed(yuv_image,p)));
cristyed231572011-07-14 02:18:59 +0000696 p+=GetPixelChannels(yuv_image);
cristyc6da28e2011-04-28 01:41:35 +0000697 (void) WriteBlobByte(image,ScaleQuantumToChar(
cristy4c08aed2011-07-01 19:47:50 +0000698 GetPixelBlue(yuv_image,s)));
cristy292ebf12011-02-08 20:21:31 +0000699 (void) WriteBlobShort(image,ScaleQuantumToShort(
cristy4c08aed2011-07-01 19:47:50 +0000700 GetPixelRed(yuv_image,p)));
cristy292ebf12011-02-08 20:21:31 +0000701 }
cristyed231572011-07-14 02:18:59 +0000702 p+=GetPixelChannels(yuv_image);
cristy3ed852e2009-09-05 21:47:34 +0000703 s++;
704 x++;
705 }
706 if (image->previous == (Image *) NULL)
707 {
cristycee97112010-05-28 00:44:52 +0000708 status=SetImageProgress(image,SaveImageTag,(MagickOffsetType) y,
709 image->rows);
cristy3ed852e2009-09-05 21:47:34 +0000710 if (status == MagickFalse)
711 break;
712 }
713 }
714 yuv_image=DestroyImage(yuv_image);
715 }
716 else
717 {
718 /*
719 Initialize Y channel.
720 */
cristybb503372010-05-27 20:51:26 +0000721 for (y=0; y < (ssize_t) yuv_image->rows; y++)
cristy3ed852e2009-09-05 21:47:34 +0000722 {
cristy3a37efd2011-08-28 20:31:03 +0000723 p=GetVirtualPixels(yuv_image,0,y,yuv_image->columns,1,exception);
cristy4c08aed2011-07-01 19:47:50 +0000724 if (p == (const Quantum *) NULL)
cristy3ed852e2009-09-05 21:47:34 +0000725 break;
cristybb503372010-05-27 20:51:26 +0000726 for (x=0; x < (ssize_t) yuv_image->columns; x++)
cristy3ed852e2009-09-05 21:47:34 +0000727 {
cristy292ebf12011-02-08 20:21:31 +0000728 if (quantum == 1)
729 (void) WriteBlobByte(image,ScaleQuantumToChar(
cristy4c08aed2011-07-01 19:47:50 +0000730 GetPixelRed(yuv_image,p)));
cristy292ebf12011-02-08 20:21:31 +0000731 else
732 (void) WriteBlobShort(image,ScaleQuantumToShort(
cristy4c08aed2011-07-01 19:47:50 +0000733 GetPixelRed(yuv_image,p)));
cristyed231572011-07-14 02:18:59 +0000734 p+=GetPixelChannels(yuv_image);
cristy3ed852e2009-09-05 21:47:34 +0000735 }
736 if (image->previous == (Image *) NULL)
737 {
cristycee97112010-05-28 00:44:52 +0000738 status=SetImageProgress(image,SaveImageTag,(MagickOffsetType) y,
739 image->rows);
cristy3ed852e2009-09-05 21:47:34 +0000740 if (status == MagickFalse)
741 break;
742 }
743 }
744 yuv_image=DestroyImage(yuv_image);
745 if (image->previous == (Image *) NULL)
746 {
747 status=SetImageProgress(image,SaveImageTag,1,3);
748 if (status == MagickFalse)
749 break;
750 }
751 /*
752 Initialize U channel.
753 */
754 if (interlace == PartitionInterlace)
755 {
756 (void) CloseBlob(image);
757 AppendImageFormat("U",image->filename);
cristyedf03fa2011-08-30 12:44:39 +0000758 status=OpenBlob(image_info,image,WriteBinaryBlobMode,exception);
cristy3ed852e2009-09-05 21:47:34 +0000759 if (status == MagickFalse)
760 return(status);
761 }
cristybb503372010-05-27 20:51:26 +0000762 for (y=0; y < (ssize_t) chroma_image->rows; y++)
cristy3ed852e2009-09-05 21:47:34 +0000763 {
764 p=GetVirtualPixels(chroma_image,0,y,chroma_image->columns,1,
cristy3a37efd2011-08-28 20:31:03 +0000765 exception);
cristy4c08aed2011-07-01 19:47:50 +0000766 if (p == (const Quantum *) NULL)
cristy3ed852e2009-09-05 21:47:34 +0000767 break;
cristybb503372010-05-27 20:51:26 +0000768 for (x=0; x < (ssize_t) chroma_image->columns; x++)
cristy3ed852e2009-09-05 21:47:34 +0000769 {
cristy292ebf12011-02-08 20:21:31 +0000770 if (quantum == 1)
771 (void) WriteBlobByte(image,ScaleQuantumToChar(
cristy4c08aed2011-07-01 19:47:50 +0000772 GetPixelGreen(chroma_image,p)));
cristy292ebf12011-02-08 20:21:31 +0000773 else
774 (void) WriteBlobShort(image,ScaleQuantumToShort(
cristy4c08aed2011-07-01 19:47:50 +0000775 GetPixelGreen(chroma_image,p)));
cristyed231572011-07-14 02:18:59 +0000776 p+=GetPixelChannels(chroma_image);
cristy3ed852e2009-09-05 21:47:34 +0000777 }
778 }
779 if (image->previous == (Image *) NULL)
780 {
781 status=SetImageProgress(image,SaveImageTag,2,3);
782 if (status == MagickFalse)
783 break;
784 }
785 /*
786 Initialize V channel.
787 */
788 if (interlace == PartitionInterlace)
789 {
790 (void) CloseBlob(image);
791 AppendImageFormat("V",image->filename);
cristyedf03fa2011-08-30 12:44:39 +0000792 status=OpenBlob(image_info,image,WriteBinaryBlobMode,exception);
cristy3ed852e2009-09-05 21:47:34 +0000793 if (status == MagickFalse)
794 return(status);
795 }
cristybb503372010-05-27 20:51:26 +0000796 for (y=0; y < (ssize_t) chroma_image->rows; y++)
cristy3ed852e2009-09-05 21:47:34 +0000797 {
798 p=GetVirtualPixels(chroma_image,0,y,chroma_image->columns,1,
cristy3a37efd2011-08-28 20:31:03 +0000799 exception);
cristy4c08aed2011-07-01 19:47:50 +0000800 if (p == (const Quantum *) NULL)
cristy3ed852e2009-09-05 21:47:34 +0000801 break;
cristybb503372010-05-27 20:51:26 +0000802 for (x=0; x < (ssize_t) chroma_image->columns; x++)
cristy3ed852e2009-09-05 21:47:34 +0000803 {
cristy292ebf12011-02-08 20:21:31 +0000804 if (quantum == 1)
805 (void) WriteBlobByte(image,ScaleQuantumToChar(
cristy4c08aed2011-07-01 19:47:50 +0000806 GetPixelBlue(chroma_image,p)));
cristy292ebf12011-02-08 20:21:31 +0000807 else
808 (void) WriteBlobShort(image,ScaleQuantumToShort(
cristy4c08aed2011-07-01 19:47:50 +0000809 GetPixelBlue(chroma_image,p)));
cristyed231572011-07-14 02:18:59 +0000810 p+=GetPixelChannels(chroma_image);
cristy3ed852e2009-09-05 21:47:34 +0000811 }
812 }
813 if (image->previous == (Image *) NULL)
814 {
815 status=SetImageProgress(image,SaveImageTag,2,3);
816 if (status == MagickFalse)
817 break;
818 }
819 }
820 chroma_image=DestroyImage(chroma_image);
821 if (interlace == PartitionInterlace)
822 (void) CopyMagickString(image->filename,image_info->filename,
823 MaxTextExtent);
824 if (GetNextImageInList(image) == (Image *) NULL)
825 break;
826 image=SyncNextImageInList(image);
827 status=SetImageProgress(image,SaveImagesTag,scene++,
828 GetImageListLength(image));
829 if (status == MagickFalse)
830 break;
831 } while (image_info->adjoin != MagickFalse);
832 (void) CloseBlob(image);
833 return(MagickTrue);
834}