dlls/avifil32/tmpfile.c

   1 /*
   2  * Copyright 2003 Michael Günnewig
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #include <stdarg.h>
  20
  21 #include "windef.h"
  22 #include "winbase.h"
  23 #include "wingdi.h"
  24 #include "winuser.h"
  25 #include "winerror.h"
  26 #include "vfw.h"
  27
  28 #include "avifile_private.h"
  29 #include "extrachunk.h"
  30
  31 #include "wine/debug.h"
  32
  33 WINE_DEFAULT_DEBUG_CHANNEL(avifile);
  34
  35 /***********************************************************************/
  36
  37 static HRESULT WINAPI ITmpFile_fnQueryInterface(IAVIFile* iface,REFIID refiid,LPVOID *obj);
  38 static ULONG   WINAPI ITmpFile_fnAddRef(IAVIFile* iface);
  39 static ULONG   WINAPI ITmpFile_fnRelease(IAVIFile* iface);
  40 static HRESULT WINAPI ITmpFile_fnInfo(IAVIFile*iface,AVIFILEINFOW*afi,LONG size);
  41 static HRESULT WINAPI ITmpFile_fnGetStream(IAVIFile*iface,PAVISTREAM*avis,DWORD fccType,LONG lParam);
  42 static HRESULT WINAPI ITmpFile_fnCreateStream(IAVIFile*iface,PAVISTREAM*avis,AVISTREAMINFOW*asi);
  43 static HRESULT WINAPI ITmpFile_fnWriteData(IAVIFile*iface,DWORD ckid,LPVOID lpData,LONG size);
  44 static HRESULT WINAPI ITmpFile_fnReadData(IAVIFile*iface,DWORD ckid,LPVOID lpData,LONG *size);
  45 static HRESULT WINAPI ITmpFile_fnEndRecord(IAVIFile*iface);
  46 static HRESULT WINAPI ITmpFile_fnDeleteStream(IAVIFile*iface,DWORD fccType,LONG lParam);
  47
  48 static const struct IAVIFileVtbl itmpft = {
  49   ITmpFile_fnQueryInterface,
  50   ITmpFile_fnAddRef,
  51   ITmpFile_fnRelease,
  52   ITmpFile_fnInfo,
  53   ITmpFile_fnGetStream,
  54   ITmpFile_fnCreateStream,
  55   ITmpFile_fnWriteData,
  56   ITmpFile_fnReadData,
  57   ITmpFile_fnEndRecord,
  58   ITmpFile_fnDeleteStream
  59 };
  60
  61 typedef struct _ITmpFileImpl {
  62   /* IUnknown stuff */
  63   const IAVIFileVtbl *lpVtbl;
  64   LONG         ref;
  65
  66   /* IAVIFile stuff */
  67   AVIFILEINFOW  fInfo;
  68   PAVISTREAM   *ppStreams;
  69 } ITmpFileImpl;
  70
  71 PAVIFILE AVIFILE_CreateAVITempFile(int nStreams, const PAVISTREAM *ppStreams) {
  72   ITmpFileImpl *tmpFile;
  73   int           i;
  74
  75   tmpFile = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(ITmpFileImpl));
  76   if (tmpFile == NULL)
  77     return NULL;
  78
  79   tmpFile->lpVtbl = &itmpft;
  80   tmpFile->ref    = 1;
  81   memset(&tmpFile->fInfo, 0, sizeof(tmpFile->fInfo));
  82
  83   tmpFile->fInfo.dwStreams = nStreams;
  84   tmpFile->ppStreams = HeapAlloc(GetProcessHeap(), 0, nStreams * sizeof(PAVISTREAM));
  85   if (tmpFile->ppStreams == NULL) {
  86     HeapFree(GetProcessHeap(), 0, tmpFile);
  87     return NULL;
  88   }
  89
  90   for (i = 0; i < nStreams; i++) {
  91     AVISTREAMINFOW sInfo;
  92
  93     tmpFile->ppStreams[i] = ppStreams[i];
  94
  95     AVIStreamAddRef(ppStreams[i]);
  96     AVIStreamInfoW(ppStreams[i], &sInfo, sizeof(sInfo));
  97     if (i == 0) {
  98       tmpFile->fInfo.dwScale = sInfo.dwScale;
  99       tmpFile->fInfo.dwRate  = sInfo.dwRate;
 100       if (!sInfo.dwScale || !sInfo.dwRate) {
 101         tmpFile->fInfo.dwScale = 1;
 102         tmpFile->fInfo.dwRate  = 100;
 103       }
 104     }
 105
 106     if (tmpFile->fInfo.dwSuggestedBufferSize < sInfo.dwSuggestedBufferSize)
 107       tmpFile->fInfo.dwSuggestedBufferSize = sInfo.dwSuggestedBufferSize;
 108
 109     {
 110       register DWORD tmp;
 111
 112       tmp = MulDiv(AVIStreamSampleToTime(ppStreams[i], sInfo.dwLength),
 113                    tmpFile->fInfo.dwScale, tmpFile->fInfo.dwRate * 1000);
 114       if (tmpFile->fInfo.dwLength < tmp)
 115         tmpFile->fInfo.dwLength = tmp;
 116
 117       tmp = sInfo.rcFrame.right - sInfo.rcFrame.left;
 118       if (tmpFile->fInfo.dwWidth < tmp)
 119         tmpFile->fInfo.dwWidth = tmp;
 120       tmp = sInfo.rcFrame.bottom - sInfo.rcFrame.top;
 121       if (tmpFile->fInfo.dwHeight < tmp)
 122         tmpFile->fInfo.dwHeight = tmp;
 123     }
 124   }
 125
 126   return (PAVIFILE)tmpFile;
 127 }
 128
 129 static HRESULT WINAPI ITmpFile_fnQueryInterface(IAVIFile *iface, REFIID refiid,
 130                                                 LPVOID *obj)
 131 {
 132   ITmpFileImpl *This = (ITmpFileImpl *)iface;
 133
 134   TRACE("(%p,%s,%p)\n", This, debugstr_guid(refiid), obj);
 135
 136   if (IsEqualGUID(&IID_IUnknown, refiid) ||
 137       IsEqualGUID(&IID_IAVIFile, refiid)) {
 138     *obj = iface;
 139     IAVIFile_AddRef(iface);
 140
 141     return S_OK;
 142   }
 143
 144   return OLE_E_ENUM_NOMORE;
 145 }
 146
 147 static ULONG   WINAPI ITmpFile_fnAddRef(IAVIFile *iface)
 148 {
 149   ITmpFileImpl *This = (ITmpFileImpl *)iface;
 150   ULONG ref = InterlockedIncrement(&This->ref);
 151
 152   TRACE("(%p) -> %d\n", iface, ref);
 153
 154   return ref;
 155 }
 156
 157 static ULONG   WINAPI ITmpFile_fnRelease(IAVIFile *iface)
 158 {
 159   ITmpFileImpl *This = (ITmpFileImpl *)iface;
 160   ULONG ref = InterlockedDecrement(&This->ref);
 161
 162   TRACE("(%p) -> %d\n", iface, ref);
 163
 164   if (!ref) {
 165     unsigned int i;
 166
 167     for (i = 0; i < This->fInfo.dwStreams; i++) {
 168       if (This->ppStreams[i] != NULL) {
 169         AVIStreamRelease(This->ppStreams[i]);
 170
 171         This->ppStreams[i] = NULL;
 172       }
 173     }
 174
 175     HeapFree(GetProcessHeap(), 0, This);
 176     return 0;
 177   }
 178
 179   return ref;
 180 }
 181
 182 static HRESULT WINAPI ITmpFile_fnInfo(IAVIFile *iface,
 183                                       AVIFILEINFOW *afi, LONG size)
 184 {
 185   ITmpFileImpl *This = (ITmpFileImpl *)iface;
 186
 187   TRACE("(%p,%p,%d)\n",iface,afi,size);
 188
 189   if (afi == NULL)
 190     return AVIERR_BADPARAM;
 191   if (size < 0)
 192     return AVIERR_BADSIZE;
 193
 194   memcpy(afi, &This->fInfo, min((DWORD)size, sizeof(This->fInfo)));
 195
 196   if ((DWORD)size < sizeof(This->fInfo))
 197     return AVIERR_BUFFERTOOSMALL;
 198   return AVIERR_OK;
 199 }
 200
 201 static HRESULT WINAPI ITmpFile_fnGetStream(IAVIFile *iface, PAVISTREAM *avis,
 202                                            DWORD fccType, LONG lParam)
 203 {
 204   ITmpFileImpl *This = (ITmpFileImpl *)iface;
 205
 206   ULONG nStream = (ULONG)-1;
 207
 208   TRACE("(%p,%p,0x%08X,%d)\n", iface, avis, fccType, lParam);
 209
 210   if (avis == NULL || lParam < 0)
 211     return AVIERR_BADPARAM;
 212
 213   if (fccType != streamtypeANY) {
 214     /* search the number of the specified stream */
 215     ULONG i;
 216
 217     for (i = 0; i < This->fInfo.dwStreams; i++) {
 218       AVISTREAMINFOW sInfo;
 219       HRESULT        hr;
 220
 221       hr = AVIStreamInfoW(This->ppStreams[i], &sInfo, sizeof(sInfo));
 222       if (FAILED(hr))
 223         return hr;
 224
 225       if (sInfo.fccType == fccType) {
 226         if (lParam == 0) {
 227           nStream = i;
 228           break;
 229         } else
 230           lParam--;
 231       }
 232     }
 233   } else
 234     nStream = lParam;
 235
 236   /* Does the requested stream exist ? */
 237   if (nStream < This->fInfo.dwStreams && This->ppStreams[nStream] != NULL) {
 238     *avis = This->ppStreams[nStream];
 239     AVIStreamAddRef(*avis);
 240
 241     return AVIERR_OK;
 242   }
 243
 244   /* Sorry, but the specified stream doesn't exist */
 245   return AVIERR_NODATA;
 246 }
 247
 248 static HRESULT WINAPI ITmpFile_fnCreateStream(IAVIFile *iface,PAVISTREAM *avis,
 249                                               AVISTREAMINFOW *asi)
 250 {
 251   TRACE("(%p,%p,%p)\n",iface,avis,asi);
 252
 253   return AVIERR_UNSUPPORTED;
 254 }
 255
 256 static HRESULT WINAPI ITmpFile_fnWriteData(IAVIFile *iface, DWORD ckid,
 257                                            LPVOID lpData, LONG size)
 258 {
 259   TRACE("(%p,0x%08X,%p,%d)\n", iface, ckid, lpData, size);
 260
 261   return AVIERR_UNSUPPORTED;
 262 }
 263
 264 static HRESULT WINAPI ITmpFile_fnReadData(IAVIFile *iface, DWORD ckid,
 265                                           LPVOID lpData, LONG *size)
 266 {
 267   TRACE("(%p,0x%08X,%p,%p)\n", iface, ckid, lpData, size);
 268
 269   return AVIERR_UNSUPPORTED;
 270 }
 271
 272 static HRESULT WINAPI ITmpFile_fnEndRecord(IAVIFile *iface)
 273 {
 274   TRACE("(%p)\n",iface);
 275
 276   return AVIERR_OK;
 277 }
 278
 279 static HRESULT WINAPI ITmpFile_fnDeleteStream(IAVIFile *iface, DWORD fccType,
 280                                               LONG lParam)
 281 {
 282   TRACE("(%p,0x%08X,%d)\n", iface, fccType, lParam);
 283
 284   return AVIERR_UNSUPPORTED;
 285 }