コード例 #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;
    }