//
//  Sound.m
//  Cog
//
//  Created by Zaphod Beeblebrox on 5/11/05.
//  Copyright 2005 __MyCompanyName__. All rights reserved.
//

#include <mach/mach_init.h>
#include <mach/thread_policy.h>
#include <mach/thread_act.h>
#include <malloc/malloc.h>

#import "Sound.h"
#import "FlacFile.h"
#import "MonkeysFile.h"
#import "MPEGFile.h"
#import "MusepackFile.h"
#import "VorbisFile.h"
#import "WaveFile.h"

#import "DBLog.h"

//128kb of floating point data gives ya ~371 milliseconds...so 512kb should give us about 1.5 seconds
#define RING_BUFFER_SIZE (512 * 1024)
//this much data is about 56 milliseconds of floating point data
#define BUFFER_WRITE_CHUNK (16 * 1024)

//#define RING_BUFFER_SIZE 1048576
//#define BUFFER_WRITE_CHUNK 32768
#define FEEDER_THREAD_IMPORTANCE 4

//timeout should be smaller than the time itd take for the buffer to run dry...looks like were ironclad
#define TIMEOUT 1

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");
}

@implementation Sound

//called from the complexfill when the audio is converted...good clean fun
static OSStatus Sound_ACInputProc(AudioConverterRef inAudioConverter, UInt32* ioNumberDataPackets, AudioBufferList* ioData, AudioStreamPacketDescription** outDataPacketDescription, void* inUserData)
{
	Sound *sound = (Sound *)inUserData;
	OSStatus err = noErr;
	
	int amountToWrite;
	int amountWritten;
	void *sourceBuf;

	amountToWrite = (*ioNumberDataPackets)*sound->sourceStreamFormat.mBytesPerPacket;
	sourceBuf = malloc(amountToWrite);
	sound->conversionBuffer = sourceBuf;
	
	amountWritten = [sound->soundFile fillBuffer:sourceBuf ofSize:amountToWrite];
	
//	DBLog(@"PACKET NUMBER RECEIVED: %i", *ioNumberDataPackets);
	ioData->mBuffers[0].mData = sourceBuf;
	ioData->mBuffers[0].mDataByteSize = amountWritten;
	ioData->mBuffers[0].mNumberChannels = sound->sourceStreamFormat.mChannelsPerFrame;
	ioData->mNumberBuffers = 1;
		
	return err;
}

//called from coreaudio, just fill the bufferlist and thats all
static OSStatus Sound_Renderer(void *inRefCon,  AudioUnitRenderActionFlags *ioActionFlags, const AudioTimeStamp  *inTimeStamp, UInt32 inBusNumber, UInt32 inNumberFrames, AudioBufferList  *ioData)
{
	Sound *sound = (Sound *)inRefCon;
	OSStatus err = noErr;
	
	int amountAvailable;
	int amountToRead;
	void *readPointer;
	
	[sound->readLock lock];
	
	amountAvailable = [sound->readRingBuffer lengthAvailableToReadReturningPointer:&readPointer];
	if (sound->playbackStatus == kCogStatusEndOfFile && amountAvailable == 0)
	{
//		DBLog(@"FILE CHANGED!!!!!");
		[sound sendPortMessage:kCogFileChangedMessage];
		sound->readRingBuffer = [sound oppositeBuffer:sound->readRingBuffer];

		[sound setPlaybackStatus:kCogStatusPlaying];
		
		sound->currentPosition = 0;
		
		double time = [sound calculateTime:sound->totalLength];
		int bitrate = [sound->soundFile bitRate];
		[sound sendPortMessage:kCogLengthUpdateMessage withData:&time ofSize:(sizeof(double))];
		[sound sendPortMessage:kCogBitrateUpdateMessage withData:&bitrate ofSize:(sizeof(int))];
	}
	if (sound->playbackStatus == kCogStatusEndOfPlaylist && amountAvailable == 0)
	{
		//Stop playback
		[sound setPlaybackStatus:kCogStatusStopped];
//		return err;
	}

	if (amountAvailable < ([sound->readRingBuffer bufferLength] - BUFFER_WRITE_CHUNK))
	{
//		DBLog(@"AVAILABLE: %i", amountAvailable);
		[sound fireFillTimer];
	}
	
	if (amountAvailable > inNumberFrames*sound->deviceFormat.mBytesPerPacket)
		amountToRead = inNumberFrames*sound->deviceFormat.mBytesPerPacket;
	else
		amountToRead = amountAvailable;
	
	memcpy(ioData->mBuffers[0].mData, readPointer, amountToRead);
	ioData->mBuffers[0].mDataByteSize = amountToRead;
	
	[sound->readRingBuffer didReadLength:amountToRead];

	sound->currentPosition += amountToRead;
	
	[sound->readLock unlock];
	
	return err;
}	

