Yahoo! Found the last three bugs in writing the avi stream. It now works :-D

2 files changed, 9 insertions, 4 deletions

diff --git a/noncore/multimedia/camera/avi.c b/noncore/multimedia/camera/avi.c
index 0c757c9..77aba33 100644
--- a/noncore/multimedia/camera/avi.c
+++ b/noncore/multimedia/camera/avi.c
@@ -1,235 +1,236 @@
 /**********************************************************************
 ** Copyright (C) 2003 Michael 'Mickey' Lauer.  All rights reserved.
 ** Based on work from Andrew Tridgell and the jpegtoavi project
 **
 ** This file is part of Opie Environment.
 **
 ** This file may be distributed and/or modified under the terms of the
 ** GNU General Public License version 2 as published by the Free Software
 ** Foundation and appearing in the file LICENSE.GPL included in the
 ** packaging of this file.
 **
 ** This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
 ** WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
 **
 **********************************************************************/
 
 #include "avi.h"
 
 #include <string.h>
 #include <stdio.h>
 
 int nframes;
 int totalsize;
 unsigned int* sizes;
 
 void fprint_quartet(int fd, unsigned int i)
 {
     char data[4];
 
     data[0] = (char) i%0x100;
     i /= 0x100;
     data[1] = (char) i%0x100;
     i /= 0x100;
     data[2] = (char) i%0x100;
     i /= 0x100;
     data[3] = (char) i%0x100;
 
     write( fd, &data, 4 );
 }
 
 // start writing an AVI file
 
 void avi_start(int fd, int frames)
 {
     int ofs = sizeof(struct riff_head)+
               sizeof(struct list_head)+
               sizeof(struct avi_head)+
               sizeof(struct list_head)+
               sizeof(struct stream_head)+
               sizeof(struct frame_head)+
               sizeof(struct list_head)+
               sizeof(struct dmlh_head)+
               sizeof(struct list_head);
 
     printf( "avi_start: frames = %d\n", frames );
 
     lseek(fd, ofs, SEEK_SET);
 
     nframes = 0;
     totalsize = 0;
 
     sizes = (unsigned int*) calloc( sizeof(unsigned int), frames );   // hold size of each frame
 }
 
 // add a jpeg frame to an AVI file
 void avi_add(int fd, u8 *buf, int size)
 {
     struct db_head db = {"00db", 0};
 
     printf( "avi_add: nframes = %d, totalsize = %d, size = %d\n", nframes, totalsize, size );
 
     // overwrite JFIF type with AVI1
     buf[6]='A';
     buf[7]='V';
     buf[8]='I';
     buf[9]='1';
 
     while( size%4 ) size++; // align 0 modulo 4*/
     db.size = size;
 
     write( fd, &db, sizeof(db) );
     write( fd, buf, size );
 
     sizes[nframes] = size;
 
     nframes++;
     totalsize += size;  // total frame size
 }
 
 // finish writing the AVI file - filling in the header
 void avi_end(int fd, int width, int height, int fps)
 {
     struct idx1_head idx = {"idx1", 16*nframes };
     struct db_head db = {"00db", 0};
     struct riff_head rh = { "RIFF", 0, "AVI "};
     struct list_head lh1 = {"LIST", 0, "hdrl"};
     struct avi_head ah;
     struct list_head lh2 = {"LIST", 0, "strl"};
     struct stream_head sh;
     struct frame_head fh;
     struct list_head lh3 = {"LIST", 0, "odml" };
     struct dmlh_head dh = {"dmlh", 4, nframes };
     struct list_head lh4 = {"LIST", 0, "movi"};
     int i;
     unsigned int offset = 4;
 
     printf( "avi_end: nframes = %d, fps = %d\n", nframes, fps );
 
     // write index
 
     write(fd, &idx, sizeof(idx));
 
     for ( i = 0; i < nframes; i++ )
     {
         write(fd, &db, 4 ); // only need the 00db
         fprint_quartet( fd, 18 );       // ???
         fprint_quartet( fd, offset );
         fprint_quartet( fd, sizes[i] );
-        offset += sizes[i];
+        offset += sizes[i] + 8; //+8 (for the additional header)
     }
 
     free( sizes );
 
     bzero( &ah, sizeof(ah) );
     strcpy(ah.avih, "avih");
     ah.time = 1000000 / fps;
     ah.maxbytespersec = 1000000.0*(totalsize/nframes)/ah.time;
+    ah.nframes = nframes;
     ah.numstreams = 1;
     ah.flags = AVIF_HASINDEX;
     ah.width = width;
     ah.height = height;
 
     bzero(&sh, sizeof(sh));
     strcpy(sh.strh, "strh");
     strcpy(sh.vids, "vids");
     strcpy(sh.codec, "MJPG");
     sh.scale = ah.time;
     sh.rate = 1000000;
     sh.length = nframes;
 
     bzero(&fh, sizeof(fh));
     strcpy(fh.strf, "strf");
     fh.width = width;
     fh.height = height;
     fh.planes = 1;
     fh.bitcount = 24;
     strcpy(fh.codec,"MJPG");
     fh.unpackedsize = 3*width*height;
 
     rh.size = sizeof(lh1)+sizeof(ah)+sizeof(lh2)+sizeof(sh)+
         sizeof(fh)+sizeof(lh3)+sizeof(dh)+sizeof(lh4)+
         nframes*sizeof(struct db_head)+
         totalsize + sizeof(struct idx1_head)+ (16*nframes) +4; // FIXME:16 bytes per nframe // the '4' - what for???
 
     lh1.size = 4+sizeof(ah)+sizeof(lh2)+sizeof(sh)+sizeof(fh)+sizeof(lh3)+sizeof(dh);
     ah.size = sizeof(ah)-8;
     lh2.size = 4+sizeof(sh)+sizeof(fh)+sizeof(lh3)+sizeof(dh);     //4+sizeof(sh)+sizeof(fh);
     sh.size = sizeof(sh)-8;
     fh.size = sizeof(fh)-8;
     fh.size2 = fh.size;
     lh3.size = 4+sizeof(dh);
     lh4.size = 4+ nframes*sizeof(struct db_head)+ totalsize;
 
     lseek(fd, 0, SEEK_SET);
 
     write(fd, &rh, sizeof(rh));
     write(fd, &lh1, sizeof(lh1));
     write(fd, &ah, sizeof(ah));
     write(fd, &lh2, sizeof(lh2));
     write(fd, &sh, sizeof(sh));
     write(fd, &fh, sizeof(fh));
     write(fd, &lh3, sizeof(lh3));
     write(fd, &dh, sizeof(dh));
     write(fd, &lh4, sizeof(lh4));
 }
 
 
 /* NOTE: This is not a general purpose routine - it is meant to only
    cope with AVIs saved using the other functions in this file
 void avi_explode(char *fname)
 {
 	struct riff_head rh;
 	struct list_head lh1;
 	struct avi_head ah;
 	struct list_head lh2;
 	struct stream_head sh;
 	struct frame_head fh;
 	struct list_head lh3;
 	int hsize, qsize;
 	u16 *htables = jpeg_huffman_tables(&hsize);
 	u16 *qtables = jpeg_quantisation_tables(&qsize, image_quality);
 	int fd, i;
 
 	fd = open(fname,O_RDONLY);
 	if (fd == -1) {
 		perror(fname);
 		return;
 	}
 
 	read(fd, &rh, sizeof(rh));
 	read(fd, &lh1, sizeof(lh1));
 	read(fd, &ah, sizeof(ah));
 	read(fd, &lh2, sizeof(lh2));
 	read(fd, &sh, sizeof(sh));
 	read(fd, &fh, sizeof(fh));
 	read(fd, &lh3, sizeof(lh3));
 
 	for (i=0; ; i++) {
 		u8 buf[500*1024];
 		struct db_head db;
 		char fname[100];
 		int fd2;
 
 		if (read(fd, &db, sizeof(db)) != sizeof(db) ||
 		    read(fd, buf, db.size) != db.size) break;
 
 		snprintf(fname, sizeof(fname)-1,"frame.%d", i);
 
 		fd2 = open(fname,O_WRONLY|O_CREAT, 0644);
 		if (fd2 == -1) {
 			perror(fname);
 			continue;
 		}
 		write(fd2, buf, 2);
 		write(fd2, qtables, qsize);
 		write(fd2, htables, hsize);
 		write(fd2, buf+2, db.size-2);
 		close(fd2);
 	}
 	close(fd);
 	printf("exploded %d frames\n", i);
 }
 
 */
 
