RIFFReader subclass:#AVIReader

	instanceVariableNames:'frames nframes frameBuffer frameBufferSize imageBuffer redPalette

		greenPalette bluePalette flags frameDelay streamType depth

		compression colorTable numColors deltaMethod'

	classVariableNames:'MAXCOLORS UnsupportedFormatErrorSignal'

	poolDictionaries:''

	category:'Graphics-Images-Support'

!

!AVIReader class methodsFor:'documentation'!

copyright

"

 COPYRIGHT (c) 1997 by eXept Software AG

              All Rights Reserved

 This software is furnished under a license and may be used

 only in accordance with the terms of that license and with the

 inclusion of the above copyright notice.   This software may not

 be provided or otherwise made available to, or used by, any

 other person.  No title to or ownership of the software is

 hereby transferred.

"

!

documentation

"

    This is not yet completed.

    its very experimental and may change.

    (will introduce a new class hierarchy based upon

     MovieReader ...).

    When used like an imageReader, #fromFile: will return

    the very first frame.

    Warning: right now, the complete movie is read and huge

    to allow stream operation ...

    [author:]

        Claus Gittinger

    [see also:]

        ImageReader

"

!

examples

"

    |reader film view tNext|

    reader := AVIReader readFile:'/usr/local/AVI/jeffmild.fli'.

    reader isNil ifTrue:[^ nil].

    film := reader images.

    view := StandardSystemView extent:(film first extent).

    view openAndWait.

    tNext := Time millisecondClockValue + (reader frameDelay).

    film do:[:frame |

        frame displayOn:view x:0 y:0.

        (Delay untilMilliseconds:tNext) wait.

        tNext := tNext + (reader frameDelay).

    ].

"

! !

!AVIReader class methodsFor:'class initialization'!

initialize

    MAXCOLORS := 256.

    UnsupportedFormatErrorSignal isNil ifTrue:[

        UnsupportedFormatErrorSignal := ErrorSignal newSignalMayProceed:true.

        UnsupportedFormatErrorSignal nameClass:self message:#unsupportedFormatErrorSignal.

    ].

    "

     AVIReader initialize

    "

    "Created: 4.4.1997 / 22:35:52 / cg"

    "Modified: 5.4.1997 / 14:37:08 / cg"

! !

!AVIReader class methodsFor:'testing'!

isValidImageFile:aFileName

    "return true, if aFileName contains an AVI-movie"

    |data1 len data3 inStream|

    inStream := self streamReadingFile:aFileName.

    inStream isNil ifTrue:[^ false].

    inStream binary.

    data1 := String new:4.

    inStream nextBytes:4 into:data1.

    len := inStream nextLongMSB:true.

    data3 := String new:4.

    inStream nextBytes:4 into:data3.

    ((data1 = 'RIFF')

    and:[data3 = 'AVI ']) ifTrue:[

        ^ true

    ].

    inStream close.

    ^ false.

    "

     AVIReader isValidImageFile:'bitmaps/magtape.xpm'    

     AVIReader isValidImageFile:'/home2/pd_stuff/movies/avi/drlair.avi'      

    "

    "Created: 4.4.1997 / 22:35:52 / cg"

    "Modified: 4.4.1997 / 23:15:39 / cg"

! !

!AVIReader methodsFor:'accessing'!

frameDelay

    "return the value of the instance variable 'frameDelay' (automatically generated)"

    ^ frameDelay

    "Created: 4.4.1997 / 22:35:52 / cg"

!

hasMultipleImages

    ^ frames size > 1

    "Modified: 4.4.1997 / 21:42:59 / cg"

    "Created: 4.4.1997 / 22:35:52 / cg"

!

images

    "return a collection of all images as represented by myself"

    |images image depth|

    depth := self bitsPerPixel.

    images := OrderedCollection new.

    frames do:[:aFrame |

        image := (Image implementorForDepth:depth) new.

        image 

            width:width 

            height:height

            photometric:photometric

            samplesPerPixel:samplesPerPixel

            bitsPerSample:bitsPerSample

            colorMap:colorMap

            bits:aFrame

            mask:mask.

        images add:image.

    ].

    ^ images

    "Modified: 20.6.1996 / 17:09:04 / cg"

    "Created: 4.4.1997 / 22:35:52 / cg"

! !

!AVIReader methodsFor:'decoders'!

AVI_Decode_CRAM:data

"

/*

 * Routine to Decode an AVI CRAM chunk

 */

