---

wav_writer.h

//----------------------------------------------------------------------------
//----------------------------------------------------------------------------
#ifndef WAVWRITER_H_INCLUDED
#define WAVWRITER_H_INCLUDED
//----------------------------------------------------------------------------
#include <string>
#include <fstream>
#include "datatypes.h"
//----------------------------------------------------------------------------
namespace ac3 {
//----------------------------------------------------------------------------
#define WAVWRITER_UNKNOWN_ERROR -1            
#define WAVWRITER_NO_ERROR 0
#define WAVWRITER_COULD_NOT_CREATE_FILE 1
#define WAVWRITER_FILE_NOT_OPEN 2
#define WAVWRITER_WRITE_ERROR 3
#define WAVWRITER_INVALID_FORMAT 4
#define WAVWRITER_UNSUPPORTED_FORMAT 5

#define WAVWRITER_ERROR_NUM 6

class CWAVWriterException
{
public:
  CWAVWriterException() : m_nReason(WAVWRITER_UNKNOWN_ERROR) {}
  CWAVWriterException(int nReason) : m_nReason(nReason) {}
  CWAVWriterException(int nReason, const std::string& sHint)
    : m_nReason(nReason), m_sHint(sHint) {}
  
public:
  int GetReason() const { return m_nReason; }
  std::string GetHint() const { return m_sHint; }
  std::string GetString() const;
  
private:
  int m_nReason;
  std::string m_sHint;

  static std::string m_sUnknownError;
  static std::string m_sError[WAVWRITER_ERROR_NUM];
};
//----------------------------------------------------------------------------
class CWAVWriter
{
public:
  CWAVWriter();
private:
  CWAVWriter(const CWAVWriter& ww) {}
public:
  CWAVWriter(const char* pszFilename, UINT32 nSampleRate,
             UINT16 nChannelNum, UINT16 nBitResolution);
  CWAVWriter(const std::string& sFilename, UINT32 nSampleRate,
             UINT16 nChannelNum, UINT16 nBitResolution);
  ~CWAVWriter();


public:
  void OpenFile(const char* pszFilename, UINT32 nSampleRate,
                UINT16 nChannelNum, UINT16 nBitResolution);
  void OpenFile(const std::string& sFilename, UINT32 nSampleRate,
                UINT16 nChannelNum, UINT16 nBitResolution);
  void CloseFile();

  void Write8BitSamples(UINT8** ppSource, UINT32 nCount);
  void Write16BitSamples(INT16** ppSource, UINT32 nCount);
  void Write24BitSamples(INT32** ppSource, UINT32 nCount);
  void Write32BitSamples(INT32** ppSource, UINT32 nCount);


private:
  void WriteHeader();
  
  void WriteFOURCC(std::string sFOURCC);
  void WriteUINT8(UINT8 nValue);
  void WriteINT8(INT8 nValue);
  void WriteUINT16(UINT16 nValue);
  void WriteINT16(INT16 nValue);
  void WriteUINT24(UINT32 nValue);
  void WriteINT24(INT32 nValue);
  void WriteUINT32(UINT32 nValue);
  void WriteINT32(INT32 nValue);
  
  CWAVWriter& operator=(const CWAVWriter& ww) { return *this; }


private:
  bool m_bFileOpen;
  std::ofstream m_file;
  UINT32 m_nSampleRate;
  UINT16 m_nChannels;
  UINT16 m_nBitsPerSample;

  UINT32 m_nAllSize;
  std::streampos m_allSizePos;
  UINT32 m_nDataSize;
  std::streampos m_dataSizePos;
};
//----------------------------------------------------------------------------
} // namespace ac3
//----------------------------------------------------------------------------
#endif // WAVWRITER_H_INCLUDED

---