コード例 #1
0
        /// <summary>
        /// AnnexB formatted h264 bitstream
        /// </summary>
        /// <param name="streamInfo"></param>
        /// <returns></returns>
        public static AMMediaType H264_AnnexB(InputstreamInfo streamInfo)
        {
            int width  = (int)streamInfo.Width;
            int height = (int)streamInfo.Height;

            if (streamInfo.ExtraData.Length > 0)
            {
                var codecData = new H264CodecData(streamInfo.ExtraData);

                SPSUnit spsUnit = new SPSUnit(codecData.SPS);
                width  = spsUnit.Width();
                height = spsUnit.Height();
            }

            VideoInfoHeader2 vi = new VideoInfoHeader2();

            vi.SrcRect.right     = width;
            vi.SrcRect.bottom    = height;
            vi.TargetRect.right  = width;
            vi.TargetRect.bottom = height;

            int hcf = HCF(width, height);

            vi.PictAspectRatioX = width / hcf;
            vi.PictAspectRatioY = height / hcf;

            vi.BmiHeader.Width       = width;
            vi.BmiHeader.Height      = height;
            vi.BmiHeader.Planes      = 1;
            vi.BmiHeader.Compression = FOURCC_H264;

            AMMediaType amt = new AMMediaType();

            amt.majorType           = MediaType.Video;
            amt.subType             = MediaSubType.H264;
            amt.temporalCompression = true;
            amt.fixedSizeSamples    = false;
            amt.sampleSize          = 1;
            amt.SetFormat(vi);
            return(amt);
        }
コード例 #2
0
        /// <summary>
        /// AVC1 formatted H264 bitstream
        /// </summary>
        /// <param name="streamInfo"></param>
        /// <returns></returns>
        public static AMMediaType H264_AVC1(InputstreamInfo streamInfo)
        {
            H264CodecData  codecData = null;
            Mpeg2VideoInfo vi        = new Mpeg2VideoInfo();

            byte[] extraData = new byte[0];
            int    width     = (int)streamInfo.Width;
            int    height    = (int)streamInfo.Height;

            if (streamInfo.ExtraData.Length > 0)
            {
                codecData = new H264CodecData(streamInfo.ExtraData);

                SPSUnit spsUnit = new SPSUnit(codecData.SPS);
                width  = spsUnit.Width();
                height = spsUnit.Height();
            }

            vi.hdr.SrcRect.right     = width;
            vi.hdr.SrcRect.bottom    = height;
            vi.hdr.TargetRect.right  = width;
            vi.hdr.TargetRect.bottom = height;

            int hcf = HCF(width, height);

            vi.hdr.PictAspectRatioX = width / hcf;
            vi.hdr.PictAspectRatioY = height / hcf;

            vi.hdr.BmiHeader.Width  = width;
            vi.hdr.BmiHeader.Height = height;

            vi.hdr.BmiHeader.Planes      = 1;
            vi.hdr.BmiHeader.Compression = FOURCC_AVC1;
            vi.hdr.BmiHeader.BitCount    = 24;

            if (codecData != null)
            {
                vi.dwProfile = (uint)codecData.Profile;
                vi.dwLevel   = (uint)codecData.Level;
                vi.dwFlags   = (uint)codecData.NALSizeMinusOne + 1;

                extraData = NaluParser.CreateAVC1ParameterSet(codecData.SPS, codecData.PPS, 2);
            }
            else
            {
                // Example: avc1.4D401F -> Main Level 3.1
                // Profile     Value
                // Baseline    42E0
                // Main        4D40
                // High        6400
                // Extended    58A0

                // Level       Hex Value
                // 3.0         1E
                // 3.1         1F
                // 4.1         29
                // 5.1         33
                string codecInfo = streamInfo.CodecInternalName.Split('.').Last();
                if (codecInfo.Length == 6)
                {
                    int codecNum;
                    if (int.TryParse(codecInfo.Substring(0, 2), NumberStyles.HexNumber, CultureInfo.InvariantCulture, out codecNum))
                    {
                        vi.dwProfile = (uint)codecNum;
                    }
                    if (int.TryParse(codecInfo.Substring(4, 2), NumberStyles.HexNumber, CultureInfo.InvariantCulture, out codecNum))
                    {
                        vi.dwLevel = (uint)codecNum;
                    }
                }
            }

            vi.cbSequenceHeader = (uint)extraData.Length;

            AMMediaType amt = new AMMediaType();

            amt.majorType           = MediaType.Video;
            amt.subType             = MEDIASUBTYPE_AVC1;
            amt.temporalCompression = true;
            amt.fixedSizeSamples    = false;
            amt.sampleSize          = 1;
            SetFormat(vi, extraData, amt);
            return(amt);
        }