PilotCore.cpp

#include "PilotCore.h"
#include "PilotCalibrator.h"
#include "BaseOutputGenerator.h"
#include "SerialOutputGenerator.h"
#include <AceButton.h>
#include "StateManager.h"
#include "Globals.h"
#include <Arduino.h>

#include "SimplePitchWheelGenerator.h"
#include "SimpleCCGenerator.h"
#include "GenericOutputGenerator.h"
#include "AnalogControlledCentreGenerator.h"

using namespace ace_button;

PilotCore::PilotCore(StateManager* stateManager) : BaseState(stateManager) {

    for(int i = 0; i < MIDI_GENERATOR_SLOT_COUNT; i++) {
        outputGenerators[i] = nullptr;
    }

    // set up digital pins as pull up inputs
    pinMode(Pins::DIGITAL_PIN_BUTTON_L, INPUT_PULLUP);
    pinMode(Pins::DIGITAL_PIN_BUTTON_R, INPUT_PULLUP);

    // set up the analog pins
    analogJoyRX.begin(Pins::ANALOG_PIN_JOY_R_X, RESPONSIVE_SLEEP_ENABLED, RESPONSIVE_SNAP_MULTIPLIER);
    analogJoyRY.begin(Pins::ANALOG_PIN_JOY_R_Y, RESPONSIVE_SLEEP_ENABLED, RESPONSIVE_SNAP_MULTIPLIER);
    analogJoyLX.begin(Pins::ANALOG_PIN_JOY_L_X, RESPONSIVE_SLEEP_ENABLED, RESPONSIVE_SNAP_MULTIPLIER);
    analogJoyLY.begin(Pins::ANALOG_PIN_JOY_L_Y, RESPONSIVE_SLEEP_ENABLED, RESPONSIVE_SNAP_MULTIPLIER);
    analogPot1.begin(Pins::ANALOG_PIN_POT_1, RESPONSIVE_SLEEP_ENABLED, RESPONSIVE_SNAP_MULTIPLIER);
    analogPot2.begin(Pins::ANALOG_PIN_POT_2, RESPONSIVE_SLEEP_ENABLED, RESPONSIVE_SNAP_MULTIPLIER);
    analogPot3.begin(Pins::ANALOG_PIN_POT_3, RESPONSIVE_SLEEP_ENABLED, RESPONSIVE_SNAP_MULTIPLIER);
    analogPot4.begin(Pins::ANALOG_PIN_POT_4, RESPONSIVE_SLEEP_ENABLED, RESPONSIVE_SNAP_MULTIPLIER);

    analogJoyRX.enableEdgeSnap();
    analogJoyRY.enableEdgeSnap();
    analogJoyLX.enableEdgeSnap();
    analogJoyLY.enableEdgeSnap();
    analogPot1.enableEdgeSnap();
    analogPot2.enableEdgeSnap();
    analogPot3.enableEdgeSnap();
    analogPot4.enableEdgeSnap();

    // load the first preset
    switchOutputGenerators(0);
}

PilotCore::~PilotCore() {

}

void PilotCore::loop() {

    analogJoyRX.update();
    analogJoyRY.update();
    analogJoyLX.update();
    analogJoyLY.update();
    analogPot1.update();
    analogPot2.update();
    analogPot3.update();
    analogPot4.update();

    analogValues[AnalogInputs::JOY_R_X] = VOLTAGE_MAX - analogMinMaxTracker->mapValueRespectingCentre(Pins::ANALOG_PIN_JOY_R_X, analogJoyRX.getValue(), 15);
    analogValues[AnalogInputs::JOY_R_Y] = analogMinMaxTracker->mapValue(Pins::ANALOG_PIN_JOY_R_Y, analogJoyRY.getValue());
    analogValues[AnalogInputs::JOY_L_X] = analogMinMaxTracker->mapValueRespectingCentre(Pins::ANALOG_PIN_JOY_L_X, analogJoyLX.getValue(), 15);
    analogValues[AnalogInputs::JOY_L_Y] = VOLTAGE_MAX - analogMinMaxTracker->mapValue(Pins::ANALOG_PIN_JOY_L_Y, analogJoyLY.getValue());

    analogValues[AnalogInputs::POT_1] = analogMinMaxTracker->mapValue(Pins::ANALOG_PIN_POT_1, analogPot1.getValue());
    analogValues[AnalogInputs::POT_2] = analogMinMaxTracker->mapValue(Pins::ANALOG_PIN_POT_2, analogPot2.getValue());
    analogValues[AnalogInputs::POT_3] = analogMinMaxTracker->mapValue(Pins::ANALOG_PIN_POT_3, analogPot3.getValue());
    analogValues[AnalogInputs::POT_4] = analogMinMaxTracker->mapValue(Pins::ANALOG_PIN_POT_4, analogPot4.getValue());

    analogChanged[AnalogInputs::JOY_R_X] = analogJoyRX.hasChanged();
    analogChanged[AnalogInputs::JOY_R_Y] = analogJoyRY.hasChanged();
    analogChanged[AnalogInputs::JOY_L_X] = analogJoyLX.hasChanged();
    analogChanged[AnalogInputs::JOY_L_Y] = analogJoyLY.hasChanged();
    analogChanged[AnalogInputs::POT_1] = analogPot1.hasChanged();
    analogChanged[AnalogInputs::POT_2] = analogPot2.hasChanged();
    analogChanged[AnalogInputs::POT_3] = analogPot3.hasChanged();
    analogChanged[AnalogInputs::POT_4] = analogPot4.hasChanged();

    digitalValues[DigitalInputs::BUTTON_L] = digitalRead(Pins::DIGITAL_PIN_BUTTON_L);
    digitalValues[DigitalInputs::BUTTON_R] = digitalRead(Pins::DIGITAL_PIN_BUTTON_R);

    // loop through the midi generators
    for (int i = 0; i < MIDI_GENERATOR_SLOT_COUNT; i++) {
        if(outputGenerators[i] != nullptr) {
            outputGenerators[i]->loop();
        }
    }
}

