/*
  Copyright (c) 2021 Alex Diener
  
  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.
  
  Alex Diener alex@ludobloom.com
*/

#include "audiolab/Globals.h"
#include "audiolab/SynthControls_frequency_stairstep.h"
#include "shell/Shell.h"
#include "utilities/IOUtilities.h"
#include <stdio.h>

#define stemobject_implementation SynthControls_frequency_stairstep

v_begin();
v_func(dispose);
v_end();

SynthControls_frequency_stairstep * SynthControls_frequency_stairstep_create(Vector2f position, FrequencyCurve_stairstep * frequencyCurve, SynthControls_liveUpdateCallback liveUpdateCallback, SynthControls_updateCompleteCallback updateCompleteCallback, void * callbackContext, UIWindowRoot * windowRoot, UIAppearance appearance) {
	stemobject_create_implementation(init, position, frequencyCurve, liveUpdateCallback, updateCompleteCallback, callbackContext, windowRoot, appearance)
}

defineFrequencyControlSliderCallback(FrequencyCurve_stairstep, self->frequencyCurve, baseValue, "base frequency");
defineSynthControlSliderCallback_extended(FrequencyCurve_stairstep, self->frequencyCurve, steps[self->editingStepIndex].duration, stepDuration, "stairstep duration", 0.05f,               self->liveUpdateCallback, self->updateCompleteCallback, self->callbackContext);
defineSynthControlSliderCallback_extended(FrequencyCurve_stairstep, self->frequencyCurve, steps[self->editingStepIndex].change,   stepChange,   "stairstep value",    FREQUENCY_STEP_SIZE, self->liveUpdateCallback, self->updateCompleteCallback, self->callbackContext);

static void updateStepDisplay(SynthControls_frequency_stairstep * self) {
	char text[15];
	snprintf_safe(text, sizeof(text), "Step %u/%u", self->editingStepIndex + 1, self->frequencyCurve->stepCount);
	UILabel_setText(self->stepLabel, STR(text));
	UIFloatEditText_setValue(self->editText_stepDuration, self->frequencyCurve->steps[self->editingStepIndex].duration);
	UISlider_setValue(self->slider_stepDuration, self->frequencyCurve->steps[self->editingStepIndex].duration);
	UIFloatEditText_setValue(self->editText_stepChange, self->frequencyCurve->steps[self->editingStepIndex].change);
	UISlider_setValue(self->slider_stepChange, self->frequencyCurve->steps[self->editingStepIndex].change);
	self->removeStepButton->visible = self->frequencyCurve->stepCount > 1;
}

static void previousStepButtonCallback(UIButton * button, unsigned int modifiers, double referenceTime, void * context) {
	SynthControls_frequency_stairstep * self = context;
	self->editingStepIndex = (self->editingStepIndex + self->frequencyCurve->stepCount - 1) % self->frequencyCurve->stepCount;
	updateStepDisplay(self);
}

static void nextStepButtonCallback(UIButton * button, unsigned int modifiers, double referenceTime, void * context) {
	SynthControls_frequency_stairstep * self = context;
	self->editingStepIndex = (self->editingStepIndex + 1) % self->frequencyCurve->stepCount;
	updateStepDisplay(self);
}

static void updateSteppers(SynthControls_frequency_stairstep * self) {
	self->repeatCountStepper->upEnabled = self->frequencyCurve->repeatCount < STAIRSTEP_REPEAT_INFINITE - 1;
	self->repeatCountStepper->downEnabled = self->frequencyCurve->repeatCount > 0;
	self->repeatStartIndexStepper->upEnabled = self->frequencyCurve->repeatStartIndex < self->frequencyCurve->stepCount - 1;
	self->repeatStartIndexStepper->downEnabled = self->frequencyCurve->repeatStartIndex > 0;
	self->infiniteRepeatButton->visible = self->frequencyCurve->repeatCount != STAIRSTEP_REPEAT_INFINITE;
}

static void addStepButtonCallback(UIButton * button, unsigned int modifiers, double referenceTime, void * context) {
	SynthControls_frequency_stairstep * self = context;
	self->frequencyCurve->steps = realloc(self->frequencyCurve->steps, (self->frequencyCurve->stepCount + 1) * sizeof(*self->frequencyCurve->steps));
	self->editingStepIndex++;
	for (unsigned int stepIndex = self->frequencyCurve->stepCount; stepIndex > self->editingStepIndex; stepIndex--) {
		self->frequencyCurve->steps[stepIndex] = self->frequencyCurve->steps[stepIndex - 1];
	}
	self->frequencyCurve->steps[self->editingStepIndex].duration = 0.25;
	self->frequencyCurve->steps[self->editingStepIndex].change = 1.0f;
	self->frequencyCurve->stepCount++;
	self->liveUpdateCallback(self->callbackContext);
	self->updateCompleteCallback(self->callbackContext, "add freq step");
	updateStepDisplay(self);
	updateSteppers(self);
}

