// FD_BufNMF, an NRT buffer NMF Processor
// A tool from the FluCoMa project, funded by the European Research Council (ERC) under the European Union’s Horizon 2020 research and innovation programme (grant agreement No 725899)

#include "SC_PlugIn.h"
#include <vector>
#include "stft.h"

static InterfaceTable *ft;

void BufNMF(World *world, struct SndBuf *dstBuf, struct sc_msg_iter *msg)
{
	int dstFrameCount = dstBuf->frames;
	int dstChanCount = dstBuf->channels;

	uint32 srcBufNum = msg->geti();
	uint32 rank = msg->geti();
	uint32 fftSize = msg->geti();
	uint32 windowSize = msg->geti();
	uint32 hopSize = msg->geti();

	if (srcBufNum >= world->mNumSndBufs){
		Print("fdNMF is not happy because the source buffer does not exist.\n");
		return;
	}

	SndBuf* srcBuf = world->mSndBufs + srcBufNum;

	if (srcBuf->data == dstBuf->data){
		Print("fdNMF is not happy because the source buffer is the same as the destination buffer.\n");
		return;
	}

	int srcFrameCount = srcBuf->frames;
	int srcChanCount = srcBuf->channels;

	// if (dstChanCount < rank) {
	// 	Print("fdNMF is not happy because the destination buffer has a lower channel count than the number of ranks.\n");
	// 	return;
	// }

	// make a vector of doubles for the samples
	std::vector<double> tmp_vec(srcFrameCount);

	//construct STFT processors
	stft::STFT stft(windowSize, fftSize, hopSize);
	stft::ISTFT istft(windowSize, fftSize, hopSize);

	//for each channels
	for (int j=0;j<srcChanCount;j++){
		//copies and casts to double the source samples
		for (int i=0;i<srcFrameCount;i++){
				tmp_vec[i] = srcBuf->data[(i*srcChanCount)+j];
		}

		// fft and ifft for fun
		stft::Spectrogram spec = stft.process(tmp_vec);
		std::vector<double>  out_vec = istft.process(spec);

		//writes the output
		for (int i=0;i<srcFrameCount;i++){
				dstBuf->data[(i*srcChanCount)+j] = (float)out_vec[i];
		}
	}
}

PluginLoad(OfflineFluidDecompositionUGens) {
	ft = inTable;
	DefineBufGen("BufNMF", BufNMF);
}