TAudioFile.hpp

Go to the documentation of this file.

#ifndef __TAudioFile__
#define __TAudioFile__

#include <Basics/CCountedObject.hpp>
#include "TWaveFileFormat.hpp"
#include "TAiffFileFormat.hpp"
#include "TAudioBuffer.hpp"
using Exponent::Audio::TAudioBuffer;
using Exponent::Audio::TWaveFileFormat;
using Exponent::Audio::TAiffFileFormat;
using Exponent::Basics::CCountedObject;

namespace Exponent
{
    namespace Audio
    {
        template <class TypeName> class TAudioFile : public CCountedObject
        {
            EXPONENT_CLASS_DECLARATION;
//  ===========================================================================

        public:

//  ===========================================================================

            TAudioFile()
            {
                EXPONENT_CLASS_CONSTRUCTION(TAudioFile<TypeName>);
            }

            virtual ~TAudioFile()
            {
                EXPONENT_CLASS_DESTRUCTION(TAudioFile<TypeName>);
            }

//  ===========================================================================

            const TAudioBuffer<TypeName> *getAudioBuffer() const { return &m_audioBuffer; }

            TAudioBuffer<TypeName> *getMutableAudioBuffer() { return &m_audioBuffer; }

//  ===========================================================================

            bool readFile(const CSystemString &filename)
            {
                TWaveFileFormat<TypeName> waveFileFormat;
                if (waveFileFormat.isValidFormat(filename))
                {
                    if (!waveFileFormat.openFile(CFileStream::e_input, filename))
                    {
                        return false;
                    }
                    
                    // Construct the file format
                    typename TWaveFileFormat<TypeName>::SAudioFileFormat format;
                    if (!waveFileFormat.readHeader(format))
                    {
                        return false;
                    }

                    m_audioBuffer.initialise(format.m_numberOfSamples, format.m_numberOfChannels);

                    // Now read in the audio data, in range  -1 to +1\n
                    if (!waveFileFormat.readData(m_audioBuffer))
                    {
                        return false;
                    }

                    return true;
                }
                else
                {
                    TAiffFileFormat<TypeName> aiffFileFormat;
                    if (!aiffFileFormat.isValidFormat(filename))
                    {
                        return false;
                    }

                    if (!aiffFileFormat.openFile(CFileStream::e_input, filename))
                    {
                        return false;
                    }
                    
                    typename IAudioFileFormat< TypeName >::SAudioFileFormat format;
                    if (!aiffFileFormat.readHeader(format))
                    {
                        return false;
                    }

                    m_audioBuffer.initialise(format.m_numberOfSamples, format.m_numberOfChannels);

                    // Now read in the audio data, in range  -1 to +1\n
                    if (!aiffFileFormat.readData(m_audioBuffer))
                    {
                        return false;
                    }

                    return true;
                }
                return false;
            }

            bool writeFile(const CSystemString &filename, const unsigned long bitDepth, const unsigned long sampleRate)
            {
                return this->writeFile(filename, bitDepth, sampleRate, m_audioBuffer);
            }

            bool writeFile(const CSystemString &filename, const unsigned long bitDepth, const unsigned long sampleRate, TAudioBuffer<TypeName> &buffer)
            {
                if (!(bitDepth == 8  ||
                      bitDepth == 16 ||
                      bitDepth == 24))
                {
                    return false;
                }

                TWaveFileFormat<TypeName> waveFileFormat;

                if (!waveFileFormat.openFile(CFileStream::e_output, filename))
                {
                    return false;
                }

                // Now create the format to write to
                typename TWaveFileFormat<TypeName>::SAudioFileFormat format;
                format.m_bitDepth         = bitDepth;   
                format.m_sampleRate       = sampleRate; 
                format.m_numberOfChannels = buffer.getNumberOfChannels();   
                format.m_numberOfSamples  = buffer.getNumberOfSamples(); 

                if (!waveFileFormat.writeHeader(format))
                {
                    return false;
                }

                if (!waveFileFormat.writeData(buffer))
                {
                    return false;
                }

                return true;
            }


//  ===========================================================================

        protected:

//  ===========================================================================
    
            TAudioBuffer<TypeName> m_audioBuffer;               
        };

        EXPONENT_TEMPLATE_CLASS_IMPLEMENTATION(TAudioFile<TypeName>, TypeName, CCountedObject);
    }
}
#endif  // End of TAudioFile.hpp

---