Cog/Plugins/CoreAudio/CoreAudioDecoder.m

/*
 *  $Id$
 *
 *  Copyright (C) 2006 Stephen F. Booth <me@sbooth.org>
 *
 *  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 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

#include <unistd.h>

#import "CoreAudioDecoder.h"

#import "Logging.h"

@interface CoreAudioDecoder (Private)
- (BOOL) readInfoFromExtAudioFileRef;
@end

static OSStatus readProc(void* clientData,
                         SInt64 position,
                         UInt32 requestCount,
                         void* buffer,
                         UInt32* actualCount)
{
    NSObject* _handle = (__bridge NSObject *)(clientData);
    CoreAudioDecoder * __unsafe_unretained pSelf = (id) _handle;
    
    id<CogSource> source = pSelf->_audioSource;
    
    if (position != pSelf->_lastPosition) {
        [source seek:position whence:SEEK_SET];
    }

    size_t copyMax = 0;
    size_t bytesRead = 0;

    if ( requestCount )
        bytesRead = [source read:(((uint8_t *)buffer) + copyMax) amount:requestCount];
    
    pSelf->_lastPosition = position + bytesRead;

    if(actualCount)
        *actualCount = (UInt32) (copyMax + bytesRead);

    return noErr;
}

static SInt64 getSizeProc(void* clientData) {
    NSObject* _handle = (__bridge NSObject *)(clientData);
    CoreAudioDecoder * __unsafe_unretained pSelf = (id) _handle;

    return pSelf->_fileSize;
}

@implementation CoreAudioDecoder

- (void) close
{
	OSStatus			err;

    if (_in_opened) {
        err = ExtAudioFileDispose(_in);
        if(noErr != err) {
            DLog(@"Error closing ExtAudioFile");
        }
        _in_opened = NO;
    }
    
    if (_audioFile_opened) {
        err = AudioFileClose(_audioFile);
        if(noErr != err) {
            DLog(@"Error closing AudioFile");
        }
        _audioFile_opened = NO;
    }
    
    _audioSource = nil;
}

- (void) dealloc
{
    [self close];
}

- (BOOL)open:(id<CogSource>)source;
{
	OSStatus						err;
    
    _audioFile_opened = NO;
    _in_opened = NO;
    
    if (![source seekable])
        return NO;
    
    _audioSource = source;
    _lastPosition = [source tell];
    [source seek:0 whence:SEEK_END];
    _fileSize = [source tell];
    [source seek:_lastPosition whence:SEEK_SET];
    
    err = AudioFileOpenWithCallbacks((__bridge void *)self, readProc, 0, getSizeProc, 0, 0, &_audioFile);
    if(noErr != err) {
        ALog(@"Error opening callback interface to file: %d", err);
        return NO;
    }
    
    _audioFile_opened = YES;
    
    err = ExtAudioFileWrapAudioFileID(_audioFile, false, &_in);
	if(noErr != err) {
		ALog(@"Error opening file: %d", err);
		return NO;
	}
    
    _in_opened = YES;
	
	return [self readInfoFromExtAudioFileRef];
}

- (BOOL) readInfoFromExtAudioFileRef
{
	OSStatus						err;
	UInt32							size;
    UInt32                          asbdSize;
	AudioStreamBasicDescription		asbd;
    AudioFileID                     afi;
	
	// Get input file information
	asbdSize = sizeof(asbd);
	err		 = ExtAudioFileGetProperty(_in, kExtAudioFileProperty_FileDataFormat, &asbdSize, &asbd);
	if(err != noErr) {
		err = ExtAudioFileDispose(_in);
		return NO;
	}
	
	SInt64 total;
	size	= sizeof(total);
	err		= ExtAudioFileGetProperty(_in, kExtAudioFileProperty_FileLengthFrames, &size, &total);
	if(err != noErr) {
		err = ExtAudioFileDispose(_in);
		return NO;
	}
	totalFrames = total;
	
    size    = sizeof(afi);
    err     = ExtAudioFileGetProperty(_in, kExtAudioFileProperty_AudioFile, &size, &afi);
    if(err != noErr) {
        err = ExtAudioFileDispose(_in);
        return NO;
    }
    
    SInt32    formatBitsPerSample;
    size    = sizeof(formatBitsPerSample);
    err     = AudioFileGetProperty(afi, kAudioFilePropertySourceBitDepth, &size, &formatBitsPerSample);
    if(err != noErr) {
        if (err == kAudioFileUnsupportedPropertyError) {
            formatBitsPerSample = 0; // floating point formats apparently don't return this any more
        }
        else {
            err = ExtAudioFileDispose(_in);
            return NO;
        }
    }
    
    UInt32    _bitrate;
    size    = sizeof(_bitrate);
    err     = AudioFileGetProperty(afi, kAudioFilePropertyBitRate, &size, &_bitrate);
    if(err != noErr) {
        err = ExtAudioFileDispose(_in);
        return NO;
    }
	
    bitrate                = (_bitrate + 500) / 1000;
    
    CFStringRef formatName;
    size    = sizeof(formatName);
    err     = AudioFormatGetProperty(kAudioFormatProperty_FormatName, asbdSize, &asbd, &size, &formatName);
    if(err != noErr) {
        err = ExtAudioFileDispose(_in);
        return NO;
    }
    
    codec = (__bridge NSString *)formatName;
    
    CFRelease(formatName);
    
    NSRange range = [codec rangeOfString:@","];
    if (range.location != NSNotFound) {
        codec = [codec substringToIndex:range.location];
    }
    
	// Set our properties
	bitsPerSample		= formatBitsPerSample;
	channels			= asbd.mChannelsPerFrame;
	frequency			= asbd.mSampleRate;
	floatingPoint       = NO;
    
	// mBitsPerChannel will only be set for lpcm formats
	if(bitsPerSample <= 0) {
		bitsPerSample = 32;
        floatingPoint = YES;
	}
    
    _audioFile_is_lossy = NO;
    
    if (floatingPoint || [[codec lowercaseString] containsString:@"adpcm"] || [[codec lowercaseString] containsString:@"gsm"])
        _audioFile_is_lossy = YES;
	
	// Set output format
	AudioStreamBasicDescription		result;
	
	bzero(&result, sizeof(AudioStreamBasicDescription));
	
	result.mFormatID			= kAudioFormatLinearPCM;
    if (floatingPoint) {
        result.mFormatFlags     = kAudioFormatFlagIsFloat | kAudioFormatFlagIsPacked;
    }
    else {
        result.mFormatFlags		= kAudioFormatFlagIsSignedInteger | kLinearPCMFormatFlagIsBigEndian;
    }
	
	result.mSampleRate			= frequency;
	result.mChannelsPerFrame	= channels;
	result.mBitsPerChannel		= bitsPerSample;
	
	result.mBytesPerPacket		= channels * (bitsPerSample / 8);
	result.mFramesPerPacket		= 1;
	result.mBytesPerFrame		= channels * (bitsPerSample / 8);
	
	err = ExtAudioFileSetProperty(_in, kExtAudioFileProperty_ClientDataFormat, sizeof(result), &result);
	if(noErr != err) {
		err = ExtAudioFileDispose(_in);
		return NO;
	}
	
	[self willChangeValueForKey:@"properties"];
	[self didChangeValueForKey:@"properties"];
	
	return YES;
}

- (int) readAudio:(void *)buf frames:(UInt32)frames
{
	OSStatus				err;
	AudioBufferList			bufferList;
	UInt32					frameCount;
	
	// Set up the AudioBufferList
	bufferList.mNumberBuffers				= 1;
	bufferList.mBuffers[0].mNumberChannels	= channels;
	bufferList.mBuffers[0].mData			= buf;
	bufferList.mBuffers[0].mDataByteSize	= frames * channels * (bitsPerSample/8);
	
	// Read a chunk of PCM input (converted from whatever format)
	frameCount	= frames;
	err			= ExtAudioFileRead(_in, &frameCount, &bufferList);
	if(err != noErr) {
		return 0;
	}	
	
	return frameCount;
}

- (long) seek:(long)frame
{
	OSStatus			err;
	
	err = ExtAudioFileSeek(_in, frame);
	if(noErr != err) {
		return -1;
	}
	
	return frame;
}

+ (NSArray *)fileTypes
{
	OSStatus			err;
	UInt32				size;
	NSArray *sAudioExtensions;
	
	size	= sizeof(sAudioExtensions);
	err		= AudioFileGetGlobalInfo(kAudioFileGlobalInfo_AllExtensions, 0, NULL, &size, &sAudioExtensions);
	if(noErr != err) {
		return nil;
	}
	
	return sAudioExtensions;
}

+ (NSArray *)mimeTypes
{
    OSStatus           err;
    UInt32             size;
    NSArray *sAudioMIMETypes;
    
    size    = sizeof(sAudioMIMETypes);
    err     = AudioFileGetGlobalInfo(kAudioFileGlobalInfo_AllMIMETypes, 0, NULL, &size, &sAudioMIMETypes);
    if(noErr != err) {
        return nil;
    }
    
	return sAudioMIMETypes;
}

+ (float)priority
{
    return 1.0;
}

+ (NSArray *)fileTypeAssociations
{
    return @[
        @[@"WAVE File", @"wav.icns", @"wav", @"w64"],
        @[@"AIFF File", @"aiff.icns", @"aif", @"aiff", @"aifc"],
        @[@"CAF File", @"song.icns", @"caf"],
        @[@"AU File", @"song.icns", @"au"],
        @[@"MPEG Audio File", @"mp3.icns", @"mp3", @"mp2", @"mp1", @"m2a", @"mpa"],
        @[@"MPEG Stream File", @"song.icns", @"mpeg"],
        @[@"MPEG-4 Audio File", @"m4a.icns", @"m4a", @"mp4", @"m4b", @"m4r"],
        @[@"MPEG-4 AAC Audio File", @"song.icns", @"aac", @"adts"],
        @[@"AMR Audio File", @"song.icns", @"amr"],
        @[@"USAC Audio File", @"song.icns", @"xhe"],
        @[@"AC-3 Audio File", @"song.icns", @"ac3"],
        @[@"FLAC Audio File", @"flac.icns", @"flac"],
        @[@"SND Audio File", @"song.icns", @"snd"]
    ];
}

- (NSDictionary *)properties
{
	return [NSDictionary dictionaryWithObjectsAndKeys:
		[NSNumber numberWithInt:channels],@"channels",
		[NSNumber numberWithInt:bitsPerSample],@"bitsPerSample",
        [NSNumber numberWithBool:floatingPoint],@"floatingPoint",
		[NSNumber numberWithInt:bitrate],@"bitrate",
		[NSNumber numberWithFloat:frequency],@"sampleRate",
		[NSNumber numberWithLong:totalFrames],@"totalFrames",
		[NSNumber numberWithBool:YES], @"seekable",
        codec, @"codec",
        floatingPoint ? @"host" : @"big", @"endian",
        _audioFile_is_lossy ? @"lossy" : @"lossless", @"encoding",
		nil];
}


@end