#define AVI_CRAM_C1(ip,clr,rdec) { \

 *ip++ = clr; *ip++ = clr; *ip++ = clr; *ip = clr; ip -= rdec; \

 *ip++ = clr; *ip++ = clr; *ip++ = clr; *ip = clr; ip -= rdec; \

 *ip++ = clr; *ip++ = clr; *ip++ = clr; *ip = clr; ip -= rdec; \

 *ip++ = clr; *ip++ = clr; *ip++ = clr; *ip = clr; }

#define AVI_CRAM_C2(ip,flag,cA,cB,rdec) { \

  *ip++ =(flag&0x01)?(cB):(cA); *ip++ =(flag&0x02)?(cB):(cA); \

  *ip++ =(flag&0x04)?(cB):(cA); *ip   =(flag&0x08)?(cB):(cA); ip-=rdec; \

  *ip++ =(flag&0x10)?(cB):(cA); *ip++ =(flag&0x20)?(cB):(cA); \

  *ip++ =(flag&0x40)?(cB):(cA); *ip   =(flag&0x80)?(cB):(cA); }

#define AVI_CRAM_C4(ip,flag,cA0,cA1,cB0,cB1,rdec) { \

  *ip++ =(flag&0x01)?(cB0):(cA0); *ip++ =(flag&0x02)?(cB0):(cA0); \

  *ip++ =(flag&0x04)?(cB1):(cA1); *ip   =(flag&0x08)?(cB1):(cA1); ip-=rdec; \

  *ip++ =(flag&0x10)?(cB0):(cA0); *ip++ =(flag&0x20)?(cB0):(cA0); \

  *ip++ =(flag&0x40)?(cB1):(cA1); *ip   =(flag&0x80)?(cB1):(cA1); }

#define AVI_MIN_MAX_CHECK(x,y,min_x,max_x,min_y,max_y) { \

    if (x < min_x) min_x = x; if (y > max_y) max_y = y; \

    if (x > max_x) max_x = x; if (y < min_y) min_y = y; } 

#define AVI_BLOCK_INC(x,y,imagex) { x += 4; if (x>=imagex) { x=0; y -= 4; } }

#define AVI_GET_16(data,dptr) { data = *dptr++; data |= (*dptr++) << 8; }

#define AVI_CRAM_rgbC1(ip,r,g,b) { \

 *ip++=r; *ip++=g; *ip++=b; *ip++=r; *ip++=g; *ip++=b; \

 *ip++=r; *ip++=g; *ip++=b; *ip++=r; *ip++=g; *ip  =b; }

#define AVI_CRAM_rgbC2(ip,flag,rA,gA,bA,rB,gB,bB) { \

  if (flag&0x01) {*ip++=rB; *ip++=gB; *ip++=bB;} \

  else           {*ip++=rA; *ip++=gA; *ip++=bA;} \

  if (flag&0x02) {*ip++=rB; *ip++=gB; *ip++=bB;} \

  else           {*ip++=rA; *ip++=gA; *ip++=bA;} \

  if (flag&0x04) {*ip++=rB; *ip++=gB; *ip++=bB;} \

  else           {*ip++=rA; *ip++=gA; *ip++=bA;} \

  if (flag&0x08) {*ip++=rB; *ip++=gB; *ip  =bB;} \

  else           {*ip++=rA; *ip++=gA; *ip  =bA;}  }

#define AVI_CRAM_rgbC4(ip,flag,rA,gA,bA,rB,gB,bB) { \

  if (flag&0x01) {*ip++=rB; *ip++=gB; *ip++=bB;} \

  else           {*ip++=rA; *ip++=gA; *ip++=bA;} \

  if (flag&0x02) {*ip++=rB; *ip++=gB; *ip  =bB;} \

  else           {*ip++=rA; *ip++=gA; *ip  =bA;} }

#define AVI_Get_RGBColor(r,g,b,color) \

{ register ULONG _r,_g,_b; \

  _r = (color >> 10) & 0x1f; r = (_r << 3) | (_r >> 2); \

  _g = (color >>  5) & 0x1f; g = (_g << 3) | (_g >> 2); \

  _b =  color & 0x1f;        b = (_b << 3) | (_b >> 2); \

  if (xa_gamma_flag==TRUE) { r = xa_gamma_adj[r]>>8;    \

     g = xa_gamma_adj[g]>>8; b = xa_gamma_adj[b]>>8; } }

ULONG

AVI_Decode_CRAM(image,delta,dsize,chdr,map,map_flag,imagex,imagey,imaged,

                                                xs,ys,xe,ye,special,extra)

UBYTE *image;           /* Image Buffer. */

UBYTE *delta;           /* delta data. */

ULONG dsize;            /* delta size */

