diff options
Diffstat (limited to 'svgbob/src/optimizer.rs')
| -rw-r--r-- | svgbob/src/optimizer.rs | 198 |
1 files changed, 0 insertions, 198 deletions
diff --git a/svgbob/src/optimizer.rs b/svgbob/src/optimizer.rs deleted file mode 100644 index 2431c2b..0000000 --- a/svgbob/src/optimizer.rs +++ /dev/null @@ -1,198 +0,0 @@ -use element::Element; -use element::Feature::{BigOpenCircle, OpenCircle}; -use element::Stroke; -use loc::Loc; -use settings::Settings; - -pub struct Optimizer { - elements: Vec<Vec<Vec<Element>>>, -} - -impl Optimizer { - pub fn new(elements: Vec<Vec<Vec<Element>>>) -> Optimizer { - Optimizer { elements: elements } - } - - fn get(&self, loc: &Loc) -> Option<&Vec<Element>> { - match self.elements.get(loc.y as usize) { - Some(row) => row.get(loc.x as usize), - None => None, - } - } - - // return the reduced element and the index of the matching element on this location - fn reduce(&self, elm1: &Element, loc2: &Loc) -> Option<(Vec<Element>, usize)> { - // try all the elments of this location - if let Some(elements2) = self.get(loc2) { - if elements2.len() > 0 { - for (i, elm2) in elements2.iter().enumerate() { - // use the element that can be reduced with - if let Some(reduced) = elm1.reduce(&elm2) { - let mut new_reduced = vec![]; - new_reduced.push(reduced); - new_reduced.sort(); - new_reduced.dedup(); - return Some((new_reduced, i)); - } - } - } - } - None - } - /// trace the and try to reduce this element against the elements - /// at this location(loc), - /// returns the reduced element, the location and index of the consumed element - fn trace_elements(&self, element: &Element, loc: &Loc) -> (Vec<Element>, Vec<(Loc, usize)>) { - //trace to the right first - let right = loc.right(); - let bottom = loc.bottom(); - let bottom_right = loc.bottom_right(); - let bottom_left = loc.bottom_left(); - if let Some((all_reduced, elm_index)) = self.reduce(element, &right) { - let mut all_consumed: Vec<(Loc, usize)> = vec![]; - let mut only_reduced = vec![]; - for reduced_elm in all_reduced { - let (reduced, consumed) = self.trace_elements(&reduced_elm, &right); - all_consumed.push((right.clone(), elm_index)); - all_consumed.extend(consumed); - only_reduced = reduced; - } - (only_reduced, all_consumed) - } else if let Some((all_reduced, elm_index)) = self.reduce(element, &bottom) { - let mut all_consumed = vec![]; - let mut only_reduced = vec![]; - for reduced_elm in all_reduced { - let (reduced, consumed) = self.trace_elements(&reduced_elm, &bottom); - all_consumed.push((bottom.clone(), elm_index)); - all_consumed.extend(consumed); - only_reduced = reduced; - } - (only_reduced, all_consumed) - } else if let Some((all_reduced, elm_index)) = self.reduce(element, &bottom_right) { - let mut all_consumed = vec![]; - let mut only_reduced = vec![]; - for reduced_elm in all_reduced { - let (reduced, consumed) = self.trace_elements(&reduced_elm, &bottom_right); - all_consumed.push((bottom_right.clone(), elm_index)); - all_consumed.extend(consumed); - only_reduced = reduced; - } - (only_reduced, all_consumed) - } else if let Some((all_reduced, elm_index)) = self.reduce(element, &bottom_left) { - let mut all_consumed = vec![]; - let mut only_reduced = vec![]; - for reduced_elm in all_reduced { - let (reduced, consumed) = self.trace_elements(&reduced_elm, &bottom_left); - all_consumed.push((bottom_left.clone(), elm_index)); - all_consumed.extend(consumed); - only_reduced = reduced; - } - (only_reduced, all_consumed) - } else { - (vec![element.to_owned()], vec![]) - } - } - - /// grouped elements together that shares some end_points - fn group_elements<'e>(&self, elements: Vec<Element>) -> Vec<Vec<Element>> { - let mut consumed = vec![]; - let mut all_group = vec![]; - for (i, elm) in elements.iter().enumerate() { - if !consumed.contains(&i) { - let mut cell_group = vec![]; - cell_group.push(elm.clone()); - consumed.push(i); - for (j, elm2) in elements.iter().enumerate() { - if !consumed.contains(&j) { - if cell_group.iter().any(|e| e.shares_endpoints(&elm2)) { - cell_group.push(elm2.clone()); - consumed.push(j); - } - } - } - all_group.push(cell_group); - } - } - all_group - } - - // TODO: order the elements in such a way that - // the start -> end -> start chains nicely - pub fn optimize(&self, _settings: &Settings) -> Vec<Vec<Element>> { - let mut tracing_consumed_locs: Vec<(Loc, usize)> = vec![]; - let mut optimized = vec![]; - - for (y, line) in self.elements.iter().enumerate() { - for (x, cell) in line.iter().enumerate() { - let loc = &Loc::new(x as i32, y as i32); - for (elm_index, elm) in cell.iter().enumerate() { - if !tracing_consumed_locs.contains(&(loc.clone(), elm_index)) { - let (traced, consumed) = self.trace_elements(elm, loc); - optimized.extend(traced); - tracing_consumed_locs.extend(consumed); - } - } - } - } - optimized.sort(); - optimized.dedup(); - let arranged = self.arrange_elements(optimized); - let grouped = self.group_elements(arranged); - grouped - } - - /// arrange elements listing in the svg document - fn arrange_elements(&self, elements: Vec<Element>) -> Vec<Element> { - let mut merged = vec![]; - let mut solid_lines = vec![]; - let mut dashed_lines = vec![]; - let mut featured_circle_solid_lines = vec![]; - let mut solid_arcs = vec![]; - let mut dashed_arcs = vec![]; - let mut text = vec![]; - let mut circles = vec![]; - for elm in elements { - match elm { - Element::Circle(_, _) => { - circles.push(elm.clone()); - } - Element::Line(_, _, ref stroke, ref start_feature, ref end_feature) => { - if *start_feature == OpenCircle - || *end_feature == OpenCircle - || *start_feature == BigOpenCircle - || *end_feature == BigOpenCircle - { - featured_circle_solid_lines.push(elm.clone()) - } else { - match *stroke { - Stroke::Solid => { - solid_lines.push(elm.clone()); - } - Stroke::Dashed => { - dashed_lines.push(elm.clone()); - } - } - } - } - Element::Arc(_, _, _, _, _, ref stroke, _, _) => match *stroke { - Stroke::Solid => { - solid_arcs.push(elm.clone()); - } - Stroke::Dashed => { - dashed_arcs.push(elm.clone()); - } - }, - Element::Text(_, _) => text.push(elm.clone()), - } - } - merged.extend(solid_lines); - merged.extend(dashed_lines); - merged.extend(solid_arcs); - merged.extend(dashed_arcs); - merged.extend(text); - merged.extend(circles); - // put last to be infront of everything - merged.extend(featured_circle_solid_lines); - merged - } -} |