1 files changed, 169 insertions, 0 deletions
diff --git a/src/renderer/rects.rs b/src/renderer/rects.rs
new file mode 100644
index 00000000..e066c365
--- /dev/null
+++ b/src/renderer/rects.rs
@@ -0,0 +1,169 @@
+// Copyright 2016 Joe Wilm, The Alacritty Project Contributors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+use std::collections::HashMap;
+
+use crate::term::cell::Flags;
+use crate::term::{RenderableCell, SizeInfo};
+use crate::term::color::Rgb;
+use font::Metrics;
+
+#[derive(Debug, Copy, Clone)]
+pub struct Rect<T> {
+    pub x: T,
+    pub y: T,
+    pub width: T,
+    pub height: T,
+}
+
+impl<T> Rect<T> {
+    pub fn new(x: T, y: T, width: T, height: T) -> Self {
+        Rect { x, y, width, height }
+    }
+}
+
+/// Rects for underline, strikeout and more.
+pub struct Rects<'a> {
+    inner: Vec<(Rect<f32>, Rgb)>,
+    last_starts: HashMap<Flags, Option<RenderableCell>>,
+    last_cell: Option<RenderableCell>,
+    metrics: &'a Metrics,
+    size: &'a SizeInfo,
+}
+
+impl<'a> Rects<'a> {
+    pub fn new(metrics: &'a Metrics, size: &'a SizeInfo) -> Self {
+        let mut last_starts = HashMap::new();
+        last_starts.insert(Flags::UNDERLINE, None);
+        last_starts.insert(Flags::STRIKEOUT, None);
+
+        Self {
+            inner: Vec::new(),
+            last_cell: None,
+            last_starts,
+            metrics,
+            size,
+        }
+    }
+
+    /// Convert the stored rects to rectangles for the renderer.
+    pub fn rects(mut self) -> Vec<(Rect<f32>, Rgb)> {
+        // If there's still a line pending, draw it until the last cell
+        for (flag, start_cell) in self.last_starts.iter_mut() {
+            if let Some(start) = start_cell {
+                self.inner.push(
+                    create_rect(
+                        &start,
+                        &self.last_cell.unwrap(),
+                        *flag,
+                        &self.metrics,
+                        &self.size,
+                    )
+                );
+            }
+        }
+
+        self.inner
+    }
+
+    /// Update the stored lines with the next cell info.
+    pub fn update_lines(&mut self, cell: &RenderableCell) {
+        for (flag, start_cell) in self.last_starts.iter_mut() {
+            let flag = *flag;
+            *start_cell = match *start_cell {
+                // Check for end if line is present
+                Some(ref mut start) => {
+                    let last_cell = self.last_cell.unwrap();
+
+                    // No change in line
+                    if cell.line == start.line
+                        && cell.flags.contains(flag)
+                        && cell.fg == start.fg
+                        && cell.column == last_cell.column + 1
+                    {
+                        continue;
+                    }
+
+                    self.inner.push(create_rect(
+                        &start,
+                        &last_cell,
+                        flag,
+                        &self.metrics,
+                        &self.size,
+                    ));
+
+                    // Start a new line if the flag is present
+                    if cell.flags.contains(flag) {
+                        Some(*cell)
+                    } else {
+                        None
+                    }
+                }
+                // Check for new start of line
+                None => if cell.flags.contains(flag) {
+                    Some(*cell)
+                } else {
+                    None
+                },
+            };
+        }
+
+        self.last_cell = Some(*cell);
+    }
+
+    // Add a rectangle
+    pub fn push(&mut self, rect: Rect<f32>, color: Rgb) {
+        self.inner.push((rect, color));
+    }
+}
+
+/// Create a rectangle that starts on the left of `start` and ends on the right
+/// of `end`, based on the given flag and size metrics.
+fn create_rect(
+    start: &RenderableCell,
+    end: &RenderableCell,
+    flag: Flags,
+    metrics: &Metrics,
+    size: &SizeInfo,
+) -> (Rect<f32>, Rgb) {
+    let start_x = start.column.0 as f32 * size.cell_width;
+    let end_x = (end.column.0 + 1) as f32 * size.cell_width;
+    let width = end_x - start_x;
+
+    let (position, mut height) = match flag {
+        Flags::UNDERLINE => (metrics.underline_position, metrics.underline_thickness),
+        Flags::STRIKEOUT => (metrics.strikeout_position, metrics.strikeout_thickness),
+        _ => unimplemented!("Invalid flag for cell line drawing specified"),
+    };
+
+    // Make sure lines are always visible
+    height = height.max(1.);
+
+    let cell_bottom = (start.line.0 as f32 + 1.) * size.cell_height;
+    let baseline = cell_bottom + metrics.descent;
+
+    let mut y = baseline - position - height / 2.;
+    let max_y = cell_bottom - height;
+    if y > max_y {
+        y = max_y;
+    }
+
+    let rect = Rect::new(
+        start_x + size.padding_x,
+        y.round() + size.padding_y,
+        width,
+        height.round(),
+    );
+
+    (rect, start.fg)
+}