Source code for suboptimumg.sweep.types

from enum import Enum

# Parameter paths for Energy Grid sweep variables
PACK_WEIGHT_PARAM = "accum.pack_weight"
COAST_TRIGGER_PARAM = "pwrtn.coast_trigger"




class SweepDatatype(str, Enum):
    """All sweep variable types for strongly typed access."""

    ACCEL_PTS = "accel_pts"
    SKIDPAD_PTS = "skidpad_pts"
    AUTOX_PTS = "autoX_pts"
    ENDURANCE_PTS = "endurance_pts"
    EFFICIENCY_PTS = "efficiency_pts"
    ACCEL_T = "accel_t"
    SKIDPAD_T = "skidpad_t"
    AUTOX_T = "autoX_t"
    ENDURANCE_T = "endurance_t"
    TOTAL_PTS = "total_pts"





class EnergyGridMetric(str, Enum):
    """All selectable z-axis variables for energy grid plots, for strongly typed access."""

    ACCEL_PTS = "accel_pts"
    SKIDPAD_PTS = "skidpad_pts"
    AUTOX_PTS = "autoX_pts"
    ENDURANCE_PTS = "endurance_pts"
    EFFICIENCY_PTS = "efficiency_pts"
    ENDURANCE_T = "endurance_t"
    TOTAL_PTS = "total_pts"
    COAST_TRIGGER = "optimal_coast_trigger"
    NET_ENERGY = "net_energy_kwh"
    CAPACITY = "capacity_kwh"
    VEHICLE_MASS = "vehicle_mass"



# Human readable labels for energy grid metrics, for use in plot axes and hover templates
ENERGY_GRID_METRIC_LABELS = {
    EnergyGridMetric.TOTAL_PTS: "Total Points",
    EnergyGridMetric.ACCEL_PTS: "Accel Points",
    EnergyGridMetric.SKIDPAD_PTS: "Skidpad Points",
    EnergyGridMetric.AUTOX_PTS: "AutoX Points",
    EnergyGridMetric.ENDURANCE_PTS: "Endurance Points",
    EnergyGridMetric.EFFICIENCY_PTS: "Efficiency Points",
    EnergyGridMetric.ENDURANCE_T: "Endurance Time (s)",
    EnergyGridMetric.COAST_TRIGGER: "Coast Trigger (m/s)",
    EnergyGridMetric.NET_ENERGY: "Net Energy (kWh)",
    EnergyGridMetric.CAPACITY: "Pack Capacity (kWh)",
    EnergyGridMetric.VEHICLE_MASS: "Vehicle Mass (kg)",
}