unsigned short PilotCore::getAnalog(AnalogInputs pin) {
    return analogValues[pin];
}

bool PilotCore::getDigital(DigitalInputs pin) {
    return digitalValues[pin];
}

void PilotCore::switchOutputGenerators(int presetId) {
    // clear any current ones
    for (int i = 0; i < MIDI_GENERATOR_SLOT_COUNT; i++) {
        if(outputGenerators[i] != nullptr) {
            delete outputGenerators[i];
            outputGenerators[i] = nullptr;
        }
    }

    // add the new ones
    switch(presetId){
        case 0:
            // TODO: add NRPM option instead of CC
            outputGenerators[0] = (new GenericOutputGenerator(AnalogInputs::JOY_R_X, this))->setOutput(OutputType::OUTPUT_TYPE_PITCH_BEND, 0);
            outputGenerators[1] = (new GenericOutputGenerator(AnalogInputs::JOY_R_Y, this))->setOutput(OutputType::OUTPUT_TYPE_CC, 1);
            outputGenerators[2] = (new GenericOutputGenerator(AnalogInputs::JOY_L_X, this))->setOutput(OutputType::OUTPUT_TYPE_CC, 71);
            outputGenerators[3] = (new GenericOutputGenerator(AnalogInputs::JOY_L_Y, this))->setOutput(OutputType::OUTPUT_TYPE_CC, 74);
            outputGenerators[4] = (new GenericOutputGenerator(AnalogInputs::POT_1, this))->setOutput(OutputType::OUTPUT_TYPE_CC, 73);
            outputGenerators[5] = (new GenericOutputGenerator(AnalogInputs::POT_2, this))->setOutput(OutputType::OUTPUT_TYPE_CC, 80);
            outputGenerators[6] = (new GenericOutputGenerator(AnalogInputs::POT_3, this))->setOutput(OutputType::OUTPUT_TYPE_CC, 11);
            outputGenerators[7] = (new GenericOutputGenerator(AnalogInputs::POT_4, this))->setOutput(OutputType::OUTPUT_TYPE_CC, 72);
            break;
        case 1:
            outputGenerators[0] = (new GenericOutputGenerator(AnalogInputs::JOY_R_X, this))->setOutput(OutputType::OUTPUT_TYPE_SERIAL, 0);
            outputGenerators[1] = (new GenericOutputGenerator(AnalogInputs::JOY_R_Y, this))->setOutput(OutputType::OUTPUT_TYPE_SERIAL, 1);
            outputGenerators[2] = (new GenericOutputGenerator(AnalogInputs::JOY_L_X, this))->setOutput(OutputType::OUTPUT_TYPE_SERIAL, 2);
            outputGenerators[3] = (new GenericOutputGenerator(AnalogInputs::JOY_L_Y, this))->setOutput(OutputType::OUTPUT_TYPE_SERIAL, 3);
            outputGenerators[4] = (new GenericOutputGenerator(AnalogInputs::POT_1, this))->setOutput(OutputType::OUTPUT_TYPE_SERIAL, 4);
            outputGenerators[5] = (new GenericOutputGenerator(AnalogInputs::POT_2, this))->setOutput(OutputType::OUTPUT_TYPE_SERIAL, 5);
            outputGenerators[6] = (new GenericOutputGenerator(AnalogInputs::POT_3, this))->setOutput(OutputType::OUTPUT_TYPE_SERIAL, 6);
            outputGenerators[7] = (new GenericOutputGenerator(AnalogInputs::POT_4, this))->setOutput(OutputType::OUTPUT_TYPE_SERIAL, 7);

            break;

    }

}

void PilotCore::enter() {
    Serial.println("Entering MIDI Mode.");
}

void PilotCore::exit() {
    Serial.println("Exiting MIDI Mode.");
}

void PilotCore::handleEvent(DigitalInputs button, uint8_t eventType) {
    if(button == DigitalInputs::BUTTON_R && eventType == AceButton::AceButton::kEventDoubleClicked) {
        stateManager->changeState(new PilotCalibrator(stateManager));
    }


    if(eventType == AceButton::AceButton::kEventClicked) {
        if(button == DigitalInputs::BUTTON_R) {
            currentPreset = (currentPreset + 1) % 2;
        } else if(button == DigitalInputs::BUTTON_L) {
            currentPreset--;
            if(currentPreset < 0) {
                currentPreset = 1;
            }
        }
        switchOutputGenerators(currentPreset);

    }

}

bool PilotCore::getAnalogChanged(AnalogInputs pin) {
    return analogChanged[pin];
}