Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@ Currently available subsystems:
- Intake stop switch
- Pivot
- Telescope
- LEDs

Planned features:
- Intake and flywheel linear speed control
Expand Down
67 changes: 67 additions & 0 deletions src/main/java/com/orbit/frc/subsystems/leds/Leds.java
Original file line number Diff line number Diff line change
@@ -0,0 +1,67 @@
/*
* Package: com.orbit.frc.subsystems.leds
* Description: This package contains classes related to controlling LEDs on a robot.
*/

package com.orbit.frc.subsystems.leds;

import java.util.HashMap;

import edu.wpi.first.wpilibj.Spark;
import edu.wpi.first.wpilibj2.command.SubsystemBase;

/*
* Class: Leds
* Description: This class represents the subsystem responsible for controlling LEDs on a robot.
*/
public class Leds {
private final LedConfig config; // Configuration object for LEDs

private double LEDColour; // Current color of LEDs
public LEDStates LEDstate; // Current state of LEDs

HashMap<LEDStates, Double> LEDStatesMap = new HashMap<>(); // Map to store LED states and corresponding colors

/*
* Constructor: Leds
* Description: Initializes the LED subsystem with the provided configuration.
* Parameters:
* - config: Configuration object for LEDs
*/
public Leds(LedConfig config) {
this.LEDController = new Spark(config.LEDPort); // Initialize LED controller
Copy link
Copy Markdown
Owner

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

remember to change references to the controller when its in the config, and also check to see if it should be inited from port or just using the object passed in the config

Copy link
Copy Markdown
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

imo it should be inited from the port, makes it less confusing in config

Copy link
Copy Markdown
Owner

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

If the user wants to provide their own spark object, we should support that

this.LEDColour = 0.0; // Default LED color
this.LEDstate = LEDStates.ENABLED; // Default LED state
}

/*
* Method: addLEDState
* Description: Adds a new LED state with the specified color to the LED states map.
* Parameters:
* - state: Name of the LED state
* - colour: Color of the LED state from revdocs
*/
public void addLEDState(String state, double colour) {
LEDStatesMap.put(state, colour); // Add LED state to the map
}

/*
* Method: setLEDState
* Description: Sets the current LED state.
* Parameters:
* - state: Name of the LED state to set
*/
public void setLEDState(String state) {
this.LEDstate = state; // Set LED state
}

/*
* Method: periodic
* Description: This method is called periodically to update the LED color based on the current LED state.
*/
@Override
public void periodic() {
this.LEDColour = LEDStatesMap.get(this.LEDstate); // Get color for the current LED state
this.LEDController.set(this.LEDColour); // Set LED color
}
}
12 changes: 12 additions & 0 deletions src/main/java/com/orbit/frc/subsystems/leds/LedsConfig.java
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
package main.java.com.orbit.frc.subsystems.leds;

import com.orbit.frc.subsystems.SubsystemConfig;

public class LedsConfig extends SubsystemConfig{

/* LED config
*
*/
public int LEDPort;
public Spark LEDController;
}