static void removeStepButtonCallback(UIButton * button, unsigned int modifiers, double referenceTime, void * context) {
	SynthControls_frequency_stairstep * self = context;
	if (self->frequencyCurve->stepCount > 1) {
		self->frequencyCurve->stepCount--;
		for (unsigned int stepIndex = self->editingStepIndex; stepIndex < self->frequencyCurve->stepCount; stepIndex++) {
			self->frequencyCurve->steps[stepIndex] = self->frequencyCurve->steps[stepIndex + 1];
		}
		if (self->editingStepIndex >= self->frequencyCurve->stepCount) {
			self->editingStepIndex--;
		}
		self->liveUpdateCallback(self->callbackContext);
		self->updateCompleteCallback(self->callbackContext, "remove freq step");
		updateStepDisplay(self);
		updateSteppers(self);
	}
}

static void reverseAllButtonCallback(UIButton * button, unsigned int modifiers, double referenceTime, void * context) {
	SynthControls_frequency_stairstep * self = context;
	unsigned int stepCount = self->frequencyCurve->stepCount;
	float totalChange = 0.0f;
	for (unsigned int stepIndex = 0; stepIndex < stepCount; stepIndex++) {
		totalChange += self->frequencyCurve->steps[stepIndex].change;
	}
	self->frequencyCurve->baseValue += totalChange;
	struct FrequencyCurve_stairstep_step * stepsCopy = memdup(self->frequencyCurve->steps, stepCount * sizeof(*stepsCopy));
	for (unsigned int stepIndex = 0; stepIndex < stepCount; stepIndex++) {
		self->frequencyCurve->steps[stepIndex] = stepsCopy[stepCount - stepIndex - 1];
		self->frequencyCurve->steps[stepIndex].change = -self->frequencyCurve->steps[stepIndex].change;
	}
	free(stepsCopy);
	self->liveUpdateCallback(self->callbackContext);
	self->updateCompleteCallback(self->callbackContext, "reverse stairstep");
	updateStepDisplay(self);
	updateSteppers(self);
}

static void infiniteRepeatButtonCallback(UIButton * button, unsigned int modifiers, double referenceTime, void * context) {
	SynthControls_frequency_stairstep * self = context;
	self->frequencyCurve->repeatCount = STAIRSTEP_REPEAT_INFINITE;
	UIEditText_setText(self->repeatCountField, STRL("Infinite"));
	updateSteppers(self);
	self->liveUpdateCallback(self->callbackContext);
	self->updateCompleteCallback(self->callbackContext, "change stairstep repeat count");
}

static bool repeatCountFieldChangeCallback(UIEditText * editText, double referenceTime, void * context) {
	SynthControls_frequency_stairstep * self = context;
	unsigned int value = self->frequencyCurve->repeatCount;
	bool changed = false;
	if (sscanf(UIEditText_getText(editText).bytes, "%u", &value) == 1) {
		if (value >= STAIRSTEP_REPEAT_INFINITE) {
			value = STAIRSTEP_REPEAT_INFINITE - 1;
		}
		if (self->frequencyCurve->repeatCount != value) {
			self->frequencyCurve->repeatCount = value;
			changed = true;
		}
	}
	if (self->frequencyCurve->repeatCount == STAIRSTEP_REPEAT_INFINITE) {
		UIEditText_setText(editText, STRL("Infinite"));
	} else {
		char text[12];
		snprintf_safe(text, sizeof(text), "%u", self->frequencyCurve->repeatCount);
		UIEditText_setText(editText, STR(text));
	}
	updateSteppers(self);
	UIContainer_setFocusedElement((UIContainer *) UIElement_getTopParent(self), NULL, NULL, UI_NONE);
	if (changed) {
		self->liveUpdateCallback(self->callbackContext);
		self->updateCompleteCallback(self->callbackContext, "change stairstep repeat count");
	} else {
		Shell_systemBeep();
	}
	return false;
}

