avmedia/source/win/framegrabber.cxx

   1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
   2 /*
   3  * This file is part of the LibreOffice project.
   4  *
   5  * This Source Code Form is subject to the terms of the Mozilla Public
   6  * License, v. 2.0. If a copy of the MPL was not distributed with this
   7  * file, You can obtain one at http://mozilla.org/MPL/2.0/.
   8  *
   9  * This file incorporates work covered by the following license notice:
  10  *
  11  *   Licensed to the Apache Software Foundation (ASF) under one or more
  12  *   contributor license agreements. See the NOTICE file distributed
  13  *   with this work for additional information regarding copyright
  14  *   ownership. The ASF licenses this file to you under the Apache
  15  *   License, Version 2.0 (the "License"); you may not use this file
  16  *   except in compliance with the License. You may obtain a copy of
  17  *   the License at http://www.apache.org/licenses/LICENSE-2.0 .
  18  */
  19
  20 #if defined _MSC_VER
  21 #pragma warning(push, 1)
  22 #pragma warning(disable: 4917)
  23 #endif
  24 #include <prewin.h>
  25 #include <postwin.h>
  26 #include <objbase.h>
  27 #include <strmif.h>
  28 #include <Amvideo.h>
  29 #include "interface.hxx"
  30 #include <uuids.h>
  31 #if defined _MSC_VER
  32 #pragma warning(pop)
  33 #endif
  34
  35 #include "framegrabber.hxx"
  36 #include "player.hxx"
  37
  38 #include <tools/stream.hxx>
  39 #include <vcl/graph.hxx>
  40 #include <unotools/localfilehelper.hxx>
  41
  42 #define AVMEDIA_WIN_FRAMEGRABBER_IMPLEMENTATIONNAME "com.sun.star.comp.avmedia.FrameGrabber_DirectX"
  43 #define AVMEDIA_WIN_FRAMEGRABBER_SERVICENAME "com.sun.star.media.FrameGrabber_DirectX"
  44
  45 using namespace ::com::sun::star;
  46
  47 namespace avmedia { namespace win {
  48
  49 // ----------------
  50 // - FrameGrabber -
  51 // ----------------
  52
  53 FrameGrabber::FrameGrabber( const uno::Reference< lang::XMultiServiceFactory >& rxMgr ) :
  54     mxMgr( rxMgr )
  55 {
  56     ::CoInitialize( NULL );
  57 }
  58
  59 // ------------------------------------------------------------------------------
  60
  61 FrameGrabber::~FrameGrabber()
  62 {
  63     ::CoUninitialize();
  64 }
  65
  66 // ------------------------------------------------------------------------------
  67
  68 IMediaDet* FrameGrabber::implCreateMediaDet( const OUString& rURL ) const
  69 {
  70     IMediaDet* pDet = NULL;
  71
  72     if( SUCCEEDED( CoCreateInstance( CLSID_MediaDet, NULL, CLSCTX_INPROC_SERVER, IID_IMediaDet, (void**) &pDet ) ) )
  73     {
  74         OUString aLocalStr;
  75
  76         if( ::utl::LocalFileHelper::ConvertURLToPhysicalName( rURL, aLocalStr ) && !aLocalStr.isEmpty() )
  77         {
  78             if( !SUCCEEDED( pDet->put_Filename( ::SysAllocString( reinterpret_cast<LPCOLESTR>(aLocalStr.getStr()) ) ) ) )
  79             {
  80                 pDet->Release();
  81                 pDet = NULL;
  82             }
  83         }
  84     }
  85
  86     return pDet;
  87 }
  88
  89 // ------------------------------------------------------------------------------
  90
  91 bool FrameGrabber::create( const OUString& rURL )
  92 {
  93     // just check if a MediaDet interface can be created with the given URL
  94     IMediaDet*  pDet = implCreateMediaDet( rURL );
  95
  96     if( pDet )
  97     {
  98         maURL = rURL;
  99         pDet->Release();
 100         pDet = NULL;
 101     }
 102     else
 103         maURL = OUString();
 104
 105     return( maURL.getLength() > 0 );
 106 }
 107
 108 // ------------------------------------------------------------------------------
 109
 110 uno::Reference< graphic::XGraphic > SAL_CALL FrameGrabber::grabFrame( double fMediaTime )
 111     throw (uno::RuntimeException)
 112 {
 113     uno::Reference< graphic::XGraphic > xRet;
 114     IMediaDet*                          pDet = implCreateMediaDet( maURL );
 115
 116     if( pDet )
 117     {
 118         double  fLength;
 119         long    nStreamCount;
 120         bool    bFound = false;
 121
 122         if( SUCCEEDED( pDet->get_OutputStreams( &nStreamCount ) ) )
 123         {
 124             for( long n = 0; ( n < nStreamCount ) && !bFound; ++n )
 125             {
 126                 GUID aMajorType;
 127
 128                 if( SUCCEEDED( pDet->put_CurrentStream( n ) )  &&
 129                     SUCCEEDED( pDet->get_StreamType( &aMajorType ) ) &&
 130                     ( aMajorType == MEDIATYPE_Video ) )
 131                 {
 132                     bFound = true;
 133                 }
 134             }
 135         }
 136
 137         if( bFound &&
 138             ( S_OK == pDet->get_StreamLength( &fLength ) ) &&
 139             ( fLength > 0.0 ) && ( fMediaTime >= 0.0 ) && ( fMediaTime <= fLength ) )
 140         {
 141             AM_MEDIA_TYPE   aMediaType;
 142             long            nWidth = 0, nHeight = 0, nSize = 0;
 143
 144             if( SUCCEEDED( pDet->get_StreamMediaType( &aMediaType ) ) )
 145             {
 146                 if( ( aMediaType.formattype == FORMAT_VideoInfo ) &&
 147                     ( aMediaType.cbFormat >= sizeof( VIDEOINFOHEADER ) ) )
 148                 {
 149                     VIDEOINFOHEADER* pVih = reinterpret_cast< VIDEOINFOHEADER* >( aMediaType.pbFormat );
 150
 151                     nWidth = pVih->bmiHeader.biWidth;
 152                     nHeight = pVih->bmiHeader.biHeight;
 153
 154                     if( nHeight < 0 )
 155                         nHeight *= -1;
 156                 }
 157
 158                 if( aMediaType.cbFormat != 0 )
 159                 {
 160                     ::CoTaskMemFree( (PVOID) aMediaType.pbFormat );
 161                     aMediaType.cbFormat = 0;
 162                     aMediaType.pbFormat = NULL;
 163                 }
 164
 165                 if( aMediaType.pUnk != NULL )
 166                 {
 167                     aMediaType.pUnk->Release();
 168                     aMediaType.pUnk = NULL;
 169                 }
 170             }
 171
 172             if( ( nWidth > 0 ) && ( nHeight > 0 ) &&
 173                 SUCCEEDED( pDet->GetBitmapBits( 0, &nSize, NULL, nWidth, nHeight ) ) &&
 174                 ( nSize > 0  ) )
 175             {
 176                 char* pBuffer = new char[ nSize ];
 177
 178                 try
 179                 {
 180                     if( SUCCEEDED( pDet->GetBitmapBits( fMediaTime, NULL, pBuffer, nWidth, nHeight ) ) )
 181                     {
 182                         SvMemoryStream  aMemStm( pBuffer, nSize, STREAM_READ | STREAM_WRITE );
 183                         Bitmap          aBmp;
 184
 185                         if( aBmp.Read( aMemStm, false ) && !aBmp.IsEmpty() )
 186                         {
 187                             const Graphic aGraphic( aBmp );
 188                             xRet = aGraphic.GetXGraphic();
 189                         }
 190                     }
 191                 }
 192                 catch( ... )
 193                 {
 194                 }
 195
 196                 delete [] pBuffer;
 197             }
 198         }
 199
 200         pDet->Release();
 201     }
 202
 203     return xRet;
 204 }
 205
 206 // ------------------------------------------------------------------------------
 207
 208 OUString SAL_CALL FrameGrabber::getImplementationName(  )
 209     throw (uno::RuntimeException)
 210 {
 211     return OUString( AVMEDIA_WIN_FRAMEGRABBER_IMPLEMENTATIONNAME );
 212 }
 213
 214 // ------------------------------------------------------------------------------
 215
 216 sal_Bool SAL_CALL FrameGrabber::supportsService( const OUString& ServiceName )
 217     throw (uno::RuntimeException)
 218 {
 219     return ServiceName == AVMEDIA_WIN_FRAMEGRABBER_SERVICENAME;
 220 }
 221
 222 // ------------------------------------------------------------------------------
 223
 224 uno::Sequence< OUString > SAL_CALL FrameGrabber::getSupportedServiceNames(  )
 225     throw (uno::RuntimeException)
 226 {
 227     uno::Sequence< OUString > aRet(1);
 228     aRet[0] = AVMEDIA_WIN_FRAMEGRABBER_SERVICENAME ;
 229
 230     return aRet;
 231 }
 232
 233 } // namespace win
 234 } // namespace avmedia
 235
 236 /* vim:set shiftwidth=4 softtabstop=4 expandtab: */