//! [Criterion]'s plotting library. //! //! [Criterion]: https://github.com/bheisler/criterion.rs //! //! **WARNING** This library is criterion's implementation detail and there no plans to stabilize //! it. In other words, the API may break at any time without notice. //! //! # Examples //! //! - Simple "curves" (based on [`simple.dem`](http://gnuplot.sourceforge.net/demo/simple.html)) //! //! ![Plot](curve.svg) //! //! ``` //! # use std::fs; //! # use std::path::Path; //! use itertools_num::linspace; //! use criterion_plot::prelude::*; //! //! # if let Err(_) = criterion_plot::version() { //! # return; //! # } //! let ref xs = linspace::(-10., 10., 51).collect::>(); //! //! # fs::create_dir_all(Path::new("target/doc/criterion_plot")).unwrap(); //! # assert_eq!(Some(String::new()), //! Figure::new() //! # .set(Font("Helvetica")) //! # .set(FontSize(12.)) //! # .set(Output(Path::new("target/doc/criterion_plot/curve.svg"))) //! # .set(Size(1280, 720)) //! .configure(Key, |k| { //! k.set(Boxed::Yes) //! .set(Position::Inside(Vertical::Top, Horizontal::Left)) //! }) //! .plot(LinesPoints { //! x: xs, //! y: xs.iter().map(|x| x.sin()), //! }, //! |lp| { //! lp.set(Color::DarkViolet) //! .set(Label("sin(x)")) //! .set(LineType::Dash) //! .set(PointSize(1.5)) //! .set(PointType::Circle) //! }) //! .plot(Steps { //! x: xs, //! y: xs.iter().map(|x| x.atan()), //! }, //! |s| { //! s.set(Color::Rgb(0, 158, 115)) //! .set(Label("atan(x)")) //! .set(LineWidth(2.)) //! }) //! .plot(Impulses { //! x: xs, //! y: xs.iter().map(|x| x.atan().cos()), //! }, //! |i| { //! i.set(Color::Rgb(86, 180, 233)) //! .set(Label("cos(atan(x))")) //! }) //! .draw() // (rest of the chain has been omitted) //! # .ok() //! # .and_then(|gnuplot| { //! # gnuplot.wait_with_output().ok().and_then(|p| String::from_utf8(p.stderr).ok()) //! # })); //! ``` //! //! - error bars (based on //! [Julia plotting tutorial](https://plot.ly/julia/error-bars/#Colored-and-Styled-Error-Bars)) //! //! ![Plot](error_bar.svg) //! //! ``` //! # use std::fs; //! # use std::path::Path; //! use std::f64::consts::PI; //! //! use itertools_num::linspace; //! use rand::{Rng, XorShiftRng}; //! use criterion_plot::prelude::*; //! //! fn sinc(mut x: f64) -> f64 { //! if x == 0. { //! 1. //! } else { //! x *= PI; //! x.sin() / x //! } //! } //! //! # if let Err(_) = criterion_plot::version() { //! # return; //! # } //! let ref xs_ = linspace::(-4., 4., 101).collect::>(); //! //! // Fake some data //! let ref mut rng: XorShiftRng = rand::thread_rng().gen(); //! let xs = linspace::(-4., 4., 13).skip(1).take(11); //! let ys = xs.map(|x| sinc(x) + 0.05 * rng.gen::() - 0.025).collect::>(); //! let y_low = ys.iter().map(|&y| y - 0.025 - 0.075 * rng.gen::()).collect::>(); //! let y_high = ys.iter().map(|&y| y + 0.025 + 0.075 * rng.gen::()).collect::>(); //! let xs = linspace::(-4., 4., 13).skip(1).take(11); //! let xs = xs.map(|x| x + 0.2 * rng.gen::() - 0.1); //! //! # fs::create_dir_all(Path::new("target/doc/criterion_plot")).unwrap(); //! # assert_eq!(Some(String::new()), //! Figure::new() //! # .set(Font("Helvetica")) //! # .set(FontSize(12.)) //! # .set(Output(Path::new("target/doc/criterion_plot/error_bar.svg"))) //! # .set(Size(1280, 720)) //! .configure(Axis::BottomX, |a| { //! a.set(TicLabels { //! labels: &["-π", "0", "π"], //! positions: &[-PI, 0., PI], //! }) //! }) //! .configure(Key, //! |k| k.set(Position::Outside(Vertical::Top, Horizontal::Right))) //! .plot(Lines { //! x: xs_, //! y: xs_.iter().cloned().map(sinc), //! }, //! |l| { //! l.set(Color::Rgb(0, 158, 115)) //! .set(Label("sinc(x)")) //! .set(LineWidth(2.)) //! }) //! .plot(YErrorBars { //! x: xs, //! y: &ys, //! y_low: &y_low, //! y_high: &y_high, //! }, //! |eb| { //! eb.set(Color::DarkViolet) //! .set(LineWidth(2.)) //! .set(PointType::FilledCircle) //! .set(Label("measured")) //! }) //! .draw() // (rest of the chain has been omitted) //! # .ok() //! # .and_then(|gnuplot| { //! # gnuplot.wait_with_output().ok().and_then(|p| String::from_utf8(p.stderr).ok()) //! # })); //! ``` //! //! - Candlesticks (based on //! [`candlesticks.dem`](http://gnuplot.sourceforge.net/demo/candlesticks.html)) //! //! ![Plot](candlesticks.svg) //! //! ``` //! # use std::fs; //! # use std::path::Path; //! use criterion_plot::prelude::*; //! use rand::Rng; //! //! # if let Err(_) = criterion_plot::version() { //! # return; //! # } //! let xs = 1..11; //! //! // Fake some data //! let mut rng = rand::thread_rng(); //! let bh = xs.clone().map(|_| 5f64 + 2.5 * rng.gen::()).collect::>(); //! let bm = xs.clone().map(|_| 2.5f64 + 2.5 * rng.gen::()).collect::>(); //! let wh = bh.iter().map(|&y| y + (10. - y) * rng.gen::()).collect::>(); //! let wm = bm.iter().map(|&y| y * rng.gen::()).collect::>(); //! let m = bm.iter().zip(bh.iter()).map(|(&l, &h)| (h - l) * rng.gen::() + l) //! .collect::>(); //! //! # fs::create_dir_all(Path::new("target/doc/criterion_plot")).unwrap(); //! # assert_eq!(Some(String::new()), //! Figure::new() //! # .set(Font("Helvetica")) //! # .set(FontSize(12.)) //! # .set(Output(Path::new("target/doc/criterion_plot/candlesticks.svg"))) //! # .set(Size(1280, 720)) //! .set(BoxWidth(0.2)) //! .configure(Axis::BottomX, |a| a.set(Range::Limits(0., 11.))) //! .plot(Candlesticks { //! x: xs.clone(), //! whisker_min: &wm, //! box_min: &bm, //! box_high: &bh, //! whisker_high: &wh, //! }, //! |cs| { //! cs.set(Color::Rgb(86, 180, 233)) //! .set(Label("Quartiles")) //! .set(LineWidth(2.)) //! }) //! // trick to plot the median //! .plot(Candlesticks { //! x: xs, //! whisker_min: &m, //! box_min: &m, //! box_high: &m, //! whisker_high: &m, //! }, //! |cs| { //! cs.set(Color::Black) //! .set(LineWidth(2.)) //! }) //! .draw() // (rest of the chain has been omitted) //! # .ok() //! # .and_then(|gnuplot| { //! # gnuplot.wait_with_output().ok().and_then(|p| String::from_utf8(p.stderr).ok()) //! # })); //! ``` //! //! - Multiaxis (based on [`multiaxis.dem`](http://gnuplot.sourceforge.net/demo/multiaxis.html)) //! //! ![Plot](multiaxis.svg) //! //! ``` //! # use std::fs; //! # use std::path::Path; //! use std::f64::consts::PI; //! //! use itertools_num::linspace; //! use num_complex::Complex; //! use criterion_plot::prelude::*; //! //! fn tf(x: f64) -> Complex { //! Complex::new(0., x) / Complex::new(10., x) / Complex::new(1., x / 10_000.) //! } //! //! # if let Err(_) = criterion_plot::version() { //! # return; //! # } //! let (start, end): (f64, f64) = (1.1, 90_000.); //! let ref xs = linspace(start.ln(), end.ln(), 101).map(|x| x.exp()).collect::>(); //! let phase = xs.iter().map(|&x| tf(x).arg() * 180. / PI); //! let magnitude = xs.iter().map(|&x| tf(x).norm()); //! //! # fs::create_dir_all(Path::new("target/doc/criterion_plot")).unwrap(); //! # assert_eq!(Some(String::new()), //! Figure::new(). //! # set(Font("Helvetica")). //! # set(FontSize(12.)). //! # set(Output(Path::new("target/doc/criterion_plot/multiaxis.svg"))). //! # set(Size(1280, 720)). //! set(Title("Frequency response")). //! configure(Axis::BottomX, |a| a. //! configure(Grid::Major, |g| g. //! show()). //! set(Label("Angular frequency (rad/s)")). //! set(Range::Limits(start, end)). //! set(Scale::Logarithmic)). //! configure(Axis::LeftY, |a| a. //! set(Label("Gain")). //! set(Scale::Logarithmic)). //! configure(Axis::RightY, |a| a. //! configure(Grid::Major, |g| g. //! show()). //! set(Label("Phase shift (°)"))). //! configure(Key, |k| k. //! set(Position::Inside(Vertical::Top, Horizontal::Center)). //! set(Title(" "))). //! plot(Lines { //! x: xs, //! y: magnitude, //! }, |l| l. //! set(Color::DarkViolet). //! set(Label("Magnitude")). //! set(LineWidth(2.))). //! plot(Lines { //! x: xs, //! y: phase, //! }, |l| l. //! set(Axes::BottomXRightY). //! set(Color::Rgb(0, 158, 115)). //! set(Label("Phase")). //! set(LineWidth(2.))). //! draw(). // (rest of the chain has been omitted) //! # ok().and_then(|gnuplot| { //! # gnuplot.wait_with_output().ok().and_then(|p| { //! # String::from_utf8(p.stderr).ok() //! # }) //! # })); //! ``` //! - Filled curves (based on //! [`transparent.dem`](http://gnuplot.sourceforge.net/demo/transparent.html)) //! //! ![Plot](filled_curve.svg) //! //! ``` //! # use std::fs; //! # use std::path::Path; //! use std::f64::consts::PI; //! use std::iter; //! //! use itertools_num::linspace; //! use criterion_plot::prelude::*; //! //! # if let Err(_) = criterion_plot::version() { //! # return; //! # } //! let (start, end) = (-5., 5.); //! let ref xs = linspace(start, end, 101).collect::>(); //! let zeros = iter::repeat(0); //! //! fn gaussian(x: f64, mu: f64, sigma: f64) -> f64 { //! 1. / (((x - mu).powi(2) / 2. / sigma.powi(2)).exp() * sigma * (2. * PI).sqrt()) //! } //! //! # fs::create_dir_all(Path::new("target/doc/criterion_plot")).unwrap(); //! # assert_eq!(Some(String::new()), //! Figure::new() //! # .set(Font("Helvetica")) //! # .set(FontSize(12.)) //! # .set(Output(Path::new("target/doc/criterion_plot/filled_curve.svg"))) //! # .set(Size(1280, 720)) //! .set(Title("Transparent filled curve")) //! .configure(Axis::BottomX, |a| a.set(Range::Limits(start, end))) //! .configure(Axis::LeftY, |a| a.set(Range::Limits(0., 1.))) //! .configure(Key, |k| { //! k.set(Justification::Left) //! .set(Order::SampleText) //! .set(Position::Inside(Vertical::Top, Horizontal::Left)) //! .set(Title("Gaussian Distribution")) //! }) //! .plot(FilledCurve { //! x: xs, //! y1: xs.iter().map(|&x| gaussian(x, 0.5, 0.5)), //! y2: zeros.clone(), //! }, //! |fc| { //! fc.set(Color::ForestGreen) //! .set(Label("μ = 0.5 σ = 0.5")) //! }) //! .plot(FilledCurve { //! x: xs, //! y1: xs.iter().map(|&x| gaussian(x, 2.0, 1.0)), //! y2: zeros.clone(), //! }, //! |fc| { //! fc.set(Color::Gold) //! .set(Label("μ = 2.0 σ = 1.0")) //! .set(Opacity(0.5)) //! }) //! .plot(FilledCurve { //! x: xs, //! y1: xs.iter().map(|&x| gaussian(x, -1.0, 2.0)), //! y2: zeros, //! }, //! |fc| { //! fc.set(Color::Red) //! .set(Label("μ = -1.0 σ = 2.0")) //! .set(Opacity(0.5)) //! }) //! .draw() //! .ok() //! .and_then(|gnuplot| { //! gnuplot.wait_with_output().ok().and_then(|p| String::from_utf8(p.stderr).ok()) //! })); //! ``` #![deny(missing_docs)] #![deny(warnings)] #![deny(bare_trait_objects)] // This lint has lots of false positives ATM, see // https://github.com/Manishearth/rust-clippy/issues/761 #![cfg_attr(feature = "cargo-clippy", allow(clippy::new_without_default))] // False positives with images #![cfg_attr(feature = "cargo-clippy", allow(clippy::doc_markdown))] #![cfg_attr(feature = "cargo-clippy", allow(clippy::many_single_char_names))] extern crate cast; #[macro_use] extern crate itertools; use std::borrow::Cow; use std::fmt; use std::fs::File; use std::io; use std::num::ParseIntError; use std::path::Path; use std::process::{Child, Command}; use std::str; use crate::data::Matrix; use crate::traits::{Configure, Set}; mod data; mod display; mod map; pub mod axis; pub mod candlestick; pub mod curve; pub mod errorbar; pub mod filledcurve; pub mod grid; pub mod key; pub mod prelude; pub mod proxy; pub mod traits; /// Plot container #[derive(Clone)] pub struct Figure { alpha: Option, axes: map::axis::Map, box_width: Option, font: Option>, font_size: Option, key: Option, output: Cow<'static, Path>, plots: Vec, size: Option<(usize, usize)>, terminal: Terminal, tics: map::axis::Map, title: Option>, } impl Figure { /// Creates an empty figure pub fn new() -> Figure { Figure { alpha: None, axes: map::axis::Map::new(), box_width: None, font: None, font_size: None, key: None, output: Cow::Borrowed(Path::new("output.plot")), plots: Vec::new(), size: None, terminal: Terminal::Svg, tics: map::axis::Map::new(), title: None, } } fn script(&self) -> Vec { let mut s = String::new(); s.push_str(&format!( "set output '{}'\n", self.output.display().to_string().replace("'", "''") )); if let Some(width) = self.box_width { s.push_str(&format!("set boxwidth {}\n", width)) } if let Some(ref title) = self.title { s.push_str(&format!("set title '{}'\n", title)) } for axis in self.axes.iter() { s.push_str(&axis.script()); } for (_, script) in self.tics.iter() { s.push_str(script); } if let Some(ref key) = self.key { s.push_str(&key.script()) } if let Some(alpha) = self.alpha { s.push_str(&format!("set style fill transparent solid {}\n", alpha)) } s.push_str(&format!("set terminal {} dashed", self.terminal.display())); if let Some((width, height)) = self.size { s.push_str(&format!(" size {}, {}", width, height)) } if let Some(ref name) = self.font { if let Some(size) = self.font_size { s.push_str(&format!(" font '{},{}'", name, size)) } else { s.push_str(&format!(" font '{}'", name)) } } // TODO This removes the crossbars from the ends of error bars, but should be configurable s.push_str("\nunset bars\n"); let mut is_first_plot = true; for plot in &self.plots { let data = plot.data(); if data.bytes().is_empty() { continue; } if is_first_plot { s.push_str("plot "); is_first_plot = false; } else { s.push_str(", "); } s.push_str(&format!( "'-' binary endian=little record={} format='%float64' using ", data.nrows() )); let mut is_first_col = true; for col in 0..data.ncols() { if is_first_col { is_first_col = false; } else { s.push(':'); } s.push_str(&(col + 1).to_string()); } s.push(' '); s.push_str(plot.script()); } let mut buffer = s.into_bytes(); let mut is_first = true; for plot in &self.plots { if is_first { is_first = false; buffer.push(b'\n'); } buffer.extend_from_slice(plot.data().bytes()); } buffer } /// Spawns a drawing child process /// /// NOTE: stderr, stdin, and stdout are piped pub fn draw(&mut self) -> io::Result { use std::process::Stdio; let mut gnuplot = Command::new("gnuplot") .stderr(Stdio::piped()) .stdin(Stdio::piped()) .stdout(Stdio::piped()) .spawn()?; self.dump(gnuplot.stdin.as_mut().unwrap())?; Ok(gnuplot) } /// Dumps the script required to produce the figure into `sink` pub fn dump(&mut self, sink: &mut W) -> io::Result<&mut Figure> where W: io::Write, { sink.write_all(&self.script())?; Ok(self) } /// Saves the script required to produce the figure to `path` pub fn save(&self, path: &Path) -> io::Result<&Figure> { use std::io::Write; File::create(path)?.write_all(&self.script())?; Ok(self) } } impl Configure for Figure { type Properties = axis::Properties; /// Configures an axis fn configure(&mut self, axis: Axis, configure: F) -> &mut Figure where F: FnOnce(&mut axis::Properties) -> &mut axis::Properties, { if self.axes.contains_key(axis) { configure(self.axes.get_mut(axis).unwrap()); } else { let mut properties = Default::default(); configure(&mut properties); self.axes.insert(axis, properties); } self } } impl Configure for Figure { type Properties = key::Properties; /// Configures the key (legend) fn configure(&mut self, _: Key, configure: F) -> &mut Figure where F: FnOnce(&mut key::Properties) -> &mut key::Properties, { if self.key.is_some() { configure(self.key.as_mut().unwrap()); } else { let mut key = Default::default(); configure(&mut key); self.key = Some(key); } self } } impl Set for Figure { /// Changes the box width of all the box related plots (bars, candlesticks, etc) /// /// **Note** The default value is 0 /// /// # Panics /// /// Panics if `width` is a negative value fn set(&mut self, width: BoxWidth) -> &mut Figure { let width = width.0; assert!(width >= 0.); self.box_width = Some(width); self } } impl Set for Figure { /// Changes the font fn set(&mut self, font: Font) -> &mut Figure { self.font = Some(font.0); self } } impl Set for Figure { /// Changes the size of the font /// /// # Panics /// /// Panics if `size` is a non-positive value fn set(&mut self, size: FontSize) -> &mut Figure { let size = size.0; assert!(size >= 0.); self.font_size = Some(size); self } } impl Set for Figure { /// Changes the output file /// /// **Note** The default output file is `output.plot` fn set(&mut self, output: Output) -> &mut Figure { self.output = output.0; self } } impl Set for Figure { /// Changes the figure size fn set(&mut self, size: Size) -> &mut Figure { self.size = Some((size.0, size.1)); self } } impl Set for Figure { /// Changes the output terminal /// /// **Note** By default, the terminal is set to `Svg` fn set(&mut self, terminal: Terminal) -> &mut Figure { self.terminal = terminal; self } } impl Set for Figure { /// Sets the title fn set(&mut self, title: Title) -> &mut Figure { self.title = Some(title.0); self } } impl Default for Figure { fn default() -> Self { Self::new() } } /// Box width for box-related plots: bars, candlesticks, etc #[derive(Clone, Copy)] pub struct BoxWidth(pub f64); /// A font name pub struct Font(Cow<'static, str>); /// The size of a font #[derive(Clone, Copy)] pub struct FontSize(pub f64); /// The key or legend #[derive(Clone, Copy)] pub struct Key; /// Plot label pub struct Label(Cow<'static, str>); /// Width of the lines #[derive(Clone, Copy)] pub struct LineWidth(pub f64); /// Fill color opacity #[derive(Clone, Copy)] pub struct Opacity(pub f64); /// Output file path pub struct Output(Cow<'static, Path>); /// Size of the points #[derive(Clone, Copy)] pub struct PointSize(pub f64); /// Axis range #[derive(Clone, Copy)] pub enum Range { /// Autoscale the axis Auto, /// Set the limits of the axis Limits(f64, f64), } /// Figure size #[derive(Clone, Copy)] pub struct Size(pub usize, pub usize); /// Labels attached to the tics of an axis pub struct TicLabels<P, L> { /// Labels to attach to the tics pub labels: L, /// Position of the tics on the axis pub positions: P, } /// Figure title pub struct Title(Cow<'static, str>); /// A pair of axes that define a coordinate system #[allow(missing_docs)] #[derive(Clone, Copy)] pub enum Axes { BottomXLeftY, BottomXRightY, TopXLeftY, TopXRightY, } /// A coordinate axis #[derive(Clone, Copy)] pub enum Axis { /// X axis on the bottom side of the figure BottomX, /// Y axis on the left side of the figure LeftY, /// Y axis on the right side of the figure RightY, /// X axis on the top side of the figure TopX, } impl Axis { fn next(self) -> Option<Axis> { use crate::Axis::*; match self { BottomX => Some(LeftY), LeftY => Some(RightY), RightY => Some(TopX), TopX => None, } } } /// Color #[allow(missing_docs)] #[derive(Clone, Copy)] pub enum Color { Black, Blue, Cyan, DarkViolet, ForestGreen, Gold, Gray, Green, Magenta, Red, /// Custom RGB color Rgb(u8, u8, u8), White, Yellow, } /// Grid line #[derive(Clone, Copy)] pub enum Grid { /// Major gridlines Major, /// Minor gridlines Minor, } impl Grid { fn next(self) -> Option<Grid> { use crate::Grid::*; match self { Major => Some(Minor), Minor => None, } } } /// Line type #[allow(missing_docs)] #[derive(Clone, Copy)] pub enum LineType { Dash, Dot, DotDash, DotDotDash, /// Line made of minimally sized dots SmallDot, Solid, } /// Point type #[allow(missing_docs)] #[derive(Clone, Copy)] pub enum PointType { Circle, FilledCircle, FilledSquare, FilledTriangle, Plus, Square, Star, Triangle, X, } /// Axis scale #[allow(missing_docs)] #[derive(Clone, Copy)] pub enum Scale { Linear, Logarithmic, } /// Axis scale factor #[allow(missing_docs)] #[derive(Clone, Copy)] pub struct ScaleFactor(pub f64); /// Output terminal #[allow(missing_docs)] #[derive(Clone, Copy)] pub enum Terminal { Svg, } /// Not public version of `std::default::Default`, used to not leak default constructors into the /// public API trait Default { /// Creates `Properties` with default configuration fn default() -> Self; } /// Enums that can produce gnuplot code trait Display<S> { /// Translates the enum in gnuplot code fn display(&self) -> S; } /// Curve variant of Default trait CurveDefault<S> { /// Creates `curve::Properties` with default configuration fn default(s: S) -> Self; } /// Error bar variant of Default trait ErrorBarDefault<S> { /// Creates `errorbar::Properties` with default configuration fn default(s: S) -> Self; } /// Structs that can produce gnuplot code trait Script { /// Translates some configuration struct into gnuplot code fn script(&self) -> String; } #[derive(Clone)] struct Plot { data: Matrix, script: String, } impl Plot { fn new<S>(data: Matrix, script: &S) -> Plot where S: Script, { Plot { data, script: script.script(), } } fn data(&self) -> &Matrix { &self.data } fn script(&self) -> &str { &self.script } } /// Possible errors when parsing gnuplot's version string #[derive(Debug)] pub enum VersionError { /// The `gnuplot` command couldn't be executed Exec(io::Error), /// The `gnuplot` command returned an error message Error(String), /// The `gnuplot` command returned invalid utf-8 OutputError, /// The `gnuplot` command returned an unparseable string ParseError(String), } impl fmt::Display for VersionError { fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { match self { VersionError::Exec(err) => write!(f, "`gnuplot --version` failed: {}", err), VersionError::Error(msg) => { write!(f, "`gnuplot --version` failed with error message:\n{}", msg) } VersionError::OutputError => write!(f, "`gnuplot --version` returned invalid utf-8"), VersionError::ParseError(msg) => write!( f, "`gnuplot --version` returned an unparseable version string: {}", msg ), } } } impl ::std::error::Error for VersionError { fn description(&self) -> &str { match self { VersionError::Exec(_) => "Execution Error", VersionError::Error(_) => "Other Error", VersionError::OutputError => "Output Error", VersionError::ParseError(_) => "Parse Error", } } fn cause(&self) -> Option<&dyn ::std::error::Error> { match self { VersionError::Exec(err) => Some(err), _ => None, } } } /// Structure representing a gnuplot version number. pub struct Version { /// The major version number pub major: usize, /// The minor version number pub minor: usize, /// The patch level pub patch: String, } /// Returns `gnuplot` version pub fn version() -> Result<Version, VersionError> { let command_output = Command::new("gnuplot") .arg("--version") .output() .map_err(VersionError::Exec)?; if !command_output.status.success() { let error = String::from_utf8(command_output.stderr).map_err(|_| VersionError::OutputError)?; return Err(VersionError::Error(error)); } let output = String::from_utf8(command_output.stdout).map_err(|_| VersionError::OutputError)?; parse_version(&output).map_err(|_| VersionError::ParseError(output.clone())) } fn parse_version(version_str: &str) -> Result<Version, Option<ParseIntError>> { let mut words = version_str.split_whitespace().skip(1); let mut version = words.next().ok_or(None)?.split('.'); let major = version.next().ok_or(None)?.parse()?; let minor = version.next().ok_or(None)?.parse()?; let patchlevel = words.nth(1).ok_or(None)?.to_owned(); Ok(Version { major, minor, patch: patchlevel, }) } fn scale_factor(map: &map::axis::Map<axis::Properties>, axes: Axes) -> (f64, f64) { use crate::Axes::*; use crate::Axis::*; match axes { BottomXLeftY => ( map.get(BottomX).map_or(1., ScaleFactorTrait::scale_factor), map.get(LeftY).map_or(1., ScaleFactorTrait::scale_factor), ), BottomXRightY => ( map.get(BottomX).map_or(1., ScaleFactorTrait::scale_factor), map.get(RightY).map_or(1., ScaleFactorTrait::scale_factor), ), TopXLeftY => ( map.get(TopX).map_or(1., ScaleFactorTrait::scale_factor), map.get(LeftY).map_or(1., ScaleFactorTrait::scale_factor), ), TopXRightY => ( map.get(TopX).map_or(1., ScaleFactorTrait::scale_factor), map.get(RightY).map_or(1., ScaleFactorTrait::scale_factor), ), } } // XXX :-1: to intra-crate privacy rules /// Private trait ScaleFactorTrait { /// Private fn scale_factor(&self) -> f64; } #[cfg(test)] mod test { #[test] fn version() { if let Ok(version) = super::version() { assert!(version.major >= 4); } else { println!("Gnuplot not installed."); } } #[test] fn test_parse_version_on_valid_string() { let string = "gnuplot 5.0 patchlevel 7"; let version = super::parse_version(&string).unwrap(); assert_eq!(5, version.major); assert_eq!(0, version.minor); assert_eq!("7", &version.patch); } #[test] fn test_parse_gentoo_version() { let string = "gnuplot 5.2 patchlevel 5a (Gentoo revision r0)"; let version = super::parse_version(&string).unwrap(); assert_eq!(5, version.major); assert_eq!(2, version.minor); assert_eq!("5a", &version.patch); } #[test] fn test_parse_version_returns_error_on_invalid_strings() { let strings = [ "", "foobar", "gnuplot 50 patchlevel 7", "gnuplot 5.0 patchlevel", "gnuplot foo.bar patchlevel 7", ]; for string in &strings { assert!(super::parse_version(string).is_err()); } } }