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

#include "frc/smartdashboard/Sendable.h"
#include "frc/smartdashboard/SendableHelper.h"
#include "frc/util/Color8Bit.h"

using namespace frc;

static constexpr char kPosition[] = "pos";

MechanismRoot2d::MechanismRoot2d(const wpi::Twine& name, double x, double y,
                                 const private_init&)
    : MechanismObject2d(name.str()), m_x{x}, m_y{y} {}

void MechanismRoot2d::SetPosition(double x, double y) {
  std::scoped_lock lock(m_mutex);
  m_x = x;
  m_y = y;
  Flush();
}

void MechanismRoot2d::UpdateEntries(std::shared_ptr<nt::NetworkTable> table) {
  m_posEntry = table->GetEntry(kPosition);
  Flush();
}

inline void MechanismRoot2d::Flush() {
  if (m_posEntry) {
    m_posEntry.SetDoubleArray({m_x, m_y});
  }
}