/*----------------------------------------------------------------------------*/
/* Copyright (c) 2019-2020 FIRST. All Rights Reserved.                        */
/* Open Source Software - may be modified and shared by FRC teams. The code   */
/* must be accompanied by the FIRST BSD license file in the root directory of */
/* the project.                                                               */
/*----------------------------------------------------------------------------*/

#include "glass/other/FMS.h"

#include <imgui.h>
#include <wpi/SmallString.h>

#include "glass/DataSource.h"

using namespace glass;

static const char* stations[] = {"Red 1",  "Red 2",  "Red 3",
                                 "Blue 1", "Blue 2", "Blue 3"};

void glass::DisplayFMS(FMSModel* model, bool* matchTimeEnabled) {
  if (!model->Exists() || model->IsReadOnly()) return DisplayFMSReadOnly(model);

  // FMS Attached
  if (auto data = model->GetFmsAttachedData()) {
    bool val = data->GetValue();
    if (ImGui::Checkbox("FMS Attached", &val)) model->SetFmsAttached(val);
    data->EmitDrag();
  }

  // DS Attached
  if (auto data = model->GetDsAttachedData()) {
    bool val = data->GetValue();
    if (ImGui::Checkbox("DS Attached", &val)) model->SetDsAttached(val);
    data->EmitDrag();
  }

  // Alliance Station
  if (auto data = model->GetAllianceStationIdData()) {
    int val = data->GetValue();
    ImGui::SetNextItemWidth(ImGui::GetFontSize() * 8);
    if (ImGui::Combo("Alliance Station", &val, stations, 6))
      model->SetAllianceStationId(val);
    data->EmitDrag();
  }

  // Match Time
  if (auto data = model->GetMatchTimeData()) {
    if (matchTimeEnabled)
      ImGui::Checkbox("Match Time Enabled", matchTimeEnabled);

    double val = data->GetValue();
    ImGui::SetNextItemWidth(ImGui::GetFontSize() * 8);
    if (ImGui::InputDouble("Match Time", &val, 0, 0, "%.1f",
                           ImGuiInputTextFlags_EnterReturnsTrue)) {
      model->SetMatchTime(val);
    }
    data->EmitDrag();
    ImGui::SameLine();
    if (ImGui::Button("Reset")) {
      model->SetMatchTime(0.0);
    }
  }

  // Game Specific Message
  // make buffer full 64 width, null terminated, for editability
  wpi::SmallString<64> gameSpecificMessage;
  model->GetGameSpecificMessage(gameSpecificMessage);
  gameSpecificMessage.resize(63);
  gameSpecificMessage.push_back('\0');
  ImGui::SetNextItemWidth(ImGui::GetFontSize() * 8);
  if (ImGui::InputText("Game Specific", gameSpecificMessage.data(),
                       gameSpecificMessage.size(),
                       ImGuiInputTextFlags_EnterReturnsTrue)) {
    model->SetGameSpecificMessage(gameSpecificMessage.data());
  }
}

void glass::DisplayFMSReadOnly(FMSModel* model) {
  bool exists = model->Exists();
  if (!exists) ImGui::PushStyleColor(ImGuiCol_Text, IM_COL32(96, 96, 96, 255));

  if (auto data = model->GetEStopData()) {
    ImGui::Selectable("E-Stopped: ");
    data->EmitDrag();
    ImGui::SameLine();
    ImGui::TextUnformatted(exists ? (data->GetValue() ? "Yes" : "No") : "?");
  }
  if (auto data = model->GetEnabledData()) {
    ImGui::Selectable("Robot Enabled: ");
    data->EmitDrag();
    ImGui::SameLine();
    ImGui::TextUnformatted(exists ? (data->GetValue() ? "Yes" : "No") : "?");
  }
  if (auto data = model->GetTestData()) {
    ImGui::Selectable("Test Mode: ");
    data->EmitDrag();
    ImGui::SameLine();
    ImGui::TextUnformatted(exists ? (data->GetValue() ? "Yes" : "No") : "?");
  }
  if (auto data = model->GetAutonomousData()) {
    ImGui::Selectable("Autonomous Mode: ");
    data->EmitDrag();
    ImGui::SameLine();
    ImGui::TextUnformatted(exists ? (data->GetValue() ? "Yes" : "No") : "?");
  }
  if (auto data = model->GetFmsAttachedData()) {
    ImGui::Selectable("FMS Attached: ");
    data->EmitDrag();
    ImGui::SameLine();
    ImGui::TextUnformatted(exists ? (data->GetValue() ? "Yes" : "No") : "?");
  }
  if (auto data = model->GetDsAttachedData()) {
    ImGui::Selectable("DS Attached: ");
    data->EmitDrag();
    ImGui::SameLine();
    ImGui::TextUnformatted(exists ? (data->GetValue() ? "Yes" : "No") : "?");
  }
  if (auto data = model->GetAllianceStationIdData()) {
    ImGui::Selectable("Alliance Station: ");
    data->EmitDrag();
    ImGui::SameLine();
    ImGui::TextUnformatted(exists ? stations[static_cast<int>(data->GetValue())]
                                  : "?");
  }
  if (auto data = model->GetMatchTimeData()) {
    ImGui::Selectable("Match Time: ");
    data->EmitDrag();
    ImGui::SameLine();
    if (exists)
      ImGui::Text("%.1f", data->GetValue());
    else
      ImGui::TextUnformatted("?");
  }

  wpi::SmallString<64> gameSpecificMessage;
  model->GetGameSpecificMessage(gameSpecificMessage);
  ImGui::Text("Game Specific: %s", exists ? gameSpecificMessage.c_str() : "?");

  if (!exists) ImGui::PopStyleColor();
}