source: orxonox.OLD/trunk/src/lib/sound/ogg_player.cc @ 7355

/*
   orxonox - the future of 3D-vertical-scrollers

   Copyright (C) 2004 orx

   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, or (at your option)
   any later version.

### File Specific:
   main-programmer: Benjamin Grauer
   co-programmer: ...

    Beware:
   The Ogg-Player is a __Threaded__ Stream, this means, that invoking play()
   creates a Thread, that loops through the Music-File. The Thread seems
   to be safe, but it is not guarantied.

   -------------------------------------------------------------------
   A Simple Version of this can be found at http://www.devmaster.net
   Thanks a lot for the nice work, and the easy portability to our Project.
*/
#define DEBUG_SPECIAL_MODULE DEBUG_MODULE_SOUND

#include <iostream>

#include "ogg_player.h"

#include "sound_engine.h"

#include "debug.h"

/**
 * initializes an Ogg-player from a file
 * @param fileName the file to load
 */
OggPlayer::OggPlayer(const std::string& fileName)
{
  this->setClassID(CL_SOUND_OGG_PLAYER, "OggPlayer");

  this->state = OggPlayer::None;

  this->source = 0;
  this->buffers[0] = 0;
  this->buffers[1] = 0;
  this->musicThreadID = NULL;
  this->musicMutex = SDL_CreateMutex();

  if (!fileName.empty())
  {
    if (this->open(fileName))
      this->setName(fileName);
  }

}

/**
 * @brief deletes the OggPlayer
 */
OggPlayer::~OggPlayer()
{
  this->release();
  SDL_DestroyMutex(this->musicMutex);
}

///////////////
// INTERFACE //
///////////////
/**
 * @brief opens a file for playback
 * @param fileName the file to open
 */
bool OggPlayer::open(const std::string& fileName)
{
  MutexLock(this->musicMutex);
  // release old Ogg-File (if loaded)
  if (this->state & OggPlayer::FileOpened)
    this->release();

  // allocating Buffers
  if (this->buffers[0] == 0)
    alGenBuffers(2, this->buffers);
  SoundEngine::checkError("Allocating Buffers", __LINE__);
  if (this->buffers[0] != 0 && this->buffers[1] != 0)
    state |= OggPlayer::BuffersAllocated;
  else
  {
    PRINTF(2)("Unable to allocate al-Buffers\n");
    this->release();
    return false;
  }
  // allocating source
  if (this->source == 0)
    SoundEngine::getInstance()->popALSource(this->source);
  if (this->source != 0)
    state |= OggPlayer::SourceAllocated;
  else
  {
    PRINTF(2)("No more Sources Availiable (maybe you should consider raising the source-count.)\n");
    this->release();
    return false;
  }

  // opening the FILE;
  int result;
  if(!(oggFile = fopen(fileName.c_str(), "rb")))
  {
    PRINTF(2)("Could not open Ogg file.");
    this->release();
    return false;
  }
  // reading the Stream.
  if((result = ov_open(oggFile, &oggStream, NULL, 0)) < 0)
  {
    PRINTF(2)("Could not open Ogg stream. %s", getVorbisError(result));
    fclose(oggFile);
    this->release();
    return false;
  }
  this->state |= OggPlayer::FileOpened;

  // acquiring the vorbis-properties.
  vorbisInfo = ov_info(&oggStream, -1);
  vorbisComment = ov_comment(&oggStream, -1);

  if(vorbisInfo->channels == 1)
    format = AL_FORMAT_MONO16;
  else
    format = AL_FORMAT_STEREO16;

  // setting the Source Properties.
  alSource3f(source, AL_POSITION,        0.0, 0.0, 0.0);
  alSource3f(source, AL_VELOCITY,        0.0, 0.0, 0.0);
  alSource3f(source, AL_DIRECTION,       0.0, 0.0, 0.0);
  alSourcef (source, AL_ROLLOFF_FACTOR,  0.0          );
  alSourcei (source, AL_SOURCE_RELATIVE, AL_TRUE      );
  alSourcef (source, AL_GAIN,            SoundEngine::getInstance()->getMusicVolume());
  SoundEngine::checkError("OggPlayer::open()::SetSourceProperties", __LINE__);

  // Set to State Stopped
  this->state |= OggPlayer::Stopped;
  return true;
}


/**
 * @brief start Playing Music.
 * @returns true on success false otherwise.
 */
bool OggPlayer::play()
{
  if (!(this->state & OggPlayer::FileOpened))
    return false;

  this->state &= ~(OggPlayer::Stopped | OggPlayer::Paused);

  if (!this->playback())
    return false;

  if (this->musicThreadID == NULL)
    return ((this->musicThreadID = SDL_CreateThread(OggPlayer::musicThread, (void*)this)) != NULL);
  return true;
}

/**
 * @brief stop the OggPlayer from Playing.
 */
