diff --git a/Userland/Libraries/LibWeb/CSS/Parser/Parser.cpp b/Userland/Libraries/LibWeb/CSS/Parser/Parser.cpp index b2aa565aa3..3de898abe2 100644 --- a/Userland/Libraries/LibWeb/CSS/Parser/Parser.cpp +++ b/Userland/Libraries/LibWeb/CSS/Parser/Parser.cpp @@ -3738,6 +3738,46 @@ ErrorOr> Parser::parse_atan_function(Function const& fun return TRY(AtanCalculationNode::create(calculation_node.release_nonnull())); } +ErrorOr> Parser::parse_atan2_function(Function const& function) +{ + TokenStream stream { function.values() }; + auto parameters = parse_a_comma_separated_list_of_component_values(stream); + + if (parameters.size() != 2) { + dbgln_if(CSS_PARSER_DEBUG, "atan2() must have exactly two parameter"sv); + return nullptr; + } + + auto node_a = TRY(parse_a_calculation(parameters[0])); + auto node_b = TRY(parse_a_calculation(parameters[1])); + + if (!node_a || !node_b) { + dbgln_if(CSS_PARSER_DEBUG, "atan2() parameters must be valid calculations"sv); + return nullptr; + } + + auto node_a_maybe_parameter_type = node_a->resolved_type(); + if (!node_a_maybe_parameter_type.has_value()) { + dbgln_if(CSS_PARSER_DEBUG, "Failed to resolve type for atan() parameter 1"sv); + return nullptr; + } + + auto node_b_maybe_parameter_type = node_b->resolved_type(); + if (!node_b_maybe_parameter_type.has_value()) { + dbgln_if(CSS_PARSER_DEBUG, "Failed to resolve type for atan() parameter 2"sv); + return nullptr; + } + + auto node_a_resolved_type = node_a_maybe_parameter_type.value(); + auto node_b_resolved_type = node_b_maybe_parameter_type.value(); + if (node_a_resolved_type != node_b_resolved_type) { + dbgln_if(CSS_PARSER_DEBUG, "atan2() parameters must be of the same type"sv); + return nullptr; + } + + return TRY(Atan2CalculationNode::create(node_a.release_nonnull(), node_b.release_nonnull())); +} + ErrorOr> Parser::parse_dynamic_value(ComponentValue const& component_value) { if (component_value.is_function()) { @@ -3800,6 +3840,9 @@ ErrorOr> Parser::parse_a_calc_function_node(Function con if (function.name().equals_ignoring_ascii_case("atan"sv)) return TRY(parse_atan_function(function)); + if (function.name().equals_ignoring_ascii_case("atan2"sv)) + return TRY(parse_atan2_function(function)); + dbgln_if(CSS_PARSER_DEBUG, "We didn't implement `{}` function yet", function.name()); return nullptr; } diff --git a/Userland/Libraries/LibWeb/CSS/Parser/Parser.h b/Userland/Libraries/LibWeb/CSS/Parser/Parser.h index 241a40b983..b1dce14e31 100644 --- a/Userland/Libraries/LibWeb/CSS/Parser/Parser.h +++ b/Userland/Libraries/LibWeb/CSS/Parser/Parser.h @@ -301,6 +301,7 @@ private: ErrorOr> parse_asin_function(Function const&); ErrorOr> parse_acos_function(Function const&); ErrorOr> parse_atan_function(Function const&); + ErrorOr> parse_atan2_function(Function const&); ErrorOr> parse_dimension_value(ComponentValue const&); ErrorOr> parse_integer_value(TokenStream&); ErrorOr> parse_number_value(TokenStream&); diff --git a/Userland/Libraries/LibWeb/CSS/StyleValues/CalculatedStyleValue.cpp b/Userland/Libraries/LibWeb/CSS/StyleValues/CalculatedStyleValue.cpp index e8428a3fee..7d405e9fa0 100644 --- a/Userland/Libraries/LibWeb/CSS/StyleValues/CalculatedStyleValue.cpp +++ b/Userland/Libraries/LibWeb/CSS/StyleValues/CalculatedStyleValue.cpp @@ -1174,6 +1174,69 @@ ErrorOr AtanCalculationNode::dump(StringBuilder& builder, int indent) cons return {}; } +ErrorOr> Atan2CalculationNode::create(NonnullOwnPtr y, NonnullOwnPtr x) +{ + return adopt_nonnull_own_or_enomem(new (nothrow) Atan2CalculationNode(move(y), move(x))); +} + +Atan2CalculationNode::Atan2CalculationNode(NonnullOwnPtr y, NonnullOwnPtr x) + : CalculationNode(Type::Atan2) + , m_y(move(y)) + , m_x(move(x)) +{ +} + +Atan2CalculationNode::~Atan2CalculationNode() = default; + +ErrorOr Atan2CalculationNode::to_string() const +{ + StringBuilder builder; + builder.append("atan2("sv); + builder.append(TRY(m_y->to_string())); + builder.append(", "sv); + builder.append(TRY(m_x->to_string())); + builder.append(")"sv); + return builder.to_string(); +} + +Optional Atan2CalculationNode::resolved_type() const +{ + return CalculatedStyleValue::ResolvedType::Angle; +} + +bool Atan2CalculationNode::contains_percentage() const +{ + return m_y->contains_percentage() || m_x->contains_percentage(); +} + +CalculatedStyleValue::CalculationResult Atan2CalculationNode::resolve(Optional context, CalculatedStyleValue::PercentageBasis const& percentage_basis) const +{ + auto node_a = m_y->resolve(context, percentage_basis); + auto node_a_value = resolve_value(node_a.value(), context); + + auto node_b = m_x->resolve(context, percentage_basis); + auto node_b_value = resolve_value(node_b.value(), context); + + auto result = atan2(node_a_value, node_b_value); + + return { Angle(result, Angle::Type::Rad) }; +} + +ErrorOr Atan2CalculationNode::for_each_child_node(Function(NonnullOwnPtr&)> const& callback) +{ + TRY(m_y->for_each_child_node(callback)); + TRY(m_x->for_each_child_node(callback)); + TRY(callback(m_y)); + TRY(callback(m_x)); + return {}; +} + +ErrorOr Atan2CalculationNode::dump(StringBuilder& builder, int indent) const +{ + TRY(builder.try_appendff("{: >{}}ATAN2: {}\n", "", indent, TRY(to_string()))); + return {}; +} + void CalculatedStyleValue::CalculationResult::add(CalculationResult const& other, Optional context, PercentageBasis const& percentage_basis) { add_or_subtract_internal(SumOperation::Add, other, context, percentage_basis); diff --git a/Userland/Libraries/LibWeb/CSS/StyleValues/CalculatedStyleValue.h b/Userland/Libraries/LibWeb/CSS/StyleValues/CalculatedStyleValue.h index 43f2c732c3..22b58bef1a 100644 --- a/Userland/Libraries/LibWeb/CSS/StyleValues/CalculatedStyleValue.h +++ b/Userland/Libraries/LibWeb/CSS/StyleValues/CalculatedStyleValue.h @@ -504,4 +504,23 @@ private: NonnullOwnPtr m_value; }; +class Atan2CalculationNode final : public CalculationNode { +public: + static ErrorOr> create(NonnullOwnPtr, NonnullOwnPtr); + ~Atan2CalculationNode(); + + virtual ErrorOr to_string() const override; + virtual Optional resolved_type() const override; + virtual bool contains_percentage() const override; + virtual CalculatedStyleValue::CalculationResult resolve(Optional, CalculatedStyleValue::PercentageBasis const&) const override; + virtual ErrorOr for_each_child_node(Function(NonnullOwnPtr&)> const&) override; + + virtual ErrorOr dump(StringBuilder&, int indent) const override; + +private: + Atan2CalculationNode(NonnullOwnPtr, NonnullOwnPtr); + NonnullOwnPtr m_y; + NonnullOwnPtr m_x; +}; + }