fix toggling extrema/roots

src/function.rs

@@ -362,6 +362,8 @@ impl FunctionEntry {
 			&self.function.get_derivative_str(),
 			(self.integral_min_x - self.integral_max_x).abs() / (self.integral_num as f64),
 			self.derivative,
+			extrema,
+			roots,
 		)
 	}
 }

src/function_output.rs

@@ -51,7 +51,7 @@ impl FunctionOutput {
 	/// disabled, it returns `f64::NAN`)
 	pub fn display(
 		&self, plot_ui: &mut PlotUi, func_str: &str, derivative_str: &str, step: f64,
-		derivative_enabled: bool,
+		derivative_enabled: bool, extrema: bool, roots: bool,
 	) -> f64 {
 		// Plot back data
 		plot_ui.line(
@@ -72,6 +72,7 @@ impl FunctionOutput {
 		}
 
 		// Plot extrema points
+		if extrema {
 			if let Some(extrema_data) = self.extrema.clone() {
 				plot_ui.points(
 					Points::new(Values::from_values(extrema_data))
@@ -80,8 +81,10 @@ impl FunctionOutput {
 						.radius(5.0),
 				);
 			}
+		}
 
 		// Plot roots points
+		if roots {
 			if let Some(roots_data) = self.roots.clone() {
 				plot_ui.points(
 					Points::new(Values::from_values(roots_data))
@@ -90,6 +93,7 @@ impl FunctionOutput {
 						.radius(5.0),
 				);
 			}
+		}
 
 		// Plot integral data
 		if let Some(integral_data) = self.integral.clone() {