/*
    SDL - Simple DirectMedia Layer
    Copyright (C) 1997, 1998, 1999, 2000, 2001, 2002  Sam Lantinga

    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Library General Public
    License as published by the Free Software Foundation; either
    version 2 of the License, or (at your option) any later version.

    This library 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
    Library General Public License for more details.

    You should have received a copy of the GNU Library General Public
    License along with this library; if not, write to the Free
    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

    Sam Lantinga
    slouken@libsdl.org

    This file based on Apple sample code. We haven't changed the file name, 
    so if you want to see the original search for it on apple.com/developer
*/

//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
//  AudioFilePlayer.cpp
//
#include "AudioFilePlayer.h"

void ThrowResult (OSStatus result, const char* str)
{
    SDL_SetError ("Error: %s %d", str, result);
    throw result;
}

#if DEBUG
void PrintStreamDesc (AudioStreamBasicDescription *inDesc)
{
    if (!inDesc) {
        printf ("Can't print a NULL desc!\n");
        return;
    }
    
    printf ("- - - - - - - - - - - - - - - - - - - -\n");
    printf ("  Sample Rate:%f\n", inDesc->mSampleRate);
    printf ("  Format ID:%s\n", (char*)&inDesc->mFormatID);
    printf ("  Format Flags:%lX\n", inDesc->mFormatFlags);
    printf ("  Bytes per Packet:%ld\n", inDesc->mBytesPerPacket);
    printf ("  Frames per Packet:%ld\n", inDesc->mFramesPerPacket);
    printf ("  Bytes per Frame:%ld\n", inDesc->mBytesPerFrame);
    printf ("  Channels per Frame:%ld\n", inDesc->mChannelsPerFrame);
    printf ("  Bits per Channel:%ld\n", inDesc->mBitsPerChannel);
    printf ("- - - - - - - - - - - - - - - - - - - -\n");
}
#endif

OSStatus    AudioFileManager::FileInputProc (void                       *inRefCon, 
                                             AudioUnitRenderActionFlags inActionFlags,
                                             const AudioTimeStamp       *inTimeStamp, 
                                             UInt32                     inBusNumber, 
                                             AudioBuffer                *ioData)
{
    AudioFileManager* THIS = (AudioFileManager*)inRefCon;
    return THIS->Render(*ioData);
}

OSStatus    AudioFileManager::Render (AudioBuffer &ioData)
{
    OSStatus result = AudioConverterFillBuffer(mParentConverter, 
                                    AudioFileManager::ACInputProc, 
                                    this, 
                                    &ioData.mDataByteSize, 
                                    ioData.mData);
    if (result) {
        SDL_SetError ("AudioConverterFillBuffer:%ld\n", result);
        mParent.DoNotification (result);
    } else {
        mByteCounter += ioData.mDataByteSize / 2;
        AfterRender();
    }
    return result;
}

OSStatus    AudioFileManager::ACInputProc (AudioConverterRef            inAudioConverter,
                                            UInt32*                     outDataSize,
                                            void**                      outData,
                                            void*                       inUserData)
{
    AudioFileManager* THIS = (AudioFileManager*)inUserData;
    return THIS->GetFileData(outData, outDataSize);
}

AudioFileManager::~AudioFileManager ()
{
    if (mFileBuffer) {
        free (mFileBuffer);
        mFileBuffer = 0;
    }
}

AudioFilePlayer::AudioFilePlayer (const FSRef           *inFileRef)
    : mConnected (false),
      mAudioFileManager (0),
      mConverter (0),
      mNotifier (0),
      mStartFrame (0)
{
    SInt64 fileDataSize  = 0;

    OpenFile (inFileRef, fileDataSize);
        
    // we want about a seconds worth of data for the buffer
    int bytesPerSecond = UInt32 (mFileDescription.mSampleRate * mFileDescription.mBytesPerFrame);
    
#if DEBUG
    printf("File format:\n");
    PrintStreamDesc (&mFileDescription);
#endif
    
    mAudioFileManager = new AudioFileManager (*this, 
                                                mForkRefNum, 
                                                fileDataSize,
                                                bytesPerSecond);
}

