Contents
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100
/*!
The candlestick element, which showing the high/low/open/close price
*/
use std::cmp::Ordering;
use crate::element::{Drawable, PointCollection};
use crate::style::ShapeStyle;
use plotters_backend::{BackendCoord, DrawingBackend, DrawingErrorKind};
/// The candlestick data point element
pub struct CandleStick<X, Y: PartialOrd> {
style: ShapeStyle,
width: u32,
points: [(X, Y); 4],
}
impl<X: Clone, Y: PartialOrd> CandleStick<X, Y> {
/// Create a new candlestick element, which requires the Y coordinate can be compared
///
/// - `x`: The x coordinate
/// - `open`: The open value
/// - `high`: The high value
/// - `low`: The low value
/// - `close`: The close value
/// - `gain_style`: The style for gain
/// - `loss_style`: The style for loss
/// - `width`: The width
/// - **returns** The newly created candlestick element
///
/// ```rust
/// use chrono::prelude::*;
/// use plotters::prelude::*;
///
/// let candlestick = CandleStick::new(Local::now(), 130.0600, 131.3700, 128.8300, 129.1500, &GREEN, &RED, 15);
/// ```
#[allow(clippy::too_many_arguments)]
pub fn new<GS: Into<ShapeStyle>, LS: Into<ShapeStyle>>(
x: X,
open: Y,
high: Y,
low: Y,
close: Y,
gain_style: GS,
loss_style: LS,
width: u32,
) -> Self {
Self {
style: match open.partial_cmp(&close) {
Some(Ordering::Less) => gain_style.into(),
_ => loss_style.into(),
},
width,
points: [
(x.clone(), open),
(x.clone(), high),
(x.clone(), low),
(x, close),
],
}
}
}
impl<'a, X: 'a, Y: PartialOrd + 'a> PointCollection<'a, (X, Y)> for &'a CandleStick<X, Y> {
type Point = &'a (X, Y);
type IntoIter = &'a [(X, Y)];
fn point_iter(self) -> &'a [(X, Y)] {
&self.points
}
}
impl<X, Y: PartialOrd, DB: DrawingBackend> Drawable<DB> for CandleStick<X, Y> {
fn draw<I: Iterator<Item = BackendCoord>>(
&self,
points: I,
backend: &mut DB,
_: (u32, u32),
) -> Result<(), DrawingErrorKind<DB::ErrorType>> {
let mut points: Vec<_> = points.take(4).collect();
if points.len() == 4 {
let fill = self.style.filled;
if points[0].1 > points[3].1 {
points.swap(0, 3);
}
let (l, r) = (
self.width as i32 / 2,
self.width as i32 - self.width as i32 / 2,
);
backend.draw_line(points[0], points[1], &self.style)?;
backend.draw_line(points[2], points[3], &self.style)?;
points[0].0 -= l;
points[3].0 += r;
backend.draw_rect(points[0], points[3], &self.style, fill)?;
}
Ok(())
}
}