#include "dsp/digital.hpp"

#include "alikins.hpp"
#include "MsDisplayWidget.hpp"

namespace rack_plugin_Alikins {

struct GateLength : Module {
    enum ParamIds {
        GATE_LENGTH_PARAM1,
        GATE_LENGTH_PARAM2,
        GATE_LENGTH_PARAM3,
        GATE_LENGTH_PARAM4,
        GATE_LENGTH_PARAM5,
        NUM_PARAMS
    };
    enum InputIds {
        TRIGGER_INPUT1,
        TRIGGER_INPUT2,
        TRIGGER_INPUT3,
        TRIGGER_INPUT4,
        TRIGGER_INPUT5,
        GATE_LENGTH_INPUT1,
        GATE_LENGTH_INPUT2,
        GATE_LENGTH_INPUT3,
        GATE_LENGTH_INPUT4,
        GATE_LENGTH_INPUT5,
        NUM_INPUTS
    };
    enum OutputIds {
        GATE_OUTPUT1,
        GATE_OUTPUT2,
        GATE_OUTPUT3,
        GATE_OUTPUT4,
        GATE_OUTPUT5,
        NUM_OUTPUTS
    };
    enum LightIds {
        NUM_LIGHTS
    };

    float gate_length[GATE_LENGTH_INPUTS];

    SchmittTrigger inputOnTrigger[GATE_LENGTH_INPUTS];

    PulseGenerator gateGenerator[GATE_LENGTH_INPUTS];

    GateLength() : Module(NUM_PARAMS, NUM_INPUTS, NUM_OUTPUTS, NUM_LIGHTS) {}
    
    void step() override;

    void onReset() override {
    }

};

void GateLength::step() {
    // FIXME: add way to support >10.0s gate length

    float sample_time = engineGetSampleTime();
    
    for (int i = 0; i < GATE_LENGTH_INPUTS; i++) {
        gate_length[i] = clamp(params[GATE_LENGTH_PARAM1 + i].value + inputs[GATE_LENGTH_INPUT1 + i].value, 0.0f, 10.0f);
    
        if (inputOnTrigger[i].process(inputs[TRIGGER_INPUT1 + i].value)) {
            // debug("GL INPUT ON TRIGGER %d gate_length: %f", i, gate_length[i]);
            gateGenerator[i].trigger(gate_length[i]);
        }

        outputs[GATE_OUTPUT1 + i].value = gateGenerator[i].process(sample_time) ? 10.0f : 0.0f;
    }
}

struct GateLengthWidget : ModuleWidget {
    GateLengthWidget(GateLength *module);
};

GateLengthWidget::GateLengthWidget(GateLength *module) : ModuleWidget(module) {

    box.size = Vec(4 * RACK_GRID_WIDTH, RACK_GRID_HEIGHT);
    setPanel(SVG::load(assetPlugin(plugin, "res/GateLength.svg")));

    float y_pos = 2.0f;
    
    for (int i = 0; i < GATE_LENGTH_INPUTS; i++) {
        float x_pos = 4.0f;
        y_pos += 39.0f;

        addInput(Port::create<PJ301MPort>(Vec(x_pos, y_pos),
                                          Port::INPUT,
                                          module,
                                          GateLength::TRIGGER_INPUT1 + i));

        x_pos += 30.0f;

        MsDisplayWidget *gate_length_display = new MsDisplayWidget();
        gate_length_display->box.pos = Vec(x_pos, y_pos + 1.0f);
        gate_length_display->box.size = Vec(84, 24);
        gate_length_display->value = &module->gate_length[i];
        addChild(gate_length_display);

        // FIXME: use new sequential box hbox/vbox thing
        x_pos += 84.0f;
        x_pos += 4.0f;
        addOutput(Port::create<PJ301MPort>(Vec(x_pos, y_pos),
                                           Port::OUTPUT,
                                           module,
                                           GateLength::GATE_OUTPUT1 + i));

        x_pos = 4.0f;
        y_pos += 26.0f;

        addInput(Port::create<PJ301MPort>(Vec(x_pos, y_pos),
                                          Port::INPUT,
                                          module,
                                          GateLength::GATE_LENGTH_INPUT1 + i));

        x_pos += 30.0f;
        addParam(ParamWidget::create<Trimpot>(Vec(x_pos, y_pos + 3.0f),
                                              module,
                                              GateLength::GATE_LENGTH_PARAM1 + i,
                                              0.0f, 10.0f, 0.1f));
    }

    addChild(Widget::create<ScrewSilver>(Vec(0.0f, 0.0f)));
    addChild(Widget::create<ScrewSilver>(Vec(box.size.x - 15.0f, 0.0f)));
    addChild(Widget::create<ScrewSilver>(Vec(0.0f, 365.0f)));
    addChild(Widget::create<ScrewSilver>(Vec(box.size.x - 15.0f, 365.0f)));

}

} // namespace rack_plugin_Alikins

using namespace rack_plugin_Alikins;

RACK_PLUGIN_MODEL_INIT(Alikins, GateLength) {
   Model *modelGateLength = Model::create<GateLength, GateLengthWidget>(
      "Alikins", "GateLength", "Gate Length", UTILITY_TAG);
   return modelGateLength;
}