avi_map.c

Go to the documentation of this file.

/* --
OpenFX version 2.0 - Modelling, Animation and Rendering Package
Copyright (C) 2000 - 2007 OpenFX Development Team
-- */

/*  AVI_MAP.C  */

// This does not work for  DV encoded video but does for  Uncompressed and  Cinepak codecs.

#include <stdlib.h>
#include <stdio.h>
#include <windows.h>
#include <vfw.h>

#include "../common/mem.h"

typedef struct SCREENBUFFER {
 unsigned char A,R,G,B;
} fullscreenbuffer;

static long LoadBmpImageMap(char *fn, long *xxx, long *yyy, BOOL size_only,
   unsigned char *pR, unsigned char *pG, unsigned char *pB, unsigned char *source_buffer);
static HINSTANCE hThisInstance;

#if __WATCOMC__
int APIENTRY LibMain(HANDLE hDLL, DWORD dwReason, LPVOID lpReserved){
#else
BOOL WINAPI DllMain(HANDLE hDLL, DWORD dwReason, LPVOID lpReserved){
#endif
  switch (dwReason) {
    case DLL_PROCESS_ATTACH: {
      hThisInstance=hDLL;
      break;
    }
    case DLL_PROCESS_DETACH:
      break;
  }
  return TRUE;
}

BOOL _PutExternalImage
 (char *szOutFileName,
  long nx, long ny,
  long quality,
  fullscreenbuffer *image){
 return FALSE;
}

BOOL _GetExternalImageSize
 (char *filename,
  long *x,
  long *y,
  long frame){
 LONG hr; 
 AVIFILEINFO fi,*pfi;
 PAVIFILE pfile; 
 pfi=&fi;
 hr = AVIFileOpen(&pfile, filename,OF_SHARE_DENY_WRITE,0L); 
 if (hr != 0)return FALSE;
 if(AVIFileInfo(pfile,pfi,sizeof(AVIFILEINFO)) != 0){AVIFileRelease(pfile); return FALSE;}
 *y = pfi->dwHeight;
 *x = pfi->dwWidth;
 AVIFileRelease(pfile);  // closes the file 
 return TRUE;
}

BOOL _GetExternalImage
 (char *filename,
  long x, long y,
  unsigned char *Red,
  unsigned char *Green,
  unsigned char *Blue,
  long frame){
 BOOL bWorked=FALSE;
 LONG hr; 
 LPSTR pformat;
 AVIFILEINFO fi,*pfi;
 PAVIFILE pfile; 
 PAVISTREAM psin[5];
 AVISTREAMINFO avis; 
 LPBITMAPINFOHEADER lpBmi;
 BITMAPINFOHEADER Bmi;
 long i,j,ff,size,vstream = -1;
 long nsamples;
 char buffer[256];
 unsigned char *r,*g,*b,*pp,*prow;
 pfi=&fi;
 hr = AVIFileOpen(&pfile, filename,OF_SHARE_DENY_WRITE,0L); 
 if (hr != 0)return FALSE;
 if(AVIFileInfo(pfile,pfi,sizeof(AVIFILEINFO)) != 0){AVIFileRelease(pfile); return FALSE;}
 if(y != pfi->dwHeight){AVIFileRelease(pfile); return FALSE;}
 if(x != pfi->dwWidth){AVIFileRelease(pfile); return FALSE;}
 for(i=0;i<5;i++){
   psin[i]=NULL;
   if(AVIFileGetStream(pfile,&psin[i],0,i) == AVIERR_OK){
     AVIStreamInfo(psin[i],&avis,sizeof(avis));
     if(avis.fccType == streamtypeVIDEO){
       vstream=i;
       AVIStreamFormatSize(psin[i],0,&size);
       if((pformat = (LPSTR)X__Malloc(size)) != NULL){
         if(AVIStreamReadFormat(psin[i],0,pformat,&size) == AVIERR_OK){
           lpBmi=(LPBITMAPINFOHEADER)pformat;
           sprintf(buffer,"Size (%ld %ld) Bits %ld",lpBmi->biWidth,lpBmi->biHeight,
                   (long)(lpBmi->biBitCount));
           memcpy(&Bmi,pformat,sizeof(BITMAPINFOHEADER)); 
           //MessageBox(NULL,buffer,"AVI_FILE",MB_OK);
           nsamples=AVIStreamEnd(psin[i]);
           sprintf(buffer,"%ld Frames (%0.3lfs)",nsamples,
           (double)AVIStreamSampleToTime(psin[i],nsamples)*0.001);
           //MessageBox(NULL,buffer,"AVI_FILE",MB_OK);
         }
         X__Free(pformat);
       }
       break;
     }
   }
 } 
 ff=frame%nsamples - 1; ff=max(0,ff); ff=min(ff,nsamples);
 if(vstream >= 0){
   LPBITMAPINFOHEADER lpbi; 
   PGETFRAME pgf;
   //MessageBox(NULL,"About to open stream","AVI_FILE",MB_OK);
   lpbi=&Bmi;
   memset(lpbi,0,sizeof(BITMAPINFOHEADER));
   lpbi->biSize=sizeof(BITMAPINFOHEADER);
   lpbi->biWidth=(DWORD)x;
   lpbi->biHeight=(DWORD)y;
   lpbi->biPlanes=1;
   lpbi->biBitCount=24;
   lpbi->biCompression=BI_RGB;
   lpbi->biSizeImage = 0;
   lpbi->biXPelsPerMeter=0;
   lpbi->biYPelsPerMeter=0;
   lpbi->biClrUsed=0;
   lpbi->biClrImportant=0;
   pgf=AVIStreamGetFrameOpen(psin[vstream],lpbi);
   if(pgf != NULL){
     //MessageBox(NULL,"StreamOpen","AVI_FILE",MB_OK);
     pp=AVIStreamGetFrame(pgf,ff); 
     if(pp != NULL){
       lpBmi=(LPBITMAPINFOHEADER)pp;
       sprintf(buffer,"Frame %ld Size (%ld %ld) Bits %ld colours %ld  size %ld",ff,lpBmi->biWidth,lpBmi->biHeight,
                   (long)(lpBmi->biBitCount),(long)lpBmi->biClrUsed,(long)lpBmi->biSizeImage);
//       MessageBox(NULL,buffer,"Writing Output",MB_OK);
       r=Red; g=Green; b=Blue; pp += sizeof(BITMAPINFOHEADER);
       for(i=0;i<y;i++){
         prow=(pp+x*(y-i-1)*3);
         for(j=0;j<x;j++){
          *b++ = *prow++;
          *g++ = *prow++;
          *r++ = *prow++;
         }
       }
       bWorked=TRUE;
     }
     AVIStreamGetFrameClose(pgf);
   }
   AVIStreamRelease(psin[vstream]);
 }
 AVIFileRelease(pfile);  
 return bWorked;
}