source: trunk/src/testing/app/hd-video-player/GrabberProc.h @ 4

/******************************************************************************
 * SAGE - Scalable Adaptive Graphics Environment
 *
 * Copyright (C) 2004 Electronic Visualization Laboratory,
 * University of Illinois at Chicago
 *
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 *  * Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 *  * Redistributions in binary form must reproduce the above
 *    copyright notice, this list of conditions and the following disclaimer
 *    in the documentation and/or other materials provided with the distribution.
 *  * Neither the name of the University of Illinois at Chicago nor
 *    the names of its contributors may be used to endorse or promote
 *    products derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 * Direct questions, comments etc about SAGE to http://www.evl.uic.edu/cavern/forum/
 *
 *****************************************************************************/
#ifndef _GRABBERCB_
#define _GRABBERCB_

#include <qedit.h>   // Sample grabber

#include "Histogram.h"
#include "Utils.h"
#include "output_sail.h"

class C_GrabProc {

//      static GrabberCB m_grabberCallback;  // Make this global so that it stays in scope while the graph exists
        // create filter------------
        AM_MEDIA_TYPE m_mt;
        AM_MEDIA_TYPE m_audiomt;
        CComPtr<IBaseFilter> m_pGrabF;
        CComPtr<IBaseFilter> m_pAudioGrabF;
        CComPtr<IBaseFilter> m_pYUVtransF;
        int m_sub_type;
        // send image---------------
    BYTE*   m_pImg;
    BYTE*   m_pAudioSamples;
    DWORD   m_dwWidth;
    DWORD   m_dwHeight;
    LONG    m_lStride;
    int     m_iBitDepth;
        int             m_out_color_mode;
        unsigned char* m_pbRGB_buff;
        DWORD m_bufLen;
        bool m_sageOn;
        CString m_ip;

        class C_Output_sail Out_sail;

public :
        // create filter------------
        C_GrabProc(void);
        void FreeFilter(void);
        HRESULT SetFilter(IGraphBuilder* pGraph, IBaseFilter* pUpstream, wchar_t* pUpstreamPinName, int out_color_mode, bool sage=false);
        HRESULT SetAudioFilter(IGraphBuilder* pGraph, IBaseFilter* pUpstream, wchar_t* pUpstreamPinName);
        IBaseFilter* GrabF(void);
        // send image---------------
        HRESULT SetVideoFormat(const VIDEOINFOHEADER& vih, int out_color_mode);
    HRESULT SetImageBuffer(BYTE *pBuffer);
    HRESULT SetAudioBuffer(BYTE *pBuffer);
        HRESULT ConvertImage();
        int SendToBmp(void);
        int SendToSage(void);
        int SendAudioToSage(int buflen);
        void setIP(CString ip);

        void SetAudioOn(bool on);
        void SetAudioCaptureMode(bool on);
        void Shutdown();

        bool GetSageOn();
};

// GrabberCB: Class to implement the callback function for the sample grabber
// Note: No ref-counting, so keep the object in scope while you're using it!
class GrabberCB : public ISampleGrabberCB
{
private:
    BITMAPINFOHEADER m_bmi;  // Holds the bitmap format
    bool m_fFirstSample;     // true if the next sample is the first one.

public:

    GrabberCB();
    ~GrabberCB();

        C_GrabProc m_GrabProc;
        int m_out_color_mode;

    // IUnknown methods
    STDMETHODIMP_(ULONG) AddRef() { return 1; }
    STDMETHODIMP_(ULONG) Release() { return 2; }
    STDMETHOD(QueryInterface)(REFIID iid, void** ppv);

    // ISampleGrabberCB methods
    STDMETHOD(SampleCB)(double SampleTime, IMediaSample *pSample);
    STDMETHODIMP BufferCB(double, BYTE *, long) { return E_NOTIMPL; }

};

// AudioGrabberCB: Class to implement the callback function for the sample grabber
// Note: No ref-counting, so keep the object in scope while you're using it!
class AudioGrabberCB : public ISampleGrabberCB
{
private:
    BITMAPINFOHEADER m_bmi;  // Holds the bitmap format
    bool m_fFirstSample;     // true if the next sample is the first one.
        int m_index;
public:

    AudioGrabberCB();
    ~AudioGrabberCB();

        C_GrabProc* m_GrabProc;

    // IUnknown methods
    STDMETHODIMP_(ULONG) AddRef() { return 1; }
    STDMETHODIMP_(ULONG) Release() { return 2; }
    STDMETHOD(QueryInterface)(REFIID iid, void** ppv);

    // ISampleGrabberCB methods
    STDMETHOD(SampleCB)(double SampleTime, IMediaSample *pSample);
    STDMETHODIMP BufferCB(double, BYTE *, long) { return E_NOTIMPL; }

};

// GUID media_type_ary[] = {MEDIASUBTYPE_UYVY, MEDIASUBTYPE_YUY2, MEDIASUBTYPE_YUYV};

#ifdef _MAIN_
GrabberCB grabberCallback;  // Make this global so that it stays in scope while the graph exists
AudioGrabberCB AudioGrabCallback;
#else
extern GrabberCB grabberCallback;  // Make this global so that it stays in scope while the graph exists
extern AudioGrabberCB AudioGrabCallback;
#endif


#endif