oggwave.cc

/*
 * SubTimer source file
 *  Copyright (C) 2005 Peter Salvi
 *   Last modification: <2005.03.23., 16:16:16>
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 */

#include <qdatetime.h>
#include <qprogressdialog.h>

#include <vorbis/vorbisfile.h>

#include "common.hh"
#include "oggwave.hh"

OggWave::OggWave() : WaveFile()
{
  ogg = new OggVorbis_File;
}

OggWave::~OggWave()
{
}

bool OggWave::open(QString fname, int xres)
{
  FILE *f;
  f = fopen(fname, "rb");

  if(ov_open(f, ogg, NULL, 0) < 0)
    return false;

  if(ov_seekable(ogg) == 0)
    return false;

  if(ov_streams(ogg) != 1) {
    // todo
  }
  active_stream = 0;

  channels = ov_info(ogg, active_stream)->channels;
  sampling = ov_info(ogg, active_stream)->rate;

  loadWave();
  changeRes(xres);

  opened = true;
  return true;
}

void OggWave::close()
{
  ov_clear(ogg);
  opened = false;
}

QTime OggWave::length() const
{
  return QTIME_MS((int)(1000.0 * ov_time_total(ogg, active_stream)));
}

void OggWave::loadWave()
{
  if(wave != NULL)
    delete[] wave;

  char *packet = new char[PACKET_LENGTH * channels];
  size_t rate = (size_t)((double)sampling / MAX_PIXPERSEC) * channels;
  char *tmp = new char[rate];
  size_t tmppos = 0, bufpos = 0;

  double const endtime = ov_time_total(ogg, active_stream);
  wavelength = (size_t)(endtime * MAX_PIXPERSEC * 2.0);

  wave = new char[wavelength];

  QProgressDialog progress(QObject::tr("Loading wave..."), 0, wavelength,
                           0, "progress", TRUE);

  ov_time_seek(ogg, 0);
  while(ov_time_tell(ogg) < endtime && bufpos <= wavelength - 2) {
    int curr;
    long i = 0;
    if((int)((double)bufpos / (double)wavelength * 1000.0) % 10 == 0)
      progress.setProgress(bufpos);
    long res = ov_read(ogg, packet, PACKET_LENGTH * channels, 0, 1, 1, &curr);
    while(i < res) {
      while(i < res && tmppos < rate)
        tmp[tmppos++] = packet[i++];
      if(tmppos == rate) {
        if(bufpos <= wavelength - 2) {
          char min = 0, max = 0;
          for(size_t j = 0; j < rate; ++j) {
            char const minmax = tmp[j];
            if(minmax < min)
              min = minmax;
            if(minmax > max)
              max = minmax;
          }
          wave[bufpos++] = min;
          wave[bufpos++] = max;
        }
        tmppos = 0;
      }
    }
    size_t newpos = ((size_t)(ov_time_tell(ogg) * 
                              MAX_PIXPERSEC * 2.0) / 2) * 2;
    if(newpos > bufpos && bufpos > 2)
      for(size_t j = bufpos; j < newpos; ++j)
        wave[j] = wave[j - 2];
    bufpos = newpos;
  }
  progress.setProgress(wavelength);

  for(size_t i = bufpos; i < wavelength; ++i)
    wave[i] = 0;

  delete[] tmp;
  delete[] packet;
  
}

void OggWave::waveFormat(WaveFormat &f) const
{
  f.channels = channels;
  f.rate = (unsigned int)sampling;
  f.format = SND_PCM_FORMAT_S16_LE; 
  f.access = SND_PCM_ACCESS_RW_INTERLEAVED;
}

size_t OggWave::getPlayBufferSize(QTime const start, QTime const end) const
{
  double const duration = (double)start.msecsTo(end) / 1000.0;
  return (size_t)(duration * (double)sampling) * channels * 2;
}

void OggWave::getPlayBuffer(QTime const start, char *buf, size_t &size)
{
  char *packet = new char[PACKET_LENGTH * channels];
  size_t bufpos = 0;

  ov_time_seek(ogg, (double)MSECS(start) / 1000.0);
  while(bufpos < size) {
    int curr;
    long i = 0;
    long res = ov_read(ogg, packet, PACKET_LENGTH * channels, 0, 2, 1, &curr);
    if(res <= 0)
      break;
    while(i < res && bufpos < size)
      buf[bufpos++] = packet[i++];
  }
  if(size > bufpos)
    size = bufpos;
  
  delete[] packet;  
}

---