- (id)init
{
	DBLog(@"HEllo");
	self = [super init];
	if (self)
	{
		readLock = [[NSLock alloc] init];
		writeLock = [[NSLock alloc] init];
		
		ringBuffer = [[VirtualRingBuffer alloc] initWithLength:RING_BUFFER_SIZE];
		auxRingBuffer = [[VirtualRingBuffer alloc] initWithLength:RING_BUFFER_SIZE];
		
		readRingBuffer = ringBuffer;
		writeRingBuffer = ringBuffer;
	}
	
	return self;
}

- (void)launchThreadWithPort:(id)inData
{
	NSAutoreleasePool *pool;
	pool = [[NSAutoreleasePool alloc] init];
	
	distantPort = (NSPort *)inData;
	
	sendPort = [NSPort port];
	if (sendPort)
	{
		[sendPort setDelegate:self];
		
		NSArray *modes = [NSArray arrayWithObjects:NSDefaultRunLoopMode, nil];//NSEventTrackingRunLoopMode, nil];
		NSEnumerator *enumerator;
		NSString *mode;
		
		enumerator = [modes objectEnumerator];
		while ((mode = [enumerator nextObject]))
			[[NSRunLoop currentRunLoop] addPort:sendPort forMode:mode];
	}
//	DBLog(@"SENDING CHECKIN MESSAGE");
	[self sendPortMessage:kCogCheckinMessage];

	[self setupAudioOutput];
	
	[self setThreadPolicy];

	[[NSRunLoop currentRunLoop] run];
	DBLog(@"THREAD EXIT!!!!!!!!!!!");
	[pool release];
}

/*- (void)printDebugInfo:(id)sender
{
	void *ptr;
	
	DBLog(@"Available r/w: %i/%i", [ringBuffer lengthAvailableToReadReturningPointer:&ptr], [ringBuffer lengthAvailableToWriteReturningPointer:&ptr]);
}
*/
- (void)setThreadPolicy
{
    // Increase this thread's priority, and turn off timesharing.  
	
    kern_return_t error;
    thread_extended_policy_data_t extendedPolicy;
    thread_precedence_policy_data_t precedencePolicy;
	
    extendedPolicy.timeshare = 0;
    error = thread_policy_set(mach_thread_self(), THREAD_EXTENDED_POLICY,  (thread_policy_t)&extendedPolicy, THREAD_EXTENDED_POLICY_COUNT);

    if (error != KERN_SUCCESS) {
        DBLog(@"Couldnt set feeder thread's extended policy");
    }
	
    precedencePolicy.importance = FEEDER_THREAD_IMPORTANCE;
    error = thread_policy_set(mach_thread_self(), THREAD_PRECEDENCE_POLICY, (thread_policy_t)&precedencePolicy, THREAD_PRECEDENCE_POLICY_COUNT);

    if (error != KERN_SUCCESS) {
        DBLog(@"Couldn't set feeder thread's precedence policy");
    }
}

- (void)sendPortMessage:(int)msgid
{
	NSPortMessage *portMessage = [[NSPortMessage alloc] initWithSendPort:distantPort receivePort:sendPort components:nil];
	
	if (portMessage)
	{
		NSDate *date = [[NSDate alloc] initWithTimeIntervalSinceNow:20.0];//[[NSDate alloc] init];
		
		[portMessage setMsgid:msgid];
		
		DBLog(@"Sending message (nodata): %i", msgid);

		[portMessage sendBeforeDate:date];
		
		[date release];
		[portMessage release];
	}
}

