//
// Created by Alex Birch on 01/10/2017.
//

#include "Pills.h"
#include "MyColours.h"
#include <algorithm>

using namespace std;

Pill::Pill(
    AudioProcessorValueTreeState& valueTreeState,
    int bank,
    bool isFirst,
    bool isLast
)
// : pills{pills}
: valueTreeState{valueTreeState}
, bank{bank}
, textButton{String(bank)}
{
    setOpaque(true);
    textButton.setConnectedEdges (
            (isFirst ? 0 : Button::ConnectedOnLeft)
            | (isLast ? 0 : Button::ConnectedOnRight)
    );
    textButton.setRadioGroupId(34567);
    loadToggleState();
    textButton.setClickingTogglesState(true);

    addAndMakeVisible(textButton);
    
    valueTreeState.addParameterListener("bank", this);
//    valueTreeState.state.addListener(this);
    textButton.addListener(this);
}

void Pill::paint (Graphics& g)
{
    // (Our component is opaque, so we must completely fill the background with a solid colour)
    g.fillAll (getLookAndFeel().findColour (ResizableWindow::backgroundColourId));
}

void Pill::resized() {
    textButton.setBounds(getLocalBounds());
}

Pill::~Pill() {
    valueTreeState.removeParameterListener("bank", this);
//    valueTreeState.state.removeListener(this);
    textButton.removeListener(this);
}

void Pill::loadToggleState() {
    RangedAudioParameter *param {valueTreeState.getParameter("bank")};
    jassert(dynamic_cast<AudioParameterInt*> (param) != nullptr);
    AudioParameterInt* castParam {dynamic_cast<AudioParameterInt*> (param)};
    int value{castParam->get()};
    textButton.setToggleState(value == bank, dontSendNotification);
}

void Pill::buttonClicked (Button* button) {
    // selected = button;
    // onItemSelected(itemToIDMapper(button->getName().toStdString()));
    RangedAudioParameter *param {valueTreeState.getParameter("bank")};
    jassert(dynamic_cast<AudioParameterInt*> (param) != nullptr);
    AudioParameterInt* castParam {dynamic_cast<AudioParameterInt*> (param)};
    *castParam = bank;
}

void Pill::parameterChanged(const String& parameterID, float newValue) {
    if (parameterID == "bank") {
        loadToggleState();
    }
}

// void Pill::valueTreePropertyChanged(
//     ValueTree& treeWhosePropertyHasChanged,
//     const Identifier& property) {
//     if (treeWhosePropertyHasChanged.getType() == StringRef("presets")) {
//         loadModelFrom(treeWhosePropertyHasChanged);
//     }
// }

Pills::Pills(
    AudioProcessorValueTreeState& valueTreeState
    // string label,
    // const vector<string> &items,
    // const function<void (int)> &onItemSelected,
    // const function<int (const string&)> &itemToIDMapper,
    // int initiallySelectedItem
)
: valueTreeState{valueTreeState}
// , label{label}
    // items(items),
    // onItemSelected(onItemSelected),
    // itemToIDMapper(itemToIDMapper)
{
    // faster (rounded edges introduce transparency)
    setOpaque (true);

    // populate(initiallySelectedItem);
    ValueTree banks{valueTreeState.state.getChildWithName("banks")};
    loadModelFrom(banks);

    valueTreeState.state.addListener(this);
}

Pills::~Pills() {
    valueTreeState.state.removeListener(this);
}

void Pills::valueTreePropertyChanged(
    ValueTree& treeWhosePropertyHasChanged,
    const Identifier& property) {
    if (treeWhosePropertyHasChanged.getType() == StringRef("banks")) {
        if (property == StringRef("synthetic")) {
            loadModelFrom(treeWhosePropertyHasChanged);
        }
    }
}

void Pills::loadModelFrom(ValueTree& banks) {
    pills.clear();
    int numChildren{banks.getNumChildren()};
    for(int i{0}; i < numChildren; i++) {
        ValueTree child{banks.getChild(i)};
        int num{child.getProperty("num")};
        // rows.push_back(unique_ptr<Pill>(new Pill(), [](Pill* pill) {
        //     pill->remo
        // }));
        unique_ptr<Pill> pill{make_unique<Pill>(
            valueTreeState,
            num,
            i == 0,
            i == numChildren - 1)};
        addAndMakeVisible(pill.get());
        pills.push_back(move(pill));
    }
    resized();
}

