/*

Tails

*/////////////////////////////////////////////////////////////////////////////



#include "pvc.hpp"
#include "dsp/digital.hpp"

namespace rack_plugin_PvC {

struct Tails : Module {
	enum ParamIds {
		PROB_UI,
		TOSS_MODE,
		TOSS_UI,
		FLIP_UI,
		SET_A_UI,
		SET_B_UI,

		NUM_PARAMS
	};
	enum InputIds {
		SIG_IN,
		PROB_CV,
		TOSS_IN,
		FLIP_IN,
		SET_A_IN,
		SET_B_IN,

		NUM_INPUTS
	};
	enum OutputIds {
		SIG_A_OUT,
		SIG_B_OUT,
		GATE_A_OUT,
		GATE_B_OUT,
		TRIG_A_OUT,
		TRIG_B_OUT,
		
		NUM_OUTPUTS
	};
	enum LightIds {
		A_LED,
		B_LED,

		DIR_LED,
		FLP_LED,

		NUM_LIGHTS
	};

	bool gate = false;
	bool flipMode = false;
	
	SchmittTrigger tossTrigger;
	SchmittTrigger flipTrigger;
	SchmittTrigger resetTrigger;
	PulseGenerator gatePulse;
	
	Tails() : Module(NUM_PARAMS, NUM_INPUTS, NUM_OUTPUTS, NUM_LIGHTS) {}

	void step() override;

	void reset() override {
		gate = false;
		flipMode = false;
	}

	json_t *toJson() override {
		json_t *rootJ = json_object();
		
		json_t *gateStateJ = json_boolean(gate);
		json_object_set_new(rootJ, "gateState", gateStateJ);

		return rootJ;
	}
	
