mirror of
https://github.com/wpilibsuite/allwpilib
synced 2026-06-20 00:51:42 +00:00
[glass, wpilib] Rewrite Mechanism2d (#3281)
Substantially improves Mechanism2d by moving it to NetworkTables and adding a robot API to create the mechanism elements, instead of requiring a JSON file. Co-authored-by: Peter Johnson <johnson.peter@gmail.com>
This commit is contained in:
@@ -0,0 +1,80 @@
|
||||
// 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/MechanismLigament2d.h"
|
||||
|
||||
#include <cstdio>
|
||||
|
||||
using namespace frc;
|
||||
|
||||
MechanismLigament2d::MechanismLigament2d(const wpi::Twine& name, double length,
|
||||
units::degree_t angle,
|
||||
double lineWeight,
|
||||
const frc::Color8Bit& color)
|
||||
: MechanismObject2d(name),
|
||||
m_length{length},
|
||||
m_angle{angle.to<double>()},
|
||||
m_weight{lineWeight} {
|
||||
SetColor(color);
|
||||
}
|
||||
|
||||
void MechanismLigament2d::UpdateEntries(std::shared_ptr<NetworkTable> table) {
|
||||
table->GetEntry(".type").SetString("line");
|
||||
|
||||
m_colorEntry = table->GetEntry("color");
|
||||
m_angleEntry = table->GetEntry("angle");
|
||||
m_weightEntry = table->GetEntry("weight");
|
||||
m_lengthEntry = table->GetEntry("length");
|
||||
Flush();
|
||||
}
|
||||
|
||||
void MechanismLigament2d::SetColor(const Color8Bit& color) {
|
||||
std::scoped_lock lock(m_mutex);
|
||||
std::snprintf(m_color, sizeof(m_color), "#%02X%02X%02X", color.red,
|
||||
color.green, color.blue);
|
||||
Flush();
|
||||
}
|
||||
|
||||
void MechanismLigament2d::SetAngle(units::degree_t angle) {
|
||||
std::scoped_lock lock(m_mutex);
|
||||
m_angle = angle.to<double>();
|
||||
Flush();
|
||||
}
|
||||
|
||||
void MechanismLigament2d::SetLineWeight(double lineWidth) {
|
||||
std::scoped_lock lock(m_mutex);
|
||||
m_weight = lineWidth;
|
||||
Flush();
|
||||
}
|
||||
|
||||
double MechanismLigament2d::GetAngle() {
|
||||
if (m_angleEntry) {
|
||||
m_angle = m_angleEntry.GetDouble(0.0);
|
||||
}
|
||||
return m_angle;
|
||||
}
|
||||
|
||||
double MechanismLigament2d::GetLength() {
|
||||
if (m_lengthEntry) {
|
||||
m_length = m_lengthEntry.GetDouble(0.0);
|
||||
}
|
||||
return m_length;
|
||||
}
|
||||
|
||||
void MechanismLigament2d::SetLength(double length) {
|
||||
std::scoped_lock lock(m_mutex);
|
||||
m_length = length;
|
||||
Flush();
|
||||
}
|
||||
|
||||
#define SAFE_WRITE(data, Type) \
|
||||
if (m_##data##Entry) { \
|
||||
m_##data##Entry.Set##Type(m_##data); \
|
||||
}
|
||||
void MechanismLigament2d::Flush() {
|
||||
SAFE_WRITE(color, String)
|
||||
SAFE_WRITE(angle, Double)
|
||||
SAFE_WRITE(length, Double)
|
||||
SAFE_WRITE(weight, Double)
|
||||
}
|
||||
Reference in New Issue
Block a user