XA_CHDR *chdr;          /* color map info */

ULONG *map;             /* used if it's going to be remapped. */

ULONG map_flag;         /* whether or not to use remap_map info. */

ULONG imagex,imagey;    /* Size of image buffer. */

ULONG imaged;           /* Depth of Image. (IFF specific) */

ULONG *xs,*ys;          /* pos of changed area. */

ULONG *xe,*ye;          /* size of changed area. */

ULONG special;          /* Special Info. */

void *extra;            /* extra info needed to decode delta */

{

  ULONG row_dec,exitflag,changed,block_cnt;

  ULONG code0,code1;

  LONG x,y,min_x,max_x,min_y,max_y;

  UBYTE *dptr;

  changed = 0;

  max_x = max_y = 0;    min_x = imagex; min_y = imagey;

  dptr = delta;

  row_dec = imagex + 3;

  x = 0;

  y = imagey - 1;

  exitflag = 0;

  block_cnt = ((imagex * imagey) >> 4) + 1;

  if (map_flag == TRUE)

  {

    if (x11_bytes_pixel == 4)

    {

      while(!!exitflag)

      {

        code0 =  *dptr++;       code1 =  *dptr++;       block_cnt--;

        if ( ((code1==0) && (code0==0) && !!block_cnt) || (y<0)) exitflag = 1;

        else

        {

          if ((code1 >= 0x84) && (code1 <= 0x87)) /* skip */

          { ULONG skip = ((code1 - 0x84) << 8) + code0;

            block_cnt -= (skip-1); while(skip--) AVI_BLOCK_INC(x,y,imagex);

          }

          else /* single block encoded */

          {

            if (code1 >= 0x90) /* 8 color quadrant encoding */

            { ULONG cA0,cA1,cB0,cB1;

              ULONG *i_ptr = (ULONG *)(image + ((y * imagex + x) << 2) );

              cB0 = (ULONG)map[*dptr++];  cA0 = (ULONG)map[*dptr++];

              cB1 = (ULONG)map[*dptr++];  cA1 = (ULONG)map[*dptr++];

              AVI_CRAM_C4(i_ptr,code0,cA0,cA1,cB0,cB1,row_dec); i_ptr -=row_dec;

              cB0 = (ULONG)map[*dptr++];  cA0 = (ULONG)map[*dptr++];

              cB1 = (ULONG)map[*dptr++];  cA1 = (ULONG)map[*dptr++];

              AVI_CRAM_C4(i_ptr,code1,cA0,cA1,cB0,cB1,row_dec);

            } else if (code1 < 0x80) /* 2 color encoding */

            { register ULONG clr_A,clr_B;

              ULONG *i_ptr = (ULONG *)(image + ((y * imagex + x) << 2) );

              clr_B = (ULONG)map[*dptr++];   clr_A = (ULONG)map[*dptr++];

              AVI_CRAM_C2(i_ptr,code0,clr_A,clr_B,row_dec); i_ptr -= row_dec;

              AVI_CRAM_C2(i_ptr,code1,clr_A,clr_B,row_dec);

            }

            else /* 1 color encoding */

            { ULONG clr = (ULONG)map[code0]; 

              ULONG *i_ptr = (ULONG *)(image + ((y * imagex + x) << 2) );

              AVI_CRAM_C1(i_ptr,clr,row_dec);

            }

            AVI_MIN_MAX_CHECK(x,y,min_x,max_x,min_y,max_y);

            changed = 1; AVI_BLOCK_INC(x,y,imagex);

          } /* end of single block */

        } /* end of not term code */

      } /* end of not while exit */

    } /* end of 4 bytes pixel */

    else if (x11_bytes_pixel == 2)

    {

      while(!!exitflag)

      {

        code0 =  *dptr++;       code1 =  *dptr++;       block_cnt--;

        if ( ((code1==0) && (code0==0) && !!block_cnt) || (y<0)) exitflag = 1;

        else

        {

          if ((code1 >= 0x84) && (code1 <= 0x87)) /* skip */

          { ULONG skip = ((code1 - 0x84) << 8) + code0;

            block_cnt -= (skip-1); while(skip--) AVI_BLOCK_INC(x,y,imagex);

          } else /* single block encoded */

          {

            if (code1 >= 0x90) /* 8 color quadrant encoding */

            {

              USHORT cA0,cA1,cB0,cB1;

              USHORT *i_ptr = (USHORT *)(image + ((y * imagex + x) << 1) );

              cB0 = map[*dptr++];  cA0 = map[*dptr++];

              cB1 = map[*dptr++];  cA1 = map[*dptr++];

              AVI_CRAM_C4(i_ptr,code0,cA0,cA1,cB0,cB1,row_dec); i_ptr -=row_dec;

              cB0 = map[*dptr++];  cA0 = map[*dptr++];

              cB1 = map[*dptr++];  cA1 = map[*dptr++];

              AVI_CRAM_C4(i_ptr,code1,cA0,cA1,cB0,cB1,row_dec);

            } /* end of 8 color quadrant encoding */

            else if (code1 < 0x80) /* 2 color encoding */

            { USHORT clr_A,clr_B;

              USHORT *i_ptr = (USHORT *)(image + ((y * imagex + x) << 1) );

              clr_B = (USHORT)map[*dptr++];   clr_A = (USHORT)map[*dptr++];

              AVI_CRAM_C2(i_ptr,code0,clr_A,clr_B,row_dec); i_ptr -= row_dec;

              AVI_CRAM_C2(i_ptr,code1,clr_A,clr_B,row_dec);

            } /* end of 2 color */

            else /* 1 color encoding */

            { USHORT clr = (USHORT)map[code0];

              USHORT *i_ptr = (USHORT *)(image + ((y * imagex + x) << 1) );

              AVI_CRAM_C1(i_ptr,clr,row_dec);

            }

            AVI_MIN_MAX_CHECK(x,y,min_x,max_x,min_y,max_y);

            changed = 1; AVI_BLOCK_INC(x,y,imagex);

          } /* end of single block */

        } /* end of not term code */

      } /* end of not while exit */

    } /* end of 2 bytes pixel */

    else /* (x11_bytes_pixel == 1) */

    {

      while(!!exitflag)

      {

        code0 =  *dptr++;       code1 =  *dptr++;       block_cnt--;

        if ( ((code1==0) && (code0==0) && !!block_cnt) || (y<0)) exitflag = 1;

        else

        {

          if ((code1 >= 0x84) && (code1 <= 0x87)) /* skip */

          { ULONG skip = ((code1 - 0x84) << 8) + code0;

            block_cnt -= (skip-1); while(skip--) AVI_BLOCK_INC(x,y,imagex);

          } else /* single block encoded */

          { 

            if (code1 >= 0x90) /* 8 color quadrant encoding */

            { UBYTE cA0,cA1,cB0,cB1;

              UBYTE *i_ptr = (UBYTE *)(image + y * imagex + x);

              cB0 = (UBYTE)map[*dptr++];  cA0 = (UBYTE)map[*dptr++];

              cB1 = (UBYTE)map[*dptr++];  cA1 = (UBYTE)map[*dptr++];

              AVI_CRAM_C4(i_ptr,code0,cA0,cA1,cB0,cB1,row_dec); i_ptr -=row_dec;

              cB0 = (UBYTE)map[*dptr++];  cA0 = (UBYTE)map[*dptr++];

              cB1 = (UBYTE)map[*dptr++];  cA1 = (UBYTE)map[*dptr++];

              AVI_CRAM_C4(i_ptr,code1,cA0,cA1,cB0,cB1,row_dec);

            } 

            else if (code1 < 0x80) /* 2 color encoding */

            { UBYTE clr_A,clr_B;

              UBYTE *i_ptr = (UBYTE *)(image + y * imagex + x);

              clr_B = (UBYTE)map[*dptr++];   clr_A = (UBYTE)map[*dptr++];

              AVI_CRAM_C2(i_ptr,code0,clr_A,clr_B,row_dec); i_ptr -= row_dec;

              AVI_CRAM_C2(i_ptr,code1,clr_A,clr_B,row_dec);

            }

            else /* 1 color encoding */

            { UBYTE clr = (UBYTE)map[code0];

              UBYTE *i_ptr = (UBYTE *)(image + y * imagex + x);

              AVI_CRAM_C1(i_ptr,clr,row_dec);

            }

            AVI_MIN_MAX_CHECK(x,y,min_x,max_x,min_y,max_y);

            changed = 1; AVI_BLOCK_INC(x,y,imagex);

          } /* end of single block */

        } /* end of not term code */

      } /* end of not while exit */

    } /* end of 1 bytes pixel */

  } /* end of map is TRUE */

  else

  {

      while(!!exitflag)

      {

        code0 =  *dptr++;       code1 =  *dptr++;       block_cnt--;

        if ( ((code1==0) && (code0==0) && !!block_cnt) || (y<0)) exitflag = 1;

        else

        {

          if ((code1 >= 0x84) && (code1 <= 0x87)) /* skip */

          { ULONG skip = ((code1 - 0x84) << 8) + code0;

            block_cnt -= (skip-1); while(skip--) AVI_BLOCK_INC(x,y,imagex);

          } else /* single block encoded */

          {

            if (code1 >= 0x90) /* 8 color quadrant encoding */

            {

              UBYTE cA0,cA1,cB0,cB1;

              UBYTE *i_ptr = (UBYTE *)(image + y * imagex + x);

              cB0 = (UBYTE)*dptr++;  cA0 = (UBYTE)*dptr++;

              cB1 = (UBYTE)*dptr++;  cA1 = (UBYTE)*dptr++;

              AVI_CRAM_C4(i_ptr,code0,cA0,cA1,cB0,cB1,row_dec); i_ptr -=row_dec;

              cB0 = (UBYTE)*dptr++;  cA0 = (UBYTE)*dptr++;

              cB1 = (UBYTE)*dptr++;  cA1 = (UBYTE)*dptr++;

              AVI_CRAM_C4(i_ptr,code1,cA0,cA1,cB0,cB1,row_dec);

            } 

            else if (code1 < 0x80) /* 2 color encoding */

            { UBYTE clr_A,clr_B;

              UBYTE *i_ptr = (UBYTE *)(image + y * imagex + x);

              clr_B = (UBYTE)*dptr++;   clr_A = (UBYTE)*dptr++;

              AVI_CRAM_C2(i_ptr,code0,clr_A,clr_B,row_dec); i_ptr -= row_dec;

              AVI_CRAM_C2(i_ptr,code1,clr_A,clr_B,row_dec);

            } /* end of 2 color */

            else /* 1 color encoding */

            {

              UBYTE clr = (UBYTE)code0;

              UBYTE *i_ptr = (UBYTE *)(image + y * imagex + x);

              AVI_CRAM_C1(i_ptr,clr,row_dec);

            }

            AVI_MIN_MAX_CHECK(x,y,min_x,max_x,min_y,max_y);

            changed = 1; AVI_BLOCK_INC(x,y,imagex);

          } /* end of single block */

        } /* end of not term code */

      } /* end of not while exit */

  }

  if (xa_optimize_flag == TRUE)

  {

    if (changed) { *xs=min_x; *ys=min_y - 3; *xe=max_x + 4; *ye=max_y + 1; }

    else  { *xs = *ys = *xe = *ye = 0; return(ACT_DLTA_NOP); }

  }

  else { *xs = *ys = 0; *xe = imagex; *ye = imagey; }

  if (map_flag) return(ACT_DLTA_MAPD);

  else return(ACT_DLTA_NORM);

}