void OggPlayer::stop()
{
  this->state &= ~(OggPlayer::Playing | OggPlayer::Paused);
  this->state |= OggPlayer::Stopped;

  this->suspend();
  this->rewind();
}

/**
 * @brief Pause the Playing.
 */
void OggPlayer::pause()
{
  this->state &= ~OggPlayer::Playing;

  if (!(this->state & OggPlayer::Stopped))
    this->state |= OggPlayer::Paused;

  this->suspend();
}

/**
 * @brief rewind to the beginning, and play (if already playing)
 */
void OggPlayer::rewind()
{
  this->jumpTo(0.0f);
}

/**
 * @brief jump to Second timeCode in the MusicFile
 * @param timeCode where to jump to.
 */
void OggPlayer::jumpTo(float timeCode)
{

  if (this->state & OggPlayer::FileOpened)
  {
    SDL_mutexP(this->musicMutex);
    ov_time_seek(&this->oggStream, timeCode);
    SDL_mutexV(this->musicMutex);
  }
}

/**
 * @returns the Length of the Music in Seconds
 */
float OggPlayer::length()
{
  if (this->state & OggPlayer::FileOpened)
    return ov_time_total(&this->oggStream, -1);
  else
    return 0.0f;
}


/**
 * @returns true if the file is playing
 */
bool OggPlayer::isPlaying()
{
  if (!(this->state & OggPlayer::FileOpened))
    return false;
  ALenum state;

  alGetSourcei(this->source, AL_SOURCE_STATE, &state);

  return (state == AL_PLAYING);
}



////////////////////////
// INTERNAL FUNCTIONS //
////////////////////////
/**
 * @brief creates a Thread for Playing back the music even if the rest of the Engine is slow
 * @param oggPlayer the OggPlayer to play back
 * @returns -1 on error.
 */
int OggPlayer::musicThread(void* oggPlayer)
{
  if (oggPlayer == NULL)
    return -1;
  OggPlayer* ogg = (OggPlayer*)oggPlayer;

  PRINTF(4)("STARTIG AUDIO THREAD\n");
  while (ogg->state & OggPlayer::Playing)
  {
    SDL_mutexP(ogg->musicMutex);
    ogg->update();
    SDL_mutexV(ogg->musicMutex);
    SDL_Delay(1);
  }
  PRINTF(4)("End the AudioThread\n");
}


/**
 * @brief plays back the sound
 * @return true if running, false otherwise
 */
bool OggPlayer::playback()
{
  if (!(this->state & OggPlayer::FileOpened))
    return false;

  if(this->state & OggPlayer::Playing)
    return true;
  this->state |= OggPlayer::Playing;

  SDL_mutexP(this->musicMutex);
  if(!this->stream(this->buffers[0]) || !this->stream(this->buffers[1]))
  {
    SDL_mutexV(this->musicMutex);
    this->state &= ~OggPlayer::Playing;
    return false;
  }

  alSourceQueueBuffers(this->source, 2, this->buffers);
  if (DEBUG >= 3)
    SoundEngine::checkError("OggPlayer::playback()::alSourceQueueBuffers", __LINE__);

  alSourcePlay(this->source);
  if (DEBUG >= 3)
    SoundEngine::checkError("OggPlayer::playback()::alSourcePlay", __LINE__);
  SDL_mutexV(this->musicMutex);
  return true;
}


/**
 * @brief waits for the AudioThread to be finished.
 */
void OggPlayer::suspend()
{
  if (this->musicThreadID != NULL)
  {
    assert (!(this->state & Playing));
    SDL_WaitThread(this->musicThreadID, NULL);
    this->musicThreadID = NULL;
  }
  if (this->state & OggPlayer::SourceAllocated)
  {
    alSourceStop(this->source);
    alSourcei(this->source, AL_BUFFER, 0);
  }
}

/**
 * @brief updates the stream, this has to be done every few parts of a second, for sound-consistency
 * @returns true, if the Sound is playing flawlessly
 */
bool OggPlayer::update()
{
  int processed = 0;
  bool active = true;

  alGetSourcei(source, AL_BUFFERS_PROCESSED, &processed);
  if (DEBUG >= 3)
    SoundEngine::checkError("OggPlayer::update()::alGetSourceI", __LINE__);

  while(processed--)
  {
    ALuint buffer;

    alSourceUnqueueBuffers(source, 1, &buffer);
    if (DEBUG >= 3)
      SoundEngine::checkError("OggPlayer::update()::unqueue", __LINE__);

    active = stream(buffer);

    alSourceQueueBuffers(source, 1, &buffer);
    if (DEBUG >= 3)
      SoundEngine::checkError("OggPlayer::update()::queue", __LINE__);
  }

  return active;
}

/**
 * @brief gets a new Stream from buffer
 * @param buffer the buffer to get the stream from
 * @return true, if everything worked as planed
 */
