Add a method to create an AttributeSet from an AttrBuilder.
[oota-llvm.git] / lib / IR / AttributeImpl.h
1 //===-- AttributeImpl.h - Attribute Internals -------------------*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 ///
10 /// \file
11 /// \brief This file defines various helper methods and classes used by
12 /// LLVMContextImpl for creating and managing attributes.
13 ///
14 //===----------------------------------------------------------------------===//
15
16 #ifndef LLVM_ATTRIBUTESIMPL_H
17 #define LLVM_ATTRIBUTESIMPL_H
18
19 #include "llvm/ADT/FoldingSet.h"
20 #include "llvm/IR/Attributes.h"
21
22 namespace llvm {
23
24 class Constant;
25 class LLVMContext;
26
27 //===----------------------------------------------------------------------===//
28 /// \class
29 /// \brief This class represents a single, uniqued attribute. That attribute
30 /// could be a single enum, a tuple, or a string.
31 class AttributeImpl : public FoldingSetNode {
32   LLVMContext &Context;
33   Constant *Data;
34   SmallVector<Constant*, 0> Vals;
35 public:
36   explicit AttributeImpl(LLVMContext &C, uint64_t data);
37   explicit AttributeImpl(LLVMContext &C, Attribute::AttrKind data);
38   AttributeImpl(LLVMContext &C, Attribute::AttrKind data,
39                 ArrayRef<Constant*> values);
40   AttributeImpl(LLVMContext &C, StringRef data);
41
42   ArrayRef<Constant*> getValues() const {
43     return Vals;
44   }
45
46   bool hasAttribute(Attribute::AttrKind A) const;
47
48   bool hasAttributes() const;
49
50   uint64_t getAlignment() const;
51   void setAlignment(unsigned Align);
52
53   uint64_t getStackAlignment() const;
54   void setStackAlignment(unsigned Align);
55
56   bool operator==(Attribute::AttrKind Kind) const;
57   bool operator!=(Attribute::AttrKind Kind) const;
58
59   bool operator==(StringRef Kind) const;
60   bool operator!=(StringRef Kind) const;
61
62   uint64_t getBitMask() const;         // FIXME: Remove.
63
64   static uint64_t getAttrMask(Attribute::AttrKind Val);
65
66   void Profile(FoldingSetNodeID &ID) const {
67     Profile(ID, Data, Vals);
68   }
69   static void Profile(FoldingSetNodeID &ID, Constant *Data,
70                       ArrayRef<Constant*> Vals) {
71     ID.AddPointer(Data);
72     for (ArrayRef<Constant*>::iterator I = Vals.begin(), E = Vals.end();
73          I != E; ++I)
74       ID.AddPointer(*I);
75   }
76 };
77
78 //===----------------------------------------------------------------------===//
79 /// \class
80 /// \brief This class represents a set of attributes.
81 class AttributeSetImpl : public FoldingSetNode {
82   LLVMContext &Context;
83   SmallVector<AttributeWithIndex, 4> AttrList;
84
85   // AttributesSet is uniqued, these should not be publicly available.
86   void operator=(const AttributeSetImpl &) LLVM_DELETED_FUNCTION;
87   AttributeSetImpl(const AttributeSetImpl &) LLVM_DELETED_FUNCTION;
88 public:
89   AttributeSetImpl(LLVMContext &C, ArrayRef<AttributeWithIndex> attrs)
90     : Context(C), AttrList(attrs.begin(), attrs.end()) {}
91
92   LLVMContext &getContext() { return Context; }
93   ArrayRef<AttributeWithIndex> getAttributes() const { return AttrList; }
94   unsigned getNumAttributes() const { return AttrList.size(); }
95
96   void Profile(FoldingSetNodeID &ID) const {
97     Profile(ID, AttrList);
98   }
99   static void Profile(FoldingSetNodeID &ID,
100                       ArrayRef<AttributeWithIndex> AttrList){
101     for (unsigned i = 0, e = AttrList.size(); i != e; ++i) {
102       ID.AddInteger(AttrList[i].Index);
103       ID.AddInteger(AttrList[i].Attrs.getBitMask());
104     }
105   }
106 };
107
108 } // end llvm namespace
109
110 #endif