avmedia/source/gstreamer/gstframegrabber.cxx

   1 /*************************************************************************
   2  *
   3  *  OpenOffice.org - a multi-platform office productivity suite
   4  *
   5  *  $RCSfile$
   6  *
   7  *  $Revision$
   8  *
   9  *  last change: $Author$ $Date$
  10  *
  11  *  The Contents of this file are made available subject to
  12  *  the terms of GNU Lesser General Public License Version 2.1.
  13  *
  14  *
  15  *    GNU Lesser General Public License Version 2.1
  16  *    =============================================
  17  *    Copyright 2005 by Sun Microsystems, Inc.
  18  *    901 San Antonio Road, Palo Alto, CA 94303, USA
  19  *
  20  *    This library is free software; you can redistribute it and/or
  21  *    modify it under the terms of the GNU Lesser General Public
  22  *    License version 2.1, as published by the Free Software Foundation.
  23  *
  24  *    This library is distributed in the hope that it will be useful,
  25  *    but WITHOUT ANY WARRANTY; without even the implied warranty of
  26  *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  27  *    Lesser General Public License for more details.
  28  *
  29  *    You should have received a copy of the GNU Lesser General Public
  30  *    License along with this library; if not, write to the Free Software
  31  *    Foundation, Inc., 59 Temple Place, Suite 330, Boston,
  32  *    MA  02111-1307  USA
  33  *
  34  ************************************************************************/
  35
  36 #include <tools/prewin.h>
  37 #include <windows.h>
  38 #include <objbase.h>
  39 #include <strmif.h>
  40 #include <Amvideo.h>
  41 #include <Qedit.h>
  42 #include <uuids.h>
  43 #include <tools/postwin.h>
  44
  45 #include "framegrabber.hxx"
  46 #include "player.hxx"
  47
  48 #include <tools/stream.hxx>
  49 #include <vcl/graph.hxx>
  50 #include <unotools/localfilehelper.hxx>
  51
  52 #define AVMEDIA_GST_FRAMEGRABBER_IMPLEMENTATIONNAME "com.sun.star.comp.avmedia.FrameGrabber_GStreamer"
  53 #define AVMEDIA_GST_FRAMEGRABBER_SERVICENAME "com.sun.star.media.FrameGrabber_GStreamer"
  54
  55 using namespace ::com::sun::star;
  56
  57 namespace avmedia { namespace gstreamer {
  58
  59 // ----------------
  60 // - FrameGrabber -
  61 // ----------------
  62
  63 FrameGrabber::FrameGrabber( const uno::Reference< lang::XMultiServiceFactory >& rxMgr ) :
  64     mxMgr( rxMgr )
  65 {
  66     ::CoInitialize( NULL );
  67 }
  68
  69 // ------------------------------------------------------------------------------
  70
  71 FrameGrabber::~FrameGrabber()
  72 {
  73     ::CoUninitialize();
  74 }
  75
  76 // ------------------------------------------------------------------------------
  77
  78 IMediaDet* FrameGrabber::implCreateMediaDet( const ::rtl::OUString& rURL ) const
  79 {
  80     IMediaDet* pDet = NULL;
  81
  82     if( SUCCEEDED( CoCreateInstance( CLSID_MediaDet, NULL, CLSCTX_INPROC_SERVER, IID_IMediaDet, (void**) &pDet ) ) )
  83     {
  84                 String aLocalStr;
  85
  86                 if( ::utl::LocalFileHelper::ConvertURLToPhysicalName( rURL, aLocalStr ) && aLocalStr.Len() )
  87                 {
  88             if( !SUCCEEDED( pDet->put_Filename( ::SysAllocString( aLocalStr.GetBuffer() ) ) ) )
  89             {
  90                 pDet->Release();
  91                 pDet = NULL;
  92             }
  93         }
  94     }
  95
  96     return pDet;
  97 }
  98
  99 // ------------------------------------------------------------------------------
 100
 101 bool FrameGrabber::create( const ::rtl::OUString& rURL )
 102 {
 103     // just check if a MediaDet interface can be created with the given URL
 104     IMediaDet*  pDet = implCreateMediaDet( rURL );
 105
 106     if( pDet )
 107     {
 108         maURL = rURL;
 109         pDet->Release();
 110         pDet = NULL;
 111     }
 112     else
 113         maURL = ::rtl::OUString();
 114
 115     return( maURL.getLength() > 0 );
 116 }
 117
 118 // ------------------------------------------------------------------------------
 119
 120 uno::Reference< graphic::XGraphic > SAL_CALL FrameGrabber::grabFrame( double fMediaTime )
 121     throw (uno::RuntimeException)
 122 {
 123     uno::Reference< graphic::XGraphic > xRet;
 124     IMediaDet*                          pDet = implCreateMediaDet( maURL );
 125
 126     if( pDet )
 127     {
 128         double  fLength;
 129         long    nStreamCount;
 130         bool    bFound = false;
 131
 132         if( SUCCEEDED( pDet->get_OutputStreams( &nStreamCount ) ) )
 133         {
 134             for( long n = 0; ( n < nStreamCount ) && !bFound; ++n )
 135             {
 136                 GUID aMajorType;
 137
 138                 if( SUCCEEDED( pDet->put_CurrentStream( n ) )  &&
 139                     SUCCEEDED( pDet->get_StreamType( &aMajorType ) ) &&
 140                     ( aMajorType == MEDIATYPE_Video ) )
 141                 {
 142                     bFound = true;
 143                 }
 144             }
 145         }
 146
 147         if( bFound &&
 148             ( S_OK == pDet->get_StreamLength( &fLength ) ) &&
 149             ( fLength > 0.0 ) && ( fMediaTime >= 0.0 ) && ( fMediaTime <= fLength ) )
 150         {
 151             AM_MEDIA_TYPE   aMediaType;
 152             long            nWidth = 0, nHeight = 0, nSize = 0;
 153
 154             if( SUCCEEDED( pDet->get_StreamMediaType( &aMediaType ) ) )
 155             {
 156                 if( ( aMediaType.formattype == FORMAT_VideoInfo ) &&
 157                     ( aMediaType.cbFormat >= sizeof( VIDEOINFOHEADER ) ) )
 158                 {
 159                     VIDEOINFOHEADER* pVih = reinterpret_cast< VIDEOINFOHEADER* >( aMediaType.pbFormat );
 160
 161                     nWidth = pVih->bmiHeader.biWidth;
 162                     nHeight = pVih->bmiHeader.biHeight;
 163
 164                     if( nHeight < 0 )
 165                         nHeight *= -1;
 166                 }
 167
 168                 if( aMediaType.cbFormat != 0 )
 169                 {
 170                     ::CoTaskMemFree( (PVOID) aMediaType.pbFormat );
 171                     aMediaType.cbFormat = 0;
 172                     aMediaType.pbFormat = NULL;
 173                 }
 174
 175                 if( aMediaType.pUnk != NULL )
 176                 {
 177                     aMediaType.pUnk->Release();
 178                     aMediaType.pUnk = NULL;
 179                 }
 180             }
 181
 182             if( ( nWidth > 0 ) && ( nHeight > 0 ) &&
 183                 SUCCEEDED( pDet->GetBitmapBits( 0, &nSize, NULL, nWidth, nHeight ) ) &&
 184                 ( nSize > 0  ) )
 185             {
 186                 char* pBuffer = new char[ nSize ];
 187
 188                 try
 189                 {
 190                     if( SUCCEEDED( pDet->GetBitmapBits( fMediaTime, NULL, pBuffer, nWidth, nHeight ) ) )
 191                     {
 192                         SvMemoryStream  aMemStm( pBuffer, nSize, STREAM_READ | STREAM_WRITE );
 193                         Bitmap          aBmp;
 194
 195                         if( aBmp.Read( aMemStm, false ) && !aBmp.IsEmpty() )
 196                         {
 197                             const Graphic aGraphic( aBmp );
 198                             xRet = aGraphic.GetXGraphic();
 199                         }
 200                     }
 201                 }
 202                 catch( ... )
 203                 {
 204                 }
 205
 206                 delete [] pBuffer;
 207             }
 208         }
 209
 210         pDet->Release();
 211     }
 212
 213     return xRet;
 214 }
 215
 216 // ------------------------------------------------------------------------------
 217
 218 ::rtl::OUString SAL_CALL FrameGrabber::getImplementationName(  )
 219     throw (uno::RuntimeException)
 220 {
 221     return ::rtl::OUString( RTL_CONSTASCII_USTRINGPARAM( AVMEDIA_GST_FRAMEGRABBER_IMPLEMENTATIONNAME ) );
 222 }
 223
 224 // ------------------------------------------------------------------------------
 225
 226 sal_Bool SAL_CALL FrameGrabber::supportsService( const ::rtl::OUString& ServiceName )
 227     throw (uno::RuntimeException)
 228 {
 229     return ServiceName.equalsAsciiL( RTL_CONSTASCII_STRINGPARAM ( AVMEDIA_GST_FRAMEGRABBER_SERVICENAME ) );
 230 }
 231
 232 // ------------------------------------------------------------------------------
 233
 234 uno::Sequence< ::rtl::OUString > SAL_CALL FrameGrabber::getSupportedServiceNames(  )
 235     throw (uno::RuntimeException)
 236 {
 237     uno::Sequence< ::rtl::OUString > aRet(1);
 238     aRet[0] = ::rtl::OUString( RTL_CONSTASCII_USTRINGPARAM ( AVMEDIA_GST_FRAMEGRABBER_SERVICENAME ) );
 239
 240     return aRet;
 241 }
 242
 243 } // namespace gstreamer
 244 } // namespace avmedia