add course selection

This commit is contained in:
Mylloon 2024-01-01 11:39:31 +01:00
parent 77c2444de1
commit 9ad0de6222
Signed by: Anri
GPG key ID: A82D63DFF8D1317F

View file

@ -13,6 +13,7 @@ const DISCLAIMER: &str = "(selection avec ESPACE, ENTRER pour valider)";
pub fn timetable(timetable: Timetable) -> Timetable {
let mut my_timetable = timetable;
choice(&mut my_timetable);
courses(&mut my_timetable);
tdtp(&mut my_timetable);
@ -20,7 +21,7 @@ pub fn timetable(timetable: Timetable) -> Timetable {
}
/// Exclude some courses
fn courses(timetable: &mut Timetable) {
fn choice(timetable: &mut Timetable) {
let mut multiselected = vec![];
timetable.1 .1.iter().for_each(|day| {
day.courses.iter().for_each(|course_opt| {
@ -56,6 +57,89 @@ fn courses(timetable: &mut Timetable) {
}
}
/// Filter the multiple courses
fn courses(timetable: &mut Timetable) {
// Entry's name used for finding duplicates
let get_entry = |course: &Course| format!("{} - {:?}", course.name, course.category);
let mut hours = vec![];
fill_hours(&mut hours);
// Names showed to the users
let get_selection = |data: &(&Course, String)| {
format!(
"{} - {} {}-{}",
data.0.name,
data.1,
hours[data.0.start].split_once('-').unwrap().0,
hours[data.0.start + data.0.size - 1]
.split_once('-')
.unwrap()
.1
)
};
// List of courses who will be courses
let mut courses = vec![];
// Counter of courses appearing to know if multiples slots are available
let mut counts = HashMap::new();
timetable.1 .1.iter().for_each(|day| {
day.courses.iter().for_each(|course_opt| {
if let Some(course) = course_opt {
if course.category == Category::Cours {
courses.push((course, day.name.to_owned()));
let count = counts.entry(get_entry(course)).or_insert(0);
*count += 1;
}
}
})
});
// Keep only elements who have multiples slots
courses.retain(|course| *counts.get(&get_entry(course.0)).unwrap() > 1);
let mut multiselected: Vec<String> = courses.iter().map(get_selection).collect();
multiselected.sort();
let mut selections = vec![];
if !multiselected.is_empty() {
let defaults = vec![false; multiselected.len()];
selections = MultiSelect::new()
.with_prompt(format!("Choisis tes horaires de Cours {}", DISCLAIMER))
.items(&multiselected[..])
.defaults(&defaults[..])
.interact()
.unwrap();
}
// Keep only wanted courses
for day in &mut timetable.1 .1 {
day.courses.retain(|course_opt| {
if let Some(course) = course_opt {
// Keep if it's a TD/TP
if course.category == Category::TD || course.category == Category::TP {
return true;
}
// Keep if only one slot is available
if *counts.get(&get_entry(course)).unwrap() == 1 {
return true;
}
// Keep only chosen courses
for i in &selections {
if get_selection(&(course, day.name.to_owned())) == multiselected[*i] {
return true;
}
}
}
false
});
}
}
/// Filter the multiples TD/TP
fn tdtp(timetable: &mut Timetable) {
// Entry's name used for finding duplicates