#include "../include/AudioBufferSource.h"

using namespace std;
using namespace openshot;

// Default constructor
AudioBufferSource::AudioBufferSource(int samples, int channels)
		: position(0), start(0), repeat(false)
{
	// Create new buffer
	buffer = new AudioSampleBuffer(channels, samples);
	buffer->clear();

//	AudioFormatManager formatManager;
//	formatManager.registerBasicFormats();
//
//	File file ("/home/jonathan/Aptana Studio Workspace/OpenShotLibrary/src/examples/piano.wav");
//	AudioFormatReader* reader = formatManager.createReaderFor (file);
//	if (reader != 0)
//	{
//		buffer = new AudioSampleBuffer(reader->numChannels, reader->lengthInSamples);
//		buffer->readFromAudioReader(reader, 0, reader->lengthInSamples, 0, true, true);
//		float* firstChannelSamples = buffer->getSampleData(0, 0);
//	}
}

// Destructor
AudioBufferSource::~AudioBufferSource()
{
	// Clear and delete the buffer
	buffer->clear();
	delete buffer;
	buffer = NULL;
};

// Get the next block of audio samples
void AudioBufferSource::getNextAudioBlock (const AudioSourceChannelInfo& info)
{
	int buffer_samples = buffer->getNumSamples();
	int buffer_channels = buffer->getNumChannels();

	if (info.numSamples > 0) {
		int start = position;
		int number_to_copy = 0;

		// Determine how many samples to copy
		if (start + info.numSamples <= buffer_samples)
		{
			// copy the full amount requested
			number_to_copy = info.numSamples;
		}
		else if (start > buffer_samples)
		{
			// copy nothing
			number_to_copy = 0;
		}
		else if (buffer_samples - start > 0)
		{
			// only copy what is left in the buffer
			number_to_copy = buffer_samples - start;
		}
		else
		{
			// copy nothing
			number_to_copy = 0;
		}

		// Determine if any samples need to be copied
		if (number_to_copy > 0)
		{
			// Loop through each channel and copy some samples
			for (int channel = 0; channel < buffer_channels; channel++)
				info.buffer->copyFrom(channel, info.startSample, *buffer, channel, start, number_to_copy);

			// Update the position of this audio source
			position += number_to_copy;
		}

	}

}

// Prepare to play this audio source
void AudioBufferSource::prepareToPlay(int, double) { };

// Release all resources
void AudioBufferSource::releaseResources()
{
	// Clear the buffer
	buffer->clear();
};

// Set the next read position of this source
void AudioBufferSource::setNextReadPosition (long long newPosition)
{
	// set position (if the new position is in range)
	if (newPosition > 0 && newPosition < buffer->getNumSamples())
		position = newPosition;
};

// Get the next read position of this source
long long AudioBufferSource::getNextReadPosition() const
{
	// return the next read position
	return position;
};

// Get the total length (in samples) of this audio source
long long AudioBufferSource::getTotalLength() const
{
	// Get the length
	return buffer->getNumSamples();
};

// Determines if this audio source should repeat when it reaches the end
bool AudioBufferSource::isLooping() const
{
	// return if this source is looping
	return repeat;
};

// Set if this audio source should repeat when it reaches the end
void AudioBufferSource::setLooping (bool shouldLoop)
{
	// Set the repeat flag
	repeat = shouldLoop;
};

// Add audio samples to a specific channel
void AudioBufferSource::AddAudio(int destChannel, int destStartSample, const float* source, int numSamples, float gainToApplyToSource = 1.0f)
{
	// Add samples to frame's audio buffer
	buffer->addFrom(destChannel, destStartSample, source, numSamples, gainToApplyToSource);
}