/**
 *	BDADVBTSink.h
 *	Copyright (C) 2004 Nate
 *	Copyright (C) 2006 Bear
 *
 *	This file is part of DigitalWatch, a free DTV watching and recording
 *	program for the VisionPlus DVB-T.
 *
 *	DigitalWatch is free software; you can redistribute it and/or modify
 *	it under the terms of the GNU General Public License as published by
 *	the Free Software Foundation; either version 2 of the License, or
 *	(at your option) any later version.
 *
 *	DigitalWatch is distributed in the hope that it will be useful,
 *	but WITHOUT ANY WARRANTY; without even the implied warranty of
 *	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *	GNU General Public License for more details.
 *
 *	You should have received a copy of the GNU General Public License
 *	along with DigitalWatch; if not, write to the Free Software
 *	Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

#ifndef BDADVBTSINK_H
#define BDADVBTSINK_H

#include "BDADVBTSourceTuner.h"
#include "BDADVBTSinkTShift.h"
#include "BDADVBTSinkDSNet.h"
#include "BDADVBTSinkFile.h"
#include "StdAfx.h"
#include <bdatif.h>
#include "BDACardCollection.h"
#include "DWGraph.h"
#include "LogMessage.h"
#include "FilterGraphTools.h"
#include "DVBMpeg2DataParser.h"
#include "DVBTChannels.h"

// {4DF35815-79C5-44C8-8753-847D5C9C3CF5}
EXTERN_GUID(CLSID_MPEG2Multiplexer,
0x4DF35815, 0x79C5, 0x44C8, 0x87, 0x53, 0x84, 0x7D, 0x5C, 0x9C, 0x3C, 0xF5);

class BDADVBTSinkTShift;
class BDADVBTSinkFile;
class BDADVBTSinkDSNet;
class BDADVBTSink : public LogMessageCaller
{
public:
	BDADVBTSink();
	virtual ~BDADVBTSink();

	virtual void SetLogCallback(LogMessageCallback *callback);

	HRESULT Initialise(IGraphBuilder *piGraphBuilder, int cardID = 0);
	HRESULT DestroyAll();

	HRESULT AddSinkFilters(DVBTChannels_Service* pService);
	void DestroyFilter(CComPtr <IBaseFilter> &pFilter);
	HRESULT RemoveSinkFilters();

	HRESULT SetTransportStreamPin(IPin* piPin);

	BOOL IsActive();

	HRESULT AddFileName(LPOLESTR *ppFileName, DVBTChannels_Service* pService, CComPtr<IBaseFilter>& pFilter, int intSinkType = 0, LPWSTR pFileName = NULL, LPWSTR pPath = NULL);
	HRESULT NullFileName(CComPtr<IBaseFilter>& pFilter, int intSinkType);
	HRESULT StartRecording(DVBTChannels_Service* pService, LPWSTR pFilename = NULL, LPWSTR pPath = NULL);
	HRESULT StopRecording(void);
	HRESULT PauseRecording(void);
	HRESULT UnPauseRecording(DVBTChannels_Service* pService);
	BOOL IsRecording(void);
	BOOL IsPaused(void);
	HRESULT GetCurFile(LPOLESTR *ppszFileName, BOOL bTimeShiftSink = TRUE);
	HRESULT GetCurFileSize(__int64 *pllFileSize, BOOL bTimeShiftSink = TRUE);
	HRESULT GetSinkSize(LPOLESTR pFileName, __int64 *pllFileSize);
	HRESULT UpdateTSFileSink(BOOL bAutoMode = FALSE);
	HRESULT ClearSinkDemuxPins();
	HRESULT GetReferenceDemux(CComPtr<IBaseFilter>&pDemux);

	BOOL SupportsRecording() { return FALSE; }

private:

	BDADVBTSinkTShift *m_pCurrentTShiftSink;
	BDADVBTSinkFile	*m_pCurrentFileSink;
	BDADVBTSinkDSNet *m_pCurrentDSNetSink;

	CComPtr <IMpeg2Demultiplexer> m_piMpeg2Demux;
	CComPtr <IFileSinkFilter> m_piFileSink;

	BOOL m_bInitialised;
	BOOL m_bActive;

	CComPtr <IGraphBuilder> m_piGraphBuilder;
	CComPtr <IMediaControl> m_piMediaControl;

	CComPtr <IBaseFilter> m_piInfinitePinTee;

	long m_intTimeShiftType;
	long m_intFileSinkType;
	long m_intDSNetworkType;
	int m_cardId;

	FilterGraphTools graphTools;
};

#endif