include/llvm/CodeGen/SchedulerRegistry.h

   1 //===-- llvm/CodeGen/SchedulerRegistry.h ------------------------*- 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 // This file contains the implementation for instruction scheduler function
  11 // pass registry (RegisterScheduler).
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #ifndef LLVM_CODEGENSCHEDULERREGISTRY_H
  16 #define LLVM_CODEGENSCHEDULERREGISTRY_H
  17
  18 #include "llvm/CodeGen/MachinePassRegistry.h"
  19
  20 namespace llvm {
  21
  22 //===----------------------------------------------------------------------===//
  23 ///
  24 /// RegisterScheduler class - Track the registration of instruction schedulers.
  25 ///
  26 //===----------------------------------------------------------------------===//
  27
  28 class SelectionDAGISel;
  29 class ScheduleDAG;
  30 class SelectionDAG;
  31 class MachineBasicBlock;
  32
  33 class RegisterScheduler : public MachinePassRegistryNode {
  34
  35 public:
  36
  37   typedef ScheduleDAG *(*FunctionPassCtor)(SelectionDAGISel*, SelectionDAG*,
  38                                         const TargetMachine *,
  39                                         MachineBasicBlock*, bool);
  40
  41   static MachinePassRegistry Registry;
  42
  43   RegisterScheduler(const char *N, const char *D, FunctionPassCtor C)
  44   : MachinePassRegistryNode(N, D, (MachinePassCtor)C)
  45   { Registry.Add(this); }
  46   ~RegisterScheduler() { Registry.Remove(this); }
  47
  48
  49   // Accessors.
  50   //
  51   RegisterScheduler *getNext() const {
  52     return (RegisterScheduler *)MachinePassRegistryNode::getNext();
  53   }
  54   static RegisterScheduler *getList() {
  55     return (RegisterScheduler *)Registry.getList();
  56   }
  57   static FunctionPassCtor getDefault() {
  58     return (FunctionPassCtor)Registry.getDefault();
  59   }
  60   static void setDefault(FunctionPassCtor C) {
  61     Registry.setDefault((MachinePassCtor)C);
  62   }
  63   static void setListener(MachinePassRegistryListener *L) {
  64     Registry.setListener(L);
  65   }
  66
  67 };
  68
  69 } // end namespace llvm
  70
  71
  72 #endif