MathFunctionImpl< VectorWidth, MathOp::Atan2 > Struct Template Reference

#include <llvmexpr/codegen/llvm/Math.hpp>

Static Public Member Functions
static llvm::Function *	generate (MathFunctionGenerator< VectorWidth > *gen)

Detailed Description

template<int VectorWidth>
struct MathFunctionImpl< VectorWidth, MathOp::Atan2 >

Definition at line 643 of file Math.hpp.

Member Function Documentation

generate()

template<int VectorWidth>

llvm::Function * MathFunctionImpl< VectorWidth, MathOp::Atan2 >::generate ( MathFunctionGenerator< VectorWidth > * gen )

inlinestatic

Definition at line 644 of file Math.hpp.

                                                                           {
        constexpr auto OP_INFO = get_math_op_info(MathOp::Atan2);
        return gen->createFunction(
            OP_INFO.name, OP_INFO.arity,
            [gen](llvm::ArrayRef<llvm::Value*> args) -> llvm::Value* {
                auto* var_y = args[0];
                auto* var_x = args[1];
                auto* atan_func =
                    MathFunctionImpl<VectorWidth, MathOp::Atan>::generate(gen);
                auto* zero = gen->getConstant(0.0F);
                auto* pi = gen->getConstant(std::numbers::pi_v<float>);
                auto* pi_div_2 = gen->getConstant(
                    1.5707963267948966F); // NOLINT(cppcoreguidelines-avoid-magic-numbers)
                auto* y_div_x = gen->builder.CreateFDiv(var_y, var_x);
                auto* atan_y_div_x =
                    gen->builder.CreateCall(atan_func, {y_div_x});
                auto* res_x_gt_0 = atan_y_div_x;
                auto* signed_pi = gen->createIntrinsicCall(
                    llvm::Intrinsic::copysign, {pi, var_y});
                auto* res_x_lt_0 =
                    gen->builder.CreateFAdd(atan_y_div_x, signed_pi);
                auto* res_x_eq_0 = gen->createIntrinsicCall(
                    llvm::Intrinsic::copysign, {pi_div_2, var_y});
                auto* x_gt_0 = gen->builder.CreateFCmpOGT(var_x, zero);
                auto* x_lt_0 = gen->builder.CreateFCmpOLT(var_x, zero);
                auto* result =
                    gen->builder.CreateSelect(x_gt_0, res_x_gt_0, res_x_lt_0);
                result = gen->builder.CreateSelect(
                    x_lt_0, res_x_lt_0,
                    gen->builder.CreateSelect(x_gt_0, res_x_gt_0, res_x_eq_0));
                auto* x_is_zero = gen->builder.CreateFCmpOEQ(var_x, zero);
                auto* y_is_zero = gen->builder.CreateFCmpOEQ(var_y, zero);
                auto* both_zero = gen->builder.CreateAnd(x_is_zero, y_is_zero);
                result = gen->builder.CreateSelect(both_zero, zero, result);
                return result;
            });
    }

References Atan2, and get_math_op_info().

---

The documentation for this struct was generated from the following file:

• llvmexpr/codegen/llvm/Math.hpp