IltmmWMProfile::AddStream Example for C++

#include "amvideo.h" 
#include "Mmreg.h" 
HRESULT AddAudioStream(IltmmWMProfileManager* mgr, IltmmWMProfile* profile, 
UINT nStreamNum, UINT nFormatTag, ULONG nBitRate, 
ULONG nSamplesPerSec, UINT nChannels, BOOL bVideo) 
{ 
   long cCodecs; 
   VARIANT varSave; 
   VariantInit(&varSave); 
   // get MEDIATYPE_Audio codec count 
   BSTR bstr = ::SysAllocString(ltmmMEDIATYPE_Audio); 
   HRESULT hr = mgr->GetCodecInfoCount(bstr, &cCodecs); 
   ::SysFreeString(bstr); 
   if (FAILED(hr)) 
      return hr; 
   // search for matching codec 
   for (long i = 0; i < cCodecs; i++) 
   { 
      long cFormats; 
      // get MEDIATYPE_Audio codec format count 
      BSTR bstr = ::SysAllocString(ltmmMEDIATYPE_Audio); 
      HRESULT hr = mgr->GetCodecFormatCount(bstr, i, &cFormats); 
      ::SysFreeString(bstr); 
      if(FAILED(hr)) 
         break; 
      for (long j = 0; j < cFormats; j++) 
      { 
         // get MEDIATYPE_Audio stream config interface 
         IltmmWMStreamConfig* format; 
         bstr = ::SysAllocString(ltmmMEDIATYPE_Audio); 
         hr = mgr->GetCodecFormat(bstr, i, j, &format); 
         ::SysFreeString(bstr); 
         if (SUCCEEDED(hr)) 
         { 
            // get this format's media type 
            IltmmMediaTypeDisp* pmt; 
            hr = format->GetMediaType(&pmt); 
            if (SUCCEEDED(hr)) 
            { 
               hr = pmt->get_FormatType(&bstr); 
               if (SUCCEEDED(hr)) 
               { 
                  // only check FORMAT_WaveFormatEx formats 
                  if (!_wcsicmp(bstr, ltmmFORMAT_WaveFormatEx)) 
                  { 
                     VARIANT var; 
                     VariantInit(&var); 
                     pmt->get_Format(&var); 
                     WAVEFORMATEX *pWfx; 
                     SafeArrayAccessData(V_ARRAY(&var), (void**) &pWfx); 
                     // does this format look like a good candidate 
                     long nDiff = (long) (pWfx->nAvgBytesPerSec * 8 - nBitRate); 
                     if (nDiff < 250 && 
                        nDiff > -250 && 
                     pWfx->nSamplesPerSec == nSamplesPerSec && 
                     pWfx->nChannels == nChannels && 
                     pWfx->wFormatTag == nFormatTag) 
                     { 
                        if (V_VT(&varSave) != VT_EMPTY) 
                        { 
                           WAVEFORMATEX *pSaveWFX; 
                           SafeArrayAccessData(V_ARRAY(&varSave), (void**) &pSaveWFX); 
                           if (bVideo) 
                           { 
                              // For audio/video configurations, we want to 
                              // find the smaller nBlockAlign. In this case, 
                              // the nBlockAlign is larger, so we want to 
                              // use the old format. 
                              if (pWfx->nBlockAlign <= pSaveWFX->nBlockAlign) 
                              { 
                                 SafeArrayUnaccessData(V_ARRAY(&varSave)); 
                                 VariantClear(&varSave); 
                                 VariantCopy(&varSave, &var); 
                              } 
                           } 
                           else 
                           { 
                              // otherwise, we want the larger nBlockAlign 
                              if (pWfx->nBlockAlign >= pSaveWFX->nBlockAlign) 
                              { 
                                 SafeArrayUnaccessData(V_ARRAY(&varSave)); 
                                 VariantClear(&varSave); 
                                 VariantCopy(&varSave, &var); 
                              } 
                           } 
                        } 
                        else 
                        { 
                           VariantClear(&varSave); 
                           VariantCopy(&varSave, &var); 
                        } 
                     } 
                     SafeArrayUnaccessData(V_ARRAY(&var)); 
                     VariantClear(&var); 
                  } 
                  ::SysFreeString(bstr); 
               } 
               pmt->Release(); 
            } 
            format->Release(); 
         } 
      } 
   } 
   // return an error, if we didn't find a good candidate 
   if (V_VT(&varSave) == VT_EMPTY) 
      return E_FAIL; 
   WAVEFORMATEX *pSaveWFX; 
   SafeArrayAccessData(V_ARRAY(&varSave), (void**) &pSaveWFX); 
   IltmmMediaTypeDisp* pmt; 
   hr = CoCreateInstance(CLSID_ltmmMediaType, NULL, CLSCTX_INPROC_SERVER, IID_IltmmMediaTypeDisp, (void**) &pmt); 
   if (FAILED(hr)) 
   { 
      SafeArrayUnaccessData(V_ARRAY(&varSave)); 
      VariantClear(&varSave); 
      return hr; 
   } 
   // setup the new stream's media type 
   bstr = ::SysAllocString(ltmmMEDIATYPE_Audio); 
   pmt->put_Type(bstr); 
   ::SysFreeString(bstr); 
   CString s; 
   s.Format(_T("{%.8X-0000-0010-8000-00AA00389B71}"), (UINT) pSaveWFX->wFormatTag); 
   bstr = s.AllocSysString(); 
   pmt->put_Subtype(bstr); 
   ::SysFreeString(bstr); 
   pmt->put_FixedSizeSamples(VARIANT_TRUE); 
   pmt->put_TemporalCompression(VARIANT_FALSE); 
   pmt->put_SampleSize(pSaveWFX->nBlockAlign); 
   bstr = ::SysAllocString(ltmmFORMAT_WaveFormatEx); 
   pmt->put_FormatType(bstr); 
   ::SysFreeString(bstr); 
   pmt->SetFormatData(-1L, varSave); 
   IltmmWMStreamConfig* stream; 
   bstr = ::SysAllocString(ltmmMEDIATYPE_Audio); 
   hr = profile->CreateNewStream(bstr, &stream); 
   ::SysFreeString(bstr); 
   if (FAILED(hr)) 
   { 
      SafeArrayUnaccessData(V_ARRAY(&varSave)); 
      VariantClear(&varSave); 
      pmt->Release(); 
      return hr; 
   } 
   stream->put_StreamNumber(nStreamNum); 
   bstr = ::SysAllocString(L"Audio Stream"); 
   stream->put_StreamName(bstr); 
   ::SysFreeString(bstr); 
   bstr = ::SysAllocString(L"Audio"); 
   stream->put_ConnectionName(bstr); 
   ::SysFreeString(bstr); 
   stream->put_Bitrate(pSaveWFX->nAvgBytesPerSec * 8); 
   stream->SetMediaType(pmt); 
   profile->AddStream(stream); 
   stream->Release(); 
   pmt->Release(); 
   SafeArrayUnaccessData(V_ARRAY(&varSave)); 
   VariantClear(&varSave); 
   return S_OK; 
} 
HRESULT AddVideoStream(IltmmWMProfileManager* mgr, IltmmWMProfile* profile, 
UINT nStreamNum, DWORD dwFourCC, ULONG nBitRate, ULONG nWidth, ULONG nHeight, 
ULONG nFPS, ULONG nQuality, ULONG nSecPerKey) 
{ 
   long cCodecs; 
   IltmmWMStreamConfig* stream = NULL; 
   IltmmMediaTypeDisp* pmt = NULL; 
   VIDEOINFOHEADER *pVIH = NULL; 
   VARIANT varFormat; 
   VariantInit(&varFormat); 
   // get MEDIATYPE_Video codec count 
   BSTR bstr = ::SysAllocString(ltmmMEDIATYPE_Video); 
   HRESULT hr = mgr->GetCodecInfoCount(bstr, &cCodecs); 
   ::SysFreeString(bstr); 
   if (FAILED(hr)) 
      return hr; 
   // search for matching codec 
   for (long i = 0; !stream && i < cCodecs; i++) 
   { 
      long cFormats; 
      // get MEDIATYPE_Video codec format count 
      BSTR bstr = ::SysAllocString(ltmmMEDIATYPE_Video); 
      HRESULT hr = mgr->GetCodecFormatCount(bstr, i, &cFormats); 
      ::SysFreeString(bstr); 
      if (FAILED(hr)) 
         break; 
      for (long j = 0; j < cFormats; j++) 
      { 
         // get MEDIATYPE_Video stream config interface 
         bstr = ::SysAllocString(ltmmMEDIATYPE_Video); 
         hr = mgr->GetCodecFormat(bstr, i, j, &stream); 
         ::SysFreeString(bstr); 
         if (SUCCEEDED(hr)) 
         { 
            // get this format's media type 
            hr = stream->GetMediaType(&pmt); 
            if (SUCCEEDED(hr)) 
            { 
               hr = pmt->get_FormatType(&bstr); 
               if (SUCCEEDED(hr)) 
               { 
                  // only check FORMAT_VideoInfo formats 
                  if (!_wcsicmp(bstr, ltmmFORMAT_VideoInfo)) 
                  { 
                     pmt->GetFormatData(-1, &varFormat); 
                     SafeArrayAccessData(V_ARRAY(&varFormat), (void**) &pVIH); 
                     if (pVIH->bmiHeader.biCompression == dwFourCC) 
                     { 
                        ::SysFreeString(bstr); 
                        break; 
                     } 
                     SafeArrayUnaccessData(V_ARRAY(&varFormat)); 
                     VariantClear(&varFormat); 
                  } 
                  ::SysFreeString(bstr); 
               } 
               pmt->Release(); 
               pmt = NULL; 
            } 
            stream->Release(); 
            stream = NULL; 
         } 
      } 
   } 
   // return an error, if we didn't find a good candidate 
   if (!stream) 
      return E_FAIL; 
   pVIH->dwBitRate = nBitRate; 
   pVIH->rcSource.right = nWidth; 
   pVIH->rcSource.bottom = nHeight; 
   pVIH->rcTarget.right = nWidth; 
   pVIH->rcTarget.bottom = nHeight; 
   pVIH->bmiHeader.biWidth = nWidth; 
   pVIH->bmiHeader.biHeight = nHeight; 
   pVIH->AvgTimePerFrame = ((LONGLONG) 10000000 ) / ((LONGLONG) nFPS ); 
   pmt->SetFormatData(-1, varFormat); 
   hr = stream->put_Quality( nQuality ); 
   hr = stream->put_MaxKeyFrameSpacing((double) nSecPerKey); 
   stream->put_StreamNumber(nStreamNum); 
   bstr = ::SysAllocString(L"Video Stream"); 
   stream->put_StreamName(bstr); 
   ::SysFreeString(bstr); 
   bstr = ::SysAllocString(L"Video"); 
   stream->put_ConnectionName(bstr); 
   ::SysFreeString(bstr); 
   stream->put_Bitrate(nBitRate); 
   stream->SetMediaType(pmt); 
   profile->AddStream(stream); 
   stream->Release(); 
   pmt->Release(); 
   SafeArrayUnaccessData(V_ARRAY(&varFormat)); 
   VariantClear(&varFormat); 
   return S_OK; 
} 
HRESULT AddMutexObject(IltmmWMProfile *profile, UINT nBaseStream, UINT nStreamCount) 
{ 
   IltmmWMMutualExclusion *excl = NULL; 
   // create an exclusion object 
   HRESULT hr = profile->CreateNewMutualExclusion(&excl); 
   if (FAILED(hr)) 
      return hr; 
   // indicate that the streams differ by bit rate 
   // see CLSID_WMMUTEX_Bitrate in the WMSDK 
   BSTR bstr = ::SysAllocString(L"{D6E22A01-35DA-11D1-9034-00A0C90349BE}"); 
   hr = excl->put_Type(bstr); 
   ::SysFreeString(bstr); 
   if (FAILED(hr)) 
   { 
      excl->Release(); 
      return hr; 
   } 
   // mark all the streams for mutual exclusion 
   for (UINT i = 0; i < nStreamCount; i++) 
   { 
      hr = excl->AddStream(nBaseStream + i); 
      if (FAILED(hr)) 
      { 
         excl->Release(); 
         return hr; 
      } 
   } 
   // assign the exclusion object to the profile 
   hr = profile->AddMutualExclusion(excl); 
   excl->Release(); 
   return hr; 
}