diff --git a/noncore/multimedia/camera/mainwindow.cpp b/noncore/multimedia/camera/mainwindow.cpp
index e27a50e..7e60e16 100644
--- a/noncore/multimedia/camera/mainwindow.cpp
+++ b/noncore/multimedia/camera/mainwindow.cpp
@@ -115,437 +115,441 @@ void CameraMainWindow::init()
     captureFormat = "JPEG";
 
     resog = new QActionGroup( 0, "reso", true );
     resog->setToggleAction( true );
     new QAction( " 64 x  48", 0, 0, resog, 0, true );
     new QAction( "128 x  96", 0, 0, resog, 0, true );
     new QAction( "192 x 144", 0, 0, resog, 0, true );
     new QAction( "256 x 192", 0, 0, resog, 0, true );
     new QAction( "320 x 240", 0, 0, resog, 0, true );
     new QAction( "384 x 288", 0, 0, resog, 0, true );
     new QAction( "448 x 336", 0, 0, resog, 0, true );
     new QAction( "512 x 384", 0, 0, resog, 0, true );
     new QAction( "576 x 432", 0, 0, resog, 0, true );
     ( new QAction( "640 x 480", 0, 0, resog, 0, true ) )->setOn( true );
 
     qualityg = new QActionGroup( 0, "quality", true );
     qualityg->setToggleAction( true );
     new QAction( "  0 (minimal)", 0, 0, qualityg, 0, true );
     new QAction( " 25 (low)", 0, 0, qualityg, 0, true );
     ( new QAction( " 50 (good)", 0, 0, qualityg, 0, true ) )->setOn( true );
     new QAction( " 75 (better)", 0, 0, qualityg, 0, true );
     new QAction( "100 (best)", 0, 0, qualityg, 0, true );
 
     zoomg = new QActionGroup( 0, "zoom", true );
     zoomg->setToggleAction( true );
     ( new QAction( "x 1", 0, 0, zoomg, 0, true ) )->setOn( true );
     new QAction( "x 2", 0, 0, zoomg, 0, true );
 
     flipg = new QActionGroup( 0, "flip", true );
     flipg->setToggleAction( true );
     ( new QAction( "Auto (recommended)", 0, 0, flipg, 0, true ) )->setOn( true );
     new QAction( "0 (always off)", 0, 0, flipg, 0, true );
     new QAction( "X (always horizontal)", 0, 0, flipg, 0, true );
     new QAction( "Y (always vertical)", 0, 0, flipg, 0, true );
     new QAction( "* (always both)", 0, 0, flipg, 0, true );
 
     outputg = new QActionGroup( 0, "output", true );
     outputg->setToggleAction( true );
     ( new QAction( "JPEG", 0, 0, outputg, 0, true ) )->setOn( true );
     new QAction( "PNG", 0, 0, outputg, 0, true );
     new QAction( "BMP", 0, 0, outputg, 0, true );
     new QAction( "AVI", 0, 0, outputg, 0, true );
 
     connect( resog, SIGNAL( selected(QAction*) ), this, SLOT( resoMenuItemClicked(QAction*) ) );
     connect( qualityg, SIGNAL( selected(QAction*) ), this, SLOT( qualityMenuItemClicked(QAction*) ) );
     connect( zoomg, SIGNAL( selected(QAction*) ), this, SLOT( zoomMenuItemClicked(QAction*) ) );
     connect( flipg, SIGNAL( selected(QAction*) ), this, SLOT( flipMenuItemClicked(QAction*) ) );
     connect( outputg, SIGNAL( selected(QAction*) ), this, SLOT( outputMenuItemClicked(QAction*) ) );
 
 }
 
 
 void CameraMainWindow::systemMessage( const QCString& msg, const QByteArray& data )
 {
     int _newrotation;
 
     QDataStream stream( data, IO_ReadOnly );
     odebug << "received system message: " << msg << oendl;
     if ( msg == "setCurrentRotation(int)" )
     {
         stream >> _newrotation;
         odebug << "received setCurrentRotation(" << _newrotation << ")" << oendl;
 
         switch ( _newrotation )
         {
             case 270: preview->resize( QSize( 240, 288 ) ); break;
             case 180: preview->resize( QSize( 320, 208 ) ); break;
             default: QMessageBox::warning( this, "opie-camera",
                 "This rotation is not supported.\n"
                 "Supported are 180° and 270°" );
         }
 
         if ( _newrotation != _rotation )
         {
             int tmp = captureX;
             captureX = captureY;
             captureY = tmp;
             _rotation = _newrotation;
         }
 
         updateCaption();
 
     }
 }
 
 
 void CameraMainWindow::changeZoom( int zoom )
 {
     int z;
     switch ( zoom )
     {
         case 0: z = 128; break;
         case 1: z = 256; break;
         case 2: z = 512; break;
         default: assert( 0 ); break;
     }
 
     ZCameraIO::instance()->setCaptureFrame( 240, 160, z );
 }
 
 
 void CameraMainWindow::showContextMenu()
 {
     QPopupMenu reso;
     reso.setCheckable( true );
     resog->addTo( &reso );
 
     QPopupMenu quality;
     quality.setCheckable( true );
     qualityg->addTo( &quality );
 
     QPopupMenu flip;
     flip.setCheckable( true );
     flipg->addTo( &flip );
 
     QPopupMenu zoom;
     zoom.setCheckable( true );
     zoomg->addTo( &zoom );
 
     QPopupMenu output;
     output.setCheckable( true );
     outputg->addTo( &output );
 
     QPopupMenu m( this );
     m.insertItem( "&Resolution", &reso );
     m.insertItem( "&Zoom", &zoom );
     m.insertItem( "&Flip", &flip );
     m.insertItem( "&Quality", &quality );
     m.insertItem( "&Output As", &output );
 
     #ifndef QT_NO_DEBUG
     m.insertItem( "&Debug!", this, SLOT( doSomething() ) );
     #endif
 
     m.exec( QCursor::pos() );
 }
 
 
 void CameraMainWindow::resoMenuItemClicked( QAction* a )
 {
     switch ( _rotation )
     {
         case 270:
             captureY = a->text().left(3).toInt();
             captureX = a->text().right(3).toInt();
             break;
         case 180:
             captureX = a->text().left(3).toInt();
             captureY = a->text().right(3).toInt();
             break;
         default: QMessageBox::warning( this, "opie-camera",
             "This rotation is not supported.\n"
             "Supported are 180° and 270°" );
     }
     odebug << "Capture Resolution now: " << captureX << ", " << captureY << oendl;
     updateCaption();
 }
 
 
 void CameraMainWindow::qualityMenuItemClicked( QAction* a )
 {
     quality = a->text().left(3).toInt();
     odebug << "Quality now: " << quality << oendl;
     updateCaption();
 }
 
 
 void CameraMainWindow::zoomMenuItemClicked( QAction* a )
 {
     zoom = QString( a->text().at(2) ).toInt();
     odebug << "Zoom now: " << zoom << oendl;
     ZCameraIO::instance()->setZoom( zoom );
     updateCaption();
 }
 
 
 void CameraMainWindow::flipMenuItemClicked( QAction* a )
 {
     flip = QString( a->text().at(0) );
     odebug << "Flip now: " << flip << oendl;
     if ( flip == "A" )
         ZCameraIO::instance()->setFlip( ZCameraIO::AUTOMATICFLIP );
     else if ( flip == "0" )
         ZCameraIO::instance()->setFlip( ZCameraIO::XNOFLIP | ZCameraIO::YNOFLIP );
     else if ( flip == "X" )
         ZCameraIO::instance()->setFlip( ZCameraIO::XFLIP );
     else if ( flip == "Y" )
         ZCameraIO::instance()->setFlip( ZCameraIO::YFLIP );
     else if ( flip == "*" )
         ZCameraIO::instance()->setFlip( ZCameraIO::XFLIP | ZCameraIO::YFLIP );
 
     updateCaption();
 }
 
 
 void CameraMainWindow::outputMenuItemClicked( QAction* a )
 {
     captureFormat = a->text();
     odebug << "Output format now: " << captureFormat << oendl;
     updateCaption();
 }
 
 
 void CameraMainWindow::shutterClicked()
 {
     if ( captureFormat != "AVI" ) // capture one photo per shutterClick
     {
         Global::statusMessage( "CAPTURING..." );
         qApp->processEvents();
 
         odebug << "Shutter has been pressed" << oendl;
         ODevice::inst()->touchSound();
 
         performCapture( captureFormat );
     }
     else // capture video! start with one shutter click and stop with the next
     {
         !_capturing ? startVideoCapture() : stopVideoCapture();
     }
 }
 
 
 void CameraMainWindow::performCapture( const QString& format )
 {
     QString name;
     name.sprintf( "/tmp/image-%d_%d_%d_q%d.%s", _pics++, captureX, captureY, quality, (const char*) captureFormat.lower() );
     QImage i;
     ZCameraIO::instance()->captureFrame( captureX, captureY, zoom, &i );
     QImage im = i.convertDepth( 32 );
     bool result = im.save( name, format, quality );
     if ( !result )
     {
         oerr << "imageio-Problem while writing." << oendl;
         Global::statusMessage( "Error!" );
     }
     else
     {
         odebug << captureFormat << "-image has been successfully captured" << oendl;
         Global::statusMessage( "Ok." );
     }
 }
 
 
 void CameraMainWindow::startVideoCapture()
 {
     //ODevice::inst()->touchSound();
     ODevice::inst()->setLedState( Led_Mail, Led_BlinkSlow );
 
     _capturefd = ::open( CAPTUREFILE, O_WRONLY | O_CREAT | O_TRUNC );
     if ( _capturefd == -1 )
     {
         owarn << "can't open capture file: " << strerror(errno) << oendl;
         return;
     }
 
     _capturebuf = new unsigned char[captureX*captureY*2];
     _capturing = true;
     _videopics = 0;
     _framerate = 0;
     updateCaption();
     _time.start();
     preview->setRefreshingRate( 1000 );
     startTimer( 100 ); // too fast but that is ok
 }
 
 
 void CameraMainWindow::timerEvent( QTimerEvent* )
 {
     if ( !_capturing )
     {
         odebug << "timer event in CameraMainWindow without capturing video ?" << oendl;
         return;
     }
 
     odebug << "timer event during video - now capturing frame #" << _videopics+1 << oendl;
 
     ZCameraIO::instance()->captureFrame( captureX, captureY, zoom, _capturebuf );
     _videopics++;
     ::write( _capturefd, _capturebuf, captureX*captureY*2 );
     setCaption( QString().sprintf( "Capturing %dx%d @ %.2f fps %d",
         captureX, captureY, 1000.0 / (_time.elapsed()/_videopics), _videopics ) );
 }
 
 
 void CameraMainWindow::stopVideoCapture()
 {
     killTimers();
     //ODevice::inst()->touchSound();
     ODevice::inst()->setLedState( Led_Mail, Led_Off );
     _capturing = false;
     updateCaption();
     ::close( _capturefd );
     _framerate = 1000.0 / (_time.elapsed()/_videopics);
 
     postProcessVideo( CAPTUREFILE, QString().sprintf( "/tmp/video-%d_%d_%d_q%d-%dfps.avi", _videos++, captureX, captureY, quality, _framerate ) );
 
     #ifndef QT_NO_DEBUG
     preview->setRefreshingRate( 1500 );
     #else
     preview->setRefreshingRate( 200 );
     #endif
 
     //delete[] _capturebuf; //FIXME: close memory leak
 }
 
 void CameraMainWindow::postProcessVideo( const QString& infile, const QString& outfile )
 {
     preview->setRefreshingRate( 0 );
 
     /*
     unsigned char buf[153600];
 
     int fd = ::open( "/var/compile/opie/noncore/multimedia/camera/capture-320x240.dat", O_RDONLY );
     ::read( fd, &buf, 153600 );
     QImage i;
     bufferToImage( 240, 320, (unsigned char*) &buf, &i );
     QPixmap p;
     p.convertFromImage( i );
     preview->setPixmap( p );
     imageToFile( &i, "/tmp/tmpfile", "JPEG", 100 );
     return;
     */
 
     QDialog* fr = new QDialog( this, "splash", false, QWidget::WStyle_StaysOnTop ); //, false, QWidget::WStyle_NoBorder | QWidget::WStyle_Customize );
     fr->setCaption( "Please wait..." );
     QVBoxLayout* box = new QVBoxLayout( fr, 2, 2 );
     QProgressBar* bar = new QProgressBar( fr );
     bar->setCenterIndicator( true );
     bar->setTotalSteps( _videopics-1 );
     QLabel* label = new QLabel( "Post processing frame bla/bla", fr );
     box->addWidget( bar );
     box->addWidget( label );
     fr->show();
     label->show();
     bar->show();
     fr->repaint();
     qApp->processEvents();
 
     // open files
 
     int infd = ::open( (const char*) infile, O_RDONLY );
     if ( infd == -1 )
     {
         owarn << "couldn't open capture file: " << strerror(errno) << oendl;
         return;
     }
 
     int outfd = ::open( (const char*) outfile, O_CREAT | O_WRONLY | O_TRUNC, 0644 );
     if ( outfd == -1 )
     {
         owarn << "couldn't open output file: " << strerror(errno) << oendl;
         return;
     }
 
     int framesize = captureX*captureY*2;
 
     unsigned char* inbuffer = new unsigned char[ framesize ];
     QImage image;
 
     avi_start( outfd, _videopics ); // write preambel
 
     // post process
 
     for ( int i = 0; i < _videopics; ++i )
     {
         odebug << "processing frame " << i << oendl;
 
         // <gui>
         label->setText( QString().sprintf( "Post processing frame %d / %d", i+1, _videopics ) );
         bar->setProgress( i );
         bar->repaint();
         qApp->processEvents();
         // </gui>
 
         int read = ::read( infd, inbuffer, framesize );
         odebug << "read " << read << " bytes" << oendl;
         bufferToImage( captureX, captureY, inbuffer, &image );
 
         QPixmap p;
         p.convertFromImage( image );
         preview->setPixmap( p );
         preview->repaint();
         qApp->processEvents();
 
+        #ifndef QT_NO_DEBUG
+        QString tmpfilename;
+        tmpfilename.sprintf( "/tmp/test/%04d.jpg", i );
+        #else
         QString tmpfilename( "/tmp/tempfile" );
-        //tmpfilename.sprintf( "/tmp/test/%d.jpg", i );
+        #endif
 
         imageToFile( &image, tmpfilename, "JPEG", quality );
 
         QFile framefile( tmpfilename );
         if ( !framefile.open( IO_ReadOnly ) )
         {
             oerr << "can't process file: %s" << strerror(errno) << oendl;
             return; // TODO: clean up temp ressources
         }
 
         int filesize = framefile.size();
         odebug << "filesize for frame " << i << " = " << filesize << oendl;
 
         unsigned char* tempbuffer = new unsigned char[ filesize ];
         framefile.readBlock( (char*) tempbuffer, filesize );
         avi_add( outfd, tempbuffer, filesize );
         delete tempbuffer;
         framefile.close();
     }
 
     avi_end( outfd, captureX, captureY, _framerate );
 
     fr->hide();
     delete fr;
 
     updateCaption();
 
 }
 
 
 void CameraMainWindow::updateCaption()
 {
     if ( !_capturing )
         setCaption( QString().sprintf( "Opie-Camera: %dx%d %s q%d z%d (%s)", captureX, captureY, (const char*) captureFormat.lower(), quality, zoom, (const char*) flip ) );
     else
         setCaption( "Opie-Camera: => CAPTURING <=" );
 }
 
 
 #ifndef QT_NO_DEBUG
 void CameraMainWindow::doSomething()
 {
     captureX = 240;
     captureY = 320;
-    _videopics = 176;
+    _videopics = 50;
     _framerate = 5;
-    postProcessVideo( "/var/compile/opie/noncore/multimedia/camera/capture-320x240.dat",
+    postProcessVideo( "/var/compile/opie/noncore/multimedia/camera/capture.dat",
     "/tmp/output.avi" );
 }
 #endif