summaryrefslogtreecommitdiff
path: root/lib/CodeGen/SelectionDAG/SDNodeOrdering.h
blob: d2269f8accf1599391eecd71283564e201185ef1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
//===-- llvm/CodeGen/SDNodeOrdering.h - SDNode Ordering ---------*- C++ -*-===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
//
// This file declares the SDNodeOrdering class.
//
//===----------------------------------------------------------------------===//

#ifndef LLVM_CODEGEN_SDNODEORDERING_H
#define LLVM_CODEGEN_SDNODEORDERING_H

#include "llvm/ADT/DenseMap.h"

namespace llvm {

class SDNode;

/// SDNodeOrdering - Maps a unique (monotonically increasing) value to each
/// SDNode that roughly corresponds to the ordering of the original LLVM
/// instruction. This is used for turning off scheduling, because we'll forgo
/// the normal scheduling algorithms and output the instructions according to
/// this ordering.
class SDNodeOrdering {
  DenseMap<const SDNode*, unsigned> OrderMap;

  void operator=(const SDNodeOrdering&) LLVM_DELETED_FUNCTION;
  SDNodeOrdering(const SDNodeOrdering&) LLVM_DELETED_FUNCTION;
public:
  SDNodeOrdering() {}

  void add(const SDNode *Node, unsigned O) {
    OrderMap[Node] = O;
  }
  void remove(const SDNode *Node) {
    DenseMap<const SDNode*, unsigned>::iterator Itr = OrderMap.find(Node);
    if (Itr != OrderMap.end())
      OrderMap.erase(Itr);
  }
  void clear() {
    OrderMap.clear();
  }
  unsigned getOrder(const SDNode *Node) {
    return OrderMap[Node];
  }
};

} // end llvm namespace

#endif