Single plot takes slices of curves + colors

This commit is contained in:
Kacper Leśniański 2022-08-13 19:00:41 +02:00
parent 0ce5278b52
commit e6257b531b
No known key found for this signature in database
GPG key ID: 10278258F8CAA50B
4 changed files with 79 additions and 21 deletions

View file

@ -0,0 +1,45 @@
use embedded_graphics::{pixelcolor::Rgb565, prelude::*};
use embedded_graphics_simulator::{OutputSettingsBuilder, SimulatorDisplay, Window};
use embedded_plots::{
axis::Scale,
curve::{Curve, PlotPoint},
single_plot::SinglePlot,
};
fn main() -> Result<(), core::convert::Infallible> {
let mut display: SimulatorDisplay<Rgb565> = SimulatorDisplay::new(Size::new(480, 272));
//min(x), max(x), min(y) and max(y) must be equal for each curve in order to display the plot correctly
let data = vec![
PlotPoint { x: 0, y: 0 },
PlotPoint { x: 1, y: 2 },
PlotPoint { x: 2, y: 2 },
PlotPoint { x: 3, y: 0 },
];
let data2 = vec![
PlotPoint { x: 0, y: 0 },
PlotPoint { x: 1, y: 2 },
PlotPoint { x: 2, y: 1 },
PlotPoint { x: 3, y: 1 },
];
let curve = Curve::from_data(data.as_slice());
let curve2 = Curve::from_data(data2.as_slice());
let curve_list = [(curve, RgbColor::YELLOW), (curve2, RgbColor::BLUE)];
let plot = SinglePlot::new(&curve_list, Scale::RangeFraction(3), Scale::RangeFraction(2))
.into_drawable(Point { x: 50, y: 10 }, Point { x: 430, y: 250 })
.set_color(RgbColor::YELLOW)
.set_text_color(RgbColor::WHITE);
plot.draw(&mut display)?;
let output_settings = OutputSettingsBuilder::new().build();
Window::new("Basic plot", &output_settings).show_static(&display);
Ok(())
}

View file

@ -19,7 +19,8 @@ fn main() -> Result<(), core::convert::Infallible> {
];
let curve = Curve::from_data(data.as_slice());
let plot = SinglePlot::new(&curve, Scale::RangeFraction(3), Scale::RangeFraction(2))
let curve_list = [(curve, BinaryColor::On)];
let plot = SinglePlot::new(&curve_list, Scale::RangeFraction(3), Scale::RangeFraction(2))
.into_drawable(Point { x: 18, y: 2 }, Point { x: 120, y: 30 })
.set_color(BinaryColor::On);

View file

@ -18,7 +18,8 @@ fn main() -> Result<(), core::convert::Infallible> {
];
let curve = Curve::from_data(data.as_slice());
let plot = SinglePlot::new(&curve, Scale::RangeFraction(3), Scale::RangeFraction(2))
let curve_list = [(curve, RgbColor::YELLOW)];
let plot = SinglePlot::new(&curve_list, Scale::RangeFraction(3), Scale::RangeFraction(2))
.into_drawable(Point { x: 50, y: 10 }, Point { x: 430, y: 250 })
.set_color(RgbColor::YELLOW)
.set_text_color(RgbColor::WHITE);

View file

@ -6,30 +6,33 @@ use embedded_graphics::{
};
/// Display agnostic single curve plot object
#[derive(Clone, Copy)]
pub struct SinglePlot<'a> {
pub struct SinglePlot<'a, C>
where
C: PixelColor + Default,
{
/// curve to be drawn on the plot
curve: &'a Curve<'a>,
curves: &'a [(Curve<'a>, C)],
/// range of X axis on which curve will be drawn
x_scale: Scale,
/// range of Y axis on which curve will be drawn
y_scale: Scale,
}
impl<'a> SinglePlot<'a> {
impl<'a, C> SinglePlot<'a, C>
where
C: PixelColor + Default,
{
/// create SinglePlot object with manual range
pub fn new(curve: &'a Curve<'a>, x_scale: Scale, y_scale: Scale) -> SinglePlot {
pub fn new(curves: &'a [(Curve<'a>, C)], x_scale: Scale, y_scale: Scale) -> SinglePlot<C> {
assert!(curves.len() > 0, "At least one curve must be given to SinglePlot constructor");
SinglePlot {
curve,
curves,
x_scale,
y_scale,
}
}
//TODO: add auto range plot constructor
/// convert to drawable form for specific display
pub fn into_drawable<C: PixelColor + Default>(
self,
top_left: Point,
bottom_right: Point,
) -> DrawableSinglePlot<'a, C> {
pub fn into_drawable(self, top_left: Point, bottom_right: Point) -> DrawableSinglePlot<'a, C> {
DrawableSinglePlot {
plot: self,
color: None,
@ -47,7 +50,7 @@ pub struct DrawableSinglePlot<'a, C>
where
C: PixelColor + Default,
{
plot: SinglePlot<'a>,
plot: SinglePlot<'a, C>,
color: Option<C>,
text_color: Option<C>,
axis_color: Option<C>,
@ -104,7 +107,11 @@ where
let thickness = self.thickness.unwrap_or(2);
let axis_thickness = self.axis_thickness.unwrap_or(thickness);
let text_style = MonoTextStyleBuilder::new().text_color(text_color).build();
Axis::new(self.plot.curve.x_range.clone())
let x_range = self.plot.curves[0].0.x_range.clone();
let y_range = self.plot.curves[0].0.y_range.clone();
Axis::new(x_range)
.set_title("X")
.set_scale(self.plot.x_scale)
.into_drawable_axis(Placement::X {
@ -117,7 +124,7 @@ where
.set_tick_size(2)
.set_thickness(axis_thickness)
.draw(display)?;
Axis::new(self.plot.curve.y_range.clone())
Axis::new(y_range)
.set_title("Y")
.set_scale(self.plot.y_scale)
.into_drawable_axis(Placement::Y {
@ -130,12 +137,16 @@ where
.set_tick_size(2)
.set_thickness(axis_thickness)
.draw(display)?;
self.plot
.curve
for curve in self.plot.curves {
curve
.0
.into_drawable_curve(&self.top_left, &self.bottom_right)
.set_color(color)
.set_color(curve.1)
.set_thickness(thickness)
.draw(display)?;
}
Ok(())
}
}