static void repeatCountStepperCallback(UIStepper * stepper, int change, unsigned int modifiers, double referenceTime, void * context) {
	SynthControls_frequency_stairstep * self = context;
	if (self->frequencyCurve->repeatCount == STAIRSTEP_REPEAT_INFINITE) {
		self->frequencyCurve->repeatCount = 0;
	} else {
		self->frequencyCurve->repeatCount += change;
	}
	updateSteppers(self);
	char text[12];
	snprintf_safe(text, sizeof(text), "%u", self->frequencyCurve->repeatCount);
	UIEditText_setText(self->repeatCountField, STR(text));
	self->liveUpdateCallback(self->callbackContext);
	self->updateCompleteCallback(self->callbackContext, "change stairstep repeat count");
}

static bool repeatStartIndexFieldChangeCallback(UIEditText * editText, double referenceTime, void * context) {
	SynthControls_frequency_stairstep * self = context;
	unsigned int value = self->frequencyCurve->repeatStartIndex;
	bool changed = false;
	if (sscanf(UIEditText_getText(editText).bytes, "%u", &value) == 1 && self->frequencyCurve->repeatStartIndex != value) {
		self->frequencyCurve->repeatStartIndex = value;
		changed = true;
	}
	char text[12];
	snprintf_safe(text, sizeof(text), "%u", self->frequencyCurve->repeatStartIndex);
	UIEditText_setText(editText, STR(text));
	updateSteppers(self);
	UIContainer_setFocusedElement((UIContainer *) UIElement_getTopParent(self), NULL, NULL, UI_NONE);
	if (changed) {
		self->liveUpdateCallback(self->callbackContext);
		self->updateCompleteCallback(self->callbackContext, "change stairstep repeat start index");
	} else {
		Shell_systemBeep();
	}
	return false;
}

static void repeatStartIndexStepperCallback(UIStepper * stepper, int change, unsigned int modifiers, double referenceTime, void * context) {
	SynthControls_frequency_stairstep * self = context;
	self->frequencyCurve->repeatStartIndex += change;
	updateSteppers(self);
	char text[12];
	snprintf_safe(text, sizeof(text), "%u", self->frequencyCurve->repeatStartIndex);
	UIEditText_setText(self->repeatStartIndexField, STR(text));
	self->liveUpdateCallback(self->callbackContext);
	self->updateCompleteCallback(self->callbackContext, "change stairstep repeat start index");
}

