2018-12-11 19:38:06 -05:00
|
|
|
// pathfinder/utils/tile-svg/main.rs
|
|
|
|
//
|
|
|
|
// Copyright © 2018 The Pathfinder Project Developers.
|
|
|
|
//
|
|
|
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
|
|
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
|
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
|
|
|
// option. This file may not be copied, modified, or distributed
|
|
|
|
// except according to those terms.
|
|
|
|
|
|
|
|
#[macro_use]
|
|
|
|
extern crate bitflags;
|
|
|
|
|
2018-12-12 17:55:53 -05:00
|
|
|
#[cfg(test)]
|
|
|
|
extern crate quickcheck;
|
|
|
|
#[cfg(test)]
|
|
|
|
extern crate rand;
|
|
|
|
|
2018-12-11 19:38:06 -05:00
|
|
|
use euclid::{Point2D, Transform2D};
|
|
|
|
use quick_xml::Reader;
|
|
|
|
use quick_xml::events::Event;
|
|
|
|
use std::env;
|
|
|
|
use std::mem;
|
2018-12-12 17:55:53 -05:00
|
|
|
use std::ops::Range;
|
2018-12-11 19:38:06 -05:00
|
|
|
use std::path::{Path, PathBuf};
|
|
|
|
use std::str::FromStr;
|
|
|
|
use svgtypes::{Color as SvgColor, PathParser, PathSegment as SvgPathSegment, TransformListParser};
|
|
|
|
use svgtypes::{TransformListToken};
|
|
|
|
|
|
|
|
#[derive(Default)]
|
|
|
|
struct GroupStyle {
|
|
|
|
fill_color: Option<SvgColor>,
|
|
|
|
stroke_width: Option<f32>,
|
|
|
|
stroke_color: Option<SvgColor>,
|
|
|
|
transform: Option<Transform2D<f32>>,
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Debug)]
|
|
|
|
struct ComputedStyle {
|
|
|
|
fill_color: Option<SvgColor>,
|
|
|
|
stroke_width: f32,
|
|
|
|
stroke_color: Option<SvgColor>,
|
|
|
|
transform: Transform2D<f32>,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl ComputedStyle {
|
|
|
|
fn new() -> ComputedStyle {
|
|
|
|
ComputedStyle {
|
|
|
|
fill_color: None,
|
|
|
|
stroke_width: 1.0,
|
|
|
|
stroke_color: None,
|
|
|
|
transform: Transform2D::identity(),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
fn main() {
|
|
|
|
let path = PathBuf::from(env::args().skip(1).next().unwrap());
|
|
|
|
let scene = Scene::from_path(&path);
|
|
|
|
println!("{:#?}", scene);
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Debug)]
|
|
|
|
struct Scene {
|
|
|
|
objects: Vec<PathObject>,
|
|
|
|
styles: Vec<ComputedStyle>,
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Debug)]
|
|
|
|
struct PathObject {
|
|
|
|
outline: Outline,
|
|
|
|
style: StyleId,
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Clone, Copy, PartialEq, Debug)]
|
|
|
|
struct StyleId(u32);
|
|
|
|
|
|
|
|
impl Scene {
|
|
|
|
fn new() -> Scene {
|
|
|
|
Scene {
|
|
|
|
objects: vec![],
|
|
|
|
styles: vec![],
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
fn from_path(path: &Path) -> Scene {
|
|
|
|
let mut reader = Reader::from_file(&path).unwrap();
|
|
|
|
|
|
|
|
let mut xml_buffer = vec![];
|
|
|
|
let mut group_styles = vec![];
|
|
|
|
let mut style = None;
|
|
|
|
|
|
|
|
let mut scene = Scene::new();
|
|
|
|
|
|
|
|
loop {
|
|
|
|
match reader.read_event(&mut xml_buffer) {
|
|
|
|
Ok(Event::Start(ref event)) |
|
|
|
|
Ok(Event::Empty(ref event)) if event.name() == b"path" => {
|
|
|
|
let attributes = event.attributes();
|
|
|
|
for attribute in attributes {
|
|
|
|
let attribute = attribute.unwrap();
|
|
|
|
if attribute.key != b"d" {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
let value = reader.decode(&attribute.value);
|
|
|
|
let style = scene.ensure_style(&mut style, &mut group_styles);
|
|
|
|
let path_parser = PathParser::from(&*value);
|
2018-12-11 19:53:57 -05:00
|
|
|
let outline =
|
|
|
|
Outline::from_svg_path_segments(path_parser, scene.get_style(style));
|
2018-12-11 19:38:06 -05:00
|
|
|
scene.objects.push(PathObject::new(outline, style));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
Ok(Event::Start(ref event)) if event.name() == b"g" => {
|
|
|
|
let mut group_style = GroupStyle::default();
|
|
|
|
let attributes = event.attributes();
|
|
|
|
for attribute in attributes {
|
|
|
|
let attribute = attribute.unwrap();
|
|
|
|
match attribute.key {
|
|
|
|
b"fill" => {
|
|
|
|
let value = reader.decode(&attribute.value);
|
|
|
|
if let Ok(color) = SvgColor::from_str(&value) {
|
|
|
|
group_style.fill_color = Some(color)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
b"stroke" => {
|
|
|
|
let value = reader.decode(&attribute.value);
|
|
|
|
if let Ok(color) = SvgColor::from_str(&value) {
|
|
|
|
group_style.stroke_color = Some(color)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
b"transform" => {
|
|
|
|
let value = reader.decode(&attribute.value);
|
|
|
|
let mut current_transform = Transform2D::identity();
|
|
|
|
let transform_list_parser = TransformListParser::from(&*value);
|
|
|
|
for transform in transform_list_parser {
|
|
|
|
match transform {
|
|
|
|
Ok(TransformListToken::Matrix { a, b, c, d, e, f }) => {
|
|
|
|
let transform: Transform2D<f32> =
|
|
|
|
Transform2D::row_major(a, b, c, d, e, f).cast();
|
|
|
|
current_transform = current_transform.pre_mul(&transform)
|
|
|
|
}
|
|
|
|
_ => {}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
group_style.transform = Some(current_transform);
|
|
|
|
}
|
|
|
|
b"stroke-width" => {
|
|
|
|
if let Ok(width) = reader.decode(&attribute.value).parse() {
|
|
|
|
group_style.stroke_width = Some(width)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
_ => {}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
group_styles.push(group_style);
|
|
|
|
style = None;
|
|
|
|
}
|
|
|
|
Ok(Event::Eof) | Err(_) => break,
|
|
|
|
Ok(_) => {}
|
|
|
|
}
|
|
|
|
xml_buffer.clear();
|
|
|
|
}
|
|
|
|
|
|
|
|
return scene;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
fn ensure_style(&mut self, current_style: &mut Option<StyleId>, group_styles: &[GroupStyle])
|
|
|
|
-> StyleId {
|
|
|
|
if let Some(current_style) = *current_style {
|
|
|
|
return current_style
|
|
|
|
}
|
|
|
|
|
|
|
|
let mut computed_style = ComputedStyle::new();
|
|
|
|
for group_style in group_styles {
|
|
|
|
if let Some(fill_color) = group_style.fill_color {
|
|
|
|
computed_style.fill_color = Some(fill_color)
|
|
|
|
}
|
|
|
|
if let Some(stroke_width) = group_style.stroke_width {
|
|
|
|
computed_style.stroke_width = stroke_width
|
|
|
|
}
|
|
|
|
if let Some(stroke_color) = group_style.stroke_color {
|
|
|
|
computed_style.stroke_color = Some(stroke_color)
|
|
|
|
}
|
|
|
|
if let Some(transform) = group_style.transform {
|
|
|
|
computed_style.transform = computed_style.transform.pre_mul(&transform)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
let id = StyleId(self.styles.len() as u32);
|
|
|
|
self.styles.push(computed_style);
|
|
|
|
id
|
|
|
|
}
|
2018-12-11 19:53:57 -05:00
|
|
|
|
|
|
|
fn get_style(&self, style: StyleId) -> &ComputedStyle {
|
|
|
|
&self.styles[style.0 as usize]
|
|
|
|
}
|
2018-12-11 19:38:06 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
impl PathObject {
|
|
|
|
fn new(outline: Outline, style: StyleId) -> PathObject {
|
|
|
|
PathObject {
|
|
|
|
outline,
|
|
|
|
style,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Outlines
|
|
|
|
|
|
|
|
#[derive(Debug)]
|
|
|
|
struct Outline {
|
|
|
|
contours: Vec<Contour>,
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Debug)]
|
|
|
|
struct Contour {
|
|
|
|
points: Vec<Point2D<f32>>,
|
|
|
|
flags: Vec<PointFlags>,
|
|
|
|
}
|
|
|
|
|
|
|
|
bitflags! {
|
|
|
|
struct PointFlags: u8 {
|
|
|
|
const CONTROL_POINT_0 = 0x01;
|
|
|
|
const CONTROL_POINT_1 = 0x02;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Outline {
|
|
|
|
fn new() -> Outline {
|
|
|
|
Outline {
|
|
|
|
contours: vec![],
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-12-11 19:53:57 -05:00
|
|
|
fn from_svg_path_segments<I>(segments: I, style: &ComputedStyle) -> Outline
|
|
|
|
where I: Iterator<Item = SvgPathSegment> {
|
2018-12-11 19:38:06 -05:00
|
|
|
let mut outline = Outline::new();
|
|
|
|
let mut current_contour = Contour::new();
|
|
|
|
let (mut first_point_in_path, mut last_ctrl_point, mut last_point) = (None, None, None);
|
|
|
|
for segment in segments {
|
|
|
|
match segment {
|
|
|
|
SvgPathSegment::MoveTo { abs, x, y } => {
|
|
|
|
if !current_contour.is_empty() {
|
|
|
|
outline.contours.push(mem::replace(&mut current_contour, Contour::new()))
|
|
|
|
}
|
|
|
|
let to = compute_point(x, y, abs, &last_point);
|
|
|
|
first_point_in_path = Some(to);
|
|
|
|
last_point = Some(to);
|
|
|
|
last_ctrl_point = None;
|
2018-12-11 19:53:57 -05:00
|
|
|
current_contour.push_transformed_point(&to,
|
|
|
|
PointFlags::empty(),
|
|
|
|
&style.transform);
|
2018-12-11 19:38:06 -05:00
|
|
|
}
|
|
|
|
SvgPathSegment::LineTo { abs, x, y } => {
|
|
|
|
let to = compute_point(x, y, abs, &last_point);
|
|
|
|
last_point = Some(to);
|
|
|
|
last_ctrl_point = None;
|
2018-12-11 19:53:57 -05:00
|
|
|
current_contour.push_transformed_point(&to,
|
|
|
|
PointFlags::empty(),
|
|
|
|
&style.transform);
|
2018-12-11 19:38:06 -05:00
|
|
|
}
|
|
|
|
SvgPathSegment::HorizontalLineTo { abs, x } => {
|
|
|
|
let to = Point2D::new(compute_point(x, 0.0, abs, &last_point).x,
|
|
|
|
last_point.unwrap_or(Point2D::zero()).y);
|
|
|
|
last_point = Some(to);
|
|
|
|
last_ctrl_point = None;
|
2018-12-11 19:53:57 -05:00
|
|
|
current_contour.push_transformed_point(&to,
|
|
|
|
PointFlags::empty(),
|
|
|
|
&style.transform);
|
2018-12-11 19:38:06 -05:00
|
|
|
}
|
|
|
|
SvgPathSegment::VerticalLineTo { abs, y } => {
|
|
|
|
let to = Point2D::new(last_point.unwrap_or(Point2D::zero()).x,
|
|
|
|
compute_point(0.0, y, abs, &last_point).y);
|
|
|
|
last_point = Some(to);
|
|
|
|
last_ctrl_point = None;
|
2018-12-11 19:53:57 -05:00
|
|
|
current_contour.push_transformed_point(&to,
|
|
|
|
PointFlags::empty(),
|
|
|
|
&style.transform);
|
2018-12-11 19:38:06 -05:00
|
|
|
}
|
|
|
|
SvgPathSegment::Quadratic { abs, x1, y1, x, y } => {
|
|
|
|
let ctrl = compute_point(x1, y1, abs, &last_point);
|
|
|
|
last_ctrl_point = Some(ctrl);
|
2018-12-11 19:53:57 -05:00
|
|
|
let to = compute_point(x, y, abs, &last_point);
|
2018-12-11 19:38:06 -05:00
|
|
|
last_point = Some(to);
|
2018-12-11 19:53:57 -05:00
|
|
|
current_contour.push_transformed_point(&ctrl,
|
|
|
|
PointFlags::CONTROL_POINT_0,
|
|
|
|
&style.transform);
|
|
|
|
current_contour.push_transformed_point(&to,
|
|
|
|
PointFlags::empty(),
|
|
|
|
&style.transform);
|
2018-12-11 19:38:06 -05:00
|
|
|
}
|
|
|
|
SvgPathSegment::SmoothQuadratic { abs, x, y } => {
|
|
|
|
let ctrl = last_point.unwrap_or(Point2D::zero()) +
|
|
|
|
(last_point.unwrap_or(Point2D::zero()) -
|
|
|
|
last_ctrl_point.unwrap_or(Point2D::zero()));
|
|
|
|
last_ctrl_point = Some(ctrl);
|
2018-12-11 19:53:57 -05:00
|
|
|
let to = compute_point(x, y, abs, &last_point);
|
2018-12-11 19:38:06 -05:00
|
|
|
last_point = Some(to);
|
2018-12-11 19:53:57 -05:00
|
|
|
current_contour.push_transformed_point(&ctrl,
|
|
|
|
PointFlags::CONTROL_POINT_0,
|
|
|
|
&style.transform);
|
|
|
|
current_contour.push_transformed_point(&to,
|
|
|
|
PointFlags::empty(),
|
|
|
|
&style.transform);
|
2018-12-11 19:38:06 -05:00
|
|
|
}
|
|
|
|
SvgPathSegment::CurveTo { abs, x1, y1, x2, y2, x, y } => {
|
|
|
|
let ctrl0 = compute_point(x1, y1, abs, &last_point);
|
2018-12-11 19:53:57 -05:00
|
|
|
let ctrl1 = compute_point(x2, y2, abs, &last_point);
|
2018-12-11 19:38:06 -05:00
|
|
|
last_ctrl_point = Some(ctrl1);
|
2018-12-11 19:53:57 -05:00
|
|
|
let to = compute_point(x, y, abs, &last_point);
|
2018-12-11 19:38:06 -05:00
|
|
|
last_point = Some(to);
|
2018-12-11 19:53:57 -05:00
|
|
|
current_contour.push_transformed_point(&ctrl0,
|
|
|
|
PointFlags::CONTROL_POINT_0,
|
|
|
|
&style.transform);
|
|
|
|
current_contour.push_transformed_point(&ctrl1,
|
|
|
|
PointFlags::CONTROL_POINT_1,
|
|
|
|
&style.transform);
|
|
|
|
current_contour.push_transformed_point(&to,
|
|
|
|
PointFlags::empty(),
|
|
|
|
&style.transform);
|
2018-12-11 19:38:06 -05:00
|
|
|
}
|
|
|
|
SvgPathSegment::SmoothCurveTo { abs, x2, y2, x, y } => {
|
|
|
|
let ctrl0 = last_point.unwrap_or(Point2D::zero()) +
|
|
|
|
(last_point.unwrap_or(Point2D::zero()) -
|
|
|
|
last_ctrl_point.unwrap_or(Point2D::zero()));
|
2018-12-11 19:53:57 -05:00
|
|
|
let ctrl1 = compute_point(x2, y2, abs, &last_point);
|
2018-12-11 19:38:06 -05:00
|
|
|
last_ctrl_point = Some(ctrl1);
|
2018-12-11 19:53:57 -05:00
|
|
|
let to = compute_point(x, y, abs, &last_point);
|
2018-12-11 19:38:06 -05:00
|
|
|
last_point = Some(to);
|
2018-12-11 19:53:57 -05:00
|
|
|
current_contour.push_transformed_point(&ctrl0,
|
|
|
|
PointFlags::CONTROL_POINT_0,
|
|
|
|
&style.transform);
|
|
|
|
current_contour.push_transformed_point(&ctrl1,
|
|
|
|
PointFlags::CONTROL_POINT_1,
|
|
|
|
&style.transform);
|
|
|
|
current_contour.push_transformed_point(&to,
|
|
|
|
PointFlags::empty(),
|
|
|
|
&style.transform);
|
2018-12-11 19:38:06 -05:00
|
|
|
}
|
|
|
|
SvgPathSegment::ClosePath { abs: _ } => {
|
|
|
|
if !current_contour.is_empty() {
|
|
|
|
outline.contours.push(mem::replace(&mut current_contour, Contour::new()));
|
|
|
|
last_point = first_point_in_path;
|
|
|
|
last_ctrl_point = None;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
SvgPathSegment::EllipticalArc { .. } => unimplemented!("arcs"),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if !current_contour.is_empty() {
|
|
|
|
outline.contours.push(current_contour)
|
|
|
|
}
|
|
|
|
return outline;
|
|
|
|
|
|
|
|
fn compute_point(x: f64, y: f64, abs: bool, last_point: &Option<Point2D<f32>>)
|
|
|
|
-> Point2D<f32> {
|
|
|
|
let point = Point2D::new(x, y).to_f32();
|
|
|
|
match *last_point {
|
|
|
|
Some(last_point) if !abs => last_point + point.to_vector(),
|
|
|
|
_ => point,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Contour {
|
|
|
|
fn new() -> Contour {
|
|
|
|
Contour {
|
|
|
|
points: vec![],
|
|
|
|
flags: vec![],
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
fn is_empty(&self) -> bool {
|
|
|
|
self.points.is_empty()
|
|
|
|
}
|
2018-12-11 19:53:57 -05:00
|
|
|
|
|
|
|
fn push_transformed_point(&mut self,
|
|
|
|
point: &Point2D<f32>,
|
|
|
|
flags: PointFlags,
|
|
|
|
transform: &Transform2D<f32>) {
|
|
|
|
self.points.push(transform.transform_point(point));
|
|
|
|
self.flags.push(flags);
|
|
|
|
}
|
2018-12-11 19:38:06 -05:00
|
|
|
}
|
2018-12-12 17:55:53 -05:00
|
|
|
|
|
|
|
// Tiling
|
|
|
|
|
|
|
|
struct Tiler {
|
|
|
|
outline: Outline,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Tiler {
|
|
|
|
fn from_outline(outline: Outline) -> Tiler {
|
|
|
|
Tiler {
|
|
|
|
outline,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Intervals
|
|
|
|
|
|
|
|
#[derive(Debug)]
|
|
|
|
struct Intervals {
|
|
|
|
ranges: Vec<IntervalRange>,
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Clone, Copy, Debug)]
|
|
|
|
struct IntervalRange {
|
|
|
|
start: f32,
|
|
|
|
end: f32,
|
|
|
|
winding: f32,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Intervals {
|
|
|
|
fn new(end: f32) -> Intervals {
|
|
|
|
Intervals {
|
|
|
|
ranges: vec![IntervalRange::new(0.0, end, 0.0)],
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
fn add(&mut self, range: IntervalRange) {
|
|
|
|
self.split_at(range.start);
|
|
|
|
self.split_at(range.end);
|
|
|
|
|
|
|
|
// Find bracketing range.
|
|
|
|
let mut start_index = 0;
|
|
|
|
while range.start < self.ranges[start_index].start {
|
|
|
|
start_index += 1
|
|
|
|
}
|
|
|
|
let mut end_index = start_index;
|
|
|
|
while range.end < self.ranges[end_index].end {
|
|
|
|
end_index += 1
|
|
|
|
}
|
|
|
|
|
|
|
|
// Adjust winding numbers.
|
|
|
|
for existing_range in &mut self.ranges[start_index..(end_index + 1)] {
|
|
|
|
existing_range.winding += range.winding
|
|
|
|
}
|
|
|
|
|
|
|
|
self.merge_adjacent();
|
|
|
|
}
|
|
|
|
|
|
|
|
fn clear(&mut self) {
|
|
|
|
let end = self.ranges.last().unwrap().end;
|
|
|
|
self.ranges.truncate(1);
|
|
|
|
self.ranges[0] = IntervalRange::new(0.0, end, 0.0);
|
|
|
|
}
|
|
|
|
|
|
|
|
fn split_at(&mut self, value: f32) {
|
|
|
|
let mut range_index = 0;
|
|
|
|
while range_index < self.ranges.len() {
|
|
|
|
let IntervalRange {
|
|
|
|
start: old_start,
|
|
|
|
end: old_end,
|
|
|
|
winding,
|
|
|
|
} = self.ranges[range_index];
|
|
|
|
if value < old_start || value > old_end {
|
|
|
|
range_index += 1;
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
|
|
|
|
self.ranges[range_index] = IntervalRange::new(old_start, value, winding);
|
|
|
|
self.ranges.insert(range_index + 1, IntervalRange::new(value, old_end, winding));
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
fn merge_adjacent(&mut self) {
|
|
|
|
let mut dest_range_index = 0;
|
|
|
|
let mut current_range = self.ranges[0];
|
|
|
|
for src_range_index in 1..self.ranges.len() {
|
|
|
|
if self.ranges[src_range_index].winding == current_range.winding {
|
|
|
|
current_range.end = self.ranges[src_range_index].end
|
|
|
|
} else {
|
|
|
|
self.ranges[dest_range_index] = current_range;
|
|
|
|
dest_range_index += 1;
|
|
|
|
current_range = self.ranges[src_range_index];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
self.ranges[dest_range_index] = current_range;
|
|
|
|
dest_range_index += 1;
|
|
|
|
self.ranges.truncate(dest_range_index);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl IntervalRange {
|
|
|
|
fn new(start: f32, end: f32, winding: f32) -> IntervalRange {
|
|
|
|
IntervalRange {
|
|
|
|
start,
|
|
|
|
end,
|
|
|
|
winding,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
fn contains(&self, value: f32) -> bool {
|
|
|
|
value >= self.start && value < self.end
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(test)]
|
|
|
|
mod test {
|
|
|
|
use crate::{IntervalRange, Intervals};
|
|
|
|
use quickcheck::{self, Arbitrary, Gen};
|
|
|
|
use rand::Rng;
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_intervals() {
|
|
|
|
quickcheck::quickcheck(prop_intervals as fn(Spec) -> bool);
|
|
|
|
|
|
|
|
fn prop_intervals(spec: Spec) -> bool {
|
|
|
|
let mut intervals = Intervals::new(spec.end);
|
|
|
|
for range in spec.ranges {
|
|
|
|
intervals.add(range);
|
|
|
|
}
|
|
|
|
|
|
|
|
assert!(intervals.ranges.len() > 0);
|
|
|
|
assert_eq!(intervals.ranges[0].start, 0.0);
|
|
|
|
assert_eq!(intervals.ranges.last().unwrap().end, spec.end);
|
|
|
|
for prev_index in 0..(intervals.ranges.len() - 1) {
|
|
|
|
let next_index = prev_index + 1;
|
|
|
|
assert_eq!(intervals.ranges[prev_index].end, intervals.ranges[next_index].start);
|
|
|
|
assert_ne!(intervals.ranges[prev_index].winding,
|
|
|
|
intervals.ranges[next_index].winding);
|
|
|
|
}
|
|
|
|
|
|
|
|
true
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Clone, Debug)]
|
|
|
|
struct Spec {
|
|
|
|
end: f32,
|
|
|
|
ranges: Vec<IntervalRange>,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Arbitrary for Spec {
|
|
|
|
fn arbitrary<G>(g: &mut G) -> Spec where G: Gen {
|
|
|
|
const EPSILON: f32 = 0.0001;
|
|
|
|
|
|
|
|
let size = g.size();
|
|
|
|
let end = g.gen_range(EPSILON, size as f32);
|
|
|
|
|
|
|
|
let mut ranges = vec![];
|
|
|
|
let range_count = g.gen_range(0, size);
|
|
|
|
for _ in 0..range_count {
|
|
|
|
let (a, b) = (g.gen_range(0.0, end), g.gen_range(0.0, end));
|
|
|
|
let winding = g.gen_range(-(size as i32), size as i32) as f32;
|
|
|
|
ranges.push(IntervalRange::new(f32::min(a, b), f32::max(a, b), winding));
|
|
|
|
}
|
|
|
|
|
|
|
|
Spec {
|
|
|
|
end,
|
|
|
|
ranges,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|