Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

visualization #140

Open
wants to merge 5 commits into
base: fastsim-2
Choose a base branch
from
Open

visualization #140

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
b0be9f5
pushed
Jun 27, 2024
535d809
Merge branch 'fastsim-2' into f2/visualization
calbaker Jul 1, 2024
9bd3155
module added as a class (SimulationDrive)in simdrive.py
Jul 3, 2024
326fd70
Debug the class
Jul 3, 2024
9fb7c08
changes
Jul 26, 2024
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
200 changes: 200 additions & 0 deletions python/fastsim/plot_module.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,200 @@
import matplotlib.pyplot as plt
import numpy as np
from fuzzywuzzy import fuzz, process



available_signals = {
'Fuel Converter Output Power Achieved': 'fc_kw_out_ach',
'ESS Output Power Achieved': 'ess_kw_out_ach',
'Fuel Converter Input Power Achieved': 'fc_kw_in_ach',
'Acceleration Buffer SOC': 'accel_buff_soc',
'Acceleration Power': 'accel_kw',
'Ascent Energy (KJ)': 'ascent_kj',
'Ascent Power': 'ascent_kw',
'Auxiliary Power Input': 'aux_in_kw',
'Auxiliary Energy (KJ)': 'aux_kj',
'Battery Energy Consumption per Mile': 'battery_kwh_per_mi',
'Brake Energy (KJ)': 'brake_kj',
'Current ESS Max Output Power': 'cur_ess_max_kw_out',
'Current Max Available Electric Power': 'cur_max_avail_elec_kw',
'Current Max Electric Power': 'cur_max_elec_kw',
'Current Max ESS Charge Power': 'cur_max_ess_chg_kw',
'Current Max Fuel Converter Output Power': 'cur_max_fc_kw_out',
'Current Max Fuel System Output Power': 'cur_max_fs_kw_out',
'Current Max Motor Controller Input Power': 'cur_max_mc_elec_kw_in',
'Current Max Motor Controller Output Power': 'cur_max_mc_kw_out',
'Current Max Mechanical Motor Controller Input Power': 'cur_max_mech_mc_kw_in',
'Current Max Roadway Charge Power': 'cur_max_roadway_chg_kw',
'Current Max Traction Power': 'cur_max_trac_kw',
'Current Max Transmission Output Power': 'cur_max_trans_kw_out',
'Current SOC Target': 'cur_soc_target',
'Cycle Friction Brake Power': 'cyc_fric_brake_kw',
'Cycle Met': 'cyc_met',
'Cycle Regen Brake Power': 'cyc_regen_brake_kw',
'Cycle Tire Inertia Power': 'cyc_tire_inertia_kw',
'Cycle Traction Power Required': 'cyc_trac_kw_req',
'Cycle Transmission Output Power Required': 'cyc_trans_kw_out_req',
'Cycle Wheel Power Required': 'cyc_whl_kw_req',
'Cycle Wheel Rad Per Sec': 'cyc_whl_rad_per_sec',
'Desired ESS Output Power for AE': 'desired_ess_kw_out_for_ae',
'Distance (m)': 'dist_m',
'Distance (mi)': 'dist_mi',
'DOD Cycles': 'dod_cycs',
'Drag Energy (KJ)': 'drag_kj',
'Drag Power': 'drag_kw',
'Electric Power Required for AE': 'elec_kw_req_4ae',
kylecarow marked this conversation as resolved.
Show resolved Hide resolved
'Electric Energy Consumption per Mile': 'electric_kwh_per_mi',
'Energy Audit Error': 'energy_audit_error',
'ER AE Output Power': 'er_ae_kw_out',
'ER Power If Fuel Cell Required': 'er_kw_if_fc_req',
'ESS to Fuel Energy (KWH)': 'ess2fuel_kwh',
'ESS Acceleration Buffer Charge Power': 'ess_accel_buff_chg_kw',
'ESS Acceleration Regen Discharge Power': 'ess_accel_regen_dischg_kw',
'ESS AE Output Power': 'ess_ae_kw_out',
'ESS Capacity Limit Charge Power': 'ess_cap_lim_chg_kw',
'ESS Capacity Limit Discharge Power': 'ess_cap_lim_dischg_kw',
'ESS Current Energy (KWH)': 'ess_cur_kwh',
'ESS Desired Power for FC Efficiency': 'ess_desired_kw_4fc_eff',
'ESS Discharge Energy (KJ)': 'ess_dischg_kj',
'ESS Efficiency Energy (KJ)': 'ess_eff_kj',
'ESS Power If Fuel Cell Required': 'ess_kw_if_fc_req',
'ESS Lim Motor Controller Regen Percent Power': 'ess_lim_mc_regen_perc_kw',
'ESS Loss Power': 'ess_loss_kw',
'ESS Percentage Dead': 'ess_perc_dead',
'ESS Regen Buffer Discharge Power': 'ess_regen_buff_dischg_kw',
'Fuel Converter Energy (KJ)': 'fc_kj',
'Fuel Converter Power Gap from Efficiency': 'fc_kw_gap_fr_eff',
'Fuel Converter Output Power Achieved Percentage': 'fc_kw_out_ach_pct',
'Fuel Converter Time On': 'fc_time_on',
'Fuel Converter Transmission Limit Power': 'fc_trans_lim_kw',
'Fuel System Cumulative MJ Output Achieved': 'fs_cumu_mj_out_ach',
'Fuel System Output Power Achieved': 'fs_kw_out_ach',
'Fuel System Output Energy (KWH) Achieved': 'fs_kwh_out_ach',
'Fuel Energy (KJ)': 'fuel_kj',
'Gap to Lead Vehicle (m)': 'gap_to_lead_vehicle_m',
'IDM Target Speed (m/s)': 'idm_target_speed_m_per_s',
'Kinetic Energy (KJ)': 'ke_kj',
'Max ESS Acceleration Buffer Discharge Power': 'max_ess_accell_buff_dischg_kw',
'Max ESS Regen Buffer Charge Power': 'max_ess_regen_buff_chg_kw',
'Max Traction Speed (m/s)': 'max_trac_mps',
'Motor Controller Input Power for Max FC Efficiency': 'mc_elec_in_kw_for_max_fc_eff',
'Motor Controller Input Power Limit': 'mc_elec_in_lim_kw',
'Motor Controller Input Power Achieved': 'mc_elec_kw_in_ach',
'Motor Controller Input Power If Fuel Cell Required': 'mc_elec_kw_in_if_fc_req',
'Motor Controller Energy (KJ)': 'mc_kj',
'Motor Controller Power If Fuel Cell Required': 'mc_kw_if_fc_req',
'Motor Controller Mechanical Power Output Achieved': 'mc_mech_kw_out_ach',
'Motor Controller Transition Limit Power': 'mc_transi_lim_kw',
'Min ESS Power to Help FC': 'min_ess_kw_2help_fc',
'Min Motor Controller Power to Help FC': 'min_mc_kw_2help_fc',
'MPGGE': 'mpgge',
'Achieved Speed (MPH)': 'mph_ach',
'Achieved Speed (m/s)': 'mps_ach',
'Net Energy (KJ)': 'net_kj',
'Regen Buffer SOC': 'regen_buff_soc',
'Regen Control Limit KW Percentage': 'regen_contrl_lim_kw_perc',
'Roadway Charge Energy (KJ)': 'roadway_chg_kj',
'Roadway Charge Output Power Achieved': 'roadway_chg_kw_out_ach',
'Rolling Resistance Energy (KJ)': 'rr_kj',
'Rolling Resistance Power': 'rr_kw',
'State of Charge (SOC)': 'soc',
'Trace Miss': 'trace_miss',
'Trace Miss Distance Fraction': 'trace_miss_dist_frac',
'Trace Miss Iterations': 'trace_miss_iters',
'Trace Miss Speed (m/s)': 'trace_miss_speed_mps',
'Trace Miss Time Fraction': 'trace_miss_time_frac',
'Transmission Energy (KJ)': 'trans_kj',
kylecarow marked this conversation as resolved.
Show resolved Hide resolved
'Transmission Input Power Achieved': 'trans_kw_in_ach',
'Transmission Output Power Achieved': 'trans_kw_out_ach'
}




