servo/components/script/dom/canvaspattern.rs

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/. */

use canvas_traits::canvas::{FillOrStrokeStyle, RepetitionStyle, SurfaceStyle};
use dom_struct::dom_struct;
use euclid::default::{Size2D, Transform2D};

use crate::dom::bindings::cell::DomRefCell;
use crate::dom::bindings::codegen::Bindings::CanvasRenderingContext2DBinding::CanvasPatternMethods;
use crate::dom::bindings::codegen::Bindings::DOMMatrixBinding::DOMMatrix2DInit;
use crate::dom::bindings::error::ErrorResult;
use crate::dom::bindings::reflector::{Reflector, reflect_dom_object};
use crate::dom::bindings::root::DomRoot;
use crate::dom::canvasgradient::ToFillOrStrokeStyle;
use crate::dom::dommatrixreadonly::dommatrix2dinit_to_matrix;
use crate::dom::globalscope::GlobalScope;
use crate::script_runtime::CanGc;

/// <https://html.spec.whatwg.org/multipage/#canvaspattern>
#[dom_struct]
pub(crate) struct CanvasPattern {
    reflector_: Reflector,
    surface_data: Vec<u8>,
    #[no_trace]
    surface_size: Size2D<u32>,
    repeat_x: bool,
    repeat_y: bool,
    #[no_trace]
    transform: DomRefCell<Transform2D<f32>>,
    origin_clean: bool,
}

impl CanvasPattern {
    fn new_inherited(
        surface_data: Vec<u8>,
        surface_size: Size2D<u32>,
        repeat: RepetitionStyle,
        origin_clean: bool,
    ) -> CanvasPattern {
        let (x, y) = match repeat {
            RepetitionStyle::Repeat => (true, true),
            RepetitionStyle::RepeatX => (true, false),
            RepetitionStyle::RepeatY => (false, true),
            RepetitionStyle::NoRepeat => (false, false),
        };

        CanvasPattern {
            reflector_: Reflector::new(),
            surface_data,
            surface_size,
            repeat_x: x,
            repeat_y: y,
            transform: DomRefCell::new(Transform2D::identity()),
            origin_clean,
        }
    }
    pub(crate) fn new(
        global: &GlobalScope,
        surface_data: Vec<u8>,
        surface_size: Size2D<u32>,
        repeat: RepetitionStyle,
        origin_clean: bool,
        can_gc: CanGc,
    ) -> DomRoot<CanvasPattern> {
        reflect_dom_object(
            Box::new(CanvasPattern::new_inherited(
                surface_data,
                surface_size,
                repeat,
                origin_clean,
            )),
            global,
            can_gc,
        )
    }
    pub(crate) fn origin_is_clean(&self) -> bool {
        self.origin_clean
    }
}

impl CanvasPatternMethods<crate::DomTypeHolder> for CanvasPattern {
    /// <https://html.spec.whatwg.org/multipage/#dom-canvaspattern-settransform>
    fn SetTransform(&self, transform: &DOMMatrix2DInit) -> ErrorResult {
        // Step 1. Let matrix be the result of creating a DOMMatrix from the 2D
        // dictionary transform.
        let matrix = dommatrix2dinit_to_matrix(transform)?;

        // Step 2. If one or more of matrix's m11 element, m12 element, m21
        // element, m22 element, m41 element, or m42 element are infinite or
        // NaN, then return.
        if !matrix.m11.is_finite() ||
            !matrix.m12.is_finite() ||
            !matrix.m21.is_finite() ||
            !matrix.m22.is_finite() ||
            !matrix.m41.is_finite() ||
            !matrix.m42.is_finite()
        {
            return Ok(());
        }

        // Step 3. Reset the pattern's transformation matrix to matrix.
        *self.transform.borrow_mut() = Transform2D::new(
            matrix.m11 as f32,
            matrix.m12 as f32,
            matrix.m21 as f32,
            matrix.m22 as f32,
            matrix.m41 as f32,
            matrix.m42 as f32,
        );

        Ok(())
    }
}

impl ToFillOrStrokeStyle for &CanvasPattern {
    fn to_fill_or_stroke_style(self) -> FillOrStrokeStyle {
        FillOrStrokeStyle::Surface(SurfaceStyle::new(
            self.surface_data.clone(),
            self.surface_size,
            self.repeat_x,
            self.repeat_y,
            *self.transform.borrow(),
        ))
    }
}