love2d.love/src/modules/sound/lullaby/ModPlugDecoder.cpp

/**
 * Copyright (c) 2006-2022 LOVE Development Team
 *
 * This software is provided 'as-is', without any express or implied
 * warranty.  In no event will the authors be held liable for any damages
 * arising from the use of this software.
 *
 * Permission is granted to anyone to use this software for any purpose,
 * including commercial applications, and to alter it and redistribute it
 * freely, subject to the following restrictions:
 *
 * 1. The origin of this software must not be misrepresented; you must not
 *    claim that you wrote the original software. If you use this software
 *    in a product, an acknowledgment in the product documentation would be
 *    appreciated but is not required.
 * 2. Altered source versions must be plainly marked as such, and must not be
 *    misrepresented as being the original software.
 * 3. This notice may not be removed or altered from any source distribution.
 **/

#include "ModPlugDecoder.h"

#ifndef LOVE_NO_MODPLUG

#include "common/Exception.h"
#include "common/Data.h"

namespace love
{
namespace sound
{
namespace lullaby
{

ModPlugDecoder::ModPlugDecoder(Stream *stream, int bufferSize)
	: Decoder(stream, bufferSize)
	, plug(0)
	, duration(-2.0)
{
	// Set some ModPlug settings.
	settings.mFlags = MODPLUG_ENABLE_OVERSAMPLING | MODPLUG_ENABLE_NOISE_REDUCTION;
	settings.mChannels = 2;
	settings.mBits = 16;
	settings.mFrequency = sampleRate;
	settings.mResamplingMode = MODPLUG_RESAMPLE_LINEAR;

	// fill with modplug defaults (modplug _memsets_, so we could get
	// garbage settings when the struct is only partially initialized)
	// This does not exist yet on Windows.

	settings.mStereoSeparation = 128;
	settings.mMaxMixChannels = 32;
	settings.mReverbDepth = 0;
	settings.mReverbDelay = 0;
	settings.mBassAmount = 0;
	settings.mBassRange = 0;
	settings.mSurroundDepth = 0;
	settings.mSurroundDelay = 0;
	settings.mLoopCount = -1;

	ModPlug_SetSettings(&settings);

	// ModPlug has no streaming API. Miserable.
	// We don't want to load the entire stream immediately if it's big, because
	// it might not be compatible with ModPlug. So we just try to load 4MB and
	// see if that works, and then load the whole thing if it does.
	if (stream->getSize() > 1024 * 1024 * 4)
	{
		data.set(stream->read(1024 * 1024 * 4), Acquire::NORETAIN);

		plug = ModPlug_Load(data->getData(), (int)data->getSize());

		if (plug == nullptr)
			throw love::Exception("Could not load file with ModPlug.");

		stream->seek(0);
		ModPlug_Unload(plug);
	}

	data.set(stream->read(stream->getSize()), Acquire::NORETAIN);

	// Load the module.
	plug = ModPlug_Load(data->getData(), (int)data->getSize());

	if (plug == nullptr)
		throw love::Exception("Could not load file with ModPlug.");

	// set master volume for delicate ears
	ModPlug_SetMasterVolume(plug, 128);
}

ModPlugDecoder::~ModPlugDecoder()
{
	if (plug != nullptr)
		ModPlug_Unload(plug);
}

int ModPlugDecoder::probe(Stream* stream)
{
	// Ideally we want to probe every single format that ModPlug supports.
	Data *data = stream->read(1024 * 1024 * 4);
	ModPlugFile *plug = ModPlug_Load(data->getData(), (int)data->getSize());

	if (plug)
		ModPlug_Unload(plug);

	data->release();
	return plug ? 80 : 0;
}

love::sound::Decoder *ModPlugDecoder::clone()
{
	StrongRef<Stream> s(stream->clone(), Acquire::NORETAIN);
	return new ModPlugDecoder(s, bufferSize);
}

int ModPlugDecoder::decode()
{
	int r =  ModPlug_Read(plug, buffer, bufferSize);

	if (r == 0)
		eof = true;

	return r;
}

bool ModPlugDecoder::seek(double s)
{
	ModPlug_Seek(plug, (int)(s*1000.0));
	return true;
}

bool ModPlugDecoder::rewind()
{
	// Let's reload.
	ModPlug_Unload(plug);
	plug = ModPlug_Load(data->getData(), (int) data->getSize());
	ModPlug_SetMasterVolume(plug, 128);
	eof = false;
	return (plug != 0);
}

bool ModPlugDecoder::isSeekable()
{
	return true;
}

int ModPlugDecoder::getChannelCount() const
{
	return 2;
}

int ModPlugDecoder::getBitDepth() const
{
	return 16;
}

double ModPlugDecoder::getDuration()
{
	// Only calculate the duration if we haven't done so already.
	if (duration == -2.0)
	{
		int lengthms = ModPlug_GetLength(plug);

		if (lengthms < 0)
			duration = -1.0;
		else
			duration = (double) lengthms / 1000.0;
	}

	return duration;
}

} // lullaby
} // sound
} // love

#endif // LOVE_NO_MODPLUG