// WAVWriter.cpp: .wav audio writer 
//
//   Copyright (C) 2005, 2006, 2007, 2008, 2009, 2010,
//   2011 Free Software Foundation, Inc
//
// 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 3 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., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

#include "WAVWriter.h"

#include <cstring> // for strncpy
#include <boost/cstdint.hpp>
#include <fstream> // for composition (file_stream)
#include <iostream> 

#include "GnashException.h" // for SoundException
#include "log.h" // will import boost::format too

namespace gnash {
namespace sound {

namespace { // anonymous

// Header of a wave file
// http://ftp.iptel.org/pub/sems/doc/full/current/wav__hdr_8c-source.html
typedef struct{
     char rID[4];            // 'RIFF'
     boost::uint32_t rLen;        
     char wID[4];            // 'WAVE'
     char fId[4];            // 'fmt '
     boost::uint32_t pcm_header_len;   // varies...
     boost::int16_t wFormatTag;
     boost::int16_t nChannels;      // 1,2 for stereo data is (l,r) pairs
     boost::uint32_t nSamplesPerSec;
     boost::uint32_t nAvgBytesPerSec;
     boost::int16_t nBlockAlign;      
     boost::int16_t nBitsPerSample;
} WAV_HDR;

// Chunk of wave file
// http://ftp.iptel.org/pub/sems/doc/full/current/wav__hdr_8c-source.html
typedef struct{
    char dId[4];            // 'data' or 'fact'
    boost::uint32_t dLen;
} CHUNK_HDR;

} // end of anonymous namespace

/* public */
WAVWriter::WAVWriter(const std::string& wavefile)
{
        file_stream.open(wavefile.c_str());
        if (file_stream.fail()) {
            boost::format fmt = boost::format(_("Unable to write file %1%"))
                % wavefile;
            throw SoundException(fmt.str());
        } 
        else {
            write_wave_header(file_stream);
            std::cout << "# Created 44100 16Mhz stereo wave file:\n" <<
                    "AUDIOFILE=" << wavefile << std::endl;
        }
}

/* public */
WAVWriter::~WAVWriter()
{
    if (file_stream) file_stream.close();
}

/* public */
void
WAVWriter::pushSamples(boost::int16_t* from, unsigned int nSamples)
{
    // NOTE: if muted, the samples will be silent already
        boost::uint8_t* stream = reinterpret_cast<boost::uint8_t*>(from);
        unsigned int len = nSamples*2;
        file_stream.write((char*) stream, len);

}

/* private */
void
WAVWriter::write_wave_header(std::ofstream& outfile)
{
 
  // allocate wav header
  WAV_HDR wav;
  CHUNK_HDR chk;
 
  // setup wav header
  std::strncpy(wav.rID, "RIFF", 4);
  std::strncpy(wav.wID, "WAVE", 4);
  std::strncpy(wav.fId, "fmt ", 4);
 
  wav.nBitsPerSample = 16;
  wav.nSamplesPerSec = 44100;
  wav.nAvgBytesPerSec = 44100;
  wav.nAvgBytesPerSec *= wav.nBitsPerSample / 8;
  wav.nAvgBytesPerSec *= 2;
  wav.nChannels = 2;
    
  wav.pcm_header_len = 16;
  wav.wFormatTag = 1;
  wav.rLen = sizeof(WAV_HDR) + sizeof(CHUNK_HDR);
  wav.nBlockAlign = 2 * wav.nBitsPerSample / 8;

  // setup chunk header
  std::strncpy(chk.dId, "data", 4);
  chk.dLen = 0;
 
  /* write riff/wav header */
  outfile.write((char *)&wav, sizeof(WAV_HDR));
 
  /* write chunk header */
  outfile.write((char *)&chk, sizeof(CHUNK_HDR));
 
}


} // gnash.sound namespace 
} // namespace gnash