bool OggPlayer::stream(ALuint buffer)
{
  if (unlikely(!(this->state & Playing)))
    return false;
  char pcm[OGG_PLAYER_BUFFER_SIZE];
  int  size = 0;
  int  section;
  int  result;

  while(size < OGG_PLAYER_BUFFER_SIZE)
  {
    result = ov_read(&this->oggStream, pcm + size, OGG_PLAYER_BUFFER_SIZE - size, 0, 2, 1, &section);

    if(result > 0)
      size += result;
    else if(result < 0)
      throw getVorbisError(result);
    else /* eof */
      ov_time_seek(&this->oggStream, 0.0);
  }

  if(size == 0)
    return false;

  alBufferData(buffer, format, pcm, size, vorbisInfo->rate);
  if (DEBUG >= 3)
    SoundEngine::checkError("OggPlayer::stream()::BUFFER", __LINE__);

  return true;
}


/**
 * @brief releases a stream
 */
void OggPlayer::release()
{
  if (this->state & OggPlayer::SourceAllocated)
  {
    assert(alIsSource(this->source));
    if (this->state & OggPlayer::Playing);
    {
      // Kill the Music Thread.
      this->state &= ~OggPlayer::Playing;
      this->suspend();

      SoundEngine::checkError("OggPlayer::release()::alSourceStop", __LINE__);
    }
    empty();
    alSourcei(this->source, AL_BUFFER, 0);
    SoundEngine::getInstance()->pushALSource(this->source);
    this->source = 0;
    this->state &= ~SourceAllocated;
  }
  if (this->state & OggPlayer::BuffersAllocated)
  {
    assert (this->buffers[0] != 0 && this->buffers[1] != 0);
    alDeleteBuffers(2, buffers);
    SoundEngine::checkError("OggPlayer::release()::alDeleteBuffers", __LINE__);
    this->buffers[0] = 0;
    this->buffers[1] = 0;
    this->state &= ~OggPlayer::BuffersAllocated;
  }

  if (this->state & OggPlayer::FileOpened)
  {
    ov_clear(&oggStream);
    this->state &= ~OggPlayer::FileOpened;
  }

}


/**
 * @brief empties the buffers
 */
void OggPlayer::empty()
{
  int queued;

  alGetSourcei(source, AL_BUFFERS_QUEUED, &queued);

  while(queued--)
  {
    ALuint buffer;

    alSourceUnqueueBuffers(source, 1, &buffer);
    SoundEngine::checkError("OggPlayer::empty()::unqueue Buffers", __LINE__);
  }
}


/////////////////////
// DEBUG FUNCTIONS //
/////////////////////
/**
 * displays some info about the ogg-file
 */
void OggPlayer::debug() const
{
  cout
  << "version         " << vorbisInfo->version         << "\n"
  << "channels        " << vorbisInfo->channels        << "\n"
  << "rate (hz)       " << vorbisInfo->rate            << "\n"
  << "bitrate upper   " << vorbisInfo->bitrate_upper   << "\n"
  << "bitrate nominal " << vorbisInfo->bitrate_nominal << "\n"
  << "bitrate lower   " << vorbisInfo->bitrate_lower   << "\n"
  << "bitrate window  " << vorbisInfo->bitrate_window  << "\n"
  << "\n"
  << "vendor " << vorbisComment->vendor << "\n";

  for(int i = 0; i < vorbisComment->comments; i++)
    cout << "   " << vorbisComment->user_comments[i] << "\n";

  cout << endl;
}


void OggPlayer::printState() const
{
  PRINTF(0)("OggPlayer is in the following States: ");
  if (this->state & OggPlayer::FileOpened)
    PRINT(0)("FileOpened ");
  if (this->state & OggPlayer::SourceAllocated)
    PRINT(0)("SourceAllocated ");
  if (this->state & OggPlayer::BuffersAllocated)
    PRINT(0)("BuffersAllocated ");
  if (this->state & OggPlayer::Stopped)
    PRINT(0)("Stopped ");
  if (this->state & OggPlayer::Playing)
    PRINT(0)("Playing ");
  if (this->state & OggPlayer::Paused)
    PRINT(0)("Paused ");
  if (this->state & OggPlayer::Error)
    PRINT(0)("Error ");
  PRINT(0)("\n");
}

/**
 * returns errors
 * @param code the error-code
 * @return the error as a String
 */
const char* OggPlayer::getVorbisError(int code)
{
  switch(code)
  {
    case OV_EREAD:
      return ("Read from media.");
    case OV_ENOTVORBIS:
      return ("Not Vorbis data.");
    case OV_EVERSION:
      return ("Vorbis version mismatch.");
    case OV_EBADHEADER:
      return ("Invalid Vorbis header.");
    case OV_EFAULT:
      return ("Internal logic fault (bug or heap/stack corruption.");
    default:
      return ("Unknown Ogg error.");
  }
}