- (void)sendPortMessage:(int)msgid withData:(void *)data ofSize:(int)size
{
	NSPortMessage *portMessage;
	NSData *d = [[NSData alloc] initWithBytes:data length:size];
	NSArray *a = [[NSArray alloc] initWithObjects:d,nil];
	portMessage = [[NSPortMessage alloc] initWithSendPort:distantPort receivePort:sendPort components:a];
	
	[a release];
	[d release];
	
	if (portMessage)
	{
		NSDate *date = [[NSDate alloc] initWithTimeIntervalSinceNow:20.0];//give shit a little time to send, just in case...may come back to bite me
		if ([date laterDate:[NSDate date]] != date)
		{
			DBLog(@"WTF");
		}
		[portMessage setMsgid:msgid];
		DBLog(@"Sending message: %i", msgid);
		
		NS_DURING
			[portMessage sendBeforeDate:date];
		NS_HANDLER
				NSRunAlertPanel(@"Error Panel", @"%@", @"OK", nil, nil, 
						localException);
		NS_ENDHANDLER

		[date release];
		[portMessage release];
	}
	else {
		DBLog(@"DIDNT SEND! ERROR");
	}
	
}

// Worker thread message handler method.
- (void)handlePortMessage:(NSPortMessage *)portMessage
{
	unsigned int msgid = [portMessage msgid];
		
	// Handle messages from the main thread.
	if (msgid == kCogPauseResumeMessage)
	{
		if (playbackStatus == kCogStatusPaused)
		{
			[self resume];
		}
		else
		{
			[self pause];
		}
	}
	else if (msgid == kCogPauseMessage)
	{
		[self pause];
	}
	else if (msgid == kCogResumeMessage)
	{
		[self resume];
	}
	else if (msgid == kCogStopMessage)
	{
		[self stop];
	}
	else if (msgid == kCogPlayFileMessage || msgid == kCogChangeFileMessage)
	{
		NSArray* components = [portMessage components];
		NSData *data = [components objectAtIndex:0];
		NSString *s = [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding];		
		
		if (msgid == kCogPlayFileMessage) //PLAY THE FARKING FILE NOW
		{
			[self playFile:s];
		}
		else if (kCogChangeFileMessage) //change the file, usually in response to a nexttrack request
		{
			[self changeFile:s];
		}
		
		[s release];
	}
	else if (msgid == kCogSeekMessage)
	{
//		DBLog(@"MESAGE RECEIVED");
		NSArray* components = [portMessage components];
		NSData *data = [components objectAtIndex:0];
		double time;
		double newTime;
		unsigned long pos;
		
		time = (*(double *)[data bytes]);
		pos = [self calculatePos:time];
		
		newTime = [soundFile seekToTime:time];
		if (newTime >= 0.0)
		{
			DBLog(@"RESETTING");
			[self resetBuffer];
			
			pos = [self calculatePos:newTime];

			[readLock lock];
			currentPosition = pos;
			[readLock unlock];
		}
		else
		{
			DBLog(@"Not resetting: %f", newTime);
			newTime = [self calculateTime:currentPosition];
		}
		//send a message with newTime
		DBLog(@"RESETING TIME TO: %f", newTime);
		[self sendPortMessage:kCogPositionUpdateMessage withData:&newTime ofSize:(sizeof(double))];
	}
	else if (msgid == kCogEndOfPlaylistMessage)
	{
		[self setPlaybackStatus:kCogStatusEndOfPlaylist];
	}
}

- (void)startPositionTimer
{
	positionTimer = [NSTimer scheduledTimerWithTimeInterval:1 target:self selector:@selector(sendPositionUpdate:) userInfo:nil repeats:YES];
}

- (void)stopPositionTimer
{
	[positionTimer invalidate];
	positionTimer = nil;
}