// you can put a rate scalar here to play the file faster or slower
// by multiplying the same rate by the desired factor 
// eg fileSampleRate * 2 -> twice as fast
// before you create the AudioConverter
void    AudioFilePlayer::SetDestination (AudioUnit  &inDestUnit, 
                                         int         inBusNumber)
{
    if (mConnected) throw static_cast<OSStatus>(-1); //can't set dest if already engaged
 
    mPlayUnit = inDestUnit;
    mBusNumber = inBusNumber;

    OSStatus result = noErr;
    
    if (mConverter) {
        result = AudioConverterDispose (mConverter);
            THROW_RESULT("AudioConverterDispose")
    }
    
    AudioStreamBasicDescription     destDesc;
    UInt32  size = sizeof (destDesc);
    result = AudioUnitGetProperty (inDestUnit,
                                   kAudioUnitProperty_StreamFormat,
                                   kAudioUnitScope_Input,
                                   inBusNumber,
                                   &destDesc,
                                   &size);
        THROW_RESULT("AudioUnitGetProperty")

#if DEBUG
    printf("Destination format:\n");
    PrintStreamDesc (&destDesc);
#endif

        //we can "down" cast a component instance to a component
    ComponentDescription desc;
    result = GetComponentInfo ((Component)inDestUnit, &desc, 0, 0, 0);
        THROW_RESULT("GetComponentInfo")
        
        // we're going to use this to know which convert routine to call
        // a v1 audio unit will have a type of 'aunt'
        // a v2 audio unit will have one of several different types.
    if (desc.componentType != kAudioUnitComponentType) {
        result = badComponentInstance;
        THROW_RESULT("BAD COMPONENT")
    }

    
    result = AudioConverterNew (&mFileDescription, &destDesc, &mConverter);
        THROW_RESULT("AudioConverterNew")

#if 0
    // this uses the better quality SRC
    UInt32 srcID = kAudioUnitSRCAlgorithm_Polyphase;
    result = AudioConverterSetProperty(mConverter,
                    kAudioConverterSampleRateConverterAlgorithm, 
                    sizeof(srcID), 
                    &srcID);
        THROW_RESULT("AudioConverterSetProperty")
#endif
}

void    AudioFilePlayer::SetStartFrame (int frame)
{
    SInt64 position = frame * 2352;

    mStartFrame = frame;
    mAudioFileManager->SetPosition (position);
}

    
int    AudioFilePlayer::GetCurrentFrame ()
{
    return mStartFrame + (mAudioFileManager->GetByteCounter() / 2352);
}
    
void    AudioFilePlayer::SetStopFrame (int frame)
{
    SInt64 position  = frame * 2352;
    
    mAudioFileManager->SetEndOfFile (position);
}
    
AudioFilePlayer::~AudioFilePlayer()
{
    Disconnect();
        
    if (mAudioFileManager) {
        delete mAudioFileManager;
        mAudioFileManager = 0;
    }
    
    if (mForkRefNum) {
        FSClose (mForkRefNum);
        mForkRefNum = 0;
    }

    if (mConverter) {
        AudioConverterDispose (mConverter);
        mConverter = 0;
    }
}

void    AudioFilePlayer::Connect()
{
#if DEBUG
    printf ("Connect:%x,%ld, engaged=%d\n", (int)mPlayUnit, mBusNumber, (mConnected ? 1 : 0));
#endif
    if (!mConnected)
    {           
        mAudioFileManager->Connect(mConverter);
                
        // set the render callback for the file data to be supplied to the sound converter AU
        mInputCallback.inputProc = AudioFileManager::FileInputProc;
        mInputCallback.inputProcRefCon = mAudioFileManager;

        OSStatus result = AudioUnitSetProperty (mPlayUnit, 
                            kAudioUnitProperty_SetInputCallback, 
                            kAudioUnitScope_Input, 
                            mBusNumber,
                            &mInputCallback, 
                            sizeof(mInputCallback));
        THROW_RESULT("AudioUnitSetProperty")
        mConnected = true;
    }
}

// warning noted, now please go away ;-)
// #warning This should redirect the calling of notification code to some other thread
void    AudioFilePlayer::DoNotification (OSStatus inStatus) const
{
    AudioFilePlayer* THIS = const_cast<AudioFilePlayer*>(this);
        
    if (mNotifier) {
        (*mNotifier) (mRefCon, inStatus);
    } else {
        SDL_SetError ("Notification posted with no notifier in place");
        
        if (inStatus == kAudioFilePlay_FileIsFinished)
            THIS->Disconnect();
        else if (inStatus != kAudioFilePlayErr_FilePlayUnderrun)
            THIS->Disconnect();
    }
}

