Changeset 1170

Timestamp:

10/21/09 23:18:10 (11 months ago)

Author:

astrange

Message:

Set video colorspace info on import via 'nclc'.

Fixes the colors for every video ever. Closes #410

Location:

trunk

Files:

• MatroskaImportPrivate.cpp (modified) (1 diff)
• ff_private.c (modified) (3 diffs)
• ff_private.h (modified) (1 diff)

trunk/MatroskaImportPrivate.cpp

@@ -472 +472 @@
         
         set_track_clean_aperture_ext(imgDesc, width, height, IntToFixed(uint32(pxl_width)), IntToFixed(uint32(pxl_height)));
+        set_track_colorspace_ext(imgDesc, width, height);
         mkvTrack.desc = (SampleDescriptionHandle) imgDesc;
         

trunk/ff_private.c

@@ -587 +587 @@
 }
 
+// Create the 'pasp' atom for video tracks. No guesswork required.
+// References http://www.uwasa.fi/~f76998/video/conversion/
 void set_track_clean_aperture_ext(ImageDescriptionHandle imgDesc, Fixed displayW, Fixed displayH, Fixed pixelW, Fixed pixelH)
 {
@@ -607 +609 @@
         
         DisposeHandle((Handle)pasp);
+}
+
+// Create the 'nclc' atom for video tracks. Guessed entirely from image size following ffdshow.
+// FIXME read H.264 VUI/MPEG2 etc and especially read chroma positioning information.
+// this needs the parsers working
+// References: http://developer.apple.com/quicktime/icefloe/dispatch019.html
+// http://www.mir.com/DMG/chroma.html
+void set_track_colorspace_ext(ImageDescriptionHandle imgDescHandle, Fixed displayW, Fixed displayH)
+{
+        ImageDescription *imgDesc = *imgDescHandle;
+        Boolean isHd, isPAL; // otherwise NTSC
+        AVRational palRatio = (AVRational){5, 4}, displayRatio = (AVRational){displayW, displayH};
+        int colorPrimaries, transferFunction, yuvMatrix;
+        
+        isHd  = imgDesc->height >  576;
+        isPAL = imgDesc->height == 576 || av_cmp_q(palRatio, displayRatio) == 0;
+        
+        NCLCColorInfoImageDescriptionExtension **nclc = (NCLCColorInfoImageDescriptionExtension**)NewHandle(sizeof(NCLCColorInfoImageDescriptionExtension));
+                
+        if (isHd) {
+                colorPrimaries = kQTPrimaries_ITU_R709_2;
+                transferFunction = kQTTransferFunction_ITU_R709_2;
+                yuvMatrix = kQTMatrix_ITU_R_709_2;
+        } else if (isPAL) {
+                colorPrimaries = kQTPrimaries_EBU_3213;
+                transferFunction = kQTTransferFunction_ITU_R709_2;
+                yuvMatrix = kQTMatrix_ITU_R_601_4;
+        } else {
+                colorPrimaries = kQTPrimaries_SMPTE_C;
+                transferFunction = kQTTransferFunction_ITU_R709_2;
+                yuvMatrix = kQTMatrix_ITU_R_601_4;
+        }
+        
+        **nclc = (NCLCColorInfoImageDescriptionExtension){EndianU32_NtoB(kVideoColorInfoImageDescriptionExtensionType),
+                                                                                                          EndianU16_NtoB(colorPrimaries),
+                                                                                                          EndianU16_NtoB(transferFunction),
+                                                                                                          EndianU16_NtoB(yuvMatrix)};
+        
+        AddImageDescriptionExtension(imgDescHandle, (Handle)nclc, kColorInfoImageDescriptionExtension);
+        
+        DisposeHandle((Handle)nclc);
 }
 
@@ -647 +690 @@
                         initialize_video_map(&map[j], track, dataRef, dataRefType, storage->firstFrames + j);
                         set_track_clean_aperture_ext((ImageDescriptionHandle)map[j].sampleHdl, width, height, IntToFixed(st->codec->width), IntToFixed(st->codec->height));
+                        set_track_colorspace_ext((ImageDescriptionHandle)map[j].sampleHdl, width, height);
                 } else if (st->codec->codec_type == CODEC_TYPE_AUDIO) {
                         if (st->codec->sample_rate > 0) {

trunk/ff_private.h

@@ -113 +113 @@
 Handle create_strf_ext(AVCodecContext *codec);
 void set_track_clean_aperture_ext(ImageDescriptionHandle imgDesc, Fixed displayW, Fixed displayH, Fixed pixelW, Fixed pixelH);
+void set_track_colorspace_ext(ImageDescriptionHandle imgDescHandle, Fixed displayW, Fixed displayH);
 
 uint8_t *write_int32(uint8_t *target, int32_t data);