bool SynthControls_frequency_stairstep_init(SynthControls_frequency_stairstep * self, Vector2f position, FrequencyCurve_stairstep * frequencyCurve, SynthControls_liveUpdateCallback liveUpdateCallback, SynthControls_updateCompleteCallback updateCompleteCallback, void * callbackContext, UIWindowRoot * windowRoot, UIAppearance appearance) {
	call_super(init, self, position, liveUpdateCallback, updateCompleteCallback, callbackContext, windowRoot, appearance);
	self->editingStepIndex = 0;
	self->frequencyCurve = frequencyCurve;
	float controlY = -CONTROL_OFFSET_Y / 2;
	addFrequencyControlSlider(self, CONTROL_DEFAULT_X, self->frequencyCurve->baseValue, "Base value", baseValue, 0.0f, 9.0f, 4.0f);
	
	controlY -= 6;
	call_virtual(addElement, self, self->stepLabel = UILabel_create(STR_NULL, VECTOR2f(30, controlY), VECTOR2f(0.0f, 0.0f), VECTOR2f(0.0f, 0.5f), ALIGN_LEFT, 0, OVERFLOW_RESIZE, OVERFLOW_RESIZE, g_uiAppearance), true);
	call_virtual(addElement, self, UIButton_create(STRL("Prev"), VECTOR2f(115, controlY), VECTOR2f(0.0f, 0.5f), 0.0f, OVERFLOW_RESIZE, previousStepButtonCallback, self, g_uiAppearance), true);
	call_virtual(addElement, self, UIButton_create(STRL("Next"), VECTOR2f(163, controlY), VECTOR2f(0.0f, 0.5f), 0.0f, OVERFLOW_RESIZE, nextStepButtonCallback, self, g_uiAppearance), true);
	call_virtual(addElement, self, UIButton_create(STRL("Add"), VECTOR2f(212, controlY), VECTOR2f(0.0f, 0.5f), 0.0f, OVERFLOW_RESIZE, addStepButtonCallback, self, g_uiAppearance), true);
	call_virtual(addElement, self, self->removeStepButton = UIButton_create(STRL("Remove"), VECTOR2f(254, controlY), VECTOR2f(0.0f, 0.5f), 0.0f, OVERFLOW_RESIZE, removeStepButtonCallback, self, g_uiAppearance), true);
	call_virtual(addElement, self, UIButton_create(STRL("Reverse All"), VECTOR2f(324, controlY), VECTOR2f(0.0f, 0.5f), 0.0f, OVERFLOW_RESIZE, reverseAllButtonCallback, self, g_uiAppearance), true);
	controlY -= CONTROL_OFFSET_Y + 6;
	
	addSynthControlSlider_extended(self, CONTROL_DEFAULT_X, self->frequencyCurve->steps[self->editingStepIndex].duration, "Duration", stepDuration, 0.0f, 1.0f, 0.25f);
	addSynthControlSlider_extended(self, CONTROL_DEFAULT_X, self->frequencyCurve->steps[self->editingStepIndex].change,   "Change",   stepChange,  -8.0f, 8.0f, 1.0f);
	
	char text[12];
	snprintf_safe(text, sizeof(text), "%u", self->frequencyCurve->repeatCount);
	call_virtual(addElement, self, UILabel_create(STRL("Repeat count"), VECTOR2f(CONTROL_DEFAULT_X - CONTROL_SLIDER_LABEL_PADDING, controlY), VECTOR2f(0.0f, 0.0f), VECTOR2f(1.0f, 0.5f), ALIGN_RIGHT, 0, OVERFLOW_RESIZE, OVERFLOW_RESIZE, g_uiAppearance), true);
	call_virtual(addElement, self, self->repeatCountField = UIEditText_create(self->frequencyCurve->repeatCount == STAIRSTEP_REPEAT_INFINITE ? STRL("Infinite") : STR(text), VECTOR2f(CONTROL_DEFAULT_X, controlY), VECTOR2f(60, 0.0f), VECTOR2f(0.0f, 0.5f), ALIGN_LEFT, 0, OVERFLOW_TRUNCATE, OVERFLOW_RESIZE, UIEditText_defaultNewlineActionCallback, NULL, repeatCountFieldChangeCallback, UIEditText_defaultCancelActionCallback, self, g_uiAppearance), true);
	call_virtual(addElement, self, self->repeatCountStepper = UIStepper_create(VECTOR2f(CONTROL_DEFAULT_X + 62, controlY), VECTOR2f(0.0f, 0.5f), STEPPER_VERTICAL, false, repeatCountStepperCallback, NULL, self, g_uiAppearance), true);
	call_virtual(addElement, self, self->infiniteRepeatButton = UIButton_create(STRL("Infinite"), VECTOR2f(CONTROL_DEFAULT_X + 82, controlY), VECTOR2f(0.0f, 0.5f), 0.0f, OVERFLOW_RESIZE, infiniteRepeatButtonCallback, self, g_uiAppearance), true);
	controlY -= CONTROL_OFFSET_Y;
	snprintf_safe(text, sizeof(text), "%u", self->frequencyCurve->repeatStartIndex);
	call_virtual(addElement, self, UILabel_create(STRL("Repeat start"), VECTOR2f(CONTROL_DEFAULT_X - CONTROL_SLIDER_LABEL_PADDING, controlY), VECTOR2f(0.0f, 0.0f), VECTOR2f(1.0f, 0.5f), ALIGN_RIGHT, 0, OVERFLOW_RESIZE, OVERFLOW_RESIZE, g_uiAppearance), true);
	call_virtual(addElement, self, self->repeatStartIndexField = UIEditText_create(STR(text), VECTOR2f(CONTROL_DEFAULT_X, controlY), VECTOR2f(60, 0.0f), VECTOR2f(0.0f, 0.5f), ALIGN_LEFT, 0, OVERFLOW_TRUNCATE, OVERFLOW_RESIZE, UIEditText_defaultNewlineActionCallback, NULL, repeatStartIndexFieldChangeCallback, UIEditText_defaultCancelActionCallback, self, g_uiAppearance), true);
	call_virtual(addElement, self, self->repeatStartIndexStepper = UIStepper_create(VECTOR2f(CONTROL_DEFAULT_X + 62, controlY), VECTOR2f(0.0f, 0.5f), STEPPER_VERTICAL, false, repeatStartIndexStepperCallback, NULL, self, g_uiAppearance), true);
	controlY -= CONTROL_OFFSET_Y;
	
	updateStepDisplay(self);
	updateSteppers(self);
	
	return true;
}

void SynthControls_frequency_stairstep_dispose(SynthControls_frequency_stairstep * self) {
	call_super_virtual(dispose, self);
}