// void Pills::populate(int initiallySelectedItem) {
//     int index = 0;
//     for (string item : items) {
//         TextButton* pill = addToList(new TextButton(
//                 item
//         ));
// //        pill->setColour (TextButton::buttonOnColourId, Colours::blueviolet.brighter());
// //        pill->setBounds(20 + index * 55, 260, 55, 24);
//         pill->setConnectedEdges (
//                 (index == 0 ? 0 : Button::ConnectedOnLeft)
//                         | (index == (items.size()-1) ? 0 : Button::ConnectedOnRight)
//         );
//         pill->setRadioGroupId(34567);
//         if (index == initiallySelectedItem) {
//             pill->setToggleState(true, dontSendNotification);
//             selected = pill;
//         }
//         pill->setClickingTogglesState(true);
//         pill->addListener(this);
//         index++;
//     }
// }

// void Pills::setItems(
//         const vector<string> &items,
//         int initiallySelectedItem
// ) {
//     this->items = items;
//     for(TextButton* t : buttons) {
//         t->removeListener(this);
//     }
//     buttons.clear(true);
//     populate(initiallySelectedItem);
//     resized();
// }

// TextButton* Pills::addToList (TextButton* newButton) {
//     buttons.add (newButton);
//     addAndMakeVisible (newButton);
//     return newButton;
// }

void Pills::cycle(bool right) {
    RangedAudioParameter *param {valueTreeState.getParameter("bank")};
    jassert(dynamic_cast<AudioParameterInt*> (param) != nullptr);
    AudioParameterInt* castParam {dynamic_cast<AudioParameterInt*> (param)};
    int currentlySelectedBank{castParam->get()};

    ValueTree banks{valueTreeState.state.getChildWithName("banks")};
//    int numChildren{banks.getNumChildren()};

    vector<int> bankInts;
    bankInts.resize(banks.getNumChildren());

    transform(banks.begin(), banks.end(), bankInts.begin(), [](ValueTree bank) -> int {
        return bank.getProperty("num");
    });
    
//    int closestBank{currentlySelectedBank};
//    for(int i{0}; i < numChildren; i++) {
//        ValueTree child{banks.getChild(i)};
//        int proposedBank{child.getProperty("num")};
//        if (right && proposedBank > currentlySelectedBank) {
//            closestBank = jmin(closestBank, proposedBank);
//        } else if (left )
//    }

    int currentIx{static_cast<const int>(distance(bankInts.begin(), find(bankInts.begin(), bankInts.end(), currentlySelectedBank)))};
    currentIx += right ? 1 : pills.size()-1;
    // pills[currentIx % pills.size()]->textButton.triggerClick();
    *castParam = bankInts[currentIx % bankInts.size()];


    // TODO: base this on valueTree
    // int currentIx = static_cast<const int>(distance(pills.begin(), find(pills.begin(), pills.end(), selected)));
    // currentIx += right ? 1 : pills.size()-1;
    // pills[currentIx % pills.size()]->textButton.triggerClick();
}

void Pills::resized() {
    int index = 0;
    Rectangle<int> r (getLocalBounds());
    const int equalWidth = r.proportionOfWidth(pills.size() <= 0 ? 1.0 : 1.0f/pills.size());
    for(auto& pill : pills) {
        Rectangle<int> r2 (getLocalBounds());
        r2.removeFromLeft(equalWidth * index);
        r2.removeFromRight(equalWidth * (static_cast<int>(pills.size())-index-1));
        pill->setBounds(r2);
        index++;
    }
}

/**
 * This is required to support setOpaque(true)
 */
void Pills::paint(Graphics& g)
{
    g.fillAll(MyColours::getUIColourIfAvailable(LookAndFeel_V4::ColourScheme::UIColour::windowBackground, Colours::lightgrey));
}