- (void)scheduleFillTimer
{
	fillTimer = [NSTimer scheduledTimerWithTimeInterval:TIMEOUT target:self selector:@selector(fillBuffer:) userInfo:nil repeats:NO];
}

//the only method in the whole fucking thing that should be called by an outside thread
- (void)fireFillTimer
{
	NSDate *date = [[NSDate alloc] init];
//	DBLog(@"FIRED");
	[fillTimer  setFireDate:date];

	[date release];
}

- (void)fillBuffer:(id)sender
{
	int amountAvailable;
	int convertedSize;
	void *writePointer;
	
	if (playbackStatus == kCogStatusStopped)
	{
		DBLog(@"STOPPING");
		[self stop];
		return;
	}
	
	[writeLock lock];
	
	//We do this so uhm, newly started songs dont have to wait for the entire buffer to fill for data to become available.
	amountAvailable = [writeRingBuffer lengthAvailableToWriteReturningPointer:&writePointer];

	int i;
	for (i = 0; i < amountAvailable; i =+ BUFFER_WRITE_CHUNK)
	{
		int amountToWrite;
		if (amountAvailable > BUFFER_WRITE_CHUNK)
			amountToWrite = BUFFER_WRITE_CHUNK;
		else
			amountToWrite = amountAvailable;
		
		convertedSize = [self convert:writePointer packets:(amountToWrite/deviceFormat.mBytesPerPacket)];
//	convertedSize = [self convert:writePointer packets:(amountAvailable/deviceFormat.mBytesPerPacket)];
		if (playbackStatus == kCogStatusPlaying && convertedSize == 0)
		{
//			DBLog(@"NEXT!!!!");
			[self sendPortMessage:kCogRequestNextFileMessage];
			writeRingBuffer = [self oppositeBuffer:writeRingBuffer];
			
			[self setPlaybackStatus:kCogStatusEndOfFile];
		}
			
//		writePointer = (char *)writePointer + convertedSize;
		[writeRingBuffer didWriteLength:convertedSize];

		amountAvailable = [writeRingBuffer lengthAvailableToWriteReturningPointer:&writePointer];
	}
	[writeLock unlock];
//	DBLog(@"WRote: %i", convertedSize);
		
	[self scheduleFillTimer];
}

- (VirtualRingBuffer *)oppositeBuffer:(VirtualRingBuffer *)buf
{
	if (buf == ringBuffer)
		return auxRingBuffer;
	else
		return ringBuffer;
}

- (double)calculateTime:(unsigned long) pos
{
	return ((pos/deviceFormat.mBytesPerPacket)/(deviceFormat.mSampleRate/1000.0));
}

- (unsigned long)calculatePos:(double) time
{
	return (unsigned long)(time * (deviceFormat.mSampleRate/1000.0)) * deviceFormat.mBytesPerPacket;
}

- (void)sendPositionUpdate:(id)sender
{
	//this may be a bad idea
	[readLock lock];
	unsigned long pos = currentPosition;
	[readLock unlock];

	double time = [self calculateTime:pos];
	
	[self sendPortMessage:kCogPositionUpdateMessage withData:&time ofSize:(sizeof(double))];
}

- (int)convert:(void *)destBuf packets:(int)numPackets
{	
	AudioBufferList ioData;
	UInt32 ioNumberFrames;
	UInt32 destSize;
	OSStatus err;
	
	destSize = numPackets*deviceFormat.mBytesPerPacket;
	
	ioNumberFrames = numPackets;
	ioData.mBuffers[0].mData = destBuf;
	ioData.mBuffers[0].mDataByteSize = destSize;
	ioData.mBuffers[0].mNumberChannels = deviceFormat.mChannelsPerFrame;
	ioData.mNumberBuffers = 1;

//	DBLog(@"THIS IS RETARDED: %i", ioData.mBuffers[0].mData);
//	DBLog(@"NUMBER OF PACKETS REQUESTED: %i", ioNumberFrames);
	err = AudioConverterFillComplexBuffer(converter, Sound_ACInputProc, self, &ioNumberFrames, &ioData, NULL);
	if (err != noErr)
		DBLog(@"Converter error: %i", err);
//	DBLog(@"THIS IS ANNOYING: %i", ioData.mBuffers[0].mData);
	free(conversionBuffer);

	
//	DBLog(@"HERE: %i/%i", destSize,  ioData.mBuffers[0].mDataByteSize);

//	DBLog(@"%i/%i",old, inNumberFrames);
	return ioData.mBuffers[0].mDataByteSize;
}

