allwpilib/glass/src/lib/native/cpp/hardware/LEDDisplay.cpp

// Copyright (c) FIRST and other WPILib contributors.
// Open Source Software; you can modify and/or share it under the terms of
// the WPILib BSD license file in the root directory of this project.

#include "glass/hardware/LEDDisplay.h"

#include "glass/Context.h"
#include "glass/support/ExtraGuiWidgets.h"

using namespace glass;

namespace {
struct IndicatorData {
  std::vector<int> values;
  std::vector<ImU32> colors;
};
}  // namespace

void glass::DisplayLEDDisplay(LEDDisplayModel* model, int index) {
  wpi::SmallVector<LEDDisplayModel::Data, 64> dataBuf;
  auto data = model->GetData(dataBuf);
  int length = data.size();
  bool running = model->IsRunning();
  auto& storage = GetStorage();

  int* numColumns = storage.GetIntRef("columns", 10);
  bool* serpentine = storage.GetBoolRef("serpentine", false);
  int* order = storage.GetIntRef("order", LEDConfig::RowMajor);
  int* start = storage.GetIntRef("start", LEDConfig::UpperLeft);

  ImGui::PushItemWidth(ImGui::GetFontSize() * 6);
  ImGui::LabelText("Length", "%d", length);
  ImGui::LabelText("Running", "%s", running ? "Yes" : "No");
  ImGui::InputInt("Columns", numColumns);
  {
    static const char* options[] = {"Row Major", "Column Major"};
    ImGui::Combo("Order", order, options, 2);
  }
  {
    static const char* options[] = {"Upper Left", "Lower Left", "Upper Right",
                                    "Lower Right"};
    ImGui::Combo("Start", start, options, 4);
  }
  ImGui::Checkbox("Serpentine", serpentine);
  if (*numColumns < 1) *numColumns = 1;
  ImGui::PopItemWidth();

  // show as LED indicators
  auto iData = storage.GetData<IndicatorData>();
  if (!iData) {
    storage.SetData(std::make_shared<IndicatorData>());
    iData = storage.GetData<IndicatorData>();
  }
  if (length > static_cast<int>(iData->values.size()))
    iData->values.resize(length);
  if (length > static_cast<int>(iData->colors.size()))
    iData->colors.resize(length);
  if (!running) {
    iData->colors[0] = IM_COL32(128, 128, 128, 255);
    for (int j = 0; j < length; ++j) iData->values[j] = -1;
  } else {
    for (int j = 0; j < length; ++j) {
      iData->values[j] = j + 1;
      iData->colors[j] = IM_COL32(data[j].r, data[j].g, data[j].b, 255);
    }
  }

  LEDConfig config;
  config.serpentine = *serpentine;
  config.order = static_cast<LEDConfig::Order>(*order);
  config.start = static_cast<LEDConfig::Start>(*start);

  DrawLEDs(iData->values.data(), length, *numColumns, iData->colors.data(), 0,
           0, config);
}

void glass::DisplayLEDDisplays(LEDDisplaysModel* model) {
  bool hasAny = false;

  model->ForEachLEDDisplay([&](LEDDisplayModel& display, int i) {
    hasAny = true;
    if (model->GetNumLEDDisplays() > 1) ImGui::Text("LEDs[%d]", i);
    PushID(i);
    DisplayLEDDisplay(&display, i);
    PopID();
  });
  if (!hasAny) ImGui::Text("No addressable LEDs");
}