ttextinputpin.cpp.svn-base

来自「ffshow源码」· SVN-BASE 代码 · 共 233 行

/*
 * Copyright (c) 2004-2006 Milan Cutka
 *
 * This program 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.
 *
 * This program 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 this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 */

#include "stdafx.h"
#include "TtextInputPin.h"
#include "ffdebug.h"
#include "TffDecoderVideo.h"
#include "ffdshow_constants.h"
#include "ffdshow_mediaguids.h"
#include "matroskaSubs.h"
#include "Tsubreader.h"
#include "TsubtitlesSettings.h"
               
TtextInputPin::TtextInputPin(TffdshowDecVideo* pFilter,HRESULT* phr,const wchar_t *pinname,int Iid):
 CDeCSSInputPin(NAME("TtextInputPin"),pFilter,phr,pinname),
 id(Iid),
 filter(pFilter),
 segmentStart(0),
 extradata(NULL),extradatasize(0),
 firsttime(true),
 needSegment(false),
 found(false)
{
 supdvddec=filter->getParam2(IDFF_supDVDdec) && filter->getParam2(IDFF_mpg2);
 supssa=!!filter->getParam2(IDFF_subTextpinSSA);
 name[0]='\0';
}
TtextInputPin::~TtextInputPin()
{
 if (extradata) free(extradata);
}

HRESULT TtextInputPin::CheckMediaType(const CMediaType *mtIn)
{
 return (filter->getParam2(IDFF_subTextpin) || (mtIn->majortype==MEDIATYPE_DVD_ENCRYPTED_PACK && supdvddec)) &&
         (mtIn->majortype==MEDIATYPE_Text ||
         (mtIn->majortype==MEDIATYPE_Subtitle && 
          (mtIn->subtype==MEDIASUBTYPE_UTF8 || 
           mtIn->subtype==MEDIASUBTYPE_VOBSUB ||
           mtIn->subtype==MEDIASUBTYPE_USF ||
           (mtIn->subtype==MEDIASUBTYPE_SSA  && supssa) ||
           (mtIn->subtype==MEDIASUBTYPE_ASS  && supssa) ||
           (mtIn->subtype==MEDIASUBTYPE_ASS2 && supssa))
         ) ||
         (
          ((mtIn->majortype==MEDIATYPE_DVD_ENCRYPTED_PACK && supdvddec) || 
           mtIn->majortype==MEDIATYPE_MPEG2_PACK || 
           mtIn->majortype==MEDIATYPE_MPEG2_PES || 
           mtIn->majortype==MEDIATYPE_Video ||
           mtIn->majortype==MEDIATYPE_Stream) && 
          (mtIn->subtype==MEDIASUBTYPE_DVD_SUBPICTURE || 
           mtIn->subtype==MEDIASUBTYPE_CVD_SUBPICTURE || 
           mtIn->subtype==MEDIASUBTYPE_SVCD_SUBPICTURE)
        ))?S_OK:VFW_E_TYPE_NOT_ACCEPTED;
}

HRESULT TtextInputPin::SetMediaType(const CMediaType* mtIn)
{
 name[0]='\0';
 HRESULT hr=CDeCSSInputPin::SetMediaType(mtIn);
 if (hr!=S_OK) return hr;
 type=Tsubreader::SUB_INVALID;
 needSegment=false;
 if (mtIn->majortype==MEDIATYPE_Subtitle)
  {
   const SUBTITLEINFO *psi=(const SUBTITLEINFO*)mtIn->Format();
   const char_t *isoname=TsubtitlesSettings::getLangDescrIso(text<char_t>(psi->IsoLang));
   char_t trackname[256];
   text<char_t>(psi->TrackName,trackname);
   tsnprintf(name,256,_l("%s%s%s"),trackname,trackname[0]?_l(" "):_l(""),isoname);
   if (extradata) free(extradata);
   extradatasize=mtIn->cbFormat-psi->dwOffset;
   if (extradatasize)
    {
     extradata=(unsigned char*)malloc(extradatasize);
     memcpy(extradata,mtIn->pbFormat+psi->dwOffset,extradatasize);
    }
   if (mtIn->subtype==MEDIASUBTYPE_UTF8) 
    type=Tsubreader::SUB_SUBVIEWER|Tsubreader::SUB_ENC_UTF8;
   else if (mtIn->subtype==MEDIASUBTYPE_VOBSUB)
    type=Tsubreader::SUB_VOBSUB;
   else if (mtIn->subtype==MEDIASUBTYPE_SSA || mtIn->subtype==MEDIASUBTYPE_ASS || mtIn->subtype==MEDIASUBTYPE_ASS2)
    type=Tsubreader::SUB_SSA;
   else if (mtIn->subtype==MEDIASUBTYPE_USF)
    type=Tsubreader::SUB_USF;
  }
 else if (mtIn->majortype==MEDIATYPE_Text)
  type=Tsubreader::SUB_SUBVIEWER/*|Tsubreader::SUB_ENC_UTF8*/; //TODO: enable for mp4 subtitles?
 else 
  if (mtIn->subtype==MEDIASUBTYPE_DVD_SUBPICTURE)
   {
    if (mtIn->majortype==MEDIATYPE_Stream)
     needSegment=true;
    type=Tsubreader::SUB_DVD;
   } 
  else if (mtIn->subtype==MEDIASUBTYPE_CVD_SUBPICTURE)
   type=Tsubreader::SUB_CVD;
  else if (mtIn->subtype==MEDIASUBTYPE_SVCD_SUBPICTURE)
   type=Tsubreader::SUB_SVCD;
 firsttime=true;
 if (filter->getParam2(IDFF_subShowEmbedded)==0)
  filter->putParam(IDFF_subShowEmbedded,id);
 return S_OK;
}
HRESULT TtextInputPin::CompleteConnect(IPin *pReceivePin)
{
 if (name[0]=='\0')
  {
   PIN_INFO pi;
   pReceivePin->QueryPinInfo(&pi);
   text<char_t>(pi.achName,name);
   if (pi.pFilter) pi.pFilter->Release();
  }
 return CDeCSSInputPin::CompleteConnect(pReceivePin);
}