# Function for fuzzy matching signals
def fuzzy_match(signal_name, available_signals, feeling_lucky=False):
matches = process.extractOne(signal_name, available_signals.keys(), scorer=fuzz.token_sort_ratio)
best_match, score = matches
if score >= 70:
if feeling_lucky:
return best_match
kylecarow marked this conversation as resolved.
Show resolved Hide resolved
else:
print(f"Did you mean '{best_match}'? (y/n)")
choice = input().strip().lower()
if choice == 'y':
return best_match
return None




def plot(self, signal, fuzzy_search=True, feeling_lucky=False, speed_trace=True, difference=None, type='temporal'):
# Ensure signal is a list
if isinstance(signal, str):
signal = [signal]


# Handle fuzzy search
if fuzzy_search:
matched_signals = []
for sig in signal:
matched_signal = fuzzy_match(sig, available_signals, feeling_lucky)
if matched_signal:
matched_signals.append(matched_signal)
else:
print(f"No close match found for '{sig}'. Please enter a valid signal name.")
return None
else:
matched_signals = signal


# Prepare data for plotting
plot_data = []
for sig in matched_signals:
if sig in available_signals:
signal_key = available_signals[sig]
# Assuming the data is available in self after simulation
plot_data.append((getattr(self, signal_key), sig))
else:
print(f"Invalid signal name: '{sig}'")
return None


# Plotting
if type == 'temporal':
fig, axs = plt.subplots(len(plot_data) + (1 if speed_trace else 0), 1, figsize=(12, 8), sharex=True)
kylecarow marked this conversation as resolved.
Show resolved Hide resolved

for idx, (data, label) in enumerate(plot_data):
axs[idx].plot(self.cyc.time_s, data, label=label)
axs[idx].set_ylabel(label)

if speed_trace:
axs[-1].plot(self.cyc.time_s, self.mph_ach, label='Achieved Speed (MPH)', color='black')
axs[-1].set_ylabel('Achieved Speed (MPH)')

fig.suptitle('Temporal Performance Comparison')
plt.xlabel('Time (s)')
plt.legend()
plt.tight_layout()
plt.show()


elif type == 'spatial':
fig, ax = plt.subplots(figsize=(12, 8))

# Assuming 'dist_m' is the spatial distance data
for idx, (data, label) in enumerate(plot_data):
ax.plot(self.dist_m, data, label=label)

ax.set_xlabel('Distance (m)')
ax.set_ylabel('Signal Value')
ax.set_title('Spatial Performance Comparison')
ax.legend()
plt.tight_layout()
plt.show()

else:
print("Invalid type selected. Please choose 'temporal' or 'spatial'.")
return None

Loading
Loading