Make the type signature more strict.
[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   Constant *Data;
33   SmallVector<Constant*, 0> Vals;
34 public:
35   explicit AttributeImpl(LLVMContext &C, uint64_t data);
36   explicit AttributeImpl(LLVMContext &C, Attribute::AttrKind data);
37   AttributeImpl(LLVMContext &C, Attribute::AttrKind data,
38                 ArrayRef<Constant*> values);
39   AttributeImpl(LLVMContext &C, StringRef data);
40
41   ArrayRef<Constant*> getValues() const {
42     return Vals;
43   }
44
45   bool contains(Attribute::AttrKind Kind) const;
46   bool contains(StringRef Kind) const;
47
48   bool hasAttribute(Attribute::AttrKind A) const;
49
50   bool hasAttributes() const;
51   bool hasAttributes(const Attribute &A) const;
52
53   uint64_t getAlignment() const;
54   uint64_t getStackAlignment() const;
55
56   bool operator==(Attribute::AttrKind Kind) const {
57     return contains(Kind);
58   }
59   bool operator!=(Attribute::AttrKind Kind) const {
60     return !contains(Kind);
61   }
62
63   bool operator==(StringRef Kind) const {
64     return contains(Kind);
65   }
66   bool operator!=(StringRef Kind) const {
67     return !contains(Kind);
68   }
69
70   uint64_t getBitMask() const;         // FIXME: Remove.
71
72   static uint64_t getAttrMask(Attribute::AttrKind Val);
73
74   void Profile(FoldingSetNodeID &ID) const {
75     Profile(ID, Data, Vals);
76   }
77   static void Profile(FoldingSetNodeID &ID, Constant *Data,
78                       ArrayRef<Constant*> Vals) {
79     ID.AddPointer(Data);
80     for (ArrayRef<Constant*>::iterator I = Vals.begin(), E = Vals.end();
81          I != E; ++I)
82       ID.AddPointer(*I);
83   }
84 };
85
86 //===----------------------------------------------------------------------===//
87 /// \class
88 /// \brief This class represents a set of attributes.
89 class AttributeSetImpl : public FoldingSetNode {
90   // AttributesSet is uniqued, these should not be publicly available.
91   void operator=(const AttributeSetImpl &) LLVM_DELETED_FUNCTION;
92   AttributeSetImpl(const AttributeSetImpl &) LLVM_DELETED_FUNCTION;
93 public:
94   LLVMContext &Context;
95   SmallVector<AttributeWithIndex, 4> Attrs;
96
97   AttributeSetImpl(LLVMContext &C, ArrayRef<AttributeWithIndex> attrs)
98     : Context(C), Attrs(attrs.begin(), attrs.end()) {}
99
100   void Profile(FoldingSetNodeID &ID) const {
101     Profile(ID, Attrs);
102   }
103   static void Profile(FoldingSetNodeID &ID, ArrayRef<AttributeWithIndex> Attrs){
104     for (unsigned i = 0, e = Attrs.size(); i != e; ++i) {
105       ID.AddInteger(Attrs[i].Attrs.getBitMask());
106       ID.AddInteger(Attrs[i].Index);
107     }
108   }
109 };
110
111 } // end llvm namespace
112
113 #endif