STDMETHODIMP TtextInputPin::NewSegment(REFERENCE_TIME tStart,REFERENCE_TIME tStop,double dRate)
{
 segmentStart=tStart;
 filter->resetSubtitles(id);
 return CBaseInputPin::NewSegment(tStart,tStop,dRate);
}
STDMETHODIMP TtextInputPin::EndOfStream(void)
{
 return S_OK;
}
STDMETHODIMP TtextInputPin::BeginFlush(void) 
{
 return S_OK;
}
STDMETHODIMP TtextInputPin::EndFlush(void)
{
 if (Tsubreader::isDVDsub(type))
  filter->resetSubtitles(id);
 return S_OK;
}

STDMETHODIMP TtextInputPin::QuerySupported(REFGUID PropSet,ULONG Id,ULONG *pTypeSupport)
{
 if (PropSet!=AM_KSPROPSETID_DvdSubPic)
  return CDeCSSInputPin::QuerySupported(PropSet,Id,pTypeSupport);
 switch (Id)
  {
   case AM_PROPERTY_DVDSUBPIC_PALETTE:*pTypeSupport=KSPROPERTY_SUPPORT_SET;return S_OK;
   case AM_PROPERTY_DVDSUBPIC_HLI:*pTypeSupport=KSPROPERTY_SUPPORT_SET;return S_OK;
   case AM_PROPERTY_DVDSUBPIC_COMPOSIT_ON:*pTypeSupport=KSPROPERTY_SUPPORT_SET;return S_OK;
   default:return E_PROP_ID_UNSUPPORTED;
  }
}
STDMETHODIMP TtextInputPin::Set(REFGUID PropSet,ULONG ctl_Id,LPVOID pInstanceData,ULONG InstanceLength,LPVOID pPropertyData,ULONG DataLength)
{
 if (PropSet!=AM_KSPROPSETID_DvdSubPic)
  return CDeCSSInputPin::Set(PropSet,ctl_Id,pInstanceData,InstanceLength,pPropertyData,DataLength);
 filter->ctlSubtitles(id,type,ctl_Id,pPropertyData,DataLength);
 firsttime=false;found=true;
 return S_OK;
}

HRESULT TtextInputPin::Receive(IMediaSample *pSample)
{
 HRESULT hr;
 ASSERT(pSample);

 hr=CDeCSSInputPin::Receive(pSample);
 if (FAILED(hr)) return hr;

 AM_MEDIA_TYPE* pmt=NULL;
 if (SUCCEEDED(pSample->GetMediaType(&pmt)) && pmt)
  {
   CMediaType mt(*pmt);
   bool oldfirsttime=firsttime;
   SetMediaType(&mt);
   DeleteMediaType(pmt);
   firsttime=oldfirsttime;
  }

 if (firsttime)
  {
   firsttime=false;
   found=filter->initSubtitles(id,type,extradata,extradatasize);
  }
 
 REFERENCE_TIME t1=-1,t2=-1;
 pSample->GetTime(&t1,&t2);
 
 BYTE *data;
 pSample->GetPointer(&data);
 long datalen=pSample->GetActualDataLength();
 if (Tsubreader::isDVDsub(type))
  StripPacket(data,datalen);
 //int sStart=float(t1+segmentStart)/REF_SECOND_MULT,sStop=float(t2+segmentStart)/REF_SECOND_MULT;
 //data[datalen]='\0';
 //DPRINTF(_l("%02i:%02i:%02i-%02i:%02i:%02i %s"),sStart/3600,(sStart%3600)/60,sStart%60,sStop/3600,(sStop%3600)/60,sStop%60,(const char_t*)text<char_t>((const char*)data));
 if (data && datalen>0)
  filter->addSubtitle(id,t1+segmentStart,t2+segmentStart,data,datalen);
 return S_OK;
}

HRESULT TtextInputPin::Transform(IMediaSample* pSample)
{
 return S_FALSE;
}

STDMETHODIMP TtextInputPin::Disconnect(void)
{
 filter->putParam(IDFF_subShowEmbedded,0);
 name[0]='\0';found=false;
 return CDeCSSInputPin::Disconnect();
}

HRESULT TtextInputPin::getInfo(const char_t* *namePtr,int *idPtr,int *foundPtr)
{
 if (namePtr) *namePtr=name;
 if (idPtr) *idPtr=id;
 if (foundPtr) *foundPtr=found;
 return S_OK;
}