path: root/noncore/multimedia/opierec/wavFile.cpp

//wavFile.cpp
#include "wavFile.h"
#include "qtrec.h"

/* OPIE */
#include <opie2/odebug.h>
#include <qpe/config.h>
using namespace Opie::Core;

/* QT */
#include <qmessagebox.h>
#include <qdir.h>

/* STD */
#include <errno.h>
#include <sys/time.h>
#include <sys/types.h>
#include <sys/vfs.h>
#include <fcntl.h>
#include <math.h>
#include <mntent.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

WavFile::WavFile( QObject * parent,const QString &fileName, bool makeNwFile, int sampleRate,
                  int channels, int resolution, int format )
        : QObject( parent)
{
		owarn << "new wave file: " << fileName << oendl;
    bool b =  makeNwFile;
    wavSampleRate=sampleRate;
    wavFormat=format;
    wavChannels=channels;
    wavResolution=resolution;
    useTmpFile=false;
    if( b) {
        newFile();
    } else {
        openFile(fileName);
    }
}

bool WavFile::newFile() {

//  odebug << "Set up new file" << oendl;
  Config cfg("OpieRec");
  cfg.setGroup("Settings");

  currentFileName=cfg.readEntry("directory",QDir::homeDirPath());
  QString date;
  QDateTime dt = QDateTime::currentDateTime();
  date = dt.toString();//TimeString::dateString( QDateTime::currentDateTime(),false,true);
  date.replace(QRegExp("'"),"");
  date.replace(QRegExp(" "),"_");
  date.replace(QRegExp(":"),"-");
  date.replace(QRegExp(","),"");

  QString currentFile=date;
  if(currentFileName.right(1).find("/",0,true) == -1)
    currentFileName += "/" + date;
  else
    currentFileName += date;
  currentFileName+=".wav";

//  odebug << "set up file for recording: "+currentFileName << oendl;
  char pointer[] = "/tmp/opierec-XXXXXX";
  int fd = 0;

  if( currentFileName.find("/mnt",0,true) == -1
      && currentFileName.find("/tmp",0,true) == -1 ) {
    // if destination file is most likely in flash (assuming jffs2)
    // we have to write to a different filesystem first

    useTmpFile = true;
    if(( fd = mkstemp( pointer)) < 0 ) {
       perror("mkstemp failed");
       return false;
    }

//    odebug << "Opening tmp file " << pointer << "" << oendl;
    track.setName( pointer);

  } else { //just use regular file.. no moving

    useTmpFile = false;
    track.setName( currentFileName);
  }
  if(!track.open( IO_ReadWrite | IO_Truncate)) {
    QString errorMsg=(QString)strerror(errno);
    odebug << errorMsg << oendl;
    QMessageBox::message("Note", "Error opening file.\n" +errorMsg);

    return false;
  } else {
    setWavHeader( track.handle() , &hdr);
  }
return true;
}

WavFile::~WavFile() {

    closeFile();
}

void WavFile::closeFile() {
    if(track.isOpen())
        track.close();
}

int WavFile::openFile(const QString &currentFileName) {
		qWarning("open play file "+currentFileName);;
    closeFile();

		track.setName(currentFileName);

		if(!track.open(IO_ReadOnly)) {
				QString errorMsg=(QString)strerror(errno);
				odebug << "<<<<<<<<<<< "+errorMsg+currentFileName << oendl;
				QMessageBox::message("Note", "Error opening file.\n" +errorMsg);
				return -1;
		} else {
				parseWavHeader( track.handle());
		}
		return track.handle();
}

bool WavFile::setWavHeader(int fd, wavhdr *hdr) {

  strncpy((*hdr).riffID, "RIFF", 4); // RIFF
  strncpy((*hdr).wavID, "WAVE", 4); //WAVE
  strncpy((*hdr).fmtID, "fmt ", 4); // fmt
  (*hdr).fmtLen = 16; // format length = 16

  if( wavFormat == WAVE_FORMAT_PCM) {
    (*hdr).fmtTag = 1; // PCM
//    odebug << "set header  WAVE_FORMAT_PCM" << oendl;
  }
  else {
    (*hdr).fmtTag = WAVE_FORMAT_DVI_ADPCM; //intel ADPCM
 //    odebug << "set header  WAVE_FORMAT_DVI_ADPCM" << oendl;
  }

  //  (*hdr).nChannels = 1;//filePara.channels;// ? 2 : 1*/; // channels
  (*hdr).nChannels = wavChannels;// ? 2 : 1*/; // channels

  (*hdr).sampleRate = wavSampleRate; //samples per second
  (*hdr).avgBytesPerSec = (wavSampleRate)*( wavChannels*(wavResolution/8)); // bytes per second
  (*hdr).nBlockAlign = wavChannels*( wavResolution/8); //block align
  (*hdr).bitsPerSample = wavResolution; //bits per sample 8, or 16

  strncpy((*hdr).dataID, "data", 4);

  write( fd,hdr, sizeof(*hdr));
//   owarn << "writing header: bitrate " << wavResolution << ", samplerate " << wavSampleRate << ",  channels " << wavChannels << oendl;
  return true;
}

