KinectAudio.cpp

Go to the documentation of this file.

 
//#include  "WinBaseLib.h"
#include  "KinectAudio.h"
 
 
#ifdef  ENABLE_KINECT_SDK
 
 
using namespace jbxl;
using namespace jbxwl;
 
 
CKinectAudio::CKinectAudio(INuiSensor* sensor, int mode)
{
    m_audioStream   = NULL;
    m_audioSource   = NULL;
    m_audioMedia    = NULL;
    m_audioProperty = NULL;
    m_pStream       = NULL;
 
    //
    if (sensor!=NULL) init(sensor, mode);
}
 
 
 
CKinectAudio::~CKinectAudio(void)
{
    DEBUG_INFO("DESTRUCTOR: CKinectAudio: START\n");
 
    free();
 
    DEBUG_INFO("DESTRUCTOR: CKinectAudio: END\n");
}
 
 
 
BOOL  CKinectAudio::init(INuiSensor* sensor, int mode)
{
    free();
 
    if (sensor!=NULL) {
        HRESULT hr = sensor->NuiGetAudioSource(&m_audioSource);
        if (SUCCEEDED(hr)) hr = m_audioSource->QueryInterface(IID_IMediaObject,   (void**)&m_audioMedia);
        if (SUCCEEDED(hr)) hr = m_audioSource->QueryInterface(IID_IPropertyStore, (void**)&m_audioProperty);
        //
        if (SUCCEEDED(hr)) {
            BOOL ret = setProperty(mode);
            if (ret) ret = setMediaType();
            if (ret) {
                m_audioStream = new CWinAudioStream(m_audioMedia);
                m_audioStream->QueryInterface(IID_IStream, (void**)&m_pStream);
                return TRUE;
            }
        }
    }
    return FALSE;
}
 
 
 
void  CKinectAudio::free(void)
{
    releaseNull(m_pStream);
    deleteNull (m_audioStream);
 
    releaseNull(m_audioProperty);
    releaseNull(m_audioMedia);
    releaseNull(m_audioSource);
}
 
 
 
BOOL  CKinectAudio::setProperty(int mode)
{
    if (m_audioProperty==NULL) return FALSE;
 
    PROPVARIANT prop;
    
    PropVariantInit(&prop);
    prop.vt   = VT_I4;
    prop.lVal = (LONG)mode;
    m_audioProperty->SetValue(MFPKEY_WMAAECMA_SYSTEM_MODE, prop);
    PropVariantClear(&prop);
 
    return TRUE;
}
 
 
 
BOOL  CKinectAudio::setMediaType(void)
{
    if (m_audioMedia==NULL) return FALSE;
 
    DMO_MEDIA_TYPE mt = {0};
    MoInitMediaType(&mt, sizeof(WAVEFORMATEX));
    mt.majortype = MEDIATYPE_Audio;
    mt.subtype   = MEDIASUBTYPE_PCM;
    mt.lSampleSize = 0;
    mt.bFixedSizeSamples    = TRUE;
    mt.bTemporalCompression = FALSE;
    mt.formattype = FORMAT_WaveFormatEx;    
    memcpy(mt.pbFormat, &getAudioFormat(), sizeof(WAVEFORMATEX));
    
    HRESULT hr = m_audioMedia->SetOutputType(0, &mt, 0);
    MoFreeMediaType(&mt);
 
    if (FAILED(hr)) return FALSE;
    return TRUE;
}
 
 
 
void  CKinectAudio::getAudioData(Buffer* pbuf)
{
    if (m_pStream==NULL || pbuf==NULL) return;
 
    ULONG len = 0;
 
    m_pStream->Read((void*)pbuf->buf, (ULONG)pbuf->bufsz, &len);
    pbuf->vldsz = (int)len;
 
    return;
}
 
 
 
WAVEFORMATEX  CKinectAudio::getAudioFormat(void)
{
    WORD   blockAlign = ((KinectAudioBitsPerSample+7)/8)*KinectAudioChannels;
    DWORD  avgBytesPerSec = blockAlign*KinectAudioSamplesPerSec;
 
    WAVEFORMATEX format = {
        KinectAudioFormat, KinectAudioChannels, KinectAudioSamplesPerSec, 
        avgBytesPerSec, blockAlign, KinectAudioBitsPerSample, 0
    };
    return format;
}
 
 
#endif      // ENABLE_KINECT_SDK