glass/src/app/native/cpp/main.cpp - RealtimeRoboticsGroup/aos - Gitiles

 // 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 <memory>

 #include <GLFW/glfw3.h>
 #include <fmt/format.h>
 #include <imgui.h>
 #include <ntcore_cpp.h>
 #include <wpigui.h>

 #include "glass/Context.h"
 #include "glass/MainMenuBar.h"
 #include "glass/Model.h"
 #include "glass/Storage.h"
 #include "glass/View.h"
 #include "glass/networktables/NetworkTables.h"
 #include "glass/networktables/NetworkTablesProvider.h"
 #include "glass/networktables/NetworkTablesSettings.h"
 #include "glass/other/Log.h"
 #include "glass/other/Plot.h"

 namespace gui = wpi::gui;

 const char* GetWPILibVersion();

 namespace glass {
 std::string_view GetResource_glass_16_png();
 std::string_view GetResource_glass_32_png();
 std::string_view GetResource_glass_48_png();
 std::string_view GetResource_glass_64_png();
 std::string_view GetResource_glass_128_png();
 std::string_view GetResource_glass_256_png();
 std::string_view GetResource_glass_512_png();
 }  // namespace glass

 static std::unique_ptr<glass::PlotProvider> gPlotProvider;
 static std::unique_ptr<glass::NetworkTablesProvider> gNtProvider;

 static std::unique_ptr<glass::NetworkTablesModel> gNetworkTablesModel;
 static std::unique_ptr<glass::NetworkTablesSettings> gNetworkTablesSettings;
 static glass::LogData gNetworkTablesLog;
 static std::unique_ptr<glass::Window> gNetworkTablesWindow;
 static std::unique_ptr<glass::Window> gNetworkTablesSettingsWindow;
 static std::unique_ptr<glass::Window> gNetworkTablesLogWindow;

 static glass::MainMenuBar gMainMenu;
 static bool gAbout = false;
 static bool gSetEnterKey = false;
 static bool gKeyEdit = false;

 static void NtInitialize() {
   // update window title when connection status changes
   auto inst = nt::GetDefaultInstance();
   auto poller = nt::CreateConnectionListenerPoller(inst);
   nt::AddPolledConnectionListener(poller, true);
   gui::AddEarlyExecute([poller] {
     auto win = gui::GetSystemWindow();
     if (!win) {
       return;
     }
     bool timedOut;
     for (auto&& event : nt::PollConnectionListener(poller, 0, &timedOut)) {
       if (event.connected) {
         glfwSetWindowTitle(
             win, fmt::format("Glass - Connected ({})", event.conn.remote_ip)
                      .c_str());
       } else {
         glfwSetWindowTitle(win, "Glass - DISCONNECTED");
       }
     }
   });

   // handle NetworkTables log messages
   auto logPoller = nt::CreateLoggerPoller(inst);
   nt::AddPolledLogger(logPoller, NT_LOG_INFO, 100);
   gui::AddEarlyExecute([logPoller] {
     bool timedOut;
     for (auto&& msg : nt::PollLogger(logPoller, 0, &timedOut)) {
       const char* level = "";
       if (msg.level >= NT_LOG_CRITICAL) {
         level = "CRITICAL: ";
       } else if (msg.level >= NT_LOG_ERROR) {
         level = "ERROR: ";
       } else if (msg.level >= NT_LOG_WARNING) {
         level = "WARNING: ";
       }
       gNetworkTablesLog.Append(fmt::format("{}{} ({}:{})\n", level, msg.message,
                                            msg.filename, msg.line));
     }
   });

   gNetworkTablesLogWindow = std::make_unique<glass::Window>(
       glass::GetStorageRoot().GetChild("NetworkTables Log"),
       "NetworkTables Log", glass::Window::kHide);
   gNetworkTablesLogWindow->SetView(
       std::make_unique<glass::LogView>(&gNetworkTablesLog));
   gNetworkTablesLogWindow->SetDefaultPos(250, 615);
   gNetworkTablesLogWindow->SetDefaultSize(600, 130);
   gNetworkTablesLogWindow->DisableRenamePopup();
   gui::AddLateExecute([] { gNetworkTablesLogWindow->Display(); });

   // NetworkTables table window
   gNetworkTablesModel = std::make_unique<glass::NetworkTablesModel>();
   gui::AddEarlyExecute([] { gNetworkTablesModel->Update(); });

   gNetworkTablesWindow = std::make_unique<glass::Window>(
       glass::GetStorageRoot().GetChild("NetworkTables View"), "NetworkTables");
   gNetworkTablesWindow->SetView(
       std::make_unique<glass::NetworkTablesView>(gNetworkTablesModel.get()));
   gNetworkTablesWindow->SetDefaultPos(250, 277);
   gNetworkTablesWindow->SetDefaultSize(750, 185);
   gNetworkTablesWindow->DisableRenamePopup();
   gui::AddLateExecute([] { gNetworkTablesWindow->Display(); });

   // NetworkTables settings window
   gNetworkTablesSettings = std::make_unique<glass::NetworkTablesSettings>(
       glass::GetStorageRoot().GetChild("NetworkTables Settings"));
   gui::AddEarlyExecute([] { gNetworkTablesSettings->Update(); });

   gNetworkTablesSettingsWindow = std::make_unique<glass::Window>(
       glass::GetStorageRoot().GetChild("NetworkTables Settings"),
       "NetworkTables Settings");
   gNetworkTablesSettingsWindow->SetView(
       glass::MakeFunctionView([] { gNetworkTablesSettings->Display(); }));
   gNetworkTablesSettingsWindow->SetDefaultPos(30, 30);
   gNetworkTablesSettingsWindow->SetFlags(ImGuiWindowFlags_AlwaysAutoResize);
   gNetworkTablesSettingsWindow->DisableRenamePopup();
   gui::AddLateExecute([] { gNetworkTablesSettingsWindow->Display(); });

   gui::AddWindowScaler([](float scale) {
     // scale default window positions
     gNetworkTablesLogWindow->ScaleDefault(scale);
     gNetworkTablesWindow->ScaleDefault(scale);
     gNetworkTablesSettingsWindow->ScaleDefault(scale);
   });
 }

 #ifdef _WIN32
 int __stdcall WinMain(void* hInstance, void* hPrevInstance, char* pCmdLine,
                       int nCmdShow) {
   int argc = __argc;
   char** argv = __argv;
 #else
 int main(int argc, char** argv) {
 #endif
   std::string_view saveDir;
   if (argc == 2) {
     saveDir = argv[1];
   }

   gui::CreateContext();
   glass::CreateContext();

   gui::AddIcon(glass::GetResource_glass_16_png());
   gui::AddIcon(glass::GetResource_glass_32_png());
   gui::AddIcon(glass::GetResource_glass_48_png());
   gui::AddIcon(glass::GetResource_glass_64_png());
   gui::AddIcon(glass::GetResource_glass_128_png());
   gui::AddIcon(glass::GetResource_glass_256_png());
   gui::AddIcon(glass::GetResource_glass_512_png());

   gPlotProvider = std::make_unique<glass::PlotProvider>(
       glass::GetStorageRoot().GetChild("Plots"));
   gNtProvider = std::make_unique<glass::NetworkTablesProvider>(
       glass::GetStorageRoot().GetChild("NetworkTables"));

   glass::SetStorageName("glass");
   glass::SetStorageDir(saveDir.empty() ? gui::GetPlatformSaveFileDir()
                                        : saveDir);
   gPlotProvider->GlobalInit();
   gui::AddInit([] { glass::ResetTime(); });
   gNtProvider->GlobalInit();
   NtInitialize();

   glass::AddStandardNetworkTablesViews(*gNtProvider);

   gui::AddLateExecute([] { gMainMenu.Display(); });

   gMainMenu.AddMainMenu([] {
     if (ImGui::BeginMenu("View")) {
       if (ImGui::MenuItem("Set Enter Key")) {
         gSetEnterKey = true;
       }
       if (ImGui::MenuItem("Reset Time")) {
         glass::ResetTime();
       }
       ImGui::EndMenu();
     }
     if (ImGui::BeginMenu("NetworkTables")) {
       if (gNetworkTablesSettingsWindow) {
         gNetworkTablesSettingsWindow->DisplayMenuItem("NetworkTables Settings");
       }
       if (gNetworkTablesWindow) {
         gNetworkTablesWindow->DisplayMenuItem("NetworkTables View");
       }
       if (gNetworkTablesLogWindow) {
         gNetworkTablesLogWindow->DisplayMenuItem("NetworkTables Log");
       }
       ImGui::Separator();
       gNtProvider->DisplayMenu();
       ImGui::EndMenu();
     }
     if (ImGui::BeginMenu("Plot")) {
       bool paused = gPlotProvider->IsPaused();
       if (ImGui::MenuItem("Pause All Plots", nullptr, &paused)) {
         gPlotProvider->SetPaused(paused);
       }
       ImGui::Separator();
       gPlotProvider->DisplayMenu();
       ImGui::EndMenu();
     }

     if (ImGui::BeginMenu("Info")) {
       if (ImGui::MenuItem("About")) {
         gAbout = true;
       }
       ImGui::EndMenu();
     }
   });

   gui::AddLateExecute([] {
     if (gAbout) {
       ImGui::OpenPopup("About");
       gAbout = false;
     }
     if (ImGui::BeginPopupModal("About")) {
       ImGui::Text("Glass: A different kind of dashboard");
       ImGui::Separator();
       ImGui::Text("v%s", GetWPILibVersion());
       ImGui::Separator();
       ImGui::Text("Save location: %s", glass::GetStorageDir().c_str());
       if (ImGui::Button("Close")) {
         ImGui::CloseCurrentPopup();
       }
       ImGui::EndPopup();
     }

     int& enterKey = glass::GetStorageRoot().GetInt("enterKey", GLFW_KEY_ENTER);

     ImGuiIO& io = ImGui::GetIO();
     io.KeyMap[ImGuiKey_Enter] = enterKey;

     if (gSetEnterKey) {
       ImGui::OpenPopup("Set Enter Key");
       gSetEnterKey = false;
     }
     if (ImGui::BeginPopupModal("Set Enter Key")) {
       ImGui::Text("Set the key to use to mean 'Enter'");
       ImGui::Text("This is useful to edit values without the DS disabling");
       ImGui::Separator();

       if (gKeyEdit) {
         for (int i = 0; i < IM_ARRAYSIZE(io.KeysDown); ++i) {
           if (io.KeysDown[i]) {
             // remove all other uses
             enterKey = i;
             gKeyEdit = false;
             break;
           }
         }
       }

       ImGui::Text("Key:");
       ImGui::SameLine();
       char editLabel[40];
       char nameBuf[32];
       const char* name = glfwGetKeyName(enterKey, 0);
       if (!name) {
         std::snprintf(nameBuf, sizeof(nameBuf), "%d", enterKey);
         name = nameBuf;
       }
       std::snprintf(editLabel, sizeof(editLabel), "%s###edit",
                     gKeyEdit ? "(press key)" : name);
       if (ImGui::SmallButton(editLabel)) {
         gKeyEdit = true;
       }
       ImGui::SameLine();
       if (ImGui::SmallButton("Reset")) {
         enterKey = GLFW_KEY_ENTER;
       }

       if (ImGui::Button("Close")) {
         ImGui::CloseCurrentPopup();
         gKeyEdit = false;
       }
       ImGui::EndPopup();
     }
   });

   gui::Initialize("Glass - DISCONNECTED", 1024, 768);
   gui::Main();

   gNetworkTablesSettingsWindow.reset();
   gNetworkTablesLogWindow.reset();
   gNetworkTablesWindow.reset();
   gNetworkTablesModel.reset();
   gNtProvider.reset();
   gPlotProvider.reset();

   glass::DestroyContext();
   gui::DestroyContext();

   return 0;
 }
	// 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 <memory>

	#include <GLFW/glfw3.h>
	#include <fmt/format.h>
	#include <imgui.h>
	#include <ntcore_cpp.h>
	#include <wpigui.h>

	#include "glass/Context.h"
	#include "glass/MainMenuBar.h"
	#include "glass/Model.h"
	#include "glass/Storage.h"
	#include "glass/View.h"
	#include "glass/networktables/NetworkTables.h"
	#include "glass/networktables/NetworkTablesProvider.h"
	#include "glass/networktables/NetworkTablesSettings.h"
	#include "glass/other/Log.h"
	#include "glass/other/Plot.h"

	namespace gui = wpi::gui;

	const char* GetWPILibVersion();

	namespace glass {
	std::string_view GetResource_glass_16_png();
	std::string_view GetResource_glass_32_png();
	std::string_view GetResource_glass_48_png();
	std::string_view GetResource_glass_64_png();
	std::string_view GetResource_glass_128_png();
	std::string_view GetResource_glass_256_png();
	std::string_view GetResource_glass_512_png();
	} // namespace glass

	static std::unique_ptr<glass::PlotProvider> gPlotProvider;
	static std::unique_ptr<glass::NetworkTablesProvider> gNtProvider;

	static std::unique_ptr<glass::NetworkTablesModel> gNetworkTablesModel;
	static std::unique_ptr<glass::NetworkTablesSettings> gNetworkTablesSettings;
	static glass::LogData gNetworkTablesLog;
	static std::unique_ptr<glass::Window> gNetworkTablesWindow;
	static std::unique_ptr<glass::Window> gNetworkTablesSettingsWindow;
	static std::unique_ptr<glass::Window> gNetworkTablesLogWindow;

	static glass::MainMenuBar gMainMenu;
	static bool gAbout = false;
	static bool gSetEnterKey = false;
	static bool gKeyEdit = false;

	static void NtInitialize() {
	// update window title when connection status changes
	auto inst = nt::GetDefaultInstance();
	auto poller = nt::CreateConnectionListenerPoller(inst);
	nt::AddPolledConnectionListener(poller, true);
	gui::AddEarlyExecute([poller] {
	auto win = gui::GetSystemWindow();
	if (!win) {
	return;
	}
	bool timedOut;
	for (auto&& event : nt::PollConnectionListener(poller, 0, &timedOut)) {
	if (event.connected) {
	glfwSetWindowTitle(
	win, fmt::format("Glass - Connected ({})", event.conn.remote_ip)
	.c_str());
	} else {
	glfwSetWindowTitle(win, "Glass - DISCONNECTED");
	}
	}
	});

	// handle NetworkTables log messages
	auto logPoller = nt::CreateLoggerPoller(inst);
	nt::AddPolledLogger(logPoller, NT_LOG_INFO, 100);
	gui::AddEarlyExecute([logPoller] {
	bool timedOut;
	for (auto&& msg : nt::PollLogger(logPoller, 0, &timedOut)) {
	const char* level = "";
	if (msg.level >= NT_LOG_CRITICAL) {
	level = "CRITICAL: ";
	} else if (msg.level >= NT_LOG_ERROR) {
	level = "ERROR: ";
	} else if (msg.level >= NT_LOG_WARNING) {
	level = "WARNING: ";
	}
	gNetworkTablesLog.Append(fmt::format("{}{} ({}:{})\n", level, msg.message,
	msg.filename, msg.line));
	}
	});

	gNetworkTablesLogWindow = std::make_unique<glass::Window>(
	glass::GetStorageRoot().GetChild("NetworkTables Log"),
	"NetworkTables Log", glass::Window::kHide);
	gNetworkTablesLogWindow->SetView(
	std::make_unique<glass::LogView>(&gNetworkTablesLog));
	gNetworkTablesLogWindow->SetDefaultPos(250, 615);
	gNetworkTablesLogWindow->SetDefaultSize(600, 130);
	gNetworkTablesLogWindow->DisableRenamePopup();
	gui::AddLateExecute([] { gNetworkTablesLogWindow->Display(); });

	// NetworkTables table window
	gNetworkTablesModel = std::make_unique<glass::NetworkTablesModel>();
	gui::AddEarlyExecute([] { gNetworkTablesModel->Update(); });

	gNetworkTablesWindow = std::make_unique<glass::Window>(
	glass::GetStorageRoot().GetChild("NetworkTables View"), "NetworkTables");
	gNetworkTablesWindow->SetView(
	std::make_unique<glass::NetworkTablesView>(gNetworkTablesModel.get()));
	gNetworkTablesWindow->SetDefaultPos(250, 277);
	gNetworkTablesWindow->SetDefaultSize(750, 185);
	gNetworkTablesWindow->DisableRenamePopup();
	gui::AddLateExecute([] { gNetworkTablesWindow->Display(); });

	// NetworkTables settings window
	gNetworkTablesSettings = std::make_unique<glass::NetworkTablesSettings>(
	glass::GetStorageRoot().GetChild("NetworkTables Settings"));
	gui::AddEarlyExecute([] { gNetworkTablesSettings->Update(); });

	gNetworkTablesSettingsWindow = std::make_unique<glass::Window>(
	glass::GetStorageRoot().GetChild("NetworkTables Settings"),
	"NetworkTables Settings");
	gNetworkTablesSettingsWindow->SetView(
	glass::MakeFunctionView([] { gNetworkTablesSettings->Display(); }));
	gNetworkTablesSettingsWindow->SetDefaultPos(30, 30);
	gNetworkTablesSettingsWindow->SetFlags(ImGuiWindowFlags_AlwaysAutoResize);
	gNetworkTablesSettingsWindow->DisableRenamePopup();
	gui::AddLateExecute([] { gNetworkTablesSettingsWindow->Display(); });

	gui::AddWindowScaler([](float scale) {
	// scale default window positions
	gNetworkTablesLogWindow->ScaleDefault(scale);
	gNetworkTablesWindow->ScaleDefault(scale);
	gNetworkTablesSettingsWindow->ScaleDefault(scale);
	});
	}

	#ifdef _WIN32
	int __stdcall WinMain(void* hInstance, void* hPrevInstance, char* pCmdLine,
	int nCmdShow) {
	int argc = __argc;
	char** argv = __argv;
	#else
	int main(int argc, char** argv) {
	#endif
	std::string_view saveDir;
	if (argc == 2) {
	saveDir = argv[1];
	}

	gui::CreateContext();
	glass::CreateContext();

	gui::AddIcon(glass::GetResource_glass_16_png());
	gui::AddIcon(glass::GetResource_glass_32_png());
	gui::AddIcon(glass::GetResource_glass_48_png());
	gui::AddIcon(glass::GetResource_glass_64_png());
	gui::AddIcon(glass::GetResource_glass_128_png());
	gui::AddIcon(glass::GetResource_glass_256_png());
	gui::AddIcon(glass::GetResource_glass_512_png());

	gPlotProvider = std::make_unique<glass::PlotProvider>(
	glass::GetStorageRoot().GetChild("Plots"));
	gNtProvider = std::make_unique<glass::NetworkTablesProvider>(
	glass::GetStorageRoot().GetChild("NetworkTables"));

	glass::SetStorageName("glass");
	glass::SetStorageDir(saveDir.empty() ? gui::GetPlatformSaveFileDir()
	: saveDir);
	gPlotProvider->GlobalInit();
	gui::AddInit([] { glass::ResetTime(); });
	gNtProvider->GlobalInit();
	NtInitialize();

	glass::AddStandardNetworkTablesViews(*gNtProvider);

	gui::AddLateExecute([] { gMainMenu.Display(); });

	gMainMenu.AddMainMenu([] {
	if (ImGui::BeginMenu("View")) {
	if (ImGui::MenuItem("Set Enter Key")) {
	gSetEnterKey = true;
	}
	if (ImGui::MenuItem("Reset Time")) {
	glass::ResetTime();
	}
	ImGui::EndMenu();
	}
	if (ImGui::BeginMenu("NetworkTables")) {
	if (gNetworkTablesSettingsWindow) {
	gNetworkTablesSettingsWindow->DisplayMenuItem("NetworkTables Settings");
	}
	if (gNetworkTablesWindow) {
	gNetworkTablesWindow->DisplayMenuItem("NetworkTables View");
	}
	if (gNetworkTablesLogWindow) {
	gNetworkTablesLogWindow->DisplayMenuItem("NetworkTables Log");
	}
	ImGui::Separator();
	gNtProvider->DisplayMenu();
	ImGui::EndMenu();
	}
	if (ImGui::BeginMenu("Plot")) {
	bool paused = gPlotProvider->IsPaused();
	if (ImGui::MenuItem("Pause All Plots", nullptr, &paused)) {
	gPlotProvider->SetPaused(paused);
	}
	ImGui::Separator();
	gPlotProvider->DisplayMenu();
	ImGui::EndMenu();
	}

	if (ImGui::BeginMenu("Info")) {
	if (ImGui::MenuItem("About")) {
	gAbout = true;
	}
	ImGui::EndMenu();
	}
	});

	gui::AddLateExecute([] {
	if (gAbout) {
	ImGui::OpenPopup("About");
	gAbout = false;
	}
	if (ImGui::BeginPopupModal("About")) {
	ImGui::Text("Glass: A different kind of dashboard");
	ImGui::Separator();
	ImGui::Text("v%s", GetWPILibVersion());
	ImGui::Separator();
	ImGui::Text("Save location: %s", glass::GetStorageDir().c_str());
	if (ImGui::Button("Close")) {
	ImGui::CloseCurrentPopup();
	}
	ImGui::EndPopup();
	}

	int& enterKey = glass::GetStorageRoot().GetInt("enterKey", GLFW_KEY_ENTER);

	ImGuiIO& io = ImGui::GetIO();
	io.KeyMap[ImGuiKey_Enter] = enterKey;

	if (gSetEnterKey) {
	ImGui::OpenPopup("Set Enter Key");
	gSetEnterKey = false;
	}
	if (ImGui::BeginPopupModal("Set Enter Key")) {
	ImGui::Text("Set the key to use to mean 'Enter'");
	ImGui::Text("This is useful to edit values without the DS disabling");
	ImGui::Separator();

	if (gKeyEdit) {
	for (int i = 0; i < IM_ARRAYSIZE(io.KeysDown); ++i) {
	if (io.KeysDown[i]) {
	// remove all other uses
	enterKey = i;
	gKeyEdit = false;
	break;
	}
	}
	}

	ImGui::Text("Key:");
	ImGui::SameLine();
	char editLabel[40];
	char nameBuf[32];
	const char* name = glfwGetKeyName(enterKey, 0);
	if (!name) {
	std::snprintf(nameBuf, sizeof(nameBuf), "%d", enterKey);
	name = nameBuf;
	}
	std::snprintf(editLabel, sizeof(editLabel), "%s###edit",
	gKeyEdit ? "(press key)" : name);
	if (ImGui::SmallButton(editLabel)) {
	gKeyEdit = true;
	}
	ImGui::SameLine();
	if (ImGui::SmallButton("Reset")) {
	enterKey = GLFW_KEY_ENTER;
	}

	if (ImGui::Button("Close")) {
	ImGui::CloseCurrentPopup();
	gKeyEdit = false;
	}
	ImGui::EndPopup();
	}
	});

	gui::Initialize("Glass - DISCONNECTED", 1024, 768);
	gui::Main();

	gNetworkTablesSettingsWindow.reset();
	gNetworkTablesLogWindow.reset();
	gNetworkTablesWindow.reset();
	gNetworkTablesModel.reset();
	gNtProvider.reset();
	gPlotProvider.reset();

	glass::DestroyContext();
	gui::DestroyContext();

	return 0;
	}