#include "tsio_raw.h"

#include "tsound_t.h"

#include "tsystem.h"

#include "tfilepath_io.h"

using namespace std;

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

TSoundTrackReaderRaw::TSoundTrackReaderRaw(const TFilePath &fp)

	: TSoundTrackReader(fp)

{

}

//------------------------------------------------------------------------------

TSoundTrackP TSoundTrackReaderRaw::load()

{

	Tifstream is(m_path);

	if (!is)

		throw TException(L"Unable to load the RAW file " +

						 m_path.getWideString() + L" : doesn't exist");

	is.seekg(0, ios_base::end);

	long sampleCount = is.tellg() / 2;

	is.seekg(0, ios_base::beg);

	TSoundTrack *track = new TSoundTrackMono16(22050, 1, sampleCount);

	is.read((char *)track->getRawData(), sampleCount * 2);

	return track;

}

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

TSoundTrackWriterRaw::TSoundTrackWriterRaw(const TFilePath &fp)

	: TSoundTrackWriter(fp)

{

}

//------------------------------------------------------------------------------

bool TSoundTrackWriterRaw::save(const TSoundTrackP &track)

{

	TFileStatus fs(m_path);

	if (fs.doesExist() && !fs.isWritable())

		throw TException(L"Unable to save the soundtrack: " +

						 m_path.getWideString() + L" is read-only");

	Tofstream os(m_path);

	//inserisco i dati nell'output stream

	// os << (ULONG) (track->getSampleCount()) << " ";

	int sampleCount = track->getSampleCount();

	int i;

	switch (track->getChannelCount()) {

	case 1: // mono

		switch (track->getSampleSize()) {

		case 1: // un byte per campione

		{

			const char *srcCharSamples = (char *)track->getRawData();

			for (i = 0; i < sampleCount; i++) {

				short sample = *(srcCharSamples + i);

				os.write((char *)&sample, sizeof(short));

			}

			break;

		}

		case 2: // due byte per campione

		{

			const short *srcShortSamples = (short *)track->getRawData();

			for (i = 0; i < sampleCount; i++)

				os.write((char *)(srcShortSamples + i), sizeof(short));

			break;

		}

		}

		break;

	case 2: // stereo

		switch (track->getSampleSize()) {

		case 2: // due byte per campione, un byte per canale

		{

			// considera solo il canale sinistro

			const char *srcCharSamples = (char *)track->getRawData();

			for (i = 0; i < sampleCount; i += 2) {

				short sample = *(srcCharSamples + i);

				os.write((char *)&sample, sizeof(short));

			}

			break;

		}

		case 4: // quattro byte per campione, due byte per canale

		{

			// considera solo il canale sinistro

			const short *srcShortSamples = (short *)track->getRawData();

			for (i = 0; i < sampleCount; i += 2)

				os.write((char *)(srcShortSamples + i), sizeof(short));

			break;

		}

		}

		break;

	default:

		break;

	}

	return true;

}