#include "audionoiseeffect.h"

#include <QDateTime>

AudioNoiseEffect::AudioNoiseEffect(Clip* c) : Effect(c, EFFECT_TYPE_AUDIO, AUDIO_NOISE_EFFECT) {
	amount_val = add_row("Amount:")->add_field(EFFECT_FIELD_DOUBLE);
	amount_val->set_double_minimum_value(0);
	amount_val->set_double_maximum_value(100);
	amount_val->set_double_default_value(20);

	mix_val = add_row("Mix:")->add_field(EFFECT_FIELD_BOOL);
	mix_val->set_bool_value(true);

	srand(QDateTime::currentMSecsSinceEpoch());
}

void AudioNoiseEffect::process_audio(double timecode_start, double timecode_end, quint8 *samples, int nb_bytes, int) {
	double interval = (timecode_end - timecode_start)/nb_bytes;
	for (int i=0;i<nb_bytes;i+=4) {
		double timecode = timecode_start+(interval*i);

		qint16 left_noise_sample = rand();
		qint16 right_noise_sample = rand();

		// set noise volume
		left_noise_sample *= amount_val->get_double_value(timecode)*0.01;
		right_noise_sample *= amount_val->get_double_value(timecode)*0.01;

		// mix with source audio
		if (mix_val->get_bool_value(timecode)) {
			qint16 left_sample = (qint16) (((samples[i+1] & 0xFF) << 8) | (samples[i] & 0xFF));
			qint16 right_sample = (qint16) (((samples[i+3] & 0xFF) << 8) | (samples[i+2] & 0xFF));
			left_noise_sample += left_sample;
			right_noise_sample += right_sample;
		}

		samples[i+3] = (quint8) (right_noise_sample >> 8);
		samples[i+2] = (quint8) right_noise_sample;
		samples[i+1] = (quint8) (left_noise_sample >> 8);
		samples[i] = (quint8) left_noise_sample;
	}
}