	void fromJson(json_t *rootJ) override {
		json_t *gateStateJ = json_object_get(rootJ, "gateState");
		if (gateStateJ)
			gate = json_boolean_value(gateStateJ);
	}
};

void Tails::step() {
	flipMode = params[TOSS_MODE].value;

	if (tossTrigger.process(inputs[TOSS_IN].value + params[TOSS_UI].value)) {
		gatePulse.trigger(0.01f);
		if (flipMode)
			gate = (randomUniform() < params[PROB_UI].value + inputs[PROB_CV].value) ? !gate : gate; //
		else
			gate = (randomUniform() < (params[PROB_UI].value + inputs[PROB_CV].value*0.1f));
	}

	if (flipTrigger.process(inputs[FLIP_IN].value + params[FLIP_UI].value)) {
		gatePulse.trigger(0.01f);
		gate = !gate;
	}

	if (resetTrigger.process(inputs[SET_A_IN].value + params[SET_A_UI].value)) {
		gatePulse.trigger(0.01f);
		gate = false;
	}

	if (resetTrigger.process(inputs[SET_B_IN].value + params[SET_B_UI].value)) {
		gatePulse.trigger(0.01f);
		gate = true;
	}

	outputs[SIG_A_OUT].value = !gate * inputs[SIG_IN].value;
	outputs[GATE_A_OUT].value = !gate * 10.0f;
	outputs[TRIG_A_OUT].value = !gate * gatePulse.process(1.0/engineGetSampleRate()) * 10.0f;
	outputs[SIG_B_OUT].value = gate * inputs[SIG_IN].value;
	outputs[GATE_B_OUT].value = gate * 10.0f;
	outputs[TRIG_B_OUT].value = gate * gatePulse.process(1.0/engineGetSampleRate()) * 10.0f;
	lights[A_LED].value = !gate;
	lights[B_LED].value = gate;
	lights[DIR_LED].value = !flipMode;
	lights[FLP_LED].value = flipMode;
}

struct ModeToggle : SVGSwitch, ToggleSwitch {
	ModeToggle() {
		addFrame(SVG::load(assetPlugin(plugin, "res/components/empty.svg")));
		box.size = Vec(12,6);
	}
};

struct TailsWidget : ModuleWidget {
	TailsWidget(Tails *module);
};

TailsWidget::TailsWidget(Tails *module) : ModuleWidget(module) {
  setPanel(SVG::load(assetPlugin(plugin, "res/panels/Tails.svg")));
  
	// screws
	// addChild(Widget::create<ScrewHead1>(Vec(0, 0)));
	addChild(Widget::create<ScrewHead2>(Vec(box.size.x - 30, 0)));
	addChild(Widget::create<ScrewHead3>(Vec(15, 365)));
	addChild(Widget::create<ScrewHead4>(Vec(box.size.x - 15, 365)));

	addInput(Port::create<InPortAud>(Vec(19,22), Port::INPUT, module, Tails::SIG_IN));

	addParam(ParamWidget::create<PvCKnob>(Vec(19,64), module, Tails::PROB_UI, 0.0f, 1.0f, 0.5f));
	addInput(Port::create<InPortCtrl>(Vec(19,88), Port::INPUT, module, Tails::PROB_CV));
	addInput(Port::create<InPortBin>(Vec(19,124), Port::INPUT, module, Tails::TOSS_IN));
	addParam(ParamWidget::create<LabelButtonL>(Vec(12,149), module, Tails::TOSS_UI, 0, 1, 0));
	addChild(ModuleLightWidget::create<FourPixLight<OrangeLED>>(Vec(25,165),module, Tails::DIR_LED));
	addChild(ModuleLightWidget::create<FourPixLight<BlueLED>>(Vec(31,165),module, Tails::FLP_LED));
	addParam(ParamWidget::create<ModeToggle>(Vec(24,164), module, Tails::TOSS_MODE, 0, 1, 0));
	addInput(Port::create<InPortBin>(Vec(19,180), Port::INPUT, module, Tails::FLIP_IN));
	addParam(ParamWidget::create<LabelButtonL>(Vec(12,205), module, Tails::FLIP_UI, 0, 1, 0));
	
	addInput(Port::create<InPortBin>(Vec(4,222), Port::INPUT, module, Tails::SET_A_IN));
	addParam(ParamWidget::create<LabelButtonS>(Vec(3,247), module, Tails::SET_A_UI, 0, 1, 0));
	addChild(ModuleLightWidget::create<FourPixLight<CyanLED>>(Vec(13,267),module, Tails::A_LED));
	addOutput(Port::create<OutPortVal>(Vec(4,276), Port::OUTPUT, module, Tails::SIG_A_OUT));
	addOutput(Port::create<OutPortBin>(Vec(4,312), Port::OUTPUT, module, Tails::GATE_A_OUT));
	addOutput(Port::create<OutPortBin>(Vec(4,336), Port::OUTPUT, module, Tails::TRIG_A_OUT));

	addInput(Port::create<InPortBin>(Vec(34,222), Port::INPUT, module, Tails::SET_B_IN));
	addParam(ParamWidget::create<LabelButtonS>(Vec(33,247), module, Tails::SET_B_UI, 0, 1, 0));
	addChild(ModuleLightWidget::create<FourPixLight<PurpleLED>>(Vec(43,267),module, Tails::B_LED));
	addOutput(Port::create<OutPortVal>(Vec(34,276), Port::OUTPUT, module, Tails::SIG_B_OUT));
	addOutput(Port::create<OutPortBin>(Vec(34,312), Port::OUTPUT, module, Tails::GATE_B_OUT));
	addOutput(Port::create<OutPortBin>(Vec(34,336), Port::OUTPUT, module, Tails::TRIG_B_OUT));
}

} // namespace rack_plugin_PvC

using namespace rack_plugin_PvC;

RACK_PLUGIN_MODEL_INIT(PvC, Tails) {
   Model *modelTails = Model::create<Tails, TailsWidget>(
      "PvC", "Tails", "Tails", LOGIC_TAG, SWITCH_TAG, RANDOM_TAG);
   return modelTails;
}