Added swerve drive widget

Gold872 · Oct 18, 2023 · 46d7533 · 46d7533
1 parent 7b1756b
commit 46d7533
Show file tree

Hide file tree

Showing 3 changed files with 576 additions and 0 deletions.
diff --git a/lib/widgets/draggable_containers/draggable_nt4_widget_container.dart b/lib/widgets/draggable_containers/draggable_nt4_widget_container.dart
@@ -14,6 +14,7 @@ import 'package:elastic_dashboard/widgets/nt4_widgets/multi-topic/combo_box_choo
 import 'package:elastic_dashboard/widgets/nt4_widgets/multi-topic/robot_preferences.dart';
 import 'package:elastic_dashboard/widgets/nt4_widgets/multi-topic/split_button_chooser.dart';
 import 'package:elastic_dashboard/widgets/nt4_widgets/multi-topic/subsystem_widget.dart';
+import 'package:elastic_dashboard/widgets/nt4_widgets/multi-topic/swerve_drive.dart';
 import 'package:elastic_dashboard/widgets/nt4_widgets/single_topic/match_time.dart';
 import 'package:elastic_dashboard/widgets/nt4_widgets/single_topic/multi_color_view.dart';
 import 'package:elastic_dashboard/widgets/nt4_widgets/single_topic/number_bar.dart';
@@ -356,6 +357,11 @@ class DraggableNT4WidgetContainer extends DraggableWidgetContainer {
           key: UniqueKey(),
           jsonData: jsonData['properties'],
         );
+      case 'SwerveDrive':
+        return SwerveDriveWidget.fromJson(
+          key: UniqueKey(),
+          jsonData: jsonData['properties'],
+        );
       case 'ComboBox Chooser':
         return ComboBoxChooser.fromJson(
           key: UniqueKey(),

diff --git a/lib/widgets/network_tree/tree_row.dart b/lib/widgets/network_tree/tree_row.dart
@@ -14,6 +14,7 @@ import 'package:elastic_dashboard/widgets/nt4_widgets/multi-topic/power_distribu
 import 'package:elastic_dashboard/widgets/nt4_widgets/multi-topic/combo_box_chooser.dart';
 import 'package:elastic_dashboard/widgets/nt4_widgets/multi-topic/robot_preferences.dart';
 import 'package:elastic_dashboard/widgets/nt4_widgets/multi-topic/subsystem_widget.dart';
+import 'package:elastic_dashboard/widgets/nt4_widgets/multi-topic/swerve_drive.dart';
 import 'package:elastic_dashboard/widgets/nt4_widgets/single_topic/boolean_box.dart';
 import 'package:elastic_dashboard/widgets/nt4_widgets/nt4_widget.dart';
 import 'package:elastic_dashboard/widgets/nt4_widgets/single_topic/text_display.dart';
@@ -159,6 +160,8 @@ class TreeRow {
         return PIDControllerWidget(key: UniqueKey(), topic: topic);
       case 'DifferentialDrive':
         return DifferentialDrive(key: UniqueKey(), topic: topic);
+      case 'SwerveDrive':
+        return SwerveDriveWidget(key: UniqueKey(), topic: topic);
       case 'String Chooser':
         return ComboBoxChooser(key: UniqueKey(), topic: topic);
       case 'Subsystem':
@@ -205,6 +208,9 @@ class TreeRow {
     } else if (primary is DifferentialDrive) {
       width = normalGridSize * 3;
       height = normalGridSize * 2;
+    } else if (primary is SwerveDriveWidget) {
+      width = normalGridSize * 2;
+      height = normalGridSize * 2;
     } else if (primary is SubsystemWidget) {
       width = normalGridSize * 2;
     } else if (primary is CommandWidget) {