`6f5395e`

annotate: add tool state machine

Authored by mfwolffe <wolffemf@dukes.jmu.edu> 3 months ago

SHA: 6f5395e01973f5f085cd511744e856ae0ac95ef9
Parents: 283a088
Tree: 66b0323

1 changed file

Status	File	+	-
A	`garshot/src/annotate/state.rs`	214	0

garshot/src/annotate/state.rsadded

 +//! Annotation state machine and core types.
++
 +use gartk_core::Color;
++
 +/// Tool type for annotation.
 +#[derive(Debug, Clone, Copy, PartialEq, Eq, Default)]
 +pub enum ToolType {
 +    #[default]
 +    Arrow,
 +    Brush,
 +    Line,
 +    Rectangle,
 +    Ellipse,
 +    Text,
 +    Blur,
 +    Highlight,
 +}
++
 +impl ToolType {
 +    /// Get the keyboard shortcut for this tool.
 +    pub fn shortcut(&self) -> char {
 +        match self {
 +            ToolType::Brush => 'b',
 +            ToolType::Line => 'l',
 +            ToolType::Arrow => 'a',
 +            ToolType::Rectangle => 'r',
 +            ToolType::Ellipse => 'e',
 +            ToolType::Text => 't',
 +            ToolType::Blur => 'x',
 +            ToolType::Highlight => 'h',
 +        }
 +    }
++
 +    /// Get tool type from keyboard shortcut.
 +    pub fn from_shortcut(c: char) -> Option<Self> {
 +        match c.to_ascii_lowercase() {
 +            'b' => Some(ToolType::Brush),
 +            'l' => Some(ToolType::Line),
 +            'a' => Some(ToolType::Arrow),
 +            'r' => Some(ToolType::Rectangle),
 +            'e' => Some(ToolType::Ellipse),
 +            't' => Some(ToolType::Text),
 +            'x' => Some(ToolType::Blur),
 +            'h' => Some(ToolType::Highlight),
 +            _ => None,
 +        }
 +    }
++
 +    /// Get display name for this tool.
 +    pub fn name(&self) -> &'static str {
 +        match self {
 +            ToolType::Brush => "Brush",
 +            ToolType::Line => "Line",
 +            ToolType::Arrow => "Arrow",
 +            ToolType::Rectangle => "Rectangle",
 +            ToolType::Ellipse => "Ellipse",
 +            ToolType::Text => "Text",
 +            ToolType::Blur => "Blur",
 +            ToolType::Highlight => "Highlight",
 +        }
 +    }
++
 +    /// Get all tool types in order.
 +    pub fn all() -> &'static [ToolType] {
 +        &[
 +            ToolType::Brush,
 +            ToolType::Line,
 +            ToolType::Arrow,
 +            ToolType::Rectangle,
 +            ToolType::Ellipse,
 +            ToolType::Text,
 +            ToolType::Blur,
 +            ToolType::Highlight,
 +        ]
 +    }
 +}
++
 +/// Tool properties for drawing.
 +#[derive(Debug, Clone)]
 +pub struct ToolProperties {
 +    /// Stroke/fill color.
 +    pub color: Color,
 +    /// Line width in pixels.
 +    pub line_width: f64,
 +    /// Whether to fill shapes (vs stroke only).
 +    pub fill: bool,
 +    /// Blur radius for blur tool.
 +    pub blur_radius: usize,
 +    /// Font size for text tool.
 +    pub font_size: f64,
 +}
++
 +impl Default for ToolProperties {
 +    fn default() -> Self {
 +        Self {
 +            color: Color::new(1.0, 0.4, 0.0, 1.0), // Orange (#ff6600)
 +            line_width: 3.0,
 +            fill: false,
 +            blur_radius: 15,
 +            font_size: 16.0,
 +        }
 +    }
 +}
++
 +/// Preset colors for quick selection (keys 1-9).
 +const PRESET_COLORS: [Color; 9] = [
 +    Color::new(1.0, 0.0, 0.0, 1.0),   // 1: Red
 +    Color::new(1.0, 0.4, 0.0, 1.0),   // 2: Orange
 +    Color::new(1.0, 1.0, 0.0, 1.0),   // 3: Yellow
 +    Color::new(0.0, 1.0, 0.0, 1.0),   // 4: Green
 +    Color::new(0.0, 1.0, 1.0, 1.0),   // 5: Cyan
 +    Color::new(0.0, 0.0, 1.0, 1.0),   // 6: Blue
 +    Color::new(1.0, 0.0, 1.0, 1.0),   // 7: Magenta
 +    Color::new(1.0, 1.0, 1.0, 1.0),   // 8: White
 +    Color::new(0.0, 0.0, 0.0, 1.0),   // 9: Black
 +];
++
 +impl ToolProperties {
 +    /// Preset colors for quick selection (keys 1-9).
 +    pub fn preset_colors() -> &'static [Color] {
 +        &PRESET_COLORS
 +    }
++
 +    /// Increase line width.
 +    pub fn increase_line_width(&mut self) {
 +        self.line_width = (self.line_width + 1.0).min(50.0);
 +    }
++
 +    /// Decrease line width.
 +    pub fn decrease_line_width(&mut self) {
 +        self.line_width = (self.line_width - 1.0).max(1.0);
 +    }
 +}
++
 +/// Result of annotation mode.
 +#[derive(Debug)]
 +pub enum AnnotationResult {
 +    /// User saved the annotated image.
 +    Save {
 +        /// Annotated image data (RGBA).
 +        data: Vec<u8>,
 +        /// Image width.
 +        width: u32,
 +        /// Image height.
 +        height: u32,
 +    },
 +    /// User cancelled annotation.
 +    Cancel,
 +}
++
 +/// Main annotation state machine.
 +pub struct AnnotationState {
 +    /// Current tool.
 +    pub current_tool: ToolType,
 +    /// Tool properties.
 +    pub properties: ToolProperties,
 +    /// Whether annotation is finished.
 +    finished: bool,
 +    /// Result when finished.
 +    result: Option<AnnotationResult>,
 +}
++
 +impl AnnotationState {
 +    /// Create new annotation state.
 +    pub fn new() -> Self {
 +        Self {
 +            current_tool: ToolType::Arrow,
 +            properties: ToolProperties::default(),
 +            finished: false,
 +            result: None,
 +        }
 +    }
++
 +    /// Check if annotation mode is finished.
 +    pub fn is_finished(&self) -> bool {
 +        self.finished
 +    }
++
 +    /// Get the result (consumes the state).
 +    pub fn take_result(&mut self) -> Option<AnnotationResult> {
 +        self.result.take()
 +    }
++
 +    /// Select a tool.
 +    pub fn select_tool(&mut self, tool: ToolType) {
 +        self.current_tool = tool;
 +    }
++
 +    /// Set color from preset index (0-8).
 +    pub fn set_color_preset(&mut self, index: usize) {
 +        let colors = ToolProperties::preset_colors();
 +        if index < colors.len() {
 +            self.properties.color = colors[index];
 +        }
 +    }
++
 +    /// Mark as finished with save result.
 +    pub fn finish_save(&mut self, data: Vec<u8>, width: u32, height: u32) {
 +        self.finished = true;
 +        self.result = Some(AnnotationResult::Save { data, width, height });
 +    }
++
 +    /// Mark as finished with cancel result.
 +    pub fn finish_cancel(&mut self) {
 +        self.finished = true;
 +        self.result = Some(AnnotationResult::Cancel);
 +    }
 +}
++
 +impl Default for AnnotationState {
 +    fn default() -> Self {
 +        Self::new()
 +    }
 +}