- (BOOL)setupAudioOutput
{
	ComponentDescription desc;  
	OSStatus err;
	
	desc.componentType = kAudioUnitType_Output;
	desc.componentSubType = kAudioUnitSubType_DefaultOutput;
	desc.componentManufacturer = kAudioUnitManufacturer_Apple;
	desc.componentFlags = 0;
	desc.componentFlagsMask = 0;
	
	Component comp = FindNextComponent(NULL, &desc);  //Finds an component that meets the desc spec's
	if (comp == NULL)
		return NO;
	
	err = OpenAComponent(comp, &outputUnit);  //gains access to the services provided by the component
	if (err)
		return NO;
	
	// Initialize AudioUnit 
	err = AudioUnitInitialize(outputUnit);
	if (err != noErr)
		return NO;
	
	
	UInt32 size = sizeof (AudioStreamBasicDescription);
	Boolean outWritable;
	//Gets the size of the Stream Format Property and if it is writable
	AudioUnitGetPropertyInfo(outputUnit,  
							 kAudioUnitProperty_StreamFormat,
							 kAudioUnitScope_Output, 
							 0, 
							 &size, 
							 &outWritable);
	//Get the current stream format of the output
	err = AudioUnitGetProperty (outputUnit,
								kAudioUnitProperty_StreamFormat,
								kAudioUnitScope_Output,
								0,
								&deviceFormat,
								&size);

	if (err != noErr)
		return NO;
	
/*	// change output format...
	deviceFormat.mFormatFlags = kLinearPCMFormatFlagIsSignedInteger;
	deviceFormat.mBytesPerFrame = 4;
	deviceFormat.mBitsPerChannel = 16;
	deviceFormat.mBytesPerPacket = 4;
	DBLog(@"stuff: %i %i %i %i", deviceFormat.mBitsPerChannel, deviceFormat.mBytesPerFrame, deviceFormat.mBytesPerPacket, deviceFormat.mFramesPerPacket);
	err = AudioUnitSetProperty (outputUnit,
								kAudioUnitProperty_StreamFormat,
								kAudioUnitScope_Output,
								0,
								&deviceFormat,
								size);
*/
	//Set the stream format of the output to match the input
	err = AudioUnitSetProperty (outputUnit,
								kAudioUnitProperty_StreamFormat,
								kAudioUnitScope_Input,
								0,
								&deviceFormat,
								size);
	
	//setup render callbacks
	renderCallback.inputProc = Sound_Renderer;
	renderCallback.inputProcRefCon = self;
	
	AudioUnitSetProperty(outputUnit, kAudioUnitProperty_SetRenderCallback, kAudioUnitScope_Input, 0, &renderCallback, sizeof(AURenderCallbackStruct));	
	
//	DBLog(@"Audio output successfully initialized");
	return (err == noErr);
}	

- (BOOL)startAudioOutput
{
	return (noErr == AudioOutputUnitStart(outputUnit));    
}

- (void)cleanUpAudioOutput
{
	AudioOutputUnitStop(outputUnit);//you must stop the audio unit
	AudioUnitUninitialize (outputUnit);
	CloseComponent(outputUnit);
}		

- (void)stopAudioOutput
{
	if (outputUnit)
        AudioOutputUnitStop(outputUnit);
}

