Skip to content

Commit

Permalink
AP_Relay: capitalize function enum
Browse files Browse the repository at this point in the history
  • Loading branch information
IamPete1 committed Dec 18, 2023
1 parent 227113f commit b177be2
Show file tree
Hide file tree
Showing 4 changed files with 32 additions and 32 deletions.
34 changes: 17 additions & 17 deletions libraries/AP_Relay/AP_Relay.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -183,28 +183,28 @@ void AP_Relay::convert_params()
}

// Work out what function this relay should be
AP_Relay_Params::Function new_fun;
AP_Relay_Params::FUNCTION new_fun;
if (i == chute_relay) {
new_fun = AP_Relay_Params::Function::parachute;
new_fun = AP_Relay_Params::FUNCTION::PARACHUTE;

} else if (i == ice_relay) {
new_fun = AP_Relay_Params::Function::ignition;
new_fun = AP_Relay_Params::FUNCTION::IGNITION;

} else if (i == cam_relay) {
new_fun = AP_Relay_Params::Function::camera;
new_fun = AP_Relay_Params::FUNCTION::CAMERA;

#if APM_BUILD_TYPE(APM_BUILD_Rover)
} else if (i == rover_relay[0]) {
new_fun = AP_Relay_Params::Function::brushed_reverse_1;
new_fun = AP_Relay_Params::FUNCTION::BRUSHED_REVERSE_1;

} else if (i == rover_relay[1]) {
new_fun = AP_Relay_Params::Function::brushed_reverse_2;
new_fun = AP_Relay_Params::FUNCTION::BRUSHED_REVERSE_2;

} else if (i == rover_relay[2]) {
new_fun = AP_Relay_Params::Function::brushed_reverse_3;
new_fun = AP_Relay_Params::FUNCTION::BRUSHED_REVERSE_3;

} else if (i == rover_relay[3]) {
new_fun = AP_Relay_Params::Function::brushed_reverse_4;
new_fun = AP_Relay_Params::FUNCTION::BRUSHED_REVERSE_4;
#endif

} else {
Expand All @@ -214,7 +214,7 @@ void AP_Relay::convert_params()
// This will result in a pre-arm promoting the user to resolve
continue;
}
new_fun = AP_Relay_Params::Function::relay;
new_fun = AP_Relay_Params::FUNCTION::RELAY;

}
_params[i].function.set_and_save(int8_t(new_fun));
Expand Down Expand Up @@ -257,13 +257,13 @@ void AP_Relay::init()
continue;
}

const AP_Relay_Params::Function function = _params[instance].function;
if (function < AP_Relay_Params::Function::relay || function >= AP_Relay_Params::Function::num_functions) {
const AP_Relay_Params::FUNCTION function = _params[instance].function;
if (function <= AP_Relay_Params::FUNCTION::NONE || function >= AP_Relay_Params::FUNCTION::NUM_FUNCTIONS) {
// invalid function, skip it
continue;
}

if (function == AP_Relay_Params::Function::relay) {
if (function == AP_Relay_Params::FUNCTION::RELAY) {
// relay by instance number, set the state to match our output
const AP_Relay_Params::DefaultState default_state = _params[instance].default_state;
if ((default_state == AP_Relay_Params::DefaultState::OFF) ||
Expand All @@ -279,8 +279,8 @@ void AP_Relay::init()
}
}

void AP_Relay::set(const AP_Relay_Params::Function function, const bool value) {
if (function <= AP_Relay_Params::Function::none && function >= AP_Relay_Params::Function::num_functions) {
void AP_Relay::set(const AP_Relay_Params::FUNCTION function, const bool value) {
if (function <= AP_Relay_Params::FUNCTION::NONE && function >= AP_Relay_Params::FUNCTION::NUM_FUNCTIONS) {
// invalid function
return;
}
Expand Down Expand Up @@ -328,7 +328,7 @@ void AP_Relay::set(const uint8_t instance, const bool value)
return;
}

if (_params[instance].function != AP_Relay_Params::Function::relay) {
if (_params[instance].function != AP_Relay_Params::FUNCTION::RELAY) {
return;
}

Expand Down Expand Up @@ -383,11 +383,11 @@ bool AP_Relay::get(uint8_t instance) const
bool AP_Relay::enabled(uint8_t instance) const
{
// Must be a valid instance with function relay and pin set
return (instance < AP_RELAY_NUM_RELAYS) && (_params[instance].pin != -1) && (_params[instance].function == AP_Relay_Params::Function::relay);
return (instance < AP_RELAY_NUM_RELAYS) && (_params[instance].pin != -1) && (_params[instance].function == AP_Relay_Params::FUNCTION::RELAY);
}

// see if the relay is enabled
bool AP_Relay::enabled(AP_Relay_Params::Function function) const
bool AP_Relay::enabled(AP_Relay_Params::FUNCTION function) const
{
for (uint8_t instance = 0; instance < AP_RELAY_NUM_RELAYS; instance++) {
if ((_params[instance].function == function) && (_params[instance].pin != -1)) {
Expand Down
4 changes: 2 additions & 2 deletions libraries/AP_Relay/AP_Relay.h
Original file line number Diff line number Diff line change
Expand Up @@ -55,10 +55,10 @@ class AP_Relay {

bool send_relay_status(const class GCS_MAVLINK &link) const;

void set(AP_Relay_Params::Function function, bool value);
void set(AP_Relay_Params::FUNCTION function, bool value);

// see if the relay is enabled
bool enabled(AP_Relay_Params::Function function) const;
bool enabled(AP_Relay_Params::FUNCTION function) const;

private:
static AP_Relay *singleton;
Expand Down
2 changes: 1 addition & 1 deletion libraries/AP_Relay/AP_Relay_Params.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ const AP_Param::GroupInfo AP_Relay_Params::var_info[] = {
// @Values{Rover}: 8:Bushed motor reverse 4 omni motor 4

// @User: Standard
AP_GROUPINFO_FLAGS("FUNCTION", 1, AP_Relay_Params, function, (float)Function::none, AP_PARAM_FLAG_ENABLE),
AP_GROUPINFO_FLAGS("FUNCTION", 1, AP_Relay_Params, function, (float)FUNCTION::NONE, AP_PARAM_FLAG_ENABLE),

// @Param: PIN
// @DisplayName: Relay pin
Expand Down
24 changes: 12 additions & 12 deletions libraries/AP_Relay/AP_Relay_Params.h
Original file line number Diff line number Diff line change
Expand Up @@ -18,20 +18,20 @@ class AP_Relay_Params {
NO_CHANGE = 2,
};

enum class Function : uint8_t {
none = 0,
relay = 1,
ignition = 2,
parachute = 3,
camera = 4,
brushed_reverse_1 = 5,
brushed_reverse_2 = 6,
brushed_reverse_3 = 7,
brushed_reverse_4 = 8,
num_functions // must be the last entry
enum class FUNCTION : uint8_t {
NONE = 0,
RELAY = 1,
IGNITION = 2,
PARACHUTE = 3,
CAMERA = 4,
BRUSHED_REVERSE_1 = 5,
BRUSHED_REVERSE_2 = 6,
BRUSHED_REVERSE_3 = 7,
BRUSHED_REVERSE_4 = 8,
NUM_FUNCTIONS // must be the last entry
};

AP_Enum<Function> function; // relay function
AP_Enum<FUNCTION> function; // relay function
AP_Int16 pin; // gpio pin number
AP_Enum<DefaultState> default_state; // default state
};

0 comments on commit b177be2

Please sign in to comment.