2 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
13 % Read/Write Raw CCIR 601 4:1:1 or 4:2:2 Image Format %
20 % Copyright 1999-2012 ImageMagick Studio LLC, a non-profit organization %
21 % dedicated to making software imaging solutions freely available. %
23 % You may not use this file except in compliance with the License. You may %
24 % obtain a copy of the License at %
26 % http://www.imagemagick.org/script/license.php %
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. %
34 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
42 #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"
69 static MagickBooleanType
70 WriteYUVImage(const ImageInfo *,Image *,ExceptionInfo *);
73 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
77 % R e a d Y U V I m a g e %
81 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
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.
88 % The format of the ReadYUVImage method is:
90 % Image *ReadYUVImage(const ImageInfo *image_info,ExceptionInfo *exception)
92 % A description of each parameter follows:
94 % o image_info: the image info.
96 % o exception: return any errors or warnings in this structure.
99 static Image *ReadYUVImage(const ImageInfo *image_info,ExceptionInfo *exception)
112 register const Quantum
121 register unsigned char
135 Allocate image structure.
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,exception);
145 if ((image->columns == 0) || (image->rows == 0))
146 ThrowReaderException(OptionError,"MustSpecifyImageSize");
147 quantum=image->depth <= 8 ? 1 : 2;
148 interlace=image_info->interlace;
151 if (image_info->sampling_factor != (char *) NULL)
159 flags=ParseGeometry(image_info->sampling_factor,&geometry_info);
160 horizontal_factor=(ssize_t) geometry_info.rho;
161 vertical_factor=(ssize_t) geometry_info.sigma;
162 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");
168 if ((interlace == UndefinedInterlace) ||
169 ((interlace == NoInterlace) && (vertical_factor == 2)))
171 interlace=NoInterlace; /* CCIR 4:2:2 */
172 if (vertical_factor == 2)
173 interlace=PlaneInterlace; /* CCIR 4:1:1 */
175 if (interlace != PartitionInterlace)
180 status=OpenBlob(image_info,image,ReadBinaryBlobMode,exception);
181 if (status == MagickFalse)
183 image=DestroyImageList(image);
184 return((Image *) NULL);
186 if (DiscardBlobBytes(image,image->offset) == MagickFalse)
187 ThrowFileException(exception,CorruptImageError,"UnexpectedEndOfFile",
191 Allocate memory for a scanline.
193 if (interlace == NoInterlace)
194 scanline=(unsigned char *) AcquireQuantumMemory((size_t) 2UL*
195 image->columns+2UL,quantum*sizeof(*scanline));
197 scanline=(unsigned char *) AcquireQuantumMemory((size_t) image->columns,
198 quantum*sizeof(*scanline));
199 if (scanline == (unsigned char *) NULL)
200 ThrowReaderException(ResourceLimitError,"MemoryAllocationFailed");
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");
208 Convert raster image to pixel packets.
210 if ((image_info->ping != MagickFalse) && (image_info->number_scenes != 0))
211 if (image->scene >= (image_info->scene+image_info->number_scenes-1))
213 if (interlace == PartitionInterlace)
215 AppendImageFormat("Y",image->filename);
216 status=OpenBlob(image_info,image,ReadBinaryBlobMode,exception);
217 if (status == MagickFalse)
219 image=DestroyImageList(image);
220 return((Image *) NULL);
223 for (y=0; y < (ssize_t) image->rows; y++)
228 if (interlace == NoInterlace)
230 if ((y > 0) || (GetPreviousImageInList(image) == (Image *) NULL))
231 count=ReadBlob(image,(size_t) (2*quantum*image->columns),scanline);
233 q=QueueAuthenticPixels(image,0,y,image->columns,1,exception);
234 if (q == (Quantum *) NULL)
236 chroma_pixels=QueueAuthenticPixels(chroma_image,0,y,
237 chroma_image->columns,1,exception);
238 if (chroma_pixels == (Quantum *) NULL)
240 for (x=0; x < (ssize_t) image->columns; x+=2)
242 SetPixelRed(image,0,chroma_pixels);
244 SetPixelGreen(image,ScaleCharToQuantum(*p++),chroma_pixels);
247 SetPixelGreen(image,ScaleShortToQuantum(((*p) << 8) | *(p+1)),
252 SetPixelRed(image,ScaleCharToQuantum(*p++),q);
255 SetPixelRed(image,ScaleShortToQuantum(((*p) << 8) | *(p+1)),q);
258 SetPixelGreen(image,0,q);
259 SetPixelBlue(image,0,q);
260 q+=GetPixelChannels(image);
261 SetPixelGreen(image,0,q);
262 SetPixelBlue(image,0,q);
264 SetPixelBlue(image,ScaleCharToQuantum(*p++),chroma_pixels);
267 SetPixelBlue(image,ScaleShortToQuantum(((*p) << 8) | *(p+1)),
272 SetPixelRed(image,ScaleCharToQuantum(*p++),q);
275 SetPixelRed(image,ScaleShortToQuantum(((*p) << 8) | *(p+1)),q);
279 q+=GetPixelChannels(image);
284 if ((y > 0) || (GetPreviousImageInList(image) == (Image *) NULL))
285 count=ReadBlob(image,(size_t) quantum*image->columns,scanline);
287 q=QueueAuthenticPixels(image,0,y,image->columns,1,exception);
288 if (q == (Quantum *) NULL)
290 for (x=0; x < (ssize_t) image->columns; x++)
293 SetPixelRed(image,ScaleCharToQuantum(*p++),q);
296 SetPixelRed(image,ScaleShortToQuantum(((*p) << 8) | *(p+1)),q);
299 SetPixelGreen(image,0,q);
300 SetPixelBlue(image,0,q);
301 q+=GetPixelChannels(image);
304 if (SyncAuthenticPixels(image,exception) == MagickFalse)
306 if (interlace == NoInterlace)
307 if (SyncAuthenticPixels(chroma_image,exception) == MagickFalse)
309 if (image->previous == (Image *) NULL)
311 status=SetImageProgress(image,LoadImageTag,(MagickOffsetType) y,
313 if (status == MagickFalse)
317 if (interlace == PartitionInterlace)
319 (void) CloseBlob(image);
320 AppendImageFormat("U",image->filename);
321 status=OpenBlob(image_info,image,ReadBinaryBlobMode,exception);
322 if (status == MagickFalse)
324 image=DestroyImageList(image);
325 return((Image *) NULL);
328 if (interlace != NoInterlace)
330 for (y=0; y < (ssize_t) chroma_image->rows; y++)
332 count=ReadBlob(image,(size_t) quantum*chroma_image->columns,scanline);
334 q=QueueAuthenticPixels(chroma_image,0,y,chroma_image->columns,1,
336 if (q == (Quantum *) NULL)
338 for (x=0; x < (ssize_t) chroma_image->columns; x++)
340 SetPixelRed(chroma_image,0,q);
342 SetPixelGreen(chroma_image,ScaleCharToQuantum(*p++),q);
345 SetPixelGreen(chroma_image,ScaleShortToQuantum(((*p) << 8) |
349 SetPixelBlue(chroma_image,0,q);
350 q+=GetPixelChannels(chroma_image);
352 if (SyncAuthenticPixels(chroma_image,exception) == MagickFalse)
355 if (interlace == PartitionInterlace)
357 (void) CloseBlob(image);
358 AppendImageFormat("V",image->filename);
359 status=OpenBlob(image_info,image,ReadBinaryBlobMode,exception);
360 if (status == MagickFalse)
362 image=DestroyImageList(image);
363 return((Image *) NULL);
366 for (y=0; y < (ssize_t) chroma_image->rows; y++)
368 count=ReadBlob(image,(size_t) quantum*chroma_image->columns,scanline);
370 q=GetAuthenticPixels(chroma_image,0,y,chroma_image->columns,1,
372 if (q == (Quantum *) NULL)
374 for (x=0; x < (ssize_t) chroma_image->columns; x++)
377 SetPixelBlue(chroma_image,ScaleCharToQuantum(*p++),q);
380 SetPixelBlue(chroma_image,ScaleShortToQuantum(((*p) << 8) |
384 q+=GetPixelChannels(chroma_image);
386 if (SyncAuthenticPixels(chroma_image,exception) == MagickFalse)
393 resize_image=ResizeImage(chroma_image,image->columns,image->rows,
394 TriangleFilter,exception);
395 chroma_image=DestroyImage(chroma_image);
396 if (resize_image == (Image *) NULL)
397 ThrowReaderException(ResourceLimitError,"MemoryAllocationFailed");
398 for (y=0; y < (ssize_t) image->rows; y++)
400 q=GetAuthenticPixels(image,0,y,image->columns,1,exception);
401 chroma_pixels=GetVirtualPixels(resize_image,0,y,resize_image->columns,1,
403 if ((q == (Quantum *) NULL) ||
404 (chroma_pixels == (const Quantum *) NULL))
406 for (x=0; x < (ssize_t) image->columns; x++)
408 SetPixelGreen(image,GetPixelGreen(image,chroma_pixels),q);
409 SetPixelBlue(image,GetPixelBlue(image,chroma_pixels),q);
411 q+=GetPixelChannels(image);
413 if (SyncAuthenticPixels(image,exception) == MagickFalse)
416 resize_image=DestroyImage(resize_image);
417 image->colorspace=YCbCrColorspace;
418 if (interlace == PartitionInterlace)
419 (void) CopyMagickString(image->filename,image_info->filename,
421 if (EOFBlob(image) != MagickFalse)
423 ThrowFileException(exception,CorruptImageError,"UnexpectedEndOfFile",
428 Proceed to next image.
430 if (image_info->number_scenes != 0)
431 if (image->scene >= (image_info->scene+image_info->number_scenes-1))
433 if (interlace == NoInterlace)
434 count=ReadBlob(image,(size_t) (2*quantum*image->columns),scanline);
436 count=ReadBlob(image,(size_t) quantum*image->columns,scanline);
440 Allocate next image structure.
442 AcquireNextImage(image_info,image,exception);
443 if (GetNextImageInList(image) == (Image *) NULL)
445 image=DestroyImageList(image);
446 return((Image *) NULL);
448 image=SyncNextImageInList(image);
449 status=SetImageProgress(image,LoadImagesTag,TellBlob(image),
451 if (status == MagickFalse)
454 } while (count != 0);
455 scanline=(unsigned char *) RelinquishMagickMemory(scanline);
456 (void) CloseBlob(image);
457 return(GetFirstImageInList(image));
461 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
465 % R e g i s t e r Y U V I m a g e %
469 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
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.
478 % The format of the RegisterYUVImage method is:
480 % size_t RegisterYUVImage(void)
483 ModuleExport size_t RegisterYUVImage(void)
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);
500 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
504 % U n r e g i s t e r Y U V I m a g e %
508 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
510 % UnregisterYUVImage() removes format registrations made by the
511 % YUV module from the list of supported formats.
513 % The format of the UnregisterYUVImage method is:
515 % UnregisterYUVImage(void)
518 ModuleExport void UnregisterYUVImage(void)
520 (void) UnregisterMagickInfo("YUV");
524 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
528 % W r i t e Y U V I m a g e %
532 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
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.
538 % The format of the WriteYUVImage method is:
540 % MagickBooleanType WriteYUVImage(const ImageInfo *image_info,
541 % Image *image,ExceptionInfo *exception)
543 % A description of each parameter follows.
545 % o image_info: the image info.
547 % o image: The image.
549 % o exception: return any errors or warnings in this structure.
552 static MagickBooleanType WriteYUVImage(const ImageInfo *image_info,Image *image,
553 ExceptionInfo *exception)
568 register const Quantum
585 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);
591 quantum=(size_t) (image->depth <= 8 ? 1 : 2);
592 interlace=image->interlace;
595 if (image_info->sampling_factor != (char *) NULL)
603 flags=ParseGeometry(image_info->sampling_factor,&geometry_info);
604 horizontal_factor=(ssize_t) geometry_info.rho;
605 vertical_factor=(ssize_t) geometry_info.sigma;
606 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");
612 if ((interlace == UndefinedInterlace) ||
613 ((interlace == NoInterlace) && (vertical_factor == 2)))
615 interlace=NoInterlace; /* CCIR 4:2:2 */
616 if (vertical_factor == 2)
617 interlace=PlaneInterlace; /* CCIR 4:1:1 */
619 if (interlace != PartitionInterlace)
622 Open output image file.
624 status=OpenBlob(image_info,image,WriteBinaryBlobMode,exception);
625 if (status == MagickFalse)
630 AppendImageFormat("Y",image->filename);
631 status=OpenBlob(image_info,image,WriteBinaryBlobMode,exception);
632 if (status == MagickFalse)
639 Sample image to an even width and height, if necessary.
641 image->depth=(size_t) (quantum == 1 ? 8 : 16);
642 width=image->columns+(image->columns & (horizontal_factor-1));
643 height=image->rows+(image->rows & (vertical_factor-1));
644 yuv_image=ResizeImage(image,width,height,TriangleFilter,exception);
645 if (yuv_image == (Image *) NULL)
647 (void) CloseBlob(image);
650 (void) TransformImageColorspace(yuv_image,YCbCrColorspace,exception);
654 chroma_image=ResizeImage(image,width/horizontal_factor,
655 height/vertical_factor,TriangleFilter,exception);
656 if (chroma_image == (Image *) NULL)
658 (void) CloseBlob(image);
661 (void) TransformImageColorspace(chroma_image,YCbCrColorspace,exception);
662 if (interlace == NoInterlace)
665 Write noninterlaced YUV.
667 for (y=0; y < (ssize_t) yuv_image->rows; y++)
669 p=GetVirtualPixels(yuv_image,0,y,yuv_image->columns,1,exception);
670 if (p == (const Quantum *) NULL)
672 s=GetVirtualPixels(chroma_image,0,y,chroma_image->columns,1,
674 if (s == (const Quantum *) NULL)
676 for (x=0; x < (ssize_t) yuv_image->columns; x++)
680 (void) WriteBlobByte(image,ScaleQuantumToChar(
681 GetPixelGreen(yuv_image,s)));
682 (void) WriteBlobByte(image,ScaleQuantumToChar(
683 GetPixelRed(yuv_image,p)));
684 p+=GetPixelChannels(yuv_image);
685 (void) WriteBlobByte(image,ScaleQuantumToChar(
686 GetPixelBlue(yuv_image,s)));
687 (void) WriteBlobByte(image,ScaleQuantumToChar(
688 GetPixelRed(yuv_image,p)));
692 (void) WriteBlobByte(image,ScaleQuantumToChar(
693 GetPixelGreen(yuv_image,s)));
694 (void) WriteBlobShort(image,ScaleQuantumToShort(
695 GetPixelRed(yuv_image,p)));
696 p+=GetPixelChannels(yuv_image);
697 (void) WriteBlobByte(image,ScaleQuantumToChar(
698 GetPixelBlue(yuv_image,s)));
699 (void) WriteBlobShort(image,ScaleQuantumToShort(
700 GetPixelRed(yuv_image,p)));
702 p+=GetPixelChannels(yuv_image);
706 if (image->previous == (Image *) NULL)
708 status=SetImageProgress(image,SaveImageTag,(MagickOffsetType) y,
710 if (status == MagickFalse)
714 yuv_image=DestroyImage(yuv_image);
719 Initialize Y channel.
721 for (y=0; y < (ssize_t) yuv_image->rows; y++)
723 p=GetVirtualPixels(yuv_image,0,y,yuv_image->columns,1,exception);
724 if (p == (const Quantum *) NULL)
726 for (x=0; x < (ssize_t) yuv_image->columns; x++)
729 (void) WriteBlobByte(image,ScaleQuantumToChar(
730 GetPixelRed(yuv_image,p)));
732 (void) WriteBlobShort(image,ScaleQuantumToShort(
733 GetPixelRed(yuv_image,p)));
734 p+=GetPixelChannels(yuv_image);
736 if (image->previous == (Image *) NULL)
738 status=SetImageProgress(image,SaveImageTag,(MagickOffsetType) y,
740 if (status == MagickFalse)
744 yuv_image=DestroyImage(yuv_image);
745 if (image->previous == (Image *) NULL)
747 status=SetImageProgress(image,SaveImageTag,1,3);
748 if (status == MagickFalse)
752 Initialize U channel.
754 if (interlace == PartitionInterlace)
756 (void) CloseBlob(image);
757 AppendImageFormat("U",image->filename);
758 status=OpenBlob(image_info,image,WriteBinaryBlobMode,exception);
759 if (status == MagickFalse)
762 for (y=0; y < (ssize_t) chroma_image->rows; y++)
764 p=GetVirtualPixels(chroma_image,0,y,chroma_image->columns,1,
766 if (p == (const Quantum *) NULL)
768 for (x=0; x < (ssize_t) chroma_image->columns; x++)
771 (void) WriteBlobByte(image,ScaleQuantumToChar(
772 GetPixelGreen(chroma_image,p)));
774 (void) WriteBlobShort(image,ScaleQuantumToShort(
775 GetPixelGreen(chroma_image,p)));
776 p+=GetPixelChannels(chroma_image);
779 if (image->previous == (Image *) NULL)
781 status=SetImageProgress(image,SaveImageTag,2,3);
782 if (status == MagickFalse)
786 Initialize V channel.
788 if (interlace == PartitionInterlace)
790 (void) CloseBlob(image);
791 AppendImageFormat("V",image->filename);
792 status=OpenBlob(image_info,image,WriteBinaryBlobMode,exception);
793 if (status == MagickFalse)
796 for (y=0; y < (ssize_t) chroma_image->rows; y++)
798 p=GetVirtualPixels(chroma_image,0,y,chroma_image->columns,1,
800 if (p == (const Quantum *) NULL)
802 for (x=0; x < (ssize_t) chroma_image->columns; x++)
805 (void) WriteBlobByte(image,ScaleQuantumToChar(
806 GetPixelBlue(chroma_image,p)));
808 (void) WriteBlobShort(image,ScaleQuantumToShort(
809 GetPixelBlue(chroma_image,p)));
810 p+=GetPixelChannels(chroma_image);
813 if (image->previous == (Image *) NULL)
815 status=SetImageProgress(image,SaveImageTag,2,3);
816 if (status == MagickFalse)
820 chroma_image=DestroyImage(chroma_image);
821 if (interlace == PartitionInterlace)
822 (void) CopyMagickString(image->filename,image_info->filename,
824 if (GetNextImageInList(image) == (Image *) NULL)
826 image=SyncNextImageInList(image);
827 status=SetImageProgress(image,SaveImagesTag,scene++,
828 GetImageListLength(image));
829 if (status == MagickFalse)
831 } while (image_info->adjoin != MagickFalse);
832 (void) CloseBlob(image);