- (BOOL)prepareSoundFile
{
//	DBLog(@"This fucker hates me");
	[soundFile getFormat:&sourceStreamFormat];
	
#ifdef DEBUG
	PrintStreamDesc(&sourceStreamFormat);
	PrintStreamDesc(&deviceFormat);
#endif
	
	//Make the converter
	OSStatus stat = noErr;
	stat = AudioConverterNew ( &sourceStreamFormat, &deviceFormat, &converter);
//	DBLog(@"Created converter");
	if (stat != noErr)
	{
		DBLog(@"Error creating converter %i", stat);
	}

	unsigned long length;
	length = (unsigned long)([soundFile totalSize] * (deviceFormat.mSampleRate/sourceStreamFormat.mSampleRate) * (deviceFormat.mBytesPerPacket/sourceStreamFormat.mBytesPerPacket));
	
	//again, may be bad
	[readLock lock];
	totalLength = length;
	[readLock unlock];
//	DBLog(@"AM I CRASHING?");
	return (stat == noErr);
}

- (void)cleanUpSoundFile
{
	AudioConverterDispose(converter);
	[soundFile close];
}

- (void)setPlaybackStatus:(int)s
{
	playbackStatus = s;
	DBLog(@"SENDING MESSAGE");
	[self sendPortMessage:kCogStatusUpdateMessage withData:&s ofSize:(sizeof(int))];
	DBLog(@"MESSAGE SENT");
}

- (void)pause
{
	[self stopAudioOutput];

	oldPlaybackStatus = playbackStatus;
	[self setPlaybackStatus:kCogStatusPaused];

	[self stopPositionTimer];
}

- (void)resume
{
	[self setPlaybackStatus:oldPlaybackStatus];
	
	[self startAudioOutput];

	[self startPositionTimer];
}

- (void)stop
{
	DBLog(@"STOPPING 2");
	[self stopAudioOutput];
	DBLog(@"Audio output stopped");
	[self resetBuffer];
	DBLog(@"BUFFERS RESET");
	
	[self setPlaybackStatus:kCogStatusStopped];

//	DBLog(@"HERE? PORT CONFLICT...FUCK");
	unsigned long pos = 0;
	DBLog(@"STOPPED 0");
	[self sendPortMessage:kCogPositionUpdateMessage withData:&pos ofSize:(sizeof(unsigned long))];
//	DBLog(@"THIS IS UBER SHITE: %@", positionTimer);

	DBLog(@"STOPPED 1");
	
	[self stopPositionTimer];
	
	//	DBLog(@"INVALIDATED");
	DBLog(@"STOPPED 2");
}

- (void)playFile:(NSString *)filename
{
	[self resetBuffer];
	[self stopPositionTimer];
//	[self stop];

	DBLog(@"PLAYING FILE");
	[self setSoundFile:filename];

	DBLog(@"DONT LIKE THIS, HUH?");

	[readLock lock];
	unsigned long length = totalLength;
	int bitrate = [soundFile bitRate];
	[readLock unlock];
	
	double time = [self calculateTime:length];
	
	[self sendPortMessage:kCogLengthUpdateMessage withData:&time ofSize:(sizeof(double))];
	[self sendPortMessage:kCogBitrateUpdateMessage withData:&bitrate ofSize:(sizeof(int))];
	
	[self setPlaybackStatus:kCogStatusPlaying];

	[self fillBuffer:self];
	[self startAudioOutput];

	[self startPositionTimer];	
}

- (void)changeFile:(NSString *)filename
{
	[self setSoundFile:filename];
	[self fireFillTimer];
}

- (void)resetBuffer
{
	[writeLock lock];
	[readLock lock];
	
	[ringBuffer empty];
	[auxRingBuffer empty];
	
	readRingBuffer = ringBuffer;
	writeRingBuffer = ringBuffer;
	
	currentPosition = 0;
	
	[readLock unlock];
	[writeLock unlock];
}

- (void)setSoundFile:(NSString *)filename
{
	[self cleanUpSoundFile];
	[soundFile release];
	
	//GO THROUGH HELLA SHIT TO DETERMINE FILE...NEED TO MAKE SOME KIND OF REGISTERING MECHANISM
	soundFile = [SoundFile open:filename];

//	DBLog(@"File opened: %s", [filename UTF8String]);
	[self prepareSoundFile];
}


@end