bool WavFile::adjustHeaders(int fd, int total) {
  lseek(fd, 4, SEEK_SET);
  int i = total + 36;
  write( fd, &i, sizeof(i));
  lseek( fd, 40, SEEK_SET);
  write( fd, &total, sizeof(total));
//  owarn << "adjusting header " << total << "" << oendl;
  return true;
}

int WavFile::parseWavHeader(int fd) {
//  owarn << "Parsing wav header" << oendl;
  char string[4];
  int found;
  short fmt;
  unsigned short ch, bitrate;
  unsigned long samplerrate, longdata;

  if (read(fd, string, 4) < 4) {
//    owarn << " Could not read from sound file." << oendl;
    return -1;
  }
  if (strncmp(string, "RIFF", 4)) {
//    owarn << " not a valid WAV file." << oendl;
    return -1;
  }
  lseek(fd, 4, SEEK_CUR);
  if (read(fd, string, 4) < 4) {
//    owarn << "Could not read from sound file." << oendl;
    return -1;
  }
  if (strncmp(string, "WAVE", 4)) {
//    owarn << "not a valid WAV file." << oendl;
    return -1;
  }
  found = 0;

  while (!found) {
    if (read(fd, string, 4) < 4) {
//      owarn << "Could not read from sound file." << oendl;
      return -1;
    }
    if (strncmp(string, "fmt ", 4)) {
      if (read(fd, &longdata, 4) < 4) {
//        owarn << "Could not read from sound file." << oendl;
        return -1;
      }
      lseek(fd, longdata, SEEK_CUR);
    } else {
      lseek(fd, 4, SEEK_CUR);
      if (read(fd, &fmt, 2) < 2) {
//        owarn << "Could not read format chunk." << oendl;
        return -1;
      }
      if (fmt != WAVE_FORMAT_PCM && fmt != WAVE_FORMAT_DVI_ADPCM) {
//        owarn << "Wave file contains unknown format. Unable to continue." << oendl;
        return -1;
      }
      wavFormat = fmt;
      //      compressionFormat=fmt;
//      owarn << "compressionFormat is " << fmt << "" << oendl;
      if (read(fd, &ch, 2) < 2) {
//        owarn << "Could not read format chunk." << oendl;
        return -1;
      } else {
        wavChannels = ch;
//        owarn << "File has " << ch << " channels" << oendl;
      }
      if (read(fd, &samplerrate, 4) < 4) {
//        owarn << "Could not read from format chunk." << oendl;
        return -1;
      } else {
        wavSampleRate = samplerrate;
        //                sampleRate = samplerrate;
//        owarn << "File has samplerate of " << (int) samplerrate << "" << oendl;
      }
      lseek(fd, 6, SEEK_CUR);
      if (read(fd, &bitrate, 2) < 2) {
//        owarn << "Could not read format chunk." << oendl;
        return -1;
      }  else {
        wavResolution=bitrate;
        //                 resolution =  bitrate;
//        owarn << "File has bitrate of " << bitrate << "" << oendl;
      }
      found++;
    }
  }
  found = 0;
  while (!found) {
    if (read(fd, string, 4) < 4) {
      odebug << "Could not read from sound file." << oendl;
      return -1;
    }

    if (strncmp(string, "data", 4)) {
      if (read(fd, &longdata, 4)<4) {
        odebug << "Could not read from sound file." << oendl;
        return -1;
      }

      lseek(fd, longdata, SEEK_CUR);
    } else {
      if (read(fd, &longdata, 4) < 4) {
        odebug << "Could not read from sound file." << oendl;
        return -1;
      } else {
        wavNumberSamples =  longdata;
         odebug << "file hase length of " << (int)longdata << ""
                << "lasting "
                << (int)(( longdata / wavSampleRate) / wavChannels) / ( wavChannels*( wavResolution/8))
                << " seconds" << oendl;
//        wavSeconds = (( longdata / wavSampleRate) / wavChannels) / ( wavChannels*( wavResolution/8));

  return longdata;
      }
    }
  }

  lseek(fd, 0, SEEK_SET);

  return 0;
}

QString WavFile::trackName() {
    return track.name();
}

int WavFile::wavHandle(){
  return track.handle();
}

int WavFile::getFormat() {
return wavFormat;
}

int WavFile::getResolution() {
return wavResolution;
}

int WavFile::getSampleRate() {
 return wavSampleRate;
}

int WavFile::getNumberSamples() {
 return wavNumberSamples;
}

bool WavFile::isTempFile() {
return useTmpFile;
}

int WavFile::getChannels() {

   return wavChannels;
}