//===-- attributes.cpp ----------------------------------------------------===//
//
//                         LDC – the LLVM D compiler
//
// This file is distributed under the BSD-style LDC license. See the LICENSE
// file for details.
//
//===----------------------------------------------------------------------===//

#include "gen/attributes.h"
#include "gen/irstate.h"

bool AttrBuilder::hasAttributes() const
{
    return builder.hasAttributes();
}

bool AttrBuilder::contains(LLAttribute attribute) const
{
    return builder.contains(attribute);
}

AttrBuilder& AttrBuilder::clear()
{
    builder.clear();
    return *this;
}

AttrBuilder& AttrBuilder::add(LLAttribute attribute)
{
    // never set 'None' explicitly
    if (attribute)
        builder.addAttribute(attribute);
    return *this;
}

AttrBuilder& AttrBuilder::remove(LLAttribute attribute)
{
    // never remove 'None' explicitly
    if (attribute)
        builder.removeAttribute(attribute);
    return *this;
}

AttrBuilder& AttrBuilder::merge(const AttrBuilder& other)
{
    builder.merge(other.builder);
    return *this;
}


AttrSet AttrSet::extractFunctionAndReturnAttributes(const llvm::Function* function)
{
    AttrSet r;

    llvm::AttributeSet old = function->getAttributes();
    llvm::AttributeSet existingAttrs[] = { old.getFnAttributes(), old.getRetAttributes() };
    r.set = llvm::AttributeSet::get(gIR->context(), existingAttrs);

    return r;
}

AttrSet& AttrSet::add(unsigned index, const AttrBuilder& builder)
{
    if (builder.hasAttributes())
    {
        llvm::AttributeSet as = llvm::AttributeSet::get(
            gIR->context(), index, builder);
        set = set.addAttributes(gIR->context(), index, as);
    }
    return *this;
}