void    AudioFilePlayer::Disconnect ()
{
#if DEBUG
    printf ("Disconnect:%x,%ld, engaged=%d\n", (int)mPlayUnit, mBusNumber, (mConnected ? 1 : 0));
#endif
    if (mConnected)
    {
        mConnected = false;
            
        mInputCallback.inputProc = 0;
        mInputCallback.inputProcRefCon = 0;
        OSStatus result = AudioUnitSetProperty (mPlayUnit, 
                                        kAudioUnitProperty_SetInputCallback, 
                                        kAudioUnitScope_Input, 
                                        mBusNumber,
                                        &mInputCallback, 
                                        sizeof(mInputCallback));
        if (result) 
            SDL_SetError ("AudioUnitSetProperty:RemoveInputCallback:%ld", result);

        mAudioFileManager->Disconnect();
    }
}

struct SSNDData {
    UInt32 offset;
    UInt32 blockSize;
};

void    AudioFilePlayer::OpenFile (const FSRef *inRef, SInt64& outFileDataSize)
{
    ContainerChunk chunkHeader;
    ChunkHeader chunk;
    SSNDData ssndData;

    OSErr result;
    HFSUniStr255 dfName;
    ByteCount actual;
    SInt64 offset;

    // Open the data fork of the input file
    result = FSGetDataForkName(&dfName);
       THROW_RESULT("AudioFilePlayer::OpenFile(): FSGetDataForkName")

    result = FSOpenFork(inRef, dfName.length, dfName.unicode, fsRdPerm, &mForkRefNum);
       THROW_RESULT("AudioFilePlayer::OpenFile(): FSOpenFork")
 
    // Read the file header, and check if it's indeed an AIFC file
    result = FSReadFork(mForkRefNum, fsAtMark, 0, sizeof(chunkHeader), &chunkHeader, &actual);
       THROW_RESULT("AudioFilePlayer::OpenFile(): FSReadFork")

    if (chunkHeader.ckID != 'FORM') {
        result = -1;
        THROW_RESULT("AudioFilePlayer::OpenFile(): chunk id is not 'FORM'");
    }

    if (chunkHeader.formType != 'AIFC') {
        result = -1;
        THROW_RESULT("AudioFilePlayer::OpenFile(): file format is not 'AIFC'");
    }

    // Search for the SSND chunk. We ignore all compression etc. information
    // in other chunks. Of course that is kind of evil, but for now we are lazy
    // and rely on the cdfs to always give us the same fixed format.
    // TODO: Parse the COMM chunk we currently skip to fill in mFileDescription.
    offset = 0;
    do {
        result = FSReadFork(mForkRefNum, fsFromMark, offset, sizeof(chunk), &chunk, &actual);
           THROW_RESULT("AudioFilePlayer::OpenFile(): FSReadFork")
            
        // Skip the chunk data
        offset = chunk.ckSize;
    } while (chunk.ckID != 'SSND');

    // Read the header of the SSND chunk. After this, we are positioned right
    // at the start of the audio data.
    result = FSReadFork(mForkRefNum, fsAtMark, 0, sizeof(ssndData), &ssndData, &actual);
       THROW_RESULT("AudioFilePlayer::OpenFile(): FSReadFork")

    result = FSSetForkPosition(mForkRefNum, fsFromMark, ssndData.offset);
       THROW_RESULT("AudioFilePlayer::OpenFile(): FSSetForkPosition")

    // Data size
    outFileDataSize = chunk.ckSize - ssndData.offset;

    // File format
    mFileDescription.mSampleRate = 44100;
    mFileDescription.mFormatID = kAudioFormatLinearPCM;
    mFileDescription.mFormatFlags = kLinearPCMFormatFlagIsPacked | kLinearPCMFormatFlagIsSignedInteger;
    mFileDescription.mBytesPerPacket = 4;
    mFileDescription.mFramesPerPacket = 1;
    mFileDescription.mBytesPerFrame = 4;
    mFileDescription.mChannelsPerFrame = 2;
    mFileDescription.mBitsPerChannel = 16;
}