// Copyright Advanced Micro Devices, Inc., or its affiliates.
// SPDX-License-Identifier: MIT

#include <rocRoller/CodeGen/Arithmetic/BitwiseNegate.hpp>
#include <rocRoller/Utilities/Component.hpp>

namespace rocRoller
{
    template <>
    std::shared_ptr<UnaryArithmeticGenerator<Expression::BitwiseNegate>>
        GetGenerator<Expression::BitwiseNegate>(Register::ValuePtr dst,
                                                Register::ValuePtr arg,
                                                Expression::BitwiseNegate const&)
    {
        return Component::Get<UnaryArithmeticGenerator<Expression::BitwiseNegate>>(
            getContextFromValues(dst, arg), dst->regType(), dst->variableType().dataType);
    }

    Generator<Instruction> BitwiseNegateGenerator::generate(Register::ValuePtr dest,
                                                            Register::ValuePtr arg,
                                                            Expression::BitwiseNegate const&)
    {
        AssertFatal(arg != nullptr);
        AssertFatal(dest != nullptr);

        auto elementBits = std::max(DataTypeInfo::Get(dest->variableType()).elementBits,
                                    DataTypeInfo::Get(arg->variableType()).elementBits);

        if(dest->regType() == Register::Type::Scalar)
        {
            if(elementBits <= 32u)
                co_yield_(Instruction("s_not_b32", {dest}, {arg}, {}, ""));
            else if(elementBits == 64u)
                co_yield_(Instruction("s_not_b64", {dest}, {arg}, {}, ""));
            else
                Throw<FatalError>("Unsupported elementBits for bitwiseNegate operation:: ",
                                  ShowValue(elementBits));
        }
        else if(dest->regType() == Register::Type::Vector)
        {
            if(elementBits <= 32u)
            {
                co_yield_(Instruction("v_not_b32", {dest}, {arg}, {}, ""));
            }
            else if(elementBits == 64u)
            {
                co_yield_(
                    Instruction("v_not_b32", {dest->subset({0})}, {arg->subset({0})}, {}, ""));
                co_yield_(
                    Instruction("v_not_b32", {dest->subset({1})}, {arg->subset({1})}, {}, ""));
            }
            else
            {
                Throw<FatalError>("Unsupported elementBits for bitwiseNegate operation:: ",
                                  ShowValue(elementBits));
            }
        }
        else
        {
            Throw<FatalError>("Unsupported register type for bitwiseNegate operation: ",
                              ShowValue(dest->regType()));
        }
    }
}