allwpilib/wpilibc/src/main/native/cpp/Preferences.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 "frc/Preferences.h"

#include <algorithm>

#include <hal/FRCUsageReporting.h>
#include <networktables/NetworkTableInstance.h>
#include <wpi/StringRef.h>

using namespace frc;

// The Preferences table name
static wpi::StringRef kTableName{"Preferences"};

Preferences& Preferences::GetInstance() {
  static Preferences instance;
  return instance;
}

std::vector<std::string> Preferences::GetKeys() {
  return m_table->GetKeys();
}

std::string Preferences::GetString(wpi::StringRef key,
                                   wpi::StringRef defaultValue) {
  return m_table->GetString(key, defaultValue);
}

int Preferences::GetInt(wpi::StringRef key, int defaultValue) {
  return static_cast<int>(m_table->GetNumber(key, defaultValue));
}

double Preferences::GetDouble(wpi::StringRef key, double defaultValue) {
  return m_table->GetNumber(key, defaultValue);
}

float Preferences::GetFloat(wpi::StringRef key, float defaultValue) {
  return m_table->GetNumber(key, defaultValue);
}

bool Preferences::GetBoolean(wpi::StringRef key, bool defaultValue) {
  return m_table->GetBoolean(key, defaultValue);
}

int64_t Preferences::GetLong(wpi::StringRef key, int64_t defaultValue) {
  return static_cast<int64_t>(m_table->GetNumber(key, defaultValue));
}

void Preferences::SetString(wpi::StringRef key, wpi::StringRef value) {
  auto entry = m_table->GetEntry(key);
  entry.SetString(value);
  entry.SetPersistent();
}

void Preferences::PutString(wpi::StringRef key, wpi::StringRef value) {
  SetString(key, value);
}

void Preferences::InitString(wpi::StringRef key, wpi::StringRef value) {
  auto entry = m_table->GetEntry(key);
  entry.SetDefaultString(value);
}

void Preferences::SetInt(wpi::StringRef key, int value) {
  auto entry = m_table->GetEntry(key);
  entry.SetDouble(value);
  entry.SetPersistent();
}

void Preferences::PutInt(wpi::StringRef key, int value) {
  SetInt(key, value);
}

void Preferences::InitInt(wpi::StringRef key, int value) {
  auto entry = m_table->GetEntry(key);
  entry.SetDefaultDouble(value);
}

void Preferences::SetDouble(wpi::StringRef key, double value) {
  auto entry = m_table->GetEntry(key);
  entry.SetDouble(value);
  entry.SetPersistent();
}

void Preferences::PutDouble(wpi::StringRef key, double value) {
  SetDouble(key, value);
}

void Preferences::InitDouble(wpi::StringRef key, double value) {
  auto entry = m_table->GetEntry(key);
  entry.SetDefaultDouble(value);
}

void Preferences::SetFloat(wpi::StringRef key, float value) {
  auto entry = m_table->GetEntry(key);
  entry.SetDouble(value);
  entry.SetPersistent();
}

void Preferences::PutFloat(wpi::StringRef key, float value) {
  SetFloat(key, value);
}

void Preferences::InitFloat(wpi::StringRef key, float value) {
  auto entry = m_table->GetEntry(key);
  entry.SetDefaultDouble(value);
}

void Preferences::SetBoolean(wpi::StringRef key, bool value) {
  auto entry = m_table->GetEntry(key);
  entry.SetBoolean(value);
  entry.SetPersistent();
}

void Preferences::PutBoolean(wpi::StringRef key, bool value) {
  SetBoolean(key, value);
}

void Preferences::InitBoolean(wpi::StringRef key, bool value) {
  auto entry = m_table->GetEntry(key);
  entry.SetDefaultBoolean(value);
}

void Preferences::SetLong(wpi::StringRef key, int64_t value) {
  auto entry = m_table->GetEntry(key);
  entry.SetDouble(value);
  entry.SetPersistent();
}

void Preferences::PutLong(wpi::StringRef key, int64_t value) {
  SetLong(key, value);
}

void Preferences::InitLong(wpi::StringRef key, int64_t value) {
  auto entry = m_table->GetEntry(key);
  entry.SetDefaultDouble(value);
}

bool Preferences::ContainsKey(wpi::StringRef key) {
  return m_table->ContainsKey(key);
}

void Preferences::Remove(wpi::StringRef key) {
  m_table->Delete(key);
}

void Preferences::RemoveAll() {
  for (auto preference : GetKeys()) {
    if (preference != ".type") {
      Remove(preference);
    }
  }
}

Preferences::Preferences()
    : m_table(nt::NetworkTableInstance::GetDefault().GetTable(kTableName)) {
  m_table->GetEntry(".type").SetString("RobotPreferences");
  m_listener = m_table->AddEntryListener(
      [=](nt::NetworkTable* table, wpi::StringRef name,
          nt::NetworkTableEntry entry, std::shared_ptr<nt::Value> value,
          int flags) { entry.SetPersistent(); },
      NT_NOTIFY_NEW | NT_NOTIFY_IMMEDIATE);
  HAL_Report(HALUsageReporting::kResourceType_Preferences, 0);
}

Preferences::~Preferences() = default;