"

    'AVI_Decode_CRAM' infoPrint.

    "

     AVIReader fromFile:'/home2/pd_stuff/movies/avi/hangldm.avi'      

    "

    "Created: 5.4.1997 / 16:06:10 / cg"

    "Modified: 11.4.1997 / 14:03:32 / cg"

! !

!AVIReader methodsFor:'helpers'!

alignHeightTo:multiple

    "make height a multiple of the argument"

    height := multiple * ((height + multiple - 1) // multiple).

    "Modified: 5.4.1997 / 14:47:36 / cg"

!

alignTo:multiple

    "make width & height a multiple of the argument"

    width := multiple * ((width + multiple - 1) // multiple).

    height := multiple * ((height + multiple - 1) // multiple).

    "Created: 5.4.1997 / 14:47:54 / cg"

!

alignWidthTo:multiple

    "make height a multiple of the argument"

    width := multiple * ((width + multiple - 1) // multiple).

    "Created: 5.4.1997 / 14:47:16 / cg"

    "Modified: 5.4.1997 / 14:47:41 / cg"

! !

!AVIReader methodsFor:'reading from stream'!

fromStream:aStream

    "read a AVI-movie from aStream. Return a frame sequence."

    inStream := aStream.

    inStream binary.

    [inStream atEnd] whileFalse:[

        self getChunk

    ].

    "/ return the first frame as image

    colorMap := Colormap 

                    redVector:redPalette 

                    greenVector:greenPalette 

                    blueVector:bluePalette.

    photometric := #palette.

    samplesPerPixel := 1.

    bitsPerSample := #(8).

    frames notNil ifTrue:[

        data := frames at:1.

    ]

    "

     AVIReader fromFile:'/home2/pd_stuff/movies/avi/drlair.avi'      

     AVIReader fromFile:'/home2/pd_stuff/movies/avi/hangldm.avi'      

    "

    "Created: 4.4.1997 / 22:35:53 / cg"

    "Modified: 5.4.1997 / 14:34:54 / cg"

! !