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.

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

Add tire radius calculation utility in Rust #66

Merged
merged 2 commits into from
Nov 2, 2023
Merged
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
51 changes: 51 additions & 0 deletions rust/fastsim-core/src/utils.rs
Original file line number Diff line number Diff line change
@@ -1,5 +1,7 @@
//! Module containing miscellaneous utility functions.

use lazy_static::lazy_static;
use regex::Regex;
use std::collections::HashSet;

use crate::imports::*;
Expand Down Expand Up @@ -294,6 +296,55 @@ pub fn interp2d(
Ok(c)
}

lazy_static! {
static ref TIRE_CODE_REGEX: Regex = Regex::new(
r"(?i)[P|LT|ST|T]?((?:[0-9]{2,3}\.)?[0-9]+)/((?:[0-9]{1,2}\.)?[0-9]+) ?[B|D|R]?[x|\-| ]?((?:[0-9]{1,2}\.)?[0-9]+)[A|B|C|D|E|F|G|H|J|L|M|N]?"
).unwrap();
}

/// Calculate tire radius (in meters) from an [ISO metric tire code](https://en.wikipedia.org/wiki/Tire_code#ISO_metric_tire_codes)
///
/// # Arguments
/// * `tire_code` - A string containing a parsable ISO metric tire code
///
/// # Examples
/// ## Example 1:
///
/// ```rust
/// // Note the floating point imprecision in the result
/// use fastsim_core::utils::tire_code_to_radius;
/// let tire_code = "225/70Rx19.5G";
/// assert_eq!(tire_code_to_radius(&tire_code).unwrap(), 0.40514999999999995);
/// ```
///
/// ## Example 2:
///
/// ```rust
/// // Either `&str`, `&String`, or `String` can be passed
/// use fastsim_core::utils::tire_code_to_radius;
/// let tire_code = String::from("P205/60R16");
/// assert_eq!(tire_code_to_radius(tire_code).unwrap(), 0.3262);
/// ```
///
pub fn tire_code_to_radius<S: AsRef<str>>(tire_code: S) -> anyhow::Result<f64> {
let tire_code = tire_code.as_ref();
let captures = TIRE_CODE_REGEX.captures(tire_code).with_context(|| {
format!(
"Regex pattern does not match for {:?}: {:?}",
tire_code,
TIRE_CODE_REGEX.as_str(),
)
})?;
let width_mm: f64 = captures[1].parse()?;
let aspect_ratio: f64 = captures[2].parse()?;
let rim_diameter_in: f64 = captures[3].parse()?;

let sidewall_height_mm = width_mm * aspect_ratio / 100.0;
let radius_mm = (rim_diameter_in * 25.4) / 2.0 + sidewall_height_mm;

Ok(radius_mm / 1000.0)
}

#[cfg(feature = "pyo3")]
pub mod array_wrappers {
use crate::proc_